]>
Commit | Line | Data |
---|---|---|
660eb9a2 | 1 | /************************************************************************** |
2 | * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * | |
3 | * * | |
4 | * Author: The ALICE Off-line Project. * | |
5 | * Contributors are mentioned in the code where appropriate. * | |
6 | * * | |
7 | * Permission to use, copy, modify and distribute this software and its * | |
8 | * documentation strictly for non-commercial purposes is hereby granted * | |
9 | * without fee, provided that the above copyright notice appears in all * | |
10 | * copies and that both the copyright notice and this permission notice * | |
11 | * appear in the supporting documentation. The authors make no claims * | |
12 | * about the suitability of this software for any purpose. It is * | |
13 | * provided "as is" without express or implied warranty. * | |
14 | **************************************************************************/ | |
15 | ||
eeacf08b | 16 | /* $Id$ */ |
660eb9a2 | 17 | |
eeacf08b | 18 | //---------------------------------------------------------------------- |
19 | // Helper class to write geometry in ALIFE format | |
20 | // in parallel with Geant geometry definition. | |
21 | // Author: A.Morsch | |
22 | //---------------------------------------------------------------------- | |
88cb7938 | 23 | |
660eb9a2 | 24 | #include <AliALIFE.h> |
25 | ||
26 | ClassImp(AliALIFE) | |
27 | ||
28 | AliALIFE::AliALIFE(const char* file1, const char* file2) | |
29 | { | |
30 | // Constructor | |
31 | fNBodies = 0; | |
32 | fNVolumes= 0; | |
33 | fBodyFile = file1; // File for Fluka bodies | |
34 | fVolumeFile = file2; | |
35 | fFile1=fopen(fBodyFile,"w"); | |
36 | fFile2=fopen(fVolumeFile,"w"); | |
37 | BodyHeader(); | |
38 | VolumeHeader(); | |
39 | fDefaultVolume1 = "DEFAU1"; | |
40 | fDefaultVolume2 = "DEFAU2"; | |
41 | } | |
42 | ||
43 | AliALIFE::AliALIFE() | |
44 | { | |
45 | // Default constructor | |
46 | fBodyFile = "FlukaBody.inp"; | |
47 | fVolumeFile = "FlukaVolume.inp"; | |
48 | fFile1=fopen(fBodyFile,"w"); | |
49 | fFile2=fopen(fVolumeFile,"w"); | |
50 | BodyHeader(); | |
51 | VolumeHeader(); | |
52 | } | |
53 | ||
54 | void AliALIFE::BodyHeader() | |
55 | { | |
56 | // Write header for body definitions | |
57 | fprintf(fFile1,"*\n"); | |
58 | fprintf(fFile1,"GEOBEGIN COMBINAT\n"); | |
59 | fprintf(fFile1," 0 0 AliRoot Generated\n"); | |
60 | fprintf(fFile1,"*\n"); | |
61 | fprintf(fFile1,"*== Body Definitions ================================================\n"); | |
62 | fprintf(fFile1,"*\n"); | |
63 | ||
64 | } | |
65 | ||
66 | ||
67 | void AliALIFE::VolumeHeader() | |
68 | { | |
69 | // Write header for region (volume) definitions | |
70 | fprintf(fFile2,"REGIONS\n"); | |
71 | fprintf(fFile2,"*\n"); | |
72 | fprintf(fFile2,"*== Region Definitions =================================================\n"); | |
73 | fprintf(fFile2,"*\n"); | |
74 | } | |
75 | ||
76 | ||
77 | void AliALIFE:: Cylinder(Float_t rmin, Float_t rmax, | |
78 | Float_t zmin, Float_t zmax, | |
79 | Float_t pos[3], | |
80 | char* Material, char* Field, char* Cuts) | |
81 | { | |
82 | // Simple cylinder | |
83 | // | |
84 | // Bodies | |
85 | // ^^^^^^ | |
86 | char name1[5], name2[5], name3[5], name4[5]; | |
87 | ||
88 | // outer radius | |
89 | sprintf(name1, "R%4.4d", fNBodies++); | |
90 | fprintf(fFile1,"%5s ZCC%10.3f%10.3f%10.3f\n",name1, pos[0], pos[1], rmax); // inner radius | |
91 | sprintf(name2, "R%4.4d", fNBodies++); | |
92 | fprintf(fFile1,"%5s ZCC%10.3f%10.3f%10.3f\n",name2, pos[0], pos[1], rmin); | |
93 | // z-max | |
94 | sprintf(name3, "Z%4.4d", fNBodies++); | |
95 | fprintf(fFile1,"%5s XYP%10.3f\n",name3, zmax); | |
96 | // z-min | |
97 | sprintf(name4, "Z%4.4d", fNBodies++); | |
98 | fprintf(fFile1,"%5s XYP%10.3f\n",name4, zmin); | |
99 | // | |
100 | // Volumes | |
101 | // ^^^^^^^ | |
102 | ||
103 | fprintf(fFile2,">%s:%s\n", Material, Field); | |
104 | fprintf(fFile2,"EMFCUT=%s\n", Cuts); | |
105 | fprintf(fFile2,"WW-FACTOR=%s\n", Cuts); | |
106 | if (rmin >0) { | |
107 | fprintf(fFile2,"+%s-%s+%s-%s\n", name1, name2, name3, name4); | |
108 | } else { | |
109 | fprintf(fFile2,"+%s+%s-%s\n", name1, name3, name4); | |
110 | } | |
111 | ||
112 | fprintf(fFile2,"\n"); | |
113 | } | |
114 | ||
115 | void AliALIFE::OnionCylinder(Float_t* r, Int_t nr, Float_t zmin, Float_t zmax, | |
116 | Float_t pos[3], char** Materials, | |
117 | char** Fields, char** Cuts) | |
118 | { | |
119 | // | |
120 | // Concentric cylinders | |
121 | // | |
122 | ||
123 | // Bodies | |
124 | // ^^^^^^ | |
125 | char nameRin[6], nameRou[6], nameZin[6], nameZou[6]; | |
126 | // z-limits | |
127 | // z-max | |
128 | sprintf(nameZou, "Z%4.4d", fNBodies++); | |
129 | nameZou[5]='\0'; | |
130 | ||
131 | fprintf(fFile1,"%5s XYP%10.3f\n",nameZou, zmax); | |
132 | // z-min | |
133 | sprintf(nameZin, "Z%4.4d", fNBodies++); | |
134 | nameZin[5]='\0'; | |
135 | fprintf(fFile1,"%5s XYP%10.3f\n",nameZin, zmin); | |
136 | ||
137 | // inner radius | |
138 | sprintf(nameRin, "R%4.4d", fNBodies++); | |
139 | nameRin[5]='\0'; | |
140 | fprintf(fFile1,"%5s ZCC%10.3f%10.3f%10.3f\n",nameRin, pos[0], pos[1], r[0]); | |
141 | ||
142 | Int_t i; | |
143 | for (i=1; i<nr; i++) { | |
144 | // outer radius | |
145 | sprintf(nameRou, "R%4.4d", fNBodies++); | |
146 | nameRou[5]='\0'; | |
147 | fprintf(fFile1,"%5s ZCC%10.3f%10.3f%10.3f\n", | |
148 | nameRou, pos[0], pos[1], r[i]); | |
149 | // | |
150 | // Volumes | |
151 | // ^^^^^^^ | |
152 | if (Fields && Cuts) { | |
153 | fprintf(fFile2,">%s:%s\n", Materials[i-1], Fields[i-1]); | |
154 | fprintf(fFile2,"EMFCUT=%s\n", Cuts[i-1]); | |
155 | fprintf(fFile2,"WW-FACTOR=%s\n", Cuts[i-1]); | |
156 | } else { | |
157 | fprintf(fFile2,">%s:%s\n", Materials[i-1], "MF"); | |
158 | fprintf(fFile2,"EMFCUT=%s\n", "$UNSHIELDED"); | |
159 | fprintf(fFile2,"WW-FACTOR=%s\n", "$UNSHIELDED"); | |
160 | } | |
161 | if (r[i-1] != 0.) { | |
162 | fprintf(fFile2,"+%5s-%5s+%5s-%5s\n", nameZou, nameZin, nameRou, nameRin); | |
163 | } else { | |
164 | fprintf(fFile2,"+%5s-%5s+%5s\n", nameZou, nameZin, nameRou); | |
165 | } | |
166 | fprintf(fFile2,"\n"); | |
886e839a | 167 | strcpy(nameRin,nameRou); |
660eb9a2 | 168 | } |
169 | } | |
170 | ||
171 | ||
172 | void AliALIFE::Cone(Float_t rmin1, Float_t rmin2, | |
173 | Float_t rmax1, Float_t rmax2, | |
174 | Float_t zmin, Float_t zmax, | |
175 | Float_t pos[3], | |
176 | char* Material, char* Field, char* Cuts) | |
177 | { | |
178 | // Simple cone | |
179 | ||
180 | // | |
181 | // Bodies | |
182 | // ^^^^^^ | |
183 | char nameCou[6], nameCin[6]; | |
184 | Float_t d, r1, r2; | |
185 | char nameZin[6], nameZou[6]; | |
186 | // z-max | |
187 | sprintf(nameZou, "Z%4.4d", fNBodies++); | |
188 | fprintf(fFile1,"%5s XYP%10.3f\n",nameZou, zmax); | |
189 | // z-min | |
190 | sprintf(nameZin, "Z%4.4d", fNBodies++); | |
191 | fprintf(fFile1,"%5s XYP%10.3f\n",nameZin, zmin); | |
192 | ||
193 | // outer radius | |
194 | d=zmax-zmin; | |
eeacf08b | 195 | sprintf(nameCou, "C%4.4d", fNBodies++); |
660eb9a2 | 196 | if (rmax1 >= 0. && rmax2 >= 0.) { |
197 | if (rmax1!=rmax2) { | |
198 | if (rmax1 > rmax2) { | |
199 | pos[2]=zmin; | |
200 | r1=rmax1; | |
201 | r2=rmax2; | |
202 | } else { | |
203 | d=-d; | |
204 | pos[2]=zmax; | |
205 | r1=rmax2; | |
206 | r2=rmax1; | |
207 | } | |
660eb9a2 | 208 | fprintf(fFile1,"%5s TRC%10.3f%10.3f%10.3f%10.3f%10.3f%10.3f\n", |
209 | nameCou, pos[0], pos[1], pos[2], 0., 0., d); | |
210 | fprintf(fFile1," %10.3f%10.3f\n",r1,r2); | |
211 | } else { | |
660eb9a2 | 212 | fprintf(fFile1,"%5s ZCC%10.3f%10.3f%10.3f\n", |
213 | nameCou, pos[0], pos[1], rmax1); | |
214 | } | |
215 | }else { | |
b7fab4b5 | 216 | fDefaultVolume1 = nameCou; |
660eb9a2 | 217 | } |
218 | ||
219 | ||
220 | ||
221 | // inner radius | |
eeacf08b | 222 | sprintf(nameCin, "C%4.4d", fNBodies++); |
660eb9a2 | 223 | if (rmin1 >= 0. && rmin2 >= 0.) { |
224 | if (rmin1!=rmin2) { | |
225 | if (rmin1 != 0 && rmin2 !=0) { | |
226 | d=zmax-zmin; | |
227 | if (rmin1 > rmin2) { | |
228 | pos[2]=zmin; | |
229 | r1=rmin1; | |
230 | r2=rmin2; | |
231 | } else { | |
232 | pos[2]=zmax; | |
233 | r1=rmin2; | |
234 | r2=rmin1; | |
235 | d=-d; | |
236 | } | |
660eb9a2 | 237 | fprintf(fFile1,"%5s TRC%10.3f%10.3f%10.3f%10.3f%10.3f%10.3f\n", |
238 | nameCin, pos[0], pos[1], pos[2], 0., 0., d); | |
239 | fprintf(fFile1," %10.3f%10.3f\n",r1,r2); | |
240 | } | |
241 | } else { | |
660eb9a2 | 242 | fprintf(fFile1,"%5s ZCC%10.3f%10.3f%10.3f\n", |
243 | nameCin, pos[0], pos[1], rmin1); | |
244 | } | |
245 | }else { | |
b7fab4b5 | 246 | fDefaultVolume2 = nameCin; |
660eb9a2 | 247 | } |
248 | ||
249 | ||
250 | ||
251 | // | |
252 | // Volumes | |
253 | // ^^^^^^^ | |
254 | fprintf(fFile2,">%s:%s\n", Material, Field); | |
255 | fprintf(fFile2,"EMFCUT=%s\n", Cuts); | |
256 | fprintf(fFile2,"WW-FACTOR=%s\n", Cuts); | |
257 | if (rmin1 != 0 && rmin2 !=0) { | |
258 | fprintf(fFile2,"+%s-%s+%s-%s\n", nameCou, nameCin, nameZou, nameZin); | |
259 | } else { | |
260 | fprintf(fFile2,"+%s+%s-%s\n", nameCou, nameZou, nameZin); | |
261 | } | |
262 | fprintf(fFile2,"\n"); | |
263 | } | |
264 | ||
265 | void AliALIFE::OnionCone (Float_t* r1, Float_t* r2, Int_t nr, | |
266 | Float_t zmin, Float_t zmax, | |
267 | Float_t pos[3], char** Materials, char** Fields, | |
268 | char** Cuts) | |
269 | { | |
270 | // Concentric cones | |
271 | // | |
272 | char nameCin[6], nameCou[6], nameZin[6], nameZou[6]; | |
273 | // | |
274 | // Bodies | |
275 | // ^^^^^^ | |
276 | Float_t ri1, ri2, ro1, ro2, d; | |
277 | ||
278 | // z-max | |
279 | sprintf(nameZou, "Z%4.4d", fNBodies++); | |
280 | fprintf(fFile1,"%5s XYP%10.3f\n",nameZou, zmax); | |
281 | // z-min | |
282 | sprintf(nameZin, "Z%4.4d", fNBodies++); | |
283 | fprintf(fFile1,"%5s XYP%10.3f\n",nameZin, zmin); | |
284 | ||
285 | // inner radius | |
286 | d=zmax-zmin; | |
287 | Bool_t hasInner=kFALSE; | |
288 | Bool_t isCylinder=kFALSE; | |
289 | if (r1[0]>=0 && r2[0]>=0) { | |
290 | if (r1[0] != 0. && r2[0] !=0.) { | |
291 | if (r1[0]!=r2[0]) { | |
292 | hasInner=kTRUE; | |
293 | if (r1[0] > r2[0]) { | |
294 | pos[2]=zmin; | |
295 | ri1=r1[0]; | |
296 | ri2=r2[0]; | |
297 | } else { | |
298 | d=-d; | |
299 | pos[2]=zmax; | |
300 | ri1=r2[0]; | |
301 | ri2=r1[0]; | |
302 | } | |
303 | sprintf(nameCin, "C%4.4d", fNBodies++); | |
304 | nameCin[5]='\0'; | |
305 | fprintf(fFile1,"%5s TRC%10.3f%10.3f%10.3f%10.3f%10.3f%10.3f\n", | |
306 | nameCin, pos[0], pos[1], pos[2], 0., 0., d); | |
307 | fprintf(fFile1," %10.3f%10.3f\n",ri1,ri2); | |
308 | } else { | |
309 | isCylinder=kTRUE; | |
310 | sprintf(nameCin, "C%4.4d", fNBodies++); | |
311 | nameCin[5]='\0'; | |
312 | fprintf(fFile1,"%5s ZCC%10.3f%10.3f%10.3f\n", | |
313 | nameCin, pos[0], pos[1], r1[0]); | |
314 | } | |
315 | } | |
316 | } else { | |
b7fab4b5 | 317 | fDefaultVolume1 = nameCin; |
660eb9a2 | 318 | } |
319 | ||
320 | ||
321 | ||
322 | // outer radius | |
323 | Int_t i; | |
324 | for (i=1; i<nr; i++) { | |
325 | if (r1[i] >= 0. && r2[i] >=0) { | |
326 | if (r1[i]!=r2[i]) { | |
327 | d=zmax-zmin; | |
328 | if (r1[i] > r2[i]) { | |
329 | pos[2]=zmin; | |
330 | ro1=r1[i]; | |
331 | ro2=r2[i]; | |
332 | } else { | |
333 | pos[2]=zmax; | |
334 | ro1=r2[i]; | |
335 | ro2=r1[i]; | |
336 | d=-d; | |
337 | } | |
338 | sprintf(nameCou, "C%4.4d", fNBodies++); | |
339 | nameCou[5]='\0'; | |
340 | fprintf(fFile1,"%5s TRC%10.3f%10.3f%10.3f%10.3f%10.3f%10.3f\n", | |
341 | nameCou, pos[0], pos[1], pos[2], 0., 0., d); | |
342 | fprintf(fFile1," %10.3f%10.3f\n",ro1,ro2); | |
343 | } else { | |
344 | isCylinder=kTRUE; | |
345 | sprintf(nameCou, "C%4.4d", fNBodies++); | |
346 | nameCou[5]='\0'; | |
347 | fprintf(fFile1,"%5s ZCC%10.3f%10.3f%10.3f\n", | |
348 | nameCou, pos[0], pos[1], r1[i]); | |
349 | } | |
350 | } else { | |
b7fab4b5 | 351 | fDefaultVolume1 = nameCou; |
660eb9a2 | 352 | } |
353 | ||
354 | // Volumes | |
355 | // ^^^^^^^ | |
356 | if (Fields && Cuts) { | |
357 | fprintf(fFile2,">%s:%s\n", Materials[i-1], Fields[i-1]); | |
358 | fprintf(fFile2,"EMFCUT=%s\n", Cuts[i-1]); | |
359 | fprintf(fFile2,"WW-FACTOR=%s\n", Cuts[i-1]); | |
360 | } else { | |
361 | fprintf(fFile2,">%s:%s\n", Materials[i-1], "MF"); | |
362 | fprintf(fFile2,"EMFCUT=%s\n", "$UNSHIELDED"); | |
363 | fprintf(fFile2,"WW-FACTOR=%s\n", "$UNSHIELDED"); | |
364 | } | |
38e826ad | 365 | if (hasInner || isCylinder) { |
366 | if (!isCylinder) fprintf(fFile2,"+%5s-%5s+%5s-%5s\n", | |
660eb9a2 | 367 | nameCou, nameCin, nameZou, nameZin); |
38e826ad | 368 | if (isCylinder) fprintf(fFile2,"+%5s+%5s-%5s\n", |
369 | nameCou, nameZou, nameZin); | |
660eb9a2 | 370 | } else { |
371 | fprintf(fFile2,"+%5s\n", nameCou); | |
372 | hasInner=kTRUE; | |
373 | } | |
374 | fprintf(fFile2,"\n"); | |
886e839a | 375 | strcpy(nameCin,nameCou); |
660eb9a2 | 376 | } |
377 | } | |
378 | ||
379 | ||
380 | void AliALIFE::PolyCone(Float_t* rmin, Float_t* rmax, Float_t* z, | |
381 | Int_t nz, | |
382 | Float_t pos[3], | |
383 | char* Material, char* Field, char* Cuts) | |
384 | { | |
385 | // | |
386 | // Equivalent to the Geant3 PCON | |
387 | Int_t i; | |
388 | for (i=0; i<nz-1; i++) { | |
389 | // skip (z_i = z_i+1) | |
390 | if (z[i] == z[i+1]) continue; | |
391 | Cone(rmin[i], rmin[i+1], rmax[i], rmax[i+1], z[i], z[i+1], pos, | |
392 | Material, Field, Cuts); | |
393 | } | |
394 | } | |
395 | ||
396 | void AliALIFE::OnionPolyCone(Float_t** r, Float_t* z, | |
397 | Int_t nr, Int_t nz, | |
398 | Float_t pos[3], | |
886e839a | 399 | char** Materials, char** Fields, char** Cuts) |
660eb9a2 | 400 | { |
401 | // | |
402 | // Concentric PCONS | |
403 | Int_t i, j; | |
404 | for (i=0; i<nz-1; i++) { | |
405 | // skip (z_i = z_i+1) | |
406 | if (z[i] == z[i+1]) continue; | |
407 | for (j=0; j<nr-1; j++) { | |
408 | if (Fields && Cuts) { | |
409 | Cone(r[i][j], r[i+1][j], r[i][j+1], r[i+1][j+1], | |
410 | z[i], z[i+1], pos, | |
411 | Materials[i], Fields[i], Cuts[i]); | |
412 | } else { | |
413 | Cone(r[i][j], r[i+1][j], r[i][j+1], r[i+1][j+1], | |
414 | z[i], z[i+1], pos, | |
415 | Materials[i]); | |
416 | } | |
417 | } | |
418 | } | |
419 | } | |
420 | ||
421 | void AliALIFE::Comment(char* Comment) | |
422 | { | |
423 | // Insert comment line | |
424 | fprintf(fFile1,"*%s\n", Comment); | |
425 | fprintf(fFile2,"*%s\n", Comment); | |
426 | } | |
427 | ||
428 | ||
429 | void AliALIFE::Finish() | |
430 | { | |
431 | // Finish geometry definition | |
432 | char s[BUFSIZ]; | |
433 | fprintf(fFile1," END\n"); | |
434 | fclose(fFile2); | |
435 | fFile2=fopen(fVolumeFile,"r"); | |
436 | while (fgets(s, BUFSIZ, fFile2)) { | |
437 | fputs(s,fFile1); | |
438 | } | |
439 | ||
440 | fclose(fFile1); | |
441 | fclose(fFile2); | |
442 | } | |
443 | ||
444 | ||
445 | ||
446 | ||
447 | ||
448 | ||
449 | ||
450 | ||
451 | ||
452 | ||
453 |