Patch to load 1DLQ PID references

[u/mrichter/AliRoot.git] / TRD / AliTRDcalibDB.cxx

diff --git a/TRD/AliTRDcalibDB.cxx b/TRD/AliTRDcalibDB.cxx
index 2559f83c24cee93371882c095d49981c3ef5ae16..8571a2784845ac233b0a5718803b1378679fc3c0 100644 (file)
--- a/TRD/AliTRDcalibDB.cxx
+++ b/TRD/AliTRDcalibDB.cxx
@@ -36,6 +36,7 @@
 #include "AliLog.h"
 
 #include "AliTRDPIDReference.h"
+#include "AliTRDPIDResponseObject.h"
 #include "AliTRDcalibDB.h"
 #include "AliTRDtrapConfig.h"
 #include "AliTRDtrapConfigHandler.h"
@@ -279,7 +280,11 @@ const TObject *AliTRDcalibDB::GetCachedCDBObject(Int_t id)
           break;
         case 7:
          // Online gain table ID 7
-          return CacheCDBEntry(kIDOnlineGainFactor  ,"TRD/Calib/Gaintbl_Uniform_FGAN8_2012-01"); 
+          return CacheCDBEntry(kIDOnlineGainFactor  ,"TRD/Calib/Gaintbl_Uniform_FGAN8_2012-01");
+          break; 
+        case 8:
+         // Online gain table ID 8
+          return CacheCDBEntry(kIDOnlineGainFactor  ,"TRD/Calib/Krypton_2012-01"); 
           break;
       }
       break;
@@ -902,19 +907,11 @@ Float_t AliTRDcalibDB::GetPRFWidth(Int_t det, Int_t col, Int_t row)
 }
   
 //_____________________________________________________________________________
-Int_t AliTRDcalibDB::GetNumberOfTimeBinsDCS()
+Int_t AliTRDcalibDB::ExtractTimeBinsFromString(TString tbstr)
 {
-  //
-  // Returns number of time bins from the DCS
-  //
-
   // default value - has not been set
   Int_t nUndef = -1;
 
-  // Get the corresponding parameter
-  TString tbstr = "";
-  GetDCSConfigParOption(kTimebin, 0, tbstr);
-
   // Check if there is any content in the string first
   if (tbstr.Length() == 0) {
     AliError("Parameter for number of timebins is empty!");
@@ -936,10 +933,23 @@ Int_t AliTRDcalibDB::GetNumberOfTimeBinsDCS()
     return nUndef;
   }
 
-  Int_t ntb = tbstr.Atoi();
-  AliInfo(Form("Number of timebins from CDB: %d", ntb));
+  return tbstr.Atoi();
+
+}
 
-  return ntb;
+//_____________________________________________________________________________
+Int_t AliTRDcalibDB::GetNumberOfTimeBinsDCS()
+{
+  //
+  // Returns number of time bins from the DCS
+  //
+
+  // Get the corresponding parameter
+  TString cfgstr = "", cfgname = "";
+  GetGlobalConfiguration(cfgname);
+  GetDCSConfigParOption(cfgname, kTimebin, 0, cfgstr);
+
+  return ExtractTimeBinsFromString(cfgstr);
 
 }
 
@@ -950,7 +960,9 @@ void AliTRDcalibDB::GetFilterType(TString &filterType)
   // Returns the filter type
   //
 
-  GetDCSConfigParOption(kFltrSet, 0, filterType);
+  TString cfgname = "";
+  GetGlobalConfiguration(cfgname);
+  GetDCSConfigParOption(cfgname, kFltrSet, 0, filterType);
 
 }
 
@@ -1025,6 +1037,10 @@ Int_t AliTRDcalibDB::GetOnlineGainTableID()
       fOnlineGainTableID = 7;
       return fOnlineGainTableID;
     }
+    if (tableName.CompareTo("Krypton_2011-03")               == 0) {
+      fOnlineGainTableID = 8;
+      return fOnlineGainTableID;
+    }
 
   } 
   else {
@@ -1163,7 +1179,41 @@ void AliTRDcalibDB::GetGlobalConfigurationVersion(TString &version)
 }
 
 //_____________________________________________________________________________
