factories/reco/SecondaryVerticesHelix_factory.h

0001 // SPDX-License-Identifier: LGPL-3.0-or-later
0002 // Copyright (C) 2025 Xin Dong
0003
0004 #pragma once
0005
0006 #include <JANA/JEvent.h>
0007 #include <memory>
0008 #include <string>
0009 #include <utility>
0010 #include <vector>
0011
0012 #include "algorithms/reco/SecondaryVerticesHelix.h"
0013 #include "algorithms/reco/SecondaryVerticesHelixConfig.h"
0014 #include "services/algorithms_init/AlgorithmsInit_service.h"
0015 #include "extensions/jana/JOmniFactory.h"
0016
0017 namespace eicrecon {
0018
0019 class SecondaryVerticesHelix_factory
0020     : public JOmniFactory<SecondaryVerticesHelix_factory, SecondaryVerticesHelixConfig> {
0021
0022 public:
0023   using AlgoT = eicrecon::SecondaryVerticesHelix;
0024
0025 private:
0026   std::unique_ptr<AlgoT> m_algo;
0027
0028   PodioInput<edm4eic::Vertex> m_rc_vertices_input{this};
0029   PodioInput<edm4eic::ReconstructedParticle> m_rc_parts_input{this};
0030
0031   // Declare outputs
0032   PodioOutput<edm4eic::Vertex> m_secondary_vertices_output{this};
0033
0034   // Declare parameters
0035   ParameterRef<bool> m_unlikesign{this, "unlikesign", config().unlikesign};
0036   ParameterRef<float> m_minDca{this, "minDca", config().minDca};
0037   ParameterRef<float> m_maxDca12{this, "maxDca12", config().maxDca12};
0038   ParameterRef<float> m_maxDca{this, "maxDca", config().maxDca};
0039   ParameterRef<float> m_minCostheta{this, "minCostheta", config().minCostheta};
0040
0041   Service<AlgorithmsInit_service> m_algorithmsInit{this};
0042
0043 public:
0044   void Configure() {
0045     m_algo = std::make_unique<AlgoT>(GetPrefix());
0046     m_algo->level((algorithms::LogLevel)logger()->level());
0047
0048     m_algo->applyConfig(config());
0049     m_algo->init();
0050   }
0051
0052   void Process(int32_t /* run_number */, uint64_t /* event_number */) {
0053     m_algo->process({m_rc_vertices_input(), m_rc_parts_input()},
0054                     {m_secondary_vertices_output().get()});
0055   }
0056 };
0057
0058 } // namespace eicrecon