]>
Commit | Line | Data |
---|---|---|
7b85e477 | 1 | /************************************************************************** |
2 | * Copyright(c) 2007-2009, ALICE Experiment at CERN, All rights reserved. * | |
3 | * * | |
4 | * Author: The ALICE Off-line Project. * | |
5 | * Contributors are mentioned in the code where appropriate. * | |
6 | * * | |
7 | * Permission to use, copy, modify and distribute this software and its * | |
8 | * documentation strictly for non-commercial purposes is hereby granted * | |
9 | * without fee, provided that the above copyright notice appears in all * | |
10 | * copies and that both the copyright notice and this permission notice * | |
11 | * appear in the supporting documentation. The authors make no claims * | |
12 | * about the suitability of this software for any purpose. It is * | |
13 | * provided "as is" without express or implied warranty. * | |
14 | **************************************************************************/ | |
15 | ||
057304f3 | 16 | /* $Id$ */ |
7b85e477 | 17 | |
18 | //----------------------------------------------------------------------------- | |
19 | /// \class AliITSAlignMille | |
20 | /// Alignment class fro the ALICE ITS detector | |
21 | /// | |
22 | /// ITS specific alignment class which interface to AliMillepede. | |
23 | /// For each track ProcessTrack calculates the local and global derivatives | |
24 | /// at each hit and fill the corresponding local equations. Provide methods for | |
25 | /// fixing or constraining detection elements for best results. | |
26 | /// | |
27 | /// \author M. Lunardon (thanks to J. Castillo) | |
28 | //----------------------------------------------------------------------------- | |
29 | ||
30 | #include <TF1.h> | |
31 | #include <TFile.h> | |
32 | #include <TClonesArray.h> | |
33 | #include <TGraph.h> | |
34 | #include <TGeoMatrix.h> | |
35 | #include <TMath.h> | |
36 | #include <TGraphErrors.h> | |
37 | #include <TVirtualFitter.h> | |
38 | ||
39 | #include "AliITSAlignMille2.h" | |
40 | #include "AliITSgeomTGeo.h" | |
41 | #include "AliGeomManager.h" | |
42 | #include "AliMillePede2.h" | |
43 | #include "AliTrackPointArray.h" | |
44 | #include "AliAlignObjParams.h" | |
45 | #include "AliLog.h" | |
46 | #include "TSystem.h" // come si fa? | |
47 | #include "AliTrackFitterRieman.h" | |
48 | ||
49 | /// \cond CLASSIMP | |
50 | ClassImp(AliITSAlignMille2) | |
51 | /// \endcond | |
52 | ||
53 | AliITSAlignMille2* AliITSAlignMille2::fgInstance = 0; | |
54 | Int_t AliITSAlignMille2::fgInstanceID = 0; | |
55 | ||
56 | AliITSAlignMille2::AliITSAlignMille2(const Char_t *configFilename, Bool_t initmille) | |
57 | : TObject(), | |
58 | fMillepede(0), | |
59 | fStartFac(16.), | |
60 | fResCutInitial(100.), | |
61 | fResCut(100.), | |
62 | fNGlobal(0), | |
63 | fNLocal(4), | |
64 | fNStdDev(3), | |
65 | fIsMilleInit(kFALSE), | |
66 | fSensorsIn(kFALSE), | |
67 | fParSigTranslations(0.0100), | |
68 | fParSigRotations(0.1), | |
69 | // | |
70 | fCurrentModule(0), | |
71 | fTrack(0), | |
72 | fCluster(), | |
73 | fGlobalDerivatives(0), | |
74 | // | |
75 | fMinNPtsPerTrack(3), | |
76 | fInitTrackParamsMeth(1), | |
77 | fTotBadLocEqPoints(0), | |
78 | fRieman(0), | |
79 | // | |
80 | fUseGlobalDelta(kFALSE), | |
81 | fRequirePoints(kFALSE), | |
82 | fTempExcludedModule(-1), | |
83 | // | |
84 | fGeometryFileName("geometry.root"), | |
85 | fPreAlignmentFileName(""), | |
86 | fGeoManager(0), | |
87 | fIsConfigured(kFALSE), | |
88 | fPreAlignQF(0), | |
89 | // | |
90 | fPrealignment(0), | |
91 | fMilleModule(2), | |
92 | fSuperModule(2), | |
93 | fNModules(0), | |
94 | fNSuperModules(0), | |
95 | fUsePreAlignment(kFALSE), | |
96 | fUseSortedTracks(kTRUE), | |
97 | fBOn(kFALSE), | |
98 | fBField(0.0), | |
99 | fBug(0) | |
100 | { | |
101 | /// main constructor that takes input from configuration file | |
102 | // | |
103 | fMillepede = new AliMillePede2(); | |
104 | for (int i=3;i--;) fSigmaFactor[i] = 1.0; | |
105 | // | |
106 | // new RS | |
107 | for (Int_t i=0; i<6; i++) { | |
108 | fNReqLayUp[i]=0; | |
109 | fNReqLayDown[i]=0; | |
110 | fNReqLay[i]=0; | |
111 | } | |
112 | for (Int_t i=0; i<3; i++) { | |
113 | fNReqDetUp[i]=0; | |
114 | fNReqDetDown[i]=0; | |
115 | fNReqDet[i]=0; | |
116 | } | |
117 | // | |
118 | Int_t lc=LoadConfig(configFilename); | |
119 | if (lc) { | |
120 | AliInfo(Form("Error %d loading configuration from %s",lc,configFilename)); | |
121 | } | |
122 | else { | |
123 | fIsConfigured=kTRUE; | |
124 | if (initmille) { | |
125 | AliInfo(Form("Initializing Millepede with %d gpar, %d lpar and %d stddev ...",fNGlobal, fNLocal, fNStdDev)); | |
126 | Init(fNGlobal, fNLocal, fNStdDev); | |
127 | ResetLocalEquation(); | |
128 | AliInfo("Parameters initialized to zero"); | |
129 | } | |
130 | else { | |
131 | AliInfo("Millepede has not been initialized ... "); | |
132 | } | |
133 | } | |
134 | // | |
135 | fgInstance = this; | |
136 | fgInstanceID++; | |
137 | // | |
138 | } | |
139 | ||
140 | AliITSAlignMille2::~AliITSAlignMille2() { | |
141 | /// Destructor | |
142 | if (fMillepede) delete fMillepede; | |
143 | if (fGlobalDerivatives) delete[] fGlobalDerivatives; | |
144 | if (fRieman) delete fRieman; | |
145 | if (fPrealignment) delete fPrealignment; | |
146 | fMilleModule.Delete(); | |
147 | fSuperModule.Delete(); | |
148 | if (--fgInstanceID==0) fgInstance = 0; | |
149 | } | |
150 | ||
151 | /////////////////////////////////////////////////////////////////////// | |
152 | ||
153 | Int_t AliITSAlignMille2::LoadConfig(const Char_t *cfile) { | |
154 | /// return 0 if success | |
155 | /// 1 if error in module index or voluid | |
156 | ||
157 | FILE *pfc=fopen(cfile,"r"); | |
158 | if (!pfc) return -1; | |
159 | ||
160 | Char_t st[200],st2[200]; | |
161 | Char_t tmp[100]; | |
162 | Int_t idx,itx,ity,itz,ith,ips,iph; | |
163 | Float_t f1,f2,f3; | |
164 | UShort_t voluid; | |
165 | Int_t nmod=0; | |
166 | // | |
167 | while (fgets(st,200,pfc)) { | |
168 | // | |
169 | for (int i=0; i<int(strlen(st)); i++) if (st[i]=='#') st[i]=0; // skip comments | |
170 | // | |
171 | if (strstr(st,"GEOMETRY_FILE")) { | |
172 | sscanf(st,"%s %s",tmp,st2); | |
173 | if (gSystem->AccessPathName(st2)) { AliInfo("*** WARNING! *** geometry file not found! "); return -1;} | |
174 | fGeometryFileName=st2; | |
175 | InitGeometry(); | |
176 | } | |
177 | // | |
178 | if (strstr(st,"PREALIGNMENT_FILE")) { | |
179 | sscanf(st,"%s %s",tmp,st2); | |
180 | if (gSystem->AccessPathName(st2)) { AliInfo("*** WARNING! *** prealignment file not found! "); return -1;} | |
181 | fPreAlignmentFileName=st2; | |
182 | itx=ApplyToGeometry(); | |
183 | if (itx) { AliInfo(Form("*** WARNING! *** error %d reading prealignment file! ",itx)); return -6;} | |
184 | } | |
185 | // | |
186 | if (strstr(st,"SUPERMODULE_FILE")) { | |
187 | sscanf(st,"%s %s",tmp,st2); | |
188 | if (gSystem->AccessPathName(st2)) { AliInfo("*** WARNING! *** supermodule file not found! "); return -1;} | |
189 | if (LoadSuperModuleFile(st2)) return -1; | |
190 | } | |
191 | // | |
192 | if (strstr(st,"SET_B_FIELD")) { | |
193 | sscanf(st,"%s %f",tmp,&f1); | |
194 | if (f1>0) { | |
195 | fBField = f1; | |
196 | fBOn = kTRUE; | |
197 | fNLocal = 5; // helices | |
198 | fRieman = new AliTrackFitterRieman(); | |
199 | } | |
200 | else { | |
201 | fBField = 0.0; | |
202 | fBOn = kFALSE; | |
203 | fNLocal = 4; | |
204 | } | |
205 | } | |
206 | // | |
207 | if (strstr(st,"SET_MINPNT_TRA")) { | |
208 | sscanf(st,"%s %d",tmp,&idx); | |
209 | fMinNPtsPerTrack=idx; | |
210 | } | |
211 | // | |
212 | if (strstr(st,"SET_PARSIG_TRA")) { | |
213 | sscanf(st,"%s %f",tmp,&f1); | |
214 | fParSigTranslations=f1; | |
215 | } | |
216 | // | |
217 | if (strstr(st,"SET_PARSIG_ROT")) { | |
218 | sscanf(st,"%s %f",tmp,&f1); | |
219 | fParSigRotations=f1; | |
220 | } | |
221 | // | |
222 | if (strstr(st,"SET_NSTDDEV")) { | |
223 | sscanf(st,"%s %d",tmp,&idx); | |
224 | fNStdDev=idx; | |
225 | } | |
226 | // | |
227 | if (strstr(st,"SET_RESCUT_INIT")) { | |
228 | sscanf(st,"%s %f",tmp,&f1); | |
229 | fResCutInitial=f1; | |
230 | } | |
231 | // | |
232 | if (strstr(st,"SET_RESCUT_OTHER")) { | |
233 | sscanf(st,"%s %f",tmp,&f1); | |
234 | fResCut=f1; | |
235 | } | |
236 | // | |
237 | if (strstr(st,"SET_LOCALSIGMAFACTOR")) { | |
238 | f1=f2=f3=0; | |
239 | sscanf(st,"%s %f %f %f",tmp,&f1,&f2,&f3); | |
240 | if (f1>0) fSigmaFactor[0] = f1; | |
241 | if (f2>0) fSigmaFactor[1] = f2; else fSigmaFactor[1]=fSigmaFactor[0]; | |
242 | if (f3>0) fSigmaFactor[2] = f3; else fSigmaFactor[2]=fSigmaFactor[1]; | |
243 | } | |
244 | // | |
245 | if (strstr(st,"SET_STARTFAC")) { | |
246 | sscanf(st,"%s %f",tmp,&f1); | |
247 | fStartFac=f1; | |
248 | } | |
249 | // | |
250 | // >> RS | |
251 | if (strstr(st,"REQUIRE_POINT")) { | |
252 | // syntax: REQUIRE_POINT where ndet updw nreqpts | |
253 | // where = LAYER or DETECTOR | |
254 | // ndet = detector number: 1-6 for LAYER and 1-3 for DETECTOR (SPD=1, SDD=2, SSD=3) | |
255 | // updw = 1 for Y>0, -1 for Y<0, 0 if not specified | |
256 | // nreqpts = minimum number of points of that type | |
257 | sscanf(st,"%s %s %d %d %d",tmp,st2,&itx,&ity,&itz); | |
258 | itx--; | |
259 | if (strstr(st2,"LAYER")) { | |
260 | if (itx<0 || itx>5) return -7; | |
261 | if (ity>0) fNReqLayUp[itx]=itz; | |
262 | else if (ity<0) fNReqLayDown[itx]=itz; | |
263 | else fNReqLay[itx]=itz; | |
264 | fRequirePoints=kTRUE; | |
265 | } | |
266 | else if (strstr(st2,"DETECTOR")) { // DETECTOR | |
267 | if (itx<0 || itx>2) return -7; | |
268 | if (ity>0) fNReqDetUp[itx]=itz; | |
269 | else if (ity<0) fNReqDetDown[itx]=itz; | |
270 | else fNReqDet[itx]=itz; | |
271 | fRequirePoints=kTRUE; | |
272 | } | |
273 | } | |
274 | // << RS | |
275 | ||
276 | // | |
277 | if (strstr(st,"MODULE_INDEX") || strstr(st,"MODULE_VOLUID")) { | |
278 | f1=f2=f3=0; | |
279 | sscanf(st,"%s %d %d %d %d %d %d %d %f %f %f",tmp,&idx,&itx,&ity,&itz,&iph,&ith,&ips,&f1,&f2,&f3); | |
280 | // | |
281 | if (idx<=kMaxITSSensID) voluid=GetModuleVolumeID(idx); | |
282 | else voluid = UShort_t(idx); | |
283 | // | |
284 | if (voluid>=kMinITSSupeModuleID) { // custom supermodule | |
285 | int ism=-1; | |
286 | for (int j=0; j<fNSuperModules; j++) if (voluid==GetSuperModule(j)->GetVolumeID()) ism=j; | |
287 | if (ism<0) return -1; // bad volid | |
288 | fMilleModule.AddAtAndExpand(new AliITSAlignMille2Module(*GetSuperModule(ism)),nmod); | |
289 | // >> RS | |
290 | // if (f1>0) { | |
291 | // for (int kk=0; kk<GetMilleModule(nmod)->GetNSensitiveVolumes(); kk++) { | |
292 | // idx=AliITSAlignMille2Module::GetIndexFromVolumeID(GetMilleModule(nmod)->GetSensitiveVolumeVolumeID()[kk]); | |
293 | // if (idx>=0) fSensVolSigmaXfactor[idx]=f1; | |
294 | // } | |
295 | // } | |
296 | // if (f2>0) { | |
297 | // for (int kk=0; kk<GetMilleModule(nmod)->GetNSensitiveVolumes(); kk++) { | |
298 | // idx=AliITSAlignMille2Module::GetIndexFromVolumeID(GetMilleModule(nmod)->GetSensitiveVolumeVolumeID()[kk]); | |
299 | // if (idx>=0) fSensVolSigmaZfactor[idx]=f2; | |
300 | // } | |
301 | // } | |
302 | // << RS | |
303 | } | |
304 | else if (idx<=kMaxITSSensVID) { | |
305 | fMilleModule.AddAtAndExpand(new AliITSAlignMille2Module(voluid),nmod); | |
306 | AliITSAlignMille2Module* md = (AliITSAlignMille2Module*) fMilleModule[nmod]; | |
307 | fSensorsIn = kTRUE; | |
308 | md->SetSensorsProvided(); | |
309 | } | |
310 | else return -1; // bad volid | |
311 | // | |
312 | AliITSAlignMille2Module* mod = GetMilleModule(nmod); | |
313 | mod->SetFreeDOF(kDOFTX,itx!=0); | |
314 | mod->SetFreeDOF(kDOFTY,ity!=0); | |
315 | mod->SetFreeDOF(kDOFTZ,itz!=0); | |
316 | mod->SetFreeDOF(kDOFPH,iph!=0); | |
317 | mod->SetFreeDOF(kDOFTH,ith!=0); | |
318 | mod->SetFreeDOF(kDOFPS,ips!=0); | |
319 | // | |
320 | mod->SetUniqueID(nmod); | |
321 | if (f1>0) mod->SetSigmaXFactor(f1); | |
322 | if (f2>0) mod->SetSigmaYFactor(f2); else mod->SetSigmaYFactor(mod->GetSigmaXFactor()); | |
323 | if (f3>0) mod->SetSigmaZFactor(f3); else mod->SetSigmaZFactor(mod->GetSigmaYFactor()); | |
324 | nmod++; | |
325 | } | |
326 | // | |
327 | } // end while | |
328 | // | |
329 | fNModules = nmod; | |
330 | fNGlobal = fNModules*kNParCh; | |
331 | // | |
332 | fclose(pfc); | |
333 | // | |
334 | // set parent/child relationship for modules to align | |
335 | printf("Setting parent/child relationships\n"); | |
336 | // | |
337 | for (int ipar=0;ipar<nmod;ipar++) { | |
338 | AliITSAlignMille2Module* parent = GetMilleModule(ipar); | |
339 | if (parent->GetIndex()<=kMaxITSSensID) continue; // sensor cannot be a parent | |
340 | // | |
341 | for (int icld=0;icld<nmod;icld++) { | |
342 | if (icld==ipar) continue; | |
343 | AliITSAlignMille2Module* child = GetMilleModule(icld); | |
344 | if (!child->BelongsTo(parent)) continue; | |
345 | // | |
346 | AliITSAlignMille2Module* parOld = child->GetParent(); | |
347 | if (parOld && parOld->GetNSensitiveVolumes()<parent->GetNSensitiveVolumes()) continue; // parOld is closer | |
348 | child->SetParent(parent); | |
349 | } | |
350 | // | |
351 | } | |
352 | // | |
353 | // reorder the modules in such a way that parents come first | |
354 | for (int icld=0;icld<nmod;icld++) { | |
057304f3 | 355 | AliITSAlignMille2Module* child = GetMilleModule(icld); |
356 | AliITSAlignMille2Module* parent; | |
357 | while ( (parent=child->GetParent()) && (parent->GetUniqueID()<child->GetUniqueID()) ) { | |
7b85e477 | 358 | // swap |
359 | fMilleModule[icld] = parent; | |
360 | fMilleModule[parent->GetUniqueID()] = child; | |
361 | child->SetUniqueID(parent->GetUniqueID()); | |
362 | parent->SetUniqueID(icld); | |
057304f3 | 363 | child = parent; |
7b85e477 | 364 | } |
057304f3 | 365 | // |
366 | } | |
7b85e477 | 367 | // |
368 | // go over the child->parent chain and mark modules with explicitly provided sensors | |
369 | for (int icld=nmod;icld--;) { | |
370 | AliITSAlignMille2Module* child = GetMilleModule(icld); | |
371 | AliITSAlignMille2Module* parent = child->GetParent(); | |
372 | if (!parent) continue; | |
373 | parent->SetSensorsProvided( child->AreSensorsProvided() ); | |
374 | if (!parent->AreSensorsProvided()) continue; | |
375 | // suppress unused sensors | |
376 | for (int isn=0;isn<parent->GetNSensitiveVolumes();isn++) { | |
377 | int snVID = parent->GetSensVolVolumeID(isn); | |
378 | // check if this sensor is explicitly requested | |
379 | for (int imd=nmod;imd--;) if (GetMilleModule(imd)->GetVolumeID() == snVID) {snVID = -1; break;} | |
380 | // | |
381 | if (snVID==-1) continue; // found this sensor, do nothing | |
382 | // | |
383 | // otherwise, remove this sensor from the module list | |
384 | AliInfo(Form("Removing sensor %d from %s",snVID,parent->GetName())); | |
385 | parent->DelSensitiveVolume(isn--); | |
386 | } | |
387 | } | |
388 | // | |
389 | fGlobalDerivatives = new Double_t[fNGlobal]; | |
390 | memset(fGlobalDerivatives,0,fNGlobal*sizeof(Double_t)); | |
391 | // | |
392 | return 0; | |
393 | } | |
394 | ||
395 | ||
396 | void AliITSAlignMille2::SetRequiredPoint(Char_t* where, Int_t ndet, Int_t updw, Int_t nreqpts) | |
397 | { | |
398 | // set minimum number of points in specific detector or layer | |
399 | // where = LAYER or DETECTOR | |
400 | // ndet = detector number: 1-6 for LAYER and 1-3 for DETECTOR (SPD=1, SDD=2, SSD=3) | |
401 | // updw = 1 for Y>0, -1 for Y<0, 0 if not specified | |
402 | // nreqpts = minimum number of points of that type | |
403 | ndet--; | |
404 | if (strstr(where,"LAYER")) { | |
405 | if (ndet<0 || ndet>5) return; | |
406 | if (updw>0) fNReqLayUp[ndet]=nreqpts; | |
407 | else if (updw<0) fNReqLayDown[ndet]=nreqpts; | |
408 | else fNReqLay[ndet]=nreqpts; | |
409 | fRequirePoints=kTRUE; | |
410 | } | |
411 | else if (strstr(where,"DETECTOR")) { | |
412 | if (ndet<0 || ndet>2) return; | |
413 | if (updw>0) fNReqDetUp[ndet]=nreqpts; | |
414 | else if (updw<0) fNReqDetDown[ndet]=nreqpts; | |
415 | else fNReqDet[ndet]=nreqpts; | |
416 | fRequirePoints=kTRUE; | |
417 | } | |
418 | } | |
419 | ||
420 | ||
421 | Int_t AliITSAlignMille2::GetModuleIndex(const Char_t *symname) { | |
422 | /// index from symname | |
423 | if (!symname) return -1; | |
424 | for (Int_t i=0;i<=kMaxITSSensID; i++) { | |
425 | if (!strcmp(symname,AliITSgeomTGeo::GetSymName(i))) return i; | |
426 | } | |
427 | return -1; | |
428 | } | |
429 | ||
430 | Int_t AliITSAlignMille2::GetModuleIndex(UShort_t voluid) { | |
431 | /// index from volume ID | |
432 | AliGeomManager::ELayerID lay = AliGeomManager::VolUIDToLayer(voluid); | |
433 | if (lay<1|| lay>6) return -1; | |
434 | Int_t idx=Int_t(voluid)-2048*lay; | |
435 | if (idx>=AliGeomManager::LayerSize(lay)) return -1; | |
436 | for (Int_t ilay=1; ilay<lay; ilay++) | |
437 | idx += AliGeomManager::LayerSize(ilay); | |
438 | return idx; | |
439 | } | |
440 | ||
441 | UShort_t AliITSAlignMille2::GetModuleVolumeID(const Char_t *symname) { | |
442 | /// volume ID from symname | |
443 | /// works for sensitive volumes only | |
444 | if (!symname) return 0; | |
445 | ||
446 | for (UShort_t voluid=2000; voluid<13300; voluid++) { | |
447 | Int_t modId; | |
448 | AliGeomManager::ELayerID layerId = AliGeomManager::VolUIDToLayer(voluid,modId); | |
449 | if (layerId>0 && layerId<7 && modId>=0 && modId<AliGeomManager::LayerSize(layerId)) { | |
450 | if (!strcmp(symname,AliGeomManager::SymName(layerId,modId))) return voluid; | |
451 | } | |
452 | } | |
453 | ||
454 | return 0; | |
455 | } | |
456 | ||
457 | UShort_t AliITSAlignMille2::GetModuleVolumeID(Int_t index) { | |
458 | /// volume ID from index | |
459 | if (index<0) return 0; | |
460 | if (index<2198) | |
461 | return GetModuleVolumeID(AliITSgeomTGeo::GetSymName(index)); | |
462 | else { | |
463 | for (int i=0; i<fNSuperModules; i++) { | |
464 | if (GetSuperModule(i)->GetIndex()==index) return GetSuperModule(i)->GetVolumeID(); | |
465 | } | |
466 | } | |
467 | return 0; | |
468 | } | |
469 | ||
470 | void AliITSAlignMille2::InitGeometry() { | |
471 | /// initialize geometry | |
472 | AliGeomManager::LoadGeometry(fGeometryFileName.Data()); | |
473 | fGeoManager = AliGeomManager::GetGeometry(); | |
474 | if (!fGeoManager) { | |
475 | AliInfo("Couldn't initialize geometry"); | |
476 | return; | |
477 | } | |
478 | } | |
479 | ||
480 | void AliITSAlignMille2::Init(Int_t nGlobal, /* number of global paramers */ | |
481 | Int_t nLocal, /* number of local parameters */ | |
482 | Int_t nStdDev /* std dev cut */ ) | |
483 | { | |
484 | /// Initialization of AliMillepede. Fix parameters, define constraints ... | |
485 | fMillepede->InitMille(nGlobal,nLocal,nStdDev,fResCut,fResCutInitial); | |
486 | fIsMilleInit = kTRUE; | |
487 | ||
488 | /// Fix non free parameters | |
489 | for (Int_t i=0; i<fNModules; i++) { | |
490 | for (Int_t j=0; j<kNParCh; j++) { | |
491 | if (!GetMilleModule(i)->IsFreeDOF(j)) FixParameter(i*kNParCh+j,0.0); | |
492 | else { | |
493 | // pepopepo: da verificare il settaggio delle sigma, ma forse va bene... | |
494 | Double_t parsig=0; | |
495 | if (j<3) parsig = fParSigTranslations; // translations (0.0100 cm) | |
496 | else parsig = fParSigRotations; // rotations (1/10 deg) | |
497 | FixParameter(i*kNParCh+j,parsig); | |
498 | } | |
499 | } | |
500 | } | |
501 | // | |
502 | // Set iterations | |
503 | if (fStartFac>1) fMillepede->SetIterations(fStartFac); | |
504 | } | |
505 | ||
506 | ||
507 | void AliITSAlignMille2::AddConstraint(Double_t *par, Double_t value) { | |
508 | /// Constrain equation defined by par to value | |
509 | if (!fIsMilleInit) { | |
510 | AliInfo("Millepede has not been initialized!"); | |
511 | return; | |
512 | } | |
513 | fMillepede->SetGlobalConstraint(par, value); | |
514 | AliInfo("Adding constraint"); | |
515 | } | |
516 | ||
517 | void AliITSAlignMille2::InitGlobalParameters(Double_t *par) { | |
518 | /// Initialize global parameters with par array | |
519 | if (!fIsMilleInit) { | |
520 | AliInfo("Millepede has not been initialized!"); | |
521 | return; | |
522 | } | |
523 | fMillepede->SetGlobalParameters(par); | |
524 | AliInfo("Init Global Parameters"); | |
525 | } | |
526 | ||
527 | void AliITSAlignMille2::FixParameter(Int_t iPar, Double_t value) { | |
528 | /// Parameter iPar is encourage to vary in [-value;value]. | |
529 | /// If value == 0, parameter is fixed | |
530 | if (!fIsMilleInit) { | |
531 | AliInfo("Millepede has not been initialized!"); | |
532 | return; | |
533 | } | |
534 | fMillepede->SetParSigma(iPar, value); | |
535 | if (value==0) AliInfo(Form("Parameter %i Fixed", iPar)); | |
536 | } | |
537 | ||
538 | void AliITSAlignMille2::ResetLocalEquation() | |
539 | { | |
540 | /// Reset the derivative vectors | |
541 | for(int i=fNLocal;i--;) fLocalDerivatives[i] = 0.0; | |
542 | for(int i=fNGlobal;i--;) fGlobalDerivatives[i] = 0.0; | |
543 | } | |
544 | ||
545 | Int_t AliITSAlignMille2::ApplyToGeometry() | |
546 | { | |
547 | // apply starting realignment to ideal geometry | |
548 | // | |
549 | if (!fGeoManager) return -1; | |
550 | TFile *pref = new TFile(fPreAlignmentFileName.Data()); | |
551 | if (!pref->IsOpen()) return -2; | |
552 | fPrealignment = (TClonesArray*)pref->Get("ITSAlignObjs"); | |
553 | if (!fPrealignment) return -3; | |
554 | Int_t nprea = fPrealignment->GetEntriesFast(); | |
555 | AliInfo(Form("Array of input misalignments with %d entries",nprea)); | |
556 | // | |
557 | for (int ix=0; ix<nprea; ix++) { | |
558 | AliAlignObjParams *preo=(AliAlignObjParams*) fPrealignment->At(ix); | |
559 | Int_t index=AliITSAlignMille2Module::GetIndexFromVolumeID(preo->GetVolUID()); | |
560 | if (index>=0) { | |
561 | if (index>=fPreAlignQF.GetSize()) fPreAlignQF.Set(index+10); | |
562 | fPreAlignQF[index] = (int) preo->GetUniqueID()+1; | |
563 | } | |
564 | //TString nms = preo->GetSymName(); | |
565 | //if (!nms.Contains("Ladder")) continue; //RRR | |
566 | if (!preo->ApplyToGeometry()) return -4; | |
567 | } | |
568 | // | |
569 | pref->Close(); | |
570 | delete pref; | |
571 | // | |
572 | fUsePreAlignment = kTRUE; | |
573 | return 0; | |
574 | } | |
575 | ||
576 | Int_t AliITSAlignMille2::GetPreAlignmentQualityFactor(Int_t index) const | |
577 | { | |
578 | if (!fUsePreAlignment || index<0 || index>=fPreAlignQF.GetSize()) return -1; | |
579 | return fPreAlignQF[index]-1; | |
580 | } | |
581 | ||
582 | AliTrackPointArray *AliITSAlignMille2::PrepareTrack(const AliTrackPointArray *atp) { | |
583 | /// create a new AliTrackPointArray keeping only defined modules | |
584 | /// move points according to a given prealignment, if any | |
585 | /// sort alitrackpoints w.r.t. global Y direction, if selected | |
586 | ||
587 | AliTrackPointArray *atps=NULL; | |
588 | Int_t idx[20]; | |
589 | Int_t npts=atp->GetNPoints(); | |
590 | ||
591 | /// checks if AliTrackPoints belong to defined modules | |
592 | Int_t ngoodpts=0; | |
593 | Int_t intidx[20]; | |
594 | ||
595 | for (int j=0; j<npts; j++) { | |
596 | intidx[j] = IsContained(atp->GetVolumeID()[j]); | |
597 | if (intidx[j]>=0) ngoodpts++; | |
598 | } | |
599 | AliDebug(3,Form("Number of points in defined modules: %d",ngoodpts)); | |
600 | ||
601 | // reject track if not enough points are left | |
602 | if (ngoodpts<fMinNPtsPerTrack) { | |
603 | AliInfo("Track with not enough points!"); | |
604 | return NULL; | |
605 | } | |
606 | // >> RS | |
607 | AliTrackPoint p; | |
608 | // check points in specific places | |
609 | if (fRequirePoints) { | |
610 | Int_t nlayup[6],nlaydown[6],nlay[6]; | |
611 | Int_t ndetup[3],ndetdown[3],ndet[3]; | |
612 | for (Int_t j=0; j<6; j++) {nlayup[j]=0; nlaydown[j]=0; nlay[j]=0;} | |
613 | for (Int_t j=0; j<3; j++) {ndetup[j]=0; ndetdown[j]=0; ndet[j]=0;} | |
614 | ||
615 | for (int i=0; i<npts; i++) { | |
616 | // skip not defined points | |
617 | if (intidx[i]<0) continue; | |
618 | Float_t xx=atp->GetX()[i]; | |
619 | Float_t yy=atp->GetY()[i]; | |
620 | Float_t r=TMath::Sqrt(xx*xx + yy*yy); | |
621 | int lay=-1; | |
622 | if (r<5) lay=0; | |
623 | else if (r>5 && r<10) lay=1; | |
624 | else if (r>10 && r<18) lay=2; | |
625 | else if (r>18 && r<30) lay=3; | |
626 | else if (r>30 && r<40) lay=4; | |
627 | else if (r>40) lay=5; | |
628 | if (lay<0) continue; | |
629 | int det=lay/2; | |
630 | //printf("Point %d - x=%f y=%f R=%f lay=%d det=%d\n",i,xx,yy,r,lay,det); | |
631 | ||
632 | if (yy>=0.0) { // UP point | |
633 | nlayup[lay]++; | |
634 | nlay[lay]++; | |
635 | ndetup[det]++; | |
636 | ndet[det]++; | |
637 | } | |
638 | else { | |
639 | nlaydown[lay]++; | |
640 | nlay[lay]++; | |
641 | ndetdown[det]++; | |
642 | ndet[det]++; | |
643 | } | |
644 | } | |
645 | ||
646 | // checks minimum values | |
647 | Bool_t isok=kTRUE; | |
648 | for (Int_t j=0; j<6; j++) { | |
649 | if (nlayup[j]<fNReqLayUp[j]) isok=kFALSE; | |
650 | if (nlaydown[j]<fNReqLayDown[j]) isok=kFALSE; | |
651 | if (nlay[j]<fNReqLay[j]) isok=kFALSE; | |
652 | } | |
653 | for (Int_t j=0; j<3; j++) { | |
654 | if (ndetup[j]<fNReqDetUp[j]) isok=kFALSE; | |
655 | if (ndetdown[j]<fNReqDetDown[j]) isok=kFALSE; | |
656 | if (ndet[j]<fNReqDet[j]) isok=kFALSE; | |
657 | } | |
658 | if (!isok) { | |
659 | AliDebug(2,Form("Track does not meet all location point requirements!")); | |
660 | return NULL; | |
661 | } | |
662 | } | |
663 | ||
664 | // << RS | |
665 | // build a new track with (sorted) (prealigned) good points | |
666 | atps=new AliTrackPointArray(ngoodpts); | |
667 | ||
668 | for (int i=0; i<npts; i++) idx[i]=i; | |
669 | // sort track if required | |
670 | if (fUseSortedTracks) TMath::Sort(npts,atp->GetY(),idx); // sort descending... | |
671 | ||
672 | Int_t npto=0; | |
673 | for (int i=0; i<npts; i++) { | |
674 | // skip not defined points | |
675 | if (intidx[idx[i]]<0) continue; | |
676 | atp->GetPoint(p,idx[i]); | |
677 | ||
678 | // prealign point if required | |
679 | // get IDEAL matrix | |
680 | TGeoHMatrix *svOrigMatrix = GetMilleModule(intidx[idx[i]])->GetSensitiveVolumeOrigGlobalMatrix(p.GetVolumeID()); | |
681 | // get back real local coordinates: use OriginalGlobalMatrix because AliTrackPoints were written | |
682 | // with idel geometry | |
683 | Double_t pg[3],pl[3]; | |
684 | pg[0]=p.GetX(); | |
685 | pg[1]=p.GetY(); | |
686 | pg[2]=p.GetZ(); | |
687 | // printf("Global coordinates of measured point : X=%f Y=%f Z=%f \n",pg[0],pg[1],pg[2]); | |
688 | AliDebug(3,Form("Global coordinates of measured point : X=%f Y=%f Z=%f \n",pg[0],pg[1],pg[2])); | |
689 | svOrigMatrix->MasterToLocal(pg,pl); | |
690 | ||
691 | AliDebug(3,Form("Local coordinates of measured point : X=%f Y=%f Z=%f \n",pl[0],pl[1],pl[2])); | |
692 | ||
693 | // update covariance matrix | |
694 | TGeoHMatrix hcov; | |
695 | Double_t hcovel[9]; | |
696 | hcovel[0]=double(p.GetCov()[0]); | |
697 | hcovel[1]=double(p.GetCov()[1]); | |
698 | hcovel[2]=double(p.GetCov()[2]); | |
699 | hcovel[3]=double(p.GetCov()[1]); | |
700 | hcovel[4]=double(p.GetCov()[3]); | |
701 | hcovel[5]=double(p.GetCov()[4]); | |
702 | hcovel[6]=double(p.GetCov()[2]); | |
703 | hcovel[7]=double(p.GetCov()[4]); | |
704 | hcovel[8]=double(p.GetCov()[5]); | |
705 | hcov.SetRotation(hcovel); | |
706 | // now rotate in local system | |
707 | hcov.Multiply(svOrigMatrix); | |
708 | hcov.MultiplyLeft(&svOrigMatrix->Inverse()); | |
709 | // now hcov is LOCAL COVARIANCE MATRIX | |
710 | ||
711 | // >> RS | |
712 | if (fBug==1) { | |
713 | // correzione bug LAYER 5 SSD temporanea.. | |
714 | int ssdidx=AliITSAlignMille2Module::GetIndexFromVolumeID(p.GetVolumeID()); | |
715 | if (ssdidx>=500 && ssdidx<1248) { | |
716 | int ladder=(ssdidx-500)%22; | |
717 | if (ladder==18) p.SetVolumeID(AliITSAlignMille2Module::GetVolumeIDFromIndex(ssdidx+1)); | |
718 | if (ladder==19) p.SetVolumeID(AliITSAlignMille2Module::GetVolumeIDFromIndex(ssdidx-1)); | |
719 | } | |
720 | } | |
721 | ||
722 | // << RS | |
723 | ||
724 | /// get (evenctually prealigned) matrix of sens. vol. | |
725 | TGeoHMatrix *svMatrix = GetMilleModule(intidx[idx[i]])->GetSensitiveVolumeMatrix(p.GetVolumeID()); | |
726 | // modify global coordinates according with pre-aligment | |
727 | svMatrix->LocalToMaster(pl,pg); | |
728 | // now rotate in local system | |
729 | hcov.Multiply(&svMatrix->Inverse()); | |
730 | hcov.MultiplyLeft(svMatrix); | |
731 | // hcov is back in GLOBAL RF | |
732 | Float_t pcov[6]; | |
733 | pcov[0]=hcov.GetRotationMatrix()[0]; | |
734 | pcov[1]=hcov.GetRotationMatrix()[1]; | |
735 | pcov[2]=hcov.GetRotationMatrix()[2]; | |
736 | pcov[3]=hcov.GetRotationMatrix()[4]; | |
737 | pcov[4]=hcov.GetRotationMatrix()[5]; | |
738 | pcov[5]=hcov.GetRotationMatrix()[8]; | |
739 | ||
740 | p.SetXYZ(pg[0],pg[1],pg[2],pcov); | |
741 | // printf("New Gl coordinates of measured point : X=%f Y=%f Z=%f \n",pg[0],pg[1],pg[2]); | |
742 | AliDebug(3,Form("New global coordinates of measured point : X=%f Y=%f Z=%f \n",pg[0],pg[1],pg[2])); | |
743 | atps->AddPoint(npto,&p); | |
744 | AliDebug(2,Form("Adding point[%d] = ( %f , %f , %f ) volid = %d",npto,atps->GetX()[npto],atps->GetY()[npto],atps->GetZ()[npto],atps->GetVolumeID()[npto] )); | |
745 | ||
746 | npto++; | |
747 | } | |
748 | ||
749 | return atps; | |
750 | } | |
751 | ||
752 | ||
753 | ||
754 | AliTrackPointArray *AliITSAlignMille2::SortTrack(const AliTrackPointArray *atp) { | |
755 | /// sort alitrackpoints w.r.t. global Y direction | |
756 | AliTrackPointArray *atps=NULL; | |
757 | Int_t idx[20]; | |
758 | Int_t npts=atp->GetNPoints(); | |
759 | AliTrackPoint p; | |
760 | atps=new AliTrackPointArray(npts); | |
761 | ||
762 | TMath::Sort(npts,atp->GetY(),idx); | |
763 | ||
764 | for (int i=0; i<npts; i++) { | |
765 | atp->GetPoint(p,idx[i]); | |
766 | atps->AddPoint(i,&p); | |
767 | AliDebug(2,Form("Point[%d] = ( %f , %f , %f ) volid = %d",i,atps->GetX()[i],atps->GetY()[i],atps->GetZ()[i],atps->GetVolumeID()[i] )); | |
768 | } | |
769 | return atps; | |
770 | } | |
771 | ||
772 | ||
773 | Int_t AliITSAlignMille2::InitModuleParams() { | |
774 | /// initialize geometry parameters for a given detector | |
775 | /// for current cluster (fCluster) | |
776 | /// fGlobalInitParam[] is set as: | |
777 | /// [tx,ty,tz,psi,theta,phi] | |
778 | /// (old was [tx,ty,tz,theta,psi,phi] ROOT's angles...) | |
779 | /// *** At the moment: using Raffalele's angles definition *** | |
780 | /// | |
781 | /// return 0 if success | |
782 | ||
783 | if (!fGeoManager) { | |
784 | AliInfo("ITS geometry not initialized!"); | |
785 | return -1; | |
786 | } | |
787 | ||
788 | // now 'voluid' is the volumeID of a SENSITIVE VOLUME (coming from a cluster) | |
789 | ||
790 | // set the internal index (index in module list) | |
791 | UShort_t voluid=fCluster.GetVolumeID(); | |
792 | // | |
793 | // IT IS VERY IMPORTANT: start from the end of the list, where the childs are located !!! | |
794 | Int_t k=fNModules-1; | |
795 | fCurrentModule = 0; | |
796 | while (k>=0 && ! (fCurrentModule=GetMilleModule(k))->IsIn(voluid)) { | |
797 | // VERY IMPORTANT: if the sensors were explicitly provided, don't look in the supermodules | |
798 | if (fSensorsIn && fCurrentModule->GetVolumeID() > kMaxITSSensVID) {k=-1; break;} | |
799 | k--; | |
800 | } | |
801 | if (k<0) return -3; | |
802 | // fCurrentModule = GetMilleModule(k); | |
803 | // | |
804 | fModuleInitParam[0] = 0.0; | |
805 | fModuleInitParam[1] = 0.0; | |
806 | fModuleInitParam[2] = 0.0; | |
807 | fModuleInitParam[3] = 0.0; // psi (X) | |
808 | fModuleInitParam[4] = 0.0; // theta (Y) | |
809 | fModuleInitParam[5] = 0.0; // phi (Z) | |
810 | ||
811 | /// get (evenctually prealigned) matrix of sens. vol. | |
812 | TGeoHMatrix *svMatrix = fCurrentModule->GetSensitiveVolumeMatrix(voluid); | |
813 | ||
814 | fMeasGlo[0] = fCluster.GetX(); | |
815 | fMeasGlo[1] = fCluster.GetY(); | |
816 | fMeasGlo[2] = fCluster.GetZ(); | |
817 | svMatrix->MasterToLocal(fMeasGlo,fMeasLoc); | |
818 | AliDebug(2,Form("Local coordinates of measured point : X=%f Y=%f Z=%f \n",fMeasLoc[0] ,fMeasLoc[1] ,fMeasLoc[2] )); | |
819 | ||
820 | // set stdev from cluster | |
821 | TGeoHMatrix hcov; | |
822 | Double_t hcovel[9]; | |
823 | hcovel[0]=double(fCluster.GetCov()[0]); | |
824 | hcovel[1]=double(fCluster.GetCov()[1]); | |
825 | hcovel[2]=double(fCluster.GetCov()[2]); | |
826 | hcovel[3]=double(fCluster.GetCov()[1]); | |
827 | hcovel[4]=double(fCluster.GetCov()[3]); | |
828 | hcovel[5]=double(fCluster.GetCov()[4]); | |
829 | hcovel[6]=double(fCluster.GetCov()[2]); | |
830 | hcovel[7]=double(fCluster.GetCov()[4]); | |
831 | hcovel[8]=double(fCluster.GetCov()[5]); | |
832 | hcov.SetRotation(hcovel); | |
833 | // now rotate in local system | |
834 | hcov.Multiply(svMatrix); | |
835 | hcov.MultiplyLeft(&svMatrix->Inverse()); | |
836 | ||
837 | // set local sigmas | |
838 | fSigmaLoc[0] = TMath::Sqrt(TMath::Abs(hcov.GetRotationMatrix()[0])); | |
839 | fSigmaLoc[1] = TMath::Sqrt(TMath::Abs(hcov.GetRotationMatrix()[4])); // RS | |
840 | fSigmaLoc[2] = TMath::Sqrt(TMath::Abs(hcov.GetRotationMatrix()[8])); | |
841 | ||
842 | // set minimum value for SigmaLoc to 10 micron | |
843 | if (fSigmaLoc[0]<0.0010) fSigmaLoc[0]=0.0010; | |
844 | if (fSigmaLoc[2]<0.0010) fSigmaLoc[2]=0.0010; | |
845 | ||
846 | // multiply local sigmas by global and module specific factor | |
847 | for (int i=3;i--;) fSigmaLoc[i] *= fSigmaFactor[i]*fCurrentModule->GetSigmaFactor(i); | |
848 | // | |
849 | AliDebug(2,Form("Setting StDev from CovMat : fSigmaLocX=%g fSigmaLocY=%g fSigmaLocZ=%g \n",fSigmaLoc[0] ,fSigmaLoc[1] ,fSigmaLoc[2] )); | |
850 | ||
851 | return 0; | |
852 | } | |
853 | ||
854 | void AliITSAlignMille2::Print(Option_t*) const | |
855 | { | |
856 | /// | |
857 | printf("*** AliMillepede for ITS ***\n"); | |
858 | printf(" Number of defined super modules: %d\n",fNModules); | |
859 | printf(" Obtained parameters refer to %s Deltas\n",fUseGlobalDelta ? "GLOBAL":"LOCAL"); | |
860 | // | |
861 | if (fGeoManager) | |
862 | printf(" geometry loaded from %s\n",fGeometryFileName.Data()); | |
863 | else | |
864 | printf(" geometry not loaded\n"); | |
865 | // | |
866 | if (fUsePreAlignment) | |
867 | printf(" using prealignment from %s \n",fPreAlignmentFileName.Data()); | |
868 | else | |
869 | printf(" prealignment not used\n"); | |
870 | // | |
871 | // | |
872 | if (fBOn) | |
873 | printf(" B Field set to %f T - using Riemann's helices\n",fBField); | |
874 | else | |
875 | printf(" B Field OFF - using straight lines \n"); | |
876 | // | |
877 | if (fRequirePoints) printf(" Required points in tracks:\n"); | |
878 | for (Int_t i=0; i<6; i++) { | |
879 | if (fNReqLayUp[i]>0) printf(" Layer %d : %d points with Y>0\n",i+1,fNReqLayUp[i]); | |
880 | if (fNReqLayDown[i]>0) printf(" Layer %d : %d points with Y<0\n",i+1,fNReqLayDown[i]); | |
881 | if (fNReqLay[i]>0) printf(" Layer %d : %d points \n",i+1,fNReqLay[i]); | |
882 | } | |
883 | for (Int_t i=0; i<3; i++) { | |
884 | if (fNReqDetUp[i]>0) printf(" Detector %d : %d points with Y>0\n",i+1,fNReqDetUp[i]); | |
885 | if (fNReqDetDown[i]>0) printf(" Detector %d : %d points with Y<0\n",i+1,fNReqDetDown[i]); | |
886 | if (fNReqDet[i]>0) printf(" Detector %d : %d points \n",i+1,fNReqDet[i]); | |
887 | } | |
888 | // | |
889 | printf("\n Millepede configuration parameters:\n"); | |
890 | printf(" init parsig for translations : %.4f\n",fParSigTranslations); | |
891 | printf(" init parsig for rotations : %.4f\n",fParSigRotations); | |
892 | printf(" init value for chi2 cut : %.4f\n",fStartFac); | |
893 | printf(" first iteration cut value : %.4f\n",fResCutInitial); | |
894 | printf(" other iterations cut value : %.4f\n",fResCut); | |
895 | printf(" number of stddev for chi2 cut : %d\n",fNStdDev); | |
896 | printf(" mult. fact. for local sigmas : %.4f %.4f %.4f\n",fSigmaFactor[0],fSigmaFactor[1],fSigmaFactor[2]); | |
897 | ||
898 | printf("List of defined modules:\n"); | |
899 | printf(" intidx\tindex\tvoluid\tname\n"); | |
900 | for (int i=0; i<fNModules; i++) { | |
901 | AliITSAlignMille2Module* md = GetMilleModule(i); | |
902 | printf(" %d\t%d\t%d\t%s\n",i,md->GetIndex(),md->GetVolumeID(),md->GetName()); | |
903 | } | |
904 | } | |
905 | ||
906 | AliITSAlignMille2Module *AliITSAlignMille2::GetMilleModuleByVID(UShort_t voluid) const | |
907 | { | |
908 | // return pointer to a define supermodule | |
909 | // return NULL if error | |
910 | Int_t i=IsDefined(voluid); | |
911 | if (i<0) return NULL; | |
912 | return GetMilleModule(i); | |
913 | } | |
914 | ||
915 | Bool_t AliITSAlignMille2::InitRiemanFit() | |
916 | { | |
917 | // Initialize Riemann Fitter for current track | |
918 | // return kFALSE if error | |
919 | ||
920 | if (!fBOn) return kFALSE; | |
921 | ||
922 | Int_t npts=0; | |
923 | AliTrackPoint ap; | |
924 | npts = fTrack->GetNPoints(); | |
925 | AliDebug(3,Form("Fitting track with %d points",npts)); | |
926 | ||
927 | fRieman->Reset(); | |
928 | fRieman->SetTrackPointArray(fTrack); | |
929 | ||
930 | TArrayI ai(npts); | |
931 | for (Int_t ipt=0; ipt<npts; ipt++) ai[ipt]=fTrack->GetVolumeID()[ipt]; | |
932 | ||
933 | // fit track with 5 params in his own tracking-rotated reference system | |
934 | // xc = -p[1]/p[0]; | |
935 | // yc = 1/p[0]; | |
936 | // R = sqrt( x0*x0 + y0*y0 - y0*p[2]); | |
937 | if (!fRieman->Fit(&ai,NULL,(AliGeomManager::ELayerID)1,(AliGeomManager::ELayerID)6)) { | |
938 | return kFALSE; | |
939 | } | |
940 | ||
941 | for (int i=0; i<5; i++) | |
942 | fLocalInitParam[i] = fRieman->GetParam()[i]; | |
943 | ||
944 | return kTRUE; | |
945 | } | |
946 | ||
947 | Bool_t fullErr2D = kTRUE; | |
948 | ||
949 | void trackFit2D(Int_t &, Double_t *, double &chi2, double *par, int) | |
950 | { | |
951 | const double kTiny = 1.e-14; | |
952 | chi2 = 0; | |
953 | static AliTrackPoint pnt; | |
954 | // | |
955 | enum {kAX,kAZ,kBX,kBZ}; | |
956 | enum {kXX=0,kXY=1,kXZ=2,kYX=kXY,kYY=3,kYZ=4,kZX=kXZ,kZY=kYZ,kZZ=5}; | |
957 | // | |
958 | AliITSAlignMille2* alig = AliITSAlignMille2::GetInstance(); | |
959 | AliTrackPointArray* track = alig->GetCurrentTrack(); | |
960 | // | |
961 | int npts = track->GetNPoints(); | |
962 | for (int ip=0;ip<npts;ip++) { | |
963 | track->GetPoint(pnt,ip); | |
964 | const float *cov = pnt.GetCov(); | |
965 | double y = pnt.GetY(); | |
966 | double dx = pnt.GetX() - (par[kAX]+y*par[kBX]); | |
967 | double dz = pnt.GetZ() - (par[kAZ]+y*par[kBZ]); | |
968 | double xxe = cov[kXX]; | |
969 | double zze = cov[kZZ]; | |
970 | double xze = cov[kXZ]; | |
971 | // | |
972 | if (fullErr2D) { | |
973 | xxe += par[kBX]*par[kBX]*cov[kYY]-2.*par[kBX]*cov[kXY]; | |
974 | zze += par[kBZ]*par[kBZ]*cov[kYY]-2.*par[kBZ]*cov[kZY]; | |
975 | xze += par[kBX]*par[kBZ]*cov[kYY]-cov[kYZ]*par[kBZ]-cov[kXY]*par[kBX]; | |
976 | } | |
977 | // | |
978 | double det = xxe*zze - xze*xze; | |
979 | if (det<kTiny) { | |
980 | printf("Negative diag. error (det=%+e) |sxx:%+e szz:%+e sxz:%+e| bx:%+e bz:%+e|\n" | |
981 | "Discarding correlation term\n",det,xxe,zze,xze,par[kBX],par[kBZ]); | |
982 | xxe = cov[kXX]; | |
983 | zze = cov[kZZ]; | |
984 | xze = cov[kXZ]; | |
985 | fullErr2D = kFALSE; | |
986 | } | |
987 | double xxeI = zze/det; | |
988 | double zzeI = xxe/det; | |
989 | double xzeI =-xze/det; | |
990 | // | |
991 | chi2 += dx*dx*xxeI + dz*dz*zzeI + 2.*dx*dz*xzeI; | |
992 | // | |
993 | // printf("%d | %+e %+e %+e %+e %+e -> %+e\n",ip,dx,dz,xxeI,zzeI,xzeI, chi2); | |
994 | } | |
995 | // | |
996 | } | |
997 | ||
998 | void AliITSAlignMille2::InitTrackParams(int meth) | |
999 | { | |
1000 | /// initialize local parameters with different methods | |
1001 | /// for current track (fTrack) | |
1002 | Int_t npts=0; | |
1003 | AliTrackPoint ap; | |
1004 | double sX=0,sXY=0,sZ=0,sZY=0,sY=0,sYY=0,det=0; | |
1005 | // simple linear interpolation | |
1006 | // get local starting parameters (to be substituted by ESD track parms) | |
1007 | // local parms (fLocalInitParam[]) are: | |
1008 | // [0] = global x coord. of straight line intersection at y=0 plane | |
1009 | // [1] = global z coord. of straight line intersection at y=0 plane | |
1010 | // [2] = px/py | |
1011 | // [3] = pz/py | |
1012 | // test #1: linear fit in x(y) and z(y) | |
1013 | npts = fTrack->GetNPoints(); | |
1014 | AliDebug(3,Form("*** initializing track with %d points ***",npts)); | |
1015 | for (int i=npts;i--;) { | |
1016 | sY += fTrack->GetY()[i]; | |
1017 | sYY += fTrack->GetY()[i]*fTrack->GetY()[i]; | |
1018 | sX += fTrack->GetX()[i]; | |
1019 | sXY += fTrack->GetX()[i]*fTrack->GetY()[i]; | |
1020 | sZ += fTrack->GetZ()[i]; | |
1021 | sZY += fTrack->GetZ()[i]*fTrack->GetY()[i]; | |
1022 | } | |
1023 | det = sYY*npts-sY*sY; | |
1024 | if (det==0) det = 1E-20; | |
1025 | fLocalInitParam[0] = (sX*sYY-sY*sXY)/det; | |
1026 | fLocalInitParam[2] = (sXY*npts-sY*sX)/det; | |
1027 | // | |
1028 | fLocalInitParam[1] = (sZ*sYY-sY*sZY)/det; | |
1029 | fLocalInitParam[3] = (sZY*npts-sY*sZ)/det; | |
1030 | //AliDebug(2,Form("X = p0gx + ugx*Y : p0gx = %f +- %f ugx = %f +- %f\n",fLocalInitParam[0],f1->GetParError(0),fLocalInitParam[2],f1->GetParError(1))); | |
1031 | AliDebug(2,Form("X = p0gx + ugx*Y : p0gx = %f ugx = %f\n",fLocalInitParam[0],fLocalInitParam[2])); | |
1032 | // | |
1033 | if (meth==1) return; | |
1034 | // | |
1035 | // perform full fit accounting for cov.matrix | |
1036 | static TVirtualFitter *minuit = 0; | |
1037 | static Double_t step[5] = {1E-3,1E-3,1E-4,1E-4,1E-5}; | |
1038 | static Double_t arglist[10]; | |
1039 | // | |
1040 | if (!minuit) { | |
1041 | minuit = TVirtualFitter::Fitter(0,4); | |
1042 | minuit->SetFCN(trackFit2D); | |
1043 | arglist[0] = 1; | |
1044 | minuit->ExecuteCommand("SET ERR",arglist, 1); | |
1045 | // | |
1046 | arglist[0] = -1; | |
1047 | minuit->ExecuteCommand("SET PRINT",arglist,1); | |
1048 | // | |
1049 | } | |
1050 | // | |
1051 | minuit->SetParameter(0, "ax", fLocalInitParam[0], step[0], 0,0); | |
1052 | minuit->SetParameter(1, "az", fLocalInitParam[1], step[1], 0,0); | |
1053 | minuit->SetParameter(2, "bx", fLocalInitParam[2], step[2], 0,0); | |
1054 | minuit->SetParameter(3, "bz", fLocalInitParam[3], step[3], 0,0); | |
1055 | // | |
1056 | arglist[0] = 1000; // number of function calls | |
1057 | arglist[1] = 0.001; // tolerance | |
1058 | fullErr2D = kTRUE; | |
1059 | minuit->ExecuteCommand("MIGRAD",arglist,2); | |
1060 | // | |
1061 | for (int i=0;i<4;i++) fLocalInitParam[i] = minuit->GetParameter(i); | |
1062 | for (int i=0;i<4;i++) for (int j=0;j<4;j++) fLocalInitParEr[i][j] = minuit->GetCovarianceMatrixElement(i,j); | |
1063 | // | |
1064 | } | |
1065 | ||
1066 | ||
1067 | Int_t AliITSAlignMille2::IsDefined(UShort_t voluid) const | |
1068 | { | |
1069 | // checks if supermodule 'voluid' is defined and return the internal index | |
1070 | // return -1 if error | |
1071 | Int_t k = fNModules-1; | |
1072 | while ( k>=0 && !(voluid==GetMilleModule(k)->GetVolumeID()) ) k--; | |
1073 | if (k<0) return -1; | |
1074 | return k; | |
1075 | } | |
1076 | ||
1077 | Int_t AliITSAlignMille2::IsContained(UShort_t voluid) const | |
1078 | { | |
1079 | // checks if the sensitive module 'voluid' is contained inside a supermodule and return the internal index of the last identified supermodule | |
1080 | // return -1 if error | |
1081 | if (AliITSAlignMille2Module::GetIndexFromVolumeID(voluid)<0) return -1; | |
1082 | Int_t k=fNModules-1; | |
1083 | while (k>=0 && !(GetMilleModule(k)->IsIn(voluid)) ) k--; | |
1084 | if (k<0) return -1; | |
1085 | return k; | |
1086 | } | |
1087 | ||
1088 | Bool_t AliITSAlignMille2::CheckVolumeID(UShort_t voluid) const | |
1089 | { | |
1090 | /// check if a sensitive volume is contained inside one of the defined supermodules | |
1091 | Int_t k=fNModules-1; | |
1092 | while (k>=0 && !(GetMilleModule(k)->IsIn(voluid)) ) k--; | |
1093 | if (k>=0) return kTRUE; | |
1094 | return kFALSE; | |
1095 | } | |
1096 | ||
1097 | Int_t AliITSAlignMille2::CheckCurrentTrack() { | |
1098 | /// checks if AliTrackPoints belongs to defined modules | |
1099 | /// return number of good poins | |
1100 | /// return 0 if not enough points | |
1101 | ||
1102 | Int_t npts = fTrack->GetNPoints(); | |
1103 | Int_t ngoodpts=0; | |
1104 | // debug points | |
1105 | for (int j=0; j<npts; j++) { | |
1106 | UShort_t voluid = fTrack->GetVolumeID()[j]; | |
1107 | if (CheckVolumeID(voluid)) { | |
1108 | ngoodpts++; | |
1109 | } | |
1110 | } | |
1111 | ||
1112 | if (ngoodpts<fMinNPtsPerTrack) return 0; | |
1113 | ||
1114 | return ngoodpts; | |
1115 | } | |
1116 | ||
1117 | Int_t AliITSAlignMille2::ProcessTrack(const AliTrackPointArray *track) { | |
1118 | /// Process track; Loop over hits and set local equations | |
1119 | /// here 'track' is a AliTrackPointArray | |
1120 | /// return 0 if success; | |
1121 | ||
1122 | if (!fIsMilleInit) { | |
1123 | AliInfo("Millepede has not been initialized!"); | |
1124 | return -1; | |
1125 | } | |
1126 | ||
1127 | Int_t npts = track->GetNPoints(); | |
1128 | AliDebug(2,Form("*** Input track with %d points ***",npts)); | |
1129 | ||
1130 | // preprocessing of the input track: keep only points in defined volumes, | |
1131 | // move points if prealignment is set, sort by Yglo if required | |
1132 | ||
1133 | fTrack=PrepareTrack(track); | |
1134 | if (!fTrack) return -1; | |
1135 | ||
1136 | npts = fTrack->GetNPoints(); | |
1137 | AliDebug(2,Form("*** Processing prepared track with %d points ***",npts)); | |
1138 | ||
1139 | if (!fBOn) { // straight lines | |
1140 | // set local starting parameters (to be substituted by ESD track parms) | |
1141 | // local parms (fLocalInitParam[]) are: | |
1142 | // [0] = global x coord. of straight line intersection at y=0 plane | |
1143 | // [1] = global z coord. of straight line intersection at y=0 plane | |
1144 | // [2] = px/py | |
1145 | // [3] = pz/py | |
1146 | InitTrackParams(fInitTrackParamsMeth); | |
1147 | } | |
1148 | else { | |
1149 | // local parms (fLocalInitParam[]) are the Riemann Fitter params | |
1150 | if (!InitRiemanFit()) { | |
1151 | AliInfo("Riemann fit failed! skipping this track..."); | |
1152 | delete fTrack; | |
1153 | fTrack=NULL; | |
1154 | return -5; | |
1155 | } | |
1156 | } | |
1157 | ||
1158 | Int_t nloceq=0; | |
1159 | static Mille2Data md[100]; | |
1160 | ||
1161 | for (Int_t ipt=0; ipt<npts; ipt++) { | |
1162 | fTrack->GetPoint(fCluster,ipt); | |
1163 | AliDebug(2,Form("\n--- processing point %d --- \n",ipt)); | |
1164 | ||
1165 | // set geometry parameters for the the current module | |
1166 | if (InitModuleParams()) continue; | |
1167 | AliDebug(2,Form(" VolID=%d Index=%d InternalIdx=%d symname=%s\n", | |
1168 | track->GetVolumeID()[ipt], fCurrentModule->GetIndex(), | |
1169 | fCurrentModule->GetUniqueID(), AliGeomManager::SymName(track->GetVolumeID()[ipt]) )); | |
1170 | AliDebug(2,Form(" Preprocessed Point = ( %f , %f , %f ) \n",fCluster.GetX(),fCluster.GetY(),fCluster.GetZ())); | |
1171 | ||
1172 | if (!AddLocalEquation(md[nloceq])) nloceq++; | |
1173 | else fTotBadLocEqPoints++; | |
1174 | } // end loop over points | |
1175 | // | |
1176 | delete fTrack; | |
1177 | fTrack=NULL; | |
1178 | // not enough good points! | |
1179 | if (nloceq<fMinNPtsPerTrack) return -1; | |
1180 | // | |
1181 | // finally send local equations to millepede | |
1182 | SetLocalEquations(md,nloceq); | |
1183 | fMillepede->SaveRecordData(); // RRR | |
1184 | // | |
1185 | return 0; | |
1186 | } | |
1187 | ||
1188 | Int_t AliITSAlignMille2::CalcIntersectionPoint(Double_t *lpar, Double_t *gpar) { | |
1189 | /// calculate track intersection point in local coordinates | |
1190 | /// according with a given set of parameters (local(4) and global(6)) | |
1191 | /// and fill fPintLoc/Glo | |
1192 | /// local are: pgx0, pgz0, ugx, ugz OR riemann fitters pars | |
1193 | /// global are: tx,ty,tz,psi,theta,phi (Raff's delta angles in deg.) | |
1194 | /// return 0 if success | |
1195 | ||
1196 | AliDebug(3,Form("lpar = %g %g %g %g %g\ngpar= %g %g %g %g %g %g\n",lpar[0],lpar[1],lpar[2],lpar[3],lpar[4],gpar[0],gpar[1],gpar[2],gpar[3],gpar[4],gpar[5])); | |
1197 | AliDebug(3,Form("deltalpar = %g %g %g %g %g\n",lpar[0]-fLocalInitParam[0],lpar[1]-fLocalInitParam[1],lpar[2]-fLocalInitParam[2],lpar[3]-fLocalInitParam[3],lpar[4]-fLocalInitParam[4])); | |
1198 | ||
1199 | ||
1200 | // prepare the TGeoHMatrix | |
1201 | TGeoHMatrix *fTempHMat = fCurrentModule->GetSensitiveVolumeModifiedMatrix(fCluster.GetVolumeID(),gpar, | |
1202 | !fUseGlobalDelta); | |
1203 | if (!fTempHMat) return -1; | |
1204 | ||
1205 | Double_t v0g[3]; // vector with straight line direction in global coord. | |
1206 | Double_t p0g[3]; // point of the straight line (glo) | |
1207 | ||
1208 | if (fBOn) { // B FIELD! | |
1209 | AliTrackPoint prf; | |
1210 | for (int ip=0; ip<5; ip++) | |
1211 | fRieman->SetParam(ip,lpar[ip]); | |
1212 | ||
1213 | if (!fRieman->GetPCA(fCluster,prf)) { | |
1214 | AliInfo(Form("error in GetPCA for point %d",fCluster.GetVolumeID())); | |
1215 | return -3; | |
1216 | } | |
1217 | // now determine straight line passing tangent to fit curve at prf | |
1218 | // ugx = dX/dY_glo = DeltaX/DeltaY_glo | |
1219 | // mo' P1=(X,Y,Z)_glo_prf | |
1220 | // => (x,y,Z)_trk_prf ruotando di alpha... | |
1221 | Double_t alpha=fRieman->GetAlpha(); | |
1222 | Double_t x1g = prf.GetX(); | |
1223 | Double_t y1g = prf.GetY(); | |
1224 | Double_t z1g = prf.GetZ(); | |
1225 | Double_t x1t = x1g*TMath::Cos(alpha) + y1g*TMath::Sin(alpha); | |
1226 | Double_t y1t = -x1g*TMath::Sin(alpha) + y1g*TMath::Cos(alpha); | |
1227 | Double_t z1t = z1g; | |
1228 | ||
1229 | Double_t x2t = x1t+1.0; | |
1230 | Double_t y2t = y1t+fRieman->GetDYat(x1t); | |
1231 | Double_t z2t = z1t+fRieman->GetDZat(x1t); | |
1232 | Double_t x2g = x2t*TMath::Cos(alpha) - y2t*TMath::Sin(alpha); | |
1233 | Double_t y2g = x2t*TMath::Sin(alpha) + y2t*TMath::Cos(alpha); | |
1234 | Double_t z2g = z2t; | |
1235 | ||
1236 | AliDebug(3,Form("Riemann frame: fAlpha = %f = %f ",alpha,alpha*180./TMath::Pi())); | |
1237 | AliDebug(3,Form(" prf_glo=( %f , %f , %f ) prf_rf=( %f , %f , %f )\n", x1g,y1g,z1g, x1t,y1t,z1t)); | |
1238 | AliDebug(3,Form(" mov_glo=( %f , %f , %f ) rf=( %f , %f , %f )\n",x2g,y2g,z2g, x2t,y2t,z2t)); | |
1239 | ||
1240 | if (TMath::Abs(y2g-y1g)<1e-15) { | |
1241 | AliInfo("DeltaY=0! Cannot proceed..."); | |
1242 | return -1; | |
1243 | } | |
1244 | // ugx,1,ugz | |
1245 | v0g[0] = (x2g-x1g)/(y2g-y1g); | |
1246 | v0g[1]=1.0; | |
1247 | v0g[2] = (z2g-z1g)/(y2g-y1g); | |
1248 | ||
1249 | // point: just keep prf | |
1250 | p0g[0]=x1g; | |
1251 | p0g[1]=y1g; | |
1252 | p0g[2]=z1g; | |
1253 | } | |
1254 | else { // staight line | |
1255 | // vector of initial straight line direction in glob. coord | |
1256 | v0g[0]=lpar[2]; | |
1257 | v0g[1]=1.0; | |
1258 | v0g[2]=lpar[3]; | |
1259 | ||
1260 | // intercept in yg=0 plane in glob coord | |
1261 | p0g[0]=lpar[0]; | |
1262 | p0g[1]=0.0; | |
1263 | p0g[2]=lpar[1]; | |
1264 | } | |
1265 | AliDebug(3,Form("Line vector: ( %f , %f , %f ) point:( %f , %f , %f )\n",v0g[0],v0g[1],v0g[2],p0g[0],p0g[1],p0g[2])); | |
1266 | ||
1267 | // same in local coord. | |
1268 | Double_t p0l[3],v0l[3]; | |
1269 | fTempHMat->MasterToLocalVect(v0g,v0l); | |
1270 | fTempHMat->MasterToLocal(p0g,p0l); | |
1271 | ||
1272 | if (TMath::Abs(v0l[1])<1e-15) { | |
1273 | AliInfo("Track Y direction in local frame is zero! Cannot proceed..."); | |
1274 | return -1; | |
1275 | } | |
1276 | ||
1277 | // local intersection point | |
1278 | fPintLoc[0] = p0l[0] - (v0l[0]/v0l[1])*p0l[1]; | |
1279 | fPintLoc[1] = 0; | |
1280 | fPintLoc[2] = p0l[2] - (v0l[2]/v0l[1])*p0l[1]; | |
1281 | ||
1282 | // global intersection point | |
1283 | fTempHMat->LocalToMaster(fPintLoc,fPintGlo); | |
1284 | AliDebug(3,Form("Intesect. point: L( %f , %f , %f ) G( %f , %f , %f )\n",fPintLoc[0],fPintLoc[1],fPintLoc[2],fPintGlo[0],fPintGlo[1],fPintGlo[2])); | |
1285 | ||
1286 | return 0; | |
1287 | } | |
1288 | ||
1289 | Int_t AliITSAlignMille2::CalcDerivatives(Int_t paridx, Bool_t islpar) { | |
1290 | /// calculate numerically (ROOT's style) the derivatives for | |
1291 | /// local X intersection and local Z intersection | |
1292 | /// parlist: local (islpar=kTRUE) pgx0, pgz0, ugx0, ugz0 OR riemann's params | |
1293 | /// global (islpar=kFALSE) tx, ty, tz, psi, theta, phi (Raf's angles in deg) | |
1294 | /// return 0 if success | |
1295 | ||
1296 | // copy initial parameters | |
1297 | Double_t lpar[ITSMILLE2_NLOCAL]; | |
1298 | Double_t gpar[ITSMILLE2_NPARCH]; | |
1299 | for (Int_t i=0; i<ITSMILLE2_NLOCAL; i++) lpar[i]=fLocalInitParam[i]; | |
1300 | for (Int_t i=0; i<ITSMILLE2_NPARCH; i++) gpar[i]=fModuleInitParam[i]; | |
1301 | ||
1302 | // trial with fixed dpar... | |
1303 | Double_t dpar=0.0; | |
1304 | ||
1305 | if (islpar) { // track parameters | |
1306 | //dpar=fLocalInitParam[paridx]*0.001; | |
1307 | // set minimum dpar | |
1308 | if (!fBOn) { | |
1309 | if (paridx<2) dpar=1.0e-4; // translations | |
1310 | else dpar=1.0e-6; // direction | |
1311 | } | |
1312 | else { // B Field | |
1313 | // pepo: proviamo con 1/1000, poi evenctually 1/100... | |
1314 | Double_t dfrac=0.01; | |
1315 | switch(paridx) { | |
1316 | case 0: | |
1317 | // RMS cosmics: 1e-4 | |
1318 | dpar = TMath::Max(1.0e-6,TMath::Abs(fLocalInitParam[paridx]*dfrac)); | |
1319 | break; | |
1320 | case 1: | |
1321 | // RMS cosmics: 0.2 | |
1322 | dpar = TMath::Max(0.002,TMath::Abs(fLocalInitParam[paridx]*dfrac)); | |
1323 | break; | |
1324 | case 2: | |
1325 | // RMS cosmics: 9 | |
1326 | dpar = TMath::Max(0.09,TMath::Abs(fLocalInitParam[paridx]*dfrac)); | |
1327 | break; | |
1328 | case 3: | |
1329 | // RMS cosmics: 7 | |
1330 | dpar = TMath::Max(0.07,TMath::Abs(fLocalInitParam[paridx]*dfrac)); | |
1331 | break; | |
1332 | case 4: | |
1333 | // RMS cosmics: 0.3 | |
1334 | dpar = TMath::Max(0.003,TMath::Abs(fLocalInitParam[paridx]*dfrac)); | |
1335 | break; | |
1336 | } | |
1337 | } | |
1338 | } | |
1339 | else { // alignment global parameters | |
1340 | //dpar=fModuleInitParam[paridx]*0.001; | |
1341 | if (paridx<3) dpar=1.0e-4; // translations | |
1342 | else dpar=1.0e-2; // angles | |
1343 | } | |
1344 | ||
1345 | AliDebug(3,Form("+++ using dpar=%g",dpar)); | |
1346 | ||
1347 | // calculate derivative ROOT's like: | |
1348 | // using f(x+h),f(x-h),f(x+h/2),f(x-h2)... | |
1349 | Double_t pintl1[3]; // f(x-h) | |
1350 | Double_t pintl2[3]; // f(x-h/2) | |
1351 | Double_t pintl3[3]; // f(x+h/2) | |
1352 | Double_t pintl4[3]; // f(x+h) | |
1353 | ||
1354 | // first values | |
1355 | if (islpar) lpar[paridx] -= dpar; | |
1356 | else gpar[paridx] -= dpar; | |
1357 | if (CalcIntersectionPoint(lpar, gpar)) return -2; | |
1358 | for (Int_t i=0; i<3; i++) pintl1[i]=fPintLoc[i]; | |
1359 | ||
1360 | // second values | |
1361 | if (islpar) lpar[paridx] += dpar/2; | |
1362 | else gpar[paridx] += dpar/2; | |
1363 | if (CalcIntersectionPoint(lpar, gpar)) return -2; | |
1364 | for (Int_t i=0; i<3; i++) pintl2[i]=fPintLoc[i]; | |
1365 | ||
1366 | // third values | |
1367 | if (islpar) lpar[paridx] += dpar; | |
1368 | else gpar[paridx] += dpar; | |
1369 | if (CalcIntersectionPoint(lpar, gpar)) return -2; | |
1370 | for (Int_t i=0; i<3; i++) pintl3[i]=fPintLoc[i]; | |
1371 | ||
1372 | // fourth values | |
1373 | if (islpar) lpar[paridx] += dpar/2; | |
1374 | else gpar[paridx] += dpar/2; | |
1375 | if (CalcIntersectionPoint(lpar, gpar)) return -2; | |
1376 | for (Int_t i=0; i<3; i++) pintl4[i]=fPintLoc[i]; | |
1377 | ||
1378 | Double_t h2 = 1./(2.*dpar); | |
1379 | Double_t d0 = pintl4[0]-pintl1[0]; | |
1380 | Double_t d2 = 2.*(pintl3[0]-pintl2[0]); | |
1381 | fDerivativeLoc[0] = h2*(4*d2 - d0)/3.; | |
1382 | if (TMath::Abs(fDerivativeLoc[0]) < 1.0e-9) fDerivativeLoc[0] = 0.0; | |
1383 | ||
1384 | d0 = pintl4[2]-pintl1[2]; | |
1385 | d2 = 2.*(pintl3[2]-pintl2[2]); | |
1386 | fDerivativeLoc[2] = h2*(4*d2 - d0)/3.; | |
1387 | if (TMath::Abs(fDerivativeLoc[2]) < 1.0e-9) fDerivativeLoc[2]=0.0; | |
1388 | ||
1389 | AliDebug(3,Form("\n+++ derivatives +++ \n")); | |
1390 | AliDebug(3,Form("+++ dXLoc/dpar = %g +++\n",fDerivativeLoc[0])); | |
1391 | AliDebug(3,Form("+++ dZLoc/dpar = %g +++\n\n",fDerivativeLoc[0])); | |
1392 | ||
1393 | return 0; | |
1394 | } | |
1395 | ||
1396 | ||
1397 | Int_t AliITSAlignMille2::AddLocalEquation(Mille2Data &m) { | |
1398 | /// Define local equation for current cluster in X and Z coor. | |
1399 | /// and store them to memory | |
1400 | /// return 0 if success | |
1401 | int nLev = 0; | |
1402 | // store first interaction point | |
1403 | if (CalcIntersectionPoint(fLocalInitParam, fModuleInitParam)) return -4; | |
1404 | for (Int_t i=0; i<3; i++) fPintLoc0[i]=fPintLoc[i]; | |
1405 | AliDebug(2,Form("Intesect. point: L( %f , %f , %f )",fPintLoc[0],fPintLoc[1],fPintLoc[2])); | |
1406 | ||
1407 | // calculate local derivatives numerically | |
1408 | Bool_t zeroX = kTRUE; | |
1409 | Bool_t zeroZ = kTRUE; | |
1410 | // | |
1411 | for (Int_t i=0; i<fNLocal; i++) { | |
1412 | if (CalcDerivatives(i,kTRUE)) return -1; | |
1413 | m.derlocX[i] = fDerivativeLoc[0]; | |
1414 | m.derlocZ[i] = fDerivativeLoc[2]; | |
1415 | if (zeroX) zeroX = fDerivativeLoc[0]==0; | |
1416 | if (zeroZ) zeroZ = fDerivativeLoc[2]==0; | |
1417 | } | |
1418 | // for (Int_t i=0; i<fNLocal; i++) AliDebug(2,Form("Local parameter %d - dXdpar = %g - dZdpar = %g\n",i,dXdL[i],dZdL[i])); | |
1419 | if (zeroX) {AliInfo("Aborting: zero local X derivatives!"); return -2;} | |
1420 | if (zeroZ) {AliInfo("Aborting: zero local Z derivatives!"); return -2;} | |
1421 | // | |
1422 | // | |
1423 | AliITSAlignMille2Module* endModule = fCurrentModule; | |
1424 | // | |
1425 | do { | |
1426 | if (nLev==0 || !fUseGlobalDelta) zeroX = zeroZ = kTRUE; | |
1427 | int shiftL = nLev*ITSMILLE2_NPARCH; | |
1428 | for (Int_t i=0; i<ITSMILLE2_NPARCH; i++) { | |
1429 | if (nLev==0 || !fUseGlobalDelta) { | |
1430 | if (CalcDerivatives(i,kFALSE)) return -1; | |
1431 | m.dergloX[shiftL + i] = fDerivativeLoc[0]; | |
1432 | m.dergloZ[shiftL + i] = fDerivativeLoc[2]; | |
1433 | if (zeroX) zeroX = fDerivativeLoc[0]==0; | |
1434 | if (zeroZ) zeroZ = fDerivativeLoc[2]==0; | |
1435 | } | |
1436 | else { // for the global delta the deravites of different levels are the same | |
1437 | m.dergloX[shiftL + i] = m.dergloX[shiftL + i - ITSMILLE2_NPARCH]; | |
1438 | m.dergloZ[shiftL + i] = m.dergloZ[shiftL + i - ITSMILLE2_NPARCH]; | |
1439 | } | |
1440 | } | |
1441 | // for (Int_t i=0; i<ITSMILLE2_NPARCH; i++) AliDebug(2,Form("Global parameter %d - dXdpar = %g - dZdpar = %g\n",i,dXdG[i],dZdG[i])); | |
1442 | if (zeroX) {AliInfo("Aborting: zero global X derivatives!");return -2;} | |
1443 | if (zeroZ) {AliInfo("Aborting: zero global Z derivatives!");return -2;} | |
1444 | // set equation for Xloc coordinate | |
1445 | m.moduleIDX[nLev] = fCurrentModule->GetUniqueID(); | |
1446 | nLev++; | |
1447 | // | |
1448 | } while( (fCurrentModule=fCurrentModule->GetParent()) ); | |
1449 | // | |
1450 | // ok, can copy to m | |
1451 | AliDebug(2,Form("Adding local equation X with fMeas=%.6f and fSigma=%.6f",(fMeasLoc[0]-fPintLoc0[0]), fSigmaLoc[0])); | |
1452 | m.measX = fMeasLoc[0]-fPintLoc0[0]; | |
1453 | m.sigmaX = fSigmaLoc[0]; | |
1454 | // | |
1455 | AliDebug(2,Form("Adding local equation Z with fMeas=%.6f and fSigma=%.6f",(fMeasLoc[2]-fPintLoc0[2]), fSigmaLoc[2])); | |
1456 | m.measZ = fMeasLoc[2]-fPintLoc0[2]; | |
1457 | m.sigmaZ = fSigmaLoc[2]; | |
1458 | // | |
1459 | m.levFilled = nLev; | |
1460 | fCurrentModule = endModule; | |
1461 | // | |
1462 | return 0; | |
1463 | } | |
1464 | ||
1465 | void AliITSAlignMille2::SetLocalEquations(const Mille2Data *marr, Int_t neq) { | |
1466 | /// Set local equations with data stored in m | |
1467 | /// return 0 if success | |
1468 | // | |
1469 | for (Int_t j=0; j<neq; j++) { | |
1470 | // | |
1471 | const Mille2Data &m = marr[j]; | |
1472 | // | |
1473 | // set equation for Xloc coordinate | |
1474 | AliDebug(2,Form("setting local equation X with fMeas=%.6f and fSigma=%.6f",m.measX, m.sigmaX)); | |
1475 | for (int i=fNLocal; i--;) SetLocalDerivative( i, m.derlocX[i] ); | |
1476 | for (int il=m.levFilled;il--;) { | |
1477 | GetMilleModule(m.moduleIDX[il])->IncNProcessedPoints(); | |
1478 | int hlev = m.moduleIDX[il]*ITSMILLE2_NPARCH; // id of the supermodule | |
1479 | int llev = il*ITSMILLE2_NPARCH; | |
1480 | for (int i=ITSMILLE2_NPARCH; i--;) SetGlobalDerivative( hlev+i, m.dergloX[llev+i] ); | |
1481 | } | |
1482 | // | |
1483 | fMillepede->SetLocalEquation(fGlobalDerivatives, fLocalDerivatives, m.measX, m.sigmaX); | |
1484 | // | |
1485 | // set equation for Zloc coordinate | |
1486 | AliDebug(2,Form("setting local equation Z with fMeas=%.6f and fSigma=%.6f",m.measZ, m.sigmaZ)); | |
1487 | for (int i=fNLocal; i--;) SetLocalDerivative( i, m.derlocZ[i] ); | |
1488 | for (int il=m.levFilled;il--;) { | |
1489 | int hlev = m.moduleIDX[il]*ITSMILLE2_NPARCH; // id of the supermodule | |
1490 | int llev = il*ITSMILLE2_NPARCH; | |
1491 | for (int i=ITSMILLE2_NPARCH; i--;) SetGlobalDerivative(hlev+i, m.dergloZ[llev+i] ); | |
1492 | } | |
1493 | fMillepede->SetLocalEquation(fGlobalDerivatives, fLocalDerivatives, m.measZ, m.sigmaZ); | |
1494 | } | |
1495 | } | |
1496 | ||
1497 | Int_t AliITSAlignMille2::GlobalFit(Double_t *parameters,Double_t *errors,Double_t *pulls) { | |
1498 | /// Call global fit; Global parameters are stored in parameters | |
1499 | if (!fIsMilleInit) { | |
1500 | AliInfo("Millepede has not been initialized!"); | |
1501 | return 0; | |
1502 | } | |
1503 | int res = fMillepede->GlobalFit(parameters,errors,pulls); | |
1504 | AliInfo(Form("%s fitting global parameters!",res ? "Done":"Failed")); | |
1505 | return res; | |
1506 | } | |
1507 | ||
1508 | Double_t AliITSAlignMille2::GetParError(Int_t iPar) { | |
1509 | /// Get error of parameter iPar | |
1510 | if (!fIsMilleInit) { | |
1511 | AliInfo("Millepede has not been initialized!"); | |
1512 | return 0; | |
1513 | } | |
1514 | Double_t lErr = fMillepede->GetParError(iPar); | |
1515 | return lErr; | |
1516 | } | |
1517 | ||
1518 | void AliITSAlignMille2::PrintGlobalParameters() { | |
1519 | /// Print global parameters | |
1520 | if (!fIsMilleInit) { | |
1521 | AliInfo("Millepede has not been initialized!"); | |
1522 | return; | |
1523 | } | |
1524 | fMillepede->PrintGlobalParameters(); | |
1525 | } | |
1526 | ||
1527 | // //_________________________________________________________________________ | |
1528 | Int_t AliITSAlignMille2::LoadSuperModuleFile(const Char_t *sfile) | |
1529 | { | |
1530 | // load definitions of supermodules from a root file | |
1531 | // return 0 if success | |
1532 | ||
1533 | TFile *smf=TFile::Open(sfile); | |
1534 | if (!smf->IsOpen()) { | |
1535 | AliInfo(Form("Cannot open supermodule file %s",sfile)); | |
1536 | return -1; | |
1537 | } | |
1538 | ||
1539 | TClonesArray *sma=(TClonesArray*)smf->Get("ITSMilleSuperModules"); | |
1540 | if (!sma) { | |
1541 | AliInfo(Form("Cannot find ITSMilleSuperModules array in file")); | |
1542 | return -2; | |
1543 | } | |
1544 | Int_t nsma=sma->GetEntriesFast(); | |
1545 | AliInfo(Form("Array of SuperModules with %d entries\n",nsma)); | |
1546 | // | |
1547 | Char_t st[250]; | |
1548 | char symname[150]; | |
1549 | UShort_t volid; | |
1550 | TGeoHMatrix m; | |
1551 | // | |
1552 | for (Int_t i=0; i<nsma; i++) { | |
1553 | AliAlignObjParams *a = (AliAlignObjParams*)sma->UncheckedAt(i); | |
1554 | volid=a->GetVolUID(); | |
1555 | strcpy(st,a->GetSymName()); | |
1556 | a->GetMatrix(m); | |
1557 | // | |
1558 | sscanf(st,"%s",symname); | |
1559 | // decode module list | |
1560 | char *stp=strstr(st,"ModuleList:"); | |
1561 | if (!stp) return -3; | |
1562 | stp += 11; | |
1563 | int idx[2200]; | |
1564 | char spp[200]; int jp=0; | |
1565 | char cl[20]; | |
1566 | strcpy(st,stp); | |
1567 | int l=strlen(st); | |
1568 | int j=0; | |
1569 | int n=0; | |
1570 | // | |
1571 | while (j<=l) { | |
1572 | if (st[j]==9 || st[j]==32 || st[j]==10 || st[j]==0) { | |
1573 | spp[jp]=0; | |
1574 | jp=0; | |
1575 | if (strlen(spp)) { | |
1576 | int k=strcspn(spp,"-"); | |
1577 | if (k<int(strlen(spp))) { // c'e' il - | |
1578 | strcpy(cl,&(spp[k+1])); | |
1579 | spp[k]=0; | |
1580 | int ifrom=atoi(spp); int ito=atoi(cl); | |
1581 | for (int b=ifrom; b<=ito; b++) { | |
1582 | idx[n]=b; | |
1583 | n++; | |
1584 | } | |
1585 | } | |
1586 | else { // numerillo singolo | |
1587 | idx[n]=atoi(spp); | |
1588 | n++; | |
1589 | } | |
1590 | } | |
1591 | } | |
1592 | else { | |
1593 | spp[jp]=st[j]; | |
1594 | jp++; | |
1595 | } | |
1596 | j++; | |
1597 | } | |
1598 | UShort_t volidsv[2198]; | |
1599 | for (j=0;j<n;j++) { | |
1600 | volidsv[j]=AliITSAlignMille2Module::GetVolumeIDFromIndex(idx[j]); | |
1601 | if (!volidsv[j]) { | |
1602 | AliInfo(Form("Index %d not valid (range 0->%d)",idx[j],kMaxITSSensID)); | |
1603 | return -5; | |
1604 | } | |
1605 | } | |
1606 | Int_t smindex=int(2198+volid-14336); // virtual index | |
1607 | fSuperModule.AddAtAndExpand(new AliITSAlignMille2Module(smindex,volid,symname,&m,n,volidsv),fNSuperModules); | |
1608 | // | |
1609 | fNSuperModules++; | |
1610 | } | |
1611 | ||
1612 | smf->Close(); | |
1613 | // | |
1614 | return 0; | |
1615 | } | |
1616 | ||
1617 | //_________________________________________________________________________ | |
1618 | void AliITSAlignMille2::ConstrainModuleSubUnits(Int_t idm, Double_t val, UInt_t pattern) | |
1619 | { | |
1620 | // require that sum of modifications for the childs of this module is = val, i.e. | |
1621 | // the internal corrections moves the module as a whole by fixed value (0 by default). | |
1622 | // pattern is the bit pattern for the parameters to constrain | |
1623 | // | |
1624 | static TObjArray childs; | |
1625 | childs.Clear(); | |
1626 | // | |
1627 | // build list of childs for this module | |
1628 | int nChilds = 0; | |
1629 | AliITSAlignMille2Module* parent = GetMilleModule(idm); | |
1630 | if (!parent) return; | |
1631 | // | |
1632 | for (int i=fNModules;i--;) { | |
1633 | AliITSAlignMille2Module* child = GetMilleModule(i); | |
1634 | if (child->GetParent() == parent) childs.AddAtAndExpand(child,nChilds++); | |
1635 | } | |
1636 | if (nChilds<1) return; | |
1637 | // | |
1638 | int npc = 0; | |
1639 | for (int ip=0;ip<kNParCh;ip++) { | |
1640 | if ( !((pattern>>ip)&0x1) /*|| !parent->IsFreeDOF(ip)*/) continue; | |
1641 | ResetLocalEquation(); | |
1642 | for (int ich=nChilds;ich--;) fGlobalDerivatives[childs[ich]->GetUniqueID()*kNParCh+ip] = 1.0; | |
1643 | AddConstraint(fGlobalDerivatives,val); | |
1644 | npc++; | |
1645 | } | |
1646 | // | |
1647 | AliInfo(Form("Constrained %d params for %d submodules of module #%d: %s",npc,nChilds,idm,parent->GetName())); | |
1648 | // | |
1649 | } | |
1650 | ||
1651 | //_________________________________________________________________________ | |
1652 | void AliITSAlignMille2::PostConstrainModuleSubUnitsMedian(Int_t idm, Double_t val, UInt_t pattern) | |
1653 | { | |
1654 | // require that median of the modifications for the childs of this module is = val, i.e. | |
1655 | // the internal corrections moves the module as a whole by fixed value (0 by default) | |
1656 | // module the outliers. | |
1657 | // pattern is the bit pattern for the parameters to constrain | |
1658 | // The difference between the mean and the median will be transfered to the parent | |
1659 | // | |
1660 | static TObjArray childs; | |
1661 | childs.Clear(); | |
1662 | // | |
1663 | // build list of childs for this module | |
1664 | int nChilds = 0; | |
1665 | AliITSAlignMille2Module* parent = GetMilleModule(idm); | |
1666 | if (!parent) return; | |
1667 | // | |
1668 | for (int i=fNModules;i--;) { | |
1669 | AliITSAlignMille2Module* child = GetMilleModule(i); | |
1670 | if (child->GetParent() == parent) childs.AddAtAndExpand(child,nChilds++); | |
1671 | } | |
1672 | if (nChilds<1) return; | |
1673 | // | |
1674 | int npc = 0; | |
1675 | double *deltas = fMillepede->GetDeltaPars(); | |
1676 | double *tmpArr = new double[nChilds]; | |
1677 | // | |
1678 | for (int ip=0;ip<kNParCh;ip++) { | |
1679 | if ( !((pattern>>ip)&0x1) /*|| !parent->IsFreeDOF(ip)*/) continue; | |
1680 | // compute the median of the deltas | |
1681 | for (int ich=nChilds;ich--;) tmpArr[ich] = deltas[childs[ich]->GetUniqueID()*kNParCh+ip]; | |
1682 | for (int ic0=0;ic0<nChilds;ic0++) // order the deltas | |
1683 | for (int ic1=ic0+1;ic1<nChilds;ic1++) | |
1684 | if (tmpArr[ic0]>tmpArr[ic1]) {double tv=tmpArr[ic0]; tmpArr[ic0]=tmpArr[ic1]; tmpArr[ic1]=tv;} | |
1685 | // | |
1686 | int kmed = nChilds/2; | |
1687 | double median = (tmpArr[kmed]+tmpArr[nChilds-kmed-1])/2.; | |
1688 | // | |
1689 | for (int ich=nChilds;ich--;) deltas[childs[ich]->GetUniqueID()*kNParCh+ip] -= median - val; | |
1690 | deltas[parent->GetUniqueID()*kNParCh+ip] += median - val; | |
1691 | npc++; | |
1692 | } | |
1693 | delete[] tmpArr; | |
1694 | // | |
1695 | AliInfo(Form("Applied median constraint to %d params for %d submodules of module #%d: %s",npc,nChilds,idm,parent->GetName())); | |
1696 | // | |
1697 | } | |
1698 | ||
1699 | //_________________________________________________________________________ | |
1700 | void AliITSAlignMille2::ConstrainOrphans(Double_t val, UInt_t pattern) | |
1701 | { | |
1702 | // require that median of the modifications for the supermodules which have no parents is = val, i.e. | |
1703 | // the corrections moves the whole setup by fixed value (0 by default) modulo the outliers. | |
1704 | // pattern is the bit pattern for the parameters to constrain | |
1705 | // | |
1706 | static TObjArray modSet; | |
1707 | modSet.Clear(); | |
1708 | // | |
1709 | // build list of childs for this module | |
1710 | int nModules = 0; | |
1711 | int *nFree = new int[kNParCh]; | |
1712 | for (int i=0;i<kNParCh;i++) nFree[i] = 0; | |
1713 | // | |
1714 | for (int i=fNModules;i--;) { | |
1715 | AliITSAlignMille2Module* module = GetMilleModule(i); | |
1716 | if (module->GetParent()) continue; // skip this | |
1717 | for (int ip=0;ip<kNParCh;ip++) if ( ((pattern>>ip)&0x1) && module->IsFreeDOF(ip)) nFree[ip]++; | |
1718 | modSet.AddAtAndExpand(module,nModules++); | |
1719 | } | |
1720 | if (nModules<1) return; | |
1721 | // | |
1722 | int npc = 0; | |
1723 | for (int ip=0;ip<kNParCh;ip++) { | |
1724 | if (nFree[ip]<1) continue; // nothing to do | |
1725 | ResetLocalEquation(); | |
1726 | for (int ich=nModules;ich--;) { | |
1727 | AliITSAlignMille2Module* module = (AliITSAlignMille2Module*) modSet[ich]; | |
1728 | if ( !((pattern>>ip)&0x1) || !module->IsFreeDOF(ip)) continue; | |
1729 | fGlobalDerivatives[module->GetUniqueID()*kNParCh+ip] = 1.0; | |
1730 | } | |
1731 | AddConstraint(fGlobalDerivatives,val); | |
1732 | npc++; | |
1733 | } | |
1734 | // | |
1735 | delete[] nFree; | |
1736 | AliInfo(Form("Constrained %d params for %d orphan modules",npc,nModules)); | |
1737 | // | |
1738 | } | |
1739 | //_________________________________________________________________________ | |
1740 | void AliITSAlignMille2::PostConstrainOrphansMedian(Double_t val, UInt_t pattern) | |
1741 | { | |
1742 | // require that sum of modifications for the supermodules which have no parents is = val, i.e. | |
1743 | // the corrections moves the whole setup by fixed value (0 by default). | |
1744 | // pattern is the bit pattern for the parameters to constrain | |
1745 | // | |
1746 | static TObjArray modSet; | |
1747 | modSet.Clear(); | |
1748 | // | |
1749 | // build list of childs for this module | |
1750 | int nModules = 0; | |
1751 | int *nFree = new int[kNParCh]; | |
1752 | for (int i=0;i<kNParCh;i++) nFree[i] = 0; | |
1753 | // | |
1754 | for (int i=fNModules;i--;) { | |
1755 | AliITSAlignMille2Module* module = GetMilleModule(i); | |
1756 | if (module->GetParent()) continue; // skip this | |
1757 | for (int ip=0;ip<kNParCh;ip++) if ( ((pattern>>ip)&0x1) && module->IsFreeDOF(ip)) nFree[ip]++; | |
1758 | modSet.AddAtAndExpand(module,nModules++); | |
1759 | } | |
1760 | if (nModules<1) return; | |
1761 | // | |
1762 | int npc = 0; | |
1763 | double *deltas = fMillepede->GetDeltaPars(); | |
1764 | double *tmpArr = new double[nModules]; | |
1765 | // | |
1766 | for (int ip=0;ip<kNParCh;ip++) { | |
1767 | if (nFree[ip]<1) continue; // nothing to do | |
1768 | // compute the median of the deltas | |
1769 | for (int ich=nModules;ich--;) tmpArr[ich] = deltas[modSet[ich]->GetUniqueID()*kNParCh+ip]; | |
1770 | for (int ic0=0;ic0<nModules;ic0++) // order the deltas | |
1771 | for (int ic1=ic0+1;ic1<nModules;ic1++) | |
1772 | if (tmpArr[ic0]>tmpArr[ic1]) {double tv=tmpArr[ic0]; tmpArr[ic0]=tmpArr[ic1]; tmpArr[ic1]=tv;}; | |
1773 | // | |
1774 | int kmed = nModules/2; | |
1775 | double median = (tmpArr[kmed]+tmpArr[nModules-kmed-1])/2.; | |
1776 | // | |
1777 | for (int ich=nModules;ich--;) { | |
1778 | AliITSAlignMille2Module* module = (AliITSAlignMille2Module*) modSet[ich]; | |
1779 | if ( !((pattern>>ip)&0x1) || !module->IsFreeDOF(ip)) continue; | |
1780 | deltas[module->GetUniqueID()*kNParCh+ip] -= median - val; | |
1781 | } | |
1782 | npc++; | |
1783 | } | |
1784 | // | |
1785 | delete[] nFree; | |
1786 | delete[] tmpArr; | |
1787 | AliInfo(Form("Applied median constraint to %d params for %d orphan modules",npc,nModules)); | |
1788 | // | |
1789 | } | |
1790 | ||
1791 | //_________________________________________________________________________ | |
1792 | void AliITSAlignMille2::ConstrainLinComb(const Int_t *vidLst, const Float_t *wghLst, Int_t nmd, Double_t val, UInt_t pattern) | |
1793 | { | |
1794 | // require that the linear combinations of the nmd modules (refered by their volume ID) from the | |
1795 | // modList with the coefficients wghLst adds up to val. | |
1796 | // pattern is the bit pattern for the parameters to constrain. | |
1797 | // | |
1798 | static TObjArray modSet; | |
1799 | modSet.Clear(); | |
1800 | // | |
1801 | // build list of childs for this module | |
1802 | int nModules = 0; | |
1803 | int *nFree = new int[kNParCh]; | |
1804 | for (int i=0;i<kNParCh;i++) nFree[i] = 0; | |
1805 | // | |
1806 | for (int imd=nmd;imd--;) { | |
1807 | UShort_t vid = (UShort_t)vidLst[imd]; | |
1808 | for (int i=fNModules;i--;) { | |
1809 | AliITSAlignMille2Module* module = GetMilleModule(i); | |
1810 | if (module->GetVolumeID() == vid) { | |
1811 | modSet.AddAtAndExpand(module,nModules++); | |
1812 | for (int ip=0;ip<kNParCh;ip++) if ( ((pattern>>ip)&0x1) && module->IsFreeDOF(ip)) nFree[ip]++; | |
1813 | break; | |
1814 | } | |
1815 | } | |
1816 | } | |
1817 | // | |
1818 | if (nModules != nmd) { | |
1819 | AliInfo(Form("Error: constraint for %d modules requested but %d are found",nmd,nModules)); | |
1820 | delete[] nFree; | |
1821 | return; | |
1822 | } | |
1823 | int npc = 0; | |
1824 | for (int ip=0;ip<kNParCh;ip++) { | |
1825 | if (nFree[ip]<1) continue; // nothing to do | |
1826 | ResetLocalEquation(); | |
1827 | for (int ich=nModules;ich--;) { | |
1828 | AliITSAlignMille2Module* module = (AliITSAlignMille2Module*) modSet[ich]; | |
1829 | if ( !((pattern>>ip)&0x1) || !module->IsFreeDOF(ip)) continue; | |
1830 | fGlobalDerivatives[module->GetUniqueID()*kNParCh+ip] = wghLst[ich]; | |
1831 | } | |
1832 | AddConstraint(fGlobalDerivatives,val); | |
1833 | npc++; | |
1834 | } | |
1835 | // | |
1836 | delete[] nFree; | |
1837 | AliInfo(Form("Constrained %d params for linerar combination of %d modules",npc,nModules)); | |
1838 | // | |
1839 | } |