Inheritance diagram for TracksterLinksProducer:

Public Member Functions
void	beginRun (edm::Run const &iEvent, edm::EventSetup const &es) override

void	produce (edm::Event &, const edm::EventSetup &) override

	TracksterLinksProducer (const edm::ParameterSet &ps, const ONNXRuntime *)

	~TracksterLinksProducer () override

Public Member Functions inherited from edm::stream::EDProducer< edm::GlobalCache< ONNXRuntime > >
	EDProducer ()=default

	EDProducer (const EDProducer &)=delete

bool	hasAbilityToProduceInBeginLumis () const final

bool	hasAbilityToProduceInBeginProcessBlocks () const final

bool	hasAbilityToProduceInBeginRuns () const final

bool	hasAbilityToProduceInEndLumis () const final

bool	hasAbilityToProduceInEndProcessBlocks () const final

bool	hasAbilityToProduceInEndRuns () const final

const EDProducer &	operator= (const EDProducer &)=delete

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &descriptions)

static void	globalEndJob (const ONNXRuntime *)

static std::unique_ptr< ONNXRuntime >	initializeGlobalCache (const edm::ParameterSet &iConfig)

Private Member Functions
void	dumpTrackster (const Trackster &) const

void	printTrackstersDebug (const std::vector< Trackster > &, const char *label) const

Private Attributes
std::string	algoType_

const edm::ESGetToken< MagneticField, IdealMagneticFieldRecord >	bfield_token_

const edm::EDGetTokenT< std::vector< reco::CaloCluster > >	clusters_token_

const edm::EDGetTokenT< edm::ValueMap< std::pair< float, float > > >	clustersTime_token_

const std::string	detector_

const edm::ESGetToken< CaloGeometry, CaloGeometryRecord >	geometry_token_

edm::ESGetToken< HGCalDDDConstants, IdealGeometryRecord >	hdc_token_

const HGCalDDDConstants *	hgcons_

std::unique_ptr< TracksterInferenceAlgoBase >	inferenceAlgo_

std::unique_ptr< TracksterLinkingAlgoBase >	linkingAlgo_

std::vector< edm::EDGetTokenT< std::vector< float > > >	original_masks_tokens_

const edm::ESGetToken< Propagator, TrackingComponentsRecord >	propagator_token_

const std::string	propName_

const bool	regressionAndPid_

hgcal::RecHitTools	rhtools_

std::vector< edm::EDGetTokenT< std::vector< Trackster > > >	tracksters_tokens_

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer< edm::GlobalCache< ONNXRuntime > >
using	CacheTypes = CacheContexts< T... >

using	GlobalCache = typename CacheTypes::GlobalCache

using	HasAbility = AbilityChecker< T... >

using	InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache

using	LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache

using	LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >

using	LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache

using	RunCache = typename CacheTypes::RunCache

using	RunContext = RunContextT< RunCache, GlobalCache >

using	RunSummaryCache = typename CacheTypes::RunSummaryCache

Detailed Description

Definition at line 52 of file TracksterLinksProducer.cc.

Constructor & Destructor Documentation

◆ TracksterLinksProducer()

TracksterLinksProducer::TracksterLinksProducer	(	const edm::ParameterSet &	ps,
		const ONNXRuntime *	onnxRuntime
	)

explicit

Definition at line 90 of file TracksterLinksProducer.cc.

