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