Fix coverity defect
[u/mrichter/AliRoot.git] / MUON / AliMUONGeometryTransformer.h
1 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
2  * See cxx source for full Copyright notice                               */
3
4 // $Id$
5
6 /// \ingroup geometry
7 /// \class AliMUONGeometryTransformer
8 /// \brief Top container class for geometry transformations
9 ///
10 /// Geometry transformations can be filled in these ways:
11 /// - by geometry builder when geometry is built via builders
12 ///   (this way is used when running simulation and building geometry
13 ///    via VirtualMC)
14 /// - from Root geometry file (*.root) or ASCII file (*.dat) using
15 ///   the method LoadGeometryData(const TString& fileName)
16 /// - from geometry loaded in AliGeomManager using
17 ///   the method LoadGeometryData() without arguments
18 /// 
19 /// \author Ivana Hrivnacova, IPN Orsay
20
21 #ifndef ALI_MUON_GEOMETRY_TRANSFORMER_H
22 #define ALI_MUON_GEOMETRY_TRANSFORMER_H
23
24 #include <TObject.h>
25 #include <TObjArray.h>
26 #include <TGeoMatrix.h>
27
28 class AliMUONGeometryModuleTransformer;
29 class AliMUONGeometryDetElement;
30
31 class TGeoManager;
32 class TClonesArray;
33 class AliMpExMap;
34 class AliMpArea;
35
36 class AliMUONGeometryTransformer : public TObject
37 {
38   public:
39     AliMUONGeometryTransformer();
40     AliMUONGeometryTransformer(TRootIOCtor* /*ioCtor*/);
41     virtual  ~AliMUONGeometryTransformer();
42     
43     // methods
44     void  AddModuleTransformer(AliMUONGeometryModuleTransformer* transformer);
45     void  AddMisAlignModule(Int_t moduleId, const TGeoHMatrix& matrix, Bool_t bGlobal = kTRUE);
46     void  AddMisAlignDetElement(Int_t detElemId, const TGeoHMatrix& matrix, Bool_t bGlobal = kTRUE);
47     void  CreateModules();
48
49     void  AddAlignableVolumes() const; 
50     TClonesArray* CreateZeroAlignmentData() const;
51     void  ClearMisAlignmentData();      
52
53     // IO
54     //
55     Bool_t  LoadTransformations(); 
56     Bool_t  LoadGeometryData(const TString& fileName);
57     Bool_t  LoadGeometryData();
58
59     Bool_t  WriteTransformations(const TString& fileName) const;
60     Bool_t  WriteMisAlignmentData(const TString& fileName) const;
61
62     // Transformation methods 
63     //
64     void Global2Local(Int_t detElemId,
65                  Float_t xg, Float_t yg, Float_t zg, 
66                  Float_t& xl, Float_t& yl, Float_t& zl) const;
67     void Global2Local(Int_t detElemId,
68                  Double_t xg, Double_t yg, Double_t zg, 
69                  Double_t& xl, Double_t& yl, Double_t& zl) const;
70
71     void Local2Global(Int_t detElemId,
72                  Float_t xl, Float_t yl, Float_t zl, 
73                  Float_t& xg, Float_t& yg, Float_t& zg) const;
74     void Local2Global(Int_t detElemId,
75                  Double_t xl, Double_t yl, Double_t zl, 
76                  Double_t& xg, Double_t& yg, Double_t& zg) const;
77                  
78     // Set methods
79     void SetDetName(const TString& detName);                 
80     void SetOwner(Bool_t isOwner);                 
81
82     // Get methods
83     //
84     Int_t GetNofModuleTransformers() const;
85     const AliMUONGeometryModuleTransformer* GetModuleTransformer(
86                                Int_t index, Bool_t warn = true) const;
87
88     const AliMUONGeometryModuleTransformer* GetModuleTransformerByDEId(
89                                Int_t detElemId, Bool_t warn = true) const;
90
91     const AliMUONGeometryDetElement* GetDetElement(
92                                Int_t detElemId, Bool_t warn = true) const;
93
94     const TClonesArray* GetMisAlignmentData() const;
95     
96     Bool_t  HasDE(Int_t detElemId) const;
97
98     AliMpArea* GetDEArea(Int_t detElemId) const;
99     
100   protected:
101     /// Not implemented
102     AliMUONGeometryTransformer(const AliMUONGeometryTransformer& right);
103     /// Not implemented
104     AliMUONGeometryTransformer&  operator = (const AliMUONGeometryTransformer& right);
105  
106   private:
107     // static methods
108     static const TString&  GetDefaultDetectorName(); 
109
110     // methods
111       
112     void CreateDEAreas() const;
113     
114     Bool_t LoadMapping() const;
115     AliMUONGeometryModuleTransformer* GetModuleTransformerNonConst(
116                                     Int_t index, Bool_t warn = true) const;
117
118     TGeoHMatrix GetTransform(
119                   Double_t x, Double_t y, Double_t z,
120                   Double_t a1, Double_t a2, Double_t a3, 
121                   Double_t a4, Double_t a5, Double_t a6) const;
122
123     void FillModuleTransform(Int_t moduleId,
124                   Double_t x, Double_t y, Double_t z,
125                   Double_t a1, Double_t a2, Double_t a3, 
126                   Double_t a4, Double_t a5, Double_t a6); 
127     void FillDetElemTransform(Int_t id, 
128                   Double_t x, Double_t y, Double_t z,
129                   Double_t a1, Double_t a2, Double_t a3, 
130                   Double_t a4, Double_t a5, Double_t a6);
131
132     TString ReadModuleTransforms(ifstream& in);
133     TString ReadDetElemTransforms(ifstream& in);
134     Bool_t  ReadTransformations(const TString& fileName);
135
136     void    WriteTransform(ofstream& out, const TGeoMatrix* transform) const;
137     void    WriteModuleTransforms(ofstream& out) const;
138     void    WriteDetElemTransforms(ofstream& out) const;
139     
140     TString GetModuleSymName(Int_t moduleId) const;
141     TString GetDESymName(Int_t detElemId) const;
142
143     // data members
144     TString        fDetectorName;       ///< Detector name
145     TObjArray*     fModuleTransformers; ///< array of module transformers
146     TClonesArray*  fMisAlignArray;      ///< array of misalignment data
147     mutable AliMpExMap*    fDEAreas; ///< areas of detection elements in global coordinates
148     
149   ClassDef(AliMUONGeometryTransformer,4)  // Geometry parametrisation
150 };
151
152 // inline methods
153
154 /// Return the number of contained module transformers
155 inline Int_t AliMUONGeometryTransformer::GetNofModuleTransformers() const
156 { return fModuleTransformers->GetEntriesFast(); }
157
158 /// Return the array of misalignment data
159 inline const TClonesArray* AliMUONGeometryTransformer::GetMisAlignmentData() const      
160 { return fMisAlignArray; }                     
161                                
162 /// Set detector name
163 inline void AliMUONGeometryTransformer::SetDetName(const TString& detName)
164 {  fDetectorName = detName; }               
165
166 /// Set ownership of array module transformers
167 inline void AliMUONGeometryTransformer::SetOwner(Bool_t isOwner)
168 {  fModuleTransformers->SetOwner(isOwner); }               
169
170 #endif //ALI_MUON_GEOMETRY_TRANSFORMER_H
171
172
173
174
175
176
177