References algoType_, detector_, get, edm::ParameterSet::getParameter(), hdc_token_, inferenceAlgo_, linkingAlgo_, hltTiclTracksterLinks_cfi::linkingPSet, original_masks_tokens_, AlCaHLTBitMon_QueryRunRegistry::string, makeGlobalPositionRcd_cfg::tag, and tracksters_tokens_.

     : algoType_(ps.getParameter<edm::ParameterSet>("linkingPSet").getParameter<std::string>("type")),
       clusters_token_(consumes<std::vector<reco::CaloCluster>>(ps.getParameter<edm::InputTag>("layer_clusters"))),
       clustersTime_token_(
           consumes<edm::ValueMap<std::pair<float, float>>>(ps.getParameter<edm::InputTag>("layer_clustersTime"))),
       regressionAndPid_(ps.getParameter<bool>("regressionAndPid")),
       geometry_token_(esConsumes<CaloGeometry, CaloGeometryRecord, edm::Transition::BeginRun>()),
       detector_(ps.getParameter<std::string>("detector")),
       propName_(ps.getParameter<std::string>("propagator")),
       bfield_token_(esConsumes<MagneticField, IdealMagneticFieldRecord, edm::Transition::BeginRun>()),
       propagator_token_(
           esConsumes<Propagator, TrackingComponentsRecord, edm::Transition::BeginRun>(edm::ESInputTag("", propName_))) {
   // Loop over the edm::VInputTag and append the token to tracksters_tokens_
   for (auto const &tag : ps.getParameter<std::vector<edm::InputTag>>("tracksters_collections")) {
     tracksters_tokens_.emplace_back(consumes<std::vector<Trackster>>(tag));
   }
   //Loop over the edm::VInputTag of masks and append the token to original_masks_tokens_
   for (auto const &tag : ps.getParameter<std::vector<edm::InputTag>>("original_masks")) {
     original_masks_tokens_.emplace_back(consumes<std::vector<float>>(tag));
   }
   // Initialize inference algorithm using the factory
   std::string inferencePlugin = ps.getParameter<std::string>("inferenceAlgo");
   edm::ParameterSet inferencePSet = ps.getParameter<edm::ParameterSet>("pluginInferenceAlgo" + inferencePlugin);
   inferenceAlgo_ = std::unique_ptr<TracksterInferenceAlgoBase>(
       TracksterInferenceAlgoFactory::get()->create(inferencePlugin, inferencePSet));
 
   // New trackster collection after linking
   produces<std::vector<Trackster>>();
 
   // Links
   produces<std::vector<std::vector<unsigned int>>>();
   produces<std::vector<std::vector<unsigned int>>>("linkedTracksterIdToInputTracksterId");
   // LayerClusters Mask
   produces<std::vector<float>>();
 
   auto linkingPSet = ps.getParameter<edm::ParameterSet>("linkingPSet");
 
   if (algoType_ == "Skeletons") {
     std::string detectorName_ = (detector_ == "HFNose") ? "HGCalHFNoseSensitive" : "HGCalEESensitive";
     hdc_token_ = esConsumes<HGCalDDDConstants, IdealGeometryRecord, edm::Transition::BeginRun>(
         edm::ESInputTag("", detectorName_));
   }
 
   linkingAlgo_ = TracksterLinkingPluginFactory::get()->create(algoType_, linkingPSet, consumesCollector(), onnxRuntime);
 }

◆ ~TracksterLinksProducer()

TracksterLinksProducer::~TracksterLinksProducer ( )

inlineoverride

Definition at line 55 of file TracksterLinksProducer.cc.

55 {};

Member Function Documentation

◆ beginRun()

void TracksterLinksProducer::beginRun	(	edm::Run const &	iEvent,
		edm::EventSetup const &	es
	)

override

Definition at line 146 of file TracksterLinksProducer.cc.

References algoType_, bfield_token_, relativeConstraints::geom, geometry_token_, edm::EventSetup::getHandle(), hdc_token_, hgcons_, linkingAlgo_, edm::ESHandle< T >::product(), TrackCandidateProducer_cfi::propagator, propagator_token_, rhtools_, and hgcal::RecHitTools::setGeometry().

                                                                                    {
   if (algoType_ == "Skeletons") {
     edm::ESHandle<HGCalDDDConstants> hdc = es.getHandle(hdc_token_);
     hgcons_ = hdc.product();
   }
 
   edm::ESHandle<CaloGeometry> geom = es.getHandle(geometry_token_);
   rhtools_.setGeometry(*geom);
 
   edm::ESHandle<MagneticField> bfield = es.getHandle(bfield_token_);
   edm::ESHandle<Propagator> propagator = es.getHandle(propagator_token_);
 
   linkingAlgo_->initialize(hgcons_, rhtools_, bfield, propagator);
 };

