Update master to aliroot
[u/mrichter/AliRoot.git] / STEER / 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 <TClonesArray.h>
17 #include <TList.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 AliTrackReference;
27 class AliTransportMonitor;
28
29 class AliMC : public TVirtualMCApplication {
30 public:
31     AliMC();
32     AliMC(const char *name, const char *title);
33     virtual ~AliMC();
34     
35 //
36 //  MC Application
37 //
38    virtual  void  ConstructGeometry();
39    virtual  Bool_t  MisalignGeometry();
40    virtual  void  ConstructOpGeometry();
41    virtual  void  AddParticles();
42    virtual  void  InitGeometry();     
43    virtual  void  SetAllAlignableVolumes();     
44    virtual  void  GeneratePrimaries();
45    virtual  void  BeginEvent();
46    virtual  void  BeginPrimary();
47    virtual  void  PreTrack();
48    virtual  void  Stepping();         
49    virtual  void  PostTrack();
50    virtual  void  FinishPrimary();
51    virtual  void  FinishEvent();
52    virtual  Double_t  TrackingZmax() const {return fTrZmax;}
53    virtual  Double_t  TrackingRmax() const {return fTrRmax;}
54    virtual  Int_t   DetFromMate(Int_t i) const { return (*fImedia)[i];}
55 //
56
57    virtual  AliGenerator* Generator() const {return fGenerator;}
58    virtual  void SetGenerator(AliGenerator *generator);
59    virtual  void ResetGenerator(AliGenerator *generator);
60
61 //
62    virtual  void ReadTransPar();
63    virtual  void MediaTable();
64    virtual  void EnergySummary();
65    virtual  void FinishRun();
66    void          AddEnergyDeposit(Int_t id, Float_t edep) 
67                                        {fEventEnergy[id]+=edep;}
68    virtual  void  ResetHits();
69    virtual  void  ResetDigits();
70    virtual  void  ResetSDigits();
71    virtual  void  TrackingLimits( Float_t rmax=1.e10, Float_t zmax=1.e10)
72        {fTrRmax=rmax; fTrZmax=zmax;}
73    virtual  void  DecayLimits( Float_t rmin = -1., Float_t rmax = -1., Int_t pdg = 0)
74        {fRDecayMin = rmin; fRDecayMax = rmax; fDecayPdg = pdg;}
75    
76    virtual  void  Init();
77    virtual  void  SetTransPar(const char *filename="$(ALICE_ROOT)/data/galice.cuts");
78    //PH
79    virtual  void  AddHit(Int_t id, Int_t track, Int_t *vol, Float_t *hits) const;
80    virtual  void  AddDigit(Int_t id, Int_t *tracks, Int_t *digits) const;
81    virtual  void  AddHitList(TCollection *hitList) {if (hitList) fHitLists->Add(hitList);}
82    Int_t          GetCurrentTrackNumber() const;
83    virtual  void  DumpPart (Int_t i) const;
84    virtual  void  DumpPStack () const;
85    TList*         GetHitLists() const {return fHitLists ;}
86    Int_t          GetNtrack() const;
87    virtual  Int_t GetPrimary(Int_t track) const;
88    const TObjArray* Particles() const;
89    TParticle     *Particle(Int_t i) const;
90    virtual  void  PushTrack(Int_t done, Int_t parent, Int_t pdg, 
91                            const Float_t *pmom, const Float_t *vpos, const Float_t *polar, 
92                            Float_t tof, TMCProcess mech, Int_t &ntr,
93                            Float_t weight = 1, Int_t is = 0) const;
94    virtual  void  PushTrack(Int_t done, Int_t parent, Int_t pdg,
95                            Double_t px, Double_t py, Double_t pz, Double_t e,
96                            Double_t vx, Double_t vy, Double_t vz, Double_t tof,
97                            Double_t polx, Double_t poly, Double_t polz,
98                            TMCProcess mech, Int_t &ntr, Float_t weight=1,
99                            Int_t is = 0) const;
100    virtual  void  SetHighWaterMark(Int_t nt) const;
101    
102    virtual  void  KeepTrack(Int_t itra) const;
103    virtual  void  FlagTrack(Int_t track) const;
104    virtual  void  SetCurrentTrack(Int_t track) const;                           
105 // Track reference related
106    AliTrackReference* AddTrackReference(Int_t label, Int_t id = -999);
107    // const TClonesArray*      TrackReferences()   const {return &fTrackReferences;}
108    virtual void       RemapTrackReferencesIDs(const Int_t *map); //remaping track references MI
109    virtual void       ResetTrackReferences();
110    virtual void       FixParticleDecaytime();
111 // Geometry related
112    void           SetGeometryFromCDB();
113    Bool_t         IsGeometryFromCDB() const;
114 // Monitor transport   
115    void           SetUseMonitoring(Bool_t flag=kTRUE)      { fUseMonitoring = flag; }
116    AliTransportMonitor *GetTransportMonitor() const        { return fMonitor; }
117 // Random number generator status
118    void           SetSaveRndmStatus(Bool_t value)          { fSaveRndmStatus = value; }  
119    void           SetSaveRndmStatusPerEvent(Bool_t value)  { fSaveRndmEventStatus = value; }  
120    void           SetReadRndmStatus(Bool_t value)          { fReadRndmStatus = value; };   
121    void           SetRndmFileName(const TString& fileName) { fRndmFileName = fileName; };   
122
123  private:
124    AliMC(const AliMC&); // Not implemented
125    AliMC& operator= (const AliMC&); // Not implemented
126    void MakeTmpTrackRefsTree();
127    void ReorderAndExpandTreeTR();
128  private:
129    void RemapHits();
130    AliGenerator  *fGenerator;         //  Generator used in the MC
131    Bool_t         fSaveRndmStatus;    //! Options to save random engine status
132    Bool_t         fSaveRndmEventStatus; //! Options to save random engine status for each event
133    Bool_t         fReadRndmStatus;    //! Options to read random engine status
134    Bool_t         fUseMonitoring;     //! Activate monitoring
135    TString        fRndmFileName;      //! The file name of random engine status to be read in
136    TArrayF        fEventEnergy;       //! Energy deposit for current event
137    TArrayF        fSummEnergy;        //! Energy per event in each volume
138    TArrayF        fSum2Energy;        //! Energy squared per event in each volume
139    Float_t        fTrRmax;            //  Maximum radius for tracking
140    Float_t        fTrZmax;            //  Maximu z for tracking
141    Float_t        fRDecayMax;         //  Maximum radius for decay
142    Float_t        fRDecayMin;         //  Minimum radius for decay
143    Int_t          fDecayPdg;          //  PDG code of particle with forced decay length
144    TArrayI       *fImedia;            //! Array of correspondence between media and detectors
145    TString        fTransParName;      //  Name of the transport parameters file
146    AliTransportMonitor *fMonitor;     //! Transport monitoring tool
147    TList         *fHitLists;          //! Lists of hits to be remapped by PurifyKine
148    //Temporary Track Reference tree related
149    TTree         *fTmpTreeTR;            //! Temporary track reference tree
150    TFile         *fTmpFileTR;            //! Temporary track reference file
151    TClonesArray   fTrackReferences;      //! List of track references - for one primary track only
152    TClonesArray   fTmpTrackReferences;   //! Temporary list of track references - for one primary track only
153    ClassDef(AliMC, 4)
154 };
155
156  
157 #endif