906e0007118c8c7bbd0856f973935403f76eb13a
[u/mrichter/AliRoot.git] / PWG2 / RESONANCES / AliRsnCutPIDTPC.cxx
1 //
2 // Class AliRsnCutPIDTPC
3 //
4 // General implementation of a single cut strategy, which can be:
5 // - a value contained in a given interval  [--> IsBetween()   ]
6 // - a value equal to a given reference     [--> MatchesValue()]
7 //
8 // In all cases, the reference value(s) is (are) given as data members
9 // and each kind of cut requires a given value type (Int, UInt, Double),
10 // but the cut check procedure is then automatized and chosen thanks to
11 // an enumeration of the implemented cut types.
12 // At the end, the user (or any other point which uses this object) has
13 // to use the method IsSelected() to check if this cut has been passed.
14 //
15 // authors: Martin Vala (martin.vala@cern.ch)
16 //          Alberto Pulvirenti (alberto.pulvirenti@ct.infn.it)
17 //
18
19 #include <Riostream.h>
20
21 #include "AliESDpid.h"
22 #include "AliTOFT0maker.h"
23 #include "AliTOFcalib.h"
24 #include "AliCDBManager.h"
25 #include "AliITSPIDResponse.h"
26
27 #include "AliRsnEvent.h"
28 #include "AliRsnDaughter.h"
29 #include "AliRsnCutPIDTPC.h"
30
31 ClassImp(AliRsnCutPIDTPC)
32
33 //_________________________________________________________________________________________________
34 AliRsnCutPIDTPC::AliRsnCutPIDTPC
35 (const char *name, AliPID::EParticleType type, Double_t min, Double_t max, Bool_t rejectOutside) :
36   AliRsnCut(name, AliRsnCut::kDaughter, min, max),
37   fRejectOutside(rejectOutside),
38   fMomMin(0.0),
39   fMomMax(1E+20),
40   fRefType(type),
41   fESDpid(),
42   fAODpid()
43 {
44 //
45 // Main constructor.
46 //
47 }
48
49 //_________________________________________________________________________________________________
50 AliRsnCutPIDTPC::AliRsnCutPIDTPC
51 (const AliRsnCutPIDTPC& copy) :
52   AliRsnCut(copy),
53   fRejectOutside(copy.fRejectOutside),
54   fMomMin(copy.fMomMin),
55   fMomMax(copy.fMomMax),
56   fRefType(copy.fRefType),
57   fESDpid(copy.fESDpid),
58   fAODpid(copy.fAODpid)
59 {
60 //
61 // Copy constructor.
62 //
63 }
64
65 //_________________________________________________________________________________________________
66 AliRsnCutPIDTPC& AliRsnCutPIDTPC::operator=(const AliRsnCutPIDTPC& copy)
67 {
68 //
69 // Assignment operator
70 //
71
72   AliRsnCut::operator=(copy);
73
74   fRejectOutside = copy.fRejectOutside;
75   fMomMin  = copy.fMomMin;
76   fMomMax  = copy.fMomMax;
77   fRefType = copy.fRefType;
78   fESDpid  = copy.fESDpid;
79   fAODpid  = copy.fAODpid;
80   
81   return (*this);
82 }
83
84 //_________________________________________________________________________________________________
85 void AliRsnCutPIDTPC::SetBBParam(Double_t p0, Double_t p1, Double_t p2, Double_t p3, Double_t p4)
86 {
87 //
88 // Properly set the Bethe-Bloch parameters in all places where it is needed.
89 //
90
91   fESDpid.GetTPCResponse().SetBetheBlochParameters(p0, p1, p2, p3, p4);
92   fAODpid.GetTPCResponse().SetBetheBlochParameters(p0, p1, p2, p3, p4);
93 }
94
95 //_________________________________________________________________________________________________
96 Bool_t AliRsnCutPIDTPC::IsSelected(TObject *object)
97 {
98 //
99 // Cut checker.
100 //
101
102   // coherence check
103   if (!TargetOK(object)) return kFALSE;
104   
105   // reject not TPC tracks
106   AliVTrack *vtrack = dynamic_cast<AliVTrack*>(fDaughter->GetRef());
107   if (!vtrack) return kFALSE;
108   if (!IsTPC(vtrack))
109   {
110     AliDebug(AliLog::kDebug + 2, "Track is not found in TPC");
111     return kFALSE;
112   }
113   
114   // common evaluation variables
115   Double_t     mom;
116   AliESDtrack *esdTrack = fDaughter->GetRefESDtrack();
117   AliAODTrack *aodTrack = fDaughter->GetRefAODtrack();
118
119   // get inner momentum and check it w.r. to allowed range:
120   // all tracks outside it will pass the cut or not, depending on 'fRejectOutside'
121   if (esdTrack)
122     mom = esdTrack->GetInnerParam()->P();
123   else if (aodTrack)
124     mom = aodTrack->GetDetPid()->GetTPCmomentum();
125   else
126   {
127     AliDebug(AliLog::kDebug + 2, Form("Impossible to process an object of type '%s'. Cut applicable only to ESD/AOD tracks", fDaughter->GetRef()->ClassName()));
128     return kFALSE;
129   }
130   if ((mom < fMomMin || mom > fMomMax))
131   {
132     AliDebug(AliLog::kDebug + 2, Form("Track momentum = %.5f, outside allowed range", mom));
133     return (!fRejectOutside);
134   }
135
136   // assign PID nsigmas to default cut check value
137   // since bad object types are rejected before, here we have an ESD track or AOD track
138   if (esdTrack) 
139     fCutValueD = fESDpid.GetTPCResponse().GetNumberOfSigmas(mom, esdTrack->GetTPCsignal(), esdTrack->GetTPCsignalN(), fRefType);
140   else
141     fCutValueD = fAODpid.NumberOfSigmasTPC(aodTrack, fRefType);
142
143   // use AliRsnCut default method to check cut
144   return OkRangeD();
145 }
146
147 //_________________________________________________________________________________________________
148 void AliRsnCutPIDTPC::Print(const Option_t *) const
149 {
150 //
151 // Print information on this cut
152 //
153
154   AliInfo(Form("Cut name                    : %s", GetName()));
155   AliInfo(Form("--> cut range (nsigma)      : %.3f %.3f", fMinD, fMaxD));
156   AliInfo(Form("--> momentum range          : %.3f %.3f", fMomMin, fMomMax));
157   AliInfo(Form("--> tracks outside range are: %s", (fRejectOutside ? "rejected" : "accepted")));
158 }