option for mass hypothesis also on AODs

diff --git a/EMCAL/AliEMCALRecoUtils.cxx b/EMCAL/AliEMCALRecoUtils.cxx
index 8c9f7b68e36a56772f339e90e9c795e828e327cb..13b469ae0b989fcbb9189422441bbdf91cdf30f3 100644 (file)
--- a/EMCAL/AliEMCALRecoUtils.cxx
+++ b/EMCAL/AliEMCALRecoUtils.cxx
@@ -2025,16 +2025,22 @@ Bool_t AliEMCALRecoUtils::ExtrapolateTrackToEMCalSurface(AliVTrack *track,
       return kFALSE;
   }
 
-  if (mass<0) {
+  // Select the mass hypothesis
+  if ( mass < 0 )
+  {
     Bool_t onlyTPC = kFALSE;
-    if (mass==-99)
-      onlyTPC=kTRUE;
-    if (esdt) {
+    if ( mass == -99 ) onlyTPC=kTRUE;
+    
+    if (esdt)
+    {
       if ( useMassForTracking ) mass = esdt->GetMassForTracking();
       else                      mass = esdt->GetMass(onlyTPC);
     }
     else
-      mass = aodt->M();
+    {
+      if ( useMassForTracking ) mass = aodt->GetMassForTracking();
+      else                      mass = aodt->M();
+    }
   }
 
   AliExternalTrackParam *trackParam = 0;