File indexing completed on 2024-06-17 07:05:45
0001
0002
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
0036
0037
0038
0039 class TrackFindingAlgorithm : public GaudiAlgorithm {
0040 public:
0041
0042
0043 using TrackFinderOptions = Acts::CombinatorialKalmanFilterOptions<IndexSourceLinkAccessor::Iterator, Acts::VectorMultiTrajectory>;
0044 using TrackFinderResult = std::vector<Acts::Result<Acts::CombinatorialKalmanFilterResult<Acts::VectorMultiTrajectory>>>;
0045
0046
0047
0048
0049 class TrackFinderFunction {
0050 public:
0051 virtual ~TrackFinderFunction() = default;
0052 virtual TrackFinderResult operator()(const TrackParametersContainer&,
0053 const TrackFinderOptions&) const = 0;
0054 };
0055
0056
0057
0058
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 }
0093
0094 #endif