83085db08e9a6bc5136f225bee317e289f612ae7
[u/mrichter/AliRoot.git] / EMCAL / AliEMCALGeometry.h
1 #ifndef ALIEMCALGEOMETRY_H
2 #define ALIEMCALGEOMETRY_H
3 /* Copyright(c) 1998-2004, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6 /* $Id$ */
7
8 //_________________________________________________________________________
9 // Geometry class  for EMCAL : singleton
10 // EMCAL consists of a layers of scintillator, and lead.
11 //                  
12 //*-- Author: Sahal Yacoob (LBL / UCT)
13 //*--   and : Yves Schutz (Subatech)
14 //*--   and : Aleksei Pavlinov (WSU) - shashlyk staff
15 //*--   and : Gustavo Conesa: Add TRU mapping. TRU parameters still not fixed.
16
17 // --- ROOT system ---
18 class TString ;
19 class TObjArray;
20 class TVector3;
21 class TGeoMatrix;
22 class TParticle ; 
23 class AliEMCALGeometry;
24 class AliEMCALShishKebabTrd1Module;
25 class AliEMCALRecPoint;
26 class TClonesArray ;
27
28 // --- AliRoot header files ---
29 #include <TArrayD.h>
30 #include <TMath.h>
31
32 #include "AliGeometry.h"
33
34 class AliEMCALGeometry : public AliGeometry {
35 public:
36   AliEMCALGeometry(const AliEMCALGeometry& geom);
37   virtual ~AliEMCALGeometry(void); 
38
39   static AliEMCALGeometry * GetInstance(const Text_t* name,
40                                         const Text_t* title="") ; 
41   static AliEMCALGeometry * GetInstance();
42   AliEMCALGeometry & operator = (const AliEMCALGeometry  & /*rvalue*/) {
43     // assignement operator requested by coding convention but not needed
44     Fatal("operator =", "not implemented");
45     return *this;
46   };
47   static Char_t* GetDefaulGeometryName() {return fgDefaultGeometryName;}
48   void PrintGeometry();                                           //*MENU*  
49   void PrintCellIndexes(Int_t absId=0, int pri=0, char *tit="");  //*MENU*
50   virtual void Browse(TBrowser* b);
51   virtual Bool_t  IsFolder() const;
52
53   void GetCellPhiEtaIndexInSModuleFromTRUIndex(Int_t itru, Int_t iphitru, Int_t ietatru, Int_t &ietaSM, Int_t &iphiSM) const ; // Tranforms Eta-Phi Cell index in TRU into Eta-Phi index in Super Module
54   
55   // Have to call GetTransformationForSM() before calculation global charachteristics 
56   void GetGlobal(const Double_t *loc, Double_t *glob, int ind) const;
57   void GetGlobal(const TVector3 &vloc, TVector3 &vglob, int ind) const;
58   void GetGlobal(Int_t absId, Double_t glob[3]) const;
59   void GetGlobal(Int_t absId, TVector3 &vglob) const;
60   // for a given tower index absId returns eta and phi of gravity center of tower.
61   void EtaPhiFromIndex(Int_t absId, Double_t &eta, Double_t &phi) const;
62   void EtaPhiFromIndex(Int_t absId, Float_t &eta, Float_t &phi) const;
63   // 
64   Bool_t GetPhiBoundariesOfSM   (Int_t nSupMod, Double_t &phiMin, Double_t &phiMax) const;
65   Bool_t GetPhiBoundariesOfSMGap(Int_t nPhiSec, Double_t &phiMin, Double_t &phiMax) const;
66   Bool_t SuperModuleNumberFromEtaPhi(Double_t eta, Double_t phi, Int_t &nSupMod) const;
67
68   Bool_t GetAbsCellIdFromEtaPhi(Double_t eta,Double_t phi, Int_t &absId) const;
69
70   //  virtual void GetGlobal(const AliEMCALRecPoint *rp, TVector3 &vglob) const;
71
72   virtual void GetGlobal(const AliRecPoint *rp, TVector3 &vglob) const;
73   // Bool_t AreInSameTower(Int_t id1, Int_t id2) const ;  
74
75   virtual void GetGlobal(const AliRecPoint *, TVector3 &, TMatrixF &) const {}
76
77   virtual Bool_t Impact(const TParticle *) const {return kTRUE;}
78
79   Bool_t IsInEMCAL(Double_t x, Double_t y, Double_t z) const;
80   // General
81   Bool_t  IsInitialized(void) const { return fgInit ; }
82   // Return EMCAL geometrical parameters
83   // geometry
84   Char_t* GetNameOfEMCALEnvelope() const {return "XEN1";}
85   Float_t GetAlFrontThickness() const { return fAlFrontThick;}
86   Float_t GetArm1PhiMin() const { return fArm1PhiMin ; }
87   Float_t GetArm1PhiMax() const { return fArm1PhiMax ; }
88   Float_t GetArm1EtaMin() const { return fArm1EtaMin;}
89   Float_t GetArm1EtaMax() const { return fArm1EtaMax;}
90   Float_t GetIPDistance() const { return fIPDistance;}   
91   Float_t GetIP2ECASection() const { return ( GetIPDistance() + GetAlFrontThickness() + GetGap2Active() ) ; }   
92   Float_t GetEnvelop(Int_t index) const { return fEnvelop[index] ; }  
93   Float_t GetShellThickness() const { return fShellThickness ; }
94   Float_t GetZLength() const { return fZLength ; } 
95   Float_t GetGap2Active() const {return  fGap2Active ;}
96   Float_t GetDeltaEta() const {return (fArm1EtaMax-fArm1EtaMin)/
97                                        ((Float_t)fNZ);}
98   Float_t GetDeltaPhi() const {return (fArm1PhiMax-fArm1PhiMin)/
99                                        ((Float_t)fNPhi);}
100   Int_t   GetNECLayers() const {return fNECLayers ;}
101   Int_t   GetNZ() const {return fNZ ;}
102   Int_t   GetNEta() const {return fNZ ;}
103   Int_t   GetNPhi() const {return fNPhi ;}
104   Int_t   GetNTowers() const {return fNPhi * fNZ ;}
105   Float_t GetECPbRadThick()const {return fECPbRadThickness;}
106   Float_t GetECScintThick() const {return fECScintThick;}
107   Float_t GetSampling() const {return fSampling ; } 
108   //  Bool_t IsInECA(Int_t index) const { if ( (index > 0 && (index <= GetNZ() * GetNPhi()))) return kTRUE; else return kFALSE ;}
109
110   Int_t   GetNumberOfSuperModules() const {return fNumberOfSuperModules;}
111   Float_t GetfPhiGapForSuperModules() const {return fPhiGapForSM;}
112   Float_t GetPhiModuleSize() const  {return fPhiModuleSize;}
113   Float_t GetEtaModuleSize() const  {return fEtaModuleSize;}
114   Float_t GetFrontSteelStrip() const {return fFrontSteelStrip;}
115   Float_t GetLateralSteelStrip() const {return fLateralSteelStrip;}
116   Float_t GetPassiveScintThick() const {return fPassiveScintThick;}
117   Float_t GetPhiTileSize() const {return fPhiTileSize;}
118   Float_t GetEtaTileSize() const {return fEtaTileSize;}
119   Int_t   GetNPhiSuperModule() const {return fNPhiSuperModule;}
120   Int_t   GetNPHIdiv() const {return fNPHIdiv ;}
121   Int_t   GetNETAdiv() const {return fNETAdiv ;}
122   Int_t   GetNCells()  const {return fNCells;}
123
124   Int_t   GetNTRU() const    {return fNTRUEta*fNTRUPhi ; }  
125   Int_t   GetNTRUEta() const {return fNTRUEta ; }  
126   Int_t   GetNTRUPhi() const {return fNTRUPhi ; }  
127
128   Int_t   GetNCellsInTRU() const    {return fNCellsInTRUEta*fNCellsInTRUPhi; }  
129   Int_t   GetNCellsInTRUEta() const {return fNCellsInTRUEta ; }  
130   Int_t   GetNCellsInTRUPhi() const {return fNCellsInTRUPhi ; }  
131
132   Float_t GetSteelFrontThickness() const { return fSteelFrontThick;}
133   Float_t GetLongModuleSize() const {return fLongModuleSize;}
134
135   Float_t GetTrd1Angle() const {return fTrd1Angle;}
136   Float_t Get2Trd1Dx2()  const {return f2Trd1Dx2;}
137   Float_t GetTrd2AngleY()const {return fTrd2AngleY;}
138   Float_t Get2Trd2Dy2()  const {return f2Trd2Dy2;}
139   Float_t GetTubsR()     const {return fTubsR;}
140   Float_t GetTubsTurnAngle() const {return fTubsTurnAngle;}
141
142   // TRD1 staff
143   void    CreateListOfTrd1Modules();
144   TList  *GetShishKebabTrd1Modules() const {return fShishKebabTrd1Modules;}
145   AliEMCALShishKebabTrd1Module *GetShishKebabModule(Int_t neta);
146
147   void     GetTransformationForSM();
148   Float_t *GetSuperModulesPars() {return fParSM;}
149   TGeoMatrix *GetTransformationForSM(int i) {
150   if(i>=0 && i<GetNumberOfSuperModules()) return fMatrixOfSM[i]; 
151                                         else return 0;}
152   // May 31, 2006; ALICE numbering scheme: 
153   // see ALICE-INT-2003-038: ALICE Coordinate System and Software Numbering Convention
154   // All indexes are stared from zero now.
155   // 
156   // abs id <-> indexes; Shish-kebab case, only TRD1 now.
157   // EMCAL -> Super Module -> module -> tower(or cell) - logic tree of EMCAL
158   // 
159   //**  Usual name of variable - Dec 18,2006 **
160   //  nSupMod - index of super module (SM)
161   //  nModule - index of module in SM
162   //  nIphi   - phi index of tower(cell) in module
163   //  nIeta   - eta index of tower(cell) in module
164   //  
165   //  Inside SM
166   //  iphim   - phi index of module in SM  
167   //  ietam   - eta index of module in SM  
168   //
169   //  iphi    - phi index of tower(cell) in SM  
170   //  ieta    - eta index of tower(cell) in SM  
171   Int_t   GetAbsCellId(Int_t nSupMod, Int_t nModule, Int_t nIphi, Int_t nIeta) const;
172   Bool_t  CheckAbsCellId(Int_t absId) const;
173   Bool_t  GetCellIndex(Int_t absId, Int_t &nSupMod, Int_t &nModule, Int_t &nIphi, Int_t &nIeta) const;
174   // Local coordinate of Super Module 
175   void    GetModulePhiEtaIndexInSModule(Int_t nSupMod, Int_t nModule, Int_t &iphim, Int_t &ietam) const;
176   void    GetCellPhiEtaIndexInSModule(Int_t nSupMod, Int_t nModule, Int_t nIphi, Int_t nIeta,
177                                       Int_t &iphi, Int_t &ieta) const ;
178   Int_t   GetSuperModuleNumber(Int_t absId)  const;
179   Int_t   GetNumberOfModuleInPhiDirection(Int_t nSupMod)  const
180   {
181     // inline function
182     if(fKey110DEG == 1 && nSupMod>=10) return fNPhi/2;
183     else                               return fNPhi;
184   }
185   // From cell indexes to abs cell id
186   void    GetModuleIndexesFromCellIndexesInSModule(Int_t nSupMod, Int_t iphi, Int_t ieta, 
187           Int_t &iphim, Int_t &ietam, Int_t &nModule) const;
188   Int_t   GetAbsCellIdFromCellIndexes(Int_t nSupMod, Int_t iphi, Int_t ieta) const;
189   // Methods for AliEMCALRecPoint - Feb 19, 2006
190   Bool_t   RelPosCellInSModule(Int_t absId, Double_t &xr, Double_t &yr, Double_t &zr) const;
191   Bool_t   RelPosCellInSModule(Int_t absId, Double_t loc[3]) const;
192   Bool_t   RelPosCellInSModule(Int_t absId, TVector3 &vloc) const;
193   // ---
194   Float_t AngleFromEta(Float_t eta) const { // returns theta in radians for a given pseudorapidity
195     return 2.0*TMath::ATan(TMath::Exp(-eta));
196   }
197   Float_t ZFromEtaR(Float_t r,Float_t eta) const { // returns z in for a given
198     // pseudorapidity and r=sqrt(x*x+y*y).
199     return r/TMath::Tan(AngleFromEta(eta));
200   }
201   void SetNZ(Int_t nz) { fNZ= nz ; printf("SetNZ: Number of modules in Z set to %d", fNZ) ; }
202   void SetNPhi(Int_t nphi) { fNPhi= nphi ; printf("SetNPhi: Number of modules in Phi set to %d", fNPhi) ; }
203
204   void SetNTRUEta(Int_t ntru) {fNTRUEta = ntru; ; printf("SetNTRU: Number of TRUs per SuperModule in Etaset to %d", fNTRUEta) ;}
205   void SetNTRUPhi(Int_t ntru) {fNTRUPhi = ntru; ; printf("SetNTRU: Number of TRUs per SuperModule in Phi set to %d", fNTRUPhi) ;}
206
207   void SetSampling(Float_t samp) { fSampling = samp; printf("SetSampling: Sampling factor set to %f", fSampling) ; }
208
209   Int_t GetNCellsInSupMod() const {return fNCellsInSupMod;}
210   Int_t GetNCellsInModule()  const {return fNCellsInModule; }
211   Int_t GetKey110DEG()      const {return fKey110DEG;}
212   Int_t GetILOSS() const {return fILOSS;}
213   Int_t GetIHADR() const {return fIHADR;}
214
215   AliEMCALGeometry(); // default ctor only for internal usage (singleton)
216
217 protected:
218   AliEMCALGeometry(const Text_t* name, const Text_t* title);// ctor only for internal usage (singleton)
219
220   void Init(void);                      // initializes the parameters of EMCAL
221   void CheckAdditionalOptions();        //
222   void DefineSamplingFraction();        // Jun 5, 2006
223   
224 private:
225   static AliEMCALGeometry * fgGeom;     // pointer to the unique instance of the singleton
226   static Bool_t  fgInit;                // Tells if geometry has been succesfully set up.
227   static Char_t* fgDefaultGeometryName; // Default name of geometry
228
229   TString fGeoName;                     //geometry name
230
231   TObjArray *fArrayOpts;                //! array of geometry options
232
233   Float_t fAlFrontThick;                // Thickness of the front Al face of the support box  
234   Float_t fECPbRadThickness;            // cm, Thickness of the Pb radiators
235   Float_t fECScintThick;                // cm, Thickness of the scintillators
236   Int_t   fNECLayers;                   // number of scintillator layers
237   
238   Float_t fArm1PhiMin;                  // Minimum angular position of EMCAL in Phi (degrees)
239   Float_t fArm1PhiMax;                  // Maximum angular position of EMCAL in Phi (degrees)
240   Float_t fArm1EtaMin;                  // Minimum pseudorapidity position of EMCAL in Eta
241   Float_t fArm1EtaMax;                  // Maximum pseudorapidity position of EMCAL in Eta
242   
243   // Geometry Parameters
244   Float_t fEnvelop[3];                  // the GEANT TUB for the detector 
245   Float_t fIPDistance;                  // Radial Distance of the inner surface of the EMCAL
246   Float_t fShellThickness;              // Total thickness in (x,y) direction
247   Float_t fZLength;                     // Total length in z direction
248   Float_t fGap2Active;                  // Gap between the envelop and the active material
249   Int_t   fNZ;                          // Number of Towers in the Z direction
250   Int_t   fNPhi;                        // Number of Towers in the PHI direction
251   Float_t fSampling;                    // Sampling factor
252
253   // Shish-kebab option - 23-aug-04 by PAI; COMPACT, TWIST, TRD1 and TRD2
254   Int_t   fNumberOfSuperModules;         // default is 12 = 6 * 2 
255   Float_t fSteelFrontThick;              // Thickness of the front stell face of the support box - 9-sep-04
256   Float_t fFrontSteelStrip;              // 13-may-05
257   Float_t fLateralSteelStrip;            // 13-may-05
258   Float_t fPassiveScintThick;            // 13-may-05
259   Float_t fPhiModuleSize;                // Phi -> X 
260   Float_t fEtaModuleSize;                // Eta -> Y
261   Float_t fPhiTileSize;                  // Size of phi tile
262   Float_t fEtaTileSize;                  // Size of eta tile
263   Float_t fLongModuleSize;               // Size of long module
264   Int_t   fNPhiSuperModule;              // 6 - number supermodule in phi direction
265   Int_t   fNPHIdiv;                      // number phi divizion of module
266   Int_t   fNETAdiv;                      // number eta divizion of module
267   //
268   Int_t   fNCells;                       // number of cells in calo
269   Int_t   fNCellsInSupMod;               // number cell in super module
270   Int_t   fNCellsInModule;               // number cell in module)
271   //TRU parameters
272   Int_t   fNTRUEta ;                     // Number of TRUs per module in eta
273   Int_t   fNTRUPhi ;                     // Number of TRUs per module in phi
274   Int_t   fNCellsInTRUEta;               // Number of cells per TRU in eta 
275   Int_t   fNCellsInTRUPhi;               // Number of cells per TRU in phi 
276
277   // TRD1 options - 30-sep-04
278   Float_t fTrd1Angle;                    // angle in x-z plane (in degree) 
279   Float_t f2Trd1Dx2;                     // 2*dx2 for TRD1
280   Float_t fPhiGapForSM;                  // Gap betweeen supermodules in phi direction
281   Int_t   fKey110DEG;                    // for calculation abs cell id; 19-oct-05 
282   TArrayD fPhiBoundariesOfSM;            // phi boundaries of SM in rad; size is fNumberOfSuperModules;
283   TArrayD fPhiCentersOfSM;                // phi of centers of SMl size is fNumberOfSuperModules/2
284   Float_t fEtaMaxOfTRD1;                 // max eta in case of TRD1 geometry (see AliEMCALShishKebabTrd1Module)
285   // TRD2 options - 27-jan-07
286   Float_t fTrd2AngleY;                   // angle in y-z plane (in degree) 
287   Float_t f2Trd2Dy2;                     // 2*dy2 for TRD2
288   Float_t fEmptySpace;                   // 2mm om fred drawing
289   // Super module as TUBS
290   Float_t fTubsR;                        // radius of tubs 
291   Float_t fTubsTurnAngle;                // turn angle of tubs in degree
292   // Local Coordinates of SM
293   TArrayD  fCentersOfCellsEtaDir;        // size fNEta*fNETAdiv (for TRD1 only) (eta or z in SM, in cm)
294   TArrayD  fCentersOfCellsXDir;          // size fNEta*fNETAdiv (for TRD1 only) (       x in SM, in cm)
295   TArrayD  fCentersOfCellsPhiDir;        // size fNPhi*fNPHIdiv (for TRD1 only) (phi or y in SM, in cm)
296   //
297   TArrayD  fEtaCentersOfCells;           // [fNEta*fNETAdiv*fNPhi*fNPHIdiv], positive direction (eta>0); eta depend from phi position; 
298   TArrayD  fPhiCentersOfCells;           // [fNPhi*fNPHIdiv] from center of SM (-10. < phi < +10.)
299   // Move from AliEMCALv0 - Feb 19, 2006
300   TList *fShishKebabTrd1Modules; //! list of modules
301   // Local coordinates of SM for TRD1
302   Float_t     fParSM[3];       // SM sizes as in GEANT (TRD1)
303   TGeoMatrix* fMatrixOfSM[12]; //![fNumberOfSuperModules]; get from gGeoManager;
304
305   char *fAdditionalOpts[6];  //! some additional options for the geometry type and name
306   int  fNAdditionalOpts;     //! size of additional options parameter
307
308   // Options for Geant (MIP business) - will call in AliEMCAL
309   Int_t fILOSS;
310   Int_t fIHADR;
311
312   ClassDef(AliEMCALGeometry, 12) // EMCAL geometry class 
313 };
314
315 #endif // AliEMCALGEOMETRY_H