PWGLF/RESONANCES/AliRsnCutDaughterD0.cxx

   1 //
   2 // All cuts for single track in D0 analysis,
   3 // based on track quality and particle identification
   4 // with TPC and TOF.
   5 // Author: Massimo Venaruzzo
   6 //
   7 //
   8
   9 #include <Riostream.h>
  10
  11 #include "AliPID.h"
  12 #include "AliPIDResponse.h"
  13 #include "AliRsnCutDaughterD0.h"
  14
  15 ClassImp(AliRsnCutDaughterD0)
  16
  17 //__________________________________________________________________________________________________
  18 AliRsnCutDaughterD0::AliRsnCutDaughterD0(const char *name, AliPID::EParticleType pid) :
  19 AliRsnCut(name, AliRsnTarget::kDaughter),
  20   fNoPID(kFALSE),
  21   fPID(pid),
  22   fCutQuality(Form("%sQuality", name)),
  23   fPionTPCPIDCut(3.0),
  24   fKaonTPCPIDCut(3.0),
  25   fPionTOFPIDCut(3.0),
  26   fKaonTOFPIDCut(3.0),
  27   fPtDepPIDCut(kFALSE)
  28 {
  29   //
  30   // Constructor
  31   // Initialize track quality cuts to 2010 defaults
  32   //
  33   fCutQuality.SetPtRange(0.15, 1E+20);
  34   fCutQuality.SetEtaRange(-0.8, 0.8);
  35   fCutQuality.SetDCARPtFormula("0.0105+0.0350/pt^1.1");
  36   fCutQuality.SetDCARmin(0.0);
  37   fCutQuality.SetDCAZmax(2.0);
  38   fCutQuality.SetSPDminNClusters(1);
  39   fCutQuality.SetITSminNClusters(0);
  40   fCutQuality.SetITSmaxChi2(1E+20);
  41   fCutQuality.SetTPCminNClusters(70);
  42   fCutQuality.SetTPCmaxChi2(4.0);
  43   fCutQuality.SetRejectKinkDaughters();
  44   fCutQuality.SetAODTestFilterBit(5);
  45 }
  46
  47 //__________________________________________________________________________________________________
  48 Bool_t AliRsnCutDaughterD0::IsSelected(TObject *obj)
  49 {
  50   //
  51   // Global check
  52   //
  53
  54   // coherence check
  55   if (!TargetOK(obj)) return kFALSE;
  56
  57   // check track
  58   AliVTrack *track = dynamic_cast<AliVTrack *>(fDaughter->GetRef());
  59   if (!track) return kFALSE;
  60
  61   AliDebugClass(2, "Checking status...");
  62   // check flags
  63   if ((track->GetStatus() & AliESDtrack::kTPCin   ) == 0) return kFALSE;
  64   if ((track->GetStatus() & AliESDtrack::kTPCrefit) == 0) return kFALSE;
  65   if ((track->GetStatus() & AliESDtrack::kITSrefit) == 0) return kFALSE;
  66   AliDebugClass(2, "...passed");
  67
  68   // quality
  69   AliDebugClass(2, "Checking quality cuts...");
  70   if (!fCutQuality.IsSelected(obj)) return kFALSE;
  71   AliDebugClass(2, "...passed");
  72
  73   // if no PID is required, accept
  74   if (fNoPID) return kTRUE;
  75
  76   // check initialization of PID object
  77   AliPIDResponse *pid = fEvent->GetPIDResponse();
  78   if (!pid) {
  79     AliFatal("NULL PID response");
  80     return kFALSE;
  81   }
  82
  83   AliDebugClass(2, "Checking TOF Matching...");
  84   // check if TOF is matched
  85   // and computes all values used in the PID cut
  86   Bool_t   isTOF  = MatchTOF(track);
  87   AliDebugClass(2, "...passed");
  88
  89   Double_t pTPC   = track->GetTPCmomentum();
  90   Double_t p      = track->P();
  91   Double_t nsTPC  = TMath::Abs(pid->NumberOfSigmasTPC(track, fPID));
  92   Double_t nsTOF  = isTOF ? TMath::Abs(pid->NumberOfSigmasTOF(track, fPID)) : 1E20;
  93   Double_t maxTPC = 1E20;
  94   Double_t maxTOF = 1E20;
  95   AliDebugClass(2, "Checking PID...");
  96
  97   if(!fPtDepPIDCut){
  98     // applies the cut differently depending on the PID and the momentum
  99     if (isTOF) {
 100       if (fPID == AliPID::kPion) {maxTPC = fPionTPCPIDCut; maxTOF = fPionTOFPIDCut;}
 101       if (fPID == AliPID::kKaon) {maxTPC = fKaonTPCPIDCut; maxTOF = fKaonTOFPIDCut;}
 102       return (nsTPC <= maxTPC && nsTOF <= maxTOF);
 103     } else {
 104       if (fPID == AliPID::kPion) maxTPC = fPionTPCPIDCut;
 105       if (fPID == AliPID::kKaon) maxTPC = fKaonTPCPIDCut;
 106       return (nsTPC <= maxTPC);
 107     }
 108   } else {
 109     // applies the cut differently depending on the PID and the momentum
 110     if (isTOF) {
 111       // TPC: 5sigma cut for all
 112       if (nsTPC > 5.0) return kFALSE;
 113       // TOF: 3sigma below 1.5 GeV, 2sigma above
 114       if (p < 1.5) maxTOF = 3.0; else maxTOF = 2.0;
 115       return (nsTOF <= maxTOF);
 116     } else {
 117       // TPC:
 118       // all   below   350         MeV: 5sigma
 119       // all   between 350 and 500 MeV: 3sigma
 120       // pions above   500         MeV: 2sigma
 121       // kaons between 500 and 700 MeV: 2sigma
 122       // kaons above   700         MeV: rejected
 123       if (pTPC <= 0.35)
 124         maxTPC = 5.0;
 125       else if (pTPC > 0.35 && pTPC <= 0.5)
 126         maxTPC = 3.0;
 127       else {
 128         if (fPID == AliPID::kPion)
 129           maxTPC = 2.0;
 130         else if (fPID == AliPID::kKaon) {
 131           if (pTPC <= 0.7)
 132             maxTPC = 2.0;
 133           else
 134             return kFALSE;
 135         }
 136       }
 137       return (nsTPC <= maxTPC);
 138     }
 139   }
 140
 141   AliDebugClass(2, "...passed");
 142   // if we reach this point, all checks were successful
 143   AliDebugClass(2, "Good Pion/Kaon Candidate Found!!");
 144 }