Extacting the OCDB in a separate module. The detectors have write permission in the...

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

diff --git a/TRD/Macros/AliTRDpidCDB.C b/TRD/Macros/AliTRDpidCDB.C
index 9dbb6b806d9915d34a3ac41f969271e4f63c632f..bd0c5e0492fca82ab1236106c19346afccc6a1b0 100644 (file)
--- a/TRD/Macros/AliTRDpidCDB.C
+++ b/TRD/Macros/AliTRDpidCDB.C
@@ -16,7 +16,7 @@ void makePIDRefs(const char *dir = ".", const char *file="Refs.root")
 //             "dir" 
 
        AliCDBManager *man = AliCDBManager::Instance();
-       man->SetDefaultStorage("local://$ALICE_ROOT");
+       man->SetDefaultStorage("local://$ALICE_ROOT/OCDB");
        man->SetRun(0);
 
        AliTRDCalPIDRefMaker maker;
@@ -24,34 +24,42 @@ void makePIDRefs(const char *dir = ".", const char *file="Refs.root")
 }
 
 //___________________________________________________________________
-void generatePIDDB(const char *file = "Refs.root")
+void generatePIDDB(const char *fileNN = "NN.root", const char *fileLQ = "LQ.root")
 {
 // Write TRD PID DB using the reference data from file "file"
 
 
        AliCDBManager *man = AliCDBManager::Instance();
-       man->SetDefaultStorage("local://$ALICE_ROOT");
+       man->SetDefaultStorage("local://$ALICE_ROOT/OCDB");
        man->SetRun(0);
 
-       AliCDBStorage *gStorLoc = man->GetStorage("local://$ALICE_ROOT");
-  if (!gStorLoc) return;
+       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();
-  md->SetObjectClassName("AliTRDCalPIDLQ");
-  md->SetResponsible("Alex Bercuci");
-  md->SetBeamPeriod(1);
-  md->SetAliRootVersion("v4-06-HEAD"); //root version
-  md->SetComment("2D PID for TRD");
-
-       gStorLoc->Put(pid, AliCDBId("TRD/Calib/PIDLQ", 0, 0), md);
+       AliTRDCalPID *pidLQ = new AliTRDCalPIDLQ("pidLQ", "LQ TRD PID object");
+       pidLQ->LoadReferences(fileLQ);
+       AliCDBMetaData *md= new AliCDBMetaData();
+       md->SetObjectClassName("AliTRDCalPIDLQ");
+       md->SetResponsible("Alex Bercuci");
+       md->SetBeamPeriod(1);
+       md->SetAliRootVersion("v4-06-HEAD"); //root version
+       md->SetComment("2D PID for TRD");
+       gStorLoc->Put(pidLQ, AliCDBId("TRD/Calib/PIDLQ", 0, 999999999, 0, 1), md);
+       
+       AliTRDCalPID *pidNN = new AliTRDCalPIDNN("pidNN", "NN TRD PID object");
+       pidNN->LoadReferences(fileNN);
+       md->SetObjectClassName("AliTRDCalPIDNN");
+       md->SetResponsible("Alex Wilk");
+       md->SetBeamPeriod(1);
+       md->SetAliRootVersion("v4-06-HEAD"); //root version
+       md->SetComment("NN PID for TRD");
+       
+       gStorLoc->Put(pidNN, AliCDBId("TRD/Calib/PIDNN", 0, 999999999, 0, 1), md);
 }
 
 //___________________________________________________________________
-AliTRDCalPID* getPIDObject()
+AliTRDCalPID* getPIDObject(const char *method="NN")
 {
 // Returns PIDLQ object.
 // In order to browse histos do:
@@ -61,10 +69,10 @@ AliTRDCalPID* getPIDObject()
        gStyle->SetOptStat(0);
        
   AliCDBManager *CDBManager = AliCDBManager::Instance();
-       CDBManager->SetDefaultStorage("local://$ALICE_ROOT");
+       CDBManager->SetDefaultStorage("local://$ALICE_ROOT/OCDB");
        CDBManager->SetRun(0);
 
-       AliCDBEntry *wrap = CDBManager->Get("TRD/Calib/PIDLQ", 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){