| 1c5acb87 |
1 | #ifndef ALIAODPWG4PARTICLE_H |
| 2 | #define ALIAODPWG4PARTICLE_H |
| 3 | /* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. * |
| 4 | * See cxx source for full Copyright notice */ |
| 5 | |
| 6 | /* $Id: AliAODPWG4Particle.h $ */ |
| 7 | |
| 8 | //------------------------------------------------------------------------- |
| 9 | // Copy of AOD photon class, adapted for particle identification |
| 10 | // and correlations analysis |
| 11 | // Author: Yves Schutz, CERN, Gustavo Conesa, INFN |
| 12 | //------------------------------------------------------------------------- |
| 13 | |
| 14 | //-- ROOT system -- |
| 15 | #include <TLorentzVector.h> |
| 16 | class TString; |
| 17 | |
| 18 | //-- Analysis system |
| 19 | #include "AliVParticle.h" |
| 20 | |
| 21 | class AliAODPWG4Particle : public AliVParticle { |
| 22 | |
| 23 | public: |
| 24 | AliAODPWG4Particle(); |
| 25 | AliAODPWG4Particle(Double_t px, Double_t py, Double_t pz, Double_t e); |
| 26 | AliAODPWG4Particle(TLorentzVector & p); |
| 27 | virtual ~AliAODPWG4Particle(); |
| 28 | AliAODPWG4Particle(const AliAODPWG4Particle& photon); |
| 29 | AliAODPWG4Particle& operator=(const AliAODPWG4Particle& photon); |
| 30 | |
| 31 | // AliVParticle methods |
| 32 | virtual Double_t Px() const { return fMomentum->Px(); } |
| 33 | virtual Double_t Py() const { return fMomentum->Py(); } |
| 34 | virtual Double_t Pz() const { return fMomentum->Pz(); } |
| 35 | virtual Double_t Pt() const { return fMomentum->Pt(); } |
| 36 | virtual Double_t P() const { return fMomentum->P(); } |
| 37 | virtual Bool_t PxPyPz(Double_t p[3]) const { p[0] = Px(); p[1] = Py(); p[2] = Pz(); return kTRUE; } |
| 38 | virtual Double_t OneOverPt() const { return 1. / fMomentum->Pt(); } |
| 39 | virtual Double_t Phi() const; |
| 40 | virtual Double_t Theta() const { return fMomentum->Theta(); } |
| 41 | virtual Double_t E() const { return fMomentum->E(); } |
| 42 | virtual Double_t M() const { return fMomentum->M(); } |
| 43 | virtual Double_t Eta() const { return fMomentum->Eta(); } |
| 44 | virtual Double_t Y() const { return fMomentum->Rapidity();} |
| 45 | virtual Double_t Xv() const {return -999.;} // put reasonable values here |
| 46 | virtual Double_t Yv() const {return -999.;} // |
| 47 | virtual Double_t Zv() const {return -999.;} // |
| 48 | virtual Bool_t XvYvZv(Double_t x[3]) const { x[0] = Xv(); x[1] = Yv(); x[2] = Zv(); return kTRUE; } |
| 49 | virtual void Print(Option_t* /*option*/) const; |
| 50 | |
| 51 | // |
| 52 | //Dummy |
| 53 | virtual Short_t Charge() const { return 0;} |
| 54 | virtual const Double_t* PID() const { return NULL;} |
| 55 | // |
| 56 | |
| 57 | virtual Int_t GetPdg() const {return fPdg ; } |
| 58 | virtual Int_t GetTag() const {return fTag ; } |
| 59 | virtual Int_t GetLabel() const {return fLabel ; } |
| 60 | virtual Int_t GetCaloLabel (Int_t i) const {return fCaloLabel[i];} |
| 61 | virtual Int_t GetTrackLabel(Int_t i) const {return fTrackLabel[i];} |
| 62 | virtual TString GetDetector() const {return fDetector ; } |
| 63 | |
| 64 | virtual Bool_t GetDispBit(void) const {return fDisp;} |
| 65 | virtual Bool_t GetTOFBit(void) const {return fTof;} |
| 66 | virtual Bool_t GetChargedBit(void) const {return fCharged;} |
| 67 | virtual Int_t DistToBad() const {return fBadDist ;} |
| 68 | |
| 69 | virtual void SetPdg(Int_t pdg) {fPdg = pdg ; } |
| 70 | virtual void SetTag(Int_t tag) {fTag = tag ; } |
| 71 | virtual void SetLabel(Int_t l) {fLabel = l ; } |
| 72 | virtual void SetCaloLabel (Int_t a, Int_t b) {fCaloLabel [0] = a; fCaloLabel [1] = b ; } |
| 73 | virtual void SetTrackLabel(Int_t a, Int_t b) {fTrackLabel[0] = a; fTrackLabel[1] = b ; } |
| 74 | virtual void SetDetector(TString d) {fDetector = d ; } |
| 75 | |
| 76 | virtual void SetDispBit(Bool_t chi2){fDisp = chi2 ;} |
| 77 | virtual void SetTOFBit(Bool_t tof){fTof = tof ;} |
| 78 | virtual void SetChargedBit(Bool_t ch){fCharged = ch; } |
| 79 | virtual void SetDistToBad(Int_t dist){fBadDist=dist;} |
| 80 | |
| 81 | TLorentzVector * Momentum() const {return fMomentum ; } |
| 82 | virtual void SetMomentum(TLorentzVector *lv) {fMomentum = lv ; } |
| 83 | |
| 84 | Bool_t IsPIDOK(const Int_t ipid, const Int_t pdgwanted) const; |
| 85 | |
| 86 | private: |
| 87 | TLorentzVector* fMomentum; // Photon 4-momentum vector |
| 88 | Int_t fPdg; // id of particle |
| 89 | Int_t fTag; // tag of particle (decay, fragment, prompt photon) |
| 90 | Int_t fLabel; // MC label |
| 91 | Int_t fCaloLabel[2]; // CaloCluster index, 1 for photons, 2 for pi0. |
| 92 | Int_t fTrackLabel[2]; // Track lable, 1 for pions, 2 for conversion photons |
| 93 | TString fDetector; // Detector where particle was measured. |
| 94 | |
| 95 | Bool_t fDisp ; //Dispersion bit |
| 96 | Bool_t fTof ; //TOF bit |
| 97 | Bool_t fCharged ; //Charged bit |
| 98 | Int_t fBadDist ; //Distance to bad module in module units |
| 99 | |
| 100 | ClassDef(AliAODPWG4Particle,1); |
| 101 | }; |
| 102 | |
| 103 | inline Double_t AliAODPWG4Particle::Phi() const |
| 104 | { |
| 105 | // Return phi |
| 106 | Double_t phi = fMomentum->Phi(); |
| 107 | if (phi < 0.) phi += 2. * TMath::Pi(); |
| 108 | return phi; |
| 109 | } |
| 110 | |
| 111 | #endif //ALIAODPWG4PARTICLE_H |
git://git.uio.no / u / mrichter / AliRoot.git / blame