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	energyRegressionAndID (const std::vector< reco::CaloCluster > &layerClusters, const tensorflow::Session *, std::vector< Trackster > &result) 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 std::string	eidInputName_

const float	eidMinClusterEnergy_

const int	eidNClusters_

const int	eidNLayers_

const std::string	eidOutputNameEnergy_

const std::string	eidOutputNameId_

tensorflow::Session *	eidSession_

const edm::ESGetToken< CaloGeometry, CaloGeometryRecord >	geometry_token_

edm::ESGetToken< HGCalDDDConstants, IdealGeometryRecord >	hdc_token_

const HGCalDDDConstants *	hgcons_

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_

const std::string	tfDnnLabel_

const edm::ESGetToken< TfGraphDefWrapper, TfGraphRecord >	tfDnnToken_

const tensorflow::Session *	tfSession_

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

Static Private Attributes
static constexpr int	eidNFeatures_ = 3

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 51 of file TracksterLinksProducer.cc.

Constructor & Destructor Documentation

◆ TracksterLinksProducer()

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

explicit

Definition at line 102 of file TracksterLinksProducer.cc.

References algoType_, detector_, get, edm::ParameterSet::getParameter(), hdc_token_, 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")),
       tfDnnLabel_(ps.getParameter<std::string>("tfDnnLabel")),
       tfDnnToken_(esConsumes(edm::ESInputTag("", tfDnnLabel_))),
       tfSession_(nullptr),
       eidInputName_(ps.getParameter<std::string>("eid_input_name")),
       eidOutputNameEnergy_(ps.getParameter<std::string>("eid_output_name_energy")),
       eidOutputNameId_(ps.getParameter<std::string>("eid_output_name_id")),
       eidMinClusterEnergy_(ps.getParameter<double>("eid_min_cluster_energy")),
       eidNLayers_(ps.getParameter<int>("eid_n_layers")),
       eidNClusters_(ps.getParameter<int>("eid_n_clusters")),
       eidSession_(nullptr),
       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));
   }
 
   // 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 54 of file TracksterLinksProducer.cc.

54 {};

Member Function Documentation

◆ beginRun()

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

override

Definition at line 163 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 178 of file TracksterLinksProducer.cc.

References alignBH_cfg::fixed, nano_mu_digi_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;
 }

◆ energyRegressionAndID()

void TracksterLinksProducer::energyRegressionAndID	(	const std::vector< reco::CaloCluster > &	layerClusters,
		const tensorflow::Session *	eidSession,
		std::vector< Trackster > &	result
	)		const

private

Definition at line 200 of file TracksterLinksProducer.cc.

References a, funct::abs(), b, HLT_FULL_cff::batchSize, ALPAKA_ACCELERATOR_NAMESPACE::brokenline::constexpr(), data, eidInputName_, eidMinClusterEnergy_, eidNClusters_, eidNFeatures_, eidNLayers_, eidOutputNameEnergy_, eidOutputNameId_, hcalRecHitTable_cff::energy, reco::CaloCluster::energy(), reco::CaloCluster::eta(), f, lowptgsfeleseed::features(), nano_mu_digi_cff::float, hgcal::RecHitTools::getLayerWithOffset(), reco::CaloCluster::hitsAndFractions(), mps_fire::i, input, crabTemplate::inputList, createfilelist::int, dqmiolumiharvest::j, dqmdumpme::k, MainPageGenerator::l, hltEgammaHGCALIDVarsL1Seeded_cfi::layerClusters, jetsAK4_CHS_cff::outputNames, PatBasicFWLiteJetAnalyzer_Selector_cfg::outputs, reco::CaloCluster::phi(), rhtools_, tensorflow::run(), l1trig_cff::shape, jetUpdater_cfi::sort, submitPVValidationJobs::t, ticl::Trackster::vertex_multiplicity(), and ticl::Trackster::vertices().

