restore first track

timadye · Apr 6, 2024 · 3d3c37f · 3d3c37f
1 parent 6940f34
commit 3d3c37f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/Core/include/Acts/TrackFinding/CombinatorialKalmanFilter.hpp b/Core/include/Acts/TrackFinding/CombinatorialKalmanFilter.hpp
@@ -671,7 +671,8 @@ class CombinatorialKalmanFilter {
           // TrackState. No storage allocation for uncalibrated/calibrated
           // measurement and filtered parameter
           auto stateMask =
-              ~(TrackStatePropMask::Calibrated | TrackStatePropMask::Filtered);
+              ~(TrackStatePropMask::Calibrated | TrackStatePropMask::Filtered |
+                TrackStatePropMask::Smoothed);
 
           // Increment of number of processed states
           tipState.nStates++;

diff --git a/Examples/Algorithms/TrackFinding/src/TrackFindingAlgorithm.cpp b/Examples/Algorithms/TrackFinding/src/TrackFindingAlgorithm.cpp
@@ -257,6 +257,9 @@ ActsExamples::ProcessCode ActsExamples::TrackFindingAlgorithm::execute(
                       extrapolationOptions, m_cfg.extrapolationStrategy,
                       logger());
               if (!secondExtrapolationResult.ok()) {
+                // restore first track
+                (*firstFirstState).previous() = Acts::kTrackIndexInvalid;
+
                 m_nFailedExtrapolation++;
                 ACTS_ERROR("Second extrapolation for seed "
                            << iSeed << " and track " << secondTrack.index()
@@ -271,6 +274,9 @@ ActsExamples::ProcessCode ActsExamples::TrackFindingAlgorithm::execute(
                 destProxy.copyFrom(secondTrack, true);
               }
 
+              // restore first track
+              (*firstFirstState).previous() = Acts::kTrackIndexInvalid;
+
               ++nSecond;
             }
           }