First implementation of neural network PID

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

diff --git a/TRD/AliTRDgtuTrack.cxx b/TRD/AliTRDgtuTrack.cxx
index abbb80977a8706bbd27f4b3c2948ce55086bdeac..3b7335bfdccbaeec20bf238f2d9136234b43f145 100644 (file)
--- a/TRD/AliTRDgtuTrack.cxx
+++ b/TRD/AliTRDgtuTrack.cxx
@@ -390,7 +390,7 @@ void AliTRDgtuTrack::MakePID()
     AliError("No instance of AliTRDcalibDB.");
     return;  
   }
-  const AliTRDCalPID *pd = calibration->GetPIDLQObject();
+  const AliTRDCalPID *pd = calibration->GetPIDObject(1);
   
   AliTRDltuTracklet *trk;
   Int_t   nTracklets = GetNtracklets();
@@ -455,8 +455,8 @@ void AliTRDgtuTrack::MakePID()
                 dedx[0] = dedx[1] = q*3.; dedx[2] = 0.;
                 Float_t length = 3.7;
 
-    probEle *= pd->GetProbability(0, TMath::Abs(fPt), dedx, length);
-    probPio *= pd->GetProbability(2, TMath::Abs(fPt), dedx, length);
+    probEle *= pd->GetProbability(0, TMath::Abs(fPt), dedx, length, 0);
+    probPio *= pd->GetProbability(2, TMath::Abs(fPt), dedx, length, 0);
 
   }