| | 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 | enum ELayerID{kInvalidLayer=0, |
| | 22 | kFirstLayer=1, |
| | 23 | kSPD1=1, kSPD2=2, |
| | 24 | kSDD1=3, kSDD2=4, |
| | 25 | kSSD1=5, kSSD2=6, |
| | 26 | kTPC1=7, kTPC2=8, |
| | 27 | kTRD1=9, kTRD2=10, kTRD3=11, kTRD4=12, kTRD5=13, kTRD6=14, |
| | 28 | kTOF=15, |
| | 29 | kPHOS1=16, kPHOS2=17, |
| | 30 | kRICH=18, |
| | 31 | kMUON=19, |
| | 32 | kLastLayer=20}; |
| | 33 | AliAlignObj(); |
| | 34 | AliAlignObj(const char* volpath, UShort_t voluid); |
| | 35 | AliAlignObj(const char* volpath, ELayerID detId, Int_t modId); |
| | 36 | AliAlignObj(const AliAlignObj& theAlignObj); |
| | 37 | AliAlignObj& operator= (const AliAlignObj& theAlignObj); |
| | 38 | AliAlignObj& operator*=(const AliAlignObj& theAlignObj); |
| | 39 | virtual ~AliAlignObj(); |
| | 40 | |
| | 41 | //Setters |
| | 42 | virtual void SetTranslation(Double_t x, Double_t y, Double_t z) = 0; |
| | 43 | virtual void SetTranslation(const TGeoMatrix& m) = 0; |
| | 44 | virtual void SetRotation(Double_t psi, Double_t theta, Double_t phi) = 0; |
| | 45 | virtual Bool_t SetRotation(const TGeoMatrix& m) = 0; |
| | 46 | virtual void SetPars(Double_t x, Double_t y, Double_t z, Double_t psi, |
| | 47 | Double_t theta, Double_t phi) = 0; |
| | 48 | virtual Bool_t SetLocalPars(Double_t x, Double_t y, Double_t z, |
| | 49 | Double_t psi, Double_t theta, Double_t phi); |
| | 50 | virtual void SetMatrix(const TGeoMatrix& m) = 0; |
| | 51 | void SetVolPath(const TString& volpath) {fVolPath=volpath;} |
| | 52 | void SetVolUID(UShort_t voluid) {fVolUID=voluid;} |
| | 53 | void SetVolUID(ELayerID layerId, Int_t modId); |
| | 54 | |
| | 55 | //Getters |
| | 56 | const char *GetVolPath() const {return fVolPath.Data();} |
| | 57 | UShort_t GetVolUID() const {return fVolUID;} |
| | 58 | void GetVolUID(ELayerID &layerId, Int_t &modId) const; |
| | 59 | virtual void GetTranslation(Double_t* tr) const=0; |
| | 60 | virtual Bool_t GetAngles(Double_t* angles) const=0; |
| | 61 | virtual void GetPars(Double_t transl[], Double_t rot[]) const=0; |
| | 62 | virtual void GetMatrix(TGeoHMatrix& m) const=0; |
| | 63 | |
| | 64 | Bool_t IsSortable() const {return kTRUE;} |
| | 65 | Int_t GetLevel() const; |
| | 66 | virtual Int_t Compare(const TObject* obj) const; |
| | 67 | |
| | 68 | virtual AliAlignObj& Inverse() const=0; |
| | 69 | |
| | 70 | void Transform(AliTrackPoint &p) const; |
| | 71 | void Transform(AliTrackPointArray &array) const; |
| | 72 | |
| | 73 | void Print(Option_t *) const; |
| | 74 | |
| | 75 | static Int_t LayerSize(Int_t layer) { return fgLayerSize[layer]; } |
| | 76 | static const char* LayerName(Int_t layer) { return fgLayerName[layer]; } |
| | 77 | |
| | 78 | static UShort_t LayerToVolUID(ELayerID layerId, Int_t modId); |
| | 79 | static UShort_t LayerToVolUID(Int_t layerId, Int_t modId); |
| | 80 | static ELayerID VolUIDToLayer(UShort_t voluid, Int_t &modId); |
| | 81 | static ELayerID VolUIDToLayer(UShort_t voluid); |
| | 82 | |
| | 83 | static const char* GetVolPath(UShort_t voluid) { |
| | 84 | Int_t modId; |
| | 85 | ELayerID layerId = VolUIDToLayer(voluid,modId); |
| | 86 | return GetVolPath(layerId,modId); |
| | 87 | } |
| | 88 | static const char* GetVolPath(ELayerID layerId, Int_t modId); |
| | 89 | |
| | 90 | Bool_t ApplyToGeometry(); |
| | 91 | static Bool_t GetFromGeometry(const char *path, AliAlignObj &alobj); |
| | 92 | |
| | 93 | static void InitAlignObjFromGeometry(); |
| | 94 | static AliAlignObj* GetAlignObj(UShort_t voluid) { |
| | 95 | Int_t modId; |
| | 96 | ELayerID layerId = VolUIDToLayer(voluid,modId); |
| | 97 | return GetAlignObj(layerId,modId); |
| | 98 | } |
| | 99 | static AliAlignObj* GetAlignObj(ELayerID layerId, Int_t modId); |
| | 100 | |
| | 101 | protected: |
| | 102 | |
| | 103 | void AnglesToMatrix(const Double_t *angles, Double_t *rot) const; |
| | 104 | Bool_t MatrixToAngles(const Double_t *rot, Double_t *angles) const; |
| | 105 | |
| | 106 | static void InitVolPaths(); |
| | 107 | |
| | 108 | //Volume identifiers |
| | 109 | TString fVolPath; // Volume path inside TGeo geometry |
| | 110 | UShort_t fVolUID; // Unique volume ID |
| | 111 | |
| | 112 | static Int_t fgLayerSize[kLastLayer - kFirstLayer]; |
| | 113 | static const char* fgLayerName[kLastLayer - kFirstLayer]; |
| | 114 | |
| | 115 | static TString* fgVolPath[kLastLayer - kFirstLayer]; |
| | 116 | |
| | 117 | static AliAlignObj** fgAlignObjs[kLastLayer - kFirstLayer]; |
| | 118 | |
| | 119 | ClassDef(AliAlignObj, 2) |
| | 120 | }; |
| | 121 | |
| | 122 | #endif |
git://git.uio.no / u / mrichter / AliRoot.git / blame_incremental