factories/fardetectors/FarDetectorLinearTracking_factory.h

0001 // SPDX-License-Identifier: LGPL-3.0-or-later
0002 // Copyright (C) 2024 - 2025, Simon Gardner
0003
0004 #pragma once
0005
0006 #include "services/geometry/dd4hep/DD4hep_service.h"
0007
0008 // Event Model related classes
0009 #include <edm4eic/EDM4eicVersion.h>
0010 #include <edm4eic/MCRecoTrackerHitAssociationCollection.h>
0011 #if EDM4EIC_BUILD_VERSION >= EDM4EIC_VERSION(8, 7, 0)
0012 #include <edm4eic/MCRecoTrackerHitLinkCollection.h>
0013 #endif
0014 #include <edm4eic/MCRecoTrackParticleAssociationCollection.h>
0015 #include <edm4eic/TrackCollection.h>
0016 #include <edm4eic/Measurement2DCollection.h>
0017 #include <algorithms/fardetectors/FarDetectorLinearTracking.h>
0018
0019 #include <extensions/jana/JOmniFactory.h>
0020 #include <spdlog/logger.h>
0021
0022 namespace eicrecon {
0023
0024 class FarDetectorLinearTracking_factory
0025     : public JOmniFactory<FarDetectorLinearTracking_factory, FarDetectorLinearTrackingConfig> {
0026
0027 public:
0028   using AlgoT = eicrecon::FarDetectorLinearTracking;
0029
0030 private:
0031   std::unique_ptr<AlgoT> m_algo;
0032
0033   VariadicPodioInput<edm4eic::Measurement2D> m_hits_input{this};
0034 #if EDM4EIC_BUILD_VERSION >= EDM4EIC_VERSION(8, 7, 0)
0035   PodioInput<edm4eic::MCRecoTrackerHitLink> m_hits_links_input{this};
0036 #endif
0037   PodioInput<edm4eic::MCRecoTrackerHitAssociation> m_hits_association_input{this};
0038   PodioOutput<edm4eic::Track> m_tracks_output{this};
0039 #if EDM4EIC_BUILD_VERSION >= EDM4EIC_VERSION(8, 7, 0)
0040   PodioOutput<edm4eic::MCRecoTrackParticleLink> m_tracks_links_output{this};
0041 #endif
0042   PodioOutput<edm4eic::MCRecoTrackParticleAssociation> m_tracks_association_output{this};
0043
0044   ParameterRef<std::size_t> n_layer{this, "numLayers", config().n_layer};
0045   ParameterRef<std::vector<double>> layer_weights{this, "layerWeights", config().layer_weights};
0046   ParameterRef<std::size_t> layer_hits_max{this, "layerHitsMax", config().layer_hits_max};
0047   ParameterRef<float> chi2_max{this, "chi2Max", config().chi2_max};
0048
0049 public:
0050   void Configure() {
0051     m_algo = std::make_unique<AlgoT>(GetPrefix());
0052     m_algo->level(static_cast<algorithms::LogLevel>(logger()->level()));
0053     m_algo->applyConfig(config());
0054     m_algo->init();
0055   }
0056
0057   void Process(int32_t /* run_number */, uint64_t /* event_number */) {
0058
0059     try {
0060       std::vector<gsl::not_null<const edm4eic::Measurement2DCollection*>> hits;
0061       for (const auto& hit : m_hits_input()) {
0062         hits.push_back(gsl::not_null<const edm4eic::Measurement2DCollection*>{hit});
0063       }
0064
0065       // Prepare the input tuple
0066       auto input = std::make_tuple(hits,
0067 #if EDM4EIC_BUILD_VERSION >= EDM4EIC_VERSION(8, 7, 0)
0068                                    m_hits_links_input(),
0069 #endif
0070                                    m_hits_association_input());
0071
0072       m_algo->process(input, {m_tracks_output().get(),
0073 #if EDM4EIC_BUILD_VERSION >= EDM4EIC_VERSION(8, 7, 0)
0074                               m_tracks_links_output().get(),
0075 #endif
0076                               m_tracks_association_output().get()});
0077     } catch (std::exception& e) {
0078       throw JException(e.what());
0079     }
0080   }
0081 };
0082
0083 } // namespace eicrecon