Protection against not accessible input OCDB (Laurent)

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

diff --git a/TRD/AliTRDpidESD.cxx b/TRD/AliTRDpidESD.cxx
index 17442854699822a1e014d32763317d863eefb48e..2ffc94af2091af9ba2f5940d1a691b6246c2ef5e 100644 (file)
--- a/TRD/AliTRDpidESD.cxx
+++ b/TRD/AliTRDpidESD.cxx
@@ -41,7 +41,7 @@
 #include "AliTRDcalibDB.h"
 #include "AliRun.h"
 #include "AliTRDtrack.h"
-#include "Cal/AliTRDCalPIDLQ.h"
+#include "Cal/AliTRDCalPID.h"
 
 
 ClassImp(AliTRDpidESD)
@@ -116,7 +116,7 @@ Int_t AliTRDpidESD::MakePID(AliESDEvent *event)
   // The method produces probabilities based on the charge
   // and the position of the maximum time bin in each layer.
   // The dE/dx information can be used as global charge or 2 to 3
-  // slices. Check AliTRDCalPIDLQ and AliTRDCalPIDLQRef for the actual
+  // slices. Check AliTRDCalPID and AliTRDCalPIDRefMaker for the actual
   // implementation.
   //
   // Author
@@ -130,10 +130,10 @@ Int_t AliTRDpidESD::MakePID(AliESDEvent *event)
        }
        
        // Retrieve the CDB container class with the probability distributions
-       const AliTRDCalPIDLQ *pd = calibration->GetPIDLQObject();
+       const AliTRDCalPID *pd = calibration->GetPIDObject(1);
        if (!pd) {
                AliErrorGeneral("AliTRDpidESD::MakePID()"
-                       ,"No access to AliTRDCalPIDLQ");
+                       ,"No access to AliTRDCalPID");
                return -1;
        }
 
@@ -180,7 +180,7 @@ Int_t AliTRDpidESD::MakePID(AliESDEvent *event)
 
                        // Get the probabilities for the different particle species
                        for (Int_t iSpecies = 0; iSpecies < AliPID::kSPECIES; iSpecies++) {
-                               p[iSpecies] *= pd->GetProbability(iSpecies, mom, dedx, length);
+                               p[iSpecies] *= pd->GetProbability(iSpecies, mom, dedx, length, iPlan);
                                //p[iSpecies] *= pd->GetProbabilityT(iSpecies, mom, timebin);
                        }
                }