factories/tracking/ActsToTracks_factory.h

0001 // SPDX-License-Identifier: LGPL-3.0-or-later
0002 // Copyright 2024, Dmitry Kalinkin
0003
0004 #pragma once
0005
0006 #include <memory>
0007
0008 #include "algorithms/tracking/ActsToTracks.h"
0009 #include "extensions/jana/JOmniFactory.h"
0010
0011 namespace eicrecon {
0012
0013 class ActsToTracks_factory : public JOmniFactory<ActsToTracks_factory, NoConfig> {
0014 public:
0015   using AlgoT = eicrecon::ActsToTracks;
0016
0017 private:
0018   std::unique_ptr<AlgoT> m_algo;
0019
0020   PodioInput<edm4eic::Measurement2D> m_measurements_input{this};
0021   Input<ActsExamples::Trajectories> m_acts_trajectories_input{this};
0022   PodioInput<edm4eic::MCRecoTrackerHitAssociation> m_raw_hit_assocs_input{this};
0023   PodioOutput<edm4eic::Trajectory> m_trajectories_output{this};
0024   PodioOutput<edm4eic::TrackParameters> m_parameters_output{this};
0025   PodioOutput<edm4eic::Track> m_tracks_output{this};
0026   PodioOutput<edm4eic::MCRecoTrackParticleAssociation> m_track_assocs_output{this};
0027
0028 public:
0029   void Configure() {
0030     m_algo = std::make_unique<AlgoT>(this->GetPrefix());
0031     m_algo->level((algorithms::LogLevel)logger()->level());
0032     m_algo->applyConfig(config());
0033     m_algo->init();
0034   };
0035
0036   void Process(int32_t /* run_number */, uint64_t /* event_number */) {
0037     std::vector<gsl::not_null<const ActsExamples::Trajectories*>> acts_trajectories_input;
0038     for (auto acts_traj : m_acts_trajectories_input()) {
0039       acts_trajectories_input.push_back(acts_traj);
0040     }
0041     m_algo->process(
0042         {
0043             m_measurements_input(),
0044             acts_trajectories_input,
0045             m_raw_hit_assocs_input(),
0046         },
0047         {
0048             m_trajectories_output().get(),
0049             m_parameters_output().get(),
0050             m_tracks_output().get(),
0051             m_track_assocs_output().get(),
0052         });
0053   }
0054 };
0055
0056 } // namespace eicrecon