[u/mrichter/AliRoot.git] / EMCAL / AliEMCALCalibTimeDepCorrection.cxx

/**************************************************************************
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/

/* $Id: $ */

// Objects of this class contain info on time-dependent corrections
//

#include <fstream>
#include <TString.h>
#include <TFile.h>
#include <TTree.h>

#include "AliEMCALCalibTimeDepCorrection.h"

using namespace std;

ClassImp(AliEMCALCalibTimeDepCorrection)

//____________________________________________________________________________
AliEMCALCalibTimeDepCorrection::AliEMCALCalibTimeDepCorrection(const int nSM) : 
  fNSuperModule(nSM),
  fSuperModuleData(),
  fStartTime(0),
  fNTimeBins(0),
  fTimeBinSize(0)
{
  //Default constructor.
  for (int i=0; i<fNSuperModule; i++) {
    fSuperModuleData.Add(new AliEMCALSuperModuleCalibTimeDepCorrection(i));
  }
  fSuperModuleData.Compress(); // compress the TObjArray
}

//____________________________________________________________________________
void AliEMCALCalibTimeDepCorrection::InitCorrection(Int_t nSM, Int_t nBins, Float_t val)
{
  // This methods assumes that you are using SuperModules 0..nSM-1
  fNSuperModule = nSM;

  Int_t iSM = 0; // SuperModule index
  Int_t iCol = 0;
  Int_t iRow = 0;
  Int_t nCorr = nBins;
  Float_t correction = val;

  Int_t nAPDPerSM = AliEMCALGeoParams::fgkEMCALCols * AliEMCALGeoParams::fgkEMCALRows;

  for (Int_t i = 0; i < fNSuperModule; i++) {
    AliEMCALSuperModuleCalibTimeDepCorrection * t = (AliEMCALSuperModuleCalibTimeDepCorrection*) fSuperModuleData[i];
    iSM = i;
    t->SetSuperModuleNum(iSM); // assume SMs are coming in order

    for (Int_t j=0; j<nAPDPerSM; j++) {
      iCol = j / AliEMCALGeoParams::fgkEMCALRows;
      iRow = j % AliEMCALGeoParams::fgkEMCALRows;

      // set size of TArray
      t->GetCorrection(iCol,iRow)->Set(nCorr);
      for (Int_t k=0; k<nCorr; k++) {
	// add to TArray
	t->GetCorrection(iCol,iRow)->AddAt(correction, k); // AddAt = SetAt..
      }
    }

  } // i, SuperModule

  return;
}

//____________________________________________________________________________
void AliEMCALCalibTimeDepCorrection::SetCorrection(Int_t supModIndex, Int_t iCol, Int_t iRow, Int_t iBin, Float_t val)
{ // if you call for non-existing data, there may be a crash..
  ((AliEMCALSuperModuleCalibTimeDepCorrection*)fSuperModuleData[supModIndex])->GetCorrection(iCol,iRow)->AddAt(val, iBin); // AddAt = SetAt..
 return; 
}

//____________________________________________________________________________
Float_t AliEMCALCalibTimeDepCorrection::GetCorrection(Int_t supModIndex, Int_t iCol, Int_t iRow, Int_t iBin) const
{ // if you call for non-existing data, there may be a crash..
  return ((AliEMCALSuperModuleCalibTimeDepCorrection*)fSuperModuleData[supModIndex])->GetCorrection(iCol,iRow)->At(iBin);
}

