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 <vector>
0013
0014 #include "Acts/AmbiguityResolution/GreedyAmbiguityResolution.hpp"
0015 #include "AmbiguitySolverConfig.h"
0016 #include "algorithms/interfaces/WithPodConfig.h"
0017
0018 namespace eicrecon {
0019
0020 /*Reco Track Filtering Based on Greedy ambiguity resolution solver adopted from ACTS*/
0021 class AmbiguitySolver : public WithPodConfig<eicrecon::AmbiguitySolverConfig> {
0022 public:
0023   AmbiguitySolver();
0024
0025   void init(std::shared_ptr<spdlog::logger> log);
0026
0027   std::tuple<std::vector<ActsExamples::ConstTrackContainer*>,
0028              std::vector<ActsExamples::Trajectories*>>
0029   process(std::vector<const ActsExamples::ConstTrackContainer*> input_container,
0030           const edm4eic::Measurement2DCollection& meas2Ds);
0031
0032 private:
0033   std::shared_ptr<spdlog::logger> m_log;
0034   Acts::GreedyAmbiguityResolution::Config m_acts_cfg;
0035   std::unique_ptr<Acts::GreedyAmbiguityResolution> m_core;
0036   /// Private access to the logging instance
0037   std::shared_ptr<const Acts::Logger> m_acts_logger{nullptr};
0038   const Acts::Logger& logger() const { return *m_acts_logger; }
0039 };
0040
0041 } // namespace eicrecon