]>
Commit | Line | Data |
---|---|---|
d81e6423 | 1 | /************************************************************************** |
2 | * Copyright(c) 1998-1999, 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 | ||
16 | /* $Id: $ */ | |
17 | ||
61917ab3 | 18 | // Objects of this class contain info on time-dependent corrections |
d81e6423 | 19 | // |
20 | ||
21 | #include <fstream> | |
22 | #include <TString.h> | |
61917ab3 | 23 | #include <TFile.h> |
24 | #include <TTree.h> | |
d81e6423 | 25 | |
26 | #include "AliEMCALCalibTimeDepCorrection.h" | |
27 | ||
28 | using namespace std; | |
29 | ||
30 | ClassImp(AliEMCALCalibTimeDepCorrection) | |
31 | ||
32 | //____________________________________________________________________________ | |
2f17a269 | 33 | AliEMCALCalibTimeDepCorrection::AliEMCALCalibTimeDepCorrection(const int nSM) : |
34 | fNSuperModule(nSM), | |
35 | fSuperModuleData(), | |
220ed45a | 36 | fStartTime(0), |
37 | fNTimeBins(0), | |
38 | fTimeBinSize(0) | |
d81e6423 | 39 | { |
40 | //Default constructor. | |
2f17a269 | 41 | for (int i=0; i<fNSuperModule; i++) { |
42 | fSuperModuleData.Add(new AliEMCALSuperModuleCalibTimeDepCorrection(i)); | |
43 | } | |
44 | fSuperModuleData.Compress(); // compress the TObjArray | |
d81e6423 | 45 | } |
46 | ||
47 | //____________________________________________________________________________ | |
716fca62 | 48 | void AliEMCALCalibTimeDepCorrection::InitCorrection(Int_t nSM, Int_t nBins, Float_t val) |
d81e6423 | 49 | { |
50 | // This methods assumes that you are using SuperModules 0..nSM-1 | |
51 | fNSuperModule = nSM; | |
d81e6423 | 52 | |
53 | Int_t iSM = 0; // SuperModule index | |
54 | Int_t iCol = 0; | |
55 | Int_t iRow = 0; | |
56 | Int_t nCorr = nBins; | |
ab962f7b | 57 | Float_t correction = val; |
d81e6423 | 58 | |
59 | Int_t nAPDPerSM = AliEMCALGeoParams::fgkEMCALCols * AliEMCALGeoParams::fgkEMCALRows; | |
60 | ||
61 | for (Int_t i = 0; i < fNSuperModule; i++) { | |
2f17a269 | 62 | AliEMCALSuperModuleCalibTimeDepCorrection * t = (AliEMCALSuperModuleCalibTimeDepCorrection*) fSuperModuleData[i]; |
716fca62 | 63 | iSM = i; |
2f17a269 | 64 | t->SetSuperModuleNum(iSM); // assume SMs are coming in order |
d81e6423 | 65 | |
66 | for (Int_t j=0; j<nAPDPerSM; j++) { | |
716fca62 | 67 | iCol = j / AliEMCALGeoParams::fgkEMCALRows; |
68 | iRow = j % AliEMCALGeoParams::fgkEMCALRows; | |
d81e6423 | 69 | |
70 | // set size of TArray | |
2f17a269 | 71 | t->GetCorrection(iCol,iRow)->Set(nCorr); |
d81e6423 | 72 | for (Int_t k=0; k<nCorr; k++) { |
73 | // add to TArray | |
ab962f7b | 74 | t->GetCorrection(iCol,iRow)->AddAt(correction, k); // AddAt = SetAt.. |
d81e6423 | 75 | } |
76 | } | |
77 | ||
78 | } // i, SuperModule | |
79 | ||
80 | return; | |
81 | } | |
82 | ||
83 | //____________________________________________________________________________ | |
84 | void AliEMCALCalibTimeDepCorrection::SetCorrection(Int_t supModIndex, Int_t iCol, Int_t iRow, Int_t iBin, Float_t val) | |
85 | { // if you call for non-existing data, there may be a crash.. | |
2f17a269 | 86 | ((AliEMCALSuperModuleCalibTimeDepCorrection*)fSuperModuleData[supModIndex])->GetCorrection(iCol,iRow)->AddAt(val, iBin); // AddAt = SetAt.. |
d81e6423 | 87 | return; |
88 | } | |
89 | ||
90 | //____________________________________________________________________________ | |
2f17a269 | 91 | Float_t AliEMCALCalibTimeDepCorrection::GetCorrection(Int_t supModIndex, Int_t iCol, Int_t iRow, Int_t iBin) const |
d81e6423 | 92 | { // if you call for non-existing data, there may be a crash.. |
0ce5c45d | 93 | return ((AliEMCALSuperModuleCalibTimeDepCorrection*)fSuperModuleData[supModIndex])->GetCorrection(iCol,iRow)->At(iBin); |
d81e6423 | 94 | } |
95 | ||
96 | //____________________________________________________________________________ | |
61917ab3 | 97 | void AliEMCALCalibTimeDepCorrection::ReadTextInfo(Int_t nSM, const TString &txtFileName, |
98 | Bool_t swapSides) | |
d81e6423 | 99 | { |
100 | //Read data from txt file. ; coordinates given on SuperModule basis | |
101 | ||
102 | std::ifstream inputFile(txtFileName.Data()); | |
103 | if (!inputFile) { | |
61917ab3 | 104 | printf("AliEMCALCalibTimeDepCorrection::ReadTextInfo - Cannot open the APD info file %s\n", txtFileName.Data()); |
d81e6423 | 105 | return; |
106 | } | |
107 | ||
108 | fNSuperModule = nSM; | |
d81e6423 | 109 | |
110 | Int_t iSM = 0; // SuperModule index | |
111 | Int_t iCol = 0; | |
112 | Int_t iRow = 0; | |
113 | Int_t nCorr = 0; | |
ab962f7b | 114 | Float_t correction = 0; |
d81e6423 | 115 | |
116 | Int_t nAPDPerSM = AliEMCALGeoParams::fgkEMCALCols * AliEMCALGeoParams::fgkEMCALRows; | |
117 | ||
61917ab3 | 118 | // read header info |
119 | inputFile >> fStartTime >> fNTimeBins >> fTimeBinSize; | |
120 | ||
d81e6423 | 121 | for (Int_t i = 0; i < fNSuperModule; i++) { |
2f17a269 | 122 | AliEMCALSuperModuleCalibTimeDepCorrection * t = (AliEMCALSuperModuleCalibTimeDepCorrection*) fSuperModuleData[i]; |
123 | ||
d81e6423 | 124 | if (!inputFile) { |
b48d1356 | 125 | printf("AliEMCALCalibTimeDepCorrection::ReadTextInfo - Error while reading input file; likely EOF..\n"); |
d81e6423 | 126 | return; |
127 | } | |
128 | inputFile >> iSM; | |
2f17a269 | 129 | t->SetSuperModuleNum(iSM); |
d81e6423 | 130 | |
131 | for (Int_t j=0; j<nAPDPerSM; j++) { | |
132 | inputFile >> iCol >> iRow >> nCorr; | |
133 | ||
b48d1356 | 134 | // check that input values are not out bounds |
135 | if (iCol<0 || iCol>(AliEMCALGeoParams::fgkEMCALCols-1) || | |
136 | iRow<0 || iRow>(AliEMCALGeoParams::fgkEMCALRows-1) ) { | |
137 | printf("AliEMCALCalibTimeDepCorrection::ReadTextInfo - Error while reading input file; j %d iCol %d iRow %d\n", j, iCol, iRow); | |
138 | return; | |
139 | } | |
140 | ||
d81e6423 | 141 | // assume that this info is already swapped and done for this basis? |
142 | if (swapSides) { | |
143 | // C side, oriented differently than A side: swap is requested | |
144 | iCol = AliEMCALGeoParams::fgkEMCALCols-1 - iCol; | |
145 | iRow = AliEMCALGeoParams::fgkEMCALRows-1 - iRow; | |
146 | } | |
147 | ||
148 | // set size of TArray | |
2f17a269 | 149 | t->GetCorrection(iCol,iRow)->Set(nCorr); |
d81e6423 | 150 | for (Int_t k=0; k<nCorr; k++) { |
ab962f7b | 151 | inputFile >> correction; |
d81e6423 | 152 | // add to TArray |
ab962f7b | 153 | t->GetCorrection(iCol,iRow)->AddAt(correction, k); |
d81e6423 | 154 | } |
155 | } | |
156 | ||
157 | } // i, SuperModule | |
158 | ||
159 | inputFile.close(); | |
160 | ||
161 | return; | |
162 | } | |
163 | ||
164 | //____________________________________________________________________________ | |
61917ab3 | 165 | void AliEMCALCalibTimeDepCorrection::WriteTextInfo(const TString &txtFileName, |
166 | Bool_t swapSides) | |
d81e6423 | 167 | { |
168 | // write data to txt file. ; coordinates given on SuperModule basis | |
169 | ||
170 | std::ofstream outputFile(txtFileName.Data()); | |
171 | if (!outputFile) { | |
61917ab3 | 172 | printf("AliEMCALCalibTimeDepCorrection::WriteTextInfo - Cannot open the APD output file %s\n", txtFileName.Data()); |
d81e6423 | 173 | return; |
174 | } | |
175 | ||
176 | Int_t iCol = 0; | |
177 | Int_t iRow = 0; | |
178 | Int_t nCorr = 0; | |
179 | ||
180 | Int_t nAPDPerSM = AliEMCALGeoParams::fgkEMCALCols * AliEMCALGeoParams::fgkEMCALRows; | |
181 | ||
61917ab3 | 182 | // write header info |
183 | outputFile << fStartTime << " " << fNTimeBins << " " << fTimeBinSize << endl; | |
184 | ||
d81e6423 | 185 | for (Int_t i = 0; i < fNSuperModule; i++) { |
2f17a269 | 186 | AliEMCALSuperModuleCalibTimeDepCorrection * t = (AliEMCALSuperModuleCalibTimeDepCorrection*) fSuperModuleData[i]; |
187 | outputFile << t->GetSuperModuleNum() << endl; | |
d81e6423 | 188 | |
189 | for (Int_t j=0; j<nAPDPerSM; j++) { | |
190 | iCol = j / AliEMCALGeoParams::fgkEMCALRows; | |
191 | iRow = j % AliEMCALGeoParams::fgkEMCALRows; | |
192 | ||
2f17a269 | 193 | nCorr = t->GetCorrection(iCol,iRow)->GetSize(); |
d81e6423 | 194 | |
195 | if (swapSides) { | |
196 | // C side, oriented differently than A side: swap is requested | |
197 | iCol = AliEMCALGeoParams::fgkEMCALCols-1 - iCol; | |
198 | iRow = AliEMCALGeoParams::fgkEMCALRows-1 - iRow; | |
199 | } | |
200 | ||
201 | outputFile << iCol << " " << iRow << " " << nCorr << endl; | |
202 | for (Int_t k=0; k<nCorr; k++) { | |
2f17a269 | 203 | outputFile << t->GetCorrection(iCol,iRow)->At(k) << " "; |
d81e6423 | 204 | } |
205 | outputFile << endl; | |
206 | ||
207 | } | |
208 | ||
209 | } // i, SuperModule | |
210 | ||
211 | outputFile.close(); | |
212 | ||
213 | return; | |
214 | } | |
215 | ||
61917ab3 | 216 | //____________________________________________________________________________ |
217 | void AliEMCALCalibTimeDepCorrection::ReadRootInfo(const TString &rootFileName, | |
218 | Bool_t swapSides) | |
219 | { | |
220 | //Read data from root file. ; coordinates given on SuperModule basis | |
221 | TFile inputFile(rootFileName, "read"); | |
222 | ||
223 | TTree *treeGlob = (TTree*) inputFile.Get("treeGlob"); | |
224 | TTree *treeCorr = (TTree*) inputFile.Get("treeCorr"); | |
225 | ||
226 | ReadTreeInfo(treeGlob, treeCorr, swapSides); | |
227 | ||
228 | inputFile.Close(); | |
229 | ||
230 | return; | |
231 | } | |
232 | ||
233 | //____________________________________________________________________________ | |
234 | void AliEMCALCalibTimeDepCorrection::ReadTreeInfo(TTree *treeGlob, TTree *treeCorr, | |
235 | Bool_t swapSides) | |
236 | { | |
237 | // how many SuperModule's worth of entries / APDs do we have? | |
238 | Int_t nAPDPerSM = AliEMCALGeoParams::fgkEMCALCols * AliEMCALGeoParams::fgkEMCALRows; | |
239 | fNSuperModule = treeCorr->GetEntries() / nAPDPerSM; | |
240 | ||
241 | // global variables : only one entry | |
242 | treeGlob->SetBranchAddress("fStartTime", &fStartTime); | |
243 | treeGlob->SetBranchAddress("fNTimeBins", &fNTimeBins); | |
244 | treeGlob->SetBranchAddress("fTimeBinSize", &fTimeBinSize); | |
245 | treeGlob->GetEntry(0); | |
246 | ||
61917ab3 | 247 | Int_t iSM = 0; // SuperModule index |
248 | Int_t iCol = 0; | |
249 | Int_t iRow = 0; | |
250 | // list of values to be read | |
251 | Int_t nCorr = 0; | |
252 | Float_t correction[fgkMaxTimeBins] = {0}; | |
253 | // make sure it's really initialized correctly | |
254 | memset(correction, 0, sizeof(correction)); // better safe than sorry | |
255 | // end - all values | |
256 | ||
257 | // declare the branches | |
258 | treeCorr->SetBranchAddress("iSM", &iSM); | |
259 | treeCorr->SetBranchAddress("iCol", &iCol); | |
260 | treeCorr->SetBranchAddress("iRow", &iRow); | |
261 | treeCorr->SetBranchAddress("nCorr", &nCorr); | |
262 | treeCorr->SetBranchAddress("correction", correction); | |
263 | ||
264 | for (int ient=0; ient<treeCorr->GetEntries(); ient++) { | |
265 | treeCorr->GetEntry(ient); | |
266 | ||
267 | // assume the index SuperModules come in order: i=iSM | |
2f17a269 | 268 | AliEMCALSuperModuleCalibTimeDepCorrection * t = (AliEMCALSuperModuleCalibTimeDepCorrection*) fSuperModuleData[iSM]; |
269 | t->SetSuperModuleNum(iSM); | |
61917ab3 | 270 | |
271 | // assume that this info is already swapped and done for this basis? | |
272 | if (swapSides) { | |
273 | // C side, oriented differently than A side: swap is requested | |
274 | iCol = AliEMCALGeoParams::fgkEMCALCols-1 - iCol; | |
275 | iRow = AliEMCALGeoParams::fgkEMCALRows-1 - iRow; | |
276 | } | |
277 | ||
278 | ||
279 | // set size of TArray | |
2f17a269 | 280 | t->GetCorrection(iCol,iRow)->Set(nCorr); |
61917ab3 | 281 | for (Int_t k=0; k<nCorr; k++) { |
282 | // add to TArray | |
2f17a269 | 283 | t->GetCorrection(iCol,iRow)->AddAt(correction[k], k); |
61917ab3 | 284 | } |
285 | ||
286 | } // entry | |
287 | ||
288 | return; | |
289 | } | |
290 | ||
291 | //____________________________________________________________________________ | |
292 | void AliEMCALCalibTimeDepCorrection::WriteRootInfo(const TString &rootFileName, | |
293 | Bool_t swapSides) | |
294 | { | |
295 | // write data to root file. ; coordinates given on SuperModule basis | |
296 | TFile destFile(rootFileName, "recreate"); | |
297 | if (destFile.IsZombie()) { | |
298 | return; | |
299 | } | |
300 | destFile.cd(); | |
301 | ||
302 | TTree *treeGlob = new TTree("treeGlob",""); | |
303 | TTree *treeCorr = new TTree("treeCorr",""); | |
304 | ||
305 | // global part only has one entry | |
306 | treeGlob->Branch("fStartTime", &fStartTime, "fStartTime/I"); // really unsigned int.. | |
307 | treeGlob->Branch("fNTimeBins", &fNTimeBins, "fNTimeBins/I"); | |
308 | treeGlob->Branch("fTimeBinSize", &fTimeBinSize, "fTimeBinSize/I"); | |
309 | treeGlob->Fill(); | |
310 | ||
311 | // variables for filling the TTree | |
312 | Int_t iSM = 0; // SuperModule index | |
313 | Int_t iCol = 0; | |
314 | Int_t iRow = 0; | |
315 | Int_t nCorr = 0; | |
316 | Float_t correction[fgkMaxTimeBins] = {0}; | |
317 | // make sure it's really initialized correctly | |
318 | memset(correction, 0, sizeof(correction)); // better safe than sorry | |
319 | ||
320 | // declare the branches | |
321 | treeCorr->Branch("iSM", &iSM, "iSM/I"); | |
322 | treeCorr->Branch("iCol", &iCol, "iCol/I"); | |
323 | treeCorr->Branch("iRow", &iRow, "iRow/I"); | |
324 | treeCorr->Branch("nCorr", &nCorr, "nCorr/I"); | |
325 | treeCorr->Branch("correction", &correction, "correction[nCorr]/F"); | |
326 | ||
327 | Int_t nAPDPerSM = AliEMCALGeoParams::fgkEMCALCols * AliEMCALGeoParams::fgkEMCALRows; | |
328 | ||
329 | for (iSM = 0; iSM < fNSuperModule; iSM++) { | |
2f17a269 | 330 | AliEMCALSuperModuleCalibTimeDepCorrection * t = (AliEMCALSuperModuleCalibTimeDepCorrection*) fSuperModuleData[iSM]; |
61917ab3 | 331 | |
332 | for (Int_t j=0; j<nAPDPerSM; j++) { | |
333 | iCol = j / AliEMCALGeoParams::fgkEMCALRows; | |
334 | iRow = j % AliEMCALGeoParams::fgkEMCALRows; | |
335 | ||
2f17a269 | 336 | nCorr = t->GetCorrection(iCol,iRow)->GetSize(); |
61917ab3 | 337 | if (nCorr > fgkMaxTimeBins) { |
338 | printf("AliEMCALCalibTimeDepCorrection::WriteRootInfo - too many correction/timebins %d kept\n", nCorr); | |
339 | return; | |
340 | } | |
341 | ||
342 | if (swapSides) { | |
343 | // C side, oriented differently than A side: swap is requested | |
344 | iCol = AliEMCALGeoParams::fgkEMCALCols-1 - iCol; | |
345 | iRow = AliEMCALGeoParams::fgkEMCALRows-1 - iRow; | |
346 | } | |
347 | ||
348 | for (Int_t k=0; k<nCorr; k++) { | |
2f17a269 | 349 | correction[k] = t->GetCorrection(iCol,iRow)->At(k); |
61917ab3 | 350 | } |
351 | ||
352 | treeCorr->Fill(); | |
353 | } | |
354 | ||
355 | } // i, SuperModule | |
356 | ||
357 | treeGlob->Write(); | |
358 | treeCorr->Write(); | |
359 | destFile.Close(); | |
360 | ||
361 | return; | |
362 | } | |
363 | ||
d81e6423 | 364 | //____________________________________________________________________________ |
365 | AliEMCALCalibTimeDepCorrection::~AliEMCALCalibTimeDepCorrection() | |
366 | { | |
2f17a269 | 367 | fSuperModuleData.Delete(); |
d81e6423 | 368 | } |
369 | ||
370 | //____________________________________________________________________________ | |
2f17a269 | 371 | AliEMCALSuperModuleCalibTimeDepCorrection * AliEMCALCalibTimeDepCorrection::GetSuperModuleCalibTimeDepCorrectionNum(Int_t supModIndex)const |
d81e6423 | 372 | { |
d81e6423 | 373 | for (int i=0; i<fNSuperModule; i++) { |
2f17a269 | 374 | AliEMCALSuperModuleCalibTimeDepCorrection * t = (AliEMCALSuperModuleCalibTimeDepCorrection*) fSuperModuleData[i]; |
375 | if (t->GetSuperModuleNum() == supModIndex) { | |
376 | return t; | |
d81e6423 | 377 | } |
378 | } | |
379 | ||
2f17a269 | 380 | // if we arrived here, then nothing was found.. just return a NULL pointer |
381 | return NULL; | |
d81e6423 | 382 | } |
383 |