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