| | 1 | #ifndef ALI_STACK_H |
| | 2 | #define ALI_STACK_H |
| | 3 | /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * |
| | 4 | * See cxx source for full Copyright notice */ |
| | 5 | |
| | 6 | /* $Id$ */ |
| | 7 | |
| | 8 | // Particles stack class |
| | 9 | // Implements the TMCVirtualStack of the Virtual Monte Carlo |
| | 10 | // Author A.Morsch |
| | 11 | |
| | 12 | class TClonesArray; |
| | 13 | class TFile; |
| | 14 | class TObjArray; |
| | 15 | class TParticle; |
| | 16 | class TString; |
| | 17 | class TTree; |
| | 18 | #include <TArrayI.h> |
| | 19 | #include <TVirtualMCStack.h> |
| | 20 | |
| | 21 | class AliHeader; |
| | 22 | #include "AliConfig.h" |
| | 23 | |
| | 24 | |
| | 25 | class AliStack : public TVirtualMCStack |
| | 26 | { |
| | 27 | public: |
| | 28 | // creators, destructors |
| | 29 | AliStack(Int_t size, const char* evfoldname = AliConfig::GetDefaultEventFolderName()); |
| | 30 | AliStack(); |
| | 31 | AliStack(const AliStack& st); |
| | 32 | virtual ~AliStack(); |
| | 33 | AliStack& operator=(const AliStack& st) |
| | 34 | {st.Copy(*this); return(*this);} |
| | 35 | |
| | 36 | // methods |
| | 37 | |
| | 38 | virtual void PushTrack(Int_t done, Int_t parent, Int_t pdg, |
| | 39 | Float_t *pmom, Float_t *vpos, Float_t *polar, |
| | 40 | Float_t tof, TMCProcess mech, Int_t &ntr, |
| | 41 | Float_t weight, Int_t is); |
| | 42 | |
| | 43 | virtual void PushTrack(Int_t done, Int_t parent, Int_t pdg, |
| | 44 | Double_t px, Double_t py, Double_t pz, Double_t e, |
| | 45 | Double_t vx, Double_t vy, Double_t vz, Double_t tof, |
| | 46 | Double_t polx, Double_t poly, Double_t polz, |
| | 47 | TMCProcess mech, Int_t &ntr, Double_t weight, |
| | 48 | Int_t is); |
| | 49 | |
| | 50 | virtual TParticle* PopNextTrack(Int_t& track); |
| | 51 | virtual TParticle* GetCurrentTrack() const {return fCurrentTrack;} |
| | 52 | virtual TParticle* PopPrimaryForTracking(Int_t i); |
| | 53 | |
| | 54 | void ConnectTree(); |
| | 55 | void BeginEvent(); |
| | 56 | void FinishRun(); |
| | 57 | Bool_t GetEvent(); |
| | 58 | void PurifyKine(); |
| | 59 | void ReorderKine(); |
| | 60 | void FinishEvent(); |
| | 61 | void FlagTrack(Int_t track); |
| | 62 | void KeepTrack(Int_t itrack); |
| | 63 | void Reset(Int_t size = 0); |
| | 64 | void DumpPart(Int_t i) const; |
| | 65 | void DumpPStack (); |
| | 66 | void DumpLoadedStack () const; |
| | 67 | |
| | 68 | // set methods |
| | 69 | void SetNtrack(Int_t ntrack); |
| | 70 | virtual void SetCurrentTrack(Int_t track); |
| | 71 | void SetHighWaterMark(Int_t hgwmk); |
| | 72 | // get methods |
| | 73 | virtual Int_t GetNtrack() const; |
| | 74 | Int_t GetNprimary() const; |
| | 75 | virtual Int_t GetCurrentTrackNumber() const; |
| | 76 | virtual Int_t GetCurrentParentTrackNumber() const; |
| | 77 | TObjArray* Particles() const; |
| | 78 | TParticle* Particle(Int_t id); |
| | 79 | Int_t GetPrimary(Int_t id); |
| | 80 | TTree* TreeK(); |
| | 81 | void SetEventFolderName(const char* foldname); |
| | 82 | TParticle* ParticleFromTreeK(Int_t id) const; |
| | 83 | Int_t TreeKEntry(Int_t id) const; |
| | 84 | |
| | 85 | protected: |
| | 86 | // methods |
| | 87 | void CleanParents(); |
| | 88 | void ResetArrays(Int_t size); |
| | 89 | TParticle* GetNextParticle(); |
| | 90 | Bool_t KeepPhysics(TParticle* part); |
| | 91 | |
| | 92 | private: |
| | 93 | void Copy(TObject &st) const; |
| | 94 | |
| | 95 | // data members |
| | 96 | TClonesArray *fParticles; //! Pointer to list of particles |
| | 97 | TObjArray *fParticleMap; //! Map of particles in the supporting TClonesArray |
| | 98 | TArrayI fParticleFileMap; // Map for particle ids |
| | 99 | TParticle *fParticleBuffer; //! Pointer to current particle for writing |
| | 100 | TParticle *fCurrentTrack; //! Pointer to particle currently transported |
| | 101 | TTree *fTreeK; //! Particle stack |
| | 102 | Int_t fNtrack; // Number of tracks |
| | 103 | Int_t fNprimary; // Number of primaries |
| | 104 | Int_t fCurrent; //! Last track returned from the stack |
| | 105 | Int_t fCurrentPrimary; //! Last primary track returned from the stack |
| | 106 | Int_t fHgwmk; //! Last track purified |
| | 107 | Int_t fLoadPoint; //! Next free position in the particle buffer |
| | 108 | |
| | 109 | TString fEventFolderName; //! Folder name where event is mounted |
| | 110 | ClassDef(AliStack,4) //Particles stack |
| | 111 | }; |
| | 112 | |
| | 113 | // inline |
| | 114 | |
| | 115 | inline void AliStack::SetNtrack(Int_t ntrack) |
| | 116 | { fNtrack = ntrack; } |
| | 117 | |
| | 118 | inline void AliStack::SetCurrentTrack(Int_t track) |
| | 119 | { fCurrent = track; } |
| | 120 | |
| | 121 | inline Int_t AliStack::GetNtrack() const |
| | 122 | { return fNtrack; } |
| | 123 | |
| | 124 | inline Int_t AliStack::GetNprimary() const |
| | 125 | { return fNprimary; } |
| | 126 | |
| | 127 | inline Int_t AliStack::GetCurrentTrackNumber() const |
| | 128 | { return fCurrent; } |
| | 129 | |
| | 130 | inline TObjArray* AliStack::Particles() const |
| | 131 | { return fParticleMap; } |
| | 132 | |
| | 133 | #endif //ALI_STACK_H |
git://git.uio.no / u / mrichter / AliRoot.git / blame_incremental