1 |
mocchiut |
1.1 |
/** |
2 |
|
|
* \file CaloFranzini.cpp |
3 |
|
|
* \author Emiliano Mocchiutti (2007/07/18) |
4 |
|
|
*/ |
5 |
|
|
// |
6 |
|
|
// headers |
7 |
|
|
// |
8 |
|
|
#include <CaloFranzini.h> |
9 |
|
|
//-------------------------------------- |
10 |
|
|
/** |
11 |
|
|
* Default constructor |
12 |
|
|
*/ |
13 |
|
|
CaloFranzini::CaloFranzini(){ |
14 |
|
|
Clear(); |
15 |
|
|
} |
16 |
|
|
|
17 |
|
|
CaloFranzini::CaloFranzini(PamLevel2 *l2p){ |
18 |
|
|
// |
19 |
|
|
file = NULL; |
20 |
|
|
brig = NULL; |
21 |
|
|
nbin = 0; |
22 |
|
|
// |
23 |
|
|
L2 = l2p; |
24 |
|
|
// |
25 |
|
|
if ( !L2->IsORB() ) printf(" WARNING: OrbitalInfo Tree is needed, the plugin could not work properly without it \n"); |
26 |
|
|
// |
27 |
|
|
// Default variables |
28 |
|
|
// |
29 |
|
|
debug = false; |
30 |
|
|
dolong = true; |
31 |
|
|
dofull = false; |
32 |
|
|
// |
33 |
|
|
Clear(); |
34 |
|
|
// |
35 |
|
|
} |
36 |
|
|
|
37 |
|
|
void CaloFranzini::Clear(){ |
38 |
|
|
// |
39 |
|
|
OBT = 0; |
40 |
|
|
PKT = 0; |
41 |
|
|
atime = 0; |
42 |
|
|
longtzeta = 0.; |
43 |
|
|
fulltzeta = 0.; |
44 |
|
|
memset(estrip, 0, 2*22*96*sizeof(Float_t)); |
45 |
|
|
memset(qplane, 0, 2*22*sizeof(Float_t)); |
46 |
|
|
// |
47 |
|
|
} |
48 |
|
|
|
49 |
|
|
void CaloFranzini::Print(){ |
50 |
|
|
// |
51 |
|
|
Process(); |
52 |
|
|
// |
53 |
|
|
printf("========================================================================\n"); |
54 |
|
|
printf(" OBT: %u PKT: %u ATIME: %u \n",OBT,PKT,atime); |
55 |
|
|
printf(" debug [debug flag]:.. %i\n",debug); |
56 |
|
|
printf(" longtzeta :.. %f\n",longtzeta); |
57 |
|
|
printf(" fulltzeta :.. %f\n",fulltzeta); |
58 |
|
|
printf("========================================================================\n"); |
59 |
|
|
// |
60 |
|
|
} |
61 |
|
|
|
62 |
|
|
void CaloFranzini::Delete(){ |
63 |
|
|
if ( file ) file->Close(); |
64 |
|
|
Clear(); |
65 |
|
|
printf(" ERROR: CaloFranzini variables _NOT_ filled \n"); |
66 |
|
|
}; |
67 |
|
|
// |
68 |
|
|
if ( debug ) this->Print(); |
69 |
|
|
if ( debug ) printf(" exit \n"); |
70 |
|
|
// |
71 |
|
|
} |
72 |
|
|
|
73 |
|
|
Bool_t CaloFranzini::CreateMatrixFile(TString matrixfile){ |
74 |
|
|
// |
75 |
|
|
file = new TFile(matrixfile.Data(),"READ"); |
76 |
|
|
// |
77 |
|
|
if ( !file || file->IsZombie() ){ |
78 |
|
|
file = new TFile(matrixfile.Data(),"RECREATE"); |
79 |
|
|
} else { |
80 |
|
|
printf(" ERROR: file %s already existing!\n Choose another name or delete the old file\n",matrixfile.Data()); |
81 |
|
|
return(false); |
82 |
|
|
}; |
83 |
|
|
// |
84 |
|
|
return(true); |
85 |
|
|
// |
86 |
|
|
} |
87 |
|
|
|
88 |
|
|
void CaloFranzini::WriteNumBin(Int_t numbin){ |
89 |
|
|
TArrayI nbi(1, &numbin); |
90 |
|
|
file->WriteObject(&nbi, "nbinenergy"); |
91 |
|
|
} |
92 |
|
|
|
93 |
|
|
void CaloFranzini::WriteRigBin(TArrayF *rigbin){ |
94 |
|
|
file->WriteObject(&rigbin, "binrig"); |
95 |
|
|
} |
96 |
|
|
|
97 |
|
|
void CaloFranzini::WriteLongMean(TArrayF *qpl, Int_t bin){ |
98 |
|
|
TString name = Form("qplmeann%i",bin); |
99 |
|
|
file->WriteObject(&qpl,name.Data()); |
100 |
|
|
} |
101 |
|
|
|
102 |
|
|
void CaloFranzini::WriteLongMatrix(TMatrixD *mat, Int_t bin){ |
103 |
|
|
TString name = Form("matrixn%i",bin); |
104 |
|
|
file->WriteObject(&(*mat),name.Data()); |
105 |
|
|
} |
106 |
|
|
|
107 |
|
|
void CaloFranzini::CloseMatrixFile(){ |
108 |
|
|
file->Close(); |
109 |
|
|
} |
110 |
|
|
|
111 |
|
|
|
112 |
|
|
Bool_t CaloFranzini::Open(TString matrixfile){ |
113 |
|
|
// |
114 |
|
|
// find matrix file |
115 |
|
|
// |
116 |
|
|
if ( !strcmp(matrixfile.Data(),"") ){ |
117 |
|
|
if (dolong) matrixfile = (TString)gSystem->ExpandPathName("$PAM_CALIB")+"/cal-param/covmatrix_longel.root"; |
118 |
|
|
if (dofull) matrixfile = (TString)gSystem->ExpandPathName("$PAM_CALIB")+"/cal-param/covmatrix_fullel.root"; |
119 |
|
|
}; |
120 |
|
|
// |
121 |
|
|
file = new TFile(matrixfile.Data(),"READ"); |
122 |
|
|
// |
123 |
|
|
if ( !file || file->IsZombie() ){ |
124 |
|
|
printf(" ERROR: cannot open file %s \n",matrixfile.Data()); |
125 |
|
|
return(false); |
126 |
|
|
}; |
127 |
|
|
// |
128 |
|
|
TArrayI *numbin = (TArrayI*)file->Get("nbinenergy"); |
129 |
|
|
if ( !numbin ){ |
130 |
|
|
printf(" ERROR: cannot read number of bins from file %s \n",matrixfile.Data()); |
131 |
|
|
return(false); |
132 |
|
|
}; |
133 |
|
|
nbin = (Int_t)numbin->At(0); |
134 |
|
|
if ( nbin <= 0 ){ |
135 |
|
|
printf(" ERROR: cannot work with 0 energy bins (from file %s) \n",matrixfile.Data()); |
136 |
|
|
return(false); |
137 |
|
|
}; |
138 |
|
|
// |
139 |
|
|
brig = (TArrayF*)file->Get("binrig"); |
140 |
|
|
if ( !brig ){ |
141 |
|
|
printf(" ERROR: cannot read rigidity binning from file %s \n",matrixfile.Data()); |
142 |
|
|
return(false); |
143 |
|
|
}; |
144 |
|
|
// |
145 |
|
|
return(true); |
146 |
|
|
// |
147 |
|
|
} |
148 |
|
|
|
149 |
|
|
TMatrixD *CaloFranzini::LoadCovarianceMatrix(Float_t rig){ |
150 |
|
|
// |
151 |
|
|
TString name; |
152 |
|
|
for (Int_t i = 0; i<nbin-1; i++){ |
153 |
|
|
if ( rig>=brig->At(i) && rig < brig->At(i+1) ){ |
154 |
|
|
name = Form("matrixn%i",i); |
155 |
|
|
break; |
156 |
|
|
}; |
157 |
|
|
}; |
158 |
|
|
if ( rig < brig->At(0) ){ |
159 |
|
|
printf(" WARNING: Event with rigidity lower than the first covariance matrix bin (rig = %f, lower limit = %f)\n",rig,brig->At(0)); |
160 |
|
|
name = "matrixn0"; |
161 |
|
|
printf(" Using matrix %s \n",name.Data()); |
162 |
|
|
}; |
163 |
|
|
if ( rig >= brig->At(nbin) ){ |
164 |
|
|
printf(" WARNING: Event with rigidity higher than the last covariance matrix bin (rig = %f, upper limit = %f)\n",rig,brig->At(nbin)); |
165 |
|
|
name = Form("matrixn%i",nbin-1); |
166 |
|
|
printf(" Using matrix %s \n",name.Data()); |
167 |
|
|
}; |
168 |
|
|
// |
169 |
|
|
TMatrixD *matrix = (TMatrixD*)file->Get(name.Data()); |
170 |
|
|
// |
171 |
|
|
return(matrix); |
172 |
|
|
// |
173 |
|
|
} |
174 |
|
|
|
175 |
|
|
|
176 |
|
|
TArrayF *CaloFranzini::LoadLongAverage(Float_t rig){ |
177 |
|
|
// |
178 |
|
|
TString name; |
179 |
|
|
for (Int_t i = 0; i<nbin-1; i++){ |
180 |
|
|
if ( rig>=brig->At(i) && rig < brig->At(i+1) ){ |
181 |
|
|
name = Form("qplmeann%i",i); |
182 |
|
|
break; |
183 |
|
|
}; |
184 |
|
|
}; |
185 |
|
|
if ( rig < brig->At(0) ){ |
186 |
|
|
printf(" WARNING: Event with rigidity lower than the first qplmean bin (rig = %f, lower limit = %f)\n",rig,brig->At(0)); |
187 |
|
|
name = "qplmeann0"; |
188 |
|
|
printf(" Using qplmean %s \n",name.Data()); |
189 |
|
|
}; |
190 |
|
|
if ( rig >= brig->At(nbin) ){ |
191 |
|
|
printf(" WARNING: Event with rigidity higher than the last qplmean bin (rig = %f, upper limit = %f)\n",rig,brig->At(nbin)); |
192 |
|
|
name = Form("qplmeann%i",nbin-1); |
193 |
|
|
printf(" Using qplmean %s \n",name.Data()); |
194 |
|
|
}; |
195 |
|
|
// |
196 |
|
|
TArrayF *qplmean = (TArrayF*)file->Get(name.Data()); |
197 |
|
|
// |
198 |
|
|
return(qplmean); |
199 |
|
|
// |
200 |
|
|
} |