-void AliTRDcalibDB::GetDCSConfigParOption(Int_t cfgType, Int_t option, TString &cfgo)
+Int_t AliTRDcalibDB::GetNumberOfParsDCS(TString cname)
+{
+  // Get the number of configuration parameters from the DCS config
+
+  TString cdelim = "_";
+  TObjArray *carr = cname.Tokenize(cdelim);
+  return carr->GetEntries() - 1; // -1 for the "cf"
+
+}
+
+//_____________________________________________________________________________
+Int_t AliTRDcalibDB::GetNumberOfOptsDCS(TString cname, Int_t cfgType)
+{
+  // Get the number of options of a given configuration parameter from DCS
+
+  TString cdelim = "_";
+  TString odelim = "-";
+
+  TObjArray *carr = cname.Tokenize(cdelim);
+  Int_t nconfig = carr->GetEntries();
+
+  // protect
+  if ((nconfig == 0) || ((nconfig-1) < cfgType)) {
+    AliError("Not enough parameters in DCS configuration name!");
+    return 0;
+  }
+
+  TString fullcfg = ((TObjString*)carr->At(cfgType))->GetString();
+  TObjArray *oarr = fullcfg.Tokenize(odelim);
+  return oarr->GetEntries() -1; // -1 for the parameter name
+
+}
+
+//_____________________________________________________________________________
+void AliTRDcalibDB::GetDCSConfigParOption(TString cname, Int_t cfgType, Int_t option, TString &cfgo)
 {
   //
   // Get a configuration (see enum in header file) or the options of a configuration
@@ -1175,52 +1225,33 @@ void AliTRDcalibDB::GetDCSConfigParOption(Int_t cfgType, Int_t option, TString &
   TString cdelim = "_";
   TString odelim = "-";
 
-  // get the full configuration name
-  TString cname;
-  GetGlobalConfiguration(cname);
   TObjArray *carr = cname.Tokenize(cdelim);
   Int_t nconfig = carr->GetEntries();
 
   // protect
   if (nconfig == 0) {
-    AliError("Bad DCS configuration name!");
+    AliError("DCS configuration name empty!");
     cfgo = "";
     return;
   } else if ((nconfig-1) < cfgType) {
-    AliError("Not enough DCS configuration parameters!");
+    AliError("Not enough parameters in DCS configuration name!");
     cfgo = "";
     return;
   }
 
   TString fullcfg = ((TObjString*)carr->At(cfgType))->GetString();
+  TObjArray *oarr = fullcfg.Tokenize(odelim);
+  Int_t noptions = oarr->GetEntries();
 
-  if (fullcfg.Contains(odelim)) {
-
-    TObjArray *oarr = fullcfg.Tokenize(odelim);
-    Int_t noptions = oarr->GetEntries();
-
-    // protect
-    if ((noptions-1) < option) {
-      AliError("Not enough DCS configuration options defined!");
-      cfgo = "";
-      return;
-    }
-
-    cfgo = ((TObjString*)oarr->At(option))->GetString();
+  // protect
+  if ((noptions-1) < option) {
+    AliError("Not enough options in DCS configuration name!");
+    cfgo = "";
     return;
-
   }
-  else {
-
-    if (option != 0) {
-      AliError("Not enough DCS configuration options defined!");
-      cfgo = "";
-      return;
-    }
-    cfgo = fullcfg;
-    return;
 
-  }
+  cfgo = ((TObjString*)oarr->At(option))->GetString();
+  return;
 
 }
 
@@ -1513,23 +1544,27 @@ AliTRDPIDResponse *AliTRDcalibDB::GetPIDResponse(AliTRDPIDResponse::ETRDPIDMetho
     fPIDResponse = new AliTRDPIDResponse;
 
     // Load Reference Histos from OCDB
-    fPIDResponse->SetPIDmethod(method);
-    const TObjArray *references = dynamic_cast<const TObjArray *>(GetCachedCDBObject(kIDPIDLQ1D));
-
-    TIter refs(references);
-    TObject *obj = NULL;
-    AliTRDPIDReference *ref = NULL;
-    Bool_t hasReference = kFALSE;
-    while ((obj = refs())){
-      if ((ref = dynamic_cast<AliTRDPIDReference *>(obj))){
-        fPIDResponse->Load(ref);
-        hasReference = kTRUE;
-        break;
+    if(method == AliTRDPIDResponse::kLQ1D){
+      fPIDResponse->SetPIDmethod(method);
+      const TObjArray *references = dynamic_cast<const TObjArray *>(GetCachedCDBObject(kIDPIDLQ1D));
+
+      TIter refs(references);
+      TObject *obj = NULL;
+      AliTRDPIDReference *ref = NULL;
+      Bool_t hasReference = kFALSE;
+      while ((obj = refs())){
+        if ((ref = dynamic_cast<AliTRDPIDReference *>(obj))){
+          AliTRDPIDResponseObject *ro = new AliTRDPIDResponseObject;
+          ro->SetPIDReference(ref);
+          fPIDResponse->SetPIDResponseObject(ro);
+          hasReference = kTRUE;
+          break;
+        }
       }
-    }
 
-    if (!hasReference) {
-      AliError("Reference histograms not found in the OCDB");
+      if (!hasReference) {
+        AliError("Reference histograms not found in the OCDB");
+      }
     }
 
   }