CMS 3D CMS Logo

List of all members | Classes | Public Member Functions | Private Types | Private Member Functions | Private Attributes
PFMultiDepthClusterizer Class Referencefinal
Inheritance diagram for PFMultiDepthClusterizer:
PFClusterBuilderBase

Classes

class  ClusterLink
 

Public Member Functions

void buildClusters (const reco::PFClusterCollection &, const std::vector< bool > &, reco::PFClusterCollection &outclus) override
 
B2DGPFoperator= (const B2DGPF &)=delete
 
 PFMultiDepthClusterizer (const edm::ParameterSet &conf, edm::ConsumesCollector &cc)
 
 PFMultiDepthClusterizer (const B2DGPF &)=delete
 
void update (const edm::EventSetup &es) override
 
 ~PFMultiDepthClusterizer () override=default
 
- Public Member Functions inherited from PFClusterBuilderBase
std::ostream & operator<< (std::ostream &o) const
 
PFCBBoperator= (const PFCBB &)=delete
 
 PFClusterBuilderBase (const edm::ParameterSet &conf, edm::ConsumesCollector &cc)
 
 PFClusterBuilderBase (const PFCBB &)=delete
 
void reset ()
 
virtual ~PFClusterBuilderBase ()=default
 

Private Types

typedef PFMultiDepthClusterizer B2DGPF
 

Private Member Functions

void absorbCluster (reco::PFCluster &, const reco::PFCluster &)
 
void calculateShowerShapes (const reco::PFClusterCollection &, std::vector< double > &, std::vector< double > &)
 
void expandCluster (reco::PFCluster &, unsigned int point, std::vector< bool > &mask, const reco::PFClusterCollection &, const std::vector< ClusterLink > &links)
 
std::vector< ClusterLinklink (const reco::PFClusterCollection &, const std::vector< double > &, const std::vector< double > &)
 
std::vector< ClusterLinkprune (std::vector< ClusterLink > &, std::vector< bool > &linkedClusters)
 

Private Attributes

std::unique_ptr< PFCPositionCalculatorBase_allCellsPosCalc
 
double nSigmaEta_
 
double nSigmaPhi_
 

Additional Inherited Members

- Public Types inherited from PFClusterBuilderBase
typedef PFCPositionCalculatorBase PosCalc
 
- Protected Attributes inherited from PFClusterBuilderBase
const float _minFractionToKeep
 
unsigned _nClustersFound
 
unsigned _nSeeds
 
std::unique_ptr< PosCalc_positionCalc
 

Detailed Description

Definition at line 15 of file PFMultiDepthClusterizer.cc.

Member Typedef Documentation

◆ B2DGPF

Definition at line 16 of file PFMultiDepthClusterizer.cc.

Constructor & Destructor Documentation

◆ PFMultiDepthClusterizer() [1/2]

PFMultiDepthClusterizer::PFMultiDepthClusterizer ( const edm::ParameterSet conf,
edm::ConsumesCollector cc 
)

Definition at line 79 of file PFMultiDepthClusterizer.cc.

References _allCellsPosCalc, gpuPixelDoublets::cc, edm::ParameterSet::exists(), get, edm::ParameterSet::getParameter(), edm::ParameterSet::getParameterSet(), nSigmaEta_, nSigmaPhi_, conifer::pow(), and AlCaHLTBitMon_QueryRunRegistry::string.

80  : PFClusterBuilderBase(conf, cc) {
81  if (conf.exists("allCellsPositionCalc")) {
82  const edm::ParameterSet& acConf = conf.getParameterSet("allCellsPositionCalc");
83  const std::string& algoac = acConf.getParameter<std::string>("algoName");
84  _allCellsPosCalc = PFCPositionCalculatorFactory::get()->create(algoac, acConf, cc);
85  }
86 
87  nSigmaEta_ = pow(conf.getParameter<double>("nSigmaEta"), 2);
88  nSigmaPhi_ = pow(conf.getParameter<double>("nSigmaPhi"), 2);
89 }
T getParameter(std::string const &) const
Definition: ParameterSet.h:307
uint32_t cc[maxCellsPerHit]
Definition: gpuFishbone.h:49
bool exists(std::string const &parameterName) const
checks if a parameter exists
ParameterSet const & getParameterSet(std::string const &) const
constexpr int pow(int x)
Definition: conifer.h:24
std::unique_ptr< PFCPositionCalculatorBase > _allCellsPosCalc
PFClusterBuilderBase(const edm::ParameterSet &conf, edm::ConsumesCollector &cc)
#define get

