#include <PatternRecognitionbyCA.h>

Inheritance diagram for ticl::PatternRecognitionbyCA< TILES >:

Public Member Functions
void	emptyTrackstersFromSeedsTRK (std::vector< Trackster > &tracksters, std::unordered_map< int, std::vector< int >> &seedToTracksterAssociation, const edm::ProductID &collectionID) const

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

void	makeTracksters (const typename PatternRecognitionAlgoBaseT< TILES >::Inputs &input, std::vector< Trackster > &result, std::unordered_map< int, std::vector< int >> &seedToTracksterAssociation) override

	PatternRecognitionbyCA (const edm::ParameterSet &conf, edm::ConsumesCollector iC)

	~PatternRecognitionbyCA () override

Public Member Functions inherited from ticl::PatternRecognitionAlgoBaseT< TILES >
virtual void	makeTracksters (const Inputs &input, std::vector< Trackster > &result, std::unordered_map< int, std::vector< int >> &seedToTracksterAssociation)=0

	PatternRecognitionAlgoBaseT (const edm::ParameterSet &conf, edm::ConsumesCollector)

virtual	~PatternRecognitionAlgoBaseT ()

Static Public Member Functions
static void	fillPSetDescription (edm::ParameterSetDescription &iDesc)

Private Member Functions
void	mergeTrackstersTRK (const std::vector< Trackster > &, const std::vector< reco::CaloCluster > &, std::vector< Trackster > &, std::unordered_map< int, std::vector< int >> &seedToTracksterAssociation) const

Private Attributes
edm::ESGetToken< CaloGeometry, CaloGeometryRecord >	caloGeomToken_

bool	check_missing_layers_ = false

const bool	computeLocalTime_

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 double	energy_em_over_total_threshold_

const float	etaLimitIncreaseWindow_

const std::vector< int >	filter_on_categories_

const float	max_delta_time_

const double	max_longitudinal_sigmaPCA_

const int	max_missing_layers_in_trackster_

const unsigned int	max_out_in_hops_

const int	min_clusters_per_ntuplet_

const float	min_cos_pointing_

const float	min_cos_theta_

const unsigned int	min_layers_per_trackster_

const bool	oneTracksterPerTrackSeed_

const bool	out_in_dfs_

const double	pid_threshold_

const bool	promoteEmptyRegionToTrackster_

hgcal::RecHitTools	rhtools_

const float	root_doublet_max_distance_from_seed_squared_

const unsigned int	shower_start_max_layer_

const std::vector< double >	siblings_maxRSquared_

const int	skip_layers_

const std::unique_ptr< HGCGraphT< TILES > >	theGraph_

Static Private Attributes
static const int	eidNFeatures_ = 3

Additional Inherited Members
Protected Attributes inherited from ticl::PatternRecognitionAlgoBaseT< TILES >
int	algo_verbosity_

Detailed Description

template<typename TILES>
class ticl::PatternRecognitionbyCA< TILES >

Definition at line 14 of file PatternRecognitionbyCA.h.

Constructor & Destructor Documentation

◆ PatternRecognitionbyCA()

template<typename TILES >

PatternRecognitionbyCA::PatternRecognitionbyCA	(	const edm::ParameterSet &	conf,
		edm::ConsumesCollector	iC
	)

