AliTrackReference added, AliDisplay modified
[u/mrichter/AliRoot.git] / STEER / AliRun.h
1 #ifndef ALIRUN_H
2 #define ALIRUN_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 class TBrowser;
9 class TList;
10 class TTree;
11 class TBranch;
12 class TGeometry;
13 class TDatabasePDG;
14 class TRandom;
15 class TParticle;
16 class TFile;
17 #include <TArrayI.h>
18 #include "TClonesArray.h"
19 #include <TArrayF.h>
20 #include <TStopwatch.h>
21 #include "TNamed.h"
22
23 class AliDetector;
24 class AliModule;
25 class AliMagF;
26 class AliMC;
27 class AliLego;
28 class AliDisplay;
29 class AliLegoGenerator;
30 class AliHeader;
31 class AliGenerator;
32 class AliLegoGenerator;
33 #include "AliMCProcess.h"
34 class AliMCQA;
35 class AliStack;
36 class AliGenEventHeader;
37
38
39 enum {kKeepBit=1, kDaughtersBit=2, kDoneBit=4};
40
41
42 class AliRun : public TNamed {
43 public:
44    // Creators - distructors
45    AliRun();
46    AliRun(const char *name, const char *title);
47    virtual ~AliRun();
48
49    virtual  void  AddHit(Int_t id, Int_t track, Int_t *vol, Float_t *hits) const;
50    virtual  void  AddDigit(Int_t id, Int_t *tracks, Int_t *digits) const;
51    virtual  void  AddHitList(TCollection *hitList) {fHitLists->Add(hitList);}
52    virtual  void  Browse(TBrowser *b);
53    virtual  void  Build();
54    virtual  void  BuildSimpleGeometry();
55    virtual  void  CleanDetectors();
56    TObjArray     *Detectors() const {return fModules;}
57    TObjArray     *Modules() const {return fModules;}
58    Int_t          CurrentTrack() const;
59    AliDisplay    *Display() { return fDisplay;}
60    virtual  Int_t DistancetoPrimitive(Int_t px, Int_t py);
61    virtual  void  DumpPart (Int_t i) const;
62    virtual  void  DumpPStack () const;
63    virtual AliMagF *Field() const {return fField;}
64    virtual  void  PreTrack();
65    virtual  void  PostTrack();
66    virtual  void  FinishPrimary();
67    virtual  void  BeginPrimary();
68    virtual  void  FinishEvent();
69    virtual  void  FinishRun();
70    virtual  void  FlagTrack(Int_t track);
71    void           AddEnergyDeposit(Int_t id, Float_t edep) 
72                                        {fEventEnergy[id]+=edep;}
73    Int_t          GetEvNumber() const {return fEvent;}
74    Int_t          GetRunNumber() const {return fRun;}
75    void           SetRunNumber(Int_t run) {fRun=run;}
76    void           SetEventNrInRun(Int_t event) {fEventNrInRun=event;}
77    Int_t          GetEventNrInRun() const {return fEventNrInRun;}
78    Int_t          GetEventsPerRun() const {return fEventsPerRun;}
79    Int_t          GetDebug() const {return fDebug;}
80    AliModule     *GetModule(const char *name) const;
81    TList*         GetHitLists() const {return fHitLists ;}
82    AliDetector   *GetDetector(const char *name) const;
83    AliMCQA       *GetMCQA() const {return fMCQA;}
84    Int_t          GetModuleID(const char *name) const;
85    virtual  const char *GetBaseFile() const 
86     {return fBaseFileName.Data();}
87    virtual  Int_t GetEvent(Int_t event);
88    virtual  void  SetEvent(Int_t event) {fEvent=event;}
89    virtual  void  SetConfigFunction(const char * config="Config();");
90    virtual  const char *GetConfigFunction() const 
91     {return fConfigFunction.Data();}
92    TGeometry     *GetGeometry();
93    AliHeader*     GetHeader() {return fHeader;}
94    virtual  void  SetGenEventHeader(AliGenEventHeader* header);
95    virtual  void  GetNextTrack(Int_t &mtrack, Int_t &ipart, Float_t *pmom,
96                                Float_t &e, Float_t *vpos, Float_t *polar, 
97                                Float_t &tof);
98    Int_t          GetNtrack() const;
99    virtual  Int_t GetPrimary(Int_t track) const;
100    virtual  void  Hits2Digits(const char *detector=0); 
101    virtual  void  Hits2SDigits(const char *detector=0)   {Tree2Tree("S",detector);}
102    virtual  void  SDigits2Digits(const char *detector=0) {Tree2Tree("D",detector);}
103    virtual  void  Digits2Reco(const char *detector=0)    {Tree2Tree("R",detector);}
104    virtual  void  InitMC(const char *setup="Config.C");
105    virtual  void  Init(const char *setup="Config.C") {InitMC(setup);}
106    Bool_t         IsFolder() const {return kTRUE;}
107    virtual AliLego* Lego() const {return fLego;}
108    virtual  void  MakeTree(Option_t *option="KH", const char *file = 0);
109    void           MakeTree(Option_t *option, TFile *file);
110
111    TObjArray     *Particles();
112    TParticle     *Particle(Int_t i);
113    virtual  void  BeginEvent();
114    virtual  void  ResetDigits();
115    virtual  void  ResetSDigits();
116    virtual  void  ResetHits();
117    virtual  void  ResetTrackReferences();
118    virtual  void  ResetPoints();
119    virtual  void  SetTransPar(char *filename="$(ALICE_ROOT)/data/galice.cuts");
120    virtual  void  SetBaseFile(char *filename="galice.root");
121    virtual  void  ReadTransPar();
122    virtual  void  RunMC(Int_t nevent=1, const char *setup="Config.C");
123    virtual  void  Run(Int_t nevent=1, const char *setup="Config.C") 
124   {RunMC(nevent,setup);}
125    virtual  void  RunLego(const char *setup="Config.C",Int_t nc1=60,Float_t c1min=2,Float_t c1max=178,
126                           Int_t nc2=60,Float_t c2min=0,Float_t c2max=360,Float_t rmin=0,
127                           Float_t rmax=430,Float_t zmax=10000, AliLegoGenerator* gener=NULL);
128    virtual  Bool_t IsLegoRun() const {return (fLego!=0);}
129    virtual  void  RunReco(const char *detector=0, Int_t first = 0, Int_t last = 0);
130    virtual  void  SetCurrentTrack(Int_t track);                           
131    virtual  void  SetDebug(const Int_t level=0) {fDebug = level;}
132    virtual  void  SetDisplay(AliDisplay *display) {fDisplay = display;}
133    virtual  void  StepManager(Int_t id);
134    virtual  void  SetField(Int_t type=2, Int_t version=1, Float_t scale=1, Float_t maxField=10, char*filename="$(ALICE_ROOT)/data/field01.dat");
135    virtual  void  SetField(AliMagF* magField);
136    virtual  void  SetTrack(Int_t done, Int_t parent, Int_t pdg, 
137                            Float_t *pmom, Float_t *vpos, Float_t *polar, 
138                            Float_t tof, AliMCProcess mech, Int_t &ntr,
139                            Float_t weight = 1, Int_t is = 0);
140    virtual  void  SetTrack(Int_t done, Int_t parent, Int_t pdg,
141                            Double_t px, Double_t py, Double_t pz, Double_t e,
142                            Double_t vx, Double_t vy, Double_t vz, Double_t tof,
143                            Double_t polx, Double_t poly, Double_t polz,
144                            AliMCProcess mech, Int_t &ntr, Float_t weight=1,
145                            Int_t is = 0);
146    virtual  void  SetHighWaterMark(const Int_t nt);
147    
148    virtual  void  KeepTrack(const Int_t itra);
149    virtual  void  MediaTable();
150    virtual  Float_t TrackingZmax() const {return fTrZmax;}
151    virtual  Float_t TrackingRmax() const {return fTrRmax;}
152    virtual  void    TrackingLimits( Float_t rmax=1.e10, Float_t zmax=1.e10) {fTrRmax=rmax; fTrZmax=zmax;}
153    virtual  Int_t   DetFromMate(Int_t i) const { return (*fImedia)[i];}
154    virtual  AliGenerator* Generator() const {return fGenerator;}
155    virtual  void SetGenerator(AliGenerator *generator);
156    virtual  void ResetGenerator(AliGenerator *generator);
157    virtual  void EnergySummary();
158    virtual  TDatabasePDG* PDGDB() const {return fPDGDB;}
159
160    TFile* InitTreeFile(Option_t *option, TString fileName);
161    TString GetTreeDFileName() const {return fTreeDFileName;}
162    TString GetTreeSFileName() const {return fTreeSFileName;}
163    TString GetTreeRFileName() const {return fTreeRFileName;}
164    void SetTreeDFileName(TString fileName){fTreeDFileName=fileName;}
165    void SetTreeSFileName(TString fileName){fTreeSFileName=fileName;}
166    void SetTreeRFileName(TString fileName){fTreeRFileName=fileName;}
167    TFile* GetTreeDFile() const {return fTreeDFile;}
168    TFile* GetTreeSFile() const {return fTreeSFile;}
169    TFile* GetTreeRFile() const {return fTreeRFile;}
170    
171
172    TTree         *TreeD() {return fTreeD;}
173    TTree         *TreeS() {return fTreeS;}
174    TTree         *TreeE() {return fTreeE;}
175    TTree         *TreeH() {return fTreeH;}
176    TTree         *TreeTR() {return fTreeTR;}
177    TTree         *TreeK() ;
178    TTree         *TreeR() {return fTreeR;}
179
180    AliStack      *Stack() {return fStack;}
181
182 protected:
183   virtual  void  Tree2Tree(Option_t *option, const char *detector=0);
184   TFile* InitFile(TString fileName="");
185   Int_t          fRun;               //! Current run number
186   Int_t          fEvent;             //! Current event number (from 1)
187   Int_t          fEventNrInRun;      //! Current unique event number in run
188   Int_t          fEventsPerRun;      //  Number of events per run
189   Int_t          fDebug;             //  Debug flag
190   AliHeader     *fHeader;            //  Header information
191   TTree         *fTreeD;             //! Pointer to Tree for Digits
192   TTree         *fTreeS;             //! Pointer to Tree for SDigits
193   TTree         *fTreeH;             //! Pointer to Tree for Hits
194   TTree         *fTreeTR;            //! Pointer to Tree for TrackRefernces
195   TTree         *fTreeE;             //! Pointer to Tree for Header
196   TTree         *fTreeR;             //! Pointer to Tree for Reconstructed Objects
197   TObjArray     *fModules;           //  List of Detectors
198   TGeometry     *fGeometry;          //  Pointer to geometry
199   AliDisplay    *fDisplay;           //! Pointer to event display
200   TStopwatch     fTimer;             //  Timer object
201   AliMagF       *fField;             //  Magnetic Field Map
202   AliMC         *fMC;                //! Pointer to MonteCarlo object
203   TArrayI       *fImedia;            //! Array of correspondence between media and detectors
204   Int_t          fNdets;             //  Number of detectors
205   Float_t        fTrRmax;            //  Maximum radius for tracking
206   Float_t        fTrZmax;            //  Maximu z for tracking
207   AliGenerator  *fGenerator;         //  Generator used in the MC
208   Bool_t         fInitDone;          //! True when initialisation done
209   AliLego       *fLego;              //! Pointer to aliLego object if it exists
210   TDatabasePDG  *fPDGDB;             //  Particle factory object
211   TList         *fHitLists;          //! Lists of hits to be remapped by PurifyKine
212   TArrayF        fEventEnergy;       //! Energy deposit for current event
213   TArrayF        fSummEnergy;        //! Energy per event in each volume
214   TArrayF        fSum2Energy;        //! Energy squared per event in each volume
215   TString        fConfigFunction;    //  Configuration file to be executed
216   TRandom       *fRandom;            //  Pointer to the random number generator
217   AliMCQA       *fMCQA;              //  Pointer to MC Quality assurance class
218   TString        fTransParName;      //  Name of the transport parameters file
219   TString        fBaseFileName;      //  Name of the base root file
220   AliStack*      fStack;             //! Particle Stack
221   TString        fTreeDFileName;     //!  name of the file with TreeD
222   TFile*         fTreeDFile;         //!  file with TreeD
223   TString        fTreeSFileName;     //!  name of the file with TreeS
224   TFile*         fTreeSFile;         //!  file with TreeS
225   TString        fTreeRFileName;     //!  name of the file with TreeR
226   TFile*         fTreeRFile;         //!  file with TreeR
227 private:
228
229    AliRun(const AliRun &right) 
230      {}  
231    AliRun& operator = (const AliRun &) {return *this;}
232    ClassDef(AliRun,6)      //Supervisor class for all Alice detectors
233 };
234  
235 R__EXTERN  AliRun *gAlice;
236  
237 #endif