◆ ~PFMultiDepthClusterizer()

PFMultiDepthClusterizer::~PFMultiDepthClusterizer ( )
overridedefault

◆ PFMultiDepthClusterizer() [2/2]

PFMultiDepthClusterizer::PFMultiDepthClusterizer ( const B2DGPF )
delete

Member Function Documentation

◆ absorbCluster()

void PFMultiDepthClusterizer::absorbCluster ( reco::PFCluster main,
const reco::PFCluster added 
)
private

Definition at line 253 of file PFMultiDepthClusterizer.cc.

References StorageManager_cfg::e1, HLT_2023v12_cff::fraction, reco::PFCluster::recHitFractions(), and reco::CaloCluster::seed().

Referenced by expandCluster().

253  {
254  double e1 = 0.0;
255  double e2 = 0.0;
256 
257  //find seeds
258  for (const auto& fraction : main.recHitFractions())
259  if (fraction.recHitRef()->detId() == main.seed()) {
260  e1 = fraction.recHitRef()->energy();
261  }
262 
263  for (const auto& fraction : added.recHitFractions()) {
264  main.addRecHitFraction(fraction);
265  if (fraction.recHitRef()->detId() == added.seed()) {
266  e2 = fraction.recHitRef()->energy();
267  }
268  }
269  if (e2 > e1)
270  main.setSeed(added.seed());
271 }
DetId seed() const
return DetId of seed
Definition: CaloCluster.h:218
const std::vector< reco::PFRecHitFraction > & recHitFractions() const
vector of rechit fractions
Definition: PFCluster.h:65
Definition: main.py:1

◆ buildClusters()

void PFMultiDepthClusterizer::buildClusters ( const reco::PFClusterCollection input,
const std::vector< bool > &  seedable,
reco::PFClusterCollection outclus 
)
overridevirtual

Implements PFClusterBuilderBase.

Definition at line 91 of file PFMultiDepthClusterizer.cc.

References _allCellsPosCalc, calculateShowerShapes(), expandCluster(), mps_fire::i, input, link(), electronStore::links, gpuClustering::pixelStatus::mask, and prune().

93  {
94  std::vector<double> etaRMS2(input.size(), 0.0);
95  std::vector<double> phiRMS2(input.size(), 0.0);
96 
97  //We need to sort the clusters for smaller to larger depth
98  // for (unsigned int i=0;i<input.size();++i)
99  // printf(" cluster%f %f \n",input[i].depth(),input[i].energy());
100 
101  //calculate cluster shapes
102  calculateShowerShapes(input, etaRMS2, phiRMS2);
103 
104  //link
105  auto&& links = link(input, etaRMS2, phiRMS2);
106  // for (const auto& link: links)
107  // printf("link %d %d %f %f\n",link.from(),link.to(),link.dR(),link.dZ());
108 
109  std::vector<bool> mask(input.size(), false);
110  std::vector<bool> linked(input.size(), false);
111 
112  //prune
113  auto&& prunedLinks = prune(links, linked);
114 
115  //printf("Pruned links\n")
116  // for (const auto& link: prunedLinks)
117  // printf("link %d %d %f %f\n",link.from(),link.to(),link.dR(),link.dZ());
118 
119  //now we need to build clusters
120  for (unsigned int i = 0; i < input.size(); ++i) {
121  //if not masked
122  if (mask[i])
123  continue;
124  //if not linked just spit it out
125  if (!linked[i]) {
126  output.push_back(input[i]);
127  // printf("Added single cluster with energy =%f \n",input[i].energy());
128  mask[i] = true;
129  continue;
130  }
131 
132  //now business: if linked and not masked gather clusters
133  reco::PFCluster cluster = input[i];
134  mask[i] = true;
135  expandCluster(cluster, i, mask, input, prunedLinks);
136  _allCellsPosCalc->calculateAndSetPosition(cluster);
137  output.push_back(cluster);
138  // printf("Added linked cluster with energy =%f\n",cluster.energy());
139  }
140 }
void calculateShowerShapes(const reco::PFClusterCollection &, std::vector< double > &, std::vector< double > &)
Particle flow cluster, see clustering algorithm in PFClusterAlgo.
Definition: PFCluster.h:42
std::vector< ClusterLink > prune(std::vector< ClusterLink > &, std::vector< bool > &linkedClusters)
constexpr uint32_t mask
Definition: gpuClustering.h:26
static std::string const input
Definition: EdmProvDump.cc:50
std::unique_ptr< PFCPositionCalculatorBase > _allCellsPosCalc
std::vector< ClusterLink > link(const reco::PFClusterCollection &, const std::vector< double > &, const std::vector< double > &)
Definition: output.py:1
void expandCluster(reco::PFCluster &, unsigned int point, std::vector< bool > &mask, const reco::PFClusterCollection &, const std::vector< ClusterLink > &links)

