Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2024-10-16 05:06:32

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