Changes for report #65872 Request for adding changes in AliESDpid for trunk. We added...

[u/mrichter/AliRoot.git] / STEER / AliESDpid.cxx

diff --git a/STEER/AliESDpid.cxx b/STEER/AliESDpid.cxx
index 0ef62acbbd14aaa561576e38ffe7b593a01dd115..280f11a5a8176ea3a7482417acdd7a1fdc3e4492 100644 (file)
--- a/STEER/AliESDpid.cxx
+++ b/STEER/AliESDpid.cxx
@@ -215,6 +215,24 @@ void AliESDpid::MakeTOFPID(AliESDtrack *track, Float_t TimeZeroTOF) const
   if (heavy) track->ResetStatus(AliESDtrack::kTOFpid);    
 }
 //_________________________________________________________________________
+void AliESDpid::MakeTRDPID(AliESDtrack *track) const
+{
+  //
+  // Method to recalculate the TRD PID probabilities
+  //
+  if((track->GetStatus()&AliESDtrack::kTRDout)==0) return;
+  Double_t prob[AliPID::kSPECIES]; Float_t mom[6];
+  Double_t dedx[48];  // Allocate space for the maximum number of TRD slices
+  for(Int_t ilayer = 0; ilayer < 6; ilayer++){
+    mom[ilayer] = track->GetTRDmomentum(ilayer);
+    for(Int_t islice = 0; islice < track->GetNumberOfTRDslices(); islice++){
+      dedx[ilayer*track->GetNumberOfTRDslices()+islice] = track->GetTRDslice(ilayer, islice);
+    }
+  }
+  fTRDResponse.GetResponse(track->GetNumberOfTRDslices(), dedx, mom, prob);
+  track->SetTRDpid(prob);
+}
+//_________________________________________________________________________
 void AliESDpid::CombinePID(AliESDtrack *track) const
 {
   //