//____________________________________________________________________________
void AliEMCALCalibTimeDepCorrection::ReadTextInfo(Int_t nSM, const TString &txtFileName,
						  Bool_t swapSides)
{
  //Read data from txt file. ; coordinates given on SuperModule basis

  std::ifstream inputFile(txtFileName.Data());
  if (!inputFile) {
    printf("AliEMCALCalibTimeDepCorrection::ReadTextInfo - Cannot open the APD info file %s\n", txtFileName.Data());
    return;
  }

  fNSuperModule = nSM;

  Int_t iSM = 0; // SuperModule index
  Int_t iCol = 0;
  Int_t iRow = 0;
  Int_t nCorr = 0;
  Float_t correction = 0;

  Int_t nAPDPerSM = AliEMCALGeoParams::fgkEMCALCols * AliEMCALGeoParams::fgkEMCALRows;

  // read header info 
  inputFile >> fStartTime >> fNTimeBins >> fTimeBinSize;

  for (Int_t i = 0; i < fNSuperModule; i++) {
    AliEMCALSuperModuleCalibTimeDepCorrection * t = (AliEMCALSuperModuleCalibTimeDepCorrection*) fSuperModuleData[i];

    if (!inputFile) {
      printf("AliEMCALCalibTimeDepCorrection::ReadTextInfo - Error while reading input file; likely EOF..\n");
      return;
    }
    inputFile >> iSM;
    t->SetSuperModuleNum(iSM);

    for (Int_t j=0; j<nAPDPerSM; j++) {
      inputFile >> iCol >> iRow >> nCorr;

      // check that input values are not out bounds
      if (iCol<0 || iCol>(AliEMCALGeoParams::fgkEMCALCols-1) ||
	  iRow<0 || iRow>(AliEMCALGeoParams::fgkEMCALRows-1) || 
	  nCorr<0 ) {
	printf("AliEMCALCalibTimeDepCorrection::ReadTextInfo - Error while reading input file; j %d iCol %d iRow %d nCorr %d\n", j, iCol, iRow, nCorr);
      return;
      }

      // assume that this info is already swapped and done for this basis?
      if (swapSides) {
	// C side, oriented differently than A side: swap is requested
	iCol = AliEMCALGeoParams::fgkEMCALCols-1 - iCol;
	iRow = AliEMCALGeoParams::fgkEMCALRows-1 - iRow;
      }

      // set size of TArray
      t->GetCorrection(iCol,iRow)->Set(nCorr);
      for (Int_t k=0; k<nCorr; k++) {
	inputFile >> correction;
	// add to TArray
	t->GetCorrection(iCol,iRow)->AddAt(correction, k);
      }
    }

  } // i, SuperModule

  inputFile.close();

  return;
}

//____________________________________________________________________________
void AliEMCALCalibTimeDepCorrection::WriteTextInfo(const TString &txtFileName,
						   Bool_t swapSides)
{
  // write data to txt file. ; coordinates given on SuperModule basis

  std::ofstream outputFile(txtFileName.Data());
  if (!outputFile) {
    printf("AliEMCALCalibTimeDepCorrection::WriteTextInfo - Cannot open the APD output file %s\n", txtFileName.Data());
    return;
  }

  Int_t iCol = 0;
  Int_t iRow = 0;
  Int_t nCorr = 0;

  Int_t nAPDPerSM = AliEMCALGeoParams::fgkEMCALCols * AliEMCALGeoParams::fgkEMCALRows;

  // write header info 
  outputFile << fStartTime << " " << fNTimeBins << " " << fTimeBinSize << endl;

  for (Int_t i = 0; i < fNSuperModule; i++) {
    AliEMCALSuperModuleCalibTimeDepCorrection * t = (AliEMCALSuperModuleCalibTimeDepCorrection*) fSuperModuleData[i];
    outputFile << t->GetSuperModuleNum() << endl;

    for (Int_t j=0; j<nAPDPerSM; j++) {
      iCol = j / AliEMCALGeoParams::fgkEMCALRows;
      iRow = j % AliEMCALGeoParams::fgkEMCALRows;

      nCorr = t->GetCorrection(iCol,iRow)->GetSize();

      if (swapSides) {
	// C side, oriented differently than A side: swap is requested
	iCol = AliEMCALGeoParams::fgkEMCALCols-1 - iCol;
	iRow = AliEMCALGeoParams::fgkEMCALRows-1 - iRow;
      }

      outputFile << iCol << " " << iRow << " " << nCorr << endl;
      for (Int_t k=0; k<nCorr; k++) {
	outputFile << t->GetCorrection(iCol,iRow)->At(k) << " ";
      }
      outputFile << endl;

    }

  } // i, SuperModule

  outputFile.close();

  return;
}

