2 // Class AliRsnCutPIDTOF
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
10 // authors: Martin Vala (martin.vala@cern.ch)
11 // Alberto Pulvirenti (alberto.pulvirenti@ct.infn.it)
14 #include "AliAnalysisManager.h"
15 #include "AliESDInputHandler.h"
17 #include "AliRsnCutPIDTOF.h"
19 ClassImp(AliRsnCutPIDTOF)
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),
26 fRejectUnmatched(rejectUnmatched),
27 fRefType(AliPID::kUnknown),
33 // Default constructor.
34 // To set the reference PID type, calls the SetRefType method,
35 // which sets the mass accordingly and coherently.
41 //_________________________________________________________________________________________________
42 AliRsnCutPIDTOF::AliRsnCutPIDTOF(const AliRsnCutPIDTOF& copy) :
45 fRejectUnmatched(copy.fRejectUnmatched),
46 fRefType(AliPID::kUnknown),
48 fESDpid(copy.fESDpid),
53 // To set the reference PID type, calls the SetRefType method,
54 // which sets the mass accordingly and coherently.
57 SetRefType(copy.fRefType);
60 //_________________________________________________________________________________________________
61 AliRsnCutPIDTOF& AliRsnCutPIDTOF::operator=(const AliRsnCutPIDTOF& copy)
64 // Assignment operator.
65 // To set the reference PID type, calls the SetRefType method,
66 // which sets the mass accordingly and coherently.
69 fInitialized = kFALSE;
70 fRejectUnmatched = copy.fRejectUnmatched;
71 fESDpid = copy.fESDpid;
72 fAODpid = copy.fAODpid;
74 SetRefType(copy.fRefType);
79 //_________________________________________________________________________________________________
80 Bool_t AliRsnCutPIDTOF::IsSelected(TObject *object)
86 // initialize if needed
87 if (!fInitialized) Initialize();
90 if (!TargetOK(object)) return kFALSE;
92 // reject always non-track objects
93 AliVTrack *vtrack = fDaughter->GetRefVtrack();
95 AliDebug(AliLog::kDebug + 2, Form("Impossible to process an object of type '%s'. Cut applicable only to ESD/AOD tracks", fDaughter->GetRef()->ClassName()));
99 // checks that track is matched in TOF:
100 // if not, the track is accepted or rejected
101 // depending on the 'fRejectUnmatched' data member:
102 // -- kTRUE --> all unmatched tracks are rejected
103 // -- kFALSE --> all unmatched tracks are accepted (it is assumed that other PIDs are done)
104 if (!IsMatched(vtrack)) {
105 AliDebug(AliLog::kDebug + 2, "Track is not matched with TOF");
106 return (!fRejectUnmatched);
109 // retrieve real object type and
110 // prepare some useful variables
111 Double_t tof, sigma, times[5];
112 Double_t &ref = times[(Int_t)fRefType];
113 AliESDtrack *esdTrack = fDaughter->GetRefESDtrack();
114 AliAODTrack *aodTrack = fDaughter->GetRefAODtrack();
116 // cut check depends on the object type
118 // setup the ESD PID object
119 AliESDEvent *esd = 0x0;
120 if (fEvent) esd = fEvent->GetRefESD();
122 AliError("Processing an ESD track, but target is not an ESD event");
125 fESDpid.SetTOFResponse(esd, AliESDpid::kTOF_T0);
127 // get time of flight, reference times and sigma
128 esdTrack->GetIntegratedTimes(times);
129 tof = (Double_t)(esdTrack->GetTOFsignal() - fESDpid.GetTOFResponse().GetStartTime(esdTrack->P()));
130 sigma = (Double_t)fESDpid.GetTOFResponse().GetExpectedSigma(esdTrack->P(), ref, fRefMass);
132 // port values to standard AliRsnCut checker
133 fCutValueD = (tof - ref) / sigma;
135 } else if (aodTrack) {
136 // for AOD tracks, all operations are done by the AOD PID utility
137 fCutValueD = (Double_t)fAODpid.NumberOfSigmasTOF(aodTrack, fRefType);
140 AliDebug(AliLog::kDebug + 2, Form("Impossible to process an object of type '%s'. Cut applicable only to ESD/AOD tracks", fDaughter->GetRef()->ClassName()));
145 //_________________________________________________________________________________________________
146 void AliRsnCutPIDTOF::Print(const Option_t *) const
149 // Print information on this cut
152 AliInfo(Form("Cut name, type : %s %s", GetName(), ClassName()));
153 AliInfo(Form("TOF PID cut range (sigmas): %.3f %.3f", fMinD, fMaxD));
154 AliInfo(Form("Unmatched tracks are : %s", (fRejectUnmatched ? "rejected" : "accepted")));
157 //_________________________________________________________________________________________________
158 void AliRsnCutPIDTOF::Initialize()
161 // Initialize ESD pid object from global one
164 AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
165 AliESDInputHandler *handler = dynamic_cast<AliESDInputHandler*>(mgr->GetInputEventHandler());
167 AliESDpid *pid = handler->GetESDpid();
168 if (pid) fESDpid = (*pid);
171 fInitialized = kTRUE;