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
0012 namespace eicrecon {
0013
0014   class FarForwardLambdaReconstruction_factory : public JOmniFactory<FarForwardLambdaReconstruction_factory,FarForwardLambdaReconstructionConfig> {
0015
0016     public:
0017       using AlgoT = eicrecon::FarForwardLambdaReconstruction;
0018     private:
0019       std::unique_ptr<AlgoT> m_algo;
0020     PodioInput<edm4eic::ReconstructedParticle> m_neutrals_input {this};
0021     PodioOutput<edm4eic::ReconstructedParticle> m_lambda_output {this};
0022     PodioOutput<edm4eic::ReconstructedParticle> m_lambda_decay_products_cm_output {this};
0023
0024     ParameterRef<double> m_rot_y {this, "globalToProtonRotation", config().globalToProtonRotation};
0025     ParameterRef<double> m_zmax {this, "zMax", config().zMax};
0026     ParameterRef<double> m_lambda_max_mass_dev {this, "lambdaMaxMassDev", config().lambdaMaxMassDev};
0027     ParameterRef<int> m_iterations {this, "iterations", config().iterations};
0028     Service<AlgorithmsInit_service> m_algorithmsInit {this};
0029
0030 public:
0031     void Configure() {
0032         m_algo = std::make_unique<AlgoT>(GetPrefix());
0033         m_algo->level((algorithms::LogLevel)logger()->level());
0034
0035         m_algo->applyConfig(config());
0036         m_algo->init();
0037     }
0038
0039     void ChangeRun(int64_t run_number) {
0040     }
0041
0042     void Process(int64_t run_number, uint64_t event_number) {
0043       m_algo->process({m_neutrals_input()},{m_lambda_output().get(), m_lambda_decay_products_cm_output().get()});
0044     }
0045 };
0046
0047 } // eicrecon