fix: Output correct track objects in GSF refitting

Examples/Algorithms/TrackFitting/include/ActsExamples/TrackFitting/RefittingAlgorithm.hpp

@@ -33,6 +33,8 @@ class RefittingAlgorithm final : public IAlgorithm {
     std::shared_ptr<TrackFitterFunction> fit;
     /// Pick a single track for debugging (-1 process all tracks)
     int pickTrack = -1;
+    /// Inflate initial covariance.
+    std::array<double, 6> initialVarInflation = {1., 1., 1., 1., 1., 1.};
   };
 
   /// Constructor of the fitting algorithm

Examples/Algorithms/TrackFitting/src/RefittingAlgorithm.cpp

@@ -80,10 +80,15 @@ ActsExamples::ProcessCode ActsExamples::RefittingAlgorithm::execute(
         Acts::PropagatorPlainOptions(ctx.geoContext, ctx.magFieldContext)};
     options.doRefit = true;
 
-    const Acts::BoundTrackParameters initialParams(
+    Acts::BoundTrackParameters initialParams(
         track.referenceSurface().getSharedPtr(), track.parameters(),
         track.covariance(), track.particleHypothesis());
 
+    // Inflate covariance
+    for (auto i = 0ul; i < m_cfg.initialVarInflation.size(); ++i) {
+      initialParams.covariance().value() *= m_cfg.initialVarInflation.at(i);
+    }
+
     trackSourceLinks.clear();
     surfSequence.clear();
 

Examples/Python/src/TrackFitting.cpp

@@ -41,7 +41,7 @@ void addTrackFitting(Context& ctx) {
 
   ACTS_PYTHON_DECLARE_ALGORITHM(ActsExamples::RefittingAlgorithm, mex,
                                 "RefittingAlgorithm", inputTracks, outputTracks,
-                                fit, pickTrack);
+                                fit, pickTrack, initialVarInflation);
 
   {
     py::class_<TrackFitterFunction, std::shared_ptr<TrackFitterFunction>>(

Examples/Scripts/Python/truth_tracking_gsf_refitting.py

@@ -44,6 +44,7 @@ def runRefittingGsf(
             acts.logging.INFO,
             inputTracks="kf_tracks",
             outputTracks="gsf_refit_tracks",
+            initialVarInflation=[100.0] * 6,
             fit=acts.examples.makeGsfFitterFunction(
                 trackingGeometry, field, **gsfOptions
             ),
@@ -76,7 +77,7 @@ def runRefittingGsf(
     s.addWriter(
         acts.examples.RootTrackSummaryWriter(
             level=acts.logging.INFO,
-            inputTracks="tracks",
+            inputTracks="gsf_refit_tracks",
             inputParticles="particles_selected",
             inputTrackParticleMatching="refit_track_particle_matching",
             filePath=str(outputDir / "tracksummary_gsf_refit.root"),
@@ -86,9 +87,9 @@ def runRefittingGsf(
     s.addWriter(
         acts.examples.TrackFitterPerformanceWriter(
             level=acts.logging.INFO,
-            inputTracks="tracks",
+            inputTracks="gsf_refit_tracks",
             inputParticles="particles_selected",
-            inputTrackParticleMatching="track_particle_matching",
+            inputTrackParticleMatching="refit_track_particle_matching",
             filePath=str(outputDir / "performance_gsf_refit.root"),
         )
     )

Examples/Scripts/Python/truth_tracking_kalman_refitting.py

@@ -41,6 +41,7 @@ def runRefittingKf(
             level=acts.logging.INFO,
             inputTracks="kf_tracks",
             outputTracks="kf_refit_tracks",
+            initialVarInflation=[100.0] * 6,
             fit=acts.examples.makeKalmanFitterFunction(
                 trackingGeometry, field, **kalmanOptions
             ),