CMS 3D CMS Logo

List of all members | Public Member Functions | Private Types | Private Member Functions | Private Attributes | Static Private Attributes
HGCalHistoClusteringImpl Class Reference

#include <HGCalHistoClusteringImpl.h>

Public Member Functions

void clusterizeHisto (const std::vector< edm::Ptr< l1t::HGCalCluster >> &clustersPtr, const std::vector< std::pair< GlobalPoint, double >> &seedPositionsEnergy, const HGCalTriggerGeometryBase &triggerGeometry, l1t::HGCalMulticlusterBxCollection &multiclusters, l1t::HGCalClusterBxCollection &rejected_clusters) const
 
float dR (const l1t::HGCalCluster &clu, const GlobalPoint &seed) const
 
 HGCalHistoClusteringImpl (const edm::ParameterSet &conf)
 
void setGeometry (const HGCalTriggerGeometryBase *const geom)
 

Private Types

enum  ClusterAssociationStrategy { NearestNeighbour, EnergySplit }
 

Private Member Functions

std::vector< l1t::HGCalMulticlusterclusterSeedMulticluster (const std::vector< edm::Ptr< l1t::HGCalCluster >> &clustersPtrs, const std::vector< std::pair< GlobalPoint, double >> &seeds, std::vector< l1t::HGCalCluster > &rejected_clusters) const
 
void finalizeClusters (std::vector< l1t::HGCalMulticluster > &, const std::vector< l1t::HGCalCluster > &, l1t::HGCalMulticlusterBxCollection &, l1t::HGCalClusterBxCollection &, const HGCalTriggerGeometryBase &) const
 

Private Attributes

std::string cluster_association_input_
 
ClusterAssociationStrategy cluster_association_strategy_
 
double dr_
 
std::vector< double > dr_byLayer_coefficientA_
 
std::vector< double > dr_byLayer_coefficientB_
 
std::unique_ptr< HGCalTriggerClusterIdentificationBaseid_
 
double ptC3dThreshold_
 
HGCalShowerShape shape_
 
HGCalTriggerTools triggerTools_
 

Static Private Attributes

static constexpr double kMidRadius_ = 2.3
 

Detailed Description

Definition at line 14 of file HGCalHistoClusteringImpl.h.

Member Enumeration Documentation

◆ ClusterAssociationStrategy

Constructor & Destructor Documentation

◆ HGCalHistoClusteringImpl()

HGCalHistoClusteringImpl::HGCalHistoClusteringImpl ( const edm::ParameterSet conf)

Definition at line 6 of file HGCalHistoClusteringImpl.cc.

References cluster_association_input_, cluster_association_strategy_, dr_, EnergySplit, Exception, get, edm::ParameterSet::getParameter(), id_, NearestNeighbour, and ptC3dThreshold_.

7  : dr_(conf.getParameter<double>("dR_multicluster")),
8  dr_byLayer_coefficientA_(conf.existsAs<std::vector<double>>("dR_multicluster_byLayer_coefficientA")
9  ? conf.getParameter<std::vector<double>>("dR_multicluster_byLayer_coefficientA")
10  : std::vector<double>()),
11  dr_byLayer_coefficientB_(conf.existsAs<std::vector<double>>("dR_multicluster_byLayer_coefficientB")
12  ? conf.getParameter<std::vector<double>>("dR_multicluster_byLayer_coefficientB")
13  : std::vector<double>()),
14  ptC3dThreshold_(conf.getParameter<double>("minPt_multicluster")),
15  cluster_association_input_(conf.getParameter<string>("cluster_association")),
16  shape_(conf) {
17  if (cluster_association_input_ == "NearestNeighbour") {
19  } else if (cluster_association_input_ == "EnergySplit") {
21  } else {
22  throw cms::Exception("HGCTriggerParameterError")
23  << "Unknown cluster association strategy'" << cluster_association_strategy_;
24  }
25 
26  edm::LogInfo("HGCalMulticlusterParameters")
27  << "Multicluster dR: " << dr_ << "\nMulticluster minimum transverse-momentum: " << ptC3dThreshold_;
28 
29  id_ = std::unique_ptr<HGCalTriggerClusterIdentificationBase>{
30  HGCalTriggerClusterIdentificationFactory::get()->create("HGCalTriggerClusterIdentificationBDT")};
31  id_->initialize(conf.getParameter<edm::ParameterSet>("EGIdentification"));
32 }
T getParameter(std::string const &) const
Definition: ParameterSet.h:303
std::unique_ptr< HGCalTriggerClusterIdentificationBase > id_
bool existsAs(std::string const &parameterName, bool trackiness=true) const
checks if a parameter exists as a given type
Definition: ParameterSet.h:171
std::vector< double > dr_byLayer_coefficientB_
Log< level::Info, false > LogInfo
ClusterAssociationStrategy cluster_association_strategy_
#define get
std::vector< double > dr_byLayer_coefficientA_

