16 {
17
18 if (inTotFile.empty()) return false;
19
20 TFile totFile(inTotFile.c_str(),"READ");
21 if (not totFile.IsOpen()) {
22 printf("Error - File %s could not be opened.\n",inTotFile.c_str());
23 totFile.Close();
24 return false;
25 } else {
26 printf("File %s opened.\n",inTotFile.c_str());
27 printf("Running TOT calibration...\n");
28 }
29
30
31 std::vector< std::vector< std::unique_ptr<TH1F> > > histogramsTOT;
32 std::vector< std::vector< std::unique_ptr<TH1F> > > histogramsTOTsig;
33 for(
unsigned int FE = 0; FE <
m_nFE; FE++){
34
35 histogramsTOT.push_back( std::vector< std::unique_ptr<TH1F> >() );
36 histogramsTOTsig.push_back( std::vector< std::unique_ptr<TH1F> >() );
37
38
40 std::string
title =
"FE"+std::to_string(FE)+
"_pixType"+std::to_string(
pixel);
41 std::unique_ptr<TH1F> htot = std::make_unique<TH1F>((title+
"_tot").c_str(), (title+
"_tot").c_str(),
m_totnbins,
m_totLo,
m_totHi);
42 htot->SetDirectory(0);
44 htotsig->SetDirectory(0);
45
46 histogramsTOT.at(FE).push_back(std::move(htot));
47
48 histogramsTOTsig.at(FE).push_back(std::move(htotsig));
49 }
50 }
51
52
54 TKey* rodKey;
55
56 while ((rodKey=static_cast<TKey*>(rodItr()))) {
57 TDirectoryFile* rodDir = static_cast<TDirectoryFile*>(rodKey->ReadObj());
59 const std::string rodName(rodKey->GetName());
60 printf("%s\n",rodName.c_str());
61 TKey* modKey;
62 while ((modKey=static_cast<TKey*>(modItr()))) {
63 const std::string
modName(modKey->GetName());
64
66 if (not pm.
contains(modName))
continue;
67
69 continue;
70 }
71
72 printf(
" -> %s\n",
modName.c_str());
73
74
75 std::array<std::array<float, m_ncharge>,
m_nFE> totArrI{};
76 std::array<std::array<float, m_ncharge>,
m_nFE> totErrArrI{};
77 std::array<std::array<float, m_ncharge>,
m_nFE> totSigArrI{};
78 std::array<std::array<float, m_ncharge>,
m_nFE> totSigErrArrI{};
79 std::array<std::array<float, m_ncharge>,
m_nFE> totLongArrI{};
80 std::array<std::array<float, m_ncharge>,
m_nFE> totErrLongArrI{};
81
82
84
85
88 if(!h2dTOTmean or !h2dTOTsig) {
89 return false;
90 }
91 h2dTOTmean->SetDirectory(0);
92 h2dTOTsig->SetDirectory(0);
93
94 for (
unsigned int ieta = 0; ieta <
m_etaBins; ieta++) {
95 for (
unsigned int iphi = 0; iphi <
m_phiBins; iphi++) {
96 float totmean = h2dTOTmean->GetBinContent(ieta + 1, iphi + 1);
97 float totsig = h2dTOTsig ->GetBinContent(ieta + 1, iphi + 1);
98
99 if (totmean<0.1) {
100 continue;
101 }
102
103 int FE =
chipId(iphi, ieta);
105
106 if(FE<0){
107 return false;
108 }
109
110 histogramsTOT.at(FE).at(
pixel)->Fill(totmean);
111 histogramsTOTsig.at(FE).at(
pixel)->Fill(totsig);
112
113 }
114 }
115
116
117 h2dTOTmean.reset();
118 h2dTOTsig.reset();
119
120
121 for(
unsigned int FE = 0; FE <
m_nFE; FE++){
123
125 totArrI.at(FE).at(c) = histogramsTOT.at(FE).at(
pixel)->GetMean();
126 totErrArrI.at(FE).at(c) = histogramsTOT.at(FE).at(
pixel)->GetMeanError();
127 totSigArrI.at(FE).at(c) = std::sqrt(std::pow(histogramsTOTsig.at(FE).at(
pixel)->GetMean() ,2)+std::pow(histogramsTOT.at(FE).at(
pixel)->GetRMS() ,2));
128 totSigErrArrI.at(FE).at(c) = std::sqrt(std::pow(histogramsTOTsig.at(FE).at(
pixel)->GetMeanError(),2)+std::pow(histogramsTOT.at(FE).at(
pixel)->GetRMSError(),2));
129
130 if(totSigErrArrI.at(FE).at(c) > 1.0){
131 totArrI.at(FE).at(c) = 0.0;
132 }
133
134 }
135 else{
136 totLongArrI.at(FE).at(c) = histogramsTOT.at(FE).at(
pixel)->GetMean();
137 totErrLongArrI.at(FE).at(c) = histogramsTOT.at(FE).at(
pixel)->GetMeanError();
138
139 if(totErrLongArrI.at(FE).at(c) > 1.0){
140 totLongArrI.at(FE).at(c) = 0.0;
141 }
142
143 }
144
145
146 histogramsTOT.at(FE).at(
pixel)->Reset(
"ICESM");
147 histogramsTOTsig.at(FE).at(
pixel)->Reset(
"ICESM");
148 }
149 }
150 }
151
152
153 for(
unsigned int FE = 0; FE <
m_nFE; FE++) {
154
155 std::string
subdir(((FE < 10) ?
"FE0" :
"FE") +std::to_string(FE));
156
157 std::vector<float> v_Q;
158 std::vector<float> v_Qerr;
159 std::vector<float> v_TOT;
160 std::vector<float> v_TOTerr;
161 std::vector<float> v_TOTsig;
162 std::vector<float> v_TOTsigerr;
163 std::vector<float> v_TOTlong;
164 std::vector<float> v_TOTlongerr;
165
168 std::copy(std::begin(totArrI[FE]) , std::end(totArrI[FE]) , std::back_inserter(v_TOT) );
169 std::copy(std::begin(totErrArrI[FE]) , std::end(totErrArrI[FE]) , std::back_inserter(v_TOTerr) );
170 std::copy(std::begin(totSigArrI[FE]) , std::end(totSigArrI[FE]) , std::back_inserter(v_TOTsig) );
171 std::copy(std::begin(totSigErrArrI[FE]) , std::end(totSigErrArrI[FE]) , std::back_inserter(v_TOTsigerr) );
172 std::copy(std::begin(totLongArrI[FE]) , std::end(totLongArrI[FE]) , std::back_inserter(v_TOTlong) );
173 std::copy(std::begin(totErrLongArrI[FE]) , std::end(totErrLongArrI[FE]) , std::back_inserter(v_TOTlongerr) );
174
175 std::vector<float> pixNormalParams;
176 std::vector<float> pixNormalParamsQuality;
177 std::vector<float> pixSigParams;
178 std::vector<float> pixSigParamsQuality;
179 std::vector<float> pixLongParams;
180 std::vector<float> pixLongParamsQuality;
181
182
184 do{
185 int vecsize = v_Q.size();
186
187 std::unique_ptr<TGraphErrors> graphnormal = std::make_unique<TGraphErrors>(vecsize, &v_Q.at(0), &v_TOT.at(0) , &v_Qerr.at(0), &v_TOTerr.at(0) );
188 std::unique_ptr<TGraphErrors> graphsig = std::make_unique<TGraphErrors>(vecsize, &v_Q.at(0), &v_TOTsig.at(0) , &v_Qerr.at(0), &v_TOTsigerr.at(0) );
189
192
193 graphnormal->Fit(functnormal.get() ,"MRQ");
194 graphsig ->Fit(functnormalsig.get(),"MRQ");
195
196 pixNormalParams =
getParams(functnormal.get() ,3 );
197 pixSigParams =
getParams(functnormalsig.get(),2 );
198
201
203
205 if( !
m_wFile->Get((rodName+
"/"+modName+
"/TOTfits/"+subdir).c_str()) ){
206 m_wFile->mkdir((rodName+
"/"+modName+
"/TOTfits/"+subdir).c_str(),rodName.c_str());
207 }
208
209 m_wFile->cd((rodName+
"/"+modName+
"/TOTfits/"+subdir).c_str());
210
211 graphTitles(graphnormal, std::string(modName+
" - "+subdir+
" - normal pixels: Fit: "+std::to_string(n_fit)).c_str(),
"TOT");
212 graphTitles(graphsig , std::string(modName+
" - "+subdir+
" - normal pixels: Fit: "+std::to_string(n_fit)).c_str(),
"Charge smearing");
213
214 graphnormal->Write(std::string("normal_fit_"+std::to_string(n_fit)).c_str(), TObject::kWriteDelete);
215 graphsig->Write(std::string("smearing_fit_"+std::to_string(n_fit)).c_str(), TObject::kWriteDelete);
216 n_fit++;
217 }
218
219 functnormal.reset();
220 functnormalsig.reset();
221
222 graphnormal.reset();
223 graphsig.reset();
224
225 }
while(
reFit_normalPix(pixNormalParams, v_Q, v_Qerr, v_TOT, v_TOTerr, v_TOTsig, v_TOTsigerr, FE ) );
226
227
228
229 v_Q.clear();
230 v_Qerr.clear();
231
234
235
236
237 n_fit = 0;
238 do{
239
240
241 int vecsize = v_Q.size();
242
243 std::unique_ptr<TGraphErrors> graflong = std::make_unique<TGraphErrors>(vecsize, &v_Q.at(0), &v_TOTlong.at(0), &v_Qerr.at(0), &v_TOTlongerr.at(0) );
244
246
247 graflong ->Fit(functlong.get() ,"MRQ");
248
249 pixLongParams =
getParams(functlong.get() ,3 );
251
253
255 if( !
m_wFile->Get((rodName+
"/"+modName+
"/TOTfits/"+subdir).c_str()) ){
256 m_wFile->mkdir((rodName+
"/"+modName+
"/TOTfits/"+subdir).c_str(),rodName.c_str());
257 }
258
259 m_wFile->cd((rodName+
"/"+modName+
"/TOTfits/"+subdir).c_str());
260
261 graphTitles(graflong, std::string(modName+
" - "+subdir+
" - long+ganged pixels: Fit: "+std::to_string(n_fit)).c_str(),
"TOT");
262
263 graflong->Write(std::string("long_ganged_fit_"+std::to_string(n_fit)).c_str(), TObject::kWriteDelete);
264 n_fit++;
265 }
266
267
268 functlong.reset();
269
270
271 graflong.reset();
272
273
274 }while( false);
275
276
277
278 int modID = pm.
getID(modName);
279 auto itr = map_info.find( modID );
280
281 if (itr != map_info.end()) {
282 (itr->second).at(FE)->set_NormalParams( pixNormalParams);
283 (itr->second).at(FE)->set_LongParams ( pixLongParams );
284 (itr->second).at(FE)->set_SigParams ( pixSigParams );
285
286 (itr->second).at(FE)->set_times_fitted ( n_fit );
287
288 (itr->second).at(FE)->set_NormalParamsQuality( pixNormalParamsQuality);
289 (itr->second).at(FE)->set_LongParamsQuality ( pixLongParamsQuality );
290 (itr->second).at(FE)->set_SigParamsQuality ( pixSigParamsQuality );
291 }
292 else{
293 printf("Error - Module not found in fitting step... Skipping.\n");
294 return false;
295 }
296
297 }
298 }
299 }
300
301
302 for(
unsigned int FE = 0; FE <
m_nFE; FE++) {
304 histogramsTOT.at(FE).at(
pixel).reset();
305 histogramsTOTsig.at(FE).at(
pixel).reset();
306 }
307 }
308
309 totFile.Close();
310 return true;
311}
static constexpr std::array< float, m_ncharge > m_chargeArr
static constexpr int m_totnbins
std::vector< float > getParams(const TF1 *f, unsigned int params)
std::vector< float > getParams_quality(const TF1 *f)
bool reFit_normalPix(std::vector< float > ¶ms, std::vector< float > &q, std::vector< float > &qerr, std::vector< float > &tot, std::vector< float > &toterr, std::vector< float > &sig, std::vector< float > &sigerr, const unsigned int fe)
static constexpr float m_totLo
static constexpr float m_totsigHi
static constexpr float m_totsigLo
static constexpr int m_totsigNBins
static constexpr std::array< float, m_ncharge > m_chargeErrArr
void graphTitles(const std::unique_ptr< TGraphErrors > &graph, const std::string &name, const std::string &Yname)
static constexpr float m_totHi