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 <AliCentrality.h>
  46 #include <AliAODpidUtil.h>
  47 #include <AliPID.h>
  48 #include <AliPIDResponse.h>
  49
  50 #include "AliDielectronPair.h"
  51 #include "AliDielectronMC.h"
  52 #include "AliDielectronPID.h"
  53 #include "AliDielectronHelper.h"
  54
  55 class AliVEvent;
  56
  57 //________________________________________________________________
  58 class AliDielectronVarManager : public TNamed {
  59
  60 public:
  61
  62   // Particle specific variables
  63   enum ValueTypes {
  64     kPx = 0,                 // px
  65     kPy,                     // py
  66     kPz,                     // pz
  67     kPt,                     // transverse momentum
  68     kP,                      // momentum
  69     kXv,                     // vertex position in x
  70     kYv,                     // vertex position in y
  71     kZv,                     // vertex position in z
  72     kOneOverPt,              // 1/pt
  73     kPhi,                    // phi angle
  74     kTheta,                  // theta angle
  75     kEta,                    // pseudo-rapidity
  76     kY,                      // rapidity
  77     kE,                      // energy
  78     kM,                      // mass
  79     kCharge,                 // charge
  80     kNclsITS,                // number of clusters assigned in the ITS
  81     kNclsTPC,                // number of clusters assigned in the TPC
  82     kNclsTPCiter1,           // number of clusters assigned in the TPC after first iteration
  83     kNFclsTPC,               // number of findable clusters in the TPC
  84     kNFclsTPCr,              // number of findable clusters in the TPC with more robust definition
  85     kNFclsTPCrFrac,          // number of found/findable clusters in the TPC with more robust definition
  86     kTPCsignalN,             // number of points used for dEdx
  87     kTPCsignalNfrac,         // fraction of points used for dEdx / cluster used for tracking
  88     kTPCchi2Cl,              // chi2/cl in TPC
  89     kTrackStatus,            // track status bits
  90
  91     kNclsTRD,                // number of clusters assigned in the TRD
  92     kTRDntracklets,          // number of TRD tracklets used for tracking/PID TODO: correct getter
  93     kTRDpidQuality,          // number of TRD tracklets used for PID
  94     kTRDprobEle,             // TRD electron pid probability
  95     kTRDprobPio,             // TRD electron pid probability
  96
  97     kImpactParXY,            // Impact parameter in XY plane
  98     kImpactParZ,             // Impact parameter in Z
  99     kTrackLength,            // Track length
 100     kPdgCode,                // PDG code
 101
 102     kPdgCodeMother,          // PDG code of the mother
 103     kPdgCodeGrandMother,          // PDG code of the mother
 104
 105     kNumberOfDaughters,      // number of daughters
 106     kHaveSameMother,         // check that particles have the same mother (MC)
 107     kIsJpsiPrimary,          // check if the particle is primary (MC)
 108     kITSsignal,              // ITS dE/dx signal
 109     kITSsignalSSD1,          // SSD1 dE/dx signal
 110     kITSsignalSSD2,          // SSD2 dE/dx signal
 111     kITSsignalSDD1,          // SDD1 dE/dx signal
 112     kITSsignalSDD2,          // SDD2 dE/dx signal
 113     kITSclusterMap,      // ITS cluster map
 114     kITSnSigmaEle,           // number of sigmas to the dE/dx electron line in the ITS
 115     kITSnSigmaPio,           // number of sigmas to the dE/dx pion line in the ITS
 116     kITSnSigmaMuo,           // number of sigmas to the dE/dx muon line in the ITS
 117     kITSnSigmaKao,           // number of sigmas to the dE/dx kaon line in the ITS
 118     kITSnSigmaPro,           // number of sigmas to the dE/dx proton line in the ITS
 119
 120     kPIn,                    // momentum at inner wall of TPC (if available), used for PID
 121     kTPCsignal,              // TPC dE/dx signal
 122
 123         kTOFsignal,              // TOF signal
 124         kTOFbeta,                // TOF beta
 125
 126     kTPCnSigmaEle,           // number of sigmas to the dE/dx electron line in the TPC
 127     kTPCnSigmaPio,           // number of sigmas to the dE/dx pion line in the TPC
 128     kTPCnSigmaMuo,           // number of sigmas to the dE/dx muon line in the TPC
 129     kTPCnSigmaKao,           // number of sigmas to the dE/dx kaon line in the TPC
 130     kTPCnSigmaPro,           // number of sigmas to the dE/dx proton line in the TPC
 131
 132         kTOFnSigmaEle,           // number of sigmas to the pion line in the TOF
 133     kTOFnSigmaPio,           // number of sigmas to the pion line in the TOF
 134     kTOFnSigmaMuo,           // number of sigmas to the muon line in the TOF
 135     kTOFnSigmaKao,           // number of sigmas to the kaon line in the TOF
 136     kTOFnSigmaPro,           // number of sigmas to the proton line in the TOF
 137
 138     kKinkIndex0,             // kink index 0
 139
 140     kParticleMax,             //
 141     // TODO: kRNClusters ??
 142   // AliDielectronPair specific variables
 143     kChi2NDF = kParticleMax, // Chi^2/NDF
 144     kDecayLength,            // decay length
 145     kR,                      // distance to the origin
 146     kOpeningAngle,           // opening angle
 147     // helicity picture: Z-axis is considered the direction of the mother's 3-momentum vector
 148     kThetaHE,                // theta in mother's rest frame in the helicity picture
 149     kPhiHE,                  // phi in mother's rest frame in the helicity picture
 150     // Collins-Soper picture: Z-axis is considered the direction of the vectorial difference between
 151     // the 3-mom vectors of target and projectile beams
 152     kThetaCS,                // theta in mother's rest frame in Collins-Soper picture
 153     kPhiCS,                  // phi in mother's rest frame in Collins-Soper picture
 154     kLegDist,                // distance of the legs
 155     kLegDistXY,              // distance of the legs in XY
 156     kDeltaEta,         // Absolute value of Delta Eta for the legs
 157     kDeltaPhi,           // Absolute value of Delta Phi for the legs
 158     kMerr,                   // error of mass calculation
 159     kDCA,                    // distance of closest approach TODO: not implemented yet
 160     kPairType,               // type of the pair, like like sign ++ unlikesign ...
 161     kPseudoProperTime,       // pseudo proper time
 162     kPairMax,                 //
 163   // Event specific variables
 164     kXvPrim=kPairMax,        // prim vertex
 165     kYvPrim,                 // prim vertex
 166     kZvPrim,                 // prim vertex
 167     kXRes,                   // primary vertex x-resolution
 168     kYRes,                   // primary vertex y-resolution
 169     kZRes,                   // primary vertex z-resolution
 170     kNTrk,                   // number of tracks (or tracklets) TODO: ambiguous
 171     kTracks,                 // ESD tracks TODO: ambiguous
 172     kNacc,                   // Number of accepted tracks
 173     kNaccTrcklts,            // Number of accepted tracklets (MUON definition)
 174     kNch,                    // Number of charged MC tracks
 175     kCentrality,             // event centrality fraction
 176     kNevents,                // event counter
 177     kNMaxValues              //
 178     // TODO: (for A+A) ZDCEnergy, impact parameter, Iflag??
 179   };
 180
 181
 182   AliDielectronVarManager();
 183   AliDielectronVarManager(const char* name, const char* title);
 184   virtual ~AliDielectronVarManager();
 185   static void Fill(const TObject* particle, Double_t * const values);
 186   static void FillVarMCParticle2(const AliVParticle *p1, const AliVParticle *p2, Double_t * const values);
 187
 188   static void InitESDpid(Int_t type=0);
 189   static void InitAODpidUtil(Int_t type=0);
 190   static void SetESDpid(AliESDpid * const pid) {fgPIDResponse=pid;}
 191   static AliESDpid* GetESDpid() {return (AliESDpid*)fgPIDResponse;}
 192   static AliAODpidUtil* GetAODpidUtil() {return (AliAODpidUtil*)fgPIDResponse;}
 193   static void SetPIDResponse(AliPIDResponse *pidResponse) {fgPIDResponse=pidResponse;}
 194   static void SetEvent(AliVEvent * const ev);
 195   static Bool_t GetDCA(const AliAODTrack *track, Double_t d0z0[2]);
 196
 197   static const AliKFVertex* GetKFVertex() {return fgKFVertex;}
 198
 199   static const char* GetValueName(Int_t i) { return (i>=0&&i<kNMaxValues)?fgkParticleNames[i]:""; }
 200
 201   static const Double_t* GetData() {return fgData;}
 202
 203   static Double_t GetValue(ValueTypes val) {return fgData[val];}
 204 private:
 205
 206   static const char* fgkParticleNames[kNMaxValues];  //variable names
 207
 208   static void FillVarVParticle(const AliVParticle *particle,         Double_t * const values);
 209   static void FillVarESDtrack(const AliESDtrack *particle,           Double_t * const values);
 210   static void FillVarAODTrack(const AliAODTrack *particle,           Double_t * const values);
 211   static void FillVarMCParticle(const AliMCParticle *particle,       Double_t * const values);
 212   static void FillVarAODMCParticle(const AliAODMCParticle *particle, Double_t * const values);
 213   static void FillVarDielectronPair(const AliDielectronPair *pair,   Double_t * const values);
 214   static void FillVarKFParticle(const AliKFParticle *pair,   Double_t * const values);
 215
 216   static void FillVarVEvent(const AliVEvent *event,                  Double_t * const values);
 217   static void FillVarESDEvent(const AliESDEvent *event,              Double_t * const values);
 218   static void FillVarAODEvent(const AliAODEvent *event,              Double_t * const values);
 219   static void FillVarMCEvent(const AliMCEvent *event,                Double_t * const values);
 220
 221   static AliPIDResponse  *fgPIDResponse;        // PID response object
 222   static AliVEvent       *fgEvent;              // current event pointer
 223   static AliKFVertex     *fgKFVertex;           // kf vertex
 224
 225   static Double_t fgData[kNMaxValues];        //! data
 226
 227   AliDielectronVarManager(const AliDielectronVarManager &c);
 228   AliDielectronVarManager &operator=(const AliDielectronVarManager &c);
 229
 230   ClassDef(AliDielectronVarManager,1);
 231 };
 232
 233
 234 //Inline functions
 235 inline void AliDielectronVarManager::Fill(const TObject* object, Double_t * const values)
 236 {
 237   //
 238   // Main function to fill all available variables according to the type of particle
 239   //
 240   if      (object->IsA() == AliESDtrack::Class())       FillVarESDtrack(static_cast<const AliESDtrack*>(object), values);
 241   else if (object->IsA() == AliAODTrack::Class())       FillVarAODTrack(static_cast<const AliAODTrack*>(object), values);
 242   else if (object->IsA() == AliMCParticle::Class())     FillVarMCParticle(static_cast<const AliMCParticle*>(object), values);
 243   else if (object->IsA() == AliAODMCParticle::Class())  FillVarAODMCParticle(static_cast<const AliAODMCParticle*>(object), values);
 244   else if (object->IsA() == AliDielectronPair::Class()) FillVarDielectronPair(static_cast<const AliDielectronPair*>(object), values);
 245   else if (object->IsA() == AliKFParticle::Class())     FillVarKFParticle(static_cast<const AliKFParticle*>(object),values);
 246   // Main function to fill all available variables according to the type of event
 247
 248   else if (object->IsA() == AliVEvent::Class())         FillVarVEvent(static_cast<const AliVEvent*>(object), values);
 249   else if (object->IsA() == AliESDEvent::Class())       FillVarESDEvent(static_cast<const AliESDEvent*>(object), values);
 250   else if (object->IsA() == AliAODEvent::Class())       FillVarAODEvent(static_cast<const AliAODEvent*>(object), values);
 251   else if (object->IsA() == AliMCEvent::Class())        FillVarMCEvent(static_cast<const AliMCEvent*>(object), values);
 252 //   else printf(Form("AliDielectronVarManager::Fill: Type %s is not supported by AliDielectronVarManager!", object->ClassName())); //TODO: implement without object needed
 253 }
 254
 255 inline void AliDielectronVarManager::FillVarVParticle(const AliVParticle *particle, Double_t * const values)
 256 {
 257   //
 258   // Fill track information available in AliVParticle into an array
 259   //
 260   values[AliDielectronVarManager::kPx]        = particle->Px();
 261   values[AliDielectronVarManager::kPy]        = particle->Py();
 262   values[AliDielectronVarManager::kPz]        = particle->Pz();
 263   values[AliDielectronVarManager::kPt]        = particle->Pt();
 264   values[AliDielectronVarManager::kP]         = particle->P();
 265
 266   values[AliDielectronVarManager::kXv]        = particle->Xv();
 267   values[AliDielectronVarManager::kYv]        = particle->Yv();
 268   values[AliDielectronVarManager::kZv]        = particle->Zv();
 269
 270   values[AliDielectronVarManager::kOneOverPt] = (particle->Pt()>1.0e-3 ? particle->OneOverPt() : 0.0);
 271   values[AliDielectronVarManager::kPhi]       = particle->Phi();
 272   values[AliDielectronVarManager::kTheta]     = particle->Theta();
 273   values[AliDielectronVarManager::kEta]       = particle->Eta();
 274   values[AliDielectronVarManager::kY]         = particle->Y();
 275
 276   values[AliDielectronVarManager::kE]         = particle->E();
 277   values[AliDielectronVarManager::kM]         = particle->M();
 278   values[AliDielectronVarManager::kCharge]    = particle->Charge();
 279
 280   values[AliDielectronVarManager::kPdgCode]   = particle->PdgCode();
 281
 282 //   if ( fgEvent ) AliDielectronVarManager::Fill(fgEvent, values);
 283   for (Int_t i=AliDielectronVarManager::kPairMax; i<AliDielectronVarManager::kNMaxValues; ++i)
 284     values[i]=fgData[i];
 285 }
 286
 287 inline void AliDielectronVarManager::FillVarESDtrack(const AliESDtrack *particle, Double_t * const values)
 288 {
 289   //
 290   // Fill track information available for histogramming into an array
 291   //
 292
 293   // Fill common AliVParticle interface information
 294   FillVarVParticle(particle, values);
 295
 296   Double_t pidProbs[AliPID::kSPECIES];
 297   // Fill AliESDtrack interface specific information
 298   Double_t tpcNcls=particle->GetTPCNcls();
 299   Double_t tpcSignalN=particle->GetTPCsignalN();
 300   values[AliDielectronVarManager::kNclsITS]       = particle->GetNcls(0); // TODO: get rid of the plain numbers
 301   values[AliDielectronVarManager::kNclsTPC]       = tpcNcls; // TODO: get rid of the plain numbers
 302   values[AliDielectronVarManager::kNclsTPCiter1]  = particle->GetTPCNclsIter1(); // TODO: get rid of the plain numbers
 303   values[AliDielectronVarManager::kNFclsTPC]      = particle->GetTPCNclsF();
 304   values[AliDielectronVarManager::kNFclsTPCr]     = particle->GetTPCClusterInfo(2,1);
 305   values[AliDielectronVarManager::kNFclsTPCrFrac] = particle->GetTPCClusterInfo(2);
 306   values[AliDielectronVarManager::kTPCsignalN]    = tpcSignalN;
 307   values[AliDielectronVarManager::kTPCsignalNfrac]= tpcNcls>0?tpcSignalN/tpcNcls:0;
 308   values[AliDielectronVarManager::kNclsTRD]       = particle->GetNcls(2); // TODO: get rid of the plain numbers
 309   values[AliDielectronVarManager::kTRDntracklets] = particle->GetTRDntracklets(); // TODO: GetTRDtracklets/GetTRDntracklets?
 310   values[AliDielectronVarManager::kTRDpidQuality] = particle->GetTRDpidQuality();
 311   values[AliDielectronVarManager::kTrackStatus]   = (Double_t)particle->GetStatus();
 312
 313
 314   values[AliDielectronVarManager::kTPCchi2Cl] = -1;
 315   if (tpcNcls>0) values[AliDielectronVarManager::kTPCchi2Cl] = particle->GetTPCchi2() / tpcNcls;
 316   //TRD pidProbs
 317   particle->GetTRDpid(pidProbs);
 318   values[AliDielectronVarManager::kTRDprobEle]    = pidProbs[AliPID::kElectron];
 319   values[AliDielectronVarManager::kTRDprobPio]    = pidProbs[AliPID::kPion];
 320
 321   values[AliDielectronVarManager::kKinkIndex0]    = particle->GetKinkIndex(0);
 322
 323   Float_t impactParXY, impactParZ;
 324   particle->GetImpactParameters(impactParXY, impactParZ);
 325   values[AliDielectronVarManager::kImpactParXY]   = impactParXY;
 326   values[AliDielectronVarManager::kImpactParZ]    = impactParZ;
 327
 328
 329   values[AliDielectronVarManager::kPdgCode]=-1;
 330   values[AliDielectronVarManager::kPdgCodeMother]=-1;
 331   values[AliDielectronVarManager::kPdgCodeGrandMother]=-1;
 332
 333   values[AliDielectronVarManager::kNumberOfDaughters]=-999;
 334
 335   AliDielectronMC *mc=AliDielectronMC::Instance();
 336
 337   if (mc->HasMC()){
 338     if (mc->GetMCTrack(particle))
 339       values[AliDielectronVarManager::kPdgCode]=mc->GetMCTrack(particle)->PdgCode();
 340
 341     AliMCParticle *motherMC=mc->GetMCTrackMother(particle); //mother
 342     if (motherMC){
 343       values[AliDielectronVarManager::kPdgCodeMother]=motherMC->PdgCode();
 344
 345       motherMC=mc->GetMCTrackMother(motherMC);  //grand motherMC
 346       if (motherMC) values[AliDielectronVarManager::kPdgCodeGrandMother]=motherMC->PdgCode();;
 347     }
 348
 349     values[AliDielectronVarManager::kNumberOfDaughters]=mc->NumberOfDaughters(particle);
 350   } //if(mc->HasMC())
 351
 352
 353
 354   values[AliDielectronVarManager::kITSsignal]   =   particle->GetITSsignal();
 355
 356   Double_t itsdEdx[4];
 357   particle->GetITSdEdxSamples(itsdEdx);
 358
 359   values[AliDielectronVarManager::kITSsignalSSD1]   =   itsdEdx[0];
 360   values[AliDielectronVarManager::kITSsignalSSD2]   =   itsdEdx[1];
 361   values[AliDielectronVarManager::kITSsignalSDD1]   =   itsdEdx[2];
 362   values[AliDielectronVarManager::kITSsignalSDD2]   =   itsdEdx[3];
 363   values[AliDielectronVarManager::kITSclusterMap]   =   particle->GetITSClusterMap();
 364
 365   values[AliDielectronVarManager::kTrackLength]   = particle->GetIntegratedLength();
 366   //dEdx information
 367   Double_t mom = particle->GetP();
 368   const AliExternalTrackParam *in=particle->GetInnerParam();
 369   if (in) mom = in->GetP();
 370   values[AliDielectronVarManager::kPIn]=mom;
 371   values[AliDielectronVarManager::kTPCsignal]=particle->GetTPCsignal();
 372
 373   values[AliDielectronVarManager::kTOFsignal]=particle->GetTOFsignal();
 374
 375   Double_t l = particle->GetIntegratedLength();  // cm
 376   Double_t t = particle->GetTOFsignal();
 377   Double_t t0 = fgPIDResponse->GetTOFResponse().GetTimeZero(); // ps
 378
 379   if( (l < 360. || l > 800.) || (t <= 0.) || (t0 >999990.0) ) {
 380         values[AliDielectronVarManager::kTOFbeta]=0.0;
 381   }
 382   else {
 383         t -= t0; // subtract the T0
 384         l *= 0.01;  // cm ->m
 385         t *= 1e-12; //ps -> s
 386
 387         Double_t v = l / t;
 388         Float_t beta = v / TMath::C();
 389         values[AliDielectronVarManager::kTOFbeta]=beta;
 390   }
 391
 392   // nsigma to Electron band
 393   // TODO: for the moment we set the bethe bloch parameters manually
 394   //       this should be changed in future!
 395
 396   values[AliDielectronVarManager::kTPCnSigmaEle]=fgPIDResponse->NumberOfSigmasTPC(particle,AliPID::kElectron)-AliDielectronPID::GetCorrVal();
 397   values[AliDielectronVarManager::kTPCnSigmaPio]=fgPIDResponse->NumberOfSigmasTPC(particle,AliPID::kPion);
 398   values[AliDielectronVarManager::kTPCnSigmaMuo]=fgPIDResponse->NumberOfSigmasTPC(particle,AliPID::kMuon);
 399   values[AliDielectronVarManager::kTPCnSigmaKao]=fgPIDResponse->NumberOfSigmasTPC(particle,AliPID::kKaon);
 400   values[AliDielectronVarManager::kTPCnSigmaPro]=fgPIDResponse->NumberOfSigmasTPC(particle,AliPID::kProton);
 401
 402   values[AliDielectronVarManager::kITSnSigmaEle]=fgPIDResponse->NumberOfSigmasITS(particle,AliPID::kElectron);
 403   values[AliDielectronVarManager::kITSnSigmaPio]=fgPIDResponse->NumberOfSigmasITS(particle,AliPID::kPion);
 404   values[AliDielectronVarManager::kITSnSigmaMuo]=fgPIDResponse->NumberOfSigmasITS(particle,AliPID::kMuon);
 405   values[AliDielectronVarManager::kITSnSigmaKao]=fgPIDResponse->NumberOfSigmasITS(particle,AliPID::kKaon);
 406   values[AliDielectronVarManager::kITSnSigmaPro]=fgPIDResponse->NumberOfSigmasITS(particle,AliPID::kProton);
 407
 408   values[AliDielectronVarManager::kTOFnSigmaEle]=fgPIDResponse->NumberOfSigmasTOF(particle,AliPID::kElectron);
 409   values[AliDielectronVarManager::kTOFnSigmaPio]=fgPIDResponse->NumberOfSigmasTOF(particle,AliPID::kPion);
 410   values[AliDielectronVarManager::kTOFnSigmaMuo]=fgPIDResponse->NumberOfSigmasTOF(particle,AliPID::kMuon);
 411   values[AliDielectronVarManager::kTOFnSigmaKao]=fgPIDResponse->NumberOfSigmasTOF(particle,AliPID::kKaon);
 412   values[AliDielectronVarManager::kTOFnSigmaPro]=fgPIDResponse->NumberOfSigmasTOF(particle,AliPID::kProton);
 413 }
 414
 415 inline void AliDielectronVarManager::FillVarAODTrack(const AliAODTrack *particle, Double_t * const values)
 416 {
 417   //
 418   // Fill track information available for histogramming into an array
 419   //
 420
 421   // Fill common AliVParticle interface information
 422   FillVarVParticle(particle, values);
 423
 424   Double_t tpcNcls=particle->GetTPCNcls();
 425   // Reset AliESDtrack interface specific information
 426   values[AliDielectronVarManager::kNclsITS]       = 0;
 427   values[AliDielectronVarManager::kNclsTPC]       = tpcNcls;
 428   values[AliDielectronVarManager::kNclsTPCiter1]  = tpcNcls; // not really available in AOD
 429   values[AliDielectronVarManager::kNFclsTPC]      = 0;
 430   values[AliDielectronVarManager::kNFclsTPCr]     = 0;
 431   values[AliDielectronVarManager::kNFclsTPCrFrac] = 0;
 432   values[AliDielectronVarManager::kNclsTRD]       = 0;
 433   values[AliDielectronVarManager::kTRDntracklets] = 0;
 434   values[AliDielectronVarManager::kTRDpidQuality] = 0;
 435
 436   values[AliDielectronVarManager::kTPCchi2Cl] = -1;
 437   values[AliDielectronVarManager::kTrackStatus]   = (Double_t)particle->GetStatus();
 438
 439   //TRD pidProbs
 440   //TODO: set correctly
 441   values[AliDielectronVarManager::kTRDprobEle]    = 0;
 442   values[AliDielectronVarManager::kTRDprobPio]    = 0;
 443
 444   values[AliDielectronVarManager::kTPCsignalN]    = 0;
 445   values[AliDielectronVarManager::kTPCsignalNfrac]= 0;
 446
 447   // Fill AliAODTrack interface information
 448   //
 449   Double_t d0z0[2];
 450   GetDCA(particle, d0z0);
 451   values[AliDielectronVarManager::kImpactParXY]   = d0z0[0];
 452   values[AliDielectronVarManager::kImpactParZ]    = d0z0[1];
 453
 454   values[AliDielectronVarManager::kPIn]=0;
 455   values[AliDielectronVarManager::kTPCsignal]=0;
 456
 457   values[AliDielectronVarManager::kTOFsignal]=0;
 458   values[AliDielectronVarManager::kTOFbeta]=0;
 459
 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
 466   values[AliDielectronVarManager::kITSclusterMap]   =   particle->GetITSClusterMap();
 467
 468   AliAODPid *pid=particle->GetDetPid();
 469   if (pid){
 470     Double_t mom =pid->GetTPCmomentum();
 471     Double_t tpcSignalN=pid->GetTPCsignalN();
 472     values[AliDielectronVarManager::kTPCsignalN] = tpcSignalN;
 473     values[AliDielectronVarManager::kTPCsignalN] = tpcNcls>0?tpcSignalN/tpcNcls:0;
 474     Double_t tpcNsigmaEle=fgPIDResponse->NumberOfSigmasTPC(particle,AliPID::kElectron);
 475     Double_t tpcNsigmaPio=fgPIDResponse->NumberOfSigmasTPC(particle,AliPID::kPion);
 476     Double_t tpcNsigmaMuo=fgPIDResponse->NumberOfSigmasTPC(particle,AliPID::kMuon);
 477     Double_t tpcNsigmaKao=fgPIDResponse->NumberOfSigmasTPC(particle,AliPID::kKaon);
 478     Double_t tpcNsigmaPro=fgPIDResponse->NumberOfSigmasTPC(particle,AliPID::kProton);
 479
 480     values[AliDielectronVarManager::kPIn]=mom;
 481     values[AliDielectronVarManager::kTPCsignal]=pid->GetTPCsignal();
 482
 483     values[AliDielectronVarManager::kTPCnSigmaEle]=tpcNsigmaEle;
 484     values[AliDielectronVarManager::kTPCnSigmaPio]=tpcNsigmaPio;
 485     values[AliDielectronVarManager::kTPCnSigmaMuo]=tpcNsigmaMuo;
 486     values[AliDielectronVarManager::kTPCnSigmaKao]=tpcNsigmaKao;
 487     values[AliDielectronVarManager::kTPCnSigmaPro]=tpcNsigmaPro;
 488
 489     values[AliDielectronVarManager::kTRDntracklets] = 0;
 490     values[AliDielectronVarManager::kTRDpidQuality] = 0;
 491
 492   }
 493
 494   values[AliDielectronVarManager::kPdgCode]=-1;
 495   values[AliDielectronVarManager::kPdgCodeMother]=-1;
 496   values[AliDielectronVarManager::kPdgCodeGrandMother]=-1;
 497
 498   values[AliDielectronVarManager::kNumberOfDaughters]=-999;
 499
 500   AliDielectronMC *mc=AliDielectronMC::Instance();
 501
 502   if (mc->HasMC()){
 503     if (mc->GetMCTrack(particle))
 504       values[AliDielectronVarManager::kPdgCode]=mc->GetMCTrack(particle)->PdgCode();
 505
 506     AliAODMCParticle *motherMC=mc->GetMCTrackMother(particle); //mother
 507     if (motherMC){
 508       values[AliDielectronVarManager::kPdgCodeMother]=motherMC->PdgCode();
 509
 510       motherMC=mc->GetMCTrackMother(motherMC);  //grand motherMC
 511       if (motherMC) values[AliDielectronVarManager::kPdgCodeGrandMother]=motherMC->PdgCode();;
 512     }
 513
 514     values[AliDielectronVarManager::kNumberOfDaughters]=mc->NumberOfDaughters(particle);
 515   } //if(mc->HasMC())
 516
 517 }
 518
 519 inline void AliDielectronVarManager::FillVarMCParticle(const AliMCParticle *particle, Double_t * const values)
 520 {
 521   //
 522   // Fill track information available for histogramming into an array
 523   //
 524
 525   values[AliDielectronVarManager::kNclsITS]       = 0;
 526   values[AliDielectronVarManager::kNclsTPC]       = 0;
 527   values[AliDielectronVarManager::kNclsTPCiter1]  = 0;
 528   values[AliDielectronVarManager::kNFclsTPC]      = 0;
 529   values[AliDielectronVarManager::kNFclsTPCr]     = 0;
 530   values[AliDielectronVarManager::kNFclsTPCrFrac] = 0;
 531   values[AliDielectronVarManager::kNclsTRD]       = 0;
 532   values[AliDielectronVarManager::kTRDntracklets] = 0;
 533   values[AliDielectronVarManager::kTRDpidQuality] = 0;
 534   values[AliDielectronVarManager::kTPCchi2Cl]     = 0;
 535   values[AliDielectronVarManager::kTrackStatus]   = 0;
 536   values[AliDielectronVarManager::kTRDprobEle]    = 0;
 537   values[AliDielectronVarManager::kTRDprobPio]    = 0;
 538   values[AliDielectronVarManager::kTPCsignalN]    = 0;
 539   values[AliDielectronVarManager::kTPCsignalNfrac]    = 0;
 540   values[AliDielectronVarManager::kImpactParXY]   = 0;
 541   values[AliDielectronVarManager::kImpactParZ]    = 0;
 542   values[AliDielectronVarManager::kPIn]           = 0;
 543   values[AliDielectronVarManager::kTPCsignal]     = 0;
 544   values[AliDielectronVarManager::kTOFsignal]     = 0;
 545   values[AliDielectronVarManager::kTOFbeta]       = 0;
 546   values[AliDielectronVarManager::kTPCnSigmaEle]  = 0;
 547   values[AliDielectronVarManager::kTPCnSigmaPio]  = 0;
 548   values[AliDielectronVarManager::kTPCnSigmaMuo]  = 0;
 549   values[AliDielectronVarManager::kTPCnSigmaKao]  = 0;
 550   values[AliDielectronVarManager::kTPCnSigmaPro]  = 0;
 551   values[AliDielectronVarManager::kITSclusterMap] = 0;
 552
 553   values[AliDielectronVarManager::kPdgCode]       = -1;
 554   values[AliDielectronVarManager::kPdgCodeMother] = -1;
 555   values[AliDielectronVarManager::kPdgCodeGrandMother] = -1;
 556
 557   // Fill common AliVParticle interface information
 558   FillVarVParticle(particle, values);
 559
 560   AliDielectronMC *mc=AliDielectronMC::Instance();
 561
 562   // Fill AliMCParticle interface specific information
 563   values[AliDielectronVarManager::kPdgCode] = particle->PdgCode();
 564
 565   AliMCParticle *motherMC = mc->GetMCTrackMother(particle);
 566   if (motherMC){
 567     values[AliDielectronVarManager::kPdgCodeMother]=motherMC->PdgCode();
 568     motherMC=mc->GetMCTrackMother(motherMC);  //grand mother
 569     if (motherMC) values[AliDielectronVarManager::kPdgCodeGrandMother]=motherMC->PdgCode();;
 570   }
 571
 572   values[AliDielectronVarManager::kIsJpsiPrimary] = mc->IsJpsiPrimary(particle);
 573   values[AliDielectronVarManager::kNumberOfDaughters]=mc->NumberOfDaughters(particle);
 574 }
 575
 576
 577 inline void AliDielectronVarManager::FillVarMCParticle2(const AliVParticle *p1, const AliVParticle *p2, Double_t * const values) {
 578   //
 579   // fill 2 track information starting from MC legs
 580   //
 581
 582   values[AliDielectronVarManager::kNclsITS]       = 0;
 583   values[AliDielectronVarManager::kNclsTPC]       = 0;
 584   values[AliDielectronVarManager::kNclsTPCiter1]  = 0;
 585   values[AliDielectronVarManager::kNFclsTPC]      = 0;
 586   values[AliDielectronVarManager::kNFclsTPCr]     = 0;
 587   values[AliDielectronVarManager::kNFclsTPCrFrac] = 0;
 588   values[AliDielectronVarManager::kNclsTRD]       = 0;
 589   values[AliDielectronVarManager::kTRDntracklets] = 0;
 590   values[AliDielectronVarManager::kTRDpidQuality] = 0;
 591   values[AliDielectronVarManager::kTPCchi2Cl]     = 0;
 592   values[AliDielectronVarManager::kTrackStatus]   = 0;
 593   values[AliDielectronVarManager::kTRDprobEle]    = 0;
 594   values[AliDielectronVarManager::kTRDprobPio]    = 0;
 595   values[AliDielectronVarManager::kTPCsignalN]    = 0;
 596   values[AliDielectronVarManager::kTPCsignalNfrac]    = 0;
 597   values[AliDielectronVarManager::kImpactParXY]   = 0;
 598   values[AliDielectronVarManager::kImpactParZ]    = 0;
 599   values[AliDielectronVarManager::kPIn]           = 0;
 600   values[AliDielectronVarManager::kTPCsignal]     = 0;
 601   values[AliDielectronVarManager::kTPCnSigmaEle]  = 0;
 602   values[AliDielectronVarManager::kTPCnSigmaPio]  = 0;
 603   values[AliDielectronVarManager::kTPCnSigmaMuo]  = 0;
 604   values[AliDielectronVarManager::kTPCnSigmaKao]  = 0;
 605   values[AliDielectronVarManager::kTPCnSigmaPro]  = 0;
 606   values[AliDielectronVarManager::kITSclusterMap] = 0;
 607
 608   values[AliDielectronVarManager::kPdgCode]       = 0;
 609   values[AliDielectronVarManager::kPdgCodeMother] = 0;
 610
 611   AliDielectronMC *mc=AliDielectronMC::Instance();
 612   AliVParticle* mother=0x0;
 613   Int_t mLabel1 = mc->GetMothersLabel(p1->GetLabel());
 614   Int_t mLabel2 = mc->GetMothersLabel(p2->GetLabel());
 615   if(mLabel1==mLabel2)
 616     mother = mc->GetMCTrackFromMCEvent(mLabel1);
 617   if(mother)     // same mother
 618     FillVarVParticle(mother, values);
 619   else {         // the 2 particles come from different mothers so 2-particles quantities are calculated here
 620     // AliVParticle part
 621     values[AliDielectronVarManager::kPx]        = p1->Px()+p2->Px();
 622     values[AliDielectronVarManager::kPy]        = p1->Py()+p2->Py();
 623     values[AliDielectronVarManager::kPz]        = p1->Pz()+p2->Pz();
 624     values[AliDielectronVarManager::kPt]        = TMath::Sqrt(values[AliDielectronVarManager::kPx]*
 625                                                               values[AliDielectronVarManager::kPx]+
 626                                                               values[AliDielectronVarManager::kPy]*
 627                                                               values[AliDielectronVarManager::kPy]);
 628     values[AliDielectronVarManager::kP]         = TMath::Sqrt(values[AliDielectronVarManager::kPt]*
 629                                                               values[AliDielectronVarManager::kPt]+
 630                                                               values[AliDielectronVarManager::kPz]*
 631                                                               values[AliDielectronVarManager::kPz]);
 632
 633     values[AliDielectronVarManager::kXv]        = 0;
 634     values[AliDielectronVarManager::kYv]        = 0;
 635     values[AliDielectronVarManager::kZv]        = 0;
 636
 637     values[AliDielectronVarManager::kOneOverPt] = (values[AliDielectronVarManager::kPt]>1.0e-6 ? 1.0/values[AliDielectronVarManager::kPt] : 0.0);
 638     values[AliDielectronVarManager::kPhi]       = TMath::ATan2(values[AliDielectronVarManager::kPy],values[AliDielectronVarManager::kPx]);
 639     values[AliDielectronVarManager::kTheta]     = TMath::ATan2(values[AliDielectronVarManager::kPt],values[AliDielectronVarManager::kPz]);
 640     values[AliDielectronVarManager::kEta]       = ((values[AliDielectronVarManager::kP]-values[AliDielectronVarManager::kPz])>1.0e-6 && (values[AliDielectronVarManager::kP]+values[AliDielectronVarManager::kPz])>1.0e-6 ? 0.5*TMath::Log((values[AliDielectronVarManager::kP]+values[AliDielectronVarManager::kPz])/(values[AliDielectronVarManager::kP]-values[AliDielectronVarManager::kPz])) : -9999.);
 641     values[AliDielectronVarManager::kE]         = p1->E()+p2->E();
 642     values[AliDielectronVarManager::kY]         = ((values[AliDielectronVarManager::kE]-values[AliDielectronVarManager::kPz])>1.0e-6 && (values[AliDielectronVarManager::kE]+values[AliDielectronVarManager::kPz])>1.0e-6 ? 0.5*TMath::Log((values[AliDielectronVarManager::kE]+values[AliDielectronVarManager::kPz])/(values[AliDielectronVarManager::kE]-values[AliDielectronVarManager::kPz])) : -9999.);
 643     values[AliDielectronVarManager::kCharge]    = p1->Charge()+p2->Charge();
 644
 645     values[AliDielectronVarManager::kM]         = p1->M()*p1->M()+p2->M()*p2->M()+
 646       2.0*(p1->E()*p2->E()-p1->Px()*p2->Px()-p1->Py()*p2->Py()-p1->Pz()*p2->Pz());
 647
 648     if ( fgEvent ) AliDielectronVarManager::Fill(fgEvent, values);
 649   }
 650 }
 651
 652
 653 inline void AliDielectronVarManager::FillVarAODMCParticle(const AliAODMCParticle *particle, Double_t * const values)
 654 {
 655   //
 656   // Fill track information available for histogramming into an array
 657   //
 658
 659   values[AliDielectronVarManager::kNclsITS]       = 0;
 660   values[AliDielectronVarManager::kNclsTPC]       = 0;
 661   values[AliDielectronVarManager::kNclsTPCiter1]  = 0;
 662   values[AliDielectronVarManager::kNFclsTPC]      = 0;
 663   values[AliDielectronVarManager::kNclsTRD]       = 0;
 664   values[AliDielectronVarManager::kTRDntracklets] = 0;
 665   values[AliDielectronVarManager::kTRDpidQuality] = 0;
 666   values[AliDielectronVarManager::kTPCchi2Cl]     = 0;
 667   values[AliDielectronVarManager::kTrackStatus]   = 0;
 668   values[AliDielectronVarManager::kTRDprobEle]    = 0;
 669   values[AliDielectronVarManager::kTRDprobPio]    = 0;
 670   values[AliDielectronVarManager::kTPCsignalN]    = 0;
 671   values[AliDielectronVarManager::kTPCsignalNfrac]= 0;
 672   values[AliDielectronVarManager::kImpactParXY]   = 0;
 673   values[AliDielectronVarManager::kImpactParZ]    = 0;
 674   values[AliDielectronVarManager::kPIn]           = 0;
 675   values[AliDielectronVarManager::kTPCsignal]     = 0;
 676   values[AliDielectronVarManager::kTPCnSigmaEle]  = 0;
 677   values[AliDielectronVarManager::kTPCnSigmaPio]  = 0;
 678   values[AliDielectronVarManager::kTPCnSigmaMuo]  = 0;
 679   values[AliDielectronVarManager::kTPCnSigmaKao]  = 0;
 680   values[AliDielectronVarManager::kTPCnSigmaPro]  = 0;
 681   values[AliDielectronVarManager::kITSclusterMap] = 0;
 682
 683   values[AliDielectronVarManager::kPdgCode]       = -1;
 684   values[AliDielectronVarManager::kPdgCodeMother] = -1;
 685   values[AliDielectronVarManager::kPdgCodeGrandMother] = -1;
 686
 687   // Fill common AliVParticle interface information
 688   FillVarVParticle(particle, values);
 689
 690   AliDielectronMC *mc=AliDielectronMC::Instance();
 691
 692
 693   // Fill AliAODMCParticle interface specific information
 694   values[AliDielectronVarManager::kPdgCode] = particle->PdgCode();
 695
 696   AliAODMCParticle *motherMC = mc->GetMCTrackMother(particle);
 697   if (motherMC){
 698     values[AliDielectronVarManager::kPdgCodeMother]=motherMC->PdgCode();
 699     motherMC=mc->GetMCTrackMother(motherMC);  //grand mother
 700     if (motherMC) values[AliDielectronVarManager::kPdgCodeGrandMother]=motherMC->PdgCode();;
 701   }
 702
 703
 704   values[AliDielectronVarManager::kIsJpsiPrimary] = mc->IsJpsiPrimary(particle);
 705
 706   values[AliDielectronVarManager::kNumberOfDaughters]=mc->NumberOfDaughters(particle);
 707 }
 708
 709 inline void AliDielectronVarManager::FillVarDielectronPair(const AliDielectronPair *pair, Double_t * const values)
 710 {
 711   //
 712   // Fill pair information available for histogramming into an array
 713   //
 714
 715   values[AliDielectronVarManager::kPdgCode]=-1;
 716   values[AliDielectronVarManager::kPdgCodeMother]=-1;
 717   values[AliDielectronVarManager::kPdgCodeGrandMother]=-1;
 718
 719   // Fill common AliVParticle interface information
 720   FillVarVParticle(pair, values);
 721
 722   // Fill AliDielectronPair specific information
 723   const AliKFParticle &kfPair = pair->GetKFParticle();
 724
 725   Double_t thetaHE=0;
 726   Double_t phiHE=0;
 727   Double_t thetaCS=0;
 728   Double_t phiCS=0;
 729
 730   pair->GetThetaPhiCM(thetaHE,phiHE,thetaCS,phiCS);
 731
 732   values[AliDielectronVarManager::kChi2NDF]      = kfPair.GetChi2()/kfPair.GetNDF();
 733   values[AliDielectronVarManager::kDecayLength]  = kfPair.GetDecayLength();
 734   values[AliDielectronVarManager::kR]            = kfPair.GetR();
 735   values[AliDielectronVarManager::kOpeningAngle] = pair->OpeningAngle();
 736   values[AliDielectronVarManager::kThetaHE]      = thetaHE;
 737   values[AliDielectronVarManager::kPhiHE]        = phiHE;
 738   values[AliDielectronVarManager::kThetaCS]      = thetaCS;
 739   values[AliDielectronVarManager::kPhiCS]        = phiCS;
 740   values[AliDielectronVarManager::kLegDist]      = pair->DistanceDaughters();
 741   values[AliDielectronVarManager::kLegDistXY]    = pair->DistanceDaughtersXY();
 742   values[AliDielectronVarManager::kDeltaEta]     = pair->DeltaEta();
 743   values[AliDielectronVarManager::kDeltaPhi]     = pair->DeltaPhi();
 744   values[AliDielectronVarManager::kMerr]         = kfPair.GetErrMass()>1e-30&&kfPair.GetMass()>1e-30?kfPair.GetErrMass()/kfPair.GetMass():1000000;
 745   values[AliDielectronVarManager::kPairType]     = pair->GetType();
 746   values[AliDielectronVarManager::kPseudoProperTime] = pair->GetPseudoProperTime(fgEvent->GetPrimaryVertex());
 747
 748
 749   AliDielectronMC *mc=AliDielectronMC::Instance();
 750
 751   if (mc->HasMC()){
 752     Bool_t samemother =  mc->HaveSameMother(pair);
 753     values[AliDielectronVarManager::kIsJpsiPrimary] = mc->IsJpsiPrimary(pair);
 754     values[AliDielectronVarManager::kHaveSameMother] = samemother ;
 755
 756   }//if (mc->HasMC())
 757
 758
 759 }
 760
 761 inline void AliDielectronVarManager::FillVarKFParticle(const AliKFParticle *particle, Double_t * const values)
 762 {
 763   //
 764   // Fill track information available in AliVParticle into an array
 765   //
 766   values[AliDielectronVarManager::kPx]        = particle->GetPx();
 767   values[AliDielectronVarManager::kPy]        = particle->GetPy();
 768   values[AliDielectronVarManager::kPz]        = particle->GetPz();
 769   values[AliDielectronVarManager::kPt]        = particle->GetPt();
 770   values[AliDielectronVarManager::kP]         = particle->GetP();
 771
 772   values[AliDielectronVarManager::kXv]        = particle->GetX();
 773   values[AliDielectronVarManager::kYv]        = particle->GetY();
 774   values[AliDielectronVarManager::kZv]        = particle->GetZ();
 775
 776   values[AliDielectronVarManager::kOneOverPt] = 0;
 777   values[AliDielectronVarManager::kPhi]       = particle->GetPhi();
 778   values[AliDielectronVarManager::kTheta]     = 0.;
 779   values[AliDielectronVarManager::kEta]       = particle->GetEta();
 780   values[AliDielectronVarManager::kY]         = ((particle->GetE()*particle->GetE()-particle->GetPx()*particle->GetPx()-particle->GetPy()*particle->GetPy()-particle->GetPz()*particle->GetPz())>0.) ? TLorentzVector(particle->GetPx(),particle->GetPy(),particle->GetPz(),particle->GetE()).Rapidity() : -1111.;
 781
 782   values[AliDielectronVarManager::kE]         = particle->GetE();
 783   values[AliDielectronVarManager::kM]         = particle->GetMass();
 784   values[AliDielectronVarManager::kCharge]    = particle->GetQ();
 785
 786   values[AliDielectronVarManager::kNclsITS]       = 0;
 787   values[AliDielectronVarManager::kNclsTPC]       = 0;
 788   values[AliDielectronVarManager::kNclsTPCiter1]  = 0;
 789   values[AliDielectronVarManager::kNFclsTPC]      = 0;
 790   values[AliDielectronVarManager::kNclsTRD]       = 0;
 791   values[AliDielectronVarManager::kTRDntracklets] = 0;
 792   values[AliDielectronVarManager::kTRDpidQuality] = 0;
 793   values[AliDielectronVarManager::kTPCchi2Cl]     = 0;
 794   values[AliDielectronVarManager::kTrackStatus]   = 0;
 795   values[AliDielectronVarManager::kTRDprobEle]    = 0;
 796   values[AliDielectronVarManager::kTRDprobPio]    = 0;
 797   values[AliDielectronVarManager::kTPCsignalN]    = 0;
 798   values[AliDielectronVarManager::kTPCsignalNfrac]= 0;
 799   values[AliDielectronVarManager::kImpactParXY]   = 0;
 800   values[AliDielectronVarManager::kImpactParZ]    = 0;
 801   values[AliDielectronVarManager::kPIn]           = 0;
 802   values[AliDielectronVarManager::kTPCsignal]     = 0;
 803   values[AliDielectronVarManager::kTOFsignal]     = 0;
 804   values[AliDielectronVarManager::kTOFbeta]       = 0;
 805   values[AliDielectronVarManager::kTPCnSigmaEle]  = 0;
 806   values[AliDielectronVarManager::kTPCnSigmaPio]  = 0;
 807   values[AliDielectronVarManager::kTPCnSigmaMuo]  = 0;
 808   values[AliDielectronVarManager::kTPCnSigmaKao]  = 0;
 809   values[AliDielectronVarManager::kTPCnSigmaPro]  = 0;
 810   values[AliDielectronVarManager::kITSclusterMap] = 0;
 811
 812   values[AliDielectronVarManager::kPdgCode]       = -1;
 813   values[AliDielectronVarManager::kPdgCodeMother] = -1;
 814   values[AliDielectronVarManager::kPdgCodeGrandMother] = -1;
 815
 816
 817   if ( fgEvent ) AliDielectronVarManager::Fill(fgEvent, values);
 818 }
 819
 820 inline void AliDielectronVarManager::FillVarVEvent(const AliVEvent *event, Double_t * const values)
 821 {
 822   //
 823   // Fill event information available for histogramming into an array
 824   //
 825   const AliVVertex *primVtx = event->GetPrimaryVertex();
 826
 827   values[AliDielectronVarManager::kXvPrim]       = 0;
 828   values[AliDielectronVarManager::kYvPrim]       = 0;
 829   values[AliDielectronVarManager::kZvPrim]       = 0;
 830 //   values[AliDielectronVarManager::kChi2NDF]      = 0; //This is the pair value!!!
 831
 832   values[AliDielectronVarManager::kNTrk]         = 0;
 833   values[AliDielectronVarManager::kNacc]         = 0;
 834   values[AliDielectronVarManager::kNaccTrcklts]  = 0;
 835   values[AliDielectronVarManager::kNevents]      = 0; //always fill bin 0;
 836
 837   if (!primVtx) return;
 838
 839   values[AliDielectronVarManager::kXvPrim]       = primVtx->GetX();
 840   values[AliDielectronVarManager::kYvPrim]       = primVtx->GetY();
 841   values[AliDielectronVarManager::kZvPrim]       = primVtx->GetZ();
 842 //   values[AliDielectronVarManager::kChi2NDF]      = primVtx->GetChi2perNDF(); //this is the pair value
 843
 844   values[AliDielectronVarManager::kNTrk]         = event->GetNumberOfTracks();
 845   values[AliDielectronVarManager::kNacc]         = AliDielectronHelper::GetNacc(event);
 846   values[AliDielectronVarManager::kNaccTrcklts]  = AliDielectronHelper::GetNaccTrcklts(event);
 847 }
 848
 849 inline void AliDielectronVarManager::FillVarESDEvent(const AliESDEvent *event, Double_t * const values)
 850 {
 851   //
 852   // Fill event information available for histogramming into an array
 853   //
 854
 855   // Fill common AliVEvent interface information
 856   FillVarVEvent(event, values);
 857
 858   Double_t centralityF=-1;
 859   AliCentrality *esdCentrality = const_cast<AliESDEvent*>(event)->GetCentrality();
 860   if (esdCentrality) centralityF = esdCentrality->GetCentralityPercentile("V0M");
 861
 862   // Fill AliESDEvent interface specific information
 863   const AliESDVertex *primVtx = event->GetPrimaryVertex();
 864   values[AliDielectronVarManager::kXRes]       = primVtx->GetXRes();
 865   values[AliDielectronVarManager::kYRes]       = primVtx->GetYRes();
 866   values[AliDielectronVarManager::kZRes]       = primVtx->GetZRes();
 867   values[AliDielectronVarManager::kCentrality] = centralityF;
 868 }
 869
 870 inline void AliDielectronVarManager::FillVarAODEvent(const AliAODEvent *event, Double_t * const values)
 871 {
 872   //
 873   // Fill event information available for histogramming into an array
 874   //
 875
 876   // Fill common AliVEvent interface information
 877   FillVarVEvent(event, values);
 878
 879   // Fill AliAODEvent interface specific information
 880 }
 881
 882 inline void AliDielectronVarManager::FillVarMCEvent(const AliMCEvent *event, Double_t * const values)
 883 {
 884   //
 885   // Fill event information available for histogramming into an array
 886   //
 887
 888   // Fill common AliVEvent interface information
 889   FillVarVEvent(event, values);
 890
 891   // Fill AliMCEvent interface specific information
 892   values[AliDielectronVarManager::kNch] = AliDielectronHelper::GetNch(event, 1.6);
 893 }
 894
 895 inline void AliDielectronVarManager::InitESDpid(Int_t type)
 896 {
 897   //
 898   // initialize PID parameters
 899   // type=0 is simulation
 900   // type=1 is data
 901
 902   if (!fgPIDResponse) fgPIDResponse=new AliESDpid((Bool_t)(type==0));
 903   Double_t alephParameters[5];
 904   // simulation
 905   alephParameters[0] = 2.15898e+00/50.;
 906   alephParameters[1] = 1.75295e+01;
 907   alephParameters[2] = 3.40030e-09;
 908   alephParameters[3] = 1.96178e+00;
 909   alephParameters[4] = 3.91720e+00;
 910   fgPIDResponse->GetTOFResponse().SetTimeResolution(80.);
 911
 912   // data
 913   if (type==1){
 914     alephParameters[0] = 0.0283086/0.97;
 915     alephParameters[1] = 2.63394e+01;
 916     alephParameters[2] = 5.04114e-11;
 917     alephParameters[3] = 2.12543e+00;
 918     alephParameters[4] = 4.88663e+00;
 919     fgPIDResponse->GetTOFResponse().SetTimeResolution(130.);
 920     fgPIDResponse->GetTPCResponse().SetMip(50.);
 921   }
 922
 923   fgPIDResponse->GetTPCResponse().SetBetheBlochParameters(
 924     alephParameters[0],alephParameters[1],alephParameters[2],
 925     alephParameters[3],alephParameters[4]);
 926
 927   fgPIDResponse->GetTPCResponse().SetSigma(3.79301e-03, 2.21280e+04);
 928 }
 929
 930 inline void AliDielectronVarManager::InitAODpidUtil(Int_t type)
 931 {
 932   if (!fgPIDResponse) fgPIDResponse=new AliAODpidUtil;
 933   Double_t alephParameters[5];
 934   // simulation
 935   alephParameters[0] = 2.15898e+00/50.;
 936   alephParameters[1] = 1.75295e+01;
 937   alephParameters[2] = 3.40030e-09;
 938   alephParameters[3] = 1.96178e+00;
 939   alephParameters[4] = 3.91720e+00;
 940   fgPIDResponse->GetTOFResponse().SetTimeResolution(80.);
 941
 942   // data
 943   if (type==1){
 944     alephParameters[0] = 0.0283086/0.97;
 945     alephParameters[1] = 2.63394e+01;
 946     alephParameters[2] = 5.04114e-11;
 947     alephParameters[3] = 2.12543e+00;
 948     alephParameters[4] = 4.88663e+00;
 949     fgPIDResponse->GetTOFResponse().SetTimeResolution(130.);
 950     fgPIDResponse->GetTPCResponse().SetMip(50.);
 951   }
 952
 953   fgPIDResponse->GetTPCResponse().SetBetheBlochParameters(
 954     alephParameters[0],alephParameters[1],alephParameters[2],
 955     alephParameters[3],alephParameters[4]);
 956
 957   fgPIDResponse->GetTPCResponse().SetSigma(3.79301e-03, 2.21280e+04);
 958 }
 959
 960
 961 inline void AliDielectronVarManager::SetEvent(AliVEvent * const ev)
 962 {
 963
 964   fgEvent = ev;
 965   if (fgKFVertex) delete fgKFVertex;
 966   fgKFVertex=0x0;
 967   if (ev && ev->GetPrimaryVertex()) fgKFVertex=new AliKFVertex(*ev->GetPrimaryVertex());
 968
 969   for (Int_t i=0; i<AliDielectronVarManager::kNMaxValues;++i) fgData[i]=0.;
 970   AliDielectronVarManager::Fill(fgEvent, fgData);
 971 }
 972
 973
 974 inline Bool_t AliDielectronVarManager::GetDCA(const AliAODTrack *track, Double_t d0z0[2])
 975 {
 976   if(track->TestBit(AliAODTrack::kIsDCA)){
 977     d0z0[0]=track->DCA();
 978     d0z0[1]=track->ZAtDCA();
 979     return kTRUE;
 980   }
 981
 982   Double_t covd0z0[3];
 983   AliAODTrack copy(*track);
 984   AliAODVertex *vtx =(AliAODVertex*)(fgEvent->GetPrimaryVertex());
 985   Double_t fBzkG = fgEvent->GetMagneticField(); // z componenent of field in kG
 986   Bool_t ok = copy.PropagateToDCA(vtx,fBzkG,kVeryBig,d0z0,covd0z0);
 987   if(!ok){
 988     d0z0[0]=-999.;
 989     d0z0[1]=-999.;
 990   }
 991   return ok;
 992 }
 993
 994 /*
 995 inline void AliDielectronVarManager::FillValues(const TParticle *particle, Double_t *values)
 996 {
 997   //
 998   // Fill track information available for histogramming into an array
 999   //
1000
1001   // Fill TParticle interface information
1002   values[AliDielectronVarManager::kPx]     = particle->Px();
1003   values[AliDielectronVarManager::kPy]     = particle->Py();
1004   values[AliDielectronVarManager::kPz]     = particle->Pz();
1005   values[AliDielectronVarManager::kPt]     = particle->Pt();
1006   values[AliDielectronVarManager::kP]      = particle->P();
1007
1008   values[AliDielectronVarManager::kXv]     = particle->Vx();
1009   values[AliDielectronVarManager::kYv]     = particle->Vy();
1010   values[AliDielectronVarManager::kZv]     = particle->Vz();
1011
1012   values[AliDielectronVarManager::kOneOverPt] = 1./particle->Pt();
1013   values[AliDielectronVarManager::kPhi]    = particle->Phi();
1014   values[AliDielectronVarManager::kTheta]  =
1015   values[AliDielectronVarManager::kEta]    = particle->Eta();
1016   values[AliDielectronVarManager::kY]      =
1017
1018   values[AliDielectronVarManager::kE]      = particle->Energy();
1019   values[AliDielectronVarManager::kM]      = particle->GetMass();
1020
1021   values[AliDielectronVarManager::kCharge] = particle->GetPDG()->Charge()/3; // uggly
1022
1023 }*/
1024
1025 #endif
1026