algorithms/tracking/AmbiguitySolver.h

0001 // SPDX-License-Identifier: LGPL-3.0-or-later
0002 // Copyright (C) 2024 Minjung Kim, Barak Schmookler
0003 #pragma once
0004
0005 #include <Acts/Utilities/Logger.hpp>
0006 #include <ActsExamples/EventData/Track.hpp>
0007 #include <ActsExamples/EventData/Trajectories.hpp>
0008 #include <edm4eic/Measurement2D.h>
0009 #include <spdlog/logger.h>
0010 #include <memory>
0011 #include <tuple>
0012 #include <variant>
0013 #include <vector>
0014
0015 #include "Acts/AmbiguityResolution/GreedyAmbiguityResolution.hpp"
0016 #include "AmbiguitySolverConfig.h"
0017 #include "algorithms/interfaces/WithPodConfig.h"
0018
0019 namespace eicrecon {
0020
0021 /*Reco Track Filtering Based on Greedy ambiguity resolution solver adopted from ACTS*/
0022 class AmbiguitySolver : public WithPodConfig<eicrecon::AmbiguitySolverConfig> {
0023 public:
0024   AmbiguitySolver();
0025
0026   void init(std::shared_ptr<spdlog::logger> log);
0027
0028   std::tuple<std::vector<ActsExamples::ConstTrackContainer*>,
0029              std::vector<ActsExamples::Trajectories*>>
0030   process(std::vector<const ActsExamples::ConstTrackContainer*> input_container,
0031           const edm4eic::Measurement2DCollection& meas2Ds);
0032
0033 private:
0034   std::shared_ptr<spdlog::logger> m_log;
0035   Acts::GreedyAmbiguityResolution::Config m_acts_cfg;
0036   std::unique_ptr<Acts::GreedyAmbiguityResolution> m_core;
0037   /// Private access to the logging instance
0038   std::shared_ptr<const Acts::Logger> m_acts_logger{nullptr};
0039   const Acts::Logger& logger() const { return *m_acts_logger; }
0040 };
0041
0042 } // namespace eicrecon