Member Function Documentation

◆ clusterizeHisto()

void HGCalHistoClusteringImpl::clusterizeHisto ( const std::vector< edm::Ptr< l1t::HGCalCluster >> &  clustersPtr,
const std::vector< std::pair< GlobalPoint, double >> &  seedPositionsEnergy,
const HGCalTriggerGeometryBase triggerGeometry,
l1t::HGCalMulticlusterBxCollection multiclusters,
l1t::HGCalClusterBxCollection rejected_clusters 
) const

Definition at line 110 of file HGCalHistoClusteringImpl.cc.

References clusterSeedMulticluster(), and finalizeClusters().

114  {
115  /* clusterize clusters around seeds */
116  std::vector<l1t::HGCalCluster> rejected_clusters_vec;
117  std::vector<l1t::HGCalMulticluster> multiclusters_vec =
118  clusterSeedMulticluster(clustersPtrs, seedPositionsEnergy, rejected_clusters_vec);
119  /* making the collection of multiclusters */
120  finalizeClusters(multiclusters_vec, rejected_clusters_vec, multiclusters, rejected_clusters, triggerGeometry);
121 }
void finalizeClusters(std::vector< l1t::HGCalMulticluster > &, const std::vector< l1t::HGCalCluster > &, l1t::HGCalMulticlusterBxCollection &, l1t::HGCalClusterBxCollection &, const HGCalTriggerGeometryBase &) const
std::vector< l1t::HGCalMulticluster > clusterSeedMulticluster(const std::vector< edm::Ptr< l1t::HGCalCluster >> &clustersPtrs, const std::vector< std::pair< GlobalPoint, double >> &seeds, std::vector< l1t::HGCalCluster > &rejected_clusters) const

◆ clusterSeedMulticluster()

std::vector< l1t::HGCalMulticluster > HGCalHistoClusteringImpl::clusterSeedMulticluster ( const std::vector< edm::Ptr< l1t::HGCalCluster >> &  clustersPtrs,
const std::vector< std::pair< GlobalPoint, double >> &  seeds,
std::vector< l1t::HGCalCluster > &  rejected_clusters 
) const
private

Definition at line 38 of file HGCalHistoClusteringImpl.cc.

References funct::abs(), cluster_association_strategy_, ztail::d, dR(), dr_, dr_byLayer_coefficientA_, dr_byLayer_coefficientB_, HCALHighEnergyHPDFilter_cfi::energy, EnergySplit, first, iseed, kMidRadius_, HGCalTriggerTools::layerWithOffset(), NearestNeighbour, DetachedQuadStep_cff::seeds, triggerTools_, PV3DBase< T, PVType, FrameType >::z(), and HGCalTriggerTools::zside().

Referenced by clusterizeHisto().

