factories/reco/FarForwardLambdaReconstruction_factory.h

0001 // SPDX-License-Identifier: LGPL-3.0-or-later
0002 // Copyright (C) 2025 Sebouh Paul
0003
0004 #pragma once
0005
0006 #include "algorithms/reco/FarForwardLambdaReconstruction.h"
0007 #include "algorithms/reco/FarForwardLambdaReconstructionConfig.h"
0008 #include "services/algorithms_init/AlgorithmsInit_service.h"
0009 #include "extensions/jana/JOmniFactory.h"
0010
0011 namespace eicrecon {
0012
0013 class FarForwardLambdaReconstruction_factory
0014     : public JOmniFactory<FarForwardLambdaReconstruction_factory,
0015                           FarForwardLambdaReconstructionConfig> {
0016
0017 public:
0018   using AlgoT = eicrecon::FarForwardLambdaReconstruction;
0019
0020 private:
0021   std::unique_ptr<AlgoT> m_algo;
0022   PodioInput<edm4eic::ReconstructedParticle> m_neutrals_input{this};
0023   PodioOutput<edm4eic::ReconstructedParticle> m_lambda_output{this};
0024   PodioOutput<edm4eic::ReconstructedParticle> m_lambda_decay_products_cm_output{this};
0025
0026   ParameterRef<std::string> m_offset_position_name{this, "offsetPositionName",
0027                                                    config().offsetPositionName};
0028   ParameterRef<double> m_rot_y{this, "globalToProtonRotation", config().globalToProtonRotation};
0029   ParameterRef<double> m_lambda_max_mass_dev{this, "lambdaMaxMassDev", config().lambdaMaxMassDev};
0030   ParameterRef<int> m_iterations{this, "iterations", config().iterations};
0031   Service<AlgorithmsInit_service> m_algorithmsInit{this};
0032
0033 public:
0034   void Configure() {
0035     m_algo = std::make_unique<AlgoT>(GetPrefix());
0036     m_algo->level((algorithms::LogLevel)logger()->level());
0037
0038     m_algo->applyConfig(config());
0039     m_algo->init();
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_neutrals_input()},
0046                     {m_lambda_output().get(), m_lambda_decay_products_cm_output().get()});
0047   }
0048 };
0049
0050 } // namespace eicrecon