Recovery ESDpid from InputHandler
[u/mrichter/AliRoot.git] / 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 }