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