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 #pragma once
0006
0007 #include "algorithms/reco/UndoAfterBurner.h"
0008 #include "algorithms/reco/UndoAfterBurnerConfig.h"
0009
0010 // Event Model related classes
0011 #include <edm4eic/MCRecoParticleAssociationCollection.h>
0012 #include <edm4eic/ReconstructedParticleCollection.h>
0013 #include <edm4hep/MCParticleCollection.h>
0014
0015 #include "extensions/jana/JOmniFactory.h"
0016
0017 namespace eicrecon {
0018
0019 class UndoAfterBurnerMCParticles_factory
0020     : public JOmniFactory<UndoAfterBurnerMCParticles_factory, UndoAfterBurnerConfig> {
0021
0022 public:
0023   using AlgoT = eicrecon::UndoAfterBurner;
0024
0025 private:
0026   std::unique_ptr<AlgoT> m_algo;
0027
0028   PodioInput<edm4hep::MCParticle> m_mcparts_input{this};
0029   PodioOutput<edm4hep::MCParticle> m_postburn_output{this};
0030
0031   ParameterRef<bool> m_pid_assume_pion_mass{this, "m_pid_assume_pion_mass",
0032                                             config().m_pid_assume_pion_mass};
0033   ParameterRef<double> m_crossing_angle{this, "m_crossing_angle", config().m_crossing_angle};
0034   ParameterRef<double> m_pid_purity{this, "m_pid_purity", config().m_pid_purity};
0035   ParameterRef<bool> m_correct_beam_FX{this, "m_correct_beam_FX", config().m_correct_beam_FX};
0036   ParameterRef<bool> m_pid_use_MC_truth{this, "m_pid_use_MC_truth", config().m_pid_use_MC_truth};
0037
0038 public:
0039   void Configure() {
0040     m_algo = std::make_unique<AlgoT>(GetPrefix());
0041     m_algo->applyConfig(config());
0042   }
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