fPIDForTracking can be also e or mu. Their overriding to pion happens in the

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

diff --git a/STEER/ESD/AliESDtrack.cxx b/STEER/ESD/AliESDtrack.cxx
index d2bd607cec9726a01d03bbd614d2e2f1629de9c2..00df3b23e2f330a7bbf9bb663982eb5d0487e5f0 100644 (file)
--- a/STEER/ESD/AliESDtrack.cxx
+++ b/STEER/ESD/AliESDtrack.cxx
@@ -818,7 +818,7 @@ AliESDtrack::AliESDtrack(TParticle * part) :
   if (pdgCode<0) pdgCode = -pdgCode;
   for (i=0;i<AliPID::kSPECIESC;i++) if (pdgCode==AliPID::ParticleCode(i)) {indexPID = i; break;}
 
-  if (indexPID < AliPID::kSPECIES) fPIDForTracking = indexPID;
+  if (indexPID < AliPID::kSPECIESC) fPIDForTracking = indexPID;
 
   // AliESD track label
   SetLabel(part->GetUniqueID());
@@ -2956,7 +2956,9 @@ Double_t AliESDtrack::GetLengthInActiveZone(const AliExternalTrackParam  *paramT
 
 Double_t AliESDtrack::GetMassForTracking() const
 {
-  double m = AliPID::ParticleMass(fPIDForTracking);
+  int pid = fPIDForTracking;
+  if (pid<AliPID::kPion) pid = AliPID::kPion;
+  double m = AliPID::ParticleMass(pid);
   return (fPIDForTracking==AliPID::kHe3 || fPIDForTracking==AliPID::kAlpha) ? -m : m;
 }