]> git.uio.no Git - u/mrichter/AliRoot.git/blame - EMCAL/AliEMCALCalibTempCoeff.cxx
fill inverse pt weighted histograms only if requested by user
[u/mrichter/AliRoot.git] / EMCAL / AliEMCALCalibTempCoeff.cxx
CommitLineData
3a282863 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 temperature-dependence coefficients
19//
20
21#include <fstream>
22#include <TString.h>
23#include <TFile.h>
24#include <TTree.h>
25
26#include "AliEMCALCalibTempCoeff.h"
27
28using namespace std;
29
30ClassImp(AliEMCALCalibTempCoeff)
31
32//____________________________________________________________________________
33AliEMCALCalibTempCoeff::AliEMCALCalibTempCoeff(const int nSM) :
34 fNSuperModule(nSM),
35 fSuperModuleData()
36{
37 //Default constructor.
38 for (int i=0; i<fNSuperModule; i++) {
39 fSuperModuleData.Add(new AliEMCALSuperModuleCalibTempCoeff(i));
40 }
41 fSuperModuleData.Compress(); // compress the TObjArray
42 fSuperModuleData.SetOwner(kTRUE);
43}
44
45//____________________________________________________________________________
46void AliEMCALCalibTempCoeff::ReadTextCalibTempCoeffInfo(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("AliEMCALCalibTempCoeff::ReadCalibTempCoeffInfo - 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 Int_t iSrc = 0;
65 Float_t tempCoeff = 0;
66 // end - all values
67
68 Int_t nAPDPerSM = AliEMCALGeoParams::fgkEMCALCols * AliEMCALGeoParams::fgkEMCALRows;
69
70 for (Int_t i = 0; i < fNSuperModule; i++) {
71 AliEMCALSuperModuleCalibTempCoeff * t = (AliEMCALSuperModuleCalibTempCoeff*) fSuperModuleData[i];
72 if (!inputFile) {
73 printf("AliEMCALCalibTempCoeff::ReadCalibTempCoeffInfo - Error while reading input file; likely EOF..\n");
74 return;
75 }
76 inputFile >> iSM;
77 t->SetSuperModuleNum(iSM);
78
79 // info for each tower
80 for (Int_t j=0; j<nAPDPerSM; j++) {
81 inputFile >> iCol >> iRow >> tempCoeff >> iSrc;
82
83 // check that input values are not out bounds
84 if (iCol<0 || iCol>(AliEMCALGeoParams::fgkEMCALCols-1) ||
85 iRow<0 || iRow>(AliEMCALGeoParams::fgkEMCALRows-1) ) {
86 printf("AliEMCALCalibTempCoeff::ReadCalibTempCoeffInfo - Error while reading input file; j %d iCol %d iRow %d\n", j, iCol, iRow);
87 return;
88 }
89
90 // assume that this info is already swapped and done for this basis?
91 if (swapSides) {
92 // C side, oriented differently than A side: swap is requested
93 iCol = AliEMCALGeoParams::fgkEMCALCols-1 - iCol;
94 iRow = AliEMCALGeoParams::fgkEMCALRows-1 - iRow;
95 }
96
97 t->SetTC(iCol, iRow, tempCoeff);
98 t->SetSrc(iCol, iRow, iSrc);
99 }
100
101 } // i, SuperModule
102
103 inputFile.close();
104
105 return;
106}
107
108//____________________________________________________________________________
109void AliEMCALCalibTempCoeff::WriteTextCalibTempCoeffInfo(const TString &txtFileName,
110 Bool_t swapSides)
111{
112 // write data to txt file. ; coordinates given on SuperModule basis
113
114 std::ofstream outputFile(txtFileName.Data());
115 if (!outputFile) {
116 printf("AliEMCALCalibTempCoeff::WriteCalibTempCoeffInfo - Cannot open the APD output file %s\n", txtFileName.Data());
117 return;
118 }
119
120 Int_t iCol = 0;
121 Int_t iRow = 0;
122
123 Int_t nAPDPerSM = AliEMCALGeoParams::fgkEMCALCols * AliEMCALGeoParams::fgkEMCALRows;
124 Float_t tempCoeff = 0;
125 Int_t iSrc = 0;
126
127 for (Int_t i = 0; i < fNSuperModule; i++) {
128 AliEMCALSuperModuleCalibTempCoeff * t = (AliEMCALSuperModuleCalibTempCoeff*) fSuperModuleData[i];
129
130 // info for each tower
131 for (Int_t j=0; j<nAPDPerSM; j++) {
132 iCol = j / AliEMCALGeoParams::fgkEMCALRows;
133 iRow = j % AliEMCALGeoParams::fgkEMCALRows;
134
135 tempCoeff = t->GetTC(iCol, iRow);
136 iSrc = t->GetSrc(iCol, iRow);
137
138 if (swapSides) {
139 // C side, oriented differently than A side: swap is requested
140 iCol = AliEMCALGeoParams::fgkEMCALCols-1 - iCol;
141 iRow = AliEMCALGeoParams::fgkEMCALRows-1 - iRow;
142 }
143
144 outputFile << iCol << " " << iRow
145 << " " << tempCoeff
146 << " " << iSrc << endl;
147 }
148
149 } // i, SuperModule
150
151 outputFile.close();
152
153 return;
154}
155
156//____________________________________________________________________________
157void AliEMCALCalibTempCoeff::ReadRootCalibTempCoeffInfo(const TString &rootFileName,
158 Bool_t swapSides)
159{
160 //Read data from root file. ; coordinates given on SuperModule basis
161 TFile inputFile(rootFileName, "read");
162
163 TTree *tree = (TTree*) inputFile.Get("tree");
164
165 ReadTreeCalibTempCoeffInfo(tree, swapSides);
166
167 inputFile.Close();
168
169 return;
170}
171
172//____________________________________________________________________________
173void AliEMCALCalibTempCoeff::ReadTreeCalibTempCoeffInfo(TTree *tree,
174 Bool_t swapSides)
175{
176 // how many SuperModule's worth of info do we have?
177 Int_t nAPDPerSM = AliEMCALGeoParams::fgkEMCALCols * AliEMCALGeoParams::fgkEMCALRows;
178 fNSuperModule = tree->GetEntries();
179
180 Int_t iSM = 0; // SuperModule index
181 // list of values to be read
182 // info for each tower
183 Float_t tempCoeff[AliEMCALGeoParams::fgkEMCALCols][AliEMCALGeoParams::fgkEMCALRows];
184 Int_t iSrc[AliEMCALGeoParams::fgkEMCALCols][AliEMCALGeoParams::fgkEMCALRows];
185 // end - all values
186
187 // just to make the initializations of the arrays are done correctly, let's use memset
188 memset(tempCoeff, 0, sizeof(tempCoeff));
189 memset(iSrc, 0, sizeof(iSrc));
190
191 // declare the branches
192 tree->SetBranchAddress("iSM", &iSM);
193 tree->SetBranchAddress("TempCoeff", tempCoeff);
194 tree->SetBranchAddress("Src", iSrc);
195
196 // indices for looping over the towers
197 Int_t iCol = 0;
198 Int_t iRow = 0;
199
200 for (int ient=0; ient<tree->GetEntries(); ient++) {
201 tree->GetEntry(ient);
202
203 // assume the index SuperModules come in order: i=iSM
204 AliEMCALSuperModuleCalibTempCoeff * t = (AliEMCALSuperModuleCalibTempCoeff*) fSuperModuleData[iSM];
205
206 t->SetSuperModuleNum(iSM);
207
208 // third: info for each tower
209 for (Int_t j=0; j<nAPDPerSM; j++) {
210 iCol = j / AliEMCALGeoParams::fgkEMCALRows;
211 iRow = j % AliEMCALGeoParams::fgkEMCALRows;
212
213 // help variables: possibly modified or swapped indices
214 int iColMod = iCol;
215 int iRowMod = iRow;
216 // assume that this info is already swapped and done for this basis?
217 if (swapSides) {
218 // C side, oriented differently than A side: swap is requested
219 iColMod = AliEMCALGeoParams::fgkEMCALCols-1 - iCol;
220 iRowMod = AliEMCALGeoParams::fgkEMCALRows-1 - iRow;
221 }
222
223 t->SetTC(iColMod, iRowMod, tempCoeff[iCol][iRow]);
224 t->SetSrc(iColMod, iRowMod, iSrc[iCol][iRow]);
225 }
226
227 } // loop over entries
228
229 return;
230}
231
232//____________________________________________________________________________
233void AliEMCALCalibTempCoeff::WriteRootCalibTempCoeffInfo(const TString &rootFileName,
234 Bool_t swapSides)
235{
236 // write data to root file. ; coordinates given on SuperModule basis
237 TFile destFile(rootFileName, "recreate");
238 if (destFile.IsZombie()) {
239 return;
240 }
241 destFile.cd();
242
243 TTree *tree = new TTree("tree","");
244
245 // variables for filling the TTree
246 Int_t iSM = 0; // SuperModule index
247 // list of values to be written
248
249 Float_t tempCoeff[AliEMCALGeoParams::fgkEMCALCols][AliEMCALGeoParams::fgkEMCALRows];
250 Int_t iSrc[AliEMCALGeoParams::fgkEMCALCols][AliEMCALGeoParams::fgkEMCALRows]; // end - all values
251
252 // just to make the initializations of the arrays are done correctly, let's use memset
253 memset(tempCoeff, 0, sizeof(tempCoeff));
254 memset(iSrc, 0, sizeof(iSrc));
255
256 Int_t nAPDPerSM = AliEMCALGeoParams::fgkEMCALCols * AliEMCALGeoParams::fgkEMCALRows;
257 // for looping over towers
258 Int_t iCol = 0;
259 Int_t iRow = 0;
260
261 // declare the branches
262 // first
263 tree->Branch("iSM", &iSM, "iSM/I");
264 // info for each tower; see if a 2D array works OK or if we'll have to use 1D arrays instead
265 tree->Branch( "TempCoeff", &tempCoeff, Form("TempCoeff[%d][%d]/F", AliEMCALGeoParams::fgkEMCALCols, AliEMCALGeoParams::fgkEMCALRows) );
266 tree->Branch( "Src", &iSrc, Form("Src[%d][%d]/I", AliEMCALGeoParams::fgkEMCALCols, AliEMCALGeoParams::fgkEMCALRows) );
267
268 for (iSM = 0; iSM < fNSuperModule; iSM++) {
269 AliEMCALSuperModuleCalibTempCoeff * t = (AliEMCALSuperModuleCalibTempCoeff*) fSuperModuleData[iSM];
270
271 iSM = t->GetSuperModuleNum();
272
273 // info for each tower
274 for (Int_t j=0; j<nAPDPerSM; j++) {
275 iCol = j / AliEMCALGeoParams::fgkEMCALRows;
276 iRow = j % AliEMCALGeoParams::fgkEMCALRows;
277
278 // help variables: possibly modified or swapped indices
279 int iColMod = iCol;
280 int iRowMod = iRow;
281 // assume that this info is already swapped and done for this basis?
282 if (swapSides) {
283 // C side, oriented differently than A side: swap is requested
284 iColMod = AliEMCALGeoParams::fgkEMCALCols-1 - iCol;
285 iRowMod = AliEMCALGeoParams::fgkEMCALRows-1 - iRow;
286 }
287
288 tempCoeff[iColMod][iRowMod] = t->GetTC(iCol, iRow);
289 iSrc[iColMod][iRowMod] = t->GetSrc(iCol, iRow);
290 }
291
292 tree->Fill();
293 } // i, SuperModule
294
295 tree->Write();
296 destFile.Close();
297
298 return;
299}
300
301//____________________________________________________________________________
302AliEMCALCalibTempCoeff::~AliEMCALCalibTempCoeff()
303{
304 fSuperModuleData.Delete();
305}
306
307//____________________________________________________________________________
308AliEMCALSuperModuleCalibTempCoeff * AliEMCALCalibTempCoeff::GetSuperModuleCalibTempCoeffNum(Int_t supModIndex)const
309{
310 for (int i=0; i<fNSuperModule; i++) {
311 AliEMCALSuperModuleCalibTempCoeff * t = (AliEMCALSuperModuleCalibTempCoeff*) fSuperModuleData[i];
312 if (t->GetSuperModuleNum() == supModIndex) {
313 return t;
314 }
315 }
316
317 // if we arrived here, then nothing was found.. just return a NULL pointer
318 return NULL;
319}
320