]>
Commit | Line | Data |
---|---|---|
b2a297fa | 1 | #ifndef ALIDIELECTRONVARMANAGER_H |
2 | #define ALIDIELECTRONVARMANAGER_H | |
3 | /* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. * | |
4 | * See cxx source for full Copyright notice */ | |
5 | ||
6 | //############################################################# | |
7 | //# # | |
8 | //# Class AliDielectronVarManager # | |
9 | //# Class for management of available variables # | |
10 | //# # | |
11 | //# Authors: # | |
12 | //# Anton Andronic, GSI / A.Andronic@gsi.de # | |
13 | //# Ionut C. Arsene, GSI / I.C.Arsene@gsi.de # | |
14 | //# Julian Book, Uni Ffm / Julian.Book@cern.ch # | |
572b0139 | 15 | //# Markus Köhler, GSI / M.Koehler@gsi.de # |
6551594b | 16 | //# Frederick Kramer, Uni Ffm / Frederick.Kramer@cern.ch # |
b2a297fa | 17 | //# Magnus Mager, CERN / Magnus.Mager@cern.ch # |
18 | //# WooJin J. Park, GSI / W.J.Park@gsi.de # | |
19 | //# Jens Wiechula, Uni HD / Jens.Wiechula@cern.ch # | |
20 | //# # | |
21 | //############################################################# | |
22 | ||
23 | ||
24 | #include <TNamed.h> | |
25 | ||
26 | #include <AliVEvent.h> | |
27 | #include <AliESDEvent.h> | |
6551594b | 28 | #include <AliAODEvent.h> |
29 | #include <AliMCEvent.h> | |
30 | #include <AliVVertex.h> | |
31 | #include <AliESDVertex.h> | |
b2a297fa | 32 | |
33 | #include <AliVParticle.h> | |
b2a297fa | 34 | #include <AliESDtrack.h> |
35 | #include <AliAODTrack.h> | |
572b0139 | 36 | #include <AliAODPid.h> |
b2a297fa | 37 | #include <AliKFParticle.h> |
572b0139 | 38 | #include <AliKFVertex.h> |
b2a297fa | 39 | #include <AliMCParticle.h> |
40 | #include <AliAODMCParticle.h> | |
41 | #include <AliVTrack.h> // ? | |
42 | ||
6551594b | 43 | #include <AliExternalTrackParam.h> |
44 | #include <AliESDpid.h> | |
4649c75b | 45 | #include <AliCentrality.h> |
2a14a7b1 | 46 | #include <AliAODpidUtil.h> |
8df8e382 | 47 | #include <AliPID.h> |
6551594b | 48 | |
b2a297fa | 49 | #include "AliDielectronPair.h" |
8df8e382 | 50 | #include "AliDielectronMC.h" |
48609e3d | 51 | #include "AliDielectronPID.h" |
b2a297fa | 52 | |
53 | class AliVEvent; | |
54 | ||
55 | //________________________________________________________________ | |
56 | class AliDielectronVarManager : public TNamed { | |
57 | ||
58 | public: | |
8df8e382 | 59 | |
b2a297fa | 60 | // Particle specific variables |
61 | enum ValueTypes { | |
62 | kPx = 0, // px | |
63 | kPy, // py | |
64 | kPz, // pz | |
65 | kPt, // transverse momentum | |
66 | kP, // momentum | |
67 | kXv, // vertex position in x | |
68 | kYv, // vertex position in y | |
69 | kZv, // vertex position in z | |
70 | kOneOverPt, // 1/pt | |
71 | kPhi, // phi angle | |
72 | kTheta, // theta angle | |
73 | kEta, // pseudo-rapidity | |
74 | kY, // rapidity | |
75 | kE, // energy | |
76 | kM, // mass | |
77 | kCharge, // charge | |
78 | kNclsITS, // number of clusters assigned in the ITS | |
79 | kNclsTPC, // number of clusters assigned in the TPC | |
554e40f8 | 80 | kNclsTPCiter1, // number of clusters assigned in the TPC after first iteration |
b2a297fa | 81 | kNFclsTPC, // number of findable clusters in the TPC |
45b2b1b8 | 82 | kNFclsTPCr, // number of findable clusters in the TPC with more robust definition |
b2a297fa | 83 | kTPCsignalN, // number of points used for dEdx |
61d106d3 | 84 | kTPCchi2Cl, // chi2/cl in TPC |
85 | kTrackStatus, // track status bits | |
8df8e382 | 86 | |
b2a297fa | 87 | kNclsTRD, // number of clusters assigned in the TRD |
88 | kTRDntracklets, // number of TRD tracklets used for tracking/PID TODO: correct getter | |
89 | kTRDpidQuality, // number of TRD tracklets used for PID | |
8df8e382 | 90 | kTRDprobEle, // TRD electron pid probability |
91 | kTRDprobPio, // TRD electron pid probability | |
92 | ||
b2a297fa | 93 | kImpactParXY, // Impact parameter in XY plane |
94 | kImpactParZ, // Impact parameter in Z | |
95 | kTrackLength, // Track length | |
96 | kPdgCode, // PDG code | |
8df8e382 | 97 | |
98 | kPdgCodeMother, // PDG code of the mother | |
99 | ||
100 | kNumberOfDaughters, // number of daughters | |
101 | kHaveSameMother, // check that particles have the same mother (MC) | |
102 | kITSsignal, // ITS dE/dx signal | |
103 | kITSsignalSSD1, // SSD1 dE/dx signal | |
104 | kITSsignalSSD2, // SSD2 dE/dx signal | |
105 | kITSsignalSDD1, // SDD1 dE/dx signal | |
106 | kITSsignalSDD2, // SDD2 dE/dx signal | |
3505bfad | 107 | kITSclusterMap, // ITS cluster map |
9143d69f | 108 | kITSnSigmaEle, // number of sigmas to the dE/dx electron line in the ITS |
109 | kITSnSigmaPio, // number of sigmas to the dE/dx pion line in the ITS | |
110 | kITSnSigmaMuo, // number of sigmas to the dE/dx muon line in the ITS | |
111 | kITSnSigmaKao, // number of sigmas to the dE/dx kaon line in the ITS | |
112 | kITSnSigmaPro, // number of sigmas to the dE/dx proton line in the ITS | |
8df8e382 | 113 | |
b2a297fa | 114 | kPIn, // momentum at inner wall of TPC (if available), used for PID |
115 | kTPCsignal, // TPC dE/dx signal | |
572b0139 | 116 | |
6551594b | 117 | kTPCnSigmaEle, // number of sigmas to the dE/dx electron line in the TPC |
572b0139 | 118 | kTPCnSigmaPio, // number of sigmas to the dE/dx pion line in the TPC |
119 | kTPCnSigmaMuo, // number of sigmas to the dE/dx muon line in the TPC | |
120 | kTPCnSigmaKao, // number of sigmas to the dE/dx kaon line in the TPC | |
121 | kTPCnSigmaPro, // number of sigmas to the dE/dx proton line in the TPC | |
122 | ||
8df8e382 | 123 | kTOFnSigmaEle, // number of sigmas to the pion line in the TOF |
572b0139 | 124 | kTOFnSigmaPio, // number of sigmas to the pion line in the TOF |
125 | kTOFnSigmaMuo, // number of sigmas to the muon line in the TOF | |
126 | kTOFnSigmaKao, // number of sigmas to the kaon line in the TOF | |
127 | kTOFnSigmaPro, // number of sigmas to the proton line in the TOF | |
8df8e382 | 128 | |
ffbede40 | 129 | kKinkIndex0, // kink index 0 |
130 | ||
b2a297fa | 131 | kParticleMax, // |
132 | // TODO: kRNClusters ?? | |
133 | // AliDielectronPair specific variables | |
134 | kChi2NDF = kParticleMax, // Chi^2/NDF | |
135 | kDecayLength, // decay length | |
136 | kR, // distance to the origin | |
137 | kOpeningAngle, // opening angle | |
8df8e382 | 138 | // helicity picture: Z-axis is considered the direction of the mother's 3-momentum vector |
139 | kThetaHE, // theta in mother's rest frame in the helicity picture | |
140 | kPhiHE, // phi in mother's rest frame in the helicity picture | |
141 | // Collins-Soper picture: Z-axis is considered the direction of the vectorial difference between | |
142 | // the 3-mom vectors of target and projectile beams | |
143 | kThetaCS, // theta in mother's rest frame in Collins-Soper picture | |
144 | kPhiCS, // phi in mother's rest frame in Collins-Soper picture | |
572b0139 | 145 | kLegDist, // distance of the legs |
146 | kLegDistXY, // distance of the legs in XY | |
61d106d3 | 147 | kDeltaEta, // Absolute value of Delta Eta for the legs |
148 | kDeltaPhi, // Absolute value of Delta Phi for the legs | |
b2a297fa | 149 | kMerr, // error of mass calculation |
150 | kDCA, // distance of closest approach TODO: not implemented yet | |
151 | kPairType, // type of the pair, like like sign ++ unlikesign ... | |
152 | kPairMax, // | |
153 | // Event specific variables | |
6551594b | 154 | kXvPrim=kPairMax, // prim vertex |
155 | kYvPrim, // prim vertex | |
156 | kZvPrim, // prim vertex | |
157 | kXRes, // primary vertex x-resolution | |
158 | kYRes, // primary vertex y-resolution | |
159 | kZRes, // primary vertex z-resolution | |
b2a297fa | 160 | kNTrk, // number of tracks (or tracklets) |
6551594b | 161 | kTracks, // ESD tracks |
ffbede40 | 162 | kCentrality, // event centrality fraction |
6551594b | 163 | kNevents, // event counter |
b2a297fa | 164 | kNMaxValues // |
165 | // TODO: (for A+A) ZDCEnergy, impact parameter, Iflag?? | |
166 | }; | |
167 | ||
168 | ||
169 | AliDielectronVarManager(); | |
170 | AliDielectronVarManager(const char* name, const char* title); | |
171 | virtual ~AliDielectronVarManager(); | |
172 | static void Fill(const TObject* particle, Double_t * const values); | |
173 | ||
6551594b | 174 | static void InitESDpid(Int_t type=0); |
2a14a7b1 | 175 | static void InitAODpidUtil(Int_t type=0); |
572b0139 | 176 | static void SetESDpid(AliESDpid * const pid) {fgESDpid=pid;} |
177 | static AliESDpid* GetESDpid() {return fgESDpid;} | |
2a14a7b1 | 178 | static AliAODpidUtil* GetAODpidUtil() {return fgAODpidUtil;} |
572b0139 | 179 | static void SetEvent(AliVEvent * const ev); |
2a14a7b1 | 180 | static Bool_t GetDCA(const AliAODTrack *track, Double_t d0z0[2]); |
181 | ||
b2a297fa | 182 | static const char* GetValueName(Int_t i) { return (i>=0&&i<kNMaxValues)?fgkParticleNames[i]:""; } |
183 | private: | |
184 | ||
185 | static const char* fgkParticleNames[kNMaxValues]; //variable names | |
186 | ||
187 | static void FillVarVParticle(const AliVParticle *particle, Double_t * const values); | |
188 | static void FillVarESDtrack(const AliESDtrack *particle, Double_t * const values); | |
189 | static void FillVarAODTrack(const AliAODTrack *particle, Double_t * const values); | |
8df8e382 | 190 | static void FillVarMCParticle(const AliMCParticle *particle, Double_t * const values); |
b2a297fa | 191 | static void FillVarAODMCParticle(const AliAODMCParticle *particle, Double_t * const values); |
192 | static void FillVarDielectronPair(const AliDielectronPair *pair, Double_t * const values); | |
6551594b | 193 | static void FillVarVEvent(const AliVEvent *event, Double_t * const values); |
194 | static void FillVarESDEvent(const AliESDEvent *event, Double_t * const values); | |
195 | static void FillVarAODEvent(const AliAODEvent *event, Double_t * const values); | |
196 | static void FillVarMCEvent(const AliMCEvent *event, Double_t * const values); | |
8df8e382 | 197 | |
2a14a7b1 | 198 | static AliESDpid *fgESDpid; // ESD pid object |
199 | static AliAODpidUtil *fgAODpidUtil; // AOD pid object | |
200 | static AliVEvent *fgEvent; // current event pointer | |
201 | static AliKFVertex *fgKFVertex; // kf vertex | |
572b0139 | 202 | |
b2a297fa | 203 | AliDielectronVarManager(const AliDielectronVarManager &c); |
204 | AliDielectronVarManager &operator=(const AliDielectronVarManager &c); | |
205 | ||
206 | ClassDef(AliDielectronVarManager,1); | |
207 | }; | |
208 | ||
209 | ||
210 | //Inline functions | |
6551594b | 211 | inline void AliDielectronVarManager::Fill(const TObject* object, Double_t * const values) |
b2a297fa | 212 | { |
213 | // | |
214 | // Main function to fill all available variables according to the type of particle | |
215 | // | |
216 | ||
6551594b | 217 | if (object->IsA() == AliESDtrack::Class()) FillVarESDtrack(static_cast<const AliESDtrack*>(object), values); |
218 | else if (object->IsA() == AliAODTrack::Class()) FillVarAODTrack(static_cast<const AliAODTrack*>(object), values); | |
219 | else if (object->IsA() == AliMCParticle::Class()) FillVarMCParticle(static_cast<const AliMCParticle*>(object), values); | |
220 | else if (object->IsA() == AliAODMCParticle::Class()) FillVarAODMCParticle(static_cast<const AliAODMCParticle*>(object), values); | |
221 | else if (object->IsA() == AliDielectronPair::Class()) FillVarDielectronPair(static_cast<const AliDielectronPair*>(object), values); | |
222 | ||
223 | // Main function to fill all available variables according to the type of event | |
224 | ||
225 | else if (object->IsA() == AliVEvent::Class()) FillVarVEvent(static_cast<const AliVEvent*>(object), values); | |
226 | else if (object->IsA() == AliESDEvent::Class()) FillVarESDEvent(static_cast<const AliESDEvent*>(object), values); | |
227 | else if (object->IsA() == AliAODEvent::Class()) FillVarAODEvent(static_cast<const AliAODEvent*>(object), values); | |
228 | else if (object->IsA() == AliMCEvent::Class()) FillVarMCEvent(static_cast<const AliMCEvent*>(object), values); | |
229 | // else Error("Fill",Form("Type %s is not supported by AliDielectronVarManager!", object->ClassName())); //TODO: implement without object needed | |
b2a297fa | 230 | } |
231 | ||
232 | inline void AliDielectronVarManager::FillVarVParticle(const AliVParticle *particle, Double_t * const values) | |
233 | { | |
234 | // | |
235 | // Fill track information available in AliVParticle into an array | |
236 | // | |
237 | values[AliDielectronVarManager::kPx] = particle->Px(); | |
238 | values[AliDielectronVarManager::kPy] = particle->Py(); | |
239 | values[AliDielectronVarManager::kPz] = particle->Pz(); | |
240 | values[AliDielectronVarManager::kPt] = particle->Pt(); | |
241 | values[AliDielectronVarManager::kP] = particle->P(); | |
242 | ||
243 | values[AliDielectronVarManager::kXv] = particle->Xv(); | |
244 | values[AliDielectronVarManager::kYv] = particle->Yv(); | |
245 | values[AliDielectronVarManager::kZv] = particle->Zv(); | |
246 | ||
247 | values[AliDielectronVarManager::kOneOverPt] = particle->OneOverPt(); | |
248 | values[AliDielectronVarManager::kPhi] = particle->Phi(); | |
249 | values[AliDielectronVarManager::kTheta] = particle->Theta(); | |
250 | values[AliDielectronVarManager::kEta] = particle->Eta(); | |
251 | values[AliDielectronVarManager::kY] = particle->Y(); | |
8df8e382 | 252 | |
b2a297fa | 253 | values[AliDielectronVarManager::kE] = particle->E(); |
254 | values[AliDielectronVarManager::kM] = particle->M(); | |
255 | values[AliDielectronVarManager::kCharge] = particle->Charge(); | |
2a14a7b1 | 256 | |
257 | values[AliDielectronVarManager::kPdgCode] = particle->PdgCode(); | |
258 | ||
6551594b | 259 | if ( fgEvent ) AliDielectronVarManager::Fill(fgEvent, values); |
b2a297fa | 260 | } |
261 | ||
262 | inline void AliDielectronVarManager::FillVarESDtrack(const AliESDtrack *particle, Double_t * const values) | |
263 | { | |
264 | // | |
265 | // Fill track information available for histogramming into an array | |
266 | // | |
267 | ||
268 | // Fill common AliVParticle interface information | |
269 | FillVarVParticle(particle, values); | |
270 | ||
8df8e382 | 271 | Double_t pidProbs[AliPID::kSPECIES]; |
b2a297fa | 272 | // Fill AliESDtrack interface specific information |
164bfb53 | 273 | Double_t tpcNcls=particle->GetTPCNcls(); |
b2a297fa | 274 | values[AliDielectronVarManager::kNclsITS] = particle->GetNcls(0); // TODO: get rid of the plain numbers |
61d106d3 | 275 | values[AliDielectronVarManager::kNclsTPC] = tpcNcls; // TODO: get rid of the plain numbers |
ffbede40 | 276 | values[AliDielectronVarManager::kNclsTPCiter1] = particle->GetTPCNclsIter1(); // TODO: get rid of the plain numbers |
b2a297fa | 277 | values[AliDielectronVarManager::kNFclsTPC] = particle->GetTPCNclsF(); |
45b2b1b8 | 278 | values[AliDielectronVarManager::kNFclsTPCr] = particle->GetTPCClusterInfo(2,1); |
b2a297fa | 279 | values[AliDielectronVarManager::kTPCsignalN] = particle->GetTPCsignalN(); |
280 | values[AliDielectronVarManager::kNclsTRD] = particle->GetNcls(2); // TODO: get rid of the plain numbers | |
281 | values[AliDielectronVarManager::kTRDntracklets] = particle->GetTRDntracklets(); // TODO: GetTRDtracklets/GetTRDntracklets? | |
282 | values[AliDielectronVarManager::kTRDpidQuality] = particle->GetTRDpidQuality(); | |
61d106d3 | 283 | values[AliDielectronVarManager::kTrackStatus] = (Double_t)particle->GetStatus(); |
284 | ||
285 | ||
286 | values[AliDielectronVarManager::kTPCchi2Cl] = -1; | |
287 | if (tpcNcls>0) values[AliDielectronVarManager::kTPCchi2Cl] = particle->GetTPCchi2() / tpcNcls; | |
8df8e382 | 288 | //TRD pidProbs |
289 | particle->GetTRDpid(pidProbs); | |
290 | values[AliDielectronVarManager::kTRDprobEle] = pidProbs[AliPID::kElectron]; | |
291 | values[AliDielectronVarManager::kTRDprobPio] = pidProbs[AliPID::kPion]; | |
ffbede40 | 292 | |
293 | values[AliDielectronVarManager::kKinkIndex0] = particle->GetKinkIndex(0); | |
8df8e382 | 294 | |
b2a297fa | 295 | Float_t impactParXY, impactParZ; |
296 | particle->GetImpactParameters(impactParXY, impactParZ); | |
297 | values[AliDielectronVarManager::kImpactParXY] = impactParXY; | |
298 | values[AliDielectronVarManager::kImpactParZ] = impactParZ; | |
299 | ||
8df8e382 | 300 | |
301 | values[AliDielectronVarManager::kPdgCode]=0; | |
302 | values[AliDielectronVarManager::kPdgCodeMother]=0; | |
303 | ||
304 | values[AliDielectronVarManager::kNumberOfDaughters]=-999; | |
305 | ||
306 | AliDielectronMC *mc=AliDielectronMC::Instance(); | |
307 | ||
308 | if (mc->HasMC()){ | |
309 | if (mc->GetMCTrack(particle)) | |
310 | values[AliDielectronVarManager::kPdgCode]= | |
311 | mc->GetMCTrack(particle)->PdgCode(); | |
312 | ||
313 | Int_t pdgMother=mc->GetMotherPDG(particle); | |
314 | if (pdgMother!=-999) | |
315 | values[AliDielectronVarManager::kPdgCodeMother]=pdgMother; | |
316 | ||
317 | values[AliDielectronVarManager::kNumberOfDaughters]=mc->NumberOfDaughters(particle); | |
318 | } //if(mc->HasMC()) | |
319 | ||
320 | ||
321 | ||
322 | values[AliDielectronVarManager::kITSsignal] = particle->GetITSsignal(); | |
323 | ||
324 | Double_t itsdEdx[4]; | |
325 | particle->GetITSdEdxSamples(itsdEdx); | |
326 | ||
327 | values[AliDielectronVarManager::kITSsignalSSD1] = itsdEdx[0]; | |
328 | values[AliDielectronVarManager::kITSsignalSSD2] = itsdEdx[1]; | |
329 | values[AliDielectronVarManager::kITSsignalSDD1] = itsdEdx[2]; | |
330 | values[AliDielectronVarManager::kITSsignalSDD2] = itsdEdx[3]; | |
3505bfad | 331 | values[AliDielectronVarManager::kITSclusterMap] = particle->GetITSClusterMap(); |
332 | ||
b2a297fa | 333 | values[AliDielectronVarManager::kTrackLength] = particle->GetIntegratedLength(); |
b2a297fa | 334 | //dEdx information |
335 | Double_t mom = particle->GetP(); | |
336 | const AliExternalTrackParam *in=particle->GetInnerParam(); | |
337 | if (in) mom = in->GetP(); | |
338 | values[AliDielectronVarManager::kPIn]=mom; | |
339 | values[AliDielectronVarManager::kTPCsignal]=particle->GetTPCsignal(); | |
6551594b | 340 | // nsigma to Electron band |
341 | // TODO: for the moment we set the bethe bloch parameters manually | |
342 | // this should be changed in future! | |
48609e3d | 343 | |
2a14a7b1 | 344 | values[AliDielectronVarManager::kTPCnSigmaEle]=fgESDpid->NumberOfSigmasTPC(particle,AliPID::kElectron)-AliDielectronPID::GetCorrVal(); |
572b0139 | 345 | values[AliDielectronVarManager::kTPCnSigmaPio]=fgESDpid->NumberOfSigmasTPC(particle,AliPID::kPion); |
346 | values[AliDielectronVarManager::kTPCnSigmaMuo]=fgESDpid->NumberOfSigmasTPC(particle,AliPID::kMuon); | |
347 | values[AliDielectronVarManager::kTPCnSigmaKao]=fgESDpid->NumberOfSigmasTPC(particle,AliPID::kKaon); | |
348 | values[AliDielectronVarManager::kTPCnSigmaPro]=fgESDpid->NumberOfSigmasTPC(particle,AliPID::kProton); | |
349 | ||
9143d69f | 350 | values[AliDielectronVarManager::kITSnSigmaEle]=fgESDpid->NumberOfSigmasITS(particle,AliPID::kElectron); |
351 | values[AliDielectronVarManager::kITSnSigmaPio]=fgESDpid->NumberOfSigmasITS(particle,AliPID::kPion); | |
352 | values[AliDielectronVarManager::kITSnSigmaMuo]=fgESDpid->NumberOfSigmasITS(particle,AliPID::kMuon); | |
353 | values[AliDielectronVarManager::kITSnSigmaKao]=fgESDpid->NumberOfSigmasITS(particle,AliPID::kKaon); | |
354 | values[AliDielectronVarManager::kITSnSigmaPro]=fgESDpid->NumberOfSigmasITS(particle,AliPID::kProton); | |
355 | ||
572b0139 | 356 | Double_t t0=fgESDpid->GetTOFResponse().GetTimeZero(); |
8df8e382 | 357 | values[AliDielectronVarManager::kTOFnSigmaEle]=fgESDpid->NumberOfSigmasTOF(particle,AliPID::kElectron,t0); |
572b0139 | 358 | values[AliDielectronVarManager::kTOFnSigmaPio]=fgESDpid->NumberOfSigmasTOF(particle,AliPID::kPion,t0); |
359 | values[AliDielectronVarManager::kTOFnSigmaMuo]=fgESDpid->NumberOfSigmasTOF(particle,AliPID::kMuon,t0); | |
360 | values[AliDielectronVarManager::kTOFnSigmaKao]=fgESDpid->NumberOfSigmasTOF(particle,AliPID::kKaon,t0); | |
361 | values[AliDielectronVarManager::kTOFnSigmaPro]=fgESDpid->NumberOfSigmasTOF(particle,AliPID::kProton,t0); | |
b2a297fa | 362 | } |
363 | ||
364 | inline void AliDielectronVarManager::FillVarAODTrack(const AliAODTrack *particle, Double_t * const values) | |
365 | { | |
366 | // | |
367 | // Fill track information available for histogramming into an array | |
368 | // | |
369 | ||
370 | // Fill common AliVParticle interface information | |
371 | FillVarVParticle(particle, values); | |
572b0139 | 372 | |
373 | // Reset AliESDtrack interface specific information | |
374 | values[AliDielectronVarManager::kNclsITS] = 0; | |
164bfb53 | 375 | values[AliDielectronVarManager::kNclsTPC] = particle->GetTPCNcls(); |
554e40f8 | 376 | values[AliDielectronVarManager::kNclsTPCiter1] = particle->GetTPCNcls(); // not really available in AOD |
572b0139 | 377 | values[AliDielectronVarManager::kNFclsTPC] = 0; |
45b2b1b8 | 378 | values[AliDielectronVarManager::kNFclsTPCr] = 0; |
572b0139 | 379 | values[AliDielectronVarManager::kNclsTRD] = 0; |
380 | values[AliDielectronVarManager::kTRDntracklets] = 0; | |
381 | values[AliDielectronVarManager::kTRDpidQuality] = 0; | |
61d106d3 | 382 | |
383 | values[AliDielectronVarManager::kTPCchi2Cl] = -1; | |
384 | values[AliDielectronVarManager::kTrackStatus] = (Double_t)particle->GetStatus(); | |
385 | ||
8df8e382 | 386 | //TRD pidProbs |
387 | //TODO: set correctly | |
388 | values[AliDielectronVarManager::kTRDprobEle] = 0; | |
389 | values[AliDielectronVarManager::kTRDprobPio] = 0; | |
390 | ||
572b0139 | 391 | //TODO: This is only an approximation!!! |
2a14a7b1 | 392 | values[AliDielectronVarManager::kTPCsignalN] = 0; |
572b0139 | 393 | |
2a14a7b1 | 394 | // Fill AliAODTrack interface information |
395 | // | |
396 | Double_t d0z0[2]; | |
397 | GetDCA(particle, d0z0); | |
398 | values[AliDielectronVarManager::kImpactParXY] = d0z0[0]; | |
399 | values[AliDielectronVarManager::kImpactParZ] = d0z0[1]; | |
572b0139 | 400 | |
401 | values[AliDielectronVarManager::kPIn]=0; | |
402 | values[AliDielectronVarManager::kTPCsignal]=0; | |
403 | ||
404 | values[AliDielectronVarManager::kTPCnSigmaEle]=0; | |
405 | values[AliDielectronVarManager::kTPCnSigmaPio]=0; | |
406 | values[AliDielectronVarManager::kTPCnSigmaMuo]=0; | |
407 | values[AliDielectronVarManager::kTPCnSigmaKao]=0; | |
408 | values[AliDielectronVarManager::kTPCnSigmaPro]=0; | |
b2a297fa | 409 | |
3505bfad | 410 | values[AliDielectronVarManager::kITSclusterMap] = particle->GetITSClusterMap(); |
572b0139 | 411 | |
412 | AliAODPid *pid=particle->GetDetPid(); | |
413 | if (pid){ | |
414 | Double_t mom =pid->GetTPCmomentum(); | |
2a14a7b1 | 415 | values[AliDielectronVarManager::kTPCsignalN] = pid->GetTPCsignalN(); |
416 | Double_t tpcNsigmaEle=fgAODpidUtil->NumberOfSigmasTPC(particle,AliPID::kElectron); | |
417 | Double_t tpcNsigmaPio=fgAODpidUtil->NumberOfSigmasTPC(particle,AliPID::kPion); | |
418 | Double_t tpcNsigmaMuo=fgAODpidUtil->NumberOfSigmasTPC(particle,AliPID::kMuon); | |
419 | Double_t tpcNsigmaKao=fgAODpidUtil->NumberOfSigmasTPC(particle,AliPID::kKaon); | |
420 | Double_t tpcNsigmaPro=fgAODpidUtil->NumberOfSigmasTPC(particle,AliPID::kProton); | |
572b0139 | 421 | |
164bfb53 | 422 | values[AliDielectronVarManager::kPIn]=mom; |
572b0139 | 423 | values[AliDielectronVarManager::kTPCsignal]=pid->GetTPCsignal(); |
424 | ||
425 | values[AliDielectronVarManager::kTPCnSigmaEle]=tpcNsigmaEle; | |
426 | values[AliDielectronVarManager::kTPCnSigmaPio]=tpcNsigmaPio; | |
427 | values[AliDielectronVarManager::kTPCnSigmaMuo]=tpcNsigmaMuo; | |
428 | values[AliDielectronVarManager::kTPCnSigmaKao]=tpcNsigmaKao; | |
429 | values[AliDielectronVarManager::kTPCnSigmaPro]=tpcNsigmaPro; | |
430 | ||
431 | values[AliDielectronVarManager::kTRDntracklets] = 0; | |
432 | values[AliDielectronVarManager::kTRDpidQuality] = 0; | |
433 | ||
434 | } | |
b2a297fa | 435 | } |
436 | ||
437 | inline void AliDielectronVarManager::FillVarMCParticle(const AliMCParticle *particle, Double_t * const values) | |
438 | { | |
439 | // | |
440 | // Fill track information available for histogramming into an array | |
441 | // | |
442 | ||
554e40f8 | 443 | values[AliDielectronVarManager::kNclsITS] = 0; |
444 | values[AliDielectronVarManager::kNclsTPC] = 0; | |
445 | values[AliDielectronVarManager::kNclsTPCiter1] = 0; | |
446 | values[AliDielectronVarManager::kNFclsTPC] = 0; | |
45b2b1b8 | 447 | values[AliDielectronVarManager::kNFclsTPCr] = 0; |
554e40f8 | 448 | values[AliDielectronVarManager::kNclsTRD] = 0; |
449 | values[AliDielectronVarManager::kTRDntracklets] = 0; | |
450 | values[AliDielectronVarManager::kTRDpidQuality] = 0; | |
451 | values[AliDielectronVarManager::kTPCchi2Cl] = 0; | |
452 | values[AliDielectronVarManager::kTrackStatus] = 0; | |
453 | values[AliDielectronVarManager::kTRDprobEle] = 0; | |
454 | values[AliDielectronVarManager::kTRDprobPio] = 0; | |
455 | values[AliDielectronVarManager::kTPCsignalN] = 0; | |
456 | values[AliDielectronVarManager::kImpactParXY] = 0; | |
457 | values[AliDielectronVarManager::kImpactParZ] = 0; | |
458 | values[AliDielectronVarManager::kPIn] = 0; | |
459 | values[AliDielectronVarManager::kTPCsignal] = 0; | |
460 | values[AliDielectronVarManager::kTPCnSigmaEle] = 0; | |
461 | values[AliDielectronVarManager::kTPCnSigmaPio] = 0; | |
462 | values[AliDielectronVarManager::kTPCnSigmaMuo] = 0; | |
463 | values[AliDielectronVarManager::kTPCnSigmaKao] = 0; | |
464 | values[AliDielectronVarManager::kTPCnSigmaPro] = 0; | |
465 | values[AliDielectronVarManager::kITSclusterMap] = 0; | |
466 | ||
467 | values[AliDielectronVarManager::kPdgCode] = 0; | |
468 | values[AliDielectronVarManager::kPdgCodeMother] = 0; | |
469 | ||
b2a297fa | 470 | // Fill common AliVParticle interface information |
471 | FillVarVParticle(particle, values); | |
8df8e382 | 472 | |
473 | AliDielectronMC *mc=AliDielectronMC::Instance(); | |
474 | ||
b2a297fa | 475 | // Fill AliMCParticle interface specific information |
476 | values[AliDielectronVarManager::kPdgCode] = particle->PdgCode(); | |
8df8e382 | 477 | |
478 | AliMCParticle *mother = mc->GetMCTrackMother(particle); | |
479 | if (mother) values[AliDielectronVarManager::kPdgCodeMother] = mother->PdgCode(); | |
480 | ||
481 | ||
482 | values[AliDielectronVarManager::kNumberOfDaughters]=mc->NumberOfDaughters(particle); | |
b2a297fa | 483 | } |
484 | ||
485 | inline void AliDielectronVarManager::FillVarAODMCParticle(const AliAODMCParticle *particle, Double_t * const values) | |
486 | { | |
487 | // | |
488 | // Fill track information available for histogramming into an array | |
489 | // | |
490 | ||
554e40f8 | 491 | values[AliDielectronVarManager::kNclsITS] = 0; |
492 | values[AliDielectronVarManager::kNclsTPC] = 0; | |
493 | values[AliDielectronVarManager::kNclsTPCiter1] = 0; | |
494 | values[AliDielectronVarManager::kNFclsTPC] = 0; | |
495 | values[AliDielectronVarManager::kNclsTRD] = 0; | |
496 | values[AliDielectronVarManager::kTRDntracklets] = 0; | |
497 | values[AliDielectronVarManager::kTRDpidQuality] = 0; | |
498 | values[AliDielectronVarManager::kTPCchi2Cl] = 0; | |
499 | values[AliDielectronVarManager::kTrackStatus] = 0; | |
500 | values[AliDielectronVarManager::kTRDprobEle] = 0; | |
501 | values[AliDielectronVarManager::kTRDprobPio] = 0; | |
502 | values[AliDielectronVarManager::kTPCsignalN] = 0; | |
503 | values[AliDielectronVarManager::kImpactParXY] = 0; | |
504 | values[AliDielectronVarManager::kImpactParZ] = 0; | |
505 | values[AliDielectronVarManager::kPIn] = 0; | |
506 | values[AliDielectronVarManager::kTPCsignal] = 0; | |
507 | values[AliDielectronVarManager::kTPCnSigmaEle] = 0; | |
508 | values[AliDielectronVarManager::kTPCnSigmaPio] = 0; | |
509 | values[AliDielectronVarManager::kTPCnSigmaMuo] = 0; | |
510 | values[AliDielectronVarManager::kTPCnSigmaKao] = 0; | |
511 | values[AliDielectronVarManager::kTPCnSigmaPro] = 0; | |
512 | values[AliDielectronVarManager::kITSclusterMap] = 0; | |
513 | ||
514 | values[AliDielectronVarManager::kPdgCode] = 0; | |
515 | values[AliDielectronVarManager::kPdgCodeMother] = 0; | |
516 | ||
b2a297fa | 517 | // Fill common AliVParticle interface information |
518 | FillVarVParticle(particle, values); | |
554e40f8 | 519 | |
8df8e382 | 520 | AliDielectronMC *mc=AliDielectronMC::Instance(); |
521 | ||
522 | ||
b2a297fa | 523 | // Fill AliAODMCParticle interface specific information |
8df8e382 | 524 | values[AliDielectronVarManager::kPdgCode] = particle->PdgCode(); |
525 | ||
526 | AliVParticle *mother = mc->GetMCTrackMother(particle); | |
527 | if (mother) values[AliDielectronVarManager::kPdgCodeMother] = mother->PdgCode(); | |
528 | ||
529 | values[AliDielectronVarManager::kNumberOfDaughters]=mc->NumberOfDaughters(particle); | |
b2a297fa | 530 | } |
531 | ||
532 | inline void AliDielectronVarManager::FillVarDielectronPair(const AliDielectronPair *pair, Double_t * const values) | |
533 | { | |
534 | // | |
535 | // Fill pair information available for histogramming into an array | |
536 | // | |
2a14a7b1 | 537 | |
538 | values[AliDielectronVarManager::kPdgCode]=0; | |
539 | values[AliDielectronVarManager::kPdgCodeMother]=0; | |
540 | ||
b2a297fa | 541 | // Fill common AliVParticle interface information |
542 | FillVarVParticle(pair, values); | |
543 | ||
544 | // Fill AliDielectronPair specific information | |
545 | const AliKFParticle &kfPair = pair->GetKFParticle(); | |
61d106d3 | 546 | |
547 | Double_t thetaHE=0; | |
548 | Double_t phiHE=0; | |
549 | Double_t thetaCS=0; | |
550 | Double_t phiCS=0; | |
551 | ||
552 | pair->GetThetaPhiCM(thetaHE,phiHE,thetaCS,phiCS); | |
553 | ||
b2a297fa | 554 | values[AliDielectronVarManager::kChi2NDF] = kfPair.GetChi2()/kfPair.GetNDF(); |
555 | values[AliDielectronVarManager::kDecayLength] = kfPair.GetDecayLength(); | |
556 | values[AliDielectronVarManager::kR] = kfPair.GetR(); | |
557 | values[AliDielectronVarManager::kOpeningAngle] = pair->OpeningAngle(); | |
61d106d3 | 558 | values[AliDielectronVarManager::kThetaHE] = thetaHE; |
559 | values[AliDielectronVarManager::kPhiHE] = phiHE; | |
560 | values[AliDielectronVarManager::kThetaCS] = thetaCS; | |
561 | values[AliDielectronVarManager::kPhiCS] = phiCS; | |
572b0139 | 562 | values[AliDielectronVarManager::kLegDist] = pair->DistanceDaughters(); |
563 | values[AliDielectronVarManager::kLegDistXY] = pair->DistanceDaughtersXY(); | |
61d106d3 | 564 | values[AliDielectronVarManager::kDeltaEta] = pair->DeltaEta(); |
565 | values[AliDielectronVarManager::kDeltaPhi] = pair->DeltaPhi(); | |
8df8e382 | 566 | values[AliDielectronVarManager::kMerr] = kfPair.GetErrMass()>1e-30&&kfPair.GetMass()>1e-30?kfPair.GetErrMass()/kfPair.GetMass():1000000; |
b2a297fa | 567 | values[AliDielectronVarManager::kPairType] = pair->GetType(); |
8df8e382 | 568 | |
569 | ||
570 | ||
571 | AliDielectronMC *mc=AliDielectronMC::Instance(); | |
572 | ||
573 | if (mc->HasMC()){ | |
574 | Bool_t samemother = mc->HaveSameMother(pair); | |
575 | values[AliDielectronVarManager::kHaveSameMother] = samemother ; | |
576 | }//if (mc->HasMC()) | |
577 | ||
578 | ||
b2a297fa | 579 | } |
580 | ||
6551594b | 581 | |
582 | inline void AliDielectronVarManager::FillVarVEvent(const AliVEvent *event, Double_t * const values) | |
583 | { | |
584 | // | |
585 | // Fill event information available for histogramming into an array | |
586 | // | |
587 | const AliVVertex *primVtx = event->GetPrimaryVertex(); | |
8df8e382 | 588 | |
589 | values[AliDielectronVarManager::kXvPrim] = 0; | |
590 | values[AliDielectronVarManager::kYvPrim] = 0; | |
591 | values[AliDielectronVarManager::kZvPrim] = 0; | |
592 | values[AliDielectronVarManager::kChi2NDF] = 0; | |
593 | ||
594 | values[AliDielectronVarManager::kNTrk] = 0; | |
595 | values[AliDielectronVarManager::kNevents] = 0; //always fill bin 0; | |
596 | ||
597 | if (!primVtx) return; | |
598 | ||
6551594b | 599 | values[AliDielectronVarManager::kXvPrim] = primVtx->GetX(); |
600 | values[AliDielectronVarManager::kYvPrim] = primVtx->GetY(); | |
601 | values[AliDielectronVarManager::kZvPrim] = primVtx->GetZ(); | |
602 | values[AliDielectronVarManager::kChi2NDF] = primVtx->GetChi2perNDF(); | |
603 | ||
604 | values[AliDielectronVarManager::kNTrk] = event->GetNumberOfTracks(); | |
6551594b | 605 | } |
606 | ||
607 | inline void AliDielectronVarManager::FillVarESDEvent(const AliESDEvent *event, Double_t * const values) | |
608 | { | |
609 | // | |
610 | // Fill event information available for histogramming into an array | |
611 | // | |
612 | ||
613 | // Fill common AliVEvent interface information | |
614 | FillVarVEvent(event, values); | |
ffbede40 | 615 | |
616 | Double_t centralityF=-1; | |
4649c75b | 617 | AliCentrality *esdCentrality = const_cast<AliESDEvent*>(event)->GetCentrality(); |
ffbede40 | 618 | if (esdCentrality) centralityF = esdCentrality->GetCentralityPercentile("V0M"); |
619 | ||
6551594b | 620 | // Fill AliESDEvent interface specific information |
621 | const AliESDVertex *primVtx = event->GetPrimaryVertex(); | |
622 | values[AliDielectronVarManager::kXRes] = primVtx->GetXRes(); | |
623 | values[AliDielectronVarManager::kYRes] = primVtx->GetYRes(); | |
624 | values[AliDielectronVarManager::kZRes] = primVtx->GetZRes(); | |
ffbede40 | 625 | values[AliDielectronVarManager::kCentrality] = centralityF; |
6551594b | 626 | } |
627 | ||
628 | inline void AliDielectronVarManager::FillVarAODEvent(const AliAODEvent *event, Double_t * const values) | |
629 | { | |
630 | // | |
631 | // Fill event information available for histogramming into an array | |
632 | // | |
633 | ||
634 | // Fill common AliVEvent interface information | |
635 | FillVarVEvent(event, values); | |
636 | ||
637 | // Fill AliAODEvent interface specific information | |
638 | } | |
639 | ||
640 | inline void AliDielectronVarManager::FillVarMCEvent(const AliMCEvent *event, Double_t * const values) | |
641 | { | |
642 | // | |
643 | // Fill event information available for histogramming into an array | |
644 | // | |
645 | ||
646 | // Fill common AliVEvent interface information | |
647 | FillVarVEvent(event, values); | |
648 | ||
649 | // Fill AliMCEvent interface specific information | |
650 | } | |
651 | ||
652 | inline void AliDielectronVarManager::InitESDpid(Int_t type) | |
653 | { | |
654 | // | |
655 | // initialize PID parameters | |
656 | // type=0 is simulation | |
657 | // type=1 is data | |
658 | ||
572b0139 | 659 | if (!fgESDpid) fgESDpid=new AliESDpid; |
6551594b | 660 | Double_t alephParameters[5]; |
661 | // simulation | |
662 | alephParameters[0] = 2.15898e+00/50.; | |
663 | alephParameters[1] = 1.75295e+01; | |
664 | alephParameters[2] = 3.40030e-09; | |
665 | alephParameters[3] = 1.96178e+00; | |
666 | alephParameters[4] = 3.91720e+00; | |
8df8e382 | 667 | fgESDpid->GetTOFResponse().SetTimeResolution(80.); |
6551594b | 668 | |
669 | // data | |
83e37742 | 670 | if (type==1){ |
671 | alephParameters[0] = 0.0283086/0.97; | |
6551594b | 672 | alephParameters[1] = 2.63394e+01; |
673 | alephParameters[2] = 5.04114e-11; | |
674 | alephParameters[3] = 2.12543e+00; | |
675 | alephParameters[4] = 4.88663e+00; | |
8df8e382 | 676 | fgESDpid->GetTOFResponse().SetTimeResolution(130.); |
83e37742 | 677 | fgESDpid->GetTPCResponse().SetMip(50.); |
6551594b | 678 | } |
679 | ||
680 | fgESDpid->GetTPCResponse().SetBetheBlochParameters( | |
681 | alephParameters[0],alephParameters[1],alephParameters[2], | |
682 | alephParameters[3],alephParameters[4]); | |
683 | ||
8df8e382 | 684 | fgESDpid->GetTPCResponse().SetSigma(3.79301e-03, 2.21280e+04); |
6551594b | 685 | } |
686 | ||
2a14a7b1 | 687 | inline void AliDielectronVarManager::InitAODpidUtil(Int_t type) |
688 | { | |
689 | if (!fgAODpidUtil) fgAODpidUtil=new AliAODpidUtil; | |
690 | Double_t alephParameters[5]; | |
691 | // simulation | |
692 | alephParameters[0] = 2.15898e+00/50.; | |
693 | alephParameters[1] = 1.75295e+01; | |
694 | alephParameters[2] = 3.40030e-09; | |
695 | alephParameters[3] = 1.96178e+00; | |
696 | alephParameters[4] = 3.91720e+00; | |
697 | fgAODpidUtil->GetTOFResponse().SetTimeResolution(80.); | |
698 | ||
699 | // data | |
700 | if (type==1){ | |
701 | alephParameters[0] = 0.0283086/0.97; | |
702 | alephParameters[1] = 2.63394e+01; | |
703 | alephParameters[2] = 5.04114e-11; | |
704 | alephParameters[3] = 2.12543e+00; | |
705 | alephParameters[4] = 4.88663e+00; | |
706 | fgAODpidUtil->GetTOFResponse().SetTimeResolution(130.); | |
707 | fgAODpidUtil->GetTPCResponse().SetMip(50.); | |
708 | } | |
709 | ||
710 | fgAODpidUtil->GetTPCResponse().SetBetheBlochParameters( | |
711 | alephParameters[0],alephParameters[1],alephParameters[2], | |
712 | alephParameters[3],alephParameters[4]); | |
713 | ||
714 | fgAODpidUtil->GetTPCResponse().SetSigma(3.79301e-03, 2.21280e+04); | |
715 | } | |
716 | ||
572b0139 | 717 | |
718 | inline void AliDielectronVarManager::SetEvent(AliVEvent * const ev) | |
719 | { | |
720 | ||
721 | fgEvent = ev; | |
722 | if (fgKFVertex) delete fgKFVertex; | |
723 | fgKFVertex=0x0; | |
8df8e382 | 724 | if (ev && ev->GetPrimaryVertex()) fgKFVertex=new AliKFVertex(*ev->GetPrimaryVertex()); |
572b0139 | 725 | } |
2a14a7b1 | 726 | |
727 | ||
728 | inline Bool_t AliDielectronVarManager::GetDCA(const AliAODTrack *track, Double_t d0z0[2]) | |
729 | { | |
730 | if(track->TestBit(AliAODTrack::kIsDCA)){ | |
731 | d0z0[0]=track->DCA(); | |
732 | d0z0[1]=track->ZAtDCA(); | |
733 | return kTRUE; | |
734 | } | |
735 | ||
736 | Double_t covd0z0[3]; | |
737 | AliAODTrack copy(*track); | |
738 | AliAODVertex *vtx =(AliAODVertex*)(fgEvent->GetPrimaryVertex()); | |
739 | Double_t fBzkG = fgEvent->GetMagneticField(); // z componenent of field in kG | |
740 | Bool_t ok = copy.PropagateToDCA(vtx,fBzkG,kVeryBig,d0z0,covd0z0); | |
741 | if(!ok){ | |
742 | d0z0[0]=-999.; | |
743 | d0z0[1]=-999.; | |
744 | } | |
745 | return ok; | |
746 | } | |
747 | ||
b2a297fa | 748 | /* |
749 | inline void AliDielectronVarManager::FillValues(const TParticle *particle, Double_t *values) | |
750 | { | |
751 | // | |
752 | // Fill track information available for histogramming into an array | |
753 | // | |
754 | ||
755 | // Fill TParticle interface information | |
756 | values[AliDielectronVarManager::kPx] = particle->Px(); | |
757 | values[AliDielectronVarManager::kPy] = particle->Py(); | |
758 | values[AliDielectronVarManager::kPz] = particle->Pz(); | |
759 | values[AliDielectronVarManager::kPt] = particle->Pt(); | |
760 | values[AliDielectronVarManager::kP] = particle->P(); | |
761 | ||
762 | values[AliDielectronVarManager::kXv] = particle->Vx(); | |
763 | values[AliDielectronVarManager::kYv] = particle->Vy(); | |
764 | values[AliDielectronVarManager::kZv] = particle->Vz(); | |
765 | ||
766 | values[AliDielectronVarManager::kOneOverPt] = 1./particle->Pt(); | |
767 | values[AliDielectronVarManager::kPhi] = particle->Phi(); | |
768 | values[AliDielectronVarManager::kTheta] = | |
769 | values[AliDielectronVarManager::kEta] = particle->Eta(); | |
770 | values[AliDielectronVarManager::kY] = | |
771 | ||
772 | values[AliDielectronVarManager::kE] = particle->Energy(); | |
773 | values[AliDielectronVarManager::kM] = particle->GetMass(); | |
774 | ||
775 | values[AliDielectronVarManager::kCharge] = particle->GetPDG()->Charge()/3; // uggly | |
776 | ||
777 | }*/ | |
778 | ||
779 | #endif |