factories/reco/HadronicFinalState_factory.h

0001 // SPDX-License-Identifier: LGPL-3.0-or-later
0002 // Copyright (C) 2024 Tyler Kutz
0003
0004 #pragma once
0005
0006 #include <JANA/JEvent.h>
0007 #include <edm4eic/HadronicFinalStateCollection.h>
0008 #include <memory>
0009 #include <string>
0010 #include <utility>
0011 #include <vector>
0012
0013 #include "extensions/jana/JOmniFactory.h"
0014 #include "services/algorithms_init/AlgorithmsInit_service.h"
0015
0016 namespace eicrecon {
0017
0018 template <typename AlgoT>
0019 class HadronicFinalState_factory
0020     : public JOmniFactory<HadronicFinalState_factory<AlgoT>, NoConfig> {
0021
0022 public:
0023   using FactoryT = JOmniFactory<HadronicFinalState_factory<AlgoT>, NoConfig>;
0024
0025 private:
0026   std::unique_ptr<AlgoT> m_algo;
0027
0028   typename FactoryT::template PodioInput<edm4hep::MCParticle> m_mc_particles_input{this};
0029   typename FactoryT::template PodioInput<edm4eic::ReconstructedParticle> m_rc_particles_input{this};
0030   typename FactoryT::template PodioInput<edm4eic::MCRecoParticleAssociation>
0031       m_rc_particles_assoc_input{this};
0032   typename FactoryT::template PodioOutput<edm4eic::HadronicFinalState>
0033       m_hadronic_final_state_output{this};
0034
0035   typename FactoryT::template Service<AlgorithmsInit_service> m_algorithmsInit{this};
0036
0037 public:
0038   void Configure() {
0039     m_algo = std::make_unique<AlgoT>(this->GetPrefix());
0040     m_algo->level(static_cast<algorithms::LogLevel>(this->logger()->level()));
0041     m_algo->applyConfig(this->config());
0042     m_algo->init();
0043   }
0044
0045   void Process(int32_t /* run_number */, uint64_t /* event_number */) {
0046     m_algo->process({m_mc_particles_input(), m_rc_particles_input(), m_rc_particles_assoc_input()},
0047                     {m_hadronic_final_state_output().get()});
0048   }
0049 };
0050
0051 } // namespace eicrecon