Remove a run from LHC10e wo AODs; added run list for the new PbPb MC LHC11a10b_bis

[u/mrichter/AliRoot.git] / TRD / Macros / AliTRDpidCDB.C

diff --git a/TRD/Macros/AliTRDpidCDB.C b/TRD/Macros/AliTRDpidCDB.C
index 9dbb6b806d9915d34a3ac41f969271e4f63c632f..3ed11235399b94121ae67db86c0a1c143d7a2e8b 100644 (file)
--- a/TRD/Macros/AliTRDpidCDB.C
+++ b/TRD/Macros/AliTRDpidCDB.C
@@ -5,73 +5,80 @@
 // 
 
 
-//___________________________________________________________________
-void makePIDRefs(const char *dir = ".", const char *file="Refs.root")
-{
-// Build the reference data for PID. The simulations have to fulfill
-// the directory structure defined inside AliTRDCalPIDRefMaker.
-// Parameters:
-// 1. "dir" - the root directory of the production
-// 2. "file" - output file containing reference data saved in directory
-//             "dir" 
-
-       AliCDBManager *man = AliCDBManager::Instance();
-       man->SetDefaultStorage("local://$ALICE_ROOT");
-       man->SetRun(0);
-
-       AliTRDCalPIDRefMaker maker;
-       maker.BuildLQReferences(file, dir);
-}
 
 //___________________________________________________________________
-void generatePIDDB(const char *file = "Refs.root")
+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");
-       man->SetRun(0);
+  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");
+  AliCDBStorage *gStorLoc = man->GetStorage("local://$ALICE_ROOT/OCDB");
   if (!gStorLoc) return;
 
-  
-       AliTRDCalPID *pid = new AliTRDCalPID("pid", "TRD PID object");
-       pid->LoadLQReferences(file);
-
-       AliCDBMetaData *md= new AliCDBMetaData();
+  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("Alex Bercuci");
+  md->SetResponsible("Alexandru Bercuci");
   md->SetBeamPeriod(1);
-  md->SetAliRootVersion("v4-06-HEAD"); //root version
+  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);
+}
 
-       gStorLoc->Put(pid, AliCDBId("TRD/Calib/PIDLQ", 0, 0), md);
+//___________________________________________________________________
+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()
+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);
-       
+  gStyle->SetOptStat(0);
+  
   AliCDBManager *CDBManager = AliCDBManager::Instance();
-       CDBManager->SetDefaultStorage("local://$ALICE_ROOT");
-       CDBManager->SetRun(0);
-
-       AliCDBEntry *wrap = CDBManager->Get("TRD/Calib/PIDLQ", 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;
+  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;
 }