Back to home page

EIC code displayed by LXR

 
 

    


File indexing completed on 2025-07-14 08:14:43

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