]>
Commit | Line | Data |
---|---|---|
1 | #ifndef TFLUKA | |
2 | #define TFLUKA | |
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 | // // | |
10 | // // | |
11 | // FLUKA implementation of the AliMC Interface // | |
12 | // // | |
13 | // // | |
14 | /////////////////////////////////////////////////////////////////////////////// | |
15 | ||
16 | ||
17 | #include "TVirtualMC.h" | |
18 | #include "TMCProcess.h" | |
19 | ||
20 | #include <map> | |
21 | #include <vector> | |
22 | ||
23 | //Forward declaration | |
24 | class TG4GeometryManager; | |
25 | class TG4DetConstruction; | |
26 | class TClonesArray; | |
27 | ||
28 | ||
29 | class TFluka : public TVirtualMC { | |
30 | ||
31 | public: | |
32 | TFluka(const char *title, Int_t verbosity = 0, Bool_t isRootGeometrySupported = 0); | |
33 | TFluka(); | |
34 | virtual ~TFluka(); | |
35 | ||
36 | // | |
37 | // methods for building/management of geometry | |
38 | // ------------------------------------------------ | |
39 | // | |
40 | ||
41 | // functions from GCONS | |
42 | virtual void Gfmate(Int_t imat, char *name, Float_t &a, Float_t &z, | |
43 | Float_t &dens, Float_t &radl, Float_t &absl, | |
44 | Float_t* ubuf, Int_t& nbuf); | |
45 | virtual void Gfmate(Int_t imat, char *name, Double_t &a, Double_t &z, | |
46 | Double_t &dens, Double_t &radl, Double_t &absl, | |
47 | Double_t* ubuf, Int_t& nbuf); | |
48 | ||
49 | // detector composition | |
50 | virtual void Material(Int_t& kmat, const char* name, Double_t a, | |
51 | Double_t z, Double_t dens, Double_t radl, Double_t absl, | |
52 | Float_t* buf, Int_t nwbuf); | |
53 | virtual void Material(Int_t& kmat, const char* name, Double_t a, | |
54 | Double_t z, Double_t dens, Double_t radl, Double_t absl, | |
55 | Double_t* buf, Int_t nwbuf); | |
56 | virtual void Mixture(Int_t& kmat, const char *name, Float_t *a, | |
57 | Float_t *z, Double_t dens, Int_t nlmat, Float_t *wmat); | |
58 | virtual void Mixture(Int_t& kmat, const char *name, Double_t *a, | |
59 | Double_t *z, Double_t dens, Int_t nlmat, Double_t *wmat); | |
60 | virtual void Medium(Int_t& kmed, const char *name, Int_t nmat, | |
61 | Int_t isvol, Int_t ifield, Double_t fieldm, Double_t tmaxfd, | |
62 | Double_t stemax, Double_t deemax, Double_t epsil, | |
63 | Double_t stmin, Float_t* ubuf, Int_t nbuf); | |
64 | virtual void Medium(Int_t& kmed, const char *name, Int_t nmat, | |
65 | Int_t isvol, Int_t ifield, Double_t fieldm, Double_t tmaxfd, | |
66 | Double_t stemax, Double_t deemax, Double_t epsil, | |
67 | Double_t stmin, Double_t* ubuf, Int_t nbuf); | |
68 | virtual void Matrix(Int_t& krot, Double_t thetaX, Double_t phiX, | |
69 | Double_t thetaY, Double_t phiY, Double_t thetaZ, | |
70 | Double_t phiZ); | |
71 | virtual void Gstpar(Int_t itmed, const char *param, Double_t parval); | |
72 | ||
73 | // functions from GGEOM | |
74 | virtual Int_t Gsvolu(const char *name, const char *shape, Int_t nmed, | |
75 | Float_t *upar, Int_t np); | |
76 | virtual Int_t Gsvolu(const char *name, const char *shape, Int_t nmed, | |
77 | Double_t *upar, Int_t np); | |
78 | virtual void Gsdvn(const char *name, const char *mother, Int_t ndiv, | |
79 | Int_t iaxis); | |
80 | virtual void Gsdvn2(const char *name, const char *mother, Int_t ndiv, | |
81 | Int_t iaxis, Double_t c0i, Int_t numed); | |
82 | virtual void Gsdvt(const char *name, const char *mother, Double_t step, | |
83 | Int_t iaxis, Int_t numed, Int_t ndvmx); | |
84 | virtual void Gsdvt2(const char *name, const char *mother, Double_t step, | |
85 | Int_t iaxis, Double_t c0, Int_t numed, Int_t ndvmx); | |
86 | virtual void Gsord(const char *name, Int_t iax); | |
87 | virtual void Gspos(const char *name, Int_t nr, const char *mother, | |
88 | Double_t x, Double_t y, Double_t z, Int_t irot, | |
89 | const char *konly="ONLY"); | |
90 | virtual void Gsposp(const char *name, Int_t nr, const char *mother, | |
91 | Double_t x, Double_t y, Double_t z, Int_t irot, | |
92 | const char *konly, Float_t *upar, Int_t np); | |
93 | virtual void Gsposp(const char *name, Int_t nr, const char *mother, | |
94 | Double_t x, Double_t y, Double_t z, Int_t irot, | |
95 | const char *konly, Double_t *upar, Int_t np); | |
96 | virtual void Gsbool(const char* onlyVolName, const char* manyVolName); | |
97 | ||
98 | virtual void SetCerenkov(Int_t itmed, Int_t npckov, Float_t *ppckov, | |
99 | Float_t *absco, Float_t *effic, Float_t *rindex); | |
100 | virtual void SetCerenkov(Int_t itmed, Int_t npckov, Double_t *ppckov, | |
101 | Double_t *absco, Double_t *effic, Double_t *rindex); | |
102 | ||
103 | ||
104 | // functions for drawing | |
105 | virtual void DrawOneSpec(const char* name) | |
106 | {printf("WARNING: DrawOneSpec not yet implemented !\n");} | |
107 | virtual void Gsatt(const char* name, const char* att, Int_t val) | |
108 | {printf("WARNING: Gsatt not yet implemented !\n");} | |
109 | virtual void Gdraw(const char*,Double_t theta = 30, Double_t phi = 30, | |
110 | Double_t psi = 0, Double_t u0 = 10, Double_t v0 = 10, | |
111 | Double_t ul = 0.01, Double_t vl = 0.01) | |
112 | {printf("WARNING: Gdraw not yet implemented !\n");} | |
113 | ||
114 | // Euclid | |
115 | virtual void WriteEuclid(const char*, const char*, Int_t, Int_t); | |
116 | ||
117 | // get methods | |
118 | virtual Int_t VolId(const Text_t* volName) const; | |
119 | virtual const char* VolName(Int_t id) const; | |
120 | virtual Int_t NofVolumes() const {return fNVolumes;} | |
121 | virtual Int_t VolId2Mate(Int_t id) const; | |
122 | // | |
123 | // methods for physics management | |
124 | // ------------------------------------------------ | |
125 | // | |
126 | ||
127 | // set methods | |
128 | virtual void SetProcess(const char* flagName, Int_t flagValue); | |
129 | virtual void SetCut(const char* cutName, Double_t cutValue); | |
130 | virtual Double_t Xsec(char*, Double_t, Int_t, Int_t); | |
131 | ||
132 | // particle table usage | |
133 | virtual Int_t IdFromPDG(Int_t id) const; | |
134 | virtual Int_t PDGFromId(Int_t pdg) const; | |
135 | virtual void DefineParticles() | |
136 | {printf("WARNING: DefineParticles not yet implemented !\n");} | |
137 | ||
138 | // | |
139 | // methods for step management | |
140 | // ------------------------------------------------ | |
141 | // | |
142 | ||
143 | // action methods | |
144 | virtual void StopTrack() | |
145 | {printf("WARNING: StopTrack not yet implemented !\n");} | |
146 | virtual void StopEvent() | |
147 | {printf("WARNING: StopEvent not yet implemented !\n");} | |
148 | ||
149 | // set methods | |
150 | virtual void SetMaxStep(Double_t); | |
151 | virtual void SetMaxNStep(Int_t); | |
152 | virtual void SetUserDecay(Int_t); | |
153 | ||
154 | // get methods | |
155 | // tracking volume(s) | |
156 | virtual Int_t CurrentVolID(Int_t& copyNo) const; | |
157 | virtual Int_t CurrentVolOffID(Int_t off, Int_t& copyNo) const; | |
158 | virtual const char* CurrentVolName() const; | |
159 | virtual const char* CurrentVolOffName(Int_t off) const; | |
160 | virtual Int_t CurrentMaterial(Float_t &a, Float_t &z, | |
161 | Float_t &dens, Float_t &radl, Float_t &absl) const; | |
162 | virtual Int_t CurrentEvent() const | |
163 | {printf("WARNING: CurrentEvent not yet implemented !\n"); return -1;} | |
164 | virtual void Gmtod(Float_t* xm, Float_t* xd, Int_t iflag); | |
165 | ||
166 | virtual void Gmtod(Double_t* xm, Double_t* xd, Int_t iflag); | |
167 | ||
168 | virtual void Gdtom(Float_t* xd, Float_t* xm, Int_t iflag); | |
169 | ||
170 | virtual void Gdtom(Double_t* xd, Double_t* xm, Int_t iflag); | |
171 | ||
172 | virtual Double_t MaxStep() const | |
173 | {printf("WARNING: MaxStep not yet implemented !\n"); return -1.;} | |
174 | virtual Int_t GetMaxNStep() const | |
175 | {printf("WARNING: GetMaxNStep not yet implemented !\n"); return -1;} | |
176 | virtual Int_t GetMedium() const; | |
177 | ||
178 | // tracking particle | |
179 | // dynamic properties | |
180 | virtual void TrackPosition(TLorentzVector& position) const; | |
181 | virtual void TrackPosition(Double_t& x, Double_t& y, Double_t& z) const; | |
182 | virtual void TrackMomentum(TLorentzVector& momentum) const; | |
183 | virtual void TrackMomentum(Double_t& px, Double_t& py, Double_t& pz, Double_t& e) const; | |
184 | virtual Double_t TrackStep() const; | |
185 | virtual Double_t TrackLength() const; | |
186 | virtual Double_t TrackTime() const; | |
187 | virtual Double_t Edep() const; | |
188 | // static properties | |
189 | virtual Int_t TrackPid() const; | |
190 | virtual Double_t TrackCharge() const; | |
191 | virtual Double_t TrackMass() const; | |
192 | virtual Double_t Etot() const; | |
193 | // track status | |
194 | virtual Bool_t IsNewTrack() const; | |
195 | virtual Bool_t IsTrackInside() const; | |
196 | virtual Bool_t IsTrackEntering() const; | |
197 | virtual Bool_t IsTrackExiting() const; | |
198 | virtual Bool_t IsTrackOut() const; | |
199 | virtual Bool_t IsTrackDisappeared() const; | |
200 | virtual Bool_t IsTrackStop() const; | |
201 | virtual Bool_t IsTrackAlive() const; | |
202 | ||
203 | // secondaries | |
204 | virtual Int_t NSecondaries() const ; | |
205 | virtual void GetSecondary(Int_t isec, Int_t& particleId, | |
206 | TLorentzVector& position, TLorentzVector& momentum); | |
207 | virtual TMCProcess ProdProcess(Int_t isec) const ; | |
208 | virtual Int_t StepProcesses(TArrayI &proc) const | |
209 | {printf("WARNING: StepProcesses not yet implemented !\n"); return -1;} | |
210 | ||
211 | ||
212 | // | |
213 | // Geant3 specific methods | |
214 | // !!! need to be transformed to common interface | |
215 | // | |
216 | virtual void Gdopt(const char*,const char*) | |
217 | {printf("WARNING: Gdopt not yet implemented !\n");} | |
218 | virtual void SetClipBox(const char*,Double_t=-9999,Double_t=0, Double_t=-9999, | |
219 | Double_t=0,Double_t=-9999,Double_t=0) | |
220 | {printf("WARNING: SetClipBox not yet implemented !\n");} | |
221 | virtual void DefaultRange() | |
222 | {printf("WARNING: DefaultRange not yet implemented !\n");} | |
223 | virtual void Gdhead(Int_t, const char*, Double_t=0) | |
224 | {printf("WARNING: Gdhead not yet implemented !\n");} | |
225 | virtual void Gdman(Double_t, Double_t, const char*) | |
226 | {printf("WARNING: Gdman not yet implemented !\n");} | |
227 | virtual void SetColors() | |
228 | {printf("WARNING: SetColors not yet implemented !\n");} | |
229 | virtual void Gtreve() | |
230 | {printf("WARNING: Gtreve not yet implemented !\n");} | |
231 | virtual void GtreveRoot() | |
232 | {printf("WARNING: GtreveRoot not yet implemented !\n");} | |
233 | virtual void Gckmat(Int_t, char*) | |
234 | {printf("WARNING: Gckmat not yet implemented !\n");} | |
235 | virtual void InitLego() | |
236 | {printf("WARNING: InitLego not yet implemented !\n");} | |
237 | virtual void Gfpart(Int_t, char*, Int_t&, Float_t&, Float_t&, Float_t&) | |
238 | {printf("WARNING: Gfpart not yet implemented !\n");} | |
239 | virtual void Gspart(Int_t, const char*, Int_t, Double_t, Double_t, Double_t) | |
240 | {printf("WARNING: Gspart not yet implemented !\n");} | |
241 | ||
242 | // | |
243 | // control methods | |
244 | // ------------------------------------------------ | |
245 | // | |
246 | ||
247 | virtual void Init(); | |
248 | virtual void InitPhysics(); | |
249 | virtual void FinishGeometry(); | |
250 | virtual void BuildPhysics(); | |
251 | virtual void ProcessEvent(); | |
252 | virtual void ProcessRun(Int_t nevent); | |
253 | ||
254 | ||
255 | // | |
256 | //New Getter and Setters | |
257 | // ------------------------------------------------ | |
258 | // | |
259 | // - Core input file name | |
260 | TString GetCoreInputFileName() const {return sCoreInputFileName;} | |
261 | void SetCoreInputFileName(const char* n) {sCoreInputFileName = n;} | |
262 | ||
263 | // - Input file name | |
264 | TString GetInputFileName() const {return sInputFileName;} | |
265 | void SetInputFileName(const char* n) {sInputFileName = n;} | |
266 | ||
267 | // - SetProcess and SetCut | |
268 | Int_t GetProcessNb() const {return iNbOfProc;} | |
269 | void SetProcessNb(Int_t l) {iNbOfProc = l;} | |
270 | Int_t GetCutNb() const {return iNbOfProc;} | |
271 | void SetCutNb(Int_t l) {iNbOfCut = l;} | |
272 | ||
273 | // - Verbosity level | |
274 | Int_t GetVerbosityLevel() const {return fVerbosityLevel;} | |
275 | void SetVerbosityLevel(Int_t l) {fVerbosityLevel = l;} | |
276 | ||
277 | // - Fluka Draw procedures identifiers | |
278 | // bxdraw = 1 inside | |
279 | // bxdraw = 11 entering | |
280 | // bxdraw = 12 exiting | |
281 | // eedraw = 2 | |
282 | // endraw = 3 | |
283 | // mgdraw = 4 | |
284 | // sodraw = 5 | |
285 | // usdraw = 6 | |
286 | Int_t GetCaller() const {return iCaller;} | |
287 | void SetCaller(Int_t l) {iCaller = l;} | |
288 | ||
289 | // - Fluka Draw procedures formal parameters | |
290 | Int_t GetIcode() const {return iIcode;} | |
291 | void SetIcode(Int_t l) {iIcode = l;} | |
292 | // in the case of sodraw iIcode=0 | |
293 | ||
294 | Int_t GetMreg() const {return fCurrentFlukaRegion;} | |
295 | void SetMreg(Int_t l) {fCurrentFlukaRegion = l;} | |
296 | ||
297 | Int_t GetNewreg() const {return iNewreg;} | |
298 | void SetNewreg(Int_t l) {iNewreg = l;} | |
299 | ||
300 | Double_t GetRull() const {return fRull;} | |
301 | void SetRull(Double_t r) {fRull = r;} | |
302 | ||
303 | Double_t GetXsco() const {return fXsco;} | |
304 | void SetXsco(Double_t x) {fXsco = x;} | |
305 | ||
306 | Double_t GetYsco() const {return fYsco;} | |
307 | void SetYsco(Double_t y) {fYsco = y;} | |
308 | ||
309 | Double_t GetZsco() const {return fZsco;} | |
310 | void SetZsco(Double_t z) {fZsco = z;} | |
311 | ||
312 | void SetCurrentFlukaRegion(Int_t reg) {fCurrentFlukaRegion=reg;} | |
313 | Int_t GetCurrentFlukaRegion() const {return fCurrentFlukaRegion;} | |
314 | ||
315 | void SetTrackIsEntering(){fTrackIsEntering = kTRUE; fTrackIsExiting = kFALSE;} | |
316 | void SetTrackIsExiting() {fTrackIsExiting = kTRUE;} | |
317 | void SetTrackIsInside() {fTrackIsExiting = kFALSE; fTrackIsEntering = kFALSE;} | |
318 | ||
319 | // | |
320 | // test | |
321 | // ------------------------------------------------ | |
322 | // | |
323 | virtual void FutoTest() ; | |
324 | ||
325 | private: | |
326 | TFluka(const TFluka &mc){} | |
327 | TFluka & operator=(const TFluka &) {return (*this);} | |
328 | ||
329 | protected: | |
330 | Int_t fVerbosityLevel; //Verbosity level (0 lowest - 3 highest) | |
331 | ||
332 | TString sInputFileName; //Name of the real input file (e.g. alice.inp) | |
333 | TString sCoreInputFileName; //Name of the input file (e.g. corealice.inp) | |
334 | ||
335 | Int_t iCaller; //Parameter to indicate who is the caller of the Fluka Draw | |
336 | Int_t iIcode; //Fluka Draw procedures formal parameter | |
337 | Int_t iNewreg; //Fluka Draw procedures formal parameter | |
338 | Double_t fRull; //Fluka Draw procedures formal parameter | |
339 | Double_t fXsco; //Fluka Draw procedures formal parameter | |
340 | Double_t fYsco; //Fluka Draw procedures formal parameter | |
341 | Double_t fZsco; //Fluka Draw procedures formal parameter | |
342 | Bool_t fTrackIsEntering; // Flag for track entering | |
343 | Bool_t fTrackIsExiting; // Flag for track exiting | |
344 | ||
345 | //variables for SetProcess and SetCut | |
346 | Int_t iNbOfProc; | |
347 | Int_t iProcessValue[100]; | |
348 | Char_t sProcessFlag[100][5]; | |
349 | Int_t iNbOfCut; | |
350 | Double_t fCutValue[100]; | |
351 | Char_t sCutFlag[100][7]; | |
352 | ||
353 | ||
354 | //Geometry through Geant4 for the time being!!! | |
355 | TG4GeometryManager* fGeometryManager; //Geometry manager | |
356 | TG4DetConstruction* fDetector; //Detector | |
357 | ||
358 | TClonesArray* fVolumeMediaMap; //!Transient list of volumes | |
359 | Int_t fNVolumes; //!Current number of volumes | |
360 | Int_t* fMediaByRegion; //!Media by Fluka region | |
361 | ||
362 | ||
363 | Int_t fCurrentFlukaRegion; //Index of fluka region at each step | |
364 | ClassDef(TFluka,1) //C++ interface to Fluka montecarlo | |
365 | }; | |
366 | ||
367 | #endif //TFLUKA | |
368 |