]>
Commit | Line | Data |
---|---|---|
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 | ||
18 | // Objects of this class contain basis for absolute calibrations | |
19 | // | |
20 | ||
21 | #include <fstream> | |
22 | #include <TString.h> | |
23 | #include <TFile.h> | |
24 | #include <TTree.h> | |
25 | ||
26 | #include "AliEMCALCalibAbs.h" | |
27 | ||
28 | using namespace std; | |
29 | ||
30 | ClassImp(AliEMCALCalibAbs) | |
31 | ||
32 | //____________________________________________________________________________ | |
33 | AliEMCALCalibAbs::AliEMCALCalibAbs(const int nSM) : | |
34 | fNSuperModule(nSM), | |
35 | fSuperModuleData() | |
36 | { | |
37 | //Default constructor. | |
38 | for (int i=0; i<fNSuperModule; i++) { | |
39 | fSuperModuleData.Add(new AliEMCALSuperModuleCalibAbs(i)); | |
40 | } | |
41 | fSuperModuleData.Compress(); // compress the TObjArray | |
42 | fSuperModuleData.SetOwner(kTRUE); | |
43 | } | |
44 | ||
45 | //____________________________________________________________________________ | |
46 | void AliEMCALCalibAbs::ReadTextCalibAbsInfo(Int_t nSM, const TString &txtFileName, | |
47 | Bool_t swapSides) | |
48 | { | |
49 | //Read data from txt file. ; coordinates given on SuperModule basis | |
50 | ||
51 | std::ifstream inputFile(txtFileName.Data()); | |
52 | if (!inputFile) { | |
53 | printf("AliEMCALCalibAbs::ReadCalibAbsInfo - Cannot open the APD info file %s\n", txtFileName.Data()); | |
54 | return; | |
55 | } | |
56 | ||
57 | fNSuperModule = nSM; | |
58 | ||
59 | Int_t iSM = 0; // SuperModule index | |
60 | Int_t iCol = 0; | |
61 | Int_t iRow = 0; | |
62 | ||
63 | // list of values to be read | |
64 | // first: overall values for the whole SuperModule | |
65 | Int_t iCalibMethod = 0; | |
66 | Int_t iCalibPass = 0; | |
67 | Float_t absoluteCalib = 0; | |
68 | // third: info for each tower | |
69 | Float_t relativeCalib = 0; // (ADC>GeV relative gain/conversion), value around 1 | |
70 | // end - all values | |
71 | ||
72 | Int_t nAPDPerSM = AliEMCALGeoParams::fgkEMCALCols * AliEMCALGeoParams::fgkEMCALRows; | |
73 | ||
74 | for (Int_t i = 0; i < fNSuperModule; i++) { | |
75 | AliEMCALSuperModuleCalibAbs * t = (AliEMCALSuperModuleCalibAbs*) fSuperModuleData[i]; | |
76 | if (!inputFile) { | |
77 | printf("AliEMCALCalibAbs::ReadCalibAbsInfo - Error while reading input file; likely EOF..\n"); | |
78 | return; | |
79 | } | |
80 | inputFile >> iSM; | |
81 | t->SetSuperModuleNum(iSM); | |
82 | ||
83 | // first: overall values for the whole SuperModule | |
84 | inputFile >> iCalibMethod >> iCalibPass >> absoluteCalib; | |
85 | t->SetCalibMethod(iCalibMethod); | |
86 | t->SetCalibPass(iCalibPass); | |
87 | t->SetAbsoluteCalib(absoluteCalib); | |
88 | ||
89 | // third: info for each tower | |
90 | for (Int_t j=0; j<nAPDPerSM; j++) { | |
91 | inputFile >> iCol >> iRow >> relativeCalib; | |
92 | ||
93 | // check that input values are not out bounds | |
94 | if (iCol<0 || iCol>(AliEMCALGeoParams::fgkEMCALCols-1) || | |
95 | iRow<0 || iRow>(AliEMCALGeoParams::fgkEMCALRows-1) ) { | |
96 | printf("AliEMCALCalibAbs::ReadCalibAbsInfo - Error while reading input file; j %d iCol %d iRow %d\n", j, iCol, iRow); | |
97 | return; | |
98 | } | |
99 | ||
100 | // assume that this info is already swapped and done for this basis? | |
101 | if (swapSides) { | |
102 | // C side, oriented differently than A side: swap is requested | |
103 | iCol = AliEMCALGeoParams::fgkEMCALCols-1 - iCol; | |
104 | iRow = AliEMCALGeoParams::fgkEMCALRows-1 - iRow; | |
105 | } | |
106 | ||
107 | t->SetRelativeCalib(iCol, iRow, relativeCalib); | |
108 | } | |
109 | ||
110 | } // i, SuperModule | |
111 | ||
112 | inputFile.close(); | |
113 | ||
114 | return; | |
115 | } | |
116 | ||
117 | //____________________________________________________________________________ | |
118 | void AliEMCALCalibAbs::WriteTextCalibAbsInfo(const TString &txtFileName, | |
119 | Bool_t swapSides) | |
120 | { | |
121 | // write data to txt file. ; coordinates given on SuperModule basis | |
122 | ||
123 | std::ofstream outputFile(txtFileName.Data()); | |
124 | if (!outputFile) { | |
125 | printf("AliEMCALCalibAbs::WriteCalibAbsInfo - Cannot open the APD output file %s\n", txtFileName.Data()); | |
126 | return; | |
127 | } | |
128 | ||
129 | Int_t iCol = 0; | |
130 | Int_t iRow = 0; | |
131 | ||
132 | Int_t nAPDPerSM = AliEMCALGeoParams::fgkEMCALCols * AliEMCALGeoParams::fgkEMCALRows; | |
133 | Float_t relativeCalib = 0; | |
134 | for (Int_t i = 0; i < fNSuperModule; i++) { | |
135 | AliEMCALSuperModuleCalibAbs * t = (AliEMCALSuperModuleCalibAbs*) fSuperModuleData[i]; | |
136 | ||
137 | // first: overall values for the whole SuperModule | |
138 | outputFile << t->GetSuperModuleNum() << endl; | |
139 | outputFile << t->GetCalibMethod() << " " | |
140 | << t->GetCalibPass() << " " | |
141 | << t->GetAbsoluteCalib() << endl; | |
142 | ||
143 | // third: info for each tower | |
144 | for (Int_t j=0; j<nAPDPerSM; j++) { | |
145 | iCol = j / AliEMCALGeoParams::fgkEMCALRows; | |
146 | iRow = j % AliEMCALGeoParams::fgkEMCALRows; | |
147 | ||
148 | relativeCalib = t->GetRelativeCalib(iCol, iRow); | |
149 | ||
150 | if (swapSides) { | |
151 | // C side, oriented differently than A side: swap is requested | |
152 | iCol = AliEMCALGeoParams::fgkEMCALCols-1 - iCol; | |
153 | iRow = AliEMCALGeoParams::fgkEMCALRows-1 - iRow; | |
154 | } | |
155 | ||
156 | outputFile << iCol << " " << iRow | |
157 | << " " << relativeCalib << endl; | |
158 | } | |
159 | ||
160 | } // i, SuperModule | |
161 | ||
162 | outputFile.close(); | |
163 | ||
164 | return; | |
165 | } | |
166 | ||
167 | //____________________________________________________________________________ | |
168 | void AliEMCALCalibAbs::ReadRootCalibAbsInfo(const TString &rootFileName, | |
169 | Bool_t swapSides) | |
170 | { | |
171 | //Read data from root file. ; coordinates given on SuperModule basis | |
172 | TFile inputFile(rootFileName, "read"); | |
173 | ||
174 | TTree *tree = (TTree*) inputFile.Get("tree"); | |
175 | ||
176 | ReadTreeCalibAbsInfo(tree, swapSides); | |
177 | ||
178 | inputFile.Close(); | |
179 | ||
180 | return; | |
181 | } | |
182 | ||
183 | //____________________________________________________________________________ | |
184 | void AliEMCALCalibAbs::ReadTreeCalibAbsInfo(TTree *tree, | |
185 | Bool_t swapSides) | |
186 | { | |
187 | // how many SuperModule's worth of info do we have? | |
188 | Int_t nAPDPerSM = AliEMCALGeoParams::fgkEMCALCols * AliEMCALGeoParams::fgkEMCALRows; | |
189 | fNSuperModule = tree->GetEntries(); | |
190 | ||
191 | Int_t iSM = 0; // SuperModule index | |
192 | // list of values to be read | |
193 | // first: overall values for the whole SuperModule | |
194 | Int_t iCalibMethod = 0; | |
195 | Int_t iCalibPass = 0; | |
196 | Float_t absoluteCalib = 0; | |
197 | // third: info for each tower | |
198 | Float_t relativeCalib[AliEMCALGeoParams::fgkEMCALCols][AliEMCALGeoParams::fgkEMCALRows]; | |
199 | // end - all values | |
200 | ||
201 | // just to make the initializations of the arrays are done correctly, let's use memset | |
202 | memset(relativeCalib, 0, sizeof(relativeCalib)); | |
203 | ||
204 | // declare the branches | |
205 | tree->SetBranchAddress("iSM", &iSM); | |
206 | tree->SetBranchAddress("CalibMethod", &iCalibMethod); | |
207 | tree->SetBranchAddress("CalibPass", &iCalibPass); | |
208 | tree->SetBranchAddress("AbsoluteCalib", &absoluteCalib); | |
209 | // | |
210 | tree->SetBranchAddress("RelativeCalib", relativeCalib); | |
211 | ||
212 | // indices for looping over the towers | |
213 | Int_t iCol = 0; | |
214 | Int_t iRow = 0; | |
215 | ||
216 | for (int ient=0; ient<tree->GetEntries(); ient++) { | |
217 | tree->GetEntry(ient); | |
218 | ||
219 | // assume the index SuperModules come in order: i=iSM | |
220 | AliEMCALSuperModuleCalibAbs * t = (AliEMCALSuperModuleCalibAbs*) fSuperModuleData[iSM]; | |
221 | ||
222 | t->SetSuperModuleNum(iSM); | |
223 | // first, overall values | |
224 | t->SetCalibMethod(iCalibMethod); | |
225 | t->SetCalibPass(iCalibPass); | |
226 | t->SetAbsoluteCalib(absoluteCalib); | |
227 | ||
228 | // third: info for each tower | |
229 | for (Int_t j=0; j<nAPDPerSM; j++) { | |
230 | iCol = j / AliEMCALGeoParams::fgkEMCALRows; | |
231 | iRow = j % AliEMCALGeoParams::fgkEMCALRows; | |
232 | ||
233 | // help variables: possibly modified or swapped indices | |
234 | int iColMod = iCol; | |
235 | int iRowMod = iRow; | |
236 | // assume that this info is already swapped and done for this basis? | |
237 | if (swapSides) { | |
238 | // C side, oriented differently than A side: swap is requested | |
239 | iColMod = AliEMCALGeoParams::fgkEMCALCols-1 - iCol; | |
240 | iRowMod = AliEMCALGeoParams::fgkEMCALRows-1 - iRow; | |
241 | } | |
242 | ||
243 | t->SetRelativeCalib(iColMod, iRowMod, relativeCalib[iCol][iRow]); | |
244 | } | |
245 | ||
246 | } // loop over entries | |
247 | ||
248 | return; | |
249 | } | |
250 | ||
251 | //____________________________________________________________________________ | |
252 | void AliEMCALCalibAbs::WriteRootCalibAbsInfo(const TString &rootFileName, | |
253 | Bool_t swapSides) | |
254 | { | |
255 | // write data to root file. ; coordinates given on SuperModule basis | |
256 | TFile destFile(rootFileName, "recreate"); | |
257 | if (destFile.IsZombie()) { | |
258 | return; | |
259 | } | |
260 | destFile.cd(); | |
261 | ||
262 | TTree *tree = new TTree("tree",""); | |
263 | ||
264 | // variables for filling the TTree | |
265 | Int_t iSM = 0; // SuperModule index | |
266 | // list of values to be written | |
267 | // first: overall values for the whole SuperModule | |
268 | Int_t iCalibMethod = 0; | |
269 | Int_t iCalibPass = 0; | |
270 | Float_t absoluteCalib = 0; | |
271 | // third: info for each tower | |
272 | Float_t relativeCalib[AliEMCALGeoParams::fgkEMCALCols][AliEMCALGeoParams::fgkEMCALRows]; | |
273 | // end - all values | |
274 | ||
275 | // just to make the initializations of the arrays are done correctly, let's use memset | |
276 | memset(relativeCalib, 0, sizeof(relativeCalib)); | |
277 | ||
278 | Int_t nAPDPerSM = AliEMCALGeoParams::fgkEMCALCols * AliEMCALGeoParams::fgkEMCALRows; | |
279 | // for looping over towers | |
280 | Int_t iCol = 0; | |
281 | Int_t iRow = 0; | |
282 | ||
283 | // declare the branches | |
284 | // first | |
285 | tree->Branch("iSM", &iSM, "iSM/I"); | |
286 | tree->Branch("CalibMethod", &iCalibMethod, "CalibMethod/I"); | |
287 | tree->Branch("CalibPass", &iCalibPass, "CalibPass/I"); | |
288 | tree->Branch("AbsoluteCalib", &absoluteCalib, "AbsoluteCalib/F"); | |
289 | // third: info for each tower; see if a 2D array works OK or if we'll have to use 1D arrays instead | |
290 | tree->Branch( "RelativeCalib", &relativeCalib, Form("RelativeCalib[%d][%d]/F", AliEMCALGeoParams::fgkEMCALCols, AliEMCALGeoParams::fgkEMCALRows) ); | |
291 | ||
292 | for (iSM = 0; iSM < fNSuperModule; iSM++) { | |
293 | AliEMCALSuperModuleCalibAbs * t = (AliEMCALSuperModuleCalibAbs*) fSuperModuleData[iSM]; | |
294 | ||
295 | iSM = t->GetSuperModuleNum(); | |
296 | // first, overall values | |
297 | iCalibMethod = t->GetCalibMethod(); | |
298 | iCalibPass = t->GetCalibPass(); | |
299 | absoluteCalib = t->GetAbsoluteCalib(); | |
300 | ||
301 | // third: info for each tower | |
302 | for (Int_t j=0; j<nAPDPerSM; j++) { | |
303 | iCol = j / AliEMCALGeoParams::fgkEMCALRows; | |
304 | iRow = j % AliEMCALGeoParams::fgkEMCALRows; | |
305 | ||
306 | // help variables: possibly modified or swapped indices | |
307 | int iColMod = iCol; | |
308 | int iRowMod = iRow; | |
309 | // assume that this info is already swapped and done for this basis? | |
310 | if (swapSides) { | |
311 | // C side, oriented differently than A side: swap is requested | |
312 | iColMod = AliEMCALGeoParams::fgkEMCALCols-1 - iCol; | |
313 | iRowMod = AliEMCALGeoParams::fgkEMCALRows-1 - iRow; | |
314 | } | |
315 | ||
316 | relativeCalib[iColMod][iRowMod] = t->GetRelativeCalib(iCol, iRow); | |
317 | } | |
318 | ||
319 | tree->Fill(); | |
320 | } // i, SuperModule | |
321 | ||
322 | tree->Write(); | |
323 | destFile.Close(); | |
324 | ||
325 | return; | |
326 | } | |
327 | ||
328 | //____________________________________________________________________________ | |
329 | AliEMCALCalibAbs::~AliEMCALCalibAbs() | |
330 | { | |
331 | fSuperModuleData.Delete(); | |
332 | } | |
333 | ||
334 | //____________________________________________________________________________ | |
335 | AliEMCALSuperModuleCalibAbs * AliEMCALCalibAbs::GetSuperModuleCalibAbsNum(Int_t supModIndex)const | |
336 | { // getter via index | |
337 | for (int i=0; i<fNSuperModule; i++) { | |
338 | AliEMCALSuperModuleCalibAbs * t = (AliEMCALSuperModuleCalibAbs*) fSuperModuleData[i]; | |
339 | if (t->GetSuperModuleNum() == supModIndex) { | |
340 | return t; | |
341 | } | |
342 | } | |
343 | ||
344 | // if we arrived here, then nothing was found.. just return a NULL pointer | |
345 | return NULL; | |
346 | } | |
347 |