◆ dumpTrackster()

void TracksterLinksProducer::dumpTrackster ( const Trackster & t ) const

private

Definition at line 161 of file TracksterLinksProducer.cc.

References alignBH_cfg::fixed, ALCARECOEcalPhiSym_cff::float, LogDebug, AlCaHLTBitMon_ParallelJobs::p, alignCSCRings::s, mathSSE::sqrt(), and submitPVValidationJobs::t.

                                                                    {
   auto e_over_h = (t.raw_em_pt() / ((t.raw_pt() - t.raw_em_pt()) != 0. ? (t.raw_pt() - t.raw_em_pt()) : 1.));
   LogDebug("TracksterLinksProducer")
       << "\nTrackster raw_pt: " << t.raw_pt() << " raw_em_pt: " << t.raw_em_pt() << " eoh: " << e_over_h
       << " barycenter: " << t.barycenter() << " eta,phi (baricenter): " << t.barycenter().eta() << ", "
       << t.barycenter().phi() << " eta,phi (eigen): " << t.eigenvectors(0).eta() << ", " << t.eigenvectors(0).phi()
       << " pt(eigen): " << std::sqrt(t.eigenvectors(0).Unit().perp2()) * t.raw_energy() << " seedID: " << t.seedID()
       << " seedIndex: " << t.seedIndex() << " size: " << t.vertices().size() << " average usage: "
       << (std::accumulate(std::begin(t.vertex_multiplicity()), std::end(t.vertex_multiplicity()), 0.) /
           (float)t.vertex_multiplicity().size())
       << " raw_energy: " << t.raw_energy() << " regressed energy: " << t.regressed_energy()
       << " probs(ga/e/mu/np/cp/nh/am/unk): ";
   for (auto const &p : t.id_probabilities()) {
     LogDebug("TracksterLinksProducer") << std::fixed << p << " ";
   }
   LogDebug("TracksterLinksProducer") << " sigmas: ";
   for (auto const &s : t.sigmas()) {
     LogDebug("TracksterLinksProducer") << s << " ";
   }
   LogDebug("TracksterLinksProducer") << std::endl;
 }

◆ fillDescriptions()

void TracksterLinksProducer::fillDescriptions ( edm::ConfigurationDescriptions & descriptions )

static

Definition at line 278 of file TracksterLinksProducer.cc.

References edm::ConfigurationDescriptions::add(), edm::ParameterSetDescription::addNode(), submitPVResolutionJobs::desc, ProducerED_cfi::InputTag, and AlCaHLTBitMon_QueryRunRegistry::string.

                                                                                         {
   edm::ParameterSetDescription desc;
   edm::ParameterSetDescription linkingDesc;
   linkingDesc.addNode(edm::PluginDescription<TracksterLinkingPluginFactory>("type", "Skeletons", true));
   // Inference Plugins
   edm::ParameterSetDescription inferenceDesc;
   inferenceDesc.addNode(edm::PluginDescription<TracksterInferenceAlgoFactory>("type", "TracksterInferenceByDNN", true));
   desc.add<edm::ParameterSetDescription>("pluginInferenceAlgoTracksterInferenceByDNN", inferenceDesc);
 
   edm::ParameterSetDescription inferenceDescCNNv4;
   inferenceDescCNNv4.addNode(
       edm::PluginDescription<TracksterInferenceAlgoFactory>("type", "TracksterInferenceByCNNv4", true));
   desc.add<edm::ParameterSetDescription>("pluginInferenceAlgoTracksterInferenceByCNNv4", inferenceDescCNNv4);
 
   desc.add<edm::ParameterSetDescription>("linkingPSet", linkingDesc);
   desc.add<std::vector<edm::InputTag>>("tracksters_collections", {edm::InputTag("ticlTrackstersCLUE3DHigh")});
   desc.add<std::vector<edm::InputTag>>("original_masks",
                                        {edm::InputTag("hgcalMergeLayerClusters", "InitialLayerClustersMask")});
   desc.add<edm::InputTag>("layer_clusters", edm::InputTag("hgcalMergeLayerClusters"));
   desc.add<edm::InputTag>("layer_clustersTime", edm::InputTag("hgcalMergeLayerClusters", "timeLayerCluster"));
   desc.add<bool>("regressionAndPid", false);
   desc.add<std::string>("detector", "HGCAL");
   desc.add<std::string>("propagator", "PropagatorWithMaterial");
   desc.add<std::string>("inferenceAlgo", "TracksterInferenceByDNN");
   descriptions.add("tracksterLinksProducer", desc);
 }

