algorithms/calorimetry/CalorimeterTruthClustering.h

0001
0002 // SPDX-License-Identifier: LGPL-3.0-or-later
0003 // Copyright (C) 2022 Sylvester Joosten, Whitney Armstrong, Wouter Deconinck
0004
0005 #pragma once
0006
0007 #include <algorithms/algorithm.h>
0008 #include <edm4eic/CalorimeterHitCollection.h>
0009 #include <edm4eic/ProtoClusterCollection.h>
0010 #include <edm4hep/SimCalorimeterHitCollection.h>
0011 #include <string>
0012 #include <string_view>
0013
0014 #include "algorithms/interfaces/WithPodConfig.h"
0015
0016 namespace eicrecon {
0017
0018 using CalorimeterTruthClusteringAlgorithm = algorithms::Algorithm<
0019     algorithms::Input<edm4eic::CalorimeterHitCollection, edm4hep::SimCalorimeterHitCollection>,
0020     algorithms::Output<edm4eic::ProtoClusterCollection>>;
0021
0022 class CalorimeterTruthClustering : public CalorimeterTruthClusteringAlgorithm,
0023                                    public WithPodConfig<NoConfig> {
0024
0025 public:
0026   CalorimeterTruthClustering(std::string_view name)
0027       : CalorimeterTruthClusteringAlgorithm{name,
0028                                             {"inputHitCollection", "inputSimHitCollection"},
0029                                             {"outputProtoClusterCollection"},
0030                                             "Use truth information for clustering."} {}
0031
0032 public:
0033   void init() final;
0034   void process(const Input&, const Output&) const final;
0035 };
0036
0037 } // namespace eicrecon