STEER/AliMCParticle.h

   1 #ifndef AliMCParticle_H
   2 #define AliMCParticle_H
   3 /* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6 /* $Id$ */
   7
   8 //-------------------------------------------------------------------------
   9 //     AliVParticle realisation for MC Particles
  10 //     Author: Andreas Morsch, CERN
  11 //-------------------------------------------------------------------------
  12
  13 #include <Rtypes.h>
  14 #include <TParticle.h>
  15 #include <TParticlePDG.h>
  16 #include <TRefArray.h>
  17
  18 #include "AliTrackReference.h"
  19 #include "AliVParticle.h"
  20
  21 class AliMCParticle: public AliVParticle {
  22 public:
  23     AliMCParticle();
  24     AliMCParticle(TParticle* part, TRefArray* rarray = 0);
  25     virtual ~AliMCParticle() {}
  26     AliMCParticle(const AliMCParticle& mcPart);
  27     AliMCParticle& operator=(const AliMCParticle& mcPart);
  28
  29     // Kinematics
  30     virtual Double_t Px()        const;
  31     virtual Double_t Py()        const;
  32     virtual Double_t Pz()        const;
  33     virtual Double_t Pt()        const;
  34     virtual Double_t P()         const;
  35     virtual Bool_t   PxPyPz(Double_t p[3]) const;
  36
  37     virtual Double_t OneOverPt() const;
  38     virtual Double_t Phi()       const;
  39     virtual Double_t Theta()     const;
  40
  41     virtual Double_t Xv() const;
  42     virtual Double_t Yv() const;
  43     virtual Double_t Zv() const;
  44     virtual Bool_t   XvYvZv(Double_t x[3]) const;
  45
  46     virtual Double_t E()         const;
  47     virtual Double_t M()         const;
  48
  49     virtual Double_t Eta()       const;
  50     virtual Double_t Y()         const;
  51
  52     virtual Short_t Charge()     const;
  53
  54     // PID
  55     virtual const Double_t *PID() const {return 0;} // return PID object (to be defined, still)
  56
  57     // Track References
  58     Int_t              GetNumberOfTrackReferences() {return fNTrackRef;}
  59     AliTrackReference* GetTrackReference(Int_t i)
  60         {return dynamic_cast<AliTrackReference*>((*fTrackReferences)[i]);}
  61  private:
  62     TParticle *fParticle;             // The wrapped TParticle
  63     TRefArray *fTrackReferences;      // Reference array to track references
  64     Int_t      fNTrackRef;            // Number of track references
  65
  66   ClassDef(AliMCParticle,0)  // AliVParticle realisation for MCParticles
  67 };
  68
  69 inline Double_t AliMCParticle::Px()        const {return fParticle->Px();}
  70 inline Double_t AliMCParticle::Py()        const {return fParticle->Py();}
  71 inline Double_t AliMCParticle::Pz()        const {return fParticle->Pz();}
  72 inline Double_t AliMCParticle::Pt()        const {return fParticle->Pt();}
  73 inline Double_t AliMCParticle::P()         const {return fParticle->P(); }
  74 inline Double_t AliMCParticle::OneOverPt() const {return 1. / fParticle->Pt();}
  75 inline Bool_t   AliMCParticle::PxPyPz(Double_t p[3]) const { p[0] = Px(); p[1] = Py(); p[2] = Pz(); return kTRUE; }
  76 inline Double_t AliMCParticle::Phi()       const {return fParticle->Phi();}
  77 inline Double_t AliMCParticle::Theta()     const {return fParticle->Theta();}
  78 inline Double_t AliMCParticle::Xv()        const {return fParticle->Vx();}
  79 inline Double_t AliMCParticle::Yv()        const {return fParticle->Vy();}
  80 inline Double_t AliMCParticle::Zv()        const {return fParticle->Vz();}
  81 inline Bool_t   AliMCParticle::XvYvZv(Double_t x[3]) const { x[0] = Xv(); x[1] = Yv(); x[2] = Zv(); return kTRUE; }
  82 inline Double_t AliMCParticle::E()         const {return fParticle->Energy();}
  83 inline Double_t AliMCParticle::Eta()       const {return fParticle->Eta();}
  84
  85 inline Double_t AliMCParticle::M()         const
  86 {
  87     TParticlePDG* pdg = fParticle->GetPDG();
  88     if (pdg) {
  89         return (pdg->Mass());
  90     } else {
  91         return (fParticle->GetCalcMass());
  92     }
  93 }
  94
  95
  96 inline Double_t AliMCParticle::Y()         const
  97 {
  98     Double_t e  = E();
  99     Double_t pz = TMath::Abs(Pz());
 100     if (e != pz) {
 101         return 0.5*TMath::Log((e+pz)/(e-pz));
 102     } else {
 103         return -999.;
 104     }
 105 }
 106
 107 inline Short_t AliMCParticle::Charge()     const
 108 {
 109     TParticlePDG* pdg = fParticle->GetPDG();
 110     if (pdg) {
 111         return (Short_t (pdg->Charge()));
 112     } else {
 113         return -99;
 114     }
 115 }
 116
 117 #endif