factories/calorimetry/CalorimeterIslandCluster_factory.h

0001 // SPDX-License-Identifier: LGPL-3.0-or-later
0002 // Copyright (C) 2023 Wouter Deconinck
0003
0004 #pragma once
0005
0006 #include "algorithms/calorimetry/CalorimeterIslandCluster.h"
0007 #include "services/algorithms_init/AlgorithmsInit_service.h"
0008 #include "extensions/jana/JOmniFactory.h"
0009
0010 namespace eicrecon {
0011
0012 class CalorimeterIslandCluster_factory
0013     : public JOmniFactory<CalorimeterIslandCluster_factory, CalorimeterIslandClusterConfig> {
0014 public:
0015   using AlgoT = eicrecon::CalorimeterIslandCluster;
0016
0017 private:
0018   std::unique_ptr<AlgoT> m_algo;
0019
0020   PodioInput<edm4eic::CalorimeterHit> m_calo_hit_input{this};
0021   PodioOutput<edm4eic::ProtoCluster> m_proto_cluster_output{this};
0022
0023   ParameterRef<double> m_sectorDist{this, "sectorDist", config().sectorDist};
0024   // ParameterRef<std::vector<double>> m_localDistXY{this, "localDistXY", config().localDistXY};
0025   ParameterRef<std::vector<double>> m_localDistXZ{this, "localDistXZ", config().localDistXZ};
0026   ParameterRef<std::vector<double>> m_localDistYZ{this, "localDistYZ", config().localDistYZ};
0027   ParameterRef<std::vector<double>> m_globallDistRPhi{this, "globalDistRPhi",
0028                                                       config().globalDistRPhi};
0029   ParameterRef<std::vector<double>> m_globalDistEtaPhi{this, "globalDistEtaPhi",
0030                                                        config().globalDistEtaPhi};
0031   ParameterRef<std::vector<double>> m_dimScalledLocalDistXY{this, "dimScaledLocalDistXY",
0032                                                             config().dimScaledLocalDistXY};
0033   ParameterRef<std::string> m_adjacencyMatrix{this, "adjacencyMatrix", config().adjacencyMatrix};
0034   ParameterRef<std::string> m_readout{this, "readoutClass", config().readout};
0035   ParameterRef<bool> m_splitCluster{this, "splitCluster", config().splitCluster};
0036   ParameterRef<double> m_minClusterHitEdep{this, "minClusterHitEdep", config().minClusterHitEdep};
0037   ParameterRef<double> m_minClusterCenterEdep{this, "minClusterCenterEdep",
0038                                               config().minClusterCenterEdep};
0039   ParameterRef<std::string> m_tepm{this, "transverseEnergyProfileMetric",
0040                                    config().transverseEnergyProfileMetric};
0041   ParameterRef<double> m_teps{this, "transverseEnergyProfileScale",
0042                               config().transverseEnergyProfileScale};
0043
0044   Service<AlgorithmsInit_service> m_algorithmsInit{this};
0045
0046 public:
0047   void Configure() {
0048     m_algo = std::make_unique<AlgoT>(GetPrefix());
0049     m_algo->level(static_cast<algorithms::LogLevel>(logger()->level()));
0050     // Remove spaces from adjacency matrix
0051     // cfg.adjacencyMatrix.erase(
0052     //  std::remove_if(cfg.adjacencyMatrix.begin(), cfg.adjacencyMatrix.end(), ::isspace), cfg.adjacencyMatrix.end());
0053     m_algo->applyConfig(config());
0054     m_algo->init();
0055   }
0056
0057   void ChangeRun(int32_t /* run_number */) {}
0058
0059   void Process(int32_t /* run_number */, uint64_t /* event_number */) {
0060     m_algo->process({m_calo_hit_input()}, {m_proto_cluster_output().get()});
0061   }
0062 };
0063
0064 } // namespace eicrecon