◆ globalEndJob()

void TracksterLinksProducer::globalEndJob ( const ONNXRuntime * )

static

Definition at line 144 of file TracksterLinksProducer.cc.

144 {}

◆ initializeGlobalCache()

std::unique_ptr< ONNXRuntime > TracksterLinksProducer::initializeGlobalCache ( const edm::ParameterSet & iConfig )

static

Definition at line 136 of file TracksterLinksProducer.cc.

References edm::FileInPath::fullPath(), and edm::ParameterSet::getParameter().

                                                                                                        {
   auto const &pluginPset = iConfig.getParameter<edm::ParameterSet>("linkingPSet");
   if (pluginPset.exists("onnxModelPath"))
     return std::make_unique<ONNXRuntime>(pluginPset.getParameter<edm::FileInPath>("onnxModelPath").fullPath());
   else
     return std::unique_ptr<ONNXRuntime>(nullptr);
 }

◆ printTrackstersDebug()

void TracksterLinksProducer::printTrackstersDebug	(	const std::vector< Trackster > &	tracksters,
		const char *	label
	)		const

private

Definition at line 253 of file TracksterLinksProducer.cc.

References alignBH_cfg::fixed, ALCARECOEcalPhiSym_cff::float, label, CrabHelper::log, LogDebug, AlCaHLTBitMon_ParallelJobs::p, alignCSCRings::s, mathSSE::sqrt(), submitPVValidationJobs::t, and HitToTracksterAssociation_cfi::tracksters.

                                                                                                                  {
   int counter = 0;
   LogDebug("TracksterLinksProducer").log([&](auto &log) {
     for (auto const &t : tracksters) {
       log << counter++ << " TracksterLinksProducer (" << label << ") obj barycenter: " << t.barycenter()
           << " eta,phi (baricenter): " << t.barycenter().eta() << ", " << t.barycenter().phi()
           << " eta,phi (eigen): " << t.eigenvectors(0).eta() << ", " << t.eigenvectors(0).phi()
           << " pt(eigen): " << std::sqrt(t.eigenvectors(0).Unit().perp2()) * t.raw_energy() << " seedID: " << t.seedID()
           << " seedIndex: " << t.seedIndex() << " size: " << t.vertices().size() << " average usage: "
           << (std::accumulate(std::begin(t.vertex_multiplicity()), std::end(t.vertex_multiplicity()), 0.) /
               (float)t.vertex_multiplicity().size())
           << " raw_energy: " << t.raw_energy() << " regressed energy: " << t.regressed_energy()
           << " probs(ga/e/mu/np/cp/nh/am/unk): ";
       for (auto const &p : t.id_probabilities()) {
         log << std::fixed << p << " ";
       }
       log << " sigmas: ";
       for (auto const &s : t.sigmas()) {
         log << s << " ";
       }
       log << "\n";
     }
   });
 }

◆ produce()

void TracksterLinksProducer::produce	(	edm::Event &	evt,
		const edm::EventSetup &	es
	)

override

Definition at line 183 of file TracksterLinksProducer.cc.

