Add missing ACTS extensions to ACTSTruthTrackingProc.

MuonColliderSoft · Aug 12, 2024 · a5c56d0 · a5c56d0
1 parent 994f078
commit a5c56d0
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/ACTSTracking/SourceLink.hxx b/ACTSTracking/SourceLink.hxx
@@ -65,5 +65,16 @@ struct SourceLinkAccessor : GeometryIdMultisetAccessor<SourceLink> {
   }
 };
 
+/// Access for the surface associated to a source link
+struct SurfaceAccessor {
+  const std::shared_ptr<const Acts::TrackingGeometry> trackingGeometry;
+
+  const Acts::Surface* operator()(const Acts::SourceLink& sourceLink) const {
+    const auto& mySourceLink = sourceLink.get<SourceLink>();
+      return trackingGeometry->findSurface(mySourceLink.geometryId());
+    }
+  };
+
+
 }  // namespace ACTSTracking
 
diff --git a/src/ACTSTruthTrackingProc.cxx b/src/ACTSTruthTrackingProc.cxx
@@ -231,6 +231,7 @@ void ACTSTruthTrackingProc::processEvent(LCEvent* evt) {
 
     // Make container
     // MeasurementContainer track;
+    ACTSTracking::MeasurementContainer measurements;
     std::vector<Acts::SourceLink> trackSourceLinks;
     ACTSTracking::MeasurementContainer track;
     for (EVENT::TrackerHit* hit : trackFilteredByRHits) {
@@ -263,6 +264,7 @@ void ACTSTruthTrackingProc::processEvent(LCEvent* evt) {
           sourceLink, loc, cov, Acts::eBoundLoc0, Acts::eBoundLoc1);
 
       track.push_back(meas);
+      measurements.push_back(meas);
       trackSourceLinks.push_back(sourceLink);
     }
 
@@ -276,13 +278,22 @@ void ACTSTruthTrackingProc::processEvent(LCEvent* evt) {
     Updater kfUpdater;
     Smoother kfSmoother;
 
+    ACTSTracking::MeasurementCalibrator measCal { measurements };
+    ACTSTracking::SurfaceAccessor surfaceAccessor {trackingGeometry()};
+
     Acts::KalmanFitterExtensions<Acts::VectorMultiTrajectory> extensions;
+    extensions.calibrator.connect<
+        &ACTSTracking::MeasurementCalibrator::calibrate>(
+        &measCal);
     extensions.updater.connect<
         &Acts::GainMatrixUpdater::operator()<Acts::VectorMultiTrajectory>>(
         &kfUpdater);
     extensions.smoother.connect<
         &Acts::GainMatrixSmoother::operator()<Acts::VectorMultiTrajectory>>(
         &kfSmoother);
+    extensions.surfaceAccessor.connect<
+        &ACTSTracking::SurfaceAccessor::operator()>(
+        &surfaceAccessor);
 
     // Set the KalmanFitter options
     // std::unique_ptr<const Acts::Logger>