3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
10 #include <TClonesArray.h>
11 #include <TMCProcess.h>
12 #include <TStopwatch.h>
13 #include <TVirtualMCApplication.h>
14 #include <TVirtualMC.h>
27 #include "AliRunLoader.h"
30 class AliGenEventHeader;
34 class AliLegoGenerator;
35 class AliLegoGenerator;
42 enum {kKeepBit=1, kDaughtersBit=2, kDoneBit=4};
45 class AliRun : public TVirtualMCApplication {
47 // Creators - distructors
49 AliRun(const char *name, const char *title);
50 AliRun(const AliRun &arun);
53 AliRun& operator = (const AliRun &arun)
54 {arun.Copy(*this); return (*this);}
55 virtual void AddHit(Int_t id, Int_t track, Int_t *vol, Float_t *hits) const;
56 virtual void AddDigit(Int_t id, Int_t *tracks, Int_t *digits) const;
57 virtual void AddHitList(TCollection *hitList) {fHitLists->Add(hitList);}
58 virtual void Browse(TBrowser *b);
60 virtual void BuildSimpleGeometry();
61 virtual void CleanDetectors();
62 TObjArray *Detectors() const {return fModules;}
63 TObjArray *Modules() const {return fModules;}
64 Int_t GetCurrentTrackNumber() const;
65 AliDisplay *Display() const { return fDisplay;}
66 virtual Int_t DistancetoPrimitive(Int_t px, Int_t py) const;
67 virtual void DumpPart (Int_t i) const;
68 virtual void DumpPStack () const;
69 virtual AliMagF *Field() const {return fField;}
70 virtual void FinishRun();
71 virtual void FlagTrack(Int_t track);
72 void AddEnergyDeposit(Int_t id, Float_t edep)
73 {fEventEnergy[id]+=edep;}
74 void AddModule(AliModule* mod);
75 Int_t GetEvNumber() const;
76 Int_t GetRunNumber() const {return fRun;}
77 void SetRunNumber(Int_t run) {fRun=run;}
78 void SetEventNrInRun(Int_t event) {fEventNrInRun=event;}
79 Int_t GetEventNrInRun() const {return fEventNrInRun;}
80 Int_t GetEventsPerRun() const {return fEventsPerRun;}
81 Int_t GetDebug() const {return fDebug;}
82 AliModule *GetModule(const char *name) const;
83 TList* GetHitLists() const {return fHitLists ;}
84 AliDetector *GetDetector(const char *name) const;
85 AliMCQA *GetMCQA() const {return fMCQA;}
86 Int_t GetModuleID(const char *name) const;
87 virtual const char *GetBaseFile() const
88 {return fBaseFileName.Data();}
89 virtual Int_t GetEvent(Int_t event);
90 virtual void SetEvent(Int_t event) {fEvent=event;}
91 virtual void SetConfigFunction(const char * config="Config();");
92 virtual const char *GetConfigFunction() const
93 {return fConfigFunction.Data();}
94 TGeometry *GetGeometry();
95 virtual void SetGenEventHeader(AliGenEventHeader* header);
96 Int_t GetNtrack() const;
97 virtual Int_t GetPrimary(Int_t track) const;
98 virtual void Hits2Digits(const char *detector=0);
99 virtual void Hits2SDigits(const char *detector=0) {Tree2Tree("S",detector);}
100 virtual void SDigits2Digits(const char *detector=0) {Tree2Tree("D",detector);}
101 virtual void Digits2Reco(const char *detector=0) {Tree2Tree("R",detector);}
102 virtual void InitMC(const char *setup="Config.C");
103 virtual void Init(const char *setup="Config.C") {InitMC(setup);}
104 Bool_t IsFolder() const {return kTRUE;}
105 virtual AliLego* Lego() const {return fLego;}
107 TObjArray *Particles() const;
108 TParticle *Particle(Int_t i) const;
109 virtual void ResetDigits();
110 virtual void ResetSDigits();
111 virtual void ResetHits();
112 virtual void ResetTrackReferences();
113 virtual void ResetPoints();
114 virtual void SetTransPar(const char *filename="$(ALICE_ROOT)/data/galice.cuts");
115 virtual void SetBaseFile(const char *filename="galice.root");
116 virtual void ReadTransPar();
117 virtual void RunMC(Int_t nevent=1, const char *setup="Config.C");
118 virtual void Run(Int_t nevent=1, const char *setup="Config.C") {RunMC(nevent,setup);}
119 virtual void RunLego(const char *setup="Config.C",Int_t nc1=60,Float_t c1min=2,Float_t c1max=178,
120 Int_t nc2=60,Float_t c2min=0,Float_t c2max=360,Float_t rmin=0,
121 Float_t rmax=430,Float_t zmax=10000, AliLegoGenerator* gener=NULL);
122 virtual Bool_t IsLegoRun() const {return (fLego!=0);}
123 virtual void RunReco(const char *detector=0, Int_t first = 0, Int_t last = 0);
124 virtual void SetCurrentTrack(Int_t track);
125 virtual void SetDebug(const Int_t level=0) {fDebug = level;}
126 virtual void SetDisplay(AliDisplay *display) {fDisplay = display;}
127 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");
128 virtual void SetField(AliMagF* magField);
129 virtual void PushTrack(Int_t done, Int_t parent, Int_t pdg,
130 Float_t *pmom, Float_t *vpos, Float_t *polar,
131 Float_t tof, TMCProcess mech, Int_t &ntr,
132 Float_t weight = 1, Int_t is = 0);
133 virtual void PushTrack(Int_t done, Int_t parent, Int_t pdg,
134 Double_t px, Double_t py, Double_t pz, Double_t e,
135 Double_t vx, Double_t vy, Double_t vz, Double_t tof,
136 Double_t polx, Double_t poly, Double_t polz,
137 TMCProcess mech, Int_t &ntr, Float_t weight=1,
139 virtual void SetHighWaterMark(const Int_t nt);
141 virtual void KeepTrack(const Int_t itra);
142 virtual void MediaTable();
143 virtual void TrackingLimits( Float_t rmax=1.e10, Float_t zmax=1.e10) {fTrRmax=rmax; fTrZmax=zmax;}
144 virtual Int_t DetFromMate(Int_t i) const { return (*fImedia)[i];}
145 virtual AliGenerator* Generator() const {return fGenerator;}
146 virtual void SetGenerator(AliGenerator *generator);
147 virtual void ResetGenerator(AliGenerator *generator);
148 virtual void EnergySummary();
149 virtual TDatabasePDG* PDGDB() const {return fPDGDB;}
153 virtual void ConstructGeometry();
154 virtual void InitGeometry();
155 virtual void GeneratePrimaries();
156 virtual void BeginEvent();
157 virtual void BeginPrimary();
158 virtual void PreTrack();
159 virtual void Stepping();
160 virtual void PostTrack();
161 virtual void FinishPrimary();
162 virtual void FinishEvent();
163 virtual Double_t TrackingZmax() const {return fTrZmax;}
164 virtual Double_t TrackingRmax() const {return fTrRmax;}
165 virtual void Field(const Double_t* x, Double_t* b) const;
168 // End of MC Application
170 TTree *TreeE() {return (fRunLoader)?fRunLoader->TreeE():0x0;}
171 TTree *TreeK() {return (fRunLoader)?fRunLoader->TreeK():0x0;}
172 AliStack *Stack() {return (fRunLoader)?fRunLoader->Stack():0x0;}
173 AliHeader* GetHeader() {return (fRunLoader)?fRunLoader->GetHeader():0x0;}
175 TTree *TreeD() {MayNotUse("TreeD"); return 0x0;}
176 TTree *TreeS() {MayNotUse("TreeS"); return 0x0;}
177 TTree *TreeR() {MayNotUse("TreeR"); return 0x0;}
180 void SetRunLoader(AliRunLoader* rloader);
181 AliRunLoader* GetRunLoader() const {return fRunLoader;}
182 // void SetEventFolderName(const char* eventfoldername);
183 virtual void Announce() const;
185 virtual void InitLoaders(); //prepares run (i.e. creates getters)
186 static void Deprecated(TObject *obj, const char *method,
187 const char *replacement) {
189 ::Warning(Form("%s::%s", obj->ClassName(), method),
190 "method is depricated\nPlease use: %s", replacement);
192 ::Warning(method, "method is depricated\nPlease use: %s", replacement);
195 virtual void Tree2Tree(Option_t *option, const char *detector=0);
196 Int_t fRun; //! Current run number
197 Int_t fEvent; //! Current event number (from 1)
198 Int_t fEventNrInRun; //! Current unique event number in run
199 Int_t fEventsPerRun; // Number of events per run
200 Int_t fDebug; // Debug flag
201 TObjArray *fModules; // List of Detectors
202 TGeometry *fGeometry; // Pointer to geometry
203 AliDisplay *fDisplay; //! Pointer to event display
204 TStopwatch fTimer; // Timer object
205 AliMagF *fField; // Magnetic Field Map
206 TVirtualMC *fMC; //! Pointer to MonteCarlo object
207 TArrayI *fImedia; //! Array of correspondence between media and detectors
208 Int_t fNdets; // Number of detectors
209 Float_t fTrRmax; // Maximum radius for tracking
210 Float_t fTrZmax; // Maximu z for tracking
211 AliGenerator *fGenerator; // Generator used in the MC
212 Bool_t fInitDone; //! True when initialisation done
213 AliLego *fLego; //! Pointer to aliLego object if it exists
214 TDatabasePDG *fPDGDB; // Particle factory object
215 TList *fHitLists; //! Lists of hits to be remapped by PurifyKine
216 TArrayF fEventEnergy; //! Energy deposit for current event
217 TArrayF fSummEnergy; //! Energy per event in each volume
218 TArrayF fSum2Energy; //! Energy squared per event in each volume
219 TString fConfigFunction; // Configuration file to be executed
220 TRandom *fRandom; // Pointer to the random number generator
221 AliMCQA *fMCQA; // Pointer to MC Quality assurance class
222 TString fTransParName; // Name of the transport parameters file
223 TString fBaseFileName; // Name of the base root file
225 AliRunLoader *fRunLoader; //!run getter - written as a separate object
227 void Copy(AliRun &arun) const;
229 ClassDef(AliRun,8) //Supervisor class for all Alice detectors
232 R__EXTERN AliRun *gAlice;