Moving to the new VMC naming convention
[u/mrichter/AliRoot.git] / STEER / AliLegoGeneratorXYZ.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 //      Lego generator in x - y - z bins
20 //    Uses geantino rays to check the material distributions and detector's
21 //    geometry
22 //    Author: A.Morsch 
23 //-------------------------------------------------------------------------
24
25 #include "AliLegoGeneratorXYZ.h"
26 #include "AliRun.h"
27
28 ClassImp(AliLegoGeneratorXYZ)
29
30
31 //___________________________________________
32     
33
34 AliLegoGeneratorXYZ::AliLegoGeneratorXYZ()
35 {
36 // Default Constructor
37     fDir1[0]=1.; fDir1[1]=0.; fDir1[2]=0.;
38     fDir2[0]=0.; fDir2[1]=1.; fDir2[2]=0.;    
39     fDir3[0]=0.; fDir3[1]=0.; fDir3[2]=1.;    
40 }
41
42 AliLegoGeneratorXYZ::AliLegoGeneratorXYZ(char* axis)
43 {
44 // Constructor
45     if (!strcmp(axis,"x") || !strcmp(axis,"X")) 
46     {
47         fDir1[0]=0.; fDir1[1]=1.; fDir1[2]=0.;
48         fDir2[0]=0.; fDir2[1]=0.; fDir2[2]=1.;    
49         fDir3[0]=1.; fDir3[1]=0.; fDir3[2]=0.;    
50     }
51     else if (!strcmp(axis,"y") || !strcmp(axis,"Y")) 
52     {
53         fDir1[0]=1.; fDir1[1]=0.; fDir1[2]=0.;
54         fDir2[0]=0.; fDir2[1]=0.; fDir2[2]=1;    
55         fDir3[0]=0.; fDir3[1]=1.; fDir3[2]=0.;    
56     }
57     else if (!strcmp(axis,"z") || !strcmp(axis,"Z")) 
58     {
59         fDir1[0]=1.; fDir1[1]=0.; fDir1[2]=0.;
60         fDir2[0]=0.; fDir2[1]=1.; fDir2[2]=0.;    
61         fDir3[0]=0.; fDir3[1]=0.; fDir3[2]=1.;    
62     }
63     else 
64     {
65         fDir1[0]=1.; fDir1[1]=0.; fDir1[2]=0.;
66         fDir2[0]=0.; fDir2[1]=1.; fDir2[2]=0.;    
67         fDir3[0]=0.; fDir3[1]=0.; fDir3[2]=1.;    
68     }
69 }
70
71
72 AliLegoGeneratorXYZ::AliLegoGeneratorXYZ(Int_t nc1, Float_t c1min,
73                                          Float_t c1max, Int_t nc2, 
74                                          Float_t c2min, Float_t c2max,
75                                          Float_t rmin, Float_t rmax, Float_t zmax) : 
76     AliLegoGenerator(nc1, c1min, c1max, nc2, c2min, c2max,
77                      rmin, rmax, zmax)
78 {
79 //  Constructor
80     fDir1[0]=1.; fDir1[1]=0.; fDir1[2]=0.;
81     fDir2[0]=0.; fDir2[1]=1.; fDir2[2]=0.;    
82     fDir3[0]=0.; fDir3[1]=0.; fDir3[2]=1.;    
83 }
84
85
86 //___________________________________________
87 void AliLegoGeneratorXYZ::Generate()
88 {
89 // Create a geantino with kinematics corresponding to the current bins
90 // Here: Coor1 =  x 
91 //       Coor2 =  z
92    
93   //
94   // Rootinos are 0
95    const Int_t kMpart = 0;
96    Float_t orig[3], pmom[3];
97    
98    // Prepare for next step
99    if(fCoor1Bin>=fNCoor1-1)
100      if(fCoor2Bin>=fNCoor2-1) {
101        Warning("Generate","End of Lego Generation");
102        return;
103      } else { 
104        fCoor2Bin++;
105        printf("Generating rays in Coordinate 2 bin:%d\n",fCoor2Bin);
106        fCoor1Bin=0;
107      } else fCoor1Bin++;
108
109    
110    fCurCoor1 = (fCoor1Min+(fCoor1Bin+0.5)*(fCoor1Max-fCoor1Min)/fNCoor1);
111    fCurCoor2 = (fCoor2Min+(fCoor2Bin+0.5)*(fCoor2Max-fCoor2Min)/fNCoor2);
112
113 // Origin and direction
114    Int_t i;
115    for (i=0; i<3; i++) {
116        pmom[i]=fDir3[i];
117        orig[i]=fCurCoor1*fDir1[i]+fCurCoor2*fDir2[i];
118    }
119    
120    Float_t dalicz = 3000;
121    if (fRadMin > 0) {
122        Float_t t = PropagateCylinder(orig,pmom,fRadMin,dalicz);
123        orig[0] = pmom[0]*t;
124        orig[1] = pmom[1]*t;
125        orig[2] = pmom[2]*t;
126        if (TMath::Abs(orig[2]) > fZMax) return;
127    }
128    
129    Float_t polar[3]={0.,0.,0.};
130    Int_t ntr;
131    gAlice->PushTrack(1, -1, kMpart, pmom, orig, polar, 0, kPPrimary, ntr);
132    
133 }
134
135
136
137