TRD PID included in the ESD schema (T.Kuhr)

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

diff --git a/STEER/AliESDtrack.cxx b/STEER/AliESDtrack.cxx
index ea507e889685c2935f8b2b5f573fd0055eca9882..8a4e2fbc7942ef43c44cda72fb5dd88a704a015a 100644 (file)
--- a/STEER/AliESDtrack.cxx
+++ b/STEER/AliESDtrack.cxx
@@ -77,6 +77,11 @@ Bool_t AliESDtrack::UpdateTrackParams(const AliKalmanTrack *t, ULong_t flags) {
     else if (mass<0.4) fR[2]=1.;    // the ITS reconstruction
     else fR[3]=1.;}                 //
     break;
+  case kTRDin: case kTRDout: case kTRDrefit:
+    fTRDncls=t->GetNumberOfClusters();
+    fTRDchi2=t->GetChi2();
+    fTRDsignal=t->GetPIDsignal();
+    break;
   default: 
     Error("UpdateTrackParams()","Wrong flag !\n");
     return kFALSE;
@@ -180,6 +185,17 @@ void AliESDtrack::GetTPCpid(Double_t *p) const {
   for (Int_t i=0; i<kSPECIES; i++) p[i]=fTPCr[i];
 }
 
+//_______________________________________________________________________
+void    AliESDtrack::SetTRDpid(Int_t iSpecies, Float_t p)
+{
+  fTRDr[iSpecies] = p;
+}
+
+Float_t AliESDtrack::GetTRDpid(Int_t iSpecies) const
+{
+  return fTRDr[iSpecies];
+}
+
 //_______________________________________________________________________
 void AliESDtrack::SetESDpid(const Double_t *p) {  
   for (Int_t i=0; i<kSPECIES; i++) fR[i]=p[i];