STEER/AliMC.h

   1 #ifndef ALIMC_H
   2 #define ALIMC_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 //
   9 // This is the ALICE implementation of TVirtualMCApplication
  10 // for simulation with different implementations
  11 // of the Virtual MonteCarlo
  12 //
  13
  14 #include <TArrayF.h>
  15 #include <TArrayI.h>
  16 #include <TList.h>
  17 #include <TClonesArray.h>
  18 #include <TMCProcess.h>
  19 #include <TVirtualMCApplication.h>
  20
  21 class TParticle;
  22 class TFile;
  23 class TTree;
  24
  25 class AliGenerator;
  26 class AliMCQA;
  27 class AliTrackReference;
  28
  29 class AliMC : public TVirtualMCApplication {
  30 public:
  31     AliMC();
  32     AliMC(const char *name, const char *title);
  33     AliMC(const AliMC &mc);
  34     virtual ~AliMC();
  35
  36     AliMC& operator= (const AliMC &mc) {
  37       // Assignment operator
  38       mc.Copy(*this);
  39       return *this;
  40     }
  41
  42 //
  43 //  MC Application
  44 //
  45    virtual  void  ConstructGeometry();
  46    virtual  Bool_t  MisalignGeometry();
  47    virtual  void  ConstructOpGeometry();
  48    virtual  void  InitGeometry();
  49    virtual  void  SetAllAlignableVolumes();
  50    virtual  void  GeneratePrimaries();
  51    virtual  void  BeginEvent();
  52    virtual  void  BeginPrimary();
  53    virtual  void  PreTrack();
  54    virtual  void  Stepping();
  55    virtual  void  PostTrack();
  56    virtual  void  FinishPrimary();
  57    virtual  void  FinishEvent();
  58    virtual  Double_t  TrackingZmax() const {return fTrZmax;}
  59    virtual  Double_t  TrackingRmax() const {return fTrRmax;}
  60    virtual  void Field(const Double_t* x, Double_t* b) const;
  61    virtual  Int_t   DetFromMate(Int_t i) const { return (*fImedia)[i];}
  62 //
  63
  64    virtual  AliGenerator* Generator() const {return fGenerator;}
  65    virtual  void SetGenerator(AliGenerator *generator);
  66    virtual  void ResetGenerator(AliGenerator *generator);
  67
  68 //
  69    virtual  void ReadTransPar();
  70    virtual  void MediaTable();
  71    virtual  void EnergySummary();
  72    virtual  void FinishRun();
  73    void          AddEnergyDeposit(Int_t id, Float_t edep)
  74                                        {fEventEnergy[id]+=edep;}
  75    virtual  void  ResetHits();
  76    virtual  void  TrackingLimits( Float_t rmax=1.e10, Float_t zmax=1.e10)
  77        {fTrRmax=rmax; fTrZmax=zmax;}
  78    virtual  void  DecayLimits( Float_t rmin = -1., Float_t rmax = -1., Int_t pdg = 0)
  79        {fRDecayMin = rmin; fRDecayMax = rmax; fDecayPdg = pdg;}
  80
  81    virtual  void  Init();
  82    virtual  void  SetTransPar(const char *filename="$(ALICE_ROOT)/data/galice.cuts");
  83    virtual  void  Browse(TBrowser *b);
  84    AliMCQA       *GetMCQA() const {return fMCQA;}
  85    //PH
  86    virtual  void  AddHit(Int_t id, Int_t track, Int_t *vol, Float_t *hits) const;
  87    virtual  void  AddDigit(Int_t id, Int_t *tracks, Int_t *digits) const;
  88    virtual  void  AddHitList(TCollection *hitList) {if (hitList) fHitLists->Add(hitList);}
  89    Int_t          GetCurrentTrackNumber() const;
  90    virtual  void  DumpPart (Int_t i) const;
  91    virtual  void  DumpPStack () const;
  92    TList*         GetHitLists() const {return fHitLists ;}
  93    Int_t          GetNtrack() const;
  94    virtual  Int_t GetPrimary(Int_t track) const;
  95    const TObjArray* Particles() const;
  96    TParticle     *Particle(Int_t i) const;
  97    virtual  void  PushTrack(Int_t done, Int_t parent, Int_t pdg,
  98                            Float_t *pmom, Float_t *vpos, Float_t *polar,
  99                            Float_t tof, TMCProcess mech, Int_t &ntr,
 100                            Float_t weight = 1, Int_t is = 0) const;
 101    virtual  void  PushTrack(Int_t done, Int_t parent, Int_t pdg,
 102                            Double_t px, Double_t py, Double_t pz, Double_t e,
 103                            Double_t vx, Double_t vy, Double_t vz, Double_t tof,
 104                            Double_t polx, Double_t poly, Double_t polz,
 105                            TMCProcess mech, Int_t &ntr, Float_t weight=1,
 106                            Int_t is = 0) const;
 107    virtual  void  SetHighWaterMark(Int_t nt) const;
 108
 109    virtual  void  KeepTrack(Int_t itra) const;
 110    virtual  void  FlagTrack(Int_t track) const;
 111    virtual  void  SetCurrentTrack(Int_t track) const;
 112 // Track reference related
 113    AliTrackReference* AddTrackReference(Int_t label, Int_t id = -999);
 114    // const TClonesArray*      TrackReferences()   const {return &fTrackReferences;}
 115    virtual void       RemapTrackReferencesIDs(Int_t *map); //remaping track references MI
 116    virtual void       ResetTrackReferences();
 117    virtual void       FixParticleDecaytime();
 118  private:
 119    void MakeTmpTrackRefsTree();
 120    void ReorderAndExpandTreeTR();
 121  private:
 122    void Copy (TObject &mc) const;
 123    void RemapHits();
 124    AliGenerator  *fGenerator;         //  Generator used in the MC
 125    TArrayF        fEventEnergy;       //! Energy deposit for current event
 126    TArrayF        fSummEnergy;        //! Energy per event in each volume
 127    TArrayF        fSum2Energy;        //! Energy squared per event in each volume
 128    Float_t        fTrRmax;            //  Maximum radius for tracking
 129    Float_t        fTrZmax;            //  Maximu z for tracking
 130    Float_t        fRDecayMax;         //  Maximum radius for decay
 131    Float_t        fRDecayMin;         //  Minimum radius for decay
 132    Int_t          fDecayPdg;          //  PDG code of particle with forced decay length
 133    TArrayI       *fImedia;            //! Array of correspondence between media and detectors
 134    TString        fTransParName;      //  Name of the transport parameters file
 135    AliMCQA       *fMCQA;              //  Pointer to MC Quality assurance class
 136    TList         *fHitLists;          //! Lists of hits to be remapped by PurifyKine
 137    //Temporary Track Reference tree related
 138    TTree         *fTmpTreeTR;            //! Temporary track reference tree
 139    TFile         *fTmpFileTR;            //! Temporary track reference file
 140    TClonesArray   fTrackReferences;      //! List of track references - for one primary track only
 141    TClonesArray   fTmpTrackReferences;   //! Temporary list of track references - for one primary track only
 142    ClassDef(AliMC, 4)
 143 };
 144
 145
 146 #endif