acts/src/TrackFindingAlgorithm.h

0001 // SPDX-License-Identifier: LGPL-3.0-or-later
0002 // Copyright (C) 2022 Whitney Armstrong, Wouter Deconinck
0003
0004 #ifndef JUGGLER_JUGRECO_TrackFindingAlgorithm_HH
0005 #define JUGGLER_JUGRECO_TrackFindingAlgorithm_HH
0006
0007 #include <functional>
0008 #include <random>
0009 #include <stdexcept>
0010 #include <vector>
0011
0012 #include "Gaudi/Property.h"
0013 #include "GaudiAlg/GaudiAlgorithm.h"
0014 #include "GaudiKernel/ToolHandle.h"
0015
0016 #include "JugBase/DataHandle.h"
0017 #include "JugBase/IGeoSvc.h"
0018 #include "JugBase/BField/DD4hepBField.h"
0019 #include "JugTrack/GeometryContainers.hpp"
0020 #include "JugTrack/Index.hpp"
0021 #include "JugTrack/IndexSourceLink.hpp"
0022 #include "JugTrack/Measurement.hpp"
0023 #include "JugTrack/Track.hpp"
0024 #include "JugTrack/Trajectories.hpp"
0025
0026 #include "edm4eic/TrackerHitCollection.h"
0027
0028 #include "Acts/Definitions/Common.hpp"
0029 #include "Acts/Geometry/TrackingGeometry.hpp"
0030 #include "Acts/TrackFinding/CombinatorialKalmanFilter.hpp"
0031 #include "Acts/TrackFinding/MeasurementSelector.hpp"
0032
0033 namespace Jug::Reco {
0034
0035 /** Fitting algorithm implmentation .
0036  *
0037  * \ingroup tracking
0038  */
0039 class TrackFindingAlgorithm : public GaudiAlgorithm {
0040 public:
0041   /// Track finder function that takes input measurements, initial trackstate
0042   /// and track finder options and returns some track-finder-specific result.
0043   using TrackFinderOptions  = Acts::CombinatorialKalmanFilterOptions<IndexSourceLinkAccessor::Iterator>;
0044   using TrackFinderResult   = std::vector<Acts::Result<Acts::CombinatorialKalmanFilterResult>>;
0045
0046   /// Find function that takes the above parameters
0047   /// @note This is separated into a virtual interface to keep compilation units
0048   /// small
0049   class TrackFinderFunction {
0050    public:
0051     virtual ~TrackFinderFunction() = default;
0052     virtual TrackFinderResult operator()(const TrackParametersContainer&,
0053                                          const TrackFinderOptions&) const = 0;
0054   };
0055
0056   /// Create the track finder function implementation.
0057   /// The magnetic field is intentionally given by-value since the variant
0058   /// contains shared_ptr anyways.
0059   static std::shared_ptr<TrackFinderFunction> makeTrackFinderFunction(
0060     std::shared_ptr<const Acts::TrackingGeometry> trackingGeometry,
0061     std::shared_ptr<const Acts::MagneticFieldProvider> magneticField);
0062
0063 public:
0064   DataHandle<IndexSourceLinkContainer> m_inputSourceLinks{"inputSourceLinks", Gaudi::DataHandle::Reader, this};
0065   DataHandle<MeasurementContainer> m_inputMeasurements{"inputMeasurements", Gaudi::DataHandle::Reader, this};
0066   DataHandle<TrackParametersContainer> m_inputInitialTrackParameters{"inputInitialTrackParameters",
0067                                                                      Gaudi::DataHandle::Reader, this};
0068   DataHandle<TrajectoriesContainer> m_outputTrajectories{"outputTrajectories", Gaudi::DataHandle::Writer, this};
0069
0070   Gaudi::Property<std::vector<double>> m_etaBins{this, "etaBins", {}};
0071   Gaudi::Property<std::vector<double>> m_chi2CutOff{this, "chi2CutOff", {15.}};
0072   Gaudi::Property<std::vector<size_t>> m_numMeasurementsCutOff{this, "numMeasurementsCutOff", {10}};
0073
0074   std::shared_ptr<TrackFinderFunction> m_trackFinderFunc;
0075   SmartIF<IGeoSvc> m_geoSvc;
0076
0077   std::shared_ptr<const Jug::BField::DD4hepBField> m_BField = nullptr;
0078   Acts::GeometryContext m_geoctx;
0079   Acts::CalibrationContext m_calibctx;
0080   Acts::MagneticFieldContext m_fieldctx;
0081
0082   Acts::MeasurementSelector::Config m_sourcelinkSelectorCfg;
0083   Acts::Logging::Level m_actsLoggingLevel = Acts::Logging::INFO;
0084
0085   TrackFindingAlgorithm(const std::string& name, ISvcLocator* svcLoc);
0086
0087   StatusCode initialize() override;
0088
0089   StatusCode execute() override;
0090 };
0091
0092 } // namespace Jug::Reco
0093
0094 #endif