References ticl::assignPCAtoTracksters(), clusters_token_, clustersTime_token_, edm::Event::get(), edm::Event::getByToken(), hgcal::RecHitTools::getPositionLayer(), mps_fire::i, inferenceAlgo_, input, dqmiolumiharvest::j, hgcal::RecHitTools::lastLayerEE(), hltEgammaHGCALIDVarsL1Seeded_cfi::layerClusters, linkingAlgo_, eostools::move(), original_masks_tokens_, edm::Event::put(), regressionAndPid_, rhtools_, tracksters_tokens_, and z.

                                                                            {
   linkingAlgo_->setEvent(evt, es);
 
   auto resultTracksters = std::make_unique<std::vector<Trackster>>();
 
   auto linkedResultTracksters = std::make_unique<std::vector<std::vector<unsigned int>>>();
 
   const auto &layerClusters = evt.get(clusters_token_);
   const auto &layerClustersTimes = evt.get(clustersTime_token_);
 
   // loop over the original_masks_tokens_ and get the original masks collections and multiply them
   // to get the global mask
   std::vector<float> original_global_mask(layerClusters.size(), 1.f);
   for (unsigned int i = 0; i < original_masks_tokens_.size(); ++i) {
     const auto &tmp_mask = evt.get(original_masks_tokens_[i]);
     for (unsigned int j = 0; j < tmp_mask.size(); ++j) {
       original_global_mask[j] *= tmp_mask[j];
     }
   }
 
   auto resultMask = std::make_unique<std::vector<float>>(original_global_mask);
 
   std::vector<edm::Handle<std::vector<Trackster>>> tracksters_h(tracksters_tokens_.size());
   MultiVectorManager<Trackster> trackstersManager;
   for (unsigned int i = 0; i < tracksters_tokens_.size(); ++i) {
     evt.getByToken(tracksters_tokens_[i], tracksters_h[i]);
     //Fill MultiVectorManager
     trackstersManager.addVector(*tracksters_h[i]);
   }
 
   // Linking
   const typename TracksterLinkingAlgoBase::Inputs input(evt, es, layerClusters, layerClustersTimes, trackstersManager);
   auto linkedTracksterIdToInputTracksterId = std::make_unique<std::vector<std::vector<unsigned int>>>();
 
   // LinkTracksters will produce a vector of vector of indices of tracksters that:
   // 1) are linked together if more than one
   // 2) are isolated if only one
   // Result tracksters contains the final version of the trackster collection
   // linkedTrackstersToInputTrackstersMap contains the mapping between the linked tracksters and the input tracksters
   linkingAlgo_->linkTracksters(input, *resultTracksters, *linkedResultTracksters, *linkedTracksterIdToInputTracksterId);
 
   // Now we need to remove the tracksters that are not linked
   // We need to emplace_back in the resultTracksters only the tracksters that are linked
 
   for (auto const &resultTrackster : *resultTracksters) {
     for (auto const &clusterIndex : resultTrackster.vertices()) {
       (*resultMask)[clusterIndex] = 0.f;
     }
   }
 
   if (regressionAndPid_) {
     // Run inference algorithm
     inferenceAlgo_->inputData(layerClusters, *resultTracksters);
     inferenceAlgo_->runInference(
         *resultTracksters);  //option to use "Linking" instead of "CLU3D"/"energyAndPid" instead of "PID"
   }
 
   assignPCAtoTracksters(*resultTracksters,
                         layerClusters,
                         layerClustersTimes,
                         rhtools_.getPositionLayer(rhtools_.lastLayerEE()).z(),
                         rhtools_,
                         true);
 
   evt.put(std::move(linkedResultTracksters));
   evt.put(std::move(resultMask));
   evt.put(std::move(resultTracksters));
   evt.put(std::move(linkedTracksterIdToInputTracksterId), "linkedTracksterIdToInputTracksterId");
 }

Member Data Documentation

◆ algoType_

std::string TracksterLinksProducer::algoType_

private

Definition at line 68 of file TracksterLinksProducer.cc.

Referenced by beginRun(), and TracksterLinksProducer().

◆ bfield_token_

const edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> TracksterLinksProducer::bfield_token_

private

Definition at line 83 of file TracksterLinksProducer.cc.

Referenced by beginRun().

