new strategy for TRD-PID ref maker (AlexW & AlexB)

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

diff --git a/TRD/Macros/AliTRDpidCDB.C b/TRD/Macros/AliTRDpidCDB.C
index b4985af73b9e01c114c0c655722317c8b9b3ea5e..b0c4871f7eda03849986ffd7b8c20f9d4ec542e6 100644 (file)
--- a/TRD/Macros/AliTRDpidCDB.C
+++ b/TRD/Macros/AliTRDpidCDB.C
@@ -48,6 +48,31 @@ void generatePIDDB()
   gStorLoc->Put(o, 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")
 {