Additional changes for HF-MUON

[u/mrichter/AliRoot.git] / PWGPP / MUON / dep / AliAnalysisTaskMuonTrackingEff.h

diff --git a/PWGPP/MUON/dep/AliAnalysisTaskMuonTrackingEff.h b/PWGPP/MUON/dep/AliAnalysisTaskMuonTrackingEff.h
new file mode 100644 (file)
index 0000000..ed555ee
--- /dev/null
+++ b/PWGPP/MUON/dep/AliAnalysisTaskMuonTrackingEff.h
@@ -0,0 +1,118 @@
+#ifndef ALIANALYSISTASKMUONTRACKINGEFF_H
+#define ALIANALYSISTASKMUONTRACKINGEFF_H
+/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
+ * See cxx source for full Copyright notice                               */
+
+/* $Id$ */ 
+
+/// \ingroup base
+/// \class AliAnalysisTaskMuonTrackingEff
+/// \brief tracking chamber efficiency from ESD data
+//Author: Nicolas LE BRIS - SUBATECH Nantes
+
+#include "AliAnalysisTaskSE.h"
+
+class AliMUONGeometryTransformer;
+class AliESDMuonTrack;
+class AliMUONTrackParam;
+class TList;
+class TString;
+class TObjArray;
+
+class AliAnalysisTaskMuonTrackingEff : public AliAnalysisTaskSE
+{
+ public:
+  AliAnalysisTaskMuonTrackingEff();
+  AliAnalysisTaskMuonTrackingEff(TString name);
+  virtual ~AliAnalysisTaskMuonTrackingEff();
+
+  /// Set location of the default OCDB storage (if not set use "raw://")
+  void SetDefaultStorage(const char* ocdbPath) { fOCDBpath = ocdbPath; }
+  
+  /// set the flag to use only tracks matched with trigger or not
+  void MatchTrigger(Bool_t flag = kTRUE) { fMatchTrig = flag; }
+  
+  /// set the flag to use only tracks passing the acceptance cuts (Rabs, eta)
+  void ApplyAccCut(Bool_t flag = kTRUE) { fApplyAccCut = flag; }
+  
+  /// set the sigma cut value on p*DCA
+  void PDCACut(Double_t cut) { fPDCACut = cut; }
+  
+  /// set the cut value on normalized chi2
+  void Chi2Cut(Double_t cut) { fChi2Cut = cut; }
+  
+  /// set the cut value on minimum pt
+  void PtCut(Double_t cut) { fPtCut = cut; }
+  
+  /// set the cut value on minimum pt
+  void UseMCLabel(Bool_t flag = kTRUE) { fUseMCLabel = flag; }
+  
+  // Implementation of interface methods
+  virtual void   UserCreateOutputObjects();
+  virtual void   UserExec(Option_t *);
+  virtual void   NotifyRun();
+  virtual void   Terminate(Option_t *);
+  
+  
+ private:
+  
+  /// Not implemented
+  AliAnalysisTaskMuonTrackingEff(const AliAnalysisTaskMuonTrackingEff& rhs);
+  /// Not implemented
+  AliAnalysisTaskMuonTrackingEff& operator = (const AliAnalysisTaskMuonTrackingEff& rhs);
+  
+  void TrackParamLoop(const TObjArray* trackParams);
+  
+  void FindAndFillMissedDetElt (const AliMUONTrackParam* trackParam,
+                               const Bool_t* trackFilter,
+                               Int_t firstMissCh, Int_t lastChamber);
+  
+  void CoordinatesOfMissingCluster(Double_t x1, Double_t y1, Double_t z1,
+                                  Double_t x2, Double_t y2, Double_t z2,
+                                  Double_t& x, Double_t& y) const;
+  
+  Bool_t CoordinatesInDetElt(Int_t DeId, Double_t x, Double_t y) const;
+  
+  void FillTDHistos (Int_t chamber, Int_t detElt, Double_t posXL, Double_t posYL);
+  
+  void FillTTHistos (Int_t chamber, Int_t detElt, Double_t posXL, Double_t posYL);
+
+  void FillSDHistos (Int_t chamber, Int_t detElt, Double_t posXL, Double_t posYL);
+  
+  Int_t FromDetElt2iDet (Int_t chamber, Int_t detElt) const;
+  Int_t FromDetElt2LocalId (Int_t chamber, Int_t detElt) const;
+  Int_t FromLocalId2DetElt(Int_t chamber, Int_t iDet) const;
+  
+  
+private:
+  
+  static const Int_t fgkNbrOfDetectionElt[10]; ///< The total number of detection element in each chamber.
+  static const Int_t fgkOffset;                ///< fFirstDetectionElt[iChamber] = fOffset * (iChamber+1).
+  
+  Bool_t   fOCDBLoaded;           //!< Determine if the OCDB and =geometry have been loaded
+  TString  fOCDBpath;             ///< OCDB path
+  Bool_t   fMatchTrig;            ///< use only tracks matched with trigger
+  Bool_t   fApplyAccCut;          ///< use only tracks passing the acceptance cuts (Rabs, eta)
+  Double_t fPDCACut;              ///< sigma cut on p*DCA
+  Double_t fChi2Cut;              ///< cut on normalized chi2
+  Double_t fPtCut;                ///< cut on minimum pt
+  Bool_t   fUseMCLabel;           ///< select tracks using MC label
+  Float_t  fCurrentCentrality;    //!< centrality of the current event
+  AliESDMuonTrack* fCurrentTrack; //!< pointer to the currently analyzed track
+
+  AliMUONGeometryTransformer *fTransformer; //!< Transformer object
+
+  TList* fDetEltTDHistList;  //!< List of histograms of the tracks detected in the detection elements. 
+  TList* fDetEltTTHistList;  //!< List of histograms of the tracks which have passed through the detection elements. 
+  TList* fDetEltSDHistList;  //!< List of histograms of the tracks only detected by one chamber of the station at the detection element level.
+  TList* fChamberTDHistList; //!< List of histograms of the tracks detected in the chambers.
+  TList* fChamberTTHistList; //!< List of histograms of the tracks which have passed through the chambers.
+  TList* fChamberSDHistList; //!< List of histograms of the tracks only detected by one chamber of the station.
+  TList* fExtraHistList;     //!< List of extra histograms.
+
+  
+  ClassDef(AliAnalysisTaskMuonTrackingEff, 4)
+};
+
+#endif
+