STEER/AliStack.h

   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 <TClonesArray.h>
  19 #include <TArrayI.h>
  20 #include <TVirtualMCStack.h>
  21
  22 class AliHeader;
  23
  24 enum {kKeepBit=1, kDaughtersBit=2, kDoneBit=4, kTransportBit=BIT(14)};
  25
  26 class AliStack : public TVirtualMCStack
  27 {
  28   public:
  29     // creators, destructors
  30     AliStack(Int_t size, const char* name = "");
  31     AliStack();
  32     AliStack(const AliStack& st);
  33     virtual ~AliStack();
  34     AliStack& operator=(const AliStack& st)
  35       {st.Copy(*this); return(*this);}
  36
  37     // methods
  38
  39     virtual void  PushTrack(Int_t done, Int_t parent, Int_t pdg,
  40                            const Float_t *pmom, const Float_t *vpos, const Float_t *polar,
  41                            Float_t tof, TMCProcess mech, Int_t &ntr,
  42                            Float_t weight, Int_t is);
  43
  44     virtual void  PushTrack(Int_t done, Int_t parent, Int_t pdg,
  45                            Double_t px, Double_t py, Double_t pz, Double_t e,
  46                            Double_t vx, Double_t vy, Double_t vz, Double_t tof,
  47                            Double_t polx, Double_t poly, Double_t polz,
  48                            TMCProcess mech, Int_t &ntr, Double_t weight,
  49                            Int_t is);
  50
  51     virtual TParticle* PopNextTrack(Int_t& track);
  52     virtual TParticle* GetCurrentTrack() const {return fCurrentTrack;}
  53     virtual TParticle* PopPrimaryForTracking(Int_t i);
  54
  55     void   ConnectTree(TTree* tree);
  56     Bool_t GetEvent();
  57     Bool_t PurifyKine();
  58     Bool_t ReorderKine();
  59     void   FinishEvent();
  60     void   FlagTrack(Int_t track);
  61     void   KeepTrack(Int_t itrack);
  62     void   Clean(Int_t size = 0);
  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     TParticle*  Particle(Int_t id);
  78     Int_t       GetPrimary(Int_t id);
  79     TTree*      TreeK() const {return fTreeK;}
  80     TParticle*  ParticleFromTreeK(Int_t id) const;
  81     Int_t       TreeKEntry(Int_t id) const;
  82     Bool_t      IsPhysicalPrimary(Int_t i);
  83     Int_t       TrackLabel(Int_t label) const {return fTrackLabelMap[label];}
  84     Int_t*      TrackLabelMap() {return fTrackLabelMap.GetArray();}
  85     const TObjArray*  Particles() const;
  86
  87   protected:
  88     // methods
  89     void  CleanParents();
  90     void  ResetArrays(Int_t size);
  91     TParticle* GetParticleMapEntry(Int_t id) const;
  92     TParticle* GetNextParticle();
  93     Bool_t KeepPhysics(const TParticle* part);
  94     Bool_t IsStable(Int_t pdg) const;
  95   private:
  96     void Copy(TObject &st) const;
  97
  98     // data members
  99     TClonesArray   fParticles;         //! Pointer to list of particles
 100     TObjArray      fParticleMap;       //! Map of particles in the supporting TClonesArray
 101     TArrayI        fParticleFileMap;   //  Map for particle ids
 102     TParticle     *fParticleBuffer;    //! Pointer to current particle for writing
 103     TParticle     *fCurrentTrack;      //! Pointer to particle currently transported
 104     TTree         *fTreeK;             //! Particle stack
 105     Int_t          fNtrack;            //  Number of tracks
 106     Int_t          fNprimary;          //  Number of primaries
 107     Int_t          fCurrent;           //! Last track returned from the stack
 108     Int_t          fCurrentPrimary;    //! Last primary track returned from the stack
 109     Int_t          fHgwmk;             //! Last track purified
 110     Int_t          fLoadPoint;         //! Next free position in the particle buffer
 111     TArrayI        fTrackLabelMap;     //! Map of track labels
 112     ClassDef(AliStack,5) //Particles stack
 113 };
 114
 115 // inline
 116
 117 inline void  AliStack::SetNtrack(Int_t ntrack)
 118 { fNtrack = ntrack; }
 119
 120 inline Int_t AliStack::GetNtrack() const
 121 { return fNtrack; }
 122
 123 inline Int_t AliStack::GetNprimary() const
 124 { return fNprimary; }
 125
 126 inline Int_t AliStack::GetCurrentTrackNumber() const
 127 { return fCurrent; }
 128
 129 inline const TObjArray* AliStack::Particles() const
 130 { return &fParticleMap; }
 131
 132 // inline protected
 133
 134 inline TParticle* AliStack::GetParticleMapEntry(Int_t id) const
 135 { return (TParticle*) fParticleMap.At(id); }
 136
 137 #endif //ALI_STACK_H