Interface to AliMillePede2 for ITS alignment + related classes (R. Shaoyan)
[u/mrichter/AliRoot.git] / ITS / AliITSAlignMille2ConstrArray.cxx
1 #include "AliITSAlignMille2ConstrArray.h"
2 #include "AliITSAlignMille2Module.h"
3
4 //========================================================================================================
5 ClassImp(AliITSAlignMille2ConstrArray)
6
7 //________________________________________________________________________________________________________
8 AliITSAlignMille2ConstrArray::AliITSAlignMille2ConstrArray() :
9 AliITSAlignMille2Constraint(),
10 fModuleIDs(0),
11 fModulePatt(0),
12 fCoeffs(0),
13 fError(0)
14 {}
15
16 //________________________________________________________________________________________________________
17 AliITSAlignMille2ConstrArray::AliITSAlignMille2ConstrArray(const Char_t* name,Double_t *parcf,Int_t npar,Double_t val,Double_t err) :
18   AliITSAlignMille2Constraint(name,kTypeGaussian,-100,val,0),
19   fModuleIDs(0),
20   fModulePatt(0),
21   fCoeffs(npar),
22   fError(err)
23 {
24   for (int i=0;i<npar;i++) {fCoeffs[i] = parcf[i]; if (parcf[i]!=0) SetBit(0x1<<i);}
25 }
26
27 //________________________________________________________________________________________________________
28 AliITSAlignMille2ConstrArray::AliITSAlignMille2ConstrArray(const AliITSAlignMille2ConstrArray& src) :
29   AliITSAlignMille2Constraint(src),
30   fModuleIDs(0),
31   fModulePatt(0),
32   fCoeffs(0),
33   fError(0)
34 {/* DUMMY */} 
35
36 //________________________________________________________________________________________________________
37 void AliITSAlignMille2ConstrArray::AddModule(AliITSAlignMille2Module* mod)
38 {
39   int nmd = GetNModules();
40   // check if its already not there
41   for (int im=nmd;im--;) {if (mod->GetUniqueID() == (UInt_t)fModuleIDs[im]) return; }
42   fModuleIDs.Set(nmd+1);
43   fModulePatt.Set(nmd+1);
44   fModuleIDs[nmd] = mod->GetUniqueID();
45   double jacobian[AliITSAlignMille2Module::kMaxParGeom][AliITSAlignMille2Module::kMaxParGeom];
46   if (mod->GeomParamsGlobal()) mod->CalcDerivLocGlo(&jacobian[0][0]);
47   //
48   Short_t patt =  GetPattern();
49   if (mod->GeomParamsGlobal()) {
50     // the constraint is defined in the module's local frame. If the alignment of geom params is
51     // done in the global frame, we need to set the real parameter involved
52     patt &= 0xffff<< AliITSAlignMille2Module::kMaxParGeom;     // reset the geometry parameters
53     for (int i=0;i<AliITSAlignMille2Module::kMaxParGeom;i++) {
54       if (!IncludesParam(i)) continue;
55       for (int j=0;j<AliITSAlignMille2Module::kMaxParGeom;j++) if (jacobian[i][j]!=0) patt |= (0x1<<j);
56     }
57   }
58   fModulePatt[nmd] = patt;
59 }
60
61 //________________________________________________________________________________________________________
62 Bool_t AliITSAlignMille2ConstrArray::IncludesModule(Int_t id) const
63 {
64   // is this module mentioned in the list?
65   int nmd = GetNModules();
66   for (int i=nmd;i--;) if (fModuleIDs[i]==id) return kTRUE;
67   return kFALSE;
68 }
69
70 //________________________________________________________________________________________________________
71 Bool_t AliITSAlignMille2ConstrArray::IncludesModPar(Int_t id,Int_t par) const
72 {
73   // is this module/parameter mentioned in the list?
74   int nmd = GetNModules();
75   for (int i=nmd;i--;) {
76     if (fModuleIDs[i]!=id) continue;
77     if (fModulePatt[i] & (0x1<<par)) return kTRUE;
78   }
79   return kFALSE;
80 }
81
82 //________________________________________________________________________________________________________
83 Bool_t AliITSAlignMille2ConstrArray::IncludesModPar(AliITSAlignMille2Module* mod, Int_t par) const
84 {
85   // is this module/parameter mentioned in the list?
86   return IncludesModPar(mod->GetUniqueID(), par);
87 }
88
89
90 //________________________________________________________________________________________________________
91 void AliITSAlignMille2ConstrArray::Print(Option_t* ) const
92 {
93   printf("#%3d Constraint %s of type %d | Value=%+e Error=%+e\n",GetConstraintID(),GetName(),GetType(),GetValue(),GetError());
94   printf("Weights on params: "); for (int i=0;i<GetNCoeffs();i++) printf("%+.3e ",GetCoeff(i)); 
95   printf("\nModules involved: \n");
96   int nmd = GetNModules();
97   for (int i=0;i<nmd;i++) {printf("%4d |",GetModuleID(i)); if ( ((i+1)%14)==0 ) printf("\n");}
98   if ( nmd%14 ) printf("\n");
99   //
100 }
101