◆ calculateShowerShapes()

void PFMultiDepthClusterizer::calculateShowerShapes ( const reco::PFClusterCollection clusters,
std::vector< double > &  etaRMS2,
std::vector< double > &  phiRMS2 
)
private

Definition at line 142 of file PFMultiDepthClusterizer.cc.

References funct::abs(), bsc_activity_cfg::clusters, SiPixelRawToDigiRegional_cfi::deltaPhi, reco::PFCluster::energy(), DivergingColor::frac, h, mps_fire::i, SiStripPI::max, reco::PFCluster::positionREP(), reco::PFCluster::recHitFractions(), and cuy::rep.

Referenced by buildClusters().

144  {
145  //shower shapes. here do not use the fractions
146 
147  for (unsigned int i = 0; i < clusters.size(); ++i) {
148  const reco::PFCluster& cluster = clusters[i];
149  double etaSum = 0.0;
150  double phiSum = 0.0;
151  auto const& crep = cluster.positionREP();
152  for (const auto& frac : cluster.recHitFractions()) {
153  auto const& h = *frac.recHitRef();
154  auto const& rep = h.positionREP();
155  etaSum += (frac.fraction() * h.energy()) * std::abs(rep.eta() - crep.eta());
156  phiSum += (frac.fraction() * h.energy()) * std::abs(deltaPhi(rep.phi(), crep.phi()));
157  }
158  //protection for single line : assign ~ tower
159  etaRMS2[i] = std::max(etaSum / cluster.energy(), 0.1);
160  etaRMS2[i] *= etaRMS2[i];
161  phiRMS2[i] = std::max(phiSum / cluster.energy(), 0.1);
162  phiRMS2[i] *= phiRMS2[i];
163  }
164 }
Particle flow cluster, see clustering algorithm in PFClusterAlgo.
Definition: PFCluster.h:42
const std::vector< reco::PFRecHitFraction > & recHitFractions() const
vector of rechit fractions
Definition: PFCluster.h:65
const REPPoint & positionREP() const
cluster position: rho, eta, phi
Definition: PFCluster.h:92
double energy() const
cluster energy
Definition: PFCluster.h:74
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
rep
Definition: cuy.py:1189
The Signals That Services Can Subscribe To This is based on ActivityRegistry h
Helper function to determine trigger accepts.
Definition: Activities.doc:4

◆ expandCluster()

void PFMultiDepthClusterizer::expandCluster ( reco::PFCluster cluster,
unsigned int  point,
std::vector< bool > &  mask,
const reco::PFClusterCollection clusters,
const std::vector< ClusterLink > &  links 
)
private

Definition at line 273 of file PFMultiDepthClusterizer.cc.

