factories/calorimetry/CalorimeterCALOROCReco_factory.h

0001 // SPDX-License-Identifier: LGPL-3.0-or-later
0002 // Copyright (C) 2026 Chun Yuen Tsang, Minho Kim
0003
0004 #pragma once
0005
0006 #include "algorithms/calorimetry/CalorimeterCALOROCReco.h"
0007 #include "services/algorithms_init/AlgorithmsInit_service.h"
0008 #include "extensions/jana/JOmniFactory.h"
0009 #include <edm4hep/SimCalorimeterHitCollection.h>
0010
0011 namespace eicrecon {
0012
0013 class CalorimeterCALOROCReco_factory
0014     : public JOmniFactory<CalorimeterCALOROCReco_factory, CalorimeterCALOROCRecoConfig> {
0015
0016 private:
0017 public:
0018   using AlgoT = eicrecon::CalorimeterCALOROCReco;
0019
0020 private:
0021   std::unique_ptr<AlgoT> m_algo;
0022
0023   PodioInput<edm4hep::SimCalorimeterHit> m_npeHitP_input{this};
0024   PodioInput<edm4eic::RawCALOROCHit> m_CALOROCP_input{this};
0025   PodioInput<edm4hep::SimCalorimeterHit> m_npeHitN_input{this};
0026   PodioInput<edm4eic::RawCALOROCHit> m_CALOROCN_input{this};
0027
0028   PodioOutput<edm4eic::CalorimeterHit> m_rec_hits_output{this};
0029   PodioOutput<edm4hep::RawCalorimeterHit> m_raw_hits_output{this};
0030   PodioOutput<edm4eic::MCRecoCalorimeterHitLink> m_raw_link_output{this};
0031   PodioOutput<edm4eic::MCRecoCalorimeterHitAssociation> m_raw_assoc_output{this};
0032
0033   ParameterRef<std::vector<double>> m_attenuationParameters{this, "attenuationParameters",
0034                                                             config().attenuationParameters};
0035   ParameterRef<std::vector<double>> m_timeWalkCorrectionParameters{
0036       this, "timeWalkCorrectionParameters", config().timeWalkCorrectionParameters};
0037   ParameterRef<std::string> m_attenuationReferencePositionNamePos{
0038       this, "attenuationReferencePositionNamePos", config().attenuationReferencePositionNamePos};
0039   ParameterRef<std::string> m_attenuationReferencePositionNameNeg{
0040       this, "attenuationReferencePositionNameNeg", config().attenuationReferencePositionNameNeg};
0041
0042   ParameterRef<double> m_slope{this, "slope", config().slope};
0043   ParameterRef<double> m_intercept{this, "intercept", config().intercept};
0044   ParameterRef<uint16_t> m_highGainDR{this, "highGainDR", config().highGainDR};
0045   ParameterRef<double> m_gainRatio{this, "gainRatio", config().gainRatio};
0046   ParameterRef<std::string> m_readout{this, "readout", config().readout};
0047   ParameterRef<std::string> m_layerField{this, "layerField", config().layerField};
0048   ParameterRef<std::string> m_sectorField{this, "sectorField", config().sectorField};
0049   ParameterRef<std::string> m_localDetElement{this, "localDetElement", config().localDetElement};
0050   ParameterRef<std::vector<std::string>> m_localDetFields{this, "localDetFields",
0051                                                           config().localDetFields};
0052   ParameterRef<eicrecon::CalorimeterCALOROCRecoConfig::ProxyType> m_proxy_type{this, "proxyType",
0053                                                                                config().proxy_type};
0054   ParameterRef<bool> m_timeWalkCor{this, "timeWalkCor", config().timeWalkCor};
0055   ParameterRef<bool> m_useNpeHitPos{this, "useNpeHitPos", config().useNpeHitPos};
0056
0057   Service<AlgorithmsInit_service> m_algorithmsInit{this};
0058
0059 public:
0060   void Configure() {
0061     m_algo = std::make_unique<AlgoT>(GetPrefix());
0062     m_algo->level(static_cast<algorithms::LogLevel>(logger()->level()));
0063     m_algo->applyConfig(config());
0064     m_algo->init();
0065   }
0066
0067   void Process(int32_t /* run_number */, uint64_t /* event_number */) {
0068     m_algo->process({m_npeHitP_input(), m_CALOROCP_input(), m_npeHitN_input(), m_CALOROCN_input()},
0069                     {m_rec_hits_output().get(), m_raw_hits_output().get(),
0070                      m_raw_link_output().get(), m_raw_assoc_output().get()});
0071   }
0072 };
0073
0074 } // namespace eicrecon