From 07ecadbee3f7ef084995adf8a962b34008009e1f Mon Sep 17 00:00:00 2001
From: Xiaocong Ai <xiaocong.ai@cern.ch>
Date: Tue, 22 Oct 2024 14:37:17 +0200
Subject: [PATCH] write all tracks to acts output

---
 Tracking/Acts/FaserActsKalmanFilter/src/CKF2.cxx            | 4 ++--
 .../Acts/FaserActsKalmanFilter/src/KalmanFitterTool.cxx     | 6 ++++++
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/Tracking/Acts/FaserActsKalmanFilter/src/CKF2.cxx b/Tracking/Acts/FaserActsKalmanFilter/src/CKF2.cxx
index 862066012..f0489f2d5 100644
--- a/Tracking/Acts/FaserActsKalmanFilter/src/CKF2.cxx
+++ b/Tracking/Acts/FaserActsKalmanFilter/src/CKF2.cxx
@@ -270,10 +270,10 @@ StatusCode CKF2::execute() {
 //  }
 
   if (m_statesWriter && !m_noDiagnostics) {
-    ATH_CHECK(m_trajectoryStatesWriterTool->write(gctx, selectedTracks, m_isMC));
+    ATH_CHECK(m_trajectoryStatesWriterTool->write(gctx, tracks, m_isMC));
   }
   if (m_summaryWriter && !m_noDiagnostics) {
-    ATH_CHECK(m_trajectorySummaryWriterTool->write(gctx, selectedTracks, m_isMC));
+    ATH_CHECK(m_trajectorySummaryWriterTool->write(gctx, tracks, m_isMC));
   }
   // ATH_CHECK(allTrackContainer.record(std::move(outputAllTracks)));
   ATH_CHECK(trackContainer.record(std::move(outputTracks)));
diff --git a/Tracking/Acts/FaserActsKalmanFilter/src/KalmanFitterTool.cxx b/Tracking/Acts/FaserActsKalmanFilter/src/KalmanFitterTool.cxx
index a339e81a9..380edcd19 100644
--- a/Tracking/Acts/FaserActsKalmanFilter/src/KalmanFitterTool.cxx
+++ b/Tracking/Acts/FaserActsKalmanFilter/src/KalmanFitterTool.cxx
@@ -367,6 +367,12 @@ KalmanFitterTool::fit(const EventContext &ctx, const Acts::GeometryContext &gctx
 
   auto [sourceLinks, measurements] = getMeasurementsFromTrack(inputTrack);
   auto trackParameters = getParametersFromTrack(inputTrack->trackParameters()->front(), inputVector, origin);
+  //Inflate the covariance of the starting track parameters
+  //@todo: make the inflation configurable
+  if(trackParameters.covariance().has_value()){ 
+    trackParameters.covariance() = (trackParameters.covariance().value())*10; 
+  }
+ 
   ATH_MSG_DEBUG("trackParameters: " << trackParameters.parameters().transpose());
   ATH_MSG_DEBUG("position: " << trackParameters.position(gctx).transpose());
   ATH_MSG_DEBUG("momentum: " << trackParameters.momentum().transpose());
-- 
GitLab