New TTask based method to do Digits To clusters. Works with files of multiple
[u/mrichter/AliRoot.git] / TGeant3 / AliNode.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 /*
17 $Log$
18 */
19
20
21 #include "AliNode.h"
22 #include "TShape.h"
23 #include "TTUBE.h"
24 #include "TBRIK.h"
25 #include "TTRD1.h"
26 #include "TTRD2.h"
27 #include "TTRAP.h"
28 #include "TTUBS.h"
29 #include "TCONE.h"
30 #include "TCONS.h"
31 #include "TSPHE.h"
32 #include "TPARA.h"
33 #include "TPGON.h"
34 #include "TPCON.h"
35 #include "TTUBS.h"
36 #include "TELTU.h"
37 #include "THYPE.h"
38 #include "TGTRA.h"
39 #include "TCTUB.h"
40
41 ClassImp(AliNode)
42     AliNode::AliNode(const char* name, const char* title, const char* shapename,
43                      Double_t x, Double_t y, Double_t z, const char* matrixname,
44                      Option_t* option) : 
45         TNode(name, title, shapename, x, y, z, matrixname, option)
46 {
47     fNDivision   = -1;
48     fAxis   =  0;
49     fStartC = 0.;
50     fStep   = 0.;
51 }
52
53   AliNode::AliNode(const char* name, const char* title, TShape* shape,
54                      Double_t x, Double_t y, Double_t z, TRotMatrix* matrix,
55                      Option_t* option) : 
56         TNode(name, title, shape, x, y, z, matrix, option)
57 {
58     fNDivision   = -1;
59     fAxis   =  0;
60     fStartC = 0.;
61     fStep   = 0.;
62 }
63
64 void   AliNode::SetDivision(Int_t ndiv, Int_t axis, Float_t start, Float_t step)
65 {
66     fNDivision   = ndiv;
67     fAxis   = axis;
68     fStartC = start;
69     fStep   = step;
70 }
71
72 void AliNode::ExpandDivisions()
73 {
74     Int_t i;
75     char vName[20];
76     char nName[20];
77     
78     char tmp[4];
79     AliNode* node;
80     TShape*  parent =  fParent->GetShape();
81     TShape*  newsh;
82     
83     strcpy(tmp, parent->GetTitle());
84     Int_t ndiv = fNDivision;
85
86 //                                      TUBE
87     
88     if (strcmp(tmp, "TUBE")==0) {
89         TTUBE * shape = (TTUBE*) parent;
90
91         Float_t dZ   = shape->GetDz();
92         Float_t rMin = shape->GetRmin();
93         Float_t rMax = shape->GetRmax();
94         
95         if (fAxis == 1) {
96 //  radial-division
97             Float_t dr = (rMax-rMin)/Float_t(fNDivision);
98             Float_t r1, r2;
99             for (i=0; i<ndiv; i++) {
100                 r1 = rMin+Float_t(i)*dr;
101                 r2 = r1+dr;
102                 sprintf(vName, "%sD%d", fShape->GetName(), i);
103                 sprintf(nName, "%sD%d", GetName(), i);
104                 newsh = new TTUBE(vName, "TUBE", "void", r1, r2, dZ);
105                 fParent->cd();
106                 node = new AliNode(nName,"", newsh, 0., 0., 0.);
107                 node->AddSons(fNodes);
108                 cd();
109             }
110     
111         } else if (fAxis == 2) {
112 //  phi-division
113             Float_t dPhi = 360./Float_t(fNDivision);
114             Float_t phi1, phi2;
115             for (i=0; i<ndiv; i++) {
116                 phi1 = Float_t(i)*dPhi;
117                 phi2 = phi1+dPhi;
118                 sprintf(vName, "%sD%d", fShape->GetName(), i);
119                 sprintf(nName, "%sD%d", GetName(), i);
120                 newsh = new TTUBS(vName, "TUBS", "void", rMin, rMax, dZ, phi1, phi2);
121                 fParent->cd();
122                 node = new AliNode(nName, "", newsh, 0., 0., 0.);
123                 node->AddSons(fNodes);
124                 cd();
125             }
126         } else {
127 //  z-division
128             Float_t delZ = dZ/Float_t(fNDivision);
129             for (i=0; i<ndiv; i++) {
130                 sprintf(vName, "%sD%d", fShape->GetName(), i);
131                 sprintf(nName, "%sD%d", GetName(), i);
132                 newsh = new TTUBE(vName, "TUBE", "void", rMin, rMax, delZ);
133                 fParent->cd();
134                 Float_t zpos = -dZ+delZ*(2.*Float_t(i)+1.);
135                 node = new AliNode(nName, "",newsh, 0., 0., zpos);
136                 node->AddSons(fNodes);
137                 cd();
138             }
139         }
140 //
141 //                            TUBS
142 //
143     } else if (strcmp(tmp, "TUBS")==0) {
144         TTUBS * shape = (TTUBS*) parent;
145         Float_t dZ   = shape->GetDz();
146         Float_t rMin = shape->GetRmin();
147         Float_t rMax = shape->GetRmax();
148         Float_t phi1 = shape->GetPhi1();
149         Float_t phi2 = shape->GetPhi2();
150         
151         if (fAxis == 1) {
152 //  radial-division
153             Float_t dr = (rMax-rMin)/Float_t(fNDivision);
154             Float_t r1, r2;
155             Int_t ndiv = fNDivision;
156             for (i=0; i<ndiv; i++) {
157                 r1 = rMin+Float_t(i)*dr;
158                 r2 = r1+dr;
159                 sprintf(vName, "%sD%d", fShape->GetName(), i);
160                 sprintf(nName, "%sD%d", GetName(), i);
161                 newsh = new TTUBS(vName, "TUBS", "void", r1, r2, dZ, phi1, phi2);
162                 fParent->cd();
163                 node = new AliNode(nName,"", newsh, 0., 0., 0.);
164                 node->AddSons(fNodes);
165                 cd();
166             }
167             
168         } else if (fAxis == 2) {
169 //  phi-division
170             Float_t dPhi = (phi2-phi1)/Float_t(fNDivision);
171             Float_t nphi1, nphi2;
172             
173             for (i=0; i<fNDivision; i++) {
174                 nphi1 = phi1+Float_t(i)*dPhi;
175                 nphi2 = nphi1+dPhi;
176                 sprintf(vName, "%sD%d", fShape->GetName(), i);
177                 sprintf(nName, "%sD%d", GetName(), i);
178
179                 newsh = new TTUBS(vName, "TUBS", "void", rMin, rMax, dZ, nphi1, nphi2);
180                 fParent->cd();
181                 node = new AliNode(nName, "", newsh, 0., 0., 0.);
182                 node->AddSons(fNodes);
183                 cd();
184             }
185         } else {
186 //  z-division
187             Float_t delZ = dZ/Float_t(fNDivision);
188             for (i=0; i<ndiv; i++) {
189                 sprintf(vName, "%sD%d", fShape->GetName(), i);
190                 sprintf(nName, "%sD%d", GetName(), i);
191                 newsh = new TTUBS(vName, "TUBS", "void", rMin, rMax, delZ, phi1, phi2);
192                 fParent->cd();
193                 Float_t zpos = -dZ+delZ*(2.*Float_t(i)+1.);
194                 node = new AliNode(nName, "",newsh, 0., 0., zpos);
195                 node->AddSons(fNodes);
196                 cd();
197             }
198         }
199         
200     } else if (strcmp(tmp, "CONE")==0) {
201         TCONE * shape = (TCONE*) parent;
202
203         Float_t dZ   = shape->GetDz();
204         Float_t rMin1 = shape->GetRmin();
205         Float_t rMax1 = shape->GetRmax();
206         Float_t rMin2 = shape->GetRmin2();
207         Float_t rMax2 = shape->GetRmax2();
208         
209         if (fAxis == 1) {
210 //  radial-division
211             Float_t dr1 = (rMax1-rMin1)/Float_t(fNDivision);
212             Float_t dr2 = (rMax2-rMin2)/Float_t(fNDivision);
213             Float_t r11, r12, r21, r22;
214             for (i=0; i<ndiv; i++) {
215                 r11 = rMin1+Float_t(i)*dr1;
216                 r12 = r11+dr1;
217                 r21 = rMin2+Float_t(i)*dr2;
218                 r22 = r21+dr2;
219
220                 sprintf(vName, "%sD%d", fShape->GetName(), i);
221                 sprintf(nName, "%sD%d", GetName(), i);
222                 newsh = new TCONE(vName, "CONE", "void", dZ, r11, r12, r21, r22);
223                 fParent->cd();
224                 node = new AliNode(nName,"", newsh, 0., 0., 0.);
225                 node->AddSons(fNodes);
226                 cd();
227             }
228         } else if (fAxis == 2) {
229 //  phi-division
230             Float_t dPhi = 360./Float_t(fNDivision);
231             Float_t phi1, phi2;
232             for (i=0; i<ndiv; i++) {
233                 phi1 = Float_t(i)*dPhi;
234                 phi2 = phi1+dPhi;
235                 sprintf(vName, "%sD%d", fShape->GetName(), i);
236                 sprintf(nName, "%sD%d", GetName(), i);
237                 newsh = new TCONS(vName, "CONS", "void", dZ, rMin1, rMax1, 
238                           rMin2, rMax2, phi1, phi2);
239                 fParent->cd();
240                 node = new AliNode(nName, "",newsh, 0., 0., 0.);
241                 node->AddSons(fNodes);
242                 cd();
243             }
244         } else {
245 //  z-division
246             Float_t delZ = dZ/Float_t(fNDivision);
247             for (i=0; i<ndiv; i++) {
248                 sprintf(vName, "%sD%d", fShape->GetName(), i);
249                 sprintf(nName, "%sD%d", GetName(), i);
250                 newsh = new TCONE(vName, "CONE", "void", delZ, rMin1, rMax1, rMin2, rMax2);
251                 fParent->cd();
252                 Float_t zpos = -dZ+delZ*(2.*Float_t(i)+1.);
253                 node = new AliNode(nName, "",newsh, 0., 0., zpos);
254                 node->AddSons(fNodes);
255                 cd();
256             }
257         }
258         
259     } else if (strcmp(tmp, "CONS")==0) {
260         TCONS * shape = (TCONS*) parent;
261         Float_t dZ   = shape->GetDz();
262         Float_t rMin1 = shape->GetRmin();
263         Float_t rMax1 = shape->GetRmax();
264         Float_t rMin2 = shape->GetRmin2();
265         Float_t rMax2 = shape->GetRmax2();
266         Float_t phi1 = shape->GetPhi1();
267         Float_t phi2 = shape->GetPhi2();
268         if (fAxis == 1) {
269 //  radial-division
270             Float_t dr1 = (rMax1-rMin1)/Float_t(fNDivision);
271             Float_t dr2 = (rMax2-rMin2)/Float_t(fNDivision);
272             Float_t r11, r12, r21, r22;
273             for (i=0; i<ndiv; i++) {
274                 r11 = rMin1+Float_t(i)*dr1;
275                 r12 = r11+dr1;
276                 r21 = rMin2+Float_t(i)*dr2;
277                 r22 = r21+dr2;
278
279                 sprintf(vName, "%sD%d", fShape->GetName(), i);
280                 sprintf(nName, "%sD%d", GetName(), i);
281                 newsh = new TCONS(vName, "CONS", "void", dZ, r11, r12, r21, r22, phi1, phi2);
282                 fParent->cd();
283                 node = new AliNode(nName,"", newsh, 0., 0., 0.);
284                 node->AddSons(fNodes);
285                 cd();
286             }
287             
288         } else if (fAxis == 2) {
289 //  phi-division
290             Float_t dPhi = (phi2-phi1)/Float_t(fNDivision);
291             Float_t nphi1, nphi2;
292             
293             for (i=0; i<fNDivision; i++) {
294                 nphi1 = phi1+Float_t(i)*dPhi;
295                 nphi2 = nphi1+dPhi;
296                 sprintf(vName, "%sD%d", fShape->GetName(), i);
297                 sprintf(nName, "%sD%d", GetName(), i);
298
299                 newsh = new TCONS(vName, "CONS", "void", dZ, rMin1, rMax1, rMin2, rMax2, nphi1, nphi2);
300                 fParent->cd();
301                 node = new AliNode(nName, "", newsh, 0., 0., 0.);
302                 node->AddSons(fNodes);
303                 cd();
304             }
305         } else {
306 //  z-division
307             Float_t delZ = dZ/Float_t(fNDivision);
308             for (i=0; i<ndiv; i++) {
309                 sprintf(vName, "%sD%d", fShape->GetName(), i);
310                 sprintf(nName, "%sD%d", GetName(), i);
311                 newsh = new TCONS(vName, "CONS", "void", delZ, rMin1, rMax1, rMin2, rMax2, phi1, phi2);
312                 fParent->cd();
313                 Float_t zpos = -dZ+delZ*(2.*Float_t(i)+1.);
314                 node = new AliNode(nName,"",newsh, 0., 0., zpos);
315                 node->AddSons(fNodes);
316                 cd();
317             }
318         }
319     } else if (strcmp(tmp, "BRIK")==0) {
320 //
321 //                          BRIK
322 //
323         TBRIK * shape = (TBRIK*) parent;
324         Float_t dX   = shape->GetDx();
325         Float_t dY   = shape->GetDy();
326         Float_t dZ   = shape->GetDz();
327
328         if (fAxis == 1) {
329 // division in x
330             Float_t delX = dX/Float_t(fNDivision);
331             for (i=0; i<ndiv; i++) {
332                 sprintf(vName, "%sD%d", fShape->GetName(), i);
333                 sprintf(nName, "%sD%d", GetName(), i);
334                 newsh = new TBRIK(vName, "BRIK", "void", delX, dY, dZ);
335                 fParent->cd();
336                 Float_t xpos = -dX+delX*(2.*Float_t(i)+1.);
337                 node = new AliNode(nName,"",newsh, xpos, 0., 0.);
338                 node->AddSons(fNodes);
339                 cd();
340             }
341         } else if (fAxis == 2) {
342 // division in y
343             Float_t delY = dY/Float_t(fNDivision);
344             for (i=0; i<ndiv; i++) {
345                 sprintf(vName, "%sD%d", fShape->GetName(), i);
346                 sprintf(nName, "%sD%d", GetName(), i);
347                 newsh = new TBRIK(vName, "BRIK", "void", dX, delY, dZ);
348                 fParent->cd();
349                 Float_t ypos = -dY+delY*(2.*Float_t(i)+1.);
350                 node = new AliNode(nName,"",newsh, 0., ypos, 0.);
351                 node->AddSons(fNodes);
352                 cd();
353             }
354         } else {
355 // division in z
356             Float_t delZ = dZ/Float_t(fNDivision);
357             for (i=0; i<ndiv; i++) {
358                 sprintf(vName, "%sD%d", fShape->GetName(), i);
359                 sprintf(nName, "%sD%d", GetName(), i);
360                 newsh = new TBRIK(vName, "BRIK", "void", dX, dY, delZ);
361                 fParent->cd();
362                 Float_t zpos = -dZ+delZ*(2.*Float_t(i)+1.);
363                 node = new AliNode(nName,"",newsh, 0., 0., zpos);
364                 node->AddSons(fNodes);
365                 cd();
366             }
367         }
368     }
369 }
370
371 void AliNode::AddSons(TList* list)
372 {
373     if (!list) return;
374     if (!fNodes) fNodes = new TList();
375     
376     TIter next(list);
377     AliNode* node;
378     
379     while((node = (AliNode*)next())) {
380         AliNode* newNode = new AliNode(*node, this);
381         fNodes->Add(newNode);
382         newNode->SetParent(this);
383         newNode->AddSons(node->GetListOfNodes());
384     }
385 }
386
387
388 AliNode::AliNode(const AliNode &node, AliNode* parent)
389 {
390     fNDivision   = node.Ndiv();
391     fAxis        = node.Axis();
392     fStartC      = node.StartC();
393     fStep        = node.Step();
394     fName        = node.GetName();
395     fTitle       = node.GetTitle();
396     fX           = node.GetX();
397     fY           = node.GetY();
398     fZ           = node.GetZ();
399     fMatrix      = node.GetMatrix();
400     fNodes       = 0;
401     fParent      = parent;
402     
403     
404     if (fNDivision > 0) {
405         fShape = new TShape(*node.GetShape());
406     } else {
407         fShape = (TShape*) node.GetShape()->Clone();
408     }
409     
410     
411 //    fShape = (TShape*) shape->Clone();
412 /*
413     char tmp[4];        
414     strcpy(tmp, shape->ClassName());
415
416     if (strcmp(tmp, "TTUBE")==0) 
417         fShape =  (TTUBE*)shape->Clone();
418     else if (strcmp(tmp, "TBRIK")==0) 
419         fShape =  (TBRIK*)shape->Clone();
420     else if (strcmp(tmp, "TCONE")==0) 
421         fShape =  (TCONE*)shape->Clone();
422     else if (strcmp(tmp, "TCONS")==0) 
423         fShape =  (TCONS*)shape->Clone();
424     else if (strcmp(tmp, "TTUBS")==0) 
425         fShape =  (TTUBS*)shape->Clone();
426     else if (strcmp(tmp, "TTRAP")==0) 
427         fShape =  (TTRAP*)shape->Clone();
428     else if (strcmp(tmp, "TTRD1")==0) 
429         fShape =  (TTRD1*)shape->Clone();
430     else if (strcmp(tmp, "TTRD2")==0)
431         fShape =  (TTRD2*)shape->Clone();
432     else if (strcmp(tmp, "TSPHE")==0) 
433         fShape =  (TSPHE*)shape->Clone();
434     else if (strcmp(tmp, "TPGON")==0) 
435         fShape =  (TPGON*)shape->Clone();
436     else if (strcmp(tmp, "TPCON")==0) 
437         fShape =  (TPCON*)shape->Clone();
438 */
439 }
440
441 void AliNode::AddSon(AliNode* node)
442 {
443     fNodes->Add(node);
444 }
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459