factories/reco/InclusiveKinematicsReconstructed_factory.h

0001 // SPDX-License-Identifier: LGPL-3.0-or-later
0002 // Copyright (C) 2022 Wouter Deconinck
0003
0004 #pragma once
0005
0006 #include <JANA/JEvent.h>
0007 #include <edm4eic/InclusiveKinematicsCollection.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 InclusiveKinematicsReconstructed_factory :
0020         public JOmniFactory<InclusiveKinematicsReconstructed_factory<AlgoT>> {
0021
0022 public:
0023     using FactoryT = JOmniFactory<InclusiveKinematicsReconstructed_factory<AlgoT>>;
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_scattered_electron_input {this};
0030     typename FactoryT::template PodioInput<edm4eic::HadronicFinalState> m_hadronic_final_state_input {this};
0031     typename FactoryT::template PodioOutput<edm4eic::InclusiveKinematics> m_inclusive_kinematics_output {this};
0032
0033     typename FactoryT::template Service<AlgorithmsInit_service> m_algorithmsInit {this};
0034
0035 public:
0036     void Configure() {
0037         m_algo = std::make_unique<AlgoT>(this->GetPrefix());
0038         m_algo->level(static_cast<algorithms::LogLevel>(this->logger()->level()));
0039         m_algo->init();
0040     }
0041
0042     void ChangeRun(int64_t run_number) {
0043     }
0044
0045     void Process(int64_t run_number, uint64_t event_number) {
0046         m_algo->process({m_mc_particles_input(), m_scattered_electron_input(), m_hadronic_final_state_input()},
0047                         {m_inclusive_kinematics_output().get()});
0048     }
0049 };
0050
0051 } // eicrecon