41  {
42  std::map<int, l1t::HGCalMulticluster> mapSeedMulticluster;
43  std::vector<l1t::HGCalMulticluster> multiclustersOut;
44 
45  for (const auto& clu : clustersPtrs) {
46  int z_side = triggerTools_.zside(clu->detId());
47 
48  double radiusCoefficientA =
50  double radiusCoefficientB =
52 
53  double minDist = radiusCoefficientA + radiusCoefficientB * (kMidRadius_ - std::abs(clu->eta()));
54 
55  std::vector<pair<int, double>> targetSeedsEnergy;
56 
57  for (unsigned int iseed = 0; iseed < seeds.size(); iseed++) {
58  GlobalPoint seedPosition = seeds[iseed].first;
59  double seedEnergy = seeds[iseed].second;
60 
61  if (z_side * seedPosition.z() < 0)
62  continue;
63  double d = this->dR(*clu, seeds[iseed].first);
64 
65  if (d < minDist) {
67  targetSeedsEnergy.emplace_back(iseed, seedEnergy);
69  minDist = d;
70 
71  if (targetSeedsEnergy.empty()) {
72  targetSeedsEnergy.emplace_back(iseed, seedEnergy);
73  } else {
74  targetSeedsEnergy.at(0).first = iseed;
75  targetSeedsEnergy.at(0).second = seedEnergy;
76  }
77  }
78  }
79  }
80 
81  if (targetSeedsEnergy.empty()) {
82  rejected_clusters.emplace_back(*clu);
83  continue;
84  }
85  //Loop over target seeds and divide up the clusters energy
86  double totalTargetSeedEnergy = 0;
87  for (const auto& energy : targetSeedsEnergy) {
88  totalTargetSeedEnergy += energy.second;
89  }
90 
91  for (const auto& energy : targetSeedsEnergy) {
92  double seedWeight = 1;
94  seedWeight = energy.second / totalTargetSeedEnergy;
95  if (mapSeedMulticluster[energy.first].size() == 0) {
96  mapSeedMulticluster[energy.first] = l1t::HGCalMulticluster(clu, seedWeight);
97  } else {
98  mapSeedMulticluster[energy.first].addConstituent(clu, true, seedWeight);
99  }
100  }
101  }
102 
103  multiclustersOut.reserve(mapSeedMulticluster.size());
104  for (const auto& mclu : mapSeedMulticluster)
105  multiclustersOut.emplace_back(mclu.second);
106 
107  return multiclustersOut;
108 }
T z() const
Definition: PV3DBase.h:61
float dR(const l1t::HGCalCluster &clu, const GlobalPoint &seed) const
int zside(const DetId &) const
static constexpr double kMidRadius_
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
std::vector< double > dr_byLayer_coefficientB_
d
Definition: ztail.py:151
unsigned layerWithOffset(const DetId &) const
ClusterAssociationStrategy cluster_association_strategy_
int iseed
Definition: AMPTWrapper.h:134
std::vector< double > dr_byLayer_coefficientA_

◆ dR()

float HGCalHistoClusteringImpl::dR ( const l1t::HGCalCluster clu,
const GlobalPoint seed 
) const

Definition at line 34 of file HGCalHistoClusteringImpl.cc.

References l1t::HGCalClusterT< C >::centreProj(), mag(), and fileCollector::seed.

Referenced by clusterSeedMulticluster().

34  {
35  return (seed - clu.centreProj()).mag();
36 }
T mag() const
The vector magnitude. Equivalent to sqrt(vec.mag2())
const GlobalPoint & centreProj() const

◆ finalizeClusters()

void HGCalHistoClusteringImpl::finalizeClusters ( std::vector< l1t::HGCalMulticluster > &  multiclusters_in,
const std::vector< l1t::HGCalCluster > &  rejected_clusters_in,
l1t::HGCalMulticlusterBxCollection multiclusters_out,
l1t::HGCalClusterBxCollection rejected_clusters_out,
const HGCalTriggerGeometryBase triggerGeometry 
) const
private

Definition at line 123 of file HGCalHistoClusteringImpl.cc.

References HGCalShowerShape::fillShapes(), id_, ptC3dThreshold_, BXVector< T >::push_back(), shape_, TtFullHadEvtBuilder_cfi::sumPt, and pfClustersFromHGC3DClusters_cfi::wp.

Referenced by clusterizeHisto().

127  {
128  for (const auto& tc : rejected_clusters_in) {
129  rejected_clusters_out.push_back(0, tc);
130  }
131 
132  for (auto& multicluster : multiclusters_in) {
133  // compute the eta, phi from its barycenter
134  // + pT as scalar sum of pT of constituents
135  double sumPt = multicluster.sumPt();
136 
137  math::PtEtaPhiMLorentzVector multiclusterP4(sumPt, multicluster.centre().eta(), multicluster.centre().phi(), 0.);
138  multicluster.setP4(multiclusterP4);
139 
140  if (multicluster.pt() > ptC3dThreshold_) {
141  //compute shower shapes
142  shape_.fillShapes(multicluster, triggerGeometry);
143  // fill quality flag
144  unsigned hwQual = 0;
145  for (unsigned wp = 0; wp < id_->working_points().size(); wp++) {
146  hwQual |= (id_->decision(multicluster, wp) << wp);
147  }
148  multicluster.setHwQual(hwQual);
149  // fill H/E
150  multicluster.saveHOverE();
151 
152  multiclusters_out.push_back(0, multicluster);
153  } else {
154  for (const auto& tc : multicluster.constituents()) {
155  rejected_clusters_out.push_back(0, *(tc.second));
156  }
157  }
158  }
159 }
std::unique_ptr< HGCalTriggerClusterIdentificationBase > id_
PtEtaPhiMLorentzVectorD PtEtaPhiMLorentzVector
Lorentz vector with cartesian internal representation.
Definition: LorentzVector.h:25
void fillShapes(l1t::HGCalMulticluster &, const HGCalTriggerGeometryBase &) const
void push_back(int bx, T object)

