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