//____________________________________________________________________________
void AliEMCALCalibTimeDepCorrection::ReadRootInfo(const TString &rootFileName,
						  Bool_t swapSides)
{
  //Read data from root file. ; coordinates given on SuperModule basis
  TFile inputFile(rootFileName, "read");  

  TTree *treeGlob = (TTree*) inputFile.Get("treeGlob");
  TTree *treeCorr = (TTree*) inputFile.Get("treeCorr");

  ReadTreeInfo(treeGlob, treeCorr, swapSides);

  inputFile.Close();

  return;
}

//____________________________________________________________________________
void AliEMCALCalibTimeDepCorrection::ReadTreeInfo(TTree *treeGlob, TTree *treeCorr,
						  Bool_t swapSides)
{
 // how many SuperModule's worth of entries / APDs do we have?
  Int_t nAPDPerSM = AliEMCALGeoParams::fgkEMCALCols * AliEMCALGeoParams::fgkEMCALRows;
  fNSuperModule = treeCorr->GetEntries() / nAPDPerSM;

  // global variables : only one entry
  treeGlob->SetBranchAddress("fStartTime", &fStartTime);
  treeGlob->SetBranchAddress("fNTimeBins", &fNTimeBins);
  treeGlob->SetBranchAddress("fTimeBinSize", &fTimeBinSize);
  treeGlob->GetEntry(0); 

  Int_t iSM = 0; // SuperModule index
  Int_t iCol = 0;
  Int_t iRow = 0;
  // list of values to be read
  Int_t nCorr = 0;
  Float_t correction[fgkMaxTimeBins] = {0};
  // make sure it's really initialized correctly
  memset(correction, 0, sizeof(correction)); // better safe than sorry
  // end - all values

  // declare the branches
  treeCorr->SetBranchAddress("iSM", &iSM);
  treeCorr->SetBranchAddress("iCol", &iCol);
  treeCorr->SetBranchAddress("iRow", &iRow);
  treeCorr->SetBranchAddress("nCorr", &nCorr);
  treeCorr->SetBranchAddress("correction", correction);

  for (int ient=0; ient<treeCorr->GetEntries(); ient++) {
    treeCorr->GetEntry(ient);

    // assume the index SuperModules come in order: i=iSM
    AliEMCALSuperModuleCalibTimeDepCorrection * t = (AliEMCALSuperModuleCalibTimeDepCorrection*) fSuperModuleData[iSM];
    t->SetSuperModuleNum(iSM);

    // assume that this info is already swapped and done for this basis?
    if (swapSides) {
      // C side, oriented differently than A side: swap is requested
      iCol = AliEMCALGeoParams::fgkEMCALCols-1 - iCol;
      iRow = AliEMCALGeoParams::fgkEMCALRows-1 - iRow;
    }


    // set size of TArray
    t->GetCorrection(iCol,iRow)->Set(nCorr);
    for (Int_t k=0; k<nCorr; k++) {
      // add to TArray
      t->GetCorrection(iCol,iRow)->AddAt(correction[k], k);
    }

  } // entry

  return;
}