◆ setGeometry()

void HGCalHistoClusteringImpl::setGeometry ( const HGCalTriggerGeometryBase *const  geom)
inline

Definition at line 18 of file HGCalHistoClusteringImpl.h.

References dr_byLayer_coefficientA_, dr_byLayer_coefficientB_, Exception, relativeConstraints::geom, HGCalTriggerTools::lastLayerBH(), HGCalShowerShape::setGeometry(), HGCalTriggerTools::setGeometry(), shape_, and triggerTools_.

18  {
23  throw cms::Exception("Configuration")
24  << "The per-layer dR values go up to " << (dr_byLayer_coefficientA_.size() - 1) << "(A) and "
25  << (dr_byLayer_coefficientB_.size() - 1) << "(B), while layers go up to " << triggerTools_.lastLayerBH()
26  << "\n";
27  }
28  }
void setGeometry(const HGCalTriggerGeometryBase *const)
void setGeometry(const HGCalTriggerGeometryBase *const geom)
std::vector< double > dr_byLayer_coefficientB_
unsigned lastLayerBH() const
std::vector< double > dr_byLayer_coefficientA_

Member Data Documentation

◆ cluster_association_input_

std::string HGCalHistoClusteringImpl::cluster_association_input_
private

Definition at line 57 of file HGCalHistoClusteringImpl.h.

Referenced by HGCalHistoClusteringImpl().

◆ cluster_association_strategy_

ClusterAssociationStrategy HGCalHistoClusteringImpl::cluster_association_strategy_
private

Definition at line 58 of file HGCalHistoClusteringImpl.h.

Referenced by clusterSeedMulticluster(), and HGCalHistoClusteringImpl().

◆ dr_

double HGCalHistoClusteringImpl::dr_
private

Definition at line 52 of file HGCalHistoClusteringImpl.h.

Referenced by clusterSeedMulticluster(), and HGCalHistoClusteringImpl().

◆ dr_byLayer_coefficientA_

std::vector<double> HGCalHistoClusteringImpl::dr_byLayer_coefficientA_
private

Definition at line 53 of file HGCalHistoClusteringImpl.h.

Referenced by clusterSeedMulticluster(), and setGeometry().

◆ dr_byLayer_coefficientB_

std::vector<double> HGCalHistoClusteringImpl::dr_byLayer_coefficientB_
private

Definition at line 54 of file HGCalHistoClusteringImpl.h.

Referenced by clusterSeedMulticluster(), and setGeometry().

◆ id_

std::unique_ptr<HGCalTriggerClusterIdentificationBase> HGCalHistoClusteringImpl::id_
private

Definition at line 62 of file HGCalHistoClusteringImpl.h.

Referenced by finalizeClusters(), and HGCalHistoClusteringImpl().

◆ kMidRadius_

constexpr double HGCalHistoClusteringImpl::kMidRadius_ = 2.3
staticprivate

Definition at line 64 of file HGCalHistoClusteringImpl.h.

Referenced by clusterSeedMulticluster().

◆ ptC3dThreshold_

double HGCalHistoClusteringImpl::ptC3dThreshold_
private

Definition at line 55 of file HGCalHistoClusteringImpl.h.

Referenced by finalizeClusters(), and HGCalHistoClusteringImpl().

◆ shape_

HGCalShowerShape HGCalHistoClusteringImpl::shape_
private

Definition at line 60 of file HGCalHistoClusteringImpl.h.

Referenced by finalizeClusters(), and setGeometry().

◆ triggerTools_

HGCalTriggerTools HGCalHistoClusteringImpl::triggerTools_
private

Definition at line 61 of file HGCalHistoClusteringImpl.h.

Referenced by clusterSeedMulticluster(), and setGeometry().