PWG2/RESONANCES/AliRsnCutPIDITS.h

   1 //
   2 // *** Class AliRsnCutPIDITS ***
   3 //
   4 // This class implements all cuts which have to be used for the 2010 runs
   5 // for phi and generic resonance analysis.
   6 // It contains an AliESDtrackCuts object for track quality selection
   7 // and some criteria for particle identification with ITS, ITS and TOF.
   8 //
   9 // authors: Martin Vala (martin.vala@cern.ch)
  10 //          Alberto Pulvirenti (alberto.pulvirenti@ct.infn.it)
  11 //
  12
  13 #ifndef ALIRSNCUTPIDITS_H
  14 #define ALIRSNCUTPIDITS_H
  15
  16 #include "AliPID.h"
  17
  18 #include "AliESDtrack.h"
  19 #include "AliESDtrackCuts.h"
  20 #include "AliESDpid.h"
  21
  22 #include "AliAODpidUtil.h"
  23
  24 #include "AliRsnDaughter.h"
  25 #include "AliRsnCut.h"
  26
  27 class AliRsnCutPIDITS : public AliRsnCut
  28 {
  29   public:
  30
  31     AliRsnCutPIDITS(const char *name          = "cutITS",
  32                     EPARTYPE    ref           = AliPID::kKaon,
  33                     Double_t    nSigmaMin     = -3.,
  34                     Double_t    nSigmaMax     =  3.,
  35                     Bool_t      rejectOutside = kTRUE);
  36
  37     AliRsnCutPIDITS(const AliRsnCutPIDITS& copy);
  38     AliRsnCutPIDITS& operator=(const AliRsnCutPIDITS& copy);
  39     virtual ~AliRsnCutPIDITS() { }
  40
  41     AliESDpid*       ESDpid()  {return &fESDpid;}
  42     AliAODpidUtil*   AODpid()  {return &fAODpid;}
  43
  44     void             SetMC(Bool_t mc = kTRUE);
  45     void             SetRejectOutside(Bool_t yn = kTRUE)           {fRejectOutside = yn;}
  46     void             SetMomentumRange(Double_t min, Double_t max)  {fMomMin = min; fMomMax = max;}
  47     void             SetNSigmaRange(Double_t min, Double_t max)    {fMinD = min; fMaxD = max;}
  48     void             SetRefType(EPARTYPE type)                     {fRefType = type;}
  49
  50     Bool_t           IsTPC(AliVTrack *vtrack);
  51     Bool_t           IsSA(AliVTrack *vtrack);
  52     virtual Bool_t   IsSelected(TObject *object);
  53     virtual void     Print(const Option_t *option = "") const;
  54
  55   private:
  56
  57     void Initialize();
  58
  59     Bool_t          fInitialized;    // a mono-usage flag which initializes the ESD pid object
  60     Bool_t          fRejectOutside;  // choose if tracks outside momentum range are rejected or not
  61     Double_t        fMomMin;         // min p in range where this cut is checked
  62     Double_t        fMomMax;         // max p in range where this cut is checked
  63     EPARTYPE        fRefType;        // particle type for which PID is checked
  64     AliESDpid       fESDpid;         // ESD PID object
  65     AliAODpidUtil   fAODpid;         // AOD PID object
  66
  67     ClassDef(AliRsnCutPIDITS, 1)
  68 };
  69
  70 inline Bool_t AliRsnCutPIDITS::IsTPC(AliVTrack *vtrack)
  71 {
  72 //
  73 // Checks if the track has the status flags required for a global track
  74 //
  75
  76   if (!vtrack)
  77   {
  78     AliWarning("NULL argument: impossible to check status");
  79     return kFALSE;
  80   }
  81
  82   Bool_t isTPCin     = ((vtrack->GetStatus() & AliESDtrack::kTPCin) != 0);
  83   Bool_t isITSrefit  = ((vtrack->GetStatus() & AliESDtrack::kITSrefit) != 0);
  84   Bool_t isITSpid    = ((vtrack->GetStatus() & AliESDtrack::kITSpid) != 0);
  85
  86   return ( isTPCin && isITSrefit && isITSpid );
  87
  88   return kTRUE;
  89 }
  90
  91 inline Bool_t AliRsnCutPIDITS::IsSA(AliVTrack *vtrack)
  92 {
  93 //
  94 // Checks if the track has the status flags required for an ITS standalone track
  95 //
  96
  97   if (!vtrack)
  98   {
  99     AliWarning("NULL argument: impossible to check status");
 100     return kFALSE;
 101   }
 102
 103   Bool_t isTPCin     = ((vtrack->GetStatus() & AliESDtrack::kTPCin) != 0);
 104   Bool_t isITSrefit  = ((vtrack->GetStatus() & AliESDtrack::kITSrefit) != 0);
 105   Bool_t isITSpureSA = ((vtrack->GetStatus() & AliESDtrack::kITSpureSA) != 0);
 106   Bool_t isITSpid    = ((vtrack->GetStatus() & AliESDtrack::kITSpid) != 0);
 107
 108   return ( (!isTPCin) && isITSrefit && (!isITSpureSA) && isITSpid );
 109
 110   return kTRUE;
 111 }
 112
 113 #endif