//____________________________________________________________________________
void AliEMCALCalibTimeDepCorrection::WriteRootInfo(const TString &rootFileName,
						   Bool_t swapSides)
{
  // write data to root file. ; coordinates given on SuperModule basis
  TFile destFile(rootFileName, "recreate");  
  if (destFile.IsZombie()) {
    return;
  }  
  destFile.cd();

  TTree *treeGlob = new TTree("treeGlob","");
  TTree *treeCorr = new TTree("treeCorr","");

  // global part only has one entry
  treeGlob->Branch("fStartTime", &fStartTime, "fStartTime/i"); // unsigned int..
  treeGlob->Branch("fNTimeBins", &fNTimeBins, "fNTimeBins/I");
  treeGlob->Branch("fTimeBinSize", &fTimeBinSize, "fTimeBinSize/I");
  treeGlob->Fill();

  // variables for filling the TTree
  Int_t iSM = 0; // SuperModule index
  Int_t iCol = 0;
  Int_t iRow = 0;
  Int_t nCorr = 0;
  Float_t correction[fgkMaxTimeBins] = {0};
  // make sure it's really initialized correctly
  memset(correction, 0, sizeof(correction)); // better safe than sorry

  // declare the branches
  treeCorr->Branch("iSM", &iSM, "iSM/I");
  treeCorr->Branch("iCol", &iCol, "iCol/I");
  treeCorr->Branch("iRow", &iRow, "iRow/I");
  treeCorr->Branch("nCorr", &nCorr, "nCorr/I");
  treeCorr->Branch("correction", &correction, "correction[nCorr]/F");

  Int_t nAPDPerSM = AliEMCALGeoParams::fgkEMCALCols * AliEMCALGeoParams::fgkEMCALRows;

  for (iSM = 0; iSM < fNSuperModule; iSM++) {
    AliEMCALSuperModuleCalibTimeDepCorrection * t = (AliEMCALSuperModuleCalibTimeDepCorrection*) fSuperModuleData[iSM];

    for (Int_t j=0; j<nAPDPerSM; j++) {
      iCol = j / AliEMCALGeoParams::fgkEMCALRows;
      iRow = j % AliEMCALGeoParams::fgkEMCALRows;

      nCorr = t->GetCorrection(iCol,iRow)->GetSize();
      if (nCorr > fgkMaxTimeBins) {
	printf("AliEMCALCalibTimeDepCorrection::WriteRootInfo - too many correction/timebins %d kept\n", nCorr);
	return;
      }

      if (swapSides) {
	// C side, oriented differently than A side: swap is requested
	iCol = AliEMCALGeoParams::fgkEMCALCols-1 - iCol;
	iRow = AliEMCALGeoParams::fgkEMCALRows-1 - iRow;
      }

      for (Int_t k=0; k<nCorr; k++) {
	correction[k] = t->GetCorrection(iCol,iRow)->At(k);
      }

      treeCorr->Fill();
    }

  } // i, SuperModule

  treeGlob->Write();
  treeCorr->Write();
  destFile.Close();

  return;
}

//____________________________________________________________________________
AliEMCALCalibTimeDepCorrection::~AliEMCALCalibTimeDepCorrection()
{
  fSuperModuleData.Delete();
}

//____________________________________________________________________________
AliEMCALSuperModuleCalibTimeDepCorrection * AliEMCALCalibTimeDepCorrection::GetSuperModuleCalibTimeDepCorrectionNum(Int_t supModIndex)const
{ // getter via index
  for (int i=0; i<fNSuperModule; i++) {
    AliEMCALSuperModuleCalibTimeDepCorrection * t = (AliEMCALSuperModuleCalibTimeDepCorrection*) fSuperModuleData[i];
    if (t->GetSuperModuleNum() == supModIndex) {
      return t;
    }
  }

  // if we arrived here, then nothing was found.. just return a NULL pointer 
  return NULL;
}
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>
61917ab3	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	{
d81e6423	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
d81e6423	66	for (Int_t j=0; j<nAPDPerSM; j++) {
716fca62	67	iCol = j / AliEMCALGeoParams::fgkEMCALRows;
716fca62	68	iRow = j % AliEMCALGeoParams::fgkEMCALRows;
d81e6423	69
d81e6423	70	// set size of TArray
2f17a269	71	t->GetCorrection(iCol,iRow)->Set(nCorr);
d81e6423	72	for (Int_t k=0; k<nCorr; k++) {
d81e6423	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,
61917ab3	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];
2f17a269	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
b48d1356	135	if (iCol<0 \|\| iCol>(AliEMCALGeoParams::fgkEMCALCols-1) \|\|
3a282863	136	iRow<0 \|\| iRow>(AliEMCALGeoParams::fgkEMCALRows-1) \|\|
	137	nCorr<0 ) {
	138	printf("AliEMCALCalibTimeDepCorrection::ReadTextInfo - Error while reading input file; j %d iCol %d iRow %d nCorr %d\n", j, iCol, iRow, nCorr);
b48d1356	139	return;
	140	}
	141
d81e6423	142	// assume that this info is already swapped and done for this basis?
	143	if (swapSides) {
	144	// C side, oriented differently than A side: swap is requested
	145	iCol = AliEMCALGeoParams::fgkEMCALCols-1 - iCol;
	146	iRow = AliEMCALGeoParams::fgkEMCALRows-1 - iRow;
	147	}
	148
	149	// set size of TArray
2f17a269	150	t->GetCorrection(iCol,iRow)->Set(nCorr);
d81e6423	151	for (Int_t k=0; k<nCorr; k++) {
ab962f7b	152	inputFile >> correction;
d81e6423	153	// add to TArray
ab962f7b	154	t->GetCorrection(iCol,iRow)->AddAt(correction, k);
d81e6423	155	}
	156	}
	157
	158	} // i, SuperModule
	159
	160	inputFile.close();
	161
	162	return;
	163	}
	164
	165	//____________________________________________________________________________
