src/components/CKFTrackingFunction.cpp

0001 // SPDX-License-Identifier: LGPL-3.0-or-later
0002 // Copyright (C) 2022 Whitney Armstrong, Wouter Deconinck, Sylvester Joosten
0003
0004 #include "Acts/EventData/TrackContainer.hpp"
0005 #include "Acts/EventData/VectorMultiTrajectory.hpp"
0006 #include "Acts/EventData/VectorTrackContainer.hpp"
0007 #include "Acts/MagneticField/MagneticFieldProvider.hpp"
0008 #include "Acts/Propagator/EigenStepper.hpp"
0009 #include "Acts/Propagator/Navigator.hpp"
0010 #include "Acts/Propagator/Propagator.hpp"
0011 #include "Acts/TrackFitting/GainMatrixSmoother.hpp"
0012 #include "Acts/TrackFitting/GainMatrixUpdater.hpp"
0013
0014 #if 0
0015 #include "Acts/Propagator/EigenStepper.hpp"
0016 #include "Acts/Propagator/Navigator.hpp"
0017 #include "Acts/Propagator/Propagator.hpp"
0018 #include "Acts/TrackFitting/GainMatrixSmoother.hpp"
0019 #include "Acts/TrackFitting/GainMatrixUpdater.hpp"
0020 #endif
0021
0022 #include "CKFTracking.h"
0023
0024 #include "JugTrack/DD4hepBField.h"
0025
0026
0027 #include <random>
0028 #include <stdexcept>
0029
0030 namespace {
0031   using Updater  = Acts::GainMatrixUpdater;
0032   using Smoother = Acts::GainMatrixSmoother;
0033
0034   using Stepper    = Acts::EigenStepper<>;
0035   using Navigator  = Acts::Navigator;
0036   using Propagator = Acts::Propagator<Stepper, Navigator>;
0037   using CKF =
0038       Acts::CombinatorialKalmanFilter<Propagator, Acts::VectorMultiTrajectory>;
0039
0040   using TrackContainer =
0041       Acts::TrackContainer<Acts::VectorTrackContainer,
0042                            Acts::VectorMultiTrajectory, std::shared_ptr>;
0043
0044   /** Finder implmentation .
0045    *
0046    * \ingroup track
0047    */
0048   struct CKFTrackingFunctionImpl
0049     : public Jug::Reco::CKFTracking::CKFTrackingFunction {
0050     CKF trackFinder;
0051
0052     CKFTrackingFunctionImpl(CKF&& f) : trackFinder(std::move(f)) {}
0053
0054     Jug::Reco::CKFTracking::TrackFinderResult
0055     operator()(const ActsExamples::TrackParameters& initialParameters,
0056                const Jug::Reco::CKFTracking::TrackFinderOptions& options,
0057                TrackContainer& tracks) const override {
0058       return trackFinder.findTracks(initialParameters, options, tracks);
0059     };
0060   };
0061
0062 } // namespace
0063
0064 namespace Jug::Reco {
0065
0066   std::shared_ptr<CKFTracking::CKFTrackingFunction>
0067   CKFTracking::makeCKFTrackingFunction(
0068       std::shared_ptr<const Acts::TrackingGeometry>      trackingGeometry,
0069       std::shared_ptr<const Acts::MagneticFieldProvider> magneticField)
0070   {
0071     Stepper   stepper(std::move(magneticField));
0072     Navigator::Config cfg{trackingGeometry};
0073     cfg.resolvePassive   = false;
0074     cfg.resolveMaterial  = true;
0075     cfg.resolveSensitive = true;
0076     Navigator navigator(cfg);
0077
0078     Propagator propagator(std::move(stepper), std::move(navigator));
0079     CKF        trackFinder(std::move(propagator));
0080
0081     // build the track finder functions. onws the track finder object.
0082     return std::make_shared<CKFTrackingFunctionImpl>(std::move(trackFinder));
0083   }
0084
0085 } // namespace Jug::Reco