factories/reco/TransformBreitFrame_factory.h

0001 // SPDX-License-Identifier: LGPL-3.0-or-later
0002 // Copyright (C) 2024 John Lajoie
0003
0004 #pragma once
0005
0006 #include <JANA/JEvent.h>
0007 #include <edm4hep/MCParticleCollection.h>
0008 #include <edm4eic/InclusiveKinematicsCollection.h>
0009 #include <edm4eic/ReconstructedParticleCollection.h>
0010 #include <memory>
0011 #include <string>
0012 #include <utility>
0013 #include <vector>
0014
0015 #include "algorithms/reco/TransformBreitFrame.h"
0016 #include "extensions/jana/JOmniFactory.h"
0017 #include "services/algorithms_init/AlgorithmsInit_service.h"
0018
0019 namespace eicrecon {
0020
0021 class TransformBreitFrame_factory : public JOmniFactory<TransformBreitFrame_factory, NoConfig> {
0022
0023 public:
0024   // algorithm to run
0025   using Algo = eicrecon::TransformBreitFrame;
0026
0027 private:
0028   std::unique_ptr<Algo> m_algo;
0029
0030   // input collection
0031   PodioInput<edm4hep::MCParticle> m_in_mcpart{this};
0032   PodioInput<edm4eic::InclusiveKinematics> m_in_kine{this};
0033   PodioInput<edm4eic::ReconstructedParticle> m_in_part{this};
0034
0035   // output collection
0036   PodioOutput<edm4eic::ReconstructedParticle> m_out_part{this};
0037
0038   Service<AlgorithmsInit_service> m_algorithmsInit{this};
0039
0040 public:
0041   void Configure() {
0042     m_algo = std::make_unique<Algo>(GetPrefix());
0043     m_algo->level(static_cast<algorithms::LogLevel>(logger()->level()));
0044     m_algo->applyConfig(config());
0045     m_algo->init();
0046   }
0047
0048   void Process(int32_t /* run_number */, uint64_t /* event_number */) {
0049     m_algo->process({m_in_mcpart(), m_in_kine(), m_in_part()}, {m_out_part().get()});
0050   }
0051
0052 }; // end TransfromBreitFrame_factory definition
0053
0054 } // namespace eicrecon