// ******* Example to Create or Read OADB Container 
// ******* for EMCal Run by Run calibration dependent 
// ******* on Temperature variations
// Author : Gustavo Conesa Balbastre (LPSC-Grenoble)

// rescale some 2011 runs slightly (0.5%):
const Float_t rescaleFactor = 1.005;
//const Float_t rescaleFactor = 0.9975;
const Int_t kFirst = 156620;
const Int_t kLast = 162740;

void CreateEMCAL_OADB_RunByRunTemperatureECalibCorrection(Int_t opt = 1, Int_t runNumber = 170387)
{
  if(opt == 0) Read(runNumber);
  if(opt == 1) Create();
}

void Create()
{
  //Create OADB container for Temperature calibration parameters

  gSystem->Load("libOADB");            

  AliOADBContainer* con = new AliOADBContainer("AliEMCALRunDepTempCalibCorrections");
      
  // Get the list of run numbers to be added to the OADB, parameters provided usually in a 
  // root file per run
  // Tar ball with all the files can be found here
  // https://twiki.cern.ch/twiki/bin/viewauth/ALICE/EMCalTimeDependentCalibrations
  
  ifstream fList;
  fList.open("CorrectionFiles/runlist.txt");
  
  Int_t runNumber  = 0;
  Float_t multiplier = 10000; // allows us to store the values in TH1S with some precision
  TString string;
  Int_t nRuns=0;
  Int_t nSM = 12;

  AliEMCALGeometry* geom = AliEMCALGeometry::GetInstance("EMCAL_COMPLETE12SMV1");
 
  if (fList.good()) 
  {

    while( string.ReadLine(fList, kFALSE) ) 
    {
      sscanf(string.Data(), "%d",&runNumber);
      
      if     (runNumber < 140000) nSM = 4;
      else if(runNumber < 200000) nSM = 10;

      if(runNumber>100000){

      multiplier = 10000; // allows us to store the values in TH1S with some precision
      if (runNumber>=kFirst && runNumber<=kLast) { // rescale some runs
	multiplier *= rescaleFactor;
      }
      printf("Run %d multiplier %5.0f\n", runNumber, multiplier);

      // Access class that contains methods to read the content of 
      // the calibration file per run
      AliEMCALCalibTimeDepCorrection  *corr =  new AliEMCALCalibTimeDepCorrection();
      corr->ReadRootInfo(Form("CorrectionFiles/Run%d_Correction.root",runNumber));

      // Init the histogram
      TH1S *h = new TH1S(Form("h%d",runNumber),"",24*48*nSM,0,24*48*nSM);
            
      for(Int_t ism = 0; ism < nSM; ism++) {
        for(Int_t icol = 0; icol < 48; icol++) {
          for(Int_t irow = 0; irow < 24; irow++) {
            Float_t recalFactor = corr->GetCorrection(ism, icol,irow,0);            
            Int_t absID = geom->GetAbsCellIdFromCellIndexes(ism, irow, icol);
            
            h->SetBinContent(absID,(Short_t)(recalFactor*multiplier));
          }
        }
      }
      
      con->AddDefaultObject(h);
      
      //Establishing run number with the correct objects
      con->AppendObject(h,runNumber,runNumber);
      
      delete corr;
      
      nRuns++;
      }
    }
  }
  fList.close();
  printf(" *** nRuns ***  %d\n",nRuns);

  // add dummy object at the end of file:
  runNumber++;
  multiplier = 10000;
  // Init the histogram
  printf("Dummy/extra Run %d at EOF multiplier %5.0f\n", runNumber, multiplier);  
  TH1S *h = new TH1S(Form("h%d",runNumber),"",24*48*nSM,0,24*48*nSM);
            
  for(Int_t ism = 0; ism < nSM; ism++) {
    for(Int_t icol = 0; icol < 48; icol++) {
      for(Int_t irow = 0; irow < 24; irow++) {
	Float_t recalFactor = 1;
	Int_t absID = geom->GetAbsCellIdFromCellIndexes(ism, irow, icol);
            
	h->SetBinContent(absID,(Short_t)(recalFactor*multiplier));
      }
    }
  }
      
  con->AddDefaultObject(h);    
  //Establishing run number with the correct objects
  con->AppendObject(h,runNumber,runNumber);
  
  con->WriteToFile("EMCALTemperatureCorrCalib.root");   
    
}

void Read(Int_t runNumber = 170387)
{
  
  gSystem->Load("libOADB");            
  
  AliOADBContainer *cont=new AliOADBContainer("");
  cont->InitFromFile("$ALICE_ROOT/OADB/EMCAL/EMCALTemperatureCorrCalib.root", "AliEMCALRunDepTempCalibCorrections");
  
  //cout<<"_________--------------- dump ---------------------___________"<<endl;
  //cont->Dump();
  
  //cout<<"cont->GetDefaultList()->Print()"<<endl;
  //cont->GetDefaultList()->Print();
  
  TH1S *h=cont->GetObject(runNumber); //GetObject(int runnumber)
  
  if (h) {
    printf("runNumber %d found\n", runNumber);
  }
  else {
    printf("runNumber %d not found\n", runNumber);
    // let's get the closest runnumber
    Int_t lower = 0;
    Int_t ic = 0;
    Int_t maxEntry = cont->GetNumberOfEntries();

    while ( (ic < maxEntry) && (cont->UpperLimit(ic) < runNumber) ) {
      lower = ic;
      ic++; 
    }

    Int_t closest = lower;
    if ( (ic<maxEntry) && 
	 (cont->LowerLimit(ic)-runNumber) < (runNumber - cont->UpperLimit(lower)) ) {
	 closest = ic;
    }

    cout << "found closest id " << closest 
	 << " from run " << cont->LowerLimit(closest) << endl;
    h = (TH1S*) cont->GetObjectByIndex(closest); 
    h->Print(); // tmp debug
    cout << endl;
  }
  

  AliEMCALGeometry* geom = AliEMCALGeometry::GetInstance("EMCAL_COMPLETE12SMV1");
  
  // Read parameter file line-by-line  
  // Get number of lines first
  
  Int_t nSM = 10;
  
  for(Int_t iabsID = 0; iabsID < 24*48*nSM; iabsID++)
  {
    printf("absID %d, content %d\n",iabsID,h->GetBinContent(iabsID));
    
  }
  
  h->Draw();

}