Definition at line 19 of file PatternRecognitionbyCA.cc.

     : PatternRecognitionAlgoBaseT<TILES>(conf, iC),
       caloGeomToken_(iC.esConsumes<CaloGeometry, CaloGeometryRecord>()),
       theGraph_(std::make_unique<HGCGraphT<TILES>>()),
       oneTracksterPerTrackSeed_(conf.getParameter<bool>("oneTracksterPerTrackSeed")),
       promoteEmptyRegionToTrackster_(conf.getParameter<bool>("promoteEmptyRegionToTrackster")),
       out_in_dfs_(conf.getParameter<bool>("out_in_dfs")),
       max_out_in_hops_(conf.getParameter<int>("max_out_in_hops")),
       min_cos_theta_(conf.getParameter<double>("min_cos_theta")),
       min_cos_pointing_(conf.getParameter<double>("min_cos_pointing")),
       root_doublet_max_distance_from_seed_squared_(
           conf.getParameter<double>("root_doublet_max_distance_from_seed_squared")),
       etaLimitIncreaseWindow_(conf.getParameter<double>("etaLimitIncreaseWindow")),
       skip_layers_(conf.getParameter<int>("skip_layers")),
       max_missing_layers_in_trackster_(conf.getParameter<int>("max_missing_layers_in_trackster")),
       check_missing_layers_(max_missing_layers_in_trackster_ < 100),
       shower_start_max_layer_(conf.getParameter<int>("shower_start_max_layer")),
       min_layers_per_trackster_(conf.getParameter<int>("min_layers_per_trackster")),
       filter_on_categories_(conf.getParameter<std::vector<int>>("filter_on_categories")),
       pid_threshold_(conf.getParameter<double>("pid_threshold")),
       energy_em_over_total_threshold_(conf.getParameter<double>("energy_em_over_total_threshold")),
       max_longitudinal_sigmaPCA_(conf.getParameter<double>("max_longitudinal_sigmaPCA")),
       min_clusters_per_ntuplet_(min_layers_per_trackster_),
       max_delta_time_(conf.getParameter<double>("max_delta_time")),
       eidInputName_(conf.getParameter<std::string>("eid_input_name")),
       eidOutputNameEnergy_(conf.getParameter<std::string>("eid_output_name_energy")),
       eidOutputNameId_(conf.getParameter<std::string>("eid_output_name_id")),
       eidMinClusterEnergy_(conf.getParameter<double>("eid_min_cluster_energy")),
       eidNLayers_(conf.getParameter<int>("eid_n_layers")),
       eidNClusters_(conf.getParameter<int>("eid_n_clusters")),
       computeLocalTime_(conf.getParameter<bool>("computeLocalTime")),
       siblings_maxRSquared_(conf.getParameter<std::vector<double>>("siblings_maxRSquared")){};

◆ ~PatternRecognitionbyCA()

template<typename TILES >

PatternRecognitionbyCA::~PatternRecognitionbyCA ( )

override

Definition at line 53 of file PatternRecognitionbyCA.cc.

53 {};

Member Function Documentation

◆ emptyTrackstersFromSeedsTRK()

template<typename TILES >

void PatternRecognitionbyCA::emptyTrackstersFromSeedsTRK	(	std::vector< Trackster > &	tracksters,
		std::unordered_map< int, std::vector< int >> &	seedToTracksterAssociation,
		const edm::ProductID &	collectionID
	)		const

Definition at line 333 of file PatternRecognitionbyCA.cc.

References ticl::Trackster::charged_hadron, f, and submitPVValidationJobs::t.

                                             {
   for (auto &thisSeed : seedToTracksterAssociation) {
     if (thisSeed.second.empty()) {
       Trackster t;
       t.setRegressedEnergy(0.f);
       t.zeroProbabilities();
       t.setIdProbability(ticl::Trackster::ParticleType::charged_hadron, 1.f);
       t.setSeed(collectionID, thisSeed.first);
       tracksters.emplace_back(t);
       thisSeed.second.emplace_back(tracksters.size() - 1);
     }
   }
 }

◆ energyRegressionAndID()

template<typename TILES >

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

Definition at line 351 of file PatternRecognitionbyCA.cc.

