Introduce an enumerator for PID methods

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

diff --git a/TRD/AliTRDpidESD.cxx b/TRD/AliTRDpidESD.cxx
index dcb38c5f87cb2a9799ff5f262884bf0e886ba880..e607b1066887f5bbd311cff7055f64f71b478a6d 100644 (file)
--- a/TRD/AliTRDpidESD.cxx
+++ b/TRD/AliTRDpidESD.cxx
@@ -37,6 +37,7 @@
 #include "AliTracker.h"
 #include "AliRun.h"
 
+#include "AliTRDReconstructor.h"
 #include "AliTRDpidESD.h"
 #include "AliTRDgeometry.h"
 #include "AliTRDcalibDB.h"
@@ -133,8 +134,14 @@ Int_t AliTRDpidESD::MakePID(AliESDEvent *event)
                return -1;
        }
        
+  AliTRDrecoParam *rec = AliTRDReconstructor::RecoParam();
+  if (!rec) {
+    AliErrorGeneral("AliTRDpidESD::MakePID()", "No TRD reco param.");
+    return 0x0;
+  }
+
        // Retrieve the CDB container class with the probability distributions
-       const AliTRDCalPID *pd = calibration->GetPIDObject(1);
+       const AliTRDCalPID *pd = calibration->GetPIDObject(rec->GetPIDMethod());
        if (!pd) {
                AliErrorGeneral("AliTRDpidESD::MakePID()"
                        ,"No access to AliTRDCalPID");