Referenced by produce().

                                                                                            {
   // Energy regression and particle identification strategy:
   //
   // 1. Set default values for regressed energy and particle id for each trackster.
   // 2. Store indices of tracksters whose total sum of cluster energies is above the
   //    eidMinClusterEnergy_ (GeV) threshold. Inference is not applied for soft tracksters.
   // 3. When no trackster passes the selection, return.
   // 4. Create input and output tensors. The batch dimension is determined by the number of
   //    selected tracksters.
   // 5. Fill input tensors with layer cluster features. Per layer, clusters are ordered descending
   //    by energy. Given that tensor data is contiguous in memory, we can use pointer arithmetic to
   //    fill values, even with batching.
   // 6. Zero-fill features for empty clusters in each layer.
   // 7. Batched inference.
   // 8. Assign the regressed energy and id probabilities to each trackster.
   //
   // Indices used throughout this method:
   // i -> batch element / trackster
   // j -> layer
   // k -> cluster
   // l -> feature
 
   // do nothing when no trackster passes the selection (3)
   int batchSize = (int)tracksters.size();
   if (batchSize == 0) {
     return;
   }
 
   for (auto &t : tracksters) {
     t.setRegressedEnergy(0.f);
     t.zeroProbabilities();
   }
 
   // create input and output tensors (4)
   tensorflow::TensorShape shape({batchSize, eidNLayers_, eidNClusters_, eidNFeatures_});
   tensorflow::Tensor input(tensorflow::DT_FLOAT, shape);
   tensorflow::NamedTensorList inputList = {{eidInputName_, input}};
   static constexpr int inputDimension = 4;
 
   std::vector<tensorflow::Tensor> outputs;
   std::vector<std::string> outputNames;
   if (!eidOutputNameEnergy_.empty()) {
     outputNames.push_back(eidOutputNameEnergy_);
   }
   if (!eidOutputNameId_.empty()) {
     outputNames.push_back(eidOutputNameId_);
   }
 
   // fill input tensor (5)
   for (int i = 0; i < batchSize; i++) {
     const Trackster &trackster = tracksters[i];
 
     // per layer, we only consider the first eidNClusters_ clusters in terms of
     // energy, so in order to avoid creating large / nested structures to do
     // the sorting for an unknown number of total clusters, create a sorted
     // list of layer cluster indices to keep track of the filled clusters
     std::vector<int> clusterIndices(trackster.vertices().size());
     for (int k = 0; k < (int)trackster.vertices().size(); k++) {
       clusterIndices[k] = k;
     }
     sort(clusterIndices.begin(), clusterIndices.end(), [&layerClusters, &trackster](const int &a, const int &b) {
       return layerClusters[trackster.vertices(a)].energy() > layerClusters[trackster.vertices(b)].energy();
     });
 
     // keep track of the number of seen clusters per layer
     std::vector<int> seenClusters(eidNLayers_);
 
     // loop through clusters by descending energy
     for (const int &k : clusterIndices) {
       // get features per layer and cluster and store the values directly in the input tensor
       const reco::CaloCluster &cluster = layerClusters[trackster.vertices(k)];
       int j = rhtools_.getLayerWithOffset(cluster.hitsAndFractions()[0].first) - 1;
       if (j < eidNLayers_ && seenClusters[j] < eidNClusters_) {
         // get the pointer to the first feature value for the current batch, layer and cluster
         float *features = &input.tensor<float, inputDimension>()(i, j, seenClusters[j], 0);
 
         // fill features
         *(features++) = float(cluster.energy() / float(trackster.vertex_multiplicity(k)));
         *(features++) = float(std::abs(cluster.eta()));
         *(features) = float(cluster.phi());
 
         // increment seen clusters
         seenClusters[j]++;
       }
     }
 
     // zero-fill features of empty clusters in each layer (6)
     for (int j = 0; j < eidNLayers_; j++) {
       for (int k = seenClusters[j]; k < eidNClusters_; k++) {
         float *features = &input.tensor<float, inputDimension>()(i, j, k, 0);
         for (int l = 0; l < eidNFeatures_; l++) {
           *(features++) = 0.f;
         }
       }
     }
   }
 
   // run the inference (7)
   tensorflow::run(eidSession, inputList, outputNames, &outputs);
 
   // store regressed energy per trackster (8)
   if (!eidOutputNameEnergy_.empty()) {
     // get the pointer to the energy tensor, dimension is batch x 1
     float *energy = outputs[0].flat<float>().data();
 
     for (int i = 0; i < batchSize; ++i) {
       float regressedEnergy =
           tracksters[i].raw_energy() > eidMinClusterEnergy_ ? energy[i] : tracksters[i].raw_energy();
       tracksters[i].setRegressedEnergy(regressedEnergy);
     }
   }
 
   // store id probabilities per trackster (8)
   if (!eidOutputNameId_.empty()) {
     // get the pointer to the id probability tensor, dimension is batch x id_probabilities.size()
     int probsIdx = !eidOutputNameEnergy_.empty();
     float *probs = outputs[probsIdx].flat<float>().data();
     int probsNumber = tracksters[0].id_probabilities().size();
     for (int i = 0; i < batchSize; ++i) {
       tracksters[i].setProbabilities(&probs[i * probsNumber]);
     }
   }
 }

