factories/digi/EICROCDigitization_factory.h

0001 // SPDX-License-Identifier: LGPL-3.0-or-later
0002 // Copyright (C) 2024 Chun Yuen Tsang
0003
0004 #pragma once
0005
0006 #include <iostream>
0007
0008 #include "algorithms/digi/EICROCDigitization.h"
0009 #include "extensions/jana/JOmniFactory.h"
0010 #include "services/algorithms_init/AlgorithmsInit_service.h"
0011
0012 namespace eicrecon {
0013
0014 class EICROCDigitization_factory
0015     : public JOmniFactory<EICROCDigitization_factory, EICROCDigitizationConfig> {
0016 public:
0017   using AlgoT = eicrecon::EICROCDigitization;
0018
0019 private:
0020   std::unique_ptr<AlgoT> m_algo;
0021
0022   PodioInput<edm4hep::RawTimeSeries> m_in_sim_track{this};
0023
0024   PodioOutput<edm4eic::RawTrackerHit> m_out_reco_particles{this};
0025
0026   ParameterRef<double> m_t_thres{this, "tThreshold", config().t_thres};
0027
0028   Service<AlgorithmsInit_service> m_algorithmsInit{this};
0029
0030 public:
0031   void Configure() {
0032     m_algo = std::make_unique<eicrecon::EICROCDigitization>(GetPrefix());
0033     m_algo->level(static_cast<algorithms::LogLevel>(logger()->level()));
0034     m_algo->applyConfig(config());
0035     m_algo->init();
0036   }
0037
0038   void ChangeRun(int32_t /* run_number */) {}
0039
0040   void Process(int32_t /* run_number */, uint64_t /* event_number */) {
0041     m_algo->process({m_in_sim_track()}, {m_out_reco_particles().get()});
0042   }
0043 };
0044 } // namespace eicrecon