add DCA switch to track propagation function

sevdokim · Mar 14, 2017 · 381e6db · 381e6db
1 parent 7a019d3
commit 381e6db
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/EMCAL/EMCALUtils/AliEMCALRecoUtils.cxx b/EMCAL/EMCALUtils/AliEMCALRecoUtils.cxx
@@ -2743,7 +2743,7 @@ Int_t  AliEMCALRecoUtils::FindMatchedClusterInClusterArr(const AliExternalTrackP
 Bool_t AliEMCALRecoUtils::ExtrapolateTrackToEMCalSurface(AliVTrack *track,
                                                          Double_t emcalR, Double_t mass,
                                                          Double_t step, Double_t minpt,
-                                                         Bool_t useMassForTracking)
+                                                         Bool_t useMassForTracking, Bool_t useDCA)
 { 
   track->SetTrackPhiEtaPtOnEMCal(-999, -999, -999);
 
@@ -2805,8 +2805,14 @@ Bool_t AliEMCALRecoUtils::ExtrapolateTrackToEMCalSurface(AliVTrack *track,
   else 
   {
     Double_t xyz[3] = {0}, pxpypz[3] = {0}, cv[21] = {0};
-    aodt->PxPyPz(pxpypz);  
-    aodt->XvYvZv(xyz);
+    aodt->PxPyPz(pxpypz);
+    if (useDCA) {
+      // Note: useDCA is by default false in this function only for backwards compatibility
+      aodt->GetXYZ(xyz);
+    }
+    else {
+      aodt->XvYvZv(xyz);
+    }
     aodt->GetCovarianceXYZPxPyPz(cv);  
     trackParam = new AliExternalTrackParam(xyz,pxpypz,cv,aodt->Charge());
   }

diff --git a/EMCAL/EMCALUtils/AliEMCALRecoUtils.h b/EMCAL/EMCALUtils/AliEMCALRecoUtils.h
@@ -296,7 +296,7 @@ class AliEMCALRecoUtils : public TNamed {
   static Bool_t ExtrapolateTrackToEMCalSurface(AliVTrack *track, /*note, on success the call will change the track*/
                                                Double_t emcalR=440, Double_t mass=0.1396,
                                                Double_t step=20, Double_t minpT=0.35,
-                                               Bool_t useMassForTracking = kFALSE);
+                                               Bool_t useMassForTracking = kFALSE, Bool_t useDCA = kFALSE);
   static Bool_t ExtrapolateTrackToEMCalSurface(AliExternalTrackParam *trkParam, 
                                                Double_t emcalR, Double_t mass, Double_t step, 
                                                Float_t &eta, Float_t &phi, Float_t &pt);