◆ fillDescriptions()

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

static

Definition at line 418 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));
 
   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>("tfDnnLabel", "tracksterSelectionTf");
   desc.add<std::string>("eid_input_name", "input");
   desc.add<std::string>("eid_output_name_energy", "output/regressed_energy");
   desc.add<std::string>("eid_output_name_id", "output/id_probabilities");
   desc.add<double>("eid_min_cluster_energy", 2.5);
   desc.add<int>("eid_n_layers", 50);
   desc.add<int>("eid_n_clusters", 10);
   desc.add<std::string>("detector", "HGCAL");
   desc.add<std::string>("propagator", "PropagatorWithMaterial");
   descriptions.add("tracksterLinksProducer", desc);
 }

◆ globalEndJob()

void TracksterLinksProducer::globalEndJob ( const ONNXRuntime * )

static

Definition at line 161 of file TracksterLinksProducer.cc.

161 {}

◆ initializeGlobalCache()

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

static

Definition at line 153 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 393 of file TracksterLinksProducer.cc.

References alignBH_cfg::fixed, nano_mu_digi_cff::float, label, dqm-mbProfile::log, LogDebug, AlCaHLTBitMon_ParallelJobs::p, alignCSCRings::s, mathSSE::sqrt(), and submitPVValidationJobs::t.

                                                                                                                  {
   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 326 of file TracksterLinksProducer.cc.

References ticl::assignPCAtoTracksters(), clusters_token_, clustersTime_token_, energyRegressionAndID(), edm::Event::get(), edm::Event::getByToken(), edm::EventSetup::getData(), hgcal::RecHitTools::getPositionLayer(), mps_fire::i, input, dqmiolumiharvest::j, hgcal::RecHitTools::lastLayerEE(), hltEgammaHGCALIDVarsL1Seeded_cfi::layerClusters, linkingAlgo_, eostools::move(), original_masks_tokens_, edm::Event::put(), regressionAndPid_, rhtools_, tfDnnToken_, tfSession_, 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_);
 
   tfSession_ = es.getData(tfDnnToken_).getSession();
   // 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_)
     energyRegressionAndID(layerClusters, tfSession_, *resultTracksters);
 
   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 70 of file TracksterLinksProducer.cc.

Referenced by beginRun(), and TracksterLinksProducer().

◆ bfield_token_

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

private

Definition at line 95 of file TracksterLinksProducer.cc.

Referenced by beginRun().

◆ clusters_token_

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

private

Definition at line 73 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 74 of file TracksterLinksProducer.cc.

Referenced by produce().

◆ detector_

const std::string TracksterLinksProducer::detector_

private

Definition at line 92 of file TracksterLinksProducer.cc.

Referenced by TracksterLinksProducer().

◆ eidInputName_

const std::string TracksterLinksProducer::eidInputName_

private

Definition at line 80 of file TracksterLinksProducer.cc.

Referenced by energyRegressionAndID().

◆ eidMinClusterEnergy_

const float TracksterLinksProducer::eidMinClusterEnergy_

private

Definition at line 83 of file TracksterLinksProducer.cc.

Referenced by energyRegressionAndID().

◆ eidNClusters_

const int TracksterLinksProducer::eidNClusters_

private

Definition at line 85 of file TracksterLinksProducer.cc.

Referenced by energyRegressionAndID().

◆ eidNFeatures_

constexpr int TracksterLinksProducer::eidNFeatures_ = 3

