File indexing completed on 2025-07-11 07:53:37
0001
0002
0003 #include "AmbiguitySolver.h"
0004
0005 #include <Acts/AmbiguityResolution/GreedyAmbiguityResolution.hpp>
0006 #include <Acts/EventData/GenericBoundTrackParameters.hpp>
0007 #include <Acts/EventData/MultiTrajectory.hpp>
0008 #include <Acts/EventData/ParticleHypothesis.hpp>
0009 #include <Acts/EventData/SourceLink.hpp>
0010 #include <Acts/EventData/TrackContainer.hpp>
0011 #include <Acts/EventData/TrackProxy.hpp>
0012 #include <Acts/EventData/TrackStatePropMask.hpp>
0013 #include <Acts/EventData/VectorMultiTrajectory.hpp>
0014 #include <Acts/EventData/VectorTrackContainer.hpp>
0015 #include <Acts/Surfaces/Surface.hpp>
0016 #include <ActsExamples/EventData/IndexSourceLink.hpp>
0017 #include <ActsExamples/EventData/Track.hpp>
0018 #include <ActsExamples/EventData/Trajectories.hpp>
0019 #include <boost/container/flat_set.hpp>
0020 #include <boost/container/vector.hpp>
0021 #include <edm4eic/Measurement2DCollection.h>
0022 #include <Eigen/Core>
0023 #include <cstddef>
0024 #include <utility>
0025
0026 #include "Acts/Utilities/Logger.hpp"
0027 #include "AmbiguitySolverConfig.h"
0028 #include "extensions/spdlog/SpdlogFormatters.h" // IWYU pragma: keep
0029 #include "extensions/spdlog/SpdlogToActs.h"
0030
0031 namespace eicrecon {
0032
0033 Acts::GreedyAmbiguityResolution::Config
0034 transformConfig(const eicrecon::AmbiguitySolverConfig& cfg) {
0035 Acts::GreedyAmbiguityResolution::Config result;
0036 result.maximumSharedHits = cfg.maximum_shared_hits;
0037 result.maximumIterations = cfg.maximum_iterations;
0038 result.nMeasurementsMin = cfg.n_measurements_min;
0039 return result;
0040 }
0041
0042 static std::size_t sourceLinkHash(const Acts::SourceLink& a) {
0043 return static_cast<std::size_t>(a.get<ActsExamples::IndexSourceLink>().index());
0044 }
0045
0046 static bool sourceLinkEquality(const Acts::SourceLink& a, const Acts::SourceLink& b) {
0047 return a.get<ActsExamples::IndexSourceLink>().index() ==
0048 b.get<ActsExamples::IndexSourceLink>().index();
0049 }
0050
0051 AmbiguitySolver::AmbiguitySolver() = default;
0052
0053 void AmbiguitySolver::init(std::shared_ptr<spdlog::logger> log) {
0054
0055 m_log = log;
0056 m_acts_logger = eicrecon::getSpdlogLogger("AmbiguitySolver", m_log);
0057 m_acts_cfg = transformConfig(m_cfg);
0058 m_core = std::make_unique<Acts::GreedyAmbiguityResolution>(m_acts_cfg, logger().clone());
0059 }
0060
0061 std::tuple<std::vector<ActsExamples::ConstTrackContainer*>,
0062 std::vector<ActsExamples::Trajectories*>>
0063 AmbiguitySolver::process(std::vector<const ActsExamples::ConstTrackContainer*> input_container,
0064 const edm4eic::Measurement2DCollection& ) {
0065
0066
0067
0068 std::vector<ActsExamples::Trajectories*> output_trajectories;
0069 std::vector<ActsExamples::ConstTrackContainer*> output_tracks;
0070
0071 auto& input_trks = input_container.front();
0072 Acts::GreedyAmbiguityResolution::State state;
0073 m_core->computeInitialState(*input_trks, state, &sourceLinkHash, &sourceLinkEquality);
0074 m_core->resolve(state);
0075
0076 ActsExamples::TrackContainer solvedTracks{std::make_shared<Acts::VectorTrackContainer>(),
0077 std::make_shared<Acts::VectorMultiTrajectory>()};
0078 solvedTracks.ensureDynamicColumns(*input_trks);
0079
0080 for (auto iTrack : state.selectedTracks) {
0081
0082 auto destProxy = solvedTracks.getTrack(solvedTracks.addTrack());
0083 auto srcProxy = input_trks->getTrack(state.trackTips.at(iTrack));
0084 destProxy.copyFrom(srcProxy, false);
0085 destProxy.tipIndex() = srcProxy.tipIndex();
0086 }
0087
0088 output_tracks.push_back(new ActsExamples::ConstTrackContainer(
0089 std::make_shared<Acts::ConstVectorTrackContainer>(std::move(solvedTracks.container())),
0090 input_trks->trackStateContainerHolder()));
0091
0092
0093 ActsExamples::Trajectories::IndexedParameters parameters;
0094 std::vector<Acts::MultiTrajectoryTraits::IndexType> tips;
0095
0096 for (const auto& track : *(output_tracks.front())) {
0097
0098 tips.clear();
0099 parameters.clear();
0100
0101 tips.push_back(track.tipIndex());
0102 parameters.emplace(std::pair{
0103 track.tipIndex(),
0104 ActsExamples::TrackParameters{track.referenceSurface().getSharedPtr(), track.parameters(),
0105 track.covariance(), track.particleHypothesis()}});
0106
0107 output_trajectories.push_back(new ActsExamples::Trajectories(
0108 ((*output_tracks.front())).trackStateContainer(), tips, parameters));
0109 }
0110
0111 return std::make_tuple(std::move(output_tracks), std::move(output_trajectories));
0112 }
0113
0114 }