git://git.uio.no / u / mrichter / AliRoot.git / blame
| Commit | Line | Data |
|---|---|---|
| e58333e0 | 1 | #ifndef AliParticleContainer_H |
| 2 | #define AliParticleContainer_H | |
| 3 | ||
| 9239b066 | 4 | // $Id$ |
| e58333e0 | 5 | |
| e58333e0 | 6 | class AliVEvent; |
| e58333e0 | 7 | class AliVParticle; |
| 8 | ||
| bf9072ca | 9 | #include "AliAODMCParticle.h" |
| 10 | ||
| e58333e0 | 11 | #include "AliEmcalContainer.h" |
| 12 | ||
| e58333e0 | 13 | class AliParticleContainer : public AliEmcalContainer { |
| 14 | public: | |
| 15 | AliParticleContainer(); | |
| 16 | AliParticleContainer(const char *name); | |
| 6421eeb0 | 17 | virtual ~AliParticleContainer(){;} |
| e58333e0 | 18 | |
| 37509f02 | 19 | Bool_t AcceptParticle(AliVParticle *vp) const; |
| ef46ebe5 | 20 | Double_t GetParticlePtCut() const { return fParticlePtCut; } |
| 21 | Double_t GetParticleEtaMin() const { return fParticleMinEta; } | |
| 22 | Double_t GetParticleEtaMax() const { return fParticleMaxEta; } | |
| 23 | Double_t GetParticlePhiMin() const { return fParticleMinPhi; } | |
| 24 | Double_t GetParticlePhiMax() const { return fParticleMaxPhi; } | |
| 25 | AliVParticle *GetLeadingParticle(const char* opt="") ; | |
| 26 | AliVParticle *GetParticle(Int_t i) const; | |
| 27 | AliVParticle *GetAcceptParticle(Int_t i) const; | |
| 28 | AliVParticle *GetParticleWithLabel(Int_t lab) const; | |
| 29 | AliVParticle *GetAcceptParticleWithLabel(Int_t lab) const; | |
| 30 | AliVParticle *GetNextAcceptParticle(Int_t i=-1) ; | |
| 31 | AliVParticle *GetNextParticle(Int_t i=-1) ; | |
| 6421eeb0 | 32 | void GetMomentum(TLorentzVector &mom, Int_t i) const; |
| ef46ebe5 | 33 | Int_t GetNParticles() const {return GetNEntries();} |
| 8612dfc8 | 34 | Int_t GetNAcceptedParticles() ; |
| b6f970ad | 35 | void SetClassName(const char *clname); |
| 37509f02 CL |
36 | void SetMCTrackBitMap(UInt_t m) { fMCTrackBitMap = m ; } |
| 37 | void SetMinMCLabel(Int_t s) { fMinMCLabel = s ; } | |
| 38 | void SetParticlePtCut(Double_t cut) { fParticlePtCut = cut ; } | |
| 39 | void SetParticleEtaLimits(Double_t min, Double_t max) { fParticleMaxEta = max ; fParticleMinEta = min ; } | |
| 40 | void SetParticlePhiLimits(Double_t min, Double_t max) { fParticleMaxPhi = max ; fParticleMinPhi = min ; } | |
| 41 | void SetTrackBitMap(UInt_t m) { fTrackBitMap = m ; } | |
| bf9072ca | 42 | void SetMCFlag(UInt_t m) { fMCFlag = m ; } |
| 43 | void SelectHIJING(Bool_t s) { if (s) fGeneratorIndex = 0; else fGeneratorIndex = -1; } | |
| 44 | void SetGeneratorIndex(Short_t i) { fGeneratorIndex = i ; } | |
| 45 | void SelectPhysicalPrimaries(Bool_t s) { if (s) fMCFlag |= AliAODMCParticle::kPhysicalPrim ; | |
| 46 | else fMCFlag &= ~AliAODMCParticle::kPhysicalPrim ; } | |
| 82aa1543 | 47 | void SetCharge(Short_t c) { fCharge = c ; } |
| e58333e0 | 48 | |
| 49 | protected: | |
| 50 | Double_t fParticlePtCut; // cut on particle pt | |
| 51 | Double_t fParticleMinEta; // cut on particle eta | |
| 52 | Double_t fParticleMaxEta; // cut on particle eta | |
| 53 | Double_t fParticleMinPhi; // cut on particle phi | |
| 54 | Double_t fParticleMaxPhi; // cut on particle phi | |
| 55 | UInt_t fTrackBitMap; // bit map of accepted tracks (non MC) | |
| 56 | UInt_t fMCTrackBitMap; // bit map of accepted MC tracks | |
| 57 | Int_t fMinMCLabel; // minimum MC label value for the tracks/clusters being considered MC particles | |
| bf9072ca | 58 | UInt_t fMCFlag; // select MC particles with flags |
| 59 | Short_t fGeneratorIndex; // select MC particles with generator index (default = -1 = switch off selection) | |
| 82aa1543 | 60 | Short_t fCharge; // select particles with charge=fCharge |
| e58333e0 | 61 | |
| 62 | private: | |
| 63 | AliParticleContainer(const AliParticleContainer& obj); // copy constructor | |
| 64 | AliParticleContainer& operator=(const AliParticleContainer& other); // assignment | |
| 65 | ||
| 82aa1543 | 66 | ClassDef(AliParticleContainer,3); |
| e58333e0 | 67 | |
| 68 | }; | |
| 69 | ||
| 70 | #endif | |
| 71 |