ANALYSIS/AliESDpidCuts.h

   1 #ifndef ALIESDPIDCUTS_H\r
   2 #define ALIESDPIDCUTS_H\r
   3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *\r
   4  * See cxx source for full Copyright notice                               */\r
   5 //\r
   6 // Class for PID cuts\r
   7 // Cuts the track based on numbers of sigmas in the detectors TPC and TOF\r
   8 // The sigma cuts can be applied symmetrically or assymetrically\r
   9 //\r
  10 \r
  11 #ifndef ALIANALYSISCUTS_H\r
  12 #include "AliAnalysisCuts.h"\r
  13 #endif\r
  14 \r
  15 #ifndef ALIPID_H\r
  16 #include "AliPID.h"\r
  17 #endif\r
  18 \r
  19 class TCollection;\r
  20 class TH1F;\r
  21 class TH1I;\r
  22 class TH2I;\r
  23 class TList;\r
  24 class AliESDtrack;\r
  25 class AliESDEvent;\r
  26 class AliESDpid;\r
  27 \r
  28 class AliESDpidCuts : public AliAnalysisCuts{\r
  29   enum{\r
  30     kHasHistograms = BIT(17)\r
  31   };\r
  32   public: \r
  33     AliESDpidCuts(const Char_t *name = "AliESDpidCuts", const Char_t *title = "");\r
  34     AliESDpidCuts(const AliESDpidCuts &ref);  // Copy constructor\r
  35     AliESDpidCuts &operator=(const AliESDpidCuts &ref);\r
  36     virtual ~AliESDpidCuts();\r
  37 \r
  38     virtual void Init();\r
  39     virtual void Copy(TObject &c) const;\r
  40     virtual Long64_t Merge(TCollection *coll);\r
  41 \r
  42     Bool_t HasHistograms() const { return TestBit(kHasHistograms); }\r
  43     void DefineHistograms(Color_t color = kRed);\r
  44     void DrawHistograms();\r
  45     void SaveHistograms(const Char_t *location = NULL);\r
  46     virtual Bool_t IsSelected(TObject *);\r
  47     virtual Bool_t IsSelected(TList * /*lst*/) {return kTRUE; }\r
  48     virtual Bool_t AcceptTrack(const AliESDtrack *track, const AliESDEvent *event);\r
  49 \r
  50     void SetPIDResponse(AliPIDResponse * pidresponse) { fPIDresponse = pidresponse; }\r
  51     AliPIDResponse *GetPIDresponse() { return fPIDresponse; };\r
  52     \r
  53     void SetTPCclusterRatioCut(Float_t clr) { fCutTPCclusterRatio = clr; }\r
  54     inline void SetTPCnSigmaCut(AliPID::EParticleType itype, Float_t nSigma);\r
  55     inline void SetTPCnSigmaCut(AliPID::EParticleType itype, Float_t negSigma, Float_t posSigma);\r
  56     inline void SetTOFnSigmaCut(AliPID::EParticleType itype, Float_t nSigma);\r
  57     inline void SetTOFnSigmaCut(AliPID::EParticleType itype, Float_t negSigma, Float_t posSigma);\r
  58     void SetMinMomentumTOF(Float_t mom) { fMinMomentumTOF = mom; }\r
  59   \r
  60   protected:\r
  61     static const Int_t kNcuts;                      // Number of Cuts\r
  62     AliPIDResponse *fPIDresponse;                   //! PID helper (n-sigma-cut)\r
  63     Char_t  fTPCsigmaCutRequired;                   // Sigma cut Requirement for TPC and Particle Species\r
  64     Char_t  fTOFsigmaCutRequired;                   // Sigma cut Requirement for TOF and Particle Species\r
  65     Float_t fCutTPCnSigma[AliPID::kSPECIES * 2];    // Species dependent cut on the distance to the TPC dE/dx line\r
  66     Float_t fCutTOFnSigma[AliPID::kSPECIES * 2];    // Species dependent cut on the distance to the TOF calculated time of flight line\r
  67     Float_t fCutTPCclusterRatio;                    // Cut on Ratio of found clusters with repect to findable clusters in the TPC\r
  68     Float_t fMinMomentumTOF;                        // Apply TOF PID only above a certain momentum\r
  69 \r
  70     //------------------------------------------\r
  71     // QA histograms\r
  72     TH1I *fHcutStatistics;                       // Cut Statistics\r
  73     TH2I *fHcutCorrelation;                      // Cut Correlation\r
  74     TH1F *fHclusterRatio[2];                     // TPC cluster Ratio\r
  75     TH1F *fHnSigmaTPC[AliPID::kSPECIES][2];      // TPC n-sigma cut\r
  76     TH1F *fHnSigmaTOF[AliPID::kSPECIES][2];      // TOF n-sigma cut\r
  77     //------------------------------------------\r
  78     \r
  79     ClassDef(AliESDpidCuts, 3)\r
  80 };\r
  81 \r
  82 //_____________________________________________________________________\r
  83 void AliESDpidCuts::SetTPCnSigmaCut(AliPID::EParticleType itype, Float_t nSigma){ \r
  84   //\r
  85   // symmetric sigma cut for TPC PID\r
  86   //\r
  87   fCutTPCnSigma[itype * 2]      = -nSigma;\r
  88   fCutTPCnSigma[itype * 2 + 1]  = nSigma; \r
  89   fTPCsigmaCutRequired |= 1 << static_cast<Int_t >(itype);\r
  90 }    \r
  91 \r
  92 //_____________________________________________________________________\r
  93 void AliESDpidCuts::SetTPCnSigmaCut(AliPID::EParticleType itype, Float_t negSigma, Float_t posSigma){\r
  94   //\r
  95   // assymetric sigma cut for TPC PID\r
  96   //\r
  97   fCutTPCnSigma[itype * 2]      = negSigma;\r
  98   fCutTPCnSigma[itype * 2 + 1]  = posSigma;\r
  99   fTPCsigmaCutRequired |= 1 << static_cast<Int_t >(itype);\r
 100 } \r
 101 \r
 102 //_____________________________________________________________________\r
 103 void AliESDpidCuts::SetTOFnSigmaCut(AliPID::EParticleType itype, Float_t nSigma){ \r
 104   //\r
 105   // symmetric sigma cut for TOF PID\r
 106   //\r
 107   fCutTOFnSigma[itype * 2]      = -nSigma;\r
 108   fCutTOFnSigma[itype * 2 + 1]  = nSigma; \r
 109   fTOFsigmaCutRequired |= 1 << static_cast<Int_t >(itype);\r
 110 }    \r
 111 \r
 112 //_____________________________________________________________________\r
 113 void AliESDpidCuts::SetTOFnSigmaCut(AliPID::EParticleType itype, Float_t negSigma, Float_t posSigma){\r
 114   //\r
 115   // assymetric sigma cut for TOF PID\r
 116   //\r
 117   fCutTOFnSigma[itype * 2]      = negSigma;\r
 118   fCutTOFnSigma[itype * 2 + 1]  = posSigma;\r
 119   fTOFsigmaCutRequired |= 1 << static_cast<Int_t >(itype);\r
 120 }\r
 121 #endif\r