TICL/plugins/TracksterLinkingRecovery.h

0001 #ifndef RecoHGCal_TICL_TracksterLinkingRecovery_H
0002 #define RecoHGCal_TICL_TracksterLinkingRecovery_H
0003
0004 #include "FWCore/Framework/interface/Frameworkfwd.h"
0005 #include "FWCore/Framework/interface/ESHandle.h"
0006 #include "FWCore/Framework/interface/MakerMacros.h"
0007 #include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
0008 #include "RecoHGCal/TICL/interface/TracksterLinkingAlgoBase.h"
0009
0010 namespace ticl {
0011
0012   class TracksterLinkingRecovery : public TracksterLinkingAlgoBase {
0013   public:
0014     TracksterLinkingRecovery(const edm::ParameterSet& conf,
0015                              edm::ConsumesCollector iC,
0016                              cms::Ort::ONNXRuntime const* onnxRuntime = nullptr)
0017         : TracksterLinkingAlgoBase(conf, iC, onnxRuntime) {}
0018
0019     ~TracksterLinkingRecovery() override {}
0020
0021     void linkTracksters(const Inputs& input,
0022                         std::vector<Trackster>& resultTracksters,
0023                         std::vector<std::vector<unsigned int>>& linkedResultTracksters,
0024                         std::vector<std::vector<unsigned int>>& linkedTracksterIdToInputTracksterId) override;
0025     void initialize(const HGCalDDDConstants* hgcons,
0026                     const hgcal::RecHitTools rhtools,
0027                     const edm::ESHandle<MagneticField> bfieldH,
0028                     const edm::ESHandle<Propagator> propH) override {};
0029
0030     static void fillPSetDescription(edm::ParameterSetDescription& iDesc) { iDesc.add<int>("algo_verbosity", 0); }
0031   };
0032
0033 }  // namespace ticl
0034
0035 #endif