/* $Id$ */

// Script to create calibration parameters and store them into CDB
// Two sets of calibration parameters can be created:
// 1) equal parameters
// 2) randomly distributed parameters for decalibrated detector silumations

void AliPHOSSetCDB()
{
   menu = new TControlBar("vertical","PHOS CDB");
   menu->AddButton("Help to run PHOS CDB","Help()",
		   "Explains how to use PHOS CDS menus");
   menu->AddButton("Equal CC","SetCC(0)",
		   "Set equal calibration coefficients");
   menu->AddButton("Decalibrate","SetCC(1)",
		   "Set random decalibration calibration coefficients");
   menu->AddButton("Real equal CC","GetCC(0)",
		   "Read initial equal calibration coefficients");
   menu->AddButton("Real random CC","GetCC(1)",
		   "Read random decalibration calibration coefficients");
   menu->Show();
}

//------------------------------------------------------------------------
Help()
{
  char *string =
    "\nSet calibration parameters and write them into ALICE CDB.
Press button \"Equal CC\" to create equal pedestals and gain factors.
Press button \"Decalibrate\" to create random pedestals and gain factors to imitate decalibrated detector\n";
  printf(string);
}

//------------------------------------------------------------------------
SetCC(Int_t flag=0)
{
  // Writing calibration coefficients into the Calibration DB
  // Arguments:
  //   flag=0: all calibration coefficients are equal
  //   flag=1: all calibration coefficients random (decalibration)
  // Author: Boris Polishchuk (Boris.Polichtchouk at cern.ch)

  TString DBFolder;
  Int_t firstRun   =  0;
  Int_t lastRun    = 10;
  Int_t beamPeriod =  1;
  char* objFormat;

  if      (flag == 0) {
    DBFolder  ="local://InitCalibDB";
    firstRun  =  0;
    lastRun   =  0;
    objFormat = "PHOS initial gain factors and pedestals";
  }
  else if (flag == 1) {
    DBFolder  ="local://deCalibDB";
    firstRun  =  0;
    lastRun   = 10;
    objFormat = "PHOS random pedestals and ADC gain factors (5x64x56)";
  }
  
  AliPHOSCalibData *calibda=new AliPHOSCalibData("PHOS");
  
  Float_t fADCpedestalEmc = 0.005;
  Float_t fADCchanelEmc   = 0.0015;
  
  TRandom rn;
  
  for(Int_t module=1; module<6; module++) {
    for(Int_t column=1; column<57; column++) {
      for(Int_t row=1; row<65; row++) {
	if (flag == 1) {
	  // Decalibration:
	  // Spread calibration coefficients uniformly with
	  // Cmax/Cmin = 5, (Cmax-Cmin)/2 = 0.0015
	  // and pedestals 0.005 +-10%
	  fADCchanelEmc  =rn.Uniform(0.00075,0.00375);
	  fADCpedestalEmc=rn.Uniform(0.0045,0.0055);
	}
	calibda->SetADCchannelEmc (module,column,row,fADCchanelEmc);
	calibda->SetADCpedestalEmc(module,column,row,fADCpedestalEmc);
      }
    }
  }

  //Store calibration data into database
  
  AliCDBMetaData md;
  md.SetComment(objFormat);
  md.SetBeamPeriod(beamPeriod);
  md.SetResponsible("Boris Polichtchouk");
  
  AliCDBId id("PHOS/Calib/GainFactors_and_Pedestals",firstRun,lastRun);

  AliCDBManager* man = AliCDBManager::Instance();  
  AliCDBStorage* loc = man->GetStorage(DBFolder.Data());
  loc->Put(calibda, id, &md);

}

//------------------------------------------------------------------------
GetCC(Int_t flag=0)
{
  // Read calibration coefficients into the Calibration DB
  // Arguments:
  //   flag=0: all calibration coefficients are equal
  //   flag=1: all calibration coefficients random (decalibration)
  // Author: Yuri.Kharlov at cern.ch

  TString DBFolder;
  Int_t firstRun   =  0;
  Int_t lastRun    = 10;
  Int_t beamPeriod =  1;
  char* objFormat;

  if      (flag == 0) {
    DBFolder  ="local://InitCalibDB";
  }
  else if (flag == 1) {
    DBFolder  ="local://deCalibDB";
  }

  AliPHOSCalibData* clb  = (AliPHOSCalibData*)(AliCDBManager::Instance()
    ->GetStorage(DBFolder.Data())->Get("PHOS/Calib/GainFactors_and_Pedestals",
    gAlice->GetRunNumber())->GetObject());

  static const Int_t nMod =  5;
  static const Int_t nCol = 56;
  static const Int_t nRow = 64;

  TH2F *hPed[nMod], *hGain[nMod];
  TCanvas *cPed  = new TCanvas("cPed" ,"Pedestals"   , 10,10,400,800);
  TCanvas *cGain = new TCanvas("cGain","Gain factors",410,10,400,800);
  cPed ->Divide(1,5);
  cGain->Divide(1,5);

  for (Int_t module=1; module<=nMod; module++) {
    TString namePed="hPed";
    namePed+=module;
    TString titlePed="Pedestals in module ";
    titlePed+=module;
    hPed[module-1] = new TH2F(namePed.Data(),titlePed.Data(),
			    nCol,1.,1.*nCol,nRow,1.,1.*nRow);

    TString nameGain="hGain";
    nameGain+=module;
    TString titleGain="Gain factors in module ";
    titleGain+=module;
    hGain[module-1] = new TH2F(nameGain.Data(),titleGain.Data(),
			    nCol,1.,1.*nCol,nRow,1.,1.*nRow);

    for (Int_t column=1; column<=nCol; column++) {
      for (Int_t row=1; row<=nRow; row++) {
	Float_t ped  = clb->GetADCpedestalEmc(module,column,row);
	Float_t gain = clb->GetADCchannelEmc (module,column,row);
	hPed[module-1] ->SetBinContent(column,row,ped);
	hGain[module-1]->SetBinContent(column,row,gain);
      }
    }
    cPed ->cd(module);
    hPed[module-1] ->Draw("lego2");
    cGain->cd(module);
    hGain[module-1]->Draw("lego2");
  }
  cPed ->Print("pedestals.eps");
  cGain->Print("gains.eps");
}