From 70c961fc8678a0b6666b3a1a37f0bb67c3579f16 Mon Sep 17 00:00:00 2001
From: noferini <9963644+noferini@users.noreply.github.com>
Date: Sun, 28 Apr 2024 17:10:13 +0200
Subject: [PATCH] return dca in fast propagator

---
 Detectors/Base/include/DetectorsBase/Propagator.h | 2 +-
 Detectors/Base/src/Propagator.cxx                 | 6 ++++--
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/Detectors/Base/include/DetectorsBase/Propagator.h b/Detectors/Base/include/DetectorsBase/Propagator.h
index 4c9424401143a..a9e2ce6e0383d 100644
--- a/Detectors/Base/include/DetectorsBase/Propagator.h
+++ b/Detectors/Base/include/DetectorsBase/Propagator.h
@@ -136,7 +136,7 @@ class PropagatorImpl
   GPUd() void setNominalBz(value_type bz) { mNominalBz = bz; }
   GPUd() bool hasMagFieldSet() const { return mField != nullptr; }
 
-  GPUd() void estimateLTFast(o2::track::TrackLTIntegral& lt, const o2::track::TrackParametrization<value_type>& trc) const;
+  GPUd() value_type estimateLTFast(o2::track::TrackLTIntegral& lt, const o2::track::TrackParametrization<value_type>& trc) const;
   GPUd() float estimateLTIncrement(const o2::track::TrackParametrization<value_type>& trc, const o2::math_utils::Point3D<value_type>& postStart, const o2::math_utils::Point3D<value_type>& posEnd) const;
 
 #ifndef GPUCA_GPUCODE
diff --git a/Detectors/Base/src/Propagator.cxx b/Detectors/Base/src/Propagator.cxx
index 5e932ebe55038..0d1b53b695536 100644
--- a/Detectors/Base/src/Propagator.cxx
+++ b/Detectors/Base/src/Propagator.cxx
@@ -668,7 +668,7 @@ GPUd() float PropagatorImpl<value_T>::estimateLTIncrement(const o2::track::Track
 
 //____________________________________________________________
 template <typename value_T>
-GPUd() void PropagatorImpl<value_T>::estimateLTFast(o2::track::TrackLTIntegral& lt, const o2::track::TrackParametrization<value_type>& trc) const
+GPUd() value_T PropagatorImpl<value_T>::estimateLTFast(o2::track::TrackLTIntegral& lt, const o2::track::TrackParametrization<value_type>& trc) const
 {
   value_T xdca = 0., ydca = 0., length = 0.; // , zdca = 0. // zdca might be used in future
   o2::math_utils::CircleXY<value_T> c;
@@ -715,8 +715,10 @@ GPUd() void PropagatorImpl<value_T>::estimateLTFast(o2::track::TrackLTIntegral&
     length = straigh_line_approx();
   }
   // since we assume the track or its parent comes from the beam-line or decay, add XY(?) distance to it
-  length += math_utils::detail::sqrt<value_type>(xdca * xdca + ydca * ydca);
+  value_T dcaT = math_utils::detail::sqrt<value_type>(xdca * xdca + ydca * ydca);
+  length += dcaT;
   lt.addStep(length, trc.getP2Inv());
+  return dcaT;
 }
 
 //____________________________________________________________