References absorbCluster(), bsc_activity_cfg::clusters, link(), electronStore::links, gpuClustering::pixelStatus::mask, and point.

Referenced by buildClusters().

277  {
278  for (const auto& link : links) {
279  if (link.from() == point) {
280  //found link that starts from this guy if not masked absorb
281  if (!mask[link.from()]) {
282  absorbCluster(cluster, clusters[link.from()]);
283  mask[link.from()] = true;
284  }
285 
286  if (!mask[link.to()]) {
287  absorbCluster(cluster, clusters[link.to()]);
288  mask[link.to()] = true;
289  expandCluster(cluster, link.to(), mask, clusters, links);
290  }
291  }
292  if (link.to() == point) {
293  //found link that starts from this guy if not masked absorb
294  if (!mask[link.to()]) {
295  absorbCluster(cluster, clusters[link.to()]);
296  mask[link.to()] = true;
297  }
298 
299  if (!mask[link.from()]) {
300  absorbCluster(cluster, clusters[link.from()]);
301  mask[link.from()] = true;
302  expandCluster(cluster, link.from(), mask, clusters, links);
303  }
304  }
305  }
306 }
constexpr uint32_t mask
Definition: gpuClustering.h:26
void absorbCluster(reco::PFCluster &, const reco::PFCluster &)
std::vector< ClusterLink > link(const reco::PFClusterCollection &, const std::vector< double > &, const std::vector< double > &)
*vegas h *****************************************************used in the default bin number in original ***version of VEGAS is ***a higher bin number might help to derive a more precise ***grade subtle point
Definition: invegas.h:5
void expandCluster(reco::PFCluster &, unsigned int point, std::vector< bool > &mask, const reco::PFClusterCollection &, const std::vector< ClusterLink > &links)

◆ link()

std::vector< PFMultiDepthClusterizer::ClusterLink > PFMultiDepthClusterizer::link ( const reco::PFClusterCollection clusters,
const std::vector< double > &  etaRMS2,
const std::vector< double > &  phiRMS2 
)
private

Definition at line 166 of file PFMultiDepthClusterizer.cc.

References funct::abs(), bsc_activity_cfg::clusters, SiPixelRawToDigiRegional_cfi::deltaPhi, reco::PFCluster::depth(), PVValHelper::dz, reco::PFCluster::energy(), mps_fire::i, dqmiolumiharvest::j, electronStore::links, nSigmaEta_, nSigmaPhi_, and reco::PFCluster::positionREP().

Referenced by buildClusters(), and expandCluster().

167  {
168  std::vector<ClusterLink> links;
169  //loop on all pairs
170  for (unsigned int i = 0; i < clusters.size(); ++i)
171  for (unsigned int j = 0; j < clusters.size(); ++j) {
172  if (i == j)
173  continue;
174 
175  const reco::PFCluster& cluster1 = clusters[i];
176  const reco::PFCluster& cluster2 = clusters[j];
177 
178  // PFCluster depth stored as double but HCAL layer clusters have integral depths only
179  auto dz = (static_cast<int>(cluster2.depth()) - static_cast<int>(cluster1.depth()));
180 
181  //Do not link at the same layer and only link inside out!
182  if (dz <= 0)
183  continue;
184 
185  auto const& crep1 = cluster1.positionREP();
186  auto const& crep2 = cluster2.positionREP();
187 
188  auto deta = crep1.eta() - crep2.eta();
189  deta = deta * deta / (etaRMS2[i] + etaRMS2[j]);
190  auto dphi = deltaPhi(crep1.phi(), crep2.phi());
191  dphi = dphi * dphi / (phiRMS2[i] + phiRMS2[j]);
192 
193  // printf("Testing Link %d -> %d (%f %f %f %f ) \n",i,j,deta,dphi,cluster1.position().Eta()-cluster2.position().Eta(),deltaPhi(cluster1.position().Phi(),cluster2.position().Phi()));
194 
195  if ((deta < nSigmaEta_) & (dphi < nSigmaPhi_))
196  links.push_back(ClusterLink(i, j, deta + dphi, std::abs(dz), cluster1.energy() + cluster2.energy()));
197  }
198 
199  return links;
200 }
Particle flow cluster, see clustering algorithm in PFClusterAlgo.
Definition: PFCluster.h:42
const REPPoint & positionREP() const
cluster position: rho, eta, phi
Definition: PFCluster.h:92
double depth() const
cluster depth
Definition: PFCluster.h:82
double energy() const
cluster energy
Definition: PFCluster.h:74
Abs< T >::type abs(const T &t)
Definition: Abs.h:22