◆ clusters_token_

const edm::EDGetTokenT<std::vector<reco::CaloCluster> > TracksterLinksProducer::clusters_token_

private

Definition at line 71 of file TracksterLinksProducer.cc.

Referenced by produce().

◆ clustersTime_token_

const edm::EDGetTokenT<edm::ValueMap<std::pair<float, float> > > TracksterLinksProducer::clustersTime_token_

private

Definition at line 72 of file TracksterLinksProducer.cc.

Referenced by produce().

◆ detector_

const std::string TracksterLinksProducer::detector_

private

Definition at line 80 of file TracksterLinksProducer.cc.

Referenced by TracksterLinksProducer().

◆ geometry_token_

const edm::ESGetToken<CaloGeometry, CaloGeometryRecord> TracksterLinksProducer::geometry_token_

private

Definition at line 79 of file TracksterLinksProducer.cc.

Referenced by beginRun().

◆ hdc_token_

edm::ESGetToken<HGCalDDDConstants, IdealGeometryRecord> TracksterLinksProducer::hdc_token_

private

Definition at line 87 of file TracksterLinksProducer.cc.

Referenced by beginRun(), and TracksterLinksProducer().

◆ hgcons_

const HGCalDDDConstants* TracksterLinksProducer::hgcons_

private

Definition at line 85 of file TracksterLinksProducer.cc.

Referenced by beginRun().

◆ inferenceAlgo_

std::unique_ptr<TracksterInferenceAlgoBase> TracksterLinksProducer::inferenceAlgo_

private

Definition at line 75 of file TracksterLinksProducer.cc.

Referenced by produce(), and TracksterLinksProducer().

◆ linkingAlgo_

std::unique_ptr<TracksterLinkingAlgoBase> TracksterLinksProducer::linkingAlgo_

private

Definition at line 67 of file TracksterLinksProducer.cc.

Referenced by beginRun(), produce(), and TracksterLinksProducer().

◆ original_masks_tokens_

std::vector<edm::EDGetTokenT<std::vector<float> > > TracksterLinksProducer::original_masks_tokens_

private

Definition at line 77 of file TracksterLinksProducer.cc.

Referenced by produce(), and TracksterLinksProducer().

◆ propagator_token_

const edm::ESGetToken<Propagator, TrackingComponentsRecord> TracksterLinksProducer::propagator_token_

private

Definition at line 84 of file TracksterLinksProducer.cc.

Referenced by beginRun().

◆ propName_

const std::string TracksterLinksProducer::propName_

private

Definition at line 81 of file TracksterLinksProducer.cc.

◆ regressionAndPid_

const bool TracksterLinksProducer::regressionAndPid_

private

Definition at line 74 of file TracksterLinksProducer.cc.

Referenced by produce().

◆ rhtools_

hgcal::RecHitTools TracksterLinksProducer::rhtools_

private

Definition at line 86 of file TracksterLinksProducer.cc.

Referenced by beginRun(), and produce().

◆ tracksters_tokens_

std::vector<edm::EDGetTokenT<std::vector<Trackster> > > TracksterLinksProducer::tracksters_tokens_

private

Definition at line 70 of file TracksterLinksProducer.cc.

Referenced by produce(), and TracksterLinksProducer().

Public Member Functions

Static Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

◆ TracksterLinksProducer()

◆ ~TracksterLinksProducer()

Member Function Documentation

◆ beginRun()

◆ dumpTrackster()

◆ fillDescriptions()

◆ globalEndJob()

◆ initializeGlobalCache()

◆ printTrackstersDebug()

◆ produce()

Member Data Documentation

◆ algoType_

◆ bfield_token_

◆ clusters_token_

◆ clustersTime_token_

◆ detector_

◆ geometry_token_

◆ hdc_token_

◆ hgcons_

◆ inferenceAlgo_

◆ linkingAlgo_

◆ original_masks_tokens_

◆ propagator_token_

◆ propName_

◆ regressionAndPid_

◆ rhtools_

◆ tracksters_tokens_