update

acts-project · Dec 19, 2024 · abd3155 · abd3155
1 parent 52ed96f
commit abd3155
Show file tree

Hide file tree

Showing 5 changed files with 11 additions and 2 deletions.
diff --git a/Examples/Algorithms/TrackFitting/include/ActsExamples/TrackFitting/RefittingAlgorithm.hpp b/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

diff --git a/Examples/Algorithms/TrackFitting/src/RefittingAlgorithm.cpp b/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();
 

diff --git a/Examples/Python/src/TrackFitting.cpp b/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>>(

diff --git a/Examples/Scripts/Python/truth_tracking_gsf_refitting.py b/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
             ),

diff --git a/Examples/Scripts/Python/truth_tracking_kalman_refitting.py b/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
             ),