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 namespace eicrecon {
0015
0016 using CalorimeterTruthClusteringAlgorithm = algorithms::Algorithm<
0017     algorithms::Input<edm4eic::CalorimeterHitCollection, edm4hep::SimCalorimeterHitCollection>,
0018     algorithms::Output<edm4eic::ProtoClusterCollection>>;
0019
0020 class CalorimeterTruthClustering : public CalorimeterTruthClusteringAlgorithm {
0021
0022 public:
0023   CalorimeterTruthClustering(std::string_view name)
0024       : CalorimeterTruthClusteringAlgorithm{name,
0025                                             {"inputHitCollection", "inputSimHitCollection"},
0026                                             {"outputProtoClusterCollection"},
0027                                             "Use truth information for clustering."} {}
0028
0029 public:
0030   void init() final;
0031   void process(const Input&, const Output&) const final;
0032 };
0033
0034 } // namespace eicrecon