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<
0028       std::vector<ActsExamples::ConstTrackContainer *>,
0029       std::vector<ActsExamples::Trajectories *>
0030       >
0031   process(std::vector<const ActsExamples::ConstTrackContainer*> input_container,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