New version of AliAlignObj. The new things are the ITS mapping and some added functio...
[u/mrichter/AliRoot.git] / STEER / AliAlignObj.h
1 #ifndef ALIALIGNOBJ_H
2 #define ALIALIGNOBJ_H
3
4 //************************************************************************
5 // AliAlignObj: alignment base class for the storage of alignment        *
6 //   information for a single volume, that is a translation, a rotation  *
7 //   and a the identity of the volume itself in form of a TGeo path and  *
8 //   as a unique integer identifier                                      *
9 //************************************************************************
10 #include "TObject.h"
11 #include "TString.h"
12 #include "TGeoMatrix.h"
13
14 class AliTrackPoint;
15 class AliTrackPointArray;
16
17 class AliAlignObj : public TObject {
18
19  public:
20
21   AliAlignObj();
22   AliAlignObj(const AliAlignObj& theAlignObj);
23   AliAlignObj& operator= (const AliAlignObj& theAlignObj);
24   virtual ~AliAlignObj();
25   enum ELayerID{kInvalidLayer=0,
26                 kFirstLayer=1,
27                 kSPD1=1, kSPD2=2,
28                 kSDD1=3, kSDD2=4,
29                 kSSD1=5, kSSD2=6,
30                 kTPC1=7, kTPC2=8,
31                 kTRD1=9, kTRD2=10, kTRD3=11, kTRD4=12, kTRD5=13, kTRD6=14,
32                 kTOF=15,
33                 kPHOS1=16, kPHOS2=17,
34                 kRICH=18,
35                 kMUON=19,
36                 kLastLayer=20};
37
38   //Setters
39   virtual void SetTranslation(Double_t x, Double_t y, Double_t z) = 0;
40   virtual void SetTranslation(const TGeoMatrix& m) = 0;
41   virtual void SetRotation(Double_t psi, Double_t theta, Double_t phi) = 0;
42   virtual Bool_t SetRotation(const TGeoMatrix& m) = 0;
43   virtual void SetPars(Double_t x, Double_t y, Double_t z, Double_t psi,
44                Double_t theta, Double_t phi) = 0;
45   virtual void SetMatrix(const TGeoMatrix& m) = 0;
46   void  SetVolPath(const TString& volpath) {fVolPath=volpath;}
47   void  SetVolUID(UShort_t voluid) {fVolUID=voluid;}
48   void  SetVolUID(ELayerID layerId, Int_t modId);
49
50   //Getters
51   const char  *GetVolPath()    const {return fVolPath.Data();}
52   UShort_t     GetVolUID()     const {return fVolUID;}
53   void         GetVolUID(ELayerID &layerId, Int_t &modId) const;
54   virtual void GetTranslation(Double_t* tr)  const=0;
55   virtual Bool_t GetAngles(Double_t* angles) const=0;
56   virtual void GetPars(Double_t transl[], Double_t rot[]) const=0;
57   virtual void GetMatrix(TGeoHMatrix& m) const=0;
58
59   virtual AliAlignObj& Inverse() const=0;
60
61   void  Transform(AliTrackPoint &p) const;
62   void  Transform(AliTrackPointArray &array) const;
63
64   void  Print(Option_t *) const;
65
66   static Int_t       LayerSize(Int_t layer) { return fgLayerSize[layer]; }
67   static const char* LayerName(Int_t layer) { return fgLayerName[layer]; }
68
69   static UShort_t LayerToVolUID(ELayerID layerId, Int_t modId);
70   static ELayerID VolUIDToLayer(UShort_t voluid, Int_t &modId);
71   static ELayerID VolUIDToLayer(UShort_t voluid);
72
73   static const char* GetVolPath(UShort_t voluid) {
74     Int_t modId;
75     ELayerID layerId = VolUIDToLayer(voluid,modId);
76     return GetVolPath(layerId,modId);
77   }
78   static const char* GetVolPath(ELayerID layerId, Int_t modId) { return fgVolPath[layerId-kFirstLayer][modId]; }
79
80  protected:
81
82   void AnglesToMatrix(const Double_t *angles, Double_t *rot) const;
83   Bool_t MatrixToAngles(const Double_t *rot, Double_t *angles) const;
84
85   void InitVolPaths();
86
87   //Volume identifiers
88   TString  fVolPath; // Volume path inside TGeo geometry
89   UShort_t fVolUID;  // Unique volume ID
90
91   static Int_t       fgLayerSize[kLastLayer - kFirstLayer];
92   static const char* fgLayerName[kLastLayer - kFirstLayer];
93
94   static const char**fgVolPath[kLastLayer - kFirstLayer];
95
96   ClassDef(AliAlignObj, 2)
97 };
98
99 #endif