References a, funct::abs(), b, HLT_FULL_cff::batchSize, data, hcalRecHitTable_cff::energy, reco::CaloCluster::energy(), reco::CaloCluster::eta(), f, lowptgsfeleseed::features(), nano_mu_digi_cff::float, 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(), tensorflow::run(), l1trig_cff::shape, jetUpdater_cfi::sort, bphysicsOniaDQM_cfi::vertex, ticl::Trackster::vertex_multiplicity(), AlignmentTracksFromVertexSelector_cfi::vertices, and ticl::Trackster::vertices().

                                                                                             {
   // 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
 
   // set default values per trackster, determine if the cluster energy threshold is passed,
   // and store indices of hard tracksters
   std::vector<int> tracksterIndices;
   for (int i = 0; i < (int)tracksters.size(); i++) {
     // calculate the cluster energy sum (2)
     // note: after the loop, sumClusterEnergy might be just above the threshold which is enough to
     // decide whether to run inference for the trackster or not
     float sumClusterEnergy = 0.;
     for (const unsigned int &vertex : tracksters[i].vertices()) {
       sumClusterEnergy += (float)layerClusters[vertex].energy();
       // there might be many clusters, so try to stop early
       if (sumClusterEnergy >= eidMinClusterEnergy_) {
         // set default values (1)
         tracksters[i].setRegressedEnergy(0.f);
         tracksters[i].zeroProbabilities();
         tracksterIndices.push_back(i);
         break;
       }
     }
   }
 
   // do nothing when no trackster passes the selection (3)
   int batchSize = (int)tracksterIndices.size();
   if (batchSize == 0) {
     return;
   }
 
   // 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}};
 
   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[tracksterIndices[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, 4>()(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, 4>()(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 (const int &i : tracksterIndices) {
       tracksters[i].setRegressedEnergy(*(energy++));
     }
   }
 
   // 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() ? 0 : 1;
     float *probs = outputs[probsIdx].flat<float>().data();
 
     for (const int &i : tracksterIndices) {
       tracksters[i].setProbabilities(probs);
       probs += tracksters[i].id_probabilities().size();
     }
   }
 }

◆ fillPSetDescription()

template<typename TILES >

void PatternRecognitionbyCA::fillPSetDescription ( edm::ParameterSetDescription & iDesc )

static

Definition at line 491 of file PatternRecognitionbyCA.cc.

References edm::ParameterSetDescription::add(), MillePedeFileConverter_cfg::e, edm::ParameterDescriptionNode::setComment(), and AlCaHLTBitMon_QueryRunRegistry::string.

                                                                                          {
   iDesc.add<int>("algo_verbosity", 0);
   iDesc.add<bool>("oneTracksterPerTrackSeed", false);
   iDesc.add<bool>("promoteEmptyRegionToTrackster", false);
   iDesc.add<bool>("out_in_dfs", true);
   iDesc.add<int>("max_out_in_hops", 10);
   iDesc.add<double>("min_cos_theta", 0.915);
   iDesc.add<double>("min_cos_pointing", -1.);
   iDesc.add<double>("root_doublet_max_distance_from_seed_squared", 9999);
   iDesc.add<double>("etaLimitIncreaseWindow", 2.1);
   iDesc.add<int>("skip_layers", 0);
   iDesc.add<int>("max_missing_layers_in_trackster", 9999);
   iDesc.add<int>("shower_start_max_layer", 9999)->setComment("make default such that no filtering is applied");
   iDesc.add<int>("min_layers_per_trackster", 10);
   iDesc.add<std::vector<int>>("filter_on_categories", {0});
   iDesc.add<double>("pid_threshold", 0.)->setComment("make default such that no filtering is applied");
   iDesc.add<double>("energy_em_over_total_threshold", -1.)
       ->setComment("make default such that no filtering is applied");
   iDesc.add<double>("max_longitudinal_sigmaPCA", 9999);
   iDesc.add<double>("max_delta_time", 3.)->setComment("nsigma");
   iDesc.add<std::string>("eid_input_name", "input");
   iDesc.add<std::string>("eid_output_name_energy", "output/regressed_energy");
   iDesc.add<std::string>("eid_output_name_id", "output/id_probabilities");
   iDesc.add<double>("eid_min_cluster_energy", 1.);
   iDesc.add<int>("eid_n_layers", 50);
   iDesc.add<int>("eid_n_clusters", 10);
   iDesc.add<bool>("computeLocalTime", false);
   iDesc.add<std::vector<double>>("siblings_maxRSquared", {6e-4, 6e-4, 6e-4});
 }

◆ makeTracksters()

template<typename TILES >

void PatternRecognitionbyCA::makeTracksters	(	const typename PatternRecognitionAlgoBaseT< TILES >::Inputs &	input,
		std::vector< Trackster > &	result,
		std::unordered_map< int, std::vector< int >> &	seedToTracksterAssociation
	)

override

Definition at line 56 of file PatternRecognitionbyCA.cc.

References ticl::Advanced, cms::cuda::assert(), ticl::assignPCAtoTracksters(), ticl::Basic, ALPAKA_ACCELERATOR_NAMESPACE::brokenline::constexpr(), filterCSVwithJSON::copy, HLT_2024v14_cff::doublets, dqmdumpme::first, caHitNtupletGeneratorKernels::foundNtuplets, relativeConstraints::geom, edm::EventSetup::getData(), mps_fire::i, input, createfilelist::int, dqmiolumiharvest::j, LogDebug, SiStripPI::min, L1TMuonDQMOffline_cfi::nEtaBins, None, ecaldqm::binning::nPhiBins, pixelTracksMonitoring_cff::ntuplet, AlCaHLTBitMon_ParallelJobs::p, mps_fire::result, jetUpdater_cfi::sort, submitPVValidationJobs::t, createJobs::tmp, tier0::unique(), and relativeConstraints::value.

                                                                        {
   // Protect from events with no seeding regions
   if (input.regions.empty())
     return;
 
   edm::EventSetup const &es = input.es;
   const CaloGeometry &geom = es.getData(caloGeomToken_);
   rhtools_.setGeometry(geom);
 
   theGraph_->setVerbosity(PatternRecognitionAlgoBaseT<TILES>::algo_verbosity_);
   theGraph_->clear();
   if (PatternRecognitionAlgoBaseT<TILES>::algo_verbosity_ > VerbosityLevel::None) {
     LogDebug("HGCPatternRecoByCA") << "Making Tracksters with CA" << std::endl;
   }
 
   constexpr auto isHFnose = std::is_same<TILES, TICLLayerTilesHFNose>::value;
   constexpr int nEtaBin = TILES::constants_type_t::nEtaBins;
   constexpr int nPhiBin = TILES::constants_type_t::nPhiBins;
 
   bool isRegionalIter = (input.regions[0].index != -1);
   std::vector<HGCDoublet::HGCntuplet> foundNtuplets;
   std::vector<int> seedIndices;
   std::vector<uint8_t> layer_cluster_usage(input.layerClusters.size(), 0);
   theGraph_->makeAndConnectDoublets(input.tiles,
                                     input.regions,
                                     nEtaBin,
                                     nPhiBin,
                                     input.layerClusters,
                                     input.mask,
                                     input.layerClustersTime,
                                     1,
                                     1,
                                     min_cos_theta_,
                                     min_cos_pointing_,
                                     root_doublet_max_distance_from_seed_squared_,
                                     etaLimitIncreaseWindow_,
                                     skip_layers_,
                                     rhtools_.lastLayer(isHFnose),
                                     max_delta_time_,
                                     rhtools_.lastLayerEE(isHFnose),
                                     rhtools_.lastLayerFH(),
                                     siblings_maxRSquared_);
 
   theGraph_->findNtuplets(foundNtuplets, seedIndices, min_clusters_per_ntuplet_, out_in_dfs_, max_out_in_hops_);
   //#ifdef FP_DEBUG
   const auto &doublets = theGraph_->getAllDoublets();
   int tracksterId = -1;
 
   // container for holding tracksters before selection
   std::vector<Trackster> tmpTracksters;
   tmpTracksters.reserve(foundNtuplets.size());
 
   for (auto const &ntuplet : foundNtuplets) {
     tracksterId++;
 
     std::set<unsigned int> effective_cluster_idx;
 
     for (auto const &doublet : ntuplet) {
       auto innerCluster = doublets[doublet].innerClusterId();
       auto outerCluster = doublets[doublet].outerClusterId();
 
       effective_cluster_idx.insert(innerCluster);
       effective_cluster_idx.insert(outerCluster);
 
       if (PatternRecognitionAlgoBaseT<TILES>::algo_verbosity_ > VerbosityLevel::Advanced) {
         LogDebug("HGCPatternRecoByCA") << " New doublet " << doublet << " for trackster: " << result.size()
                                        << " InnerCl " << innerCluster << " " << input.layerClusters[innerCluster].x()
                                        << " " << input.layerClusters[innerCluster].y() << " "
                                        << input.layerClusters[innerCluster].z() << " OuterCl " << outerCluster << " "
                                        << input.layerClusters[outerCluster].x() << " "
                                        << input.layerClusters[outerCluster].y() << " "
                                        << input.layerClusters[outerCluster].z() << " " << tracksterId << std::endl;
       }
     }
     unsigned showerMinLayerId = 99999;
     std::vector<unsigned int> uniqueLayerIds;
     uniqueLayerIds.reserve(effective_cluster_idx.size());
     std::vector<std::pair<unsigned int, unsigned int>> lcIdAndLayer;
     lcIdAndLayer.reserve(effective_cluster_idx.size());
     for (auto const i : effective_cluster_idx) {
       auto const &haf = input.layerClusters[i].hitsAndFractions();
       auto layerId = rhtools_.getLayerWithOffset(haf[0].first);
       showerMinLayerId = std::min(layerId, showerMinLayerId);
       uniqueLayerIds.push_back(layerId);
       lcIdAndLayer.emplace_back(i, layerId);
     }
     std::sort(uniqueLayerIds.begin(), uniqueLayerIds.end());
     uniqueLayerIds.erase(std::unique(uniqueLayerIds.begin(), uniqueLayerIds.end()), uniqueLayerIds.end());
     unsigned int numberOfLayersInTrackster = uniqueLayerIds.size();
     if (check_missing_layers_) {
       int numberOfMissingLayers = 0;
       unsigned int j = showerMinLayerId;
       unsigned int indexInVec = 0;
       for (const auto &layer : uniqueLayerIds) {
         if (layer != j) {
           numberOfMissingLayers++;
           j++;
           if (numberOfMissingLayers > max_missing_layers_in_trackster_) {
             numberOfLayersInTrackster = indexInVec;
             for (auto &llpair : lcIdAndLayer) {
               if (llpair.second >= layer) {
                 effective_cluster_idx.erase(llpair.first);
               }
             }
             break;
           }
         }
         indexInVec++;
         j++;
       }
     }
     if ((numberOfLayersInTrackster >= min_layers_per_trackster_) and (showerMinLayerId <= shower_start_max_layer_)) {
       // Put back indices, in the form of a Trackster, into the results vector
       Trackster tmp;
       tmp.vertices().reserve(effective_cluster_idx.size());
       tmp.vertex_multiplicity().resize(effective_cluster_idx.size(), 1);
       //regions and seedIndices can have different size
       //if a seeding region does not lead to any trackster
       tmp.setSeed(input.regions[0].collectionID, seedIndices[tracksterId]);
 
       std::copy(std::begin(effective_cluster_idx), std::end(effective_cluster_idx), std::back_inserter(tmp.vertices()));
       tmpTracksters.push_back(tmp);
     }
   }
   ticl::assignPCAtoTracksters(tmpTracksters,
                               input.layerClusters,
                               input.layerClustersTime,
                               rhtools_.getPositionLayer(rhtools_.lastLayerEE(isHFnose), isHFnose).z(),
                               rhtools_,
                               computeLocalTime_);
 
   // run energy regression and ID
   energyRegressionAndID(input.layerClusters, input.tfSession, tmpTracksters);
   // Filter results based on PID criteria or EM/Total energy ratio.
   // We want to **keep** tracksters whose cumulative
   // probability summed up over the selected categories
   // is greater than the chosen threshold. Therefore
   // the filtering function should **discard** all
   // tracksters **below** the threshold.
   auto filter_on_pids = [&](Trackster &t) -> bool {
     auto cumulative_prob = 0.;
     for (auto index : filter_on_categories_) {
       cumulative_prob += t.id_probabilities(index);
     }
     return (cumulative_prob <= pid_threshold_) &&
            (t.raw_em_energy() < energy_em_over_total_threshold_ * t.raw_energy());
   };
 
   std::vector<unsigned int> selectedTrackstersIds;
   for (unsigned i = 0; i < tmpTracksters.size(); ++i) {
     if (!filter_on_pids(tmpTracksters[i]) and tmpTracksters[i].sigmasPCA()[0] < max_longitudinal_sigmaPCA_) {
       selectedTrackstersIds.push_back(i);
     }
   }
 
   result.reserve(selectedTrackstersIds.size());
 
   for (unsigned i = 0; i < selectedTrackstersIds.size(); ++i) {
     const auto &t = tmpTracksters[selectedTrackstersIds[i]];
     for (auto const lcId : t.vertices()) {
       layer_cluster_usage[lcId]++;
       if (PatternRecognitionAlgoBaseT<TILES>::algo_verbosity_ > VerbosityLevel::Basic)
         LogDebug("HGCPatternRecoByCA") << "LayerID: " << lcId << " count: " << (int)layer_cluster_usage[lcId]
                                        << std::endl;
     }
     if (isRegionalIter) {
       seedToTracksterAssociation[t.seedIndex()].push_back(i);
     }
     result.push_back(t);
   }
 
   for (auto &trackster : result) {
     assert(trackster.vertices().size() <= trackster.vertex_multiplicity().size());
     for (size_t i = 0; i < trackster.vertices().size(); ++i) {
       trackster.vertex_multiplicity()[i] = layer_cluster_usage[trackster.vertices(i)];
       if (PatternRecognitionAlgoBaseT<TILES>::algo_verbosity_ > VerbosityLevel::Basic)
         LogDebug("HGCPatternRecoByCA") << "LayerID: " << trackster.vertices(i)
                                        << " count: " << (int)trackster.vertex_multiplicity(i) << std::endl;
     }
   }
   // Now decide if the tracksters from the track-based iterations have to be merged
   if (oneTracksterPerTrackSeed_) {
     std::vector<Trackster> tmp;
     mergeTrackstersTRK(result, input.layerClusters, tmp, seedToTracksterAssociation);
     tmp.swap(result);
   }
 
   ticl::assignPCAtoTracksters(result,
                               input.layerClusters,
                               input.layerClustersTime,
                               rhtools_.getPositionLayer(rhtools_.lastLayerEE(isHFnose), isHFnose).z(),
                               rhtools_,
                               computeLocalTime_);
 
   // run energy regression and ID
   energyRegressionAndID(input.layerClusters, input.tfSession, result);
 
   // now adding dummy tracksters from seeds not connected to any shower in the result collection
   // these are marked as charged hadrons with probability 1.
   if (promoteEmptyRegionToTrackster_) {
     emptyTrackstersFromSeedsTRK(result, seedToTracksterAssociation, input.regions[0].collectionID);
   }
 
   if (PatternRecognitionAlgoBaseT<TILES>::algo_verbosity_ > VerbosityLevel::Advanced) {
     for (auto &trackster : result) {
       LogDebug("HGCPatternRecoByCA") << "Trackster characteristics: " << std::endl;
       LogDebug("HGCPatternRecoByCA") << "Size: " << trackster.vertices().size() << std::endl;
       auto counter = 0;
       for (auto const &p : trackster.id_probabilities()) {
         LogDebug("HGCPatternRecoByCA") << counter++ << ": " << p << std::endl;
       }
     }
   }
   theGraph_->clear();
 }

◆ mergeTrackstersTRK()

template<typename TILES >

void PatternRecognitionbyCA::mergeTrackstersTRK	(	const std::vector< Trackster > &	input,
		const std::vector< reco::CaloCluster > &	layerClusters,
		std::vector< Trackster > &	output,
		std::unordered_map< int, std::vector< int >> &	seedToTracksterAssociation
	)		const

private

Definition at line 276 of file PatternRecognitionbyCA.cc.

References ticl::Basic, filterCSVwithJSON::copy, HLT_2024v14_cff::distance, spr::find(), input, dqmiolumiharvest::j, LogDebug, GetRecoTauVFromDQM_MC_cff::next, and jetUpdater_cfi::sort.

                                                                              {
   output.reserve(input.size());
   for (auto &thisSeed : seedToTracksterAssociation) {
     auto &tracksters = thisSeed.second;
     if (!tracksters.empty()) {
       auto numberOfTrackstersInSeed = tracksters.size();
       output.emplace_back(input[tracksters[0]]);
       auto &outTrackster = output.back();
       tracksters[0] = output.size() - 1;
       auto updated_size = outTrackster.vertices().size();
       for (unsigned int j = 1; j < numberOfTrackstersInSeed; ++j) {
         auto &thisTrackster = input[tracksters[j]];
         updated_size += thisTrackster.vertices().size();
         if (PatternRecognitionAlgoBaseT<TILES>::algo_verbosity_ > VerbosityLevel::Basic) {
           LogDebug("HGCPatternRecoByCA") << "Updated size: " << updated_size << std::endl;
         }
         outTrackster.vertices().reserve(updated_size);
         outTrackster.vertex_multiplicity().reserve(updated_size);
         std::copy(std::begin(thisTrackster.vertices()),
                   std::end(thisTrackster.vertices()),
                   std::back_inserter(outTrackster.vertices()));
         std::copy(std::begin(thisTrackster.vertex_multiplicity()),
                   std::end(thisTrackster.vertex_multiplicity()),
                   std::back_inserter(outTrackster.vertex_multiplicity()));
       }
       tracksters.resize(1);
 
       // Find duplicate LCs
       auto &orig_vtx = outTrackster.vertices();
       auto vtx_sorted{orig_vtx};
       std::sort(std::begin(vtx_sorted), std::end(vtx_sorted));
       for (unsigned int iLC = 1; iLC < vtx_sorted.size(); ++iLC) {
         if (vtx_sorted[iLC] == vtx_sorted[iLC - 1]) {
           // Clean up duplicate LCs
           const auto lcIdx = vtx_sorted[iLC];
           const auto firstEl = std::find(orig_vtx.begin(), orig_vtx.end(), lcIdx);
           const auto firstPos = std::distance(std::begin(orig_vtx), firstEl);
           auto iDup = std::find(std::next(firstEl), orig_vtx.end(), lcIdx);
           while (iDup != orig_vtx.end()) {
             orig_vtx.erase(iDup);
             outTrackster.vertex_multiplicity().erase(outTrackster.vertex_multiplicity().begin() +
                                                      std::distance(std::begin(orig_vtx), iDup));
             outTrackster.vertex_multiplicity()[firstPos] -= 1;
             iDup = std::find(std::next(firstEl), orig_vtx.end(), lcIdx);
           };
         }
       }
     }
   }
   output.shrink_to_fit();
 }

Member Data Documentation

◆ caloGeomToken_

template<typename TILES >

edm::ESGetToken<CaloGeometry, CaloGeometryRecord> ticl::PatternRecognitionbyCA< TILES >::caloGeomToken_

private

Definition at line 37 of file PatternRecognitionbyCA.h.

◆ check_missing_layers_

template<typename TILES >

bool ticl::PatternRecognitionbyCA< TILES >::check_missing_layers_ = false

private

Definition at line 49 of file PatternRecognitionbyCA.h.

◆ computeLocalTime_

template<typename TILES >

const bool ticl::PatternRecognitionbyCA< TILES >::computeLocalTime_

private

Definition at line 64 of file PatternRecognitionbyCA.h.

◆ eidInputName_

template<typename TILES >

const std::string ticl::PatternRecognitionbyCA< TILES >::eidInputName_

private

Definition at line 58 of file PatternRecognitionbyCA.h.

◆ eidMinClusterEnergy_

template<typename TILES >

const float ticl::PatternRecognitionbyCA< TILES >::eidMinClusterEnergy_

private

Definition at line 61 of file PatternRecognitionbyCA.h.

◆ eidNClusters_

template<typename TILES >

const int ticl::PatternRecognitionbyCA< TILES >::eidNClusters_

private

Definition at line 63 of file PatternRecognitionbyCA.h.

◆ eidNFeatures_

template<typename TILES >

const int ticl::PatternRecognitionbyCA< TILES >::eidNFeatures_ = 3

staticprivate

Definition at line 70 of file PatternRecognitionbyCA.h.

◆ eidNLayers_

template<typename TILES >

const int ticl::PatternRecognitionbyCA< TILES >::eidNLayers_

private

Definition at line 62 of file PatternRecognitionbyCA.h.

◆ eidOutputNameEnergy_

template<typename TILES >

const std::string ticl::PatternRecognitionbyCA< TILES >::eidOutputNameEnergy_

private

Definition at line 59 of file PatternRecognitionbyCA.h.

◆ eidOutputNameId_

template<typename TILES >

const std::string ticl::PatternRecognitionbyCA< TILES >::eidOutputNameId_

private

Definition at line 60 of file PatternRecognitionbyCA.h.

◆ eidSession_

template<typename TILES >

tensorflow::Session* ticl::PatternRecognitionbyCA< TILES >::eidSession_

private

Definition at line 67 of file PatternRecognitionbyCA.h.

◆ energy_em_over_total_threshold_

template<typename TILES >

const double ticl::PatternRecognitionbyCA< TILES >::energy_em_over_total_threshold_

private

Definition at line 54 of file PatternRecognitionbyCA.h.

◆ etaLimitIncreaseWindow_

template<typename TILES >

const float ticl::PatternRecognitionbyCA< TILES >::etaLimitIncreaseWindow_

private

Definition at line 46 of file PatternRecognitionbyCA.h.

◆ filter_on_categories_

template<typename TILES >

const std::vector<int> ticl::PatternRecognitionbyCA< TILES >::filter_on_categories_

private

Definition at line 52 of file PatternRecognitionbyCA.h.

◆ max_delta_time_

template<typename TILES >

const float ticl::PatternRecognitionbyCA< TILES >::max_delta_time_

private

Definition at line 57 of file PatternRecognitionbyCA.h.

◆ max_longitudinal_sigmaPCA_

template<typename TILES >

const double ticl::PatternRecognitionbyCA< TILES >::max_longitudinal_sigmaPCA_

private

Definition at line 55 of file PatternRecognitionbyCA.h.

◆ max_missing_layers_in_trackster_

template<typename TILES >

const int ticl::PatternRecognitionbyCA< TILES >::max_missing_layers_in_trackster_

private

Definition at line 48 of file PatternRecognitionbyCA.h.

◆ max_out_in_hops_

template<typename TILES >

const unsigned int ticl::PatternRecognitionbyCA< TILES >::max_out_in_hops_

private

Definition at line 42 of file PatternRecognitionbyCA.h.

◆ min_clusters_per_ntuplet_

template<typename TILES >

const int ticl::PatternRecognitionbyCA< TILES >::min_clusters_per_ntuplet_

private

Definition at line 56 of file PatternRecognitionbyCA.h.

◆ min_cos_pointing_

template<typename TILES >

const float ticl::PatternRecognitionbyCA< TILES >::min_cos_pointing_

private

Definition at line 44 of file PatternRecognitionbyCA.h.

◆ min_cos_theta_

template<typename TILES >

const float ticl::PatternRecognitionbyCA< TILES >::min_cos_theta_

private

Definition at line 43 of file PatternRecognitionbyCA.h.

◆ min_layers_per_trackster_

template<typename TILES >

const unsigned int ticl::PatternRecognitionbyCA< TILES >::min_layers_per_trackster_

private

Definition at line 51 of file PatternRecognitionbyCA.h.

◆ oneTracksterPerTrackSeed_

template<typename TILES >

const bool ticl::PatternRecognitionbyCA< TILES >::oneTracksterPerTrackSeed_

private

Definition at line 39 of file PatternRecognitionbyCA.h.

◆ out_in_dfs_

template<typename TILES >

const bool ticl::PatternRecognitionbyCA< TILES >::out_in_dfs_

private

Definition at line 41 of file PatternRecognitionbyCA.h.

◆ pid_threshold_

template<typename TILES >

const double ticl::PatternRecognitionbyCA< TILES >::pid_threshold_

private

Definition at line 53 of file PatternRecognitionbyCA.h.

◆ promoteEmptyRegionToTrackster_

template<typename TILES >

const bool ticl::PatternRecognitionbyCA< TILES >::promoteEmptyRegionToTrackster_

private

Definition at line 40 of file PatternRecognitionbyCA.h.

◆ rhtools_

template<typename TILES >

hgcal::RecHitTools ticl::PatternRecognitionbyCA< TILES >::rhtools_

private

Definition at line 66 of file PatternRecognitionbyCA.h.

◆ root_doublet_max_distance_from_seed_squared_

template<typename TILES >

const float ticl::PatternRecognitionbyCA< TILES >::root_doublet_max_distance_from_seed_squared_

private

Definition at line 45 of file PatternRecognitionbyCA.h.

◆ shower_start_max_layer_

template<typename TILES >

const unsigned int ticl::PatternRecognitionbyCA< TILES >::shower_start_max_layer_

private

Definition at line 50 of file PatternRecognitionbyCA.h.

◆ siblings_maxRSquared_

template<typename TILES >

const std::vector<double> ticl::PatternRecognitionbyCA< TILES >::siblings_maxRSquared_

private

Definition at line 68 of file PatternRecognitionbyCA.h.

◆ skip_layers_

template<typename TILES >

const int ticl::PatternRecognitionbyCA< TILES >::skip_layers_

private

Definition at line 47 of file PatternRecognitionbyCA.h.

◆ theGraph_

template<typename TILES >

const std::unique_ptr<HGCGraphT<TILES> > ticl::PatternRecognitionbyCA< TILES >::theGraph_

private

Definition at line 38 of file PatternRecognitionbyCA.h.

Public Member Functions

Static Public Member Functions

Private Member Functions

Private Attributes

Static Private Attributes

Additional Inherited Members

Detailed Description

template<typename TILES> class ticl::PatternRecognitionbyCA< TILES >

Constructor & Destructor Documentation

◆ PatternRecognitionbyCA()

◆ ~PatternRecognitionbyCA()

Member Function Documentation

◆ emptyTrackstersFromSeedsTRK()

◆ energyRegressionAndID()

◆ fillPSetDescription()

◆ makeTracksters()

◆ mergeTrackstersTRK()

Member Data Documentation

◆ caloGeomToken_

◆ check_missing_layers_

◆ computeLocalTime_

◆ eidInputName_

◆ eidMinClusterEnergy_

◆ eidNClusters_

◆ eidNFeatures_

◆ eidNLayers_

◆ eidOutputNameEnergy_

◆ eidOutputNameId_

◆ eidSession_

◆ energy_em_over_total_threshold_

◆ etaLimitIncreaseWindow_

◆ filter_on_categories_

◆ max_delta_time_

◆ max_longitudinal_sigmaPCA_

◆ max_missing_layers_in_trackster_

◆ max_out_in_hops_

◆ min_clusters_per_ntuplet_

◆ min_cos_pointing_

◆ min_cos_theta_

◆ min_layers_per_trackster_

◆ oneTracksterPerTrackSeed_

◆ out_in_dfs_

◆ pid_threshold_

◆ promoteEmptyRegionToTrackster_

◆ rhtools_

◆ root_doublet_max_distance_from_seed_squared_

◆ shower_start_max_layer_

◆ siblings_maxRSquared_

◆ skip_layers_

◆ theGraph_

template<typename TILES>
class ticl::PatternRecognitionbyCA< TILES >