61917ab3	166	void AliEMCALCalibTimeDepCorrection::WriteTextInfo(const TString &txtFileName,
61917ab3	167	Bool_t swapSides)
d81e6423	168	{
	169	// write data to txt file. ; coordinates given on SuperModule basis
	170
	171	std::ofstream outputFile(txtFileName.Data());
	172	if (!outputFile) {
61917ab3	173	printf("AliEMCALCalibTimeDepCorrection::WriteTextInfo - Cannot open the APD output file %s\n", txtFileName.Data());
d81e6423	174	return;
	175	}
	176
	177	Int_t iCol = 0;
	178	Int_t iRow = 0;
	179	Int_t nCorr = 0;
	180
	181	Int_t nAPDPerSM = AliEMCALGeoParams::fgkEMCALCols * AliEMCALGeoParams::fgkEMCALRows;
	182
61917ab3	183	// write header info
	184	outputFile << fStartTime << " " << fNTimeBins << " " << fTimeBinSize << endl;
	185
d81e6423	186	for (Int_t i = 0; i < fNSuperModule; i++) {
2f17a269	187	AliEMCALSuperModuleCalibTimeDepCorrection * t = (AliEMCALSuperModuleCalibTimeDepCorrection*) fSuperModuleData[i];
2f17a269	188	outputFile << t->GetSuperModuleNum() << endl;
d81e6423	189
	190	for (Int_t j=0; j<nAPDPerSM; j++) {
	191	iCol = j / AliEMCALGeoParams::fgkEMCALRows;
	192	iRow = j % AliEMCALGeoParams::fgkEMCALRows;
	193
2f17a269	194	nCorr = t->GetCorrection(iCol,iRow)->GetSize();
d81e6423	195
	196	if (swapSides) {
	197	// C side, oriented differently than A side: swap is requested
	198	iCol = AliEMCALGeoParams::fgkEMCALCols-1 - iCol;
	199	iRow = AliEMCALGeoParams::fgkEMCALRows-1 - iRow;
	200	}
	201
	202	outputFile << iCol << " " << iRow << " " << nCorr << endl;
	203	for (Int_t k=0; k<nCorr; k++) {
2f17a269	204	outputFile << t->GetCorrection(iCol,iRow)->At(k) << " ";
d81e6423	205	}
	206	outputFile << endl;
	207
	208	}
	209
	210	} // i, SuperModule
	211
	212	outputFile.close();
	213
	214	return;
	215	}
	216
