ANALYSIS/AliESDpidCuts.h

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