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