algorithms/tracking/AmbiguitySolver.cc

0001 // SPDX-License-Identifier: LGPL-3.0-or-later
0002 // Copyright (C) 2024 Minjung Kim, Barak Schmookler
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& /* meas2Ds */) {
0065
0066   // Assuming ActsExamples::ConstTrackContainer is compatible with Acts::ConstVectorTrackContainer
0067   // Create track container
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   //Make output trajectories
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 } // namespace eicrecon