acts/src/TrackFittingAlgorithm.h

0001 // SPDX-License-Identifier: LGPL-3.0-or-later
0002 // Copyright (C) 2022 Whitney Armstrong
0003
0004 #ifndef JUGGLER_JUGRECO_TrackFittingAlgorithm_HH
0005 #define JUGGLER_JUGRECO_TrackFittingAlgorithm_HH 1
0006
0007 #include <functional>
0008 #include <stdexcept>
0009 #include <vector>
0010 #include <random>
0011 #include <stdexcept>
0012
0013 #include "GaudiAlg/GaudiAlgorithm.h"
0014 #include "GaudiKernel/ToolHandle.h"
0015 #include "Gaudi/Property.h"
0016
0017 #include "JugBase/DataHandle.h"
0018 #include "JugBase/IGeoSvc.h"
0019 #include "JugBase/BField/DD4hepBField.h"
0020 #include "JugTrack/GeometryContainers.hpp"
0021 #include "JugTrack/IndexSourceLink.hpp"
0022 #include "JugTrack/Track.hpp"
0023 #include "JugTrack/Measurement.hpp"
0024 #include "JugTrack/Trajectories.hpp"
0025 #include "JugTrack/ProtoTrack.hpp"
0026
0027 #include "edm4eic/TrackerHitCollection.h"
0028
0029 #include "Acts/Definitions/Common.hpp"
0030 #include "Acts/Geometry/TrackingGeometry.hpp"
0031 #include "Acts/MagneticField/MagneticFieldProvider.hpp"
0032 #include "Acts/TrackFitting/KalmanFitter.hpp"
0033 #include "Acts/TrackFitting/GainMatrixSmoother.hpp"
0034 #include "Acts/TrackFitting/GainMatrixUpdater.hpp"
0035 #include "Acts/Geometry/TrackingGeometry.hpp"
0036 #include "Acts/Definitions/Common.hpp"
0037
0038
0039 namespace Jug::Reco {
0040
0041   /** Fitting algorithm implmentation .
0042    *
0043    * \ingroup tracking
0044    */
0045   class TrackFittingAlgorithm : public GaudiAlgorithm {
0046   public:
0047     /// Track fitter function that takes input measurements, initial trackstate
0048     /// and fitter options and returns some track-fitter-specific result.
0049     using TrackFitterOptions =
0050         Acts::KalmanFitterOptions;
0051
0052     //using TrackFinderResult = Acts::Result<Acts::CombinatorialKalmanFilterResult<SourceLink>>;
0053     using FitterResult = Acts::Result<Acts::KalmanFitterResult>;
0054
0055     /// Fit function that takes input measurements, initial trackstate and fitter
0056     using FitterFunction = std::function<FitterResult(
0057       const std::vector<IndexSourceLink>&, const TrackParameters&, const TrackFitterOptions&)>;
0058
0059   public:
0060     DataHandle<IndexSourceLinkContainer> m_inputSourceLinks{"inputSourceLinks", Gaudi::DataHandle::Reader, this};
0061     DataHandle<TrackParametersContainer> m_initialTrackParameters{"initialTrackParameters", Gaudi::DataHandle::Reader, this};
0062     DataHandle<MeasurementContainer>     m_inputMeasurements{"inputMeasurements", Gaudi::DataHandle::Reader, this};
0063     DataHandle<ProtoTrackContainer>      m_inputProtoTracks{"inputProtoTracks", Gaudi::DataHandle::Reader, this};
0064     DataHandle<TrajectoriesContainer>    m_foundTracks{"foundTracks", Gaudi::DataHandle::Reader, this};
0065     DataHandle<TrajectoriesContainer>    m_outputTrajectories{"outputTrajectories", Gaudi::DataHandle::Writer, this};
0066
0067     FitterFunction                        m_trackFittingFunc;
0068     SmartIF<IGeoSvc>                      m_geoSvc;
0069     std::shared_ptr<const Jug::BField::DD4hepBField> m_BField = nullptr;
0070     Acts::GeometryContext                 m_geoctx;
0071     Acts::CalibrationContext              m_calibctx;
0072     Acts::MagneticFieldContext            m_fieldctx;
0073
0074     //Acts::CKFSourceLinkSelector::Config m_sourcelinkSelectorCfg;
0075
0076     TrackFittingAlgorithm(const std::string& name, ISvcLocator* svcLoc);
0077
0078     /** Create the track finder function implementation.
0079      *  The magnetic field is intentionally given by-value since the variant
0080      *  contains shared_ptr anyways.
0081      */
0082     static FitterFunction
0083     makeTrackFittingFunction(std::shared_ptr<const Acts::TrackingGeometry>      trackingGeometry,
0084                              std::shared_ptr<const Acts::MagneticFieldProvider> magneticField);
0085     // BFieldVariant                                 magneticField);
0086
0087     StatusCode initialize() override;
0088
0089     StatusCode execute() override;
0090    private:
0091     /// Helper function to call correct FitterFunction
0092     FitterResult fitTrack(
0093         const std::vector<IndexSourceLink>& sourceLinks,
0094         const TrackParameters& initialParameters,
0095         const TrackFitterOptions& options
0096         ) const;
0097         //, const std::vector<const Acts::Surface*>& surfSequence) const;
0098   };
0099
0100   inline TrackFittingAlgorithm::FitterResult
0101   TrackFittingAlgorithm::fitTrack(const std::vector<IndexSourceLink>& sourceLinks,
0102                                   const TrackParameters&              initialParameters,
0103                                   const TrackFitterOptions&           options) const
0104   {
0105     // const std::vector<const Acts::Surface*>& surfSequence) const
0106     // if (m_cfg.directNavigation) {
0107     //  return m_cfg.dFit(sourceLinks, initialParameters, options, surfSequence);
0108     //}
0109     return m_trackFittingFunc(sourceLinks, initialParameters, options);
0110   }
0111
0112 } // namespace Jug::Reco
0113
0114 #endif