factories/reco/UndoAfterBurnerMCParticles_factory.h

0001 // SPDX-License-Identifier: LGPL-3.0-or-later
0002 // Copyright (C) 2024 Alex Jentsch, Jihee Kim, Brian Page
0003 //
0004
0005 #include "algorithms/reco/UndoAfterBurner.h"
0006 #include "algorithms/reco/UndoAfterBurnerConfig.h"
0007
0008 // Event Model related classes
0009 #include <edm4eic/MCRecoParticleAssociationCollection.h>
0010 #include <edm4eic/ReconstructedParticleCollection.h>
0011 #include <edm4hep/MCParticleCollection.h>
0012
0013 #include "extensions/jana/JOmniFactory.h"
0014
0015 namespace eicrecon {
0016
0017 class UndoAfterBurnerMCParticles_factory
0018     : public JOmniFactory<UndoAfterBurnerMCParticles_factory, UndoAfterBurnerConfig> {
0019
0020 public:
0021   using AlgoT = eicrecon::UndoAfterBurner;
0022
0023 private:
0024   std::unique_ptr<AlgoT> m_algo;
0025
0026   PodioInput<edm4hep::MCParticle> m_mcparts_input{this};
0027   PodioOutput<edm4hep::MCParticle> m_postburn_output{this};
0028
0029   ParameterRef<bool> m_pid_assume_pion_mass{this, "m_pid_assume_pion_mass",
0030                                             config().m_pid_assume_pion_mass};
0031   ParameterRef<double> m_crossing_angle{this, "m_crossing_angle", config().m_crossing_angle};
0032   ParameterRef<double> m_pid_purity{this, "m_pid_purity", config().m_pid_purity};
0033   ParameterRef<bool> m_correct_beam_FX{this, "m_correct_beam_FX", config().m_correct_beam_FX};
0034   ParameterRef<bool> m_pid_use_MC_truth{this, "m_pid_use_MC_truth", config().m_pid_use_MC_truth};
0035
0036 public:
0037   void Configure() {
0038     m_algo = std::make_unique<AlgoT>(GetPrefix());
0039     m_algo->applyConfig(config());
0040   }
0041
0042   void ChangeRun(int32_t /* run_number */) {}
0043
0044   void Process(int32_t /* run_number */, uint64_t /* event_number */) {
0045     m_algo->process({m_mcparts_input()}, {m_postburn_output().get()});
0046   }
0047 };
0048
0049 } // namespace eicrecon