1 #ifndef ALIITSALIGNMILLE_H
2 #define ALIITSALIGNMILLE_H
3 /* Copyright(c) 2007-2009 , ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
10 /// \class AliITSAlignMille
11 /// \brief Class for alignment of ITS
13 // Authors: Marcello Lunardon
17 #include "AliTrackPointArray.h"
19 // number of used objects
20 #define ITSMILLENDETELEM 2198
21 #define ITSMILLENPARCH 6
22 #define ITSMILLENLOCAL 5
23 #define ITSMILLENSTDEV 3
26 class AliAlignObjParams;
29 class AliITSAlignMilleModule;
30 class AliTrackFitterRieman;
31 class AliITSAlignMilleData;
33 class AliITSAlignMille:public TObject
36 AliITSAlignMille(const Char_t *configFilename="AliITSAlignMille.conf", Bool_t initmille=kTRUE);
37 virtual ~AliITSAlignMille();
40 Int_t GetModuleIndex(const Char_t *symname);
41 Int_t GetModuleIndex(UShort_t voluid);
42 UShort_t GetModuleVolumeID(const Char_t *symname);
43 UShort_t GetModuleVolumeID(Int_t index);
44 void SetCurrentModule(Int_t index);
45 void SetCurrentSensitiveModule(Int_t index); // set as current the SENSITIVE module with index 'index'
47 // configuration methods
48 void SetGeometryFileName(const Char_t* filename="geometry.root")
49 { fGeometryFileName = filename; }
50 const Char_t* GetGeometryFileName() {return fGeometryFileName.Data();}
51 const Char_t* GetPreAlignmentFileName() {return fPreAlignmentFileName.Data();}
52 void PrintCurrentModuleInfo();
53 void Print(Option_t*) const;
54 Bool_t IsConfigured() const {return fIsConfigured;}
55 void SetRequiredPoint(Char_t* where, Int_t ndet, Int_t updw, Int_t nreqpts);
58 void SetMinNPtsPerTrack(Int_t pts=3) {fMinNPtsPerTrack=pts;}
59 Int_t ProcessTrack(AliTrackPointArray *track);
60 AliTrackPointArray *PrepareTrack(const AliTrackPointArray *track); // build a new AliTrackPointArray with selected conditions
61 void InitTrackParams(int meth=1);
62 Bool_t InitRiemanFit();
63 AliTrackFitterRieman *GetRiemanFitter() const {return fRieman;}
64 Int_t InitModuleParams();
65 Int_t CheckCurrentTrack();
66 Bool_t CheckVolumeID(UShort_t voluid) const; // checks voluid for sensitive volumes
67 Int_t IsDefined(UShort_t voluid) const;
68 Int_t IsContained(UShort_t voluid) const;
69 Int_t CalcIntersectionPoint(const Double_t *lpar, const Double_t *gpar);
70 Int_t CalcDerivatives(Int_t paridx, Bool_t islpar);
71 const Double_t* GetLocalIntersectionPoint() const {return fPintLoc;}
72 const Double_t* GetGlobalIntersectionPoint() const {return fPintGlo;}
73 void SetInitTrackParamsMeth(Int_t meth=1) {fInitTrackParamsMeth=meth;}
74 AliTrackPointArray *SortTrack(const AliTrackPointArray *atp);
75 void SetTemporaryExcludedModule(Int_t index) {fTempExcludedModule=index;}
78 void FixParameter(Int_t param, Double_t value);
79 void AddConstraint(Double_t *factor, Double_t value );
80 void InitGlobalParameters(Double_t *par);
81 void SetLocalDerivative(Int_t index, Double_t value)
82 {fLocalDerivatives[index] = value;}
83 void SetGlobalDerivative(Int_t index, Double_t value)
84 {fGlobalDerivatives[index] = value;}
85 void LocalFit(Int_t iTrack, Double_t *lTrackParam, Int_t lSingleFit);
86 void GlobalFit(Double_t *parameters,Double_t *errors,Double_t *pulls);
87 void PrintGlobalParameters();
88 Double_t GetParError(Int_t iPar);
89 Int_t AddLocalEquation(AliITSAlignMilleData &m);
90 void SetLocalEquations(const AliITSAlignMilleData *m, Int_t neq);
93 AliTrackPointArray *GetCurrentTrack() const {return fTrack;}
94 const AliTrackPoint *GetCurrentCluster() const {return &fCluster;}
95 void SetCurrentTrack(AliTrackPointArray * const atp) {fTrack=atp;}
96 void SetCurrentCluster(const AliTrackPoint &atp) {fCluster=atp;}
99 Int_t GetNModules() const {return fNModules;}
100 Int_t GetCurrentModuleIndex() const {return fCurrentModuleIndex;}
101 TGeoHMatrix *GetCurrentModuleHMatrix() const {return fCurrentModuleHMatrix;}
102 const Double_t *GetCurrentModuleTranslation() const {return fCurrentModuleTranslation;}
103 Int_t GetCurrentModuleInternalIndex() const {return fCurrentModuleInternalIndex;}
104 const Int_t *GetModuleIndexArray() const {return fModuleIndex;}
105 const Int_t *GetProcessedPoints() const {return fProcessedPoints;}
106 Int_t GetTotBadLocEqPoints() const {return fTotBadLocEqPoints;}
107 AliITSAlignMilleModule *GetMilleModule(UShort_t voluid) const; // get pointer to the defined supermodule
108 AliITSAlignMilleModule *GetCurrentModule() const;
109 const UShort_t *GetModuleVolumeIDArray() const {return fModuleVolumeID;}
112 const Double_t *GetMeasLoc() const { return fMeasLoc;}
113 const Double_t *GetSigmaLoc() const { return fSigmaLoc;}
114 Double_t GetBField() const {return fBField;}
115 const Double_t *GetLocalInitParam() const {return fLocalInitParam;}
116 Double_t GetLocalDX() const {return fDerivativeXLoc;}
117 Double_t GetLocalDZ() const {return fDerivativeZLoc;}
118 Double_t GetParSigTranslations() const {return fParSigTranslations;}
119 Double_t GetParSigRotations() const {return fParSigRotations;}
120 Int_t GetPreAlignmentQualityFactor(Int_t index) const; // if not prealign. return -1
121 void SetBug(Int_t bug) {fBug=bug;} // 1:SSD inversion sens.18-19
125 // configuration methods
126 Int_t LoadConfig(const Char_t *cfile="AliITSAlignMille.conf");
127 Int_t LoadSuperModuleFile(const Char_t *cfile="ITSMilleSuperModules.root");
128 void ResetLocalEquation();
130 Int_t ApplyToGeometry();
133 void Init(Int_t nGlobal, Int_t nLocal, Int_t nStdDev);
136 AliMillepede *fMillepede; ///< Detector independent alignment class
137 static Int_t fgNParCh; ///< Number of degrees of freedom per chamber
138 static Int_t fgNDetElem; ///< Total number of detection elements
139 Double_t fStartFac; ///< Initial value for chi2 cut
140 ///< if > 1 Iterations in AliMil. are turned on
141 Double_t fResCutInitial; ///< Cut on residual for first iteration
142 Double_t fResCut; ///< Cut on residual for other iterations
143 Int_t fNGlobal; ///< Number of global parameters
144 Int_t fNLocal; ///< Number of local parameters
145 Int_t fNStdDev; ///< Number of standard deviations for chi2 cut
146 Bool_t fIsMilleInit; ///
147 Double_t fParSigTranslations; ///< init sigma for transl. params [cm]
148 Double_t fParSigRotations; ///< init sigma for rot. params [deg]
151 AliTrackPointArray *fTrack; ///< pointer to current track
152 AliTrackPoint fCluster; ///< current cluster
153 Double_t *fGlobalDerivatives; ///< Array of global derivatives
154 Double_t fLocalDerivatives[ITSMILLENLOCAL]; ///< Array of local deriv.
155 Double_t fLocalInitParam[ITSMILLENLOCAL]; ///< Array with inital values for local parameters for current track
156 Double_t fModuleInitParam[ITSMILLENPARCH]; ///< Array with inital values for current module parameters (init geometry)
157 Double_t fPintLoc[3]; ///
158 Double_t fPintLoc0[3]; ///
159 Double_t fPintGlo[3]; ///
160 Double_t fMeasLoc[3]; // current point local coordinates (the original ones)
161 Double_t fMeasGlo[3]; // current point glob. coord (AliTrackPoint)
162 Double_t fSigmaLoc[3]; // stdev current point
163 Double_t fSigmaXfactor; ///
164 Double_t fSigmaZfactor; ///
165 AliAlignObjParams *fTempAlignObj; ///
166 Double_t fDerivativeXLoc; // localX deriv.
167 Double_t fDerivativeZLoc; // localZ deriv.
168 Int_t fMinNPtsPerTrack; ///
169 Int_t fInitTrackParamsMeth; ///
170 Int_t *fProcessedPoints; /// array of statistics of used points per module
171 Int_t fTotBadLocEqPoints; /// total number of reject points because of bad EqLoc
172 AliTrackFitterRieman *fRieman; /// riemann fitter for helices
173 Bool_t fRequirePoints; // required points in specific layers
174 Int_t fNReqLayUp[6]; /// number of points required in layer[n] with Y>0
175 Int_t fNReqLayDown[6]; /// number of points required in layer[n] with Y<0
176 Int_t fNReqLay[6]; /// number of points required in layer[n]
177 Int_t fNReqDetUp[3]; /// number of points required in Detector[n] with Y>0
178 Int_t fNReqDetDown[3]; /// number of points required in Detector[n] with Y<0
179 Int_t fNReqDet[3]; /// number of points required in Detector[n]
180 Int_t fTempExcludedModule; /// single module temporary excluded from initial fit
183 TString fGeometryFileName; ///
184 TString fPreAlignmentFileName; ///
185 TGeoManager *fGeoManager; ///
186 Int_t fCurrentModuleIndex; /// SuperModule index
187 Int_t fCurrentModuleInternalIndex; /// SuperModule internal index
188 Int_t fCurrentSensVolIndex; /// Current point (sens. vol.) index
189 Double_t fCurrentModuleTranslation[3]; ///
190 Int_t fNModules; /// number of defined modules from config file
191 Int_t fModuleIndex[ITSMILLENDETELEM*2]; ///
192 UShort_t fModuleVolumeID[ITSMILLENDETELEM*2]; ///
193 Bool_t fFreeParam[ITSMILLENDETELEM*2][ITSMILLENPARCH]; ///
194 Bool_t fUseLocalShifts; ///
195 Bool_t fUseSuperModules; ///
196 Bool_t fUsePreAlignment; ///
197 Bool_t fUseSortedTracks; /// default is kTRUE
198 Bool_t fBOn; /// magentic field ON
199 Double_t fBField; /// value of magnetic field
200 Int_t fNSuperModules; /// number of custom supermodules in SM file
201 TGeoHMatrix *fCurrentModuleHMatrix; /// SuperModule matrix
202 Bool_t fIsConfigured; ///
203 Int_t fPreAlignQF[ITSMILLENDETELEM*2]; ///
204 Double_t fSensVolSigmaXfactor[ITSMILLENDETELEM*2]; ///
205 Double_t fSensVolSigmaZfactor[ITSMILLENDETELEM*2]; ///
206 Int_t fBug; /// tag for temporary bug correction
208 AliITSAlignMilleModule *fMilleModule[ITSMILLENDETELEM*2]; /// array of super modules to be aligned
210 AliITSAlignMilleModule *fSuperModule[ITSMILLENDETELEM*2]; /// array of super modules defined in supermodule file
212 AliITSAlignMille(const AliITSAlignMille& rhs);
213 AliITSAlignMille& operator=(const AliITSAlignMille& rhs);
216 ClassDef(AliITSAlignMille, 0)