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