◆ operator=()

B2DGPF& PFMultiDepthClusterizer::operator= ( const B2DGPF )
delete

◆ prune()

std::vector< PFMultiDepthClusterizer::ClusterLink > PFMultiDepthClusterizer::prune ( std::vector< ClusterLink > &  links,
std::vector< bool > &  linkedClusters 
)
private

Definition at line 202 of file PFMultiDepthClusterizer.cc.

References PFMultiDepthClusterizer::ClusterLink::dR(), PFMultiDepthClusterizer::ClusterLink::dZ(), PFMultiDepthClusterizer::ClusterLink::energy(), mps_fire::i, dqmiolumiharvest::j, electronStore::links, gpuClustering::pixelStatus::mask, and PFMultiDepthClusterizer::ClusterLink::to().

Referenced by buildClusters(), and dirstructure.Directory::prune().

203  {
204  std::vector<ClusterLink> goodLinks;
205  std::vector<bool> mask(links.size(), false);
206  if (links.empty())
207  return goodLinks;
208 
209  for (unsigned int i = 0; i < links.size() - 1; ++i) {
210  if (mask[i])
211  continue;
212  for (unsigned int j = i + 1; j < links.size(); ++j) {
213  if (mask[j])
214  continue;
215 
216  const ClusterLink& link1 = links[i];
217  const ClusterLink& link2 = links[j];
218 
219  if (link1.to() == link2.to()) { //found two links going to the same spot,kill one
220  //first prefer nearby layers
221  if (link1.dZ() < link2.dZ()) {
222  mask[j] = true;
223  } else if (link1.dZ() > link2.dZ()) {
224  mask[i] = true;
225  } else { //if same layer-pick based on transverse compatibility
226  if (link1.dR() < link2.dR()) {
227  mask[j] = true;
228  } else if (link1.dR() > link2.dR()) {
229  mask[i] = true;
230  } else {
231  //same distance as well -> can happen !!!!! Pick the highest SUME
232  if (link1.energy() < link2.energy())
233  mask[i] = true;
234  else
235  mask[j] = true;
236  }
237  }
238  }
239  }
240  }
241 
242  for (unsigned int i = 0; i < links.size(); ++i) {
243  if (mask[i])
244  continue;
245  goodLinks.push_back(links[i]);
246  linkedClusters[links[i].from()] = true;
247  linkedClusters[links[i].to()] = true;
248  }
249 
250  return goodLinks;
251 }
constexpr uint32_t mask
Definition: gpuClustering.h:26

◆ update()

void PFMultiDepthClusterizer::update ( const edm::EventSetup es)
inlineoverridevirtual

Member Data Documentation

◆ _allCellsPosCalc

std::unique_ptr<PFCPositionCalculatorBase> PFMultiDepthClusterizer::_allCellsPosCalc
private

Definition at line 32 of file PFMultiDepthClusterizer.cc.

Referenced by buildClusters(), PFMultiDepthClusterizer(), and update().

◆ nSigmaEta_

double PFMultiDepthClusterizer::nSigmaEta_
private

Definition at line 33 of file PFMultiDepthClusterizer.cc.

Referenced by link(), and PFMultiDepthClusterizer().

◆ nSigmaPhi_

double PFMultiDepthClusterizer::nSigmaPhi_
private

Definition at line 34 of file PFMultiDepthClusterizer.cc.

Referenced by link(), and PFMultiDepthClusterizer().