staticprivate

Definition at line 86 of file TracksterLinksProducer.cc.

Referenced by energyRegressionAndID().

◆ eidNLayers_

const int TracksterLinksProducer::eidNLayers_

private

Definition at line 84 of file TracksterLinksProducer.cc.

Referenced by energyRegressionAndID().

◆ eidOutputNameEnergy_

const std::string TracksterLinksProducer::eidOutputNameEnergy_

private

Definition at line 81 of file TracksterLinksProducer.cc.

Referenced by energyRegressionAndID().

◆ eidOutputNameId_

const std::string TracksterLinksProducer::eidOutputNameId_

private

Definition at line 82 of file TracksterLinksProducer.cc.

Referenced by energyRegressionAndID().

◆ eidSession_

tensorflow::Session* TracksterLinksProducer::eidSession_

private

Definition at line 87 of file TracksterLinksProducer.cc.

◆ geometry_token_

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

private

Definition at line 91 of file TracksterLinksProducer.cc.

Referenced by beginRun().

◆ hdc_token_

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

private

Definition at line 99 of file TracksterLinksProducer.cc.

Referenced by beginRun(), and TracksterLinksProducer().

◆ hgcons_

const HGCalDDDConstants* TracksterLinksProducer::hgcons_

private

Definition at line 97 of file TracksterLinksProducer.cc.

Referenced by beginRun().

◆ linkingAlgo_

std::unique_ptr<TracksterLinkingAlgoBase> TracksterLinksProducer::linkingAlgo_

private

Definition at line 69 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 89 of file TracksterLinksProducer.cc.

Referenced by produce(), and TracksterLinksProducer().

◆ propagator_token_

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

private

Definition at line 96 of file TracksterLinksProducer.cc.

Referenced by beginRun().

◆ propName_

const std::string TracksterLinksProducer::propName_

private

Definition at line 93 of file TracksterLinksProducer.cc.

◆ regressionAndPid_

const bool TracksterLinksProducer::regressionAndPid_

private

Definition at line 76 of file TracksterLinksProducer.cc.

Referenced by produce().

◆ rhtools_

hgcal::RecHitTools TracksterLinksProducer::rhtools_

private

Definition at line 98 of file TracksterLinksProducer.cc.

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

◆ tfDnnLabel_

const std::string TracksterLinksProducer::tfDnnLabel_

private

Definition at line 77 of file TracksterLinksProducer.cc.

◆ tfDnnToken_

const edm::ESGetToken<TfGraphDefWrapper, TfGraphRecord> TracksterLinksProducer::tfDnnToken_

private

Definition at line 78 of file TracksterLinksProducer.cc.

Referenced by produce().

◆ tfSession_

const tensorflow::Session* TracksterLinksProducer::tfSession_

private

Definition at line 79 of file TracksterLinksProducer.cc.

Referenced by produce().

◆ tracksters_tokens_

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

private

Definition at line 72 of file TracksterLinksProducer.cc.

Referenced by produce(), and TracksterLinksProducer().

Public Member Functions

Static Public Member Functions

Private Member Functions

Private Attributes

Static Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

◆ TracksterLinksProducer()

◆ ~TracksterLinksProducer()

Member Function Documentation

◆ beginRun()

◆ dumpTrackster()

◆ energyRegressionAndID()

◆ fillDescriptions()

◆ globalEndJob()

◆ initializeGlobalCache()

◆ printTrackstersDebug()

◆ produce()

Member Data Documentation

◆ algoType_

◆ bfield_token_

◆ clusters_token_

◆ clustersTime_token_

◆ detector_

◆ eidInputName_

◆ eidMinClusterEnergy_

◆ eidNClusters_

◆ eidNFeatures_

◆ eidNLayers_

◆ eidOutputNameEnergy_

◆ eidOutputNameId_

◆ eidSession_

◆ geometry_token_

◆ hdc_token_

◆ hgcons_

◆ linkingAlgo_

◆ original_masks_tokens_

◆ propagator_token_

◆ propName_

◆ regressionAndPid_

◆ rhtools_

◆ tfDnnLabel_

◆ tfDnnToken_

◆ tfSession_

◆ tracksters_tokens_