EMCAL/AliEMCALCalibTimeDepCorrection.cxx

   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 info on time-dependent corrections
  19 //
  20
  21 #include <fstream>
  22 #include <TString.h>
  23 #include <TFile.h>
  24 #include <TTree.h>
  25
  26 #include "AliEMCALCalibTimeDepCorrection.h"
  27
  28 using namespace std;
  29
  30 ClassImp(AliEMCALCalibTimeDepCorrection)
  31
  32 //____________________________________________________________________________
  33 AliEMCALCalibTimeDepCorrection::AliEMCALCalibTimeDepCorrection(const int nSM) :
  34   fNSuperModule(nSM),
  35   fSuperModuleData(),
  36   fStartTime(0),
  37   fNTimeBins(0),
  38   fTimeBinSize(0)
  39 {
  40   //Default constructor.
  41   for (int i=0; i<fNSuperModule; i++) {
  42     fSuperModuleData.Add(new AliEMCALSuperModuleCalibTimeDepCorrection(i));
  43   }
  44   fSuperModuleData.Compress(); // compress the TObjArray
  45 }
  46
  47 //____________________________________________________________________________
  48 void AliEMCALCalibTimeDepCorrection::InitCorrection(Int_t nSM, Int_t nBins, Float_t val)
  49 {
  50   // This methods assumes that you are using SuperModules 0..nSM-1
  51   fNSuperModule = nSM;
  52
  53   Int_t iSM = 0; // SuperModule index
  54   Int_t iCol = 0;
  55   Int_t iRow = 0;
  56   Int_t nCorr = nBins;
  57   Float_t correction = val;
  58
  59   Int_t nAPDPerSM = AliEMCALGeoParams::fgkEMCALCols * AliEMCALGeoParams::fgkEMCALRows;
  60
  61   for (Int_t i = 0; i < fNSuperModule; i++) {
  62     AliEMCALSuperModuleCalibTimeDepCorrection * t = (AliEMCALSuperModuleCalibTimeDepCorrection*) fSuperModuleData[i];
  63     iSM = i;
  64     t->SetSuperModuleNum(iSM); // assume SMs are coming in order
  65
  66     for (Int_t j=0; j<nAPDPerSM; j++) {
  67       iCol = j / AliEMCALGeoParams::fgkEMCALRows;
  68       iRow = j % AliEMCALGeoParams::fgkEMCALRows;
  69
  70       // set size of TArray
  71       t->GetCorrection(iCol,iRow)->Set(nCorr);
  72       for (Int_t k=0; k<nCorr; k++) {
  73         // add to TArray
  74         t->GetCorrection(iCol,iRow)->AddAt(correction, k); // AddAt = SetAt..
  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..
  86   ((AliEMCALSuperModuleCalibTimeDepCorrection*)fSuperModuleData[supModIndex])->GetCorrection(iCol,iRow)->AddAt(val, iBin); // AddAt = SetAt..
  87  return;
  88 }
  89
  90 //____________________________________________________________________________
  91 Float_t AliEMCALCalibTimeDepCorrection::GetCorrection(Int_t supModIndex, Int_t iCol, Int_t iRow, Int_t iBin) const
  92 { // if you call for non-existing data, there may be a crash..
  93   return ((AliEMCALSuperModuleCalibTimeDepCorrection*)fSuperModuleData[supModIndex])->GetCorrection(iCol,iRow)->At(iBin);
  94 }
  95
  96 //____________________________________________________________________________
  97 void AliEMCALCalibTimeDepCorrection::ReadTextInfo(Int_t nSM, const TString &txtFileName,
  98                                                   Bool_t swapSides)
  99 {
 100   //Read data from txt file. ; coordinates given on SuperModule basis
 101
 102   std::ifstream inputFile(txtFileName.Data());
 103   if (!inputFile) {
 104     printf("AliEMCALCalibTimeDepCorrection::ReadTextInfo - Cannot open the APD info file %s\n", txtFileName.Data());
 105     return;
 106   }
 107
 108   fNSuperModule = nSM;
 109
 110   Int_t iSM = 0; // SuperModule index
 111   Int_t iCol = 0;
 112   Int_t iRow = 0;
 113   Int_t nCorr = 0;
 114   Float_t correction = 0;
 115
 116   Int_t nAPDPerSM = AliEMCALGeoParams::fgkEMCALCols * AliEMCALGeoParams::fgkEMCALRows;
 117
 118   // read header info
 119   inputFile >> fStartTime >> fNTimeBins >> fTimeBinSize;
 120
 121   for (Int_t i = 0; i < fNSuperModule; i++) {
 122     AliEMCALSuperModuleCalibTimeDepCorrection * t = (AliEMCALSuperModuleCalibTimeDepCorrection*) fSuperModuleData[i];
 123
 124     if (!inputFile) {
 125       printf("AliEMCALCalibTimeDepCorrection::ReadTextInfo - Error while reading input file; likely EOF..\n");
 126       return;
 127     }
 128     inputFile >> iSM;
 129     t->SetSuperModuleNum(iSM);
 130
 131     for (Int_t j=0; j<nAPDPerSM; j++) {
 132       inputFile >> iCol >> iRow >> nCorr;
 133
 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
 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
 149       t->GetCorrection(iCol,iRow)->Set(nCorr);
 150       for (Int_t k=0; k<nCorr; k++) {
 151         inputFile >> correction;
 152         // add to TArray
 153         t->GetCorrection(iCol,iRow)->AddAt(correction, k);
 154       }
 155     }
 156
 157   } // i, SuperModule
 158
 159   inputFile.close();
 160
 161   return;
 162 }
 163
 164 //____________________________________________________________________________
 165 void AliEMCALCalibTimeDepCorrection::WriteTextInfo(const TString &txtFileName,
 166                                                    Bool_t swapSides)
 167 {
 168   // write data to txt file. ; coordinates given on SuperModule basis
 169
 170   std::ofstream outputFile(txtFileName.Data());
 171   if (!outputFile) {
 172     printf("AliEMCALCalibTimeDepCorrection::WriteTextInfo - Cannot open the APD output file %s\n", txtFileName.Data());
 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
 182   // write header info
 183   outputFile << fStartTime << " " << fNTimeBins << " " << fTimeBinSize << endl;
 184
 185   for (Int_t i = 0; i < fNSuperModule; i++) {
 186     AliEMCALSuperModuleCalibTimeDepCorrection * t = (AliEMCALSuperModuleCalibTimeDepCorrection*) fSuperModuleData[i];
 187     outputFile << t->GetSuperModuleNum() << endl;
 188
 189     for (Int_t j=0; j<nAPDPerSM; j++) {
 190       iCol = j / AliEMCALGeoParams::fgkEMCALRows;
 191       iRow = j % AliEMCALGeoParams::fgkEMCALRows;
 192
 193       nCorr = t->GetCorrection(iCol,iRow)->GetSize();
 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++) {
 203         outputFile << t->GetCorrection(iCol,iRow)->At(k) << " ";
 204       }
 205       outputFile << endl;
 206
 207     }
 208
 209   } // i, SuperModule
 210
 211   outputFile.close();
 212
 213   return;
 214 }
 215
 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
 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
 268     AliEMCALSuperModuleCalibTimeDepCorrection * t = (AliEMCALSuperModuleCalibTimeDepCorrection*) fSuperModuleData[iSM];
 269     t->SetSuperModuleNum(iSM);
 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
 280     t->GetCorrection(iCol,iRow)->Set(nCorr);
 281     for (Int_t k=0; k<nCorr; k++) {
 282       // add to TArray
 283       t->GetCorrection(iCol,iRow)->AddAt(correction[k], k);
 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++) {
 330     AliEMCALSuperModuleCalibTimeDepCorrection * t = (AliEMCALSuperModuleCalibTimeDepCorrection*) fSuperModuleData[iSM];
 331
 332     for (Int_t j=0; j<nAPDPerSM; j++) {
 333       iCol = j / AliEMCALGeoParams::fgkEMCALRows;
 334       iRow = j % AliEMCALGeoParams::fgkEMCALRows;
 335
 336       nCorr = t->GetCorrection(iCol,iRow)->GetSize();
 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++) {
 349         correction[k] = t->GetCorrection(iCol,iRow)->At(k);
 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
 364 //____________________________________________________________________________
 365 AliEMCALCalibTimeDepCorrection::~AliEMCALCalibTimeDepCorrection()
 366 {
 367   fSuperModuleData.Delete();
 368 }
 369
 370 //____________________________________________________________________________
 371 AliEMCALSuperModuleCalibTimeDepCorrection * AliEMCALCalibTimeDepCorrection::GetSuperModuleCalibTimeDepCorrectionNum(Int_t supModIndex)const
 372 {
 373   for (int i=0; i<fNSuperModule; i++) {
 374     AliEMCALSuperModuleCalibTimeDepCorrection * t = (AliEMCALSuperModuleCalibTimeDepCorrection*) fSuperModuleData[i];
 375     if (t->GetSuperModuleNum() == supModIndex) {
 376       return t;
 377     }
 378   }
 379
 380   // if we arrived here, then nothing was found.. just return a NULL pointer
 381   return NULL;
 382 }
 383