// Functions to create/store/inspect TRD PID DB.
// 
// Author:
// Alex Bercuci (A.Bercuci@gsi.de)
// 



//___________________________________________________________________
void generatePIDDBLQ(const char *file = "TRD.CalibPIDrefMakerLQ.root")
{
// Write TRD PID DB using the reference data from file "file"


  AliCDBManager *man = AliCDBManager::Instance();
  man->SetDefaultStorage("local://$ALICE_ROOT/OCDB");
  man->SetSpecificStorage("TRD/Calib/PIDLQ", "local://.");
  man->SetRun(0);

  AliCDBStorage *gStorLoc = man->GetStorage("local://$ALICE_ROOT/OCDB");
  if (!gStorLoc) return;

  if(gSystem->Load("libSTAT.so")<0) return;
  AliTRDCalPID *pid = new AliTRDCalPIDLQ("pidLQ", "LQ TRD PID object");    
  if(!pid->LoadReferences(file)) return;
  AliCDBMetaData *md= new AliCDBMetaData();
  md->SetObjectClassName("AliTRDCalPIDLQ");
  md->SetResponsible("Alexandru Bercuci");
  md->SetBeamPeriod(1);
  md->SetAliRootVersion("v4-17-Release"); //root version
  md->SetComment("2D PID for TRD");
  gStorLoc->Put(pid, AliCDBId("TRD/Calib/PIDLQ", 0, 999999999, 0, 1), md, AliCDBManager::kReference);
}

//___________________________________________________________________
void generatePIDDBNN(const char *fileNN = "NN.root")
{
// Write TRD PID DB using the reference data from file "file"


  AliCDBManager *man = AliCDBManager::Instance();
  man->SetDefaultStorage("local://$ALICE_ROOT/OCDB");
  man->SetSpecificStorage("TRD/Calib/PIDNN", "local://.");
  man->SetRun(0);

  AliCDBStorage *gStorLoc = man->GetStorage("local://$ALICE_ROOT/OCDB");
  if (!gStorLoc) return;

  AliTRDCalPID *pidNN = new AliTRDCalPIDNN("pidNN", "NN TRD PID object"); 	 
  pidNN->LoadReferences(fileNN); 	 
  AliCDBMetaData *md= new AliCDBMetaData();
  md->SetObjectClassName("AliTRDCalPIDNN");
  md->SetResponsible("Alexander Wilk");
  md->SetBeamPeriod(1);
  md->SetAliRootVersion("v4-16-Release"); //root version
  md->SetComment("NN PID for TRD");
  gStorLoc->Put(pidNN, AliCDBId("TRD/Calib/PIDNN", 0, 999999999, 0, 1), md, AliCDBManager::kReference);
}

//___________________________________________________________________
AliTRDCalPID* getPIDObject(const char *method="NN")
{
// Returns PIDLQ object.
// In order to browse histos do:
//   > AliTRDCalPID *pid = getPIDObject();
//   > pid->GetHistogram(0, 3);

  gStyle->SetOptStat(0);
  
  AliCDBManager *CDBManager = AliCDBManager::Instance();
  CDBManager->SetDefaultStorage("local://$ALICE_ROOT/OCDB");
  CDBManager->SetRun(0);

  AliCDBEntry *wrap = CDBManager->Get(Form("TRD/Calib/PID%s", method), 0);
  AliTRDCalPID *pid = dynamic_cast<const AliTRDCalPID *>wrap->GetObject();
  AliCDBMetaData *meta = wrap->GetMetaData();
  if(!pid){
    printf("Error while retriving pid object from DB.\n");
    return 0x0;
  }
  meta->PrintMetaData();
  return pid;
}