File indexing completed on 2025-07-14 08:14:43
0001
0002
0003
0004
0005 #include "algorithms/reco/UndoAfterBurner.h"
0006 #include "algorithms/reco/UndoAfterBurnerConfig.h"
0007
0008
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 ) {}
0043
0044 void Process(int32_t , uint64_t ) {
0045 m_algo->process({m_mcparts_input()}, {m_postburn_output().get()});
0046 }
0047 };
0048
0049 }