61917ab3	217	//____________________________________________________________________________
	218	void AliEMCALCalibTimeDepCorrection::ReadRootInfo(const TString &rootFileName,
	219	Bool_t swapSides)
	220	{
	221	//Read data from root file. ; coordinates given on SuperModule basis
	222	TFile inputFile(rootFileName, "read");
	223
	224	TTree treeGlob = (TTree) inputFile.Get("treeGlob");
	225	TTree treeCorr = (TTree) inputFile.Get("treeCorr");
	226
	227	ReadTreeInfo(treeGlob, treeCorr, swapSides);
	228
	229	inputFile.Close();
	230
	231	return;
	232	}
	233
	234	//____________________________________________________________________________
	235	void AliEMCALCalibTimeDepCorrection::ReadTreeInfo(TTree treeGlob, TTree treeCorr,
	236	Bool_t swapSides)
	237	{
	238	// how many SuperModule's worth of entries / APDs do we have?
	239	Int_t nAPDPerSM = AliEMCALGeoParams::fgkEMCALCols * AliEMCALGeoParams::fgkEMCALRows;
	240	fNSuperModule = treeCorr->GetEntries() / nAPDPerSM;
	241
	242	// global variables : only one entry
	243	treeGlob->SetBranchAddress("fStartTime", &fStartTime);
	244	treeGlob->SetBranchAddress("fNTimeBins", &fNTimeBins);
	245	treeGlob->SetBranchAddress("fTimeBinSize", &fTimeBinSize);
	246	treeGlob->GetEntry(0);
	247
61917ab3	248	Int_t iSM = 0; // SuperModule index
	249	Int_t iCol = 0;
	250	Int_t iRow = 0;
	251	// list of values to be read
	252	Int_t nCorr = 0;
	253	Float_t correction[fgkMaxTimeBins] = {0};
	254	// make sure it's really initialized correctly
	255	memset(correction, 0, sizeof(correction)); // better safe than sorry
	256	// end - all values
	257
	258	// declare the branches
	259	treeCorr->SetBranchAddress("iSM", &iSM);
	260	treeCorr->SetBranchAddress("iCol", &iCol);
	261	treeCorr->SetBranchAddress("iRow", &iRow);
	262	treeCorr->SetBranchAddress("nCorr", &nCorr);
	263	treeCorr->SetBranchAddress("correction", correction);
	264
	265	for (int ient=0; ient<treeCorr->GetEntries(); ient++) {
	266	treeCorr->GetEntry(ient);
	267
	268	// assume the index SuperModules come in order: i=iSM
2f17a269	269	AliEMCALSuperModuleCalibTimeDepCorrection * t = (AliEMCALSuperModuleCalibTimeDepCorrection*) fSuperModuleData[iSM];
2f17a269	270	t->SetSuperModuleNum(iSM);
61917ab3	271
	272	// assume that this info is already swapped and done for this basis?
	273	if (swapSides) {
	274	// C side, oriented differently than A side: swap is requested
	275	iCol = AliEMCALGeoParams::fgkEMCALCols-1 - iCol;
	276	iRow = AliEMCALGeoParams::fgkEMCALRows-1 - iRow;
	277	}
	278
	279
	280	// set size of TArray
2f17a269	281	t->GetCorrection(iCol,iRow)->Set(nCorr);
61917ab3	282	for (Int_t k=0; k<nCorr; k++) {
61917ab3	283	// add to TArray
2f17a269	284	t->GetCorrection(iCol,iRow)->AddAt(correction[k], k);
61917ab3	285	}
	286
	287	} // entry
	288
	289	return;
	290	}
	291
	292	//____________________________________________________________________________
	293	void AliEMCALCalibTimeDepCorrection::WriteRootInfo(const TString &rootFileName,
	294	Bool_t swapSides)
	295	{
	296	// write data to root file. ; coordinates given on SuperModule basis
	297	TFile destFile(rootFileName, "recreate");
	298	if (destFile.IsZombie()) {
	299	return;
	300	}
	301	destFile.cd();
	302
	303	TTree *treeGlob = new TTree("treeGlob","");
	304	TTree *treeCorr = new TTree("treeCorr","");
	305
	306	// global part only has one entry
2c62d6a3	307	treeGlob->Branch("fStartTime", &fStartTime, "fStartTime/i"); // unsigned int..
61917ab3	308	treeGlob->Branch("fNTimeBins", &fNTimeBins, "fNTimeBins/I");
	309	treeGlob->Branch("fTimeBinSize", &fTimeBinSize, "fTimeBinSize/I");
	310	treeGlob->Fill();
	311
	312	// variables for filling the TTree
	313	Int_t iSM = 0; // SuperModule index
	314	Int_t iCol = 0;
	315	Int_t iRow = 0;
	316	Int_t nCorr = 0;
	317	Float_t correction[fgkMaxTimeBins] = {0};
	318	// make sure it's really initialized correctly
	319	memset(correction, 0, sizeof(correction)); // better safe than sorry
	320
	321	// declare the branches
	322	treeCorr->Branch("iSM", &iSM, "iSM/I");
	323	treeCorr->Branch("iCol", &iCol, "iCol/I");
	324	treeCorr->Branch("iRow", &iRow, "iRow/I");
	325	treeCorr->Branch("nCorr", &nCorr, "nCorr/I");
	326	treeCorr->Branch("correction", &correction, "correction[nCorr]/F");
	327
	328	Int_t nAPDPerSM = AliEMCALGeoParams::fgkEMCALCols * AliEMCALGeoParams::fgkEMCALRows;
	329
	330	for (iSM = 0; iSM < fNSuperModule; iSM++) {
2f17a269	331	AliEMCALSuperModuleCalibTimeDepCorrection * t = (AliEMCALSuperModuleCalibTimeDepCorrection*) fSuperModuleData[iSM];
61917ab3	332
	333	for (Int_t j=0; j<nAPDPerSM; j++) {
	334	iCol = j / AliEMCALGeoParams::fgkEMCALRows;
	335	iRow = j % AliEMCALGeoParams::fgkEMCALRows;
	336
2f17a269	337	nCorr = t->GetCorrection(iCol,iRow)->GetSize();
61917ab3	338	if (nCorr > fgkMaxTimeBins) {
	339	printf("AliEMCALCalibTimeDepCorrection::WriteRootInfo - too many correction/timebins %d kept\n", nCorr);
	340	return;
	341	}
	342
	343	if (swapSides) {
	344	// C side, oriented differently than A side: swap is requested
	345	iCol = AliEMCALGeoParams::fgkEMCALCols-1 - iCol;
	346	iRow = AliEMCALGeoParams::fgkEMCALRows-1 - iRow;
	347	}
	348
	349	for (Int_t k=0; k<nCorr; k++) {
2f17a269	350	correction[k] = t->GetCorrection(iCol,iRow)->At(k);
61917ab3	351	}
	352
	353	treeCorr->Fill();
	354	}
	355
	356	} // i, SuperModule
	357
	358	treeGlob->Write();
	359	treeCorr->Write();
	360	destFile.Close();
	361
	362	return;
	363	}
	364
d81e6423	365	//____________________________________________________________________________
	366	AliEMCALCalibTimeDepCorrection::~AliEMCALCalibTimeDepCorrection()
	367	{
2f17a269	368	fSuperModuleData.Delete();
d81e6423	369	}
	370
	371	//____________________________________________________________________________
2f17a269	372	AliEMCALSuperModuleCalibTimeDepCorrection * AliEMCALCalibTimeDepCorrection::GetSuperModuleCalibTimeDepCorrectionNum(Int_t supModIndex)const
afae9650	373	{ // getter via index
d81e6423	374	for (int i=0; i<fNSuperModule; i++) {
2f17a269	375	AliEMCALSuperModuleCalibTimeDepCorrection * t = (AliEMCALSuperModuleCalibTimeDepCorrection*) fSuperModuleData[i];
	376	if (t->GetSuperModuleNum() == supModIndex) {
	377	return t;
d81e6423	378	}
	379	}
	380
2f17a269	381	// if we arrived here, then nothing was found.. just return a NULL pointer
2f17a269	382	return NULL;
d81e6423	383	}
d81e6423	384