algorithms/reco/ScatteredElectronsTruth.h

0001 // SPDX-License-Identifier: LGPL-3.0-or-later
0002 // Copyright (C) 2024 Daniel Brandenburg
0003
0004 #pragma once
0005
0006 #include <algorithms/algorithm.h>
0007 #include <edm4eic/MCRecoParticleAssociationCollection.h>
0008 #include <edm4eic/ReconstructedParticleCollection.h>
0009 #include <edm4hep/MCParticleCollection.h>
0010 #include <string>
0011 #include <string_view>
0012
0013 #include "algorithms/interfaces/WithPodConfig.h"
0014 #include "services/particle/ParticleSvc.h"
0015
0016 namespace eicrecon {
0017
0018 using ScatteredElectronsTruthAlgorithm = algorithms::Algorithm<
0019     algorithms::Input<edm4hep::MCParticleCollection, edm4eic::ReconstructedParticleCollection,
0020                       edm4eic::MCRecoParticleAssociationCollection>,
0021     algorithms::Output<edm4eic::ReconstructedParticleCollection>>;
0022
0023 class ScatteredElectronsTruth : public ScatteredElectronsTruthAlgorithm,
0024                                 public WithPodConfig<NoConfig> {
0025
0026 public:
0027   ScatteredElectronsTruth(std::string_view name)
0028       : ScatteredElectronsTruthAlgorithm{name,
0029                                          {"MCParticles", "inputParticles", "inputAssociations"},
0030                                          {"ReconstructedParticles"},
0031                                          "Output a list of possible scattered electrons using "
0032                                          "truth MC Particle associations."} {}
0033
0034   void init() final;
0035   void process(const Input&, const Output&) const final;
0036
0037 private:
0038   const algorithms::ParticleSvc& m_particleSvc = algorithms::ParticleSvc::instance();
0039 };
0040
0041 } // namespace eicrecon