PWG2/RESONANCES/AliRsnCutPIDTOF.cxx

   1 //
   2 // Class AliRsnCutPIDTOF
   3 //
   4 // Implements the PID check with TOF detector,
   5 // computed as a compatibility within a given range
   6 // expressed in number of sigmas w.r. to expected time.
   7 // Uses the default cut checking facilities of AliRsnCut
   8 // to check this.
   9 //
  10 // authors: Martin Vala (martin.vala@cern.ch)
  11 //          Alberto Pulvirenti (alberto.pulvirenti@ct.infn.it)
  12 //
  13
  14 #include "AliAnalysisManager.h"
  15 #include "AliESDInputHandler.h"
  16
  17 #include "AliRsnCutPIDTOF.h"
  18
  19 ClassImp(AliRsnCutPIDTOF)
  20
  21 //_________________________________________________________________________________________________
  22 AliRsnCutPIDTOF::AliRsnCutPIDTOF
  23 (const char *name, AliPID::EParticleType ref, Double_t min, Double_t max, Bool_t rejectUnmatched) :
  24   AliRsnCut(name, AliRsnCut::kDaughter, min, max),
  25   fInitialized(kFALSE),
  26   fRejectUnmatched(rejectUnmatched),
  27   fRefType(AliPID::kUnknown),
  28   fRefMass(0.0),
  29   fESDpid(),
  30   fAODpid()
  31 {
  32 //
  33 // Default constructor.
  34 // To set the reference PID type, calls the SetRefType method,
  35 // which sets the mass accordingly and coherently.
  36 //
  37
  38   SetRefType(ref);
  39 }
  40
  41 //_________________________________________________________________________________________________
  42 AliRsnCutPIDTOF::AliRsnCutPIDTOF(const AliRsnCutPIDTOF& copy) :
  43   AliRsnCut(copy),
  44   fInitialized(kFALSE),
  45   fRejectUnmatched(copy.fRejectUnmatched),
  46   fRefType(AliPID::kUnknown),
  47   fRefMass(0.0),
  48   fESDpid(copy.fESDpid),
  49   fAODpid(copy.fAODpid)
  50 {
  51 //
  52 // Copy constructor.
  53 // To set the reference PID type, calls the SetRefType method,
  54 // which sets the mass accordingly and coherently.
  55 //
  56
  57   SetRefType(copy.fRefType);
  58 }
  59
  60 //_________________________________________________________________________________________________
  61 AliRsnCutPIDTOF& AliRsnCutPIDTOF::operator=(const AliRsnCutPIDTOF& copy)
  62 {
  63 //
  64 // Assignment operator.
  65 // To set the reference PID type, calls the SetRefType method,
  66 // which sets the mass accordingly and coherently.
  67 //
  68
  69   fInitialized     = kFALSE;
  70   fRejectUnmatched = copy.fRejectUnmatched;
  71   fESDpid          = copy.fESDpid;
  72   fAODpid          = copy.fAODpid;
  73
  74   SetRefType(copy.fRefType);
  75
  76   return (*this);
  77 }
  78
  79 //_________________________________________________________________________________________________
  80 Bool_t AliRsnCutPIDTOF::IsSelected(TObject *object)
  81 {
  82 //
  83 // Cut checker.
  84 //
  85
  86   // initialize if needed
  87   if (!fInitialized) Initialize();
  88
  89   // coherence check
  90   if (!TargetOK(object)) return kFALSE;
  91
  92   // reject always non-track objects
  93   AliVTrack *vtrack = dynamic_cast<AliVTrack*>(fDaughter->GetRef());
  94   if (!vtrack)
  95   {
  96     AliDebug(AliLog::kDebug + 2, Form("Impossible to process an object of type '%s'. Cut applicable only to ESD/AOD tracks", fDaughter->GetRef()->ClassName()));
  97     return kFALSE;
  98   }
  99
 100   // checks that track is matched in TOF:
 101   // if not, the track is accepted or rejected
 102   // depending on the 'fRejectUnmatched' data member:
 103   // -- kTRUE  --> all unmatched tracks are rejected
 104   // -- kFALSE --> all unmatched tracks are accepted (it is assumed that other PIDs are done)
 105   if (!IsMatched(vtrack))
 106   {
 107     AliDebug(AliLog::kDebug + 2, "Track is not matched with TOF");
 108     return (!fRejectUnmatched);
 109   }
 110
 111   // retrieve real object type and
 112   // prepare some useful variables
 113   Double_t     tof, sigma, times[5];
 114   Double_t    &ref = times[(Int_t)fRefType];
 115   AliESDtrack *esdTrack = fDaughter->GetRefESDtrack();
 116   AliAODTrack *aodTrack = fDaughter->GetRefAODtrack();
 117
 118   // cut check depends on the object type
 119   if (esdTrack)
 120   {
 121     // setup the ESD PID object
 122     AliESDEvent *esd = AliRsnTarget::GetCurrentEvent()->GetRefESD();
 123     if (!esd)
 124     {
 125       AliError("Processing an ESD track, but target is not an ESD event");
 126       return kFALSE;
 127     }
 128     fESDpid.SetTOFResponse(esd, AliESDpid::kTOF_T0);
 129
 130     // get time of flight, reference times and sigma
 131     esdTrack->GetIntegratedTimes(times);
 132     tof   = (Double_t)(esdTrack->GetTOFsignal() - fESDpid.GetTOFResponse().GetStartTime(esdTrack->P()));
 133     sigma = (Double_t)fESDpid.GetTOFResponse().GetExpectedSigma(esdTrack->P(), ref, fRefMass);
 134
 135     // port values to standard AliRsnCut checker
 136     fCutValueD = (tof - ref) / sigma;
 137     return OkRangeD();
 138   }
 139   else if (aodTrack)
 140   {
 141     // for AOD tracks, all operations are done by the AOD PID utility
 142     fCutValueD = (Double_t)fAODpid.NumberOfSigmasTOF(aodTrack, fRefType);
 143     return OkRangeD();
 144   }
 145   else
 146   {
 147     AliDebug(AliLog::kDebug + 2, Form("Impossible to process an object of type '%s'. Cut applicable only to ESD/AOD tracks", fDaughter->GetRef()->ClassName()));
 148     return kFALSE;
 149   }
 150 }
 151
 152 //_________________________________________________________________________________________________
 153 void AliRsnCutPIDTOF::Print(const Option_t *) const
 154 {
 155 //
 156 // Print information on this cut
 157 //
 158
 159   AliInfo(Form("Cut name, type            : %s %s", GetName(), ClassName()));
 160   AliInfo(Form("TOF PID cut range (sigmas): %.3f %.3f", fMinD, fMaxD));
 161   AliInfo(Form("Unmatched tracks are      : %s", (fRejectUnmatched ? "rejected" : "accepted")));
 162 }
 163
 164 //_________________________________________________________________________________________________
 165 void AliRsnCutPIDTOF::Initialize()
 166 {
 167 //
 168 // Initialize ESD pid object from global one
 169 //
 170
 171   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
 172   AliESDInputHandler *handler = dynamic_cast<AliESDInputHandler*>(mgr->GetInputEventHandler());
 173   if (handler)
 174   {
 175     AliESDpid *pid = handler->GetESDpid();
 176     fESDpid = (*pid);
 177   }
 178
 179   fInitialized = kTRUE;
 180 }