CMS 3D CMS Logo
List of all members | Public Member Functions | Private Attributes
RBXAndHPDCleaner Class Reference

#include <RBXAndHPDCleaner.h>

Inheritance diagram for RBXAndHPDCleaner:
RecHitTopologicalCleanerBase

Public Member Functions

void clean (const edm::Handle< reco::PFRecHitCollection > &input, std::vector< bool > &mask) override
 
RBXAndHPDCleaner & operator= (const RBXAndHPDCleaner &)=delete
 
 RBXAndHPDCleaner (const edm::ParameterSet &conf)
 
 RBXAndHPDCleaner (const RBXAndHPDCleaner &)=delete
 
- Public Member Functions inherited from RecHitTopologicalCleanerBase
const std::string & name () const
 
RecHitTopologicalCleanerBase & operator= (const RecHitTopologicalCleanerBase &)=delete
 
 RecHitTopologicalCleanerBase (const edm::ParameterSet &conf)
 
 RecHitTopologicalCleanerBase (const RecHitTopologicalCleanerBase &)=delete
 
virtual ~RecHitTopologicalCleanerBase ()=default
 

Private Attributes

std::unordered_map< int, std::vector< unsigned > > _hpds
 
std::unordered_map< int, std::vector< unsigned > > _rbxs
 

Detailed Description

Definition at line 8 of file RBXAndHPDCleaner.h.

Constructor & Destructor Documentation

RBXAndHPDCleaner::RBXAndHPDCleaner ( const edm::ParameterSet &  conf)
inline

Definition at line 10 of file RBXAndHPDCleaner.h.

References clean(), input, and operator=().

10 : RecHitTopologicalCleanerBase(conf) {}
RecHitTopologicalCleanerBase::RecHitTopologicalCleanerBase
RecHitTopologicalCleanerBase(const edm::ParameterSet &conf)
Definition: RecHitTopologicalCleanerBase.h:13
RBXAndHPDCleaner::RBXAndHPDCleaner ( const RBXAndHPDCleaner &  )
delete

Member Function Documentation

void RBXAndHPDCleaner::clean ( const edm::Handle< reco::PFRecHitCollection > &  input,
std::vector< bool > &  mask 
)
overridevirtual

Implements RecHitTopologicalCleanerBase.

Definition at line 12 of file RBXAndHPDCleaner.cc.

References _hpds, _rbxs, funct::abs(), reco::PFRecHit::detId(), reco::PFRecHit::energy(), PFLayer::HCAL_BARREL1, PFLayer::HCAL_ENDCAP, mps_fire::i, training_settings::idx, LEDCalibrationChannels::ieta, LEDCalibrationChannels::iphi, dqmdumpme::k, reco::PFRecHit::layer(), reco::PFRecHit::neighbours4(), groupFilesInBlocks::nn, TrackInfoProducer_cfi::rechits, MessageLogger_cff::threshold, and heppy_batch::val.

Referenced by RBXAndHPDCleaner().

12  {
13  _hpds.clear();
14  _rbxs.clear();
15  //need to run over energy sorted rechits
16  std::vector<std::pair<unsigned, double> > ordered_hits;
17  ordered_hits.reserve(input->size());
18  for (unsigned i = 0; i < input->size(); ++i) {
19  std::pair<unsigned, double> val = std::make_pair(i, input->at(i).energy());
20  auto pos = std::upper_bound(ordered_hits.begin(), ordered_hits.end(), val, greaterByEnergy);
21  ordered_hits.insert(pos, val);
22  }
23 
24  for (const auto& idx_e : ordered_hits) {
25  if (!mask[idx_e.first])
26  continue;
27  const unsigned idx = idx_e.first;
28  const reco::PFRecHit& rechit = input->at(idx);
29  int layer = rechit.layer();
30  if (layer != PFLayer::HCAL_BARREL1 && layer != PFLayer::HCAL_ENDCAP) {
31  break;
32  }
33  HcalDetId theHcalDetId(rechit.detId());
34  int ieta = theHcalDetId.ieta();
35  int iphi = theHcalDetId.iphi();
36  int ihpd = 0, irbx = 0;
37  switch (layer) {
38  case PFLayer::HCAL_BARREL1:
39  ihpd = (ieta < 0 ? -iphi : iphi);
40  irbx = (ieta < 0 ? -(iphi + 5) / 4 : (iphi + 5) / 4);
41  break;
42  case PFLayer::HCAL_ENDCAP:
43  ihpd = (ieta < 0 ? -(iphi + 1) / 2 - 100 : (iphi + 1) / 2 + 100);
44  irbx = (ieta < 0 ? -(iphi + 5) / 4 - 20 : (iphi + 5) / 4 + 20);
45  break;
46  default:
47  break;
48  }
49  switch (std::abs(irbx)) {
50  case 19:
51  irbx = (irbx < 0 ? -1 : 1);
52  break;
53  case 39:
54  irbx = (irbx < 0 ? -21 : 21);
55  break;
56  default:
57  break;
58  }
59  _hpds[ihpd].push_back(idx);
60  _rbxs[irbx].push_back(idx);
61  }
62  // loop on the rbx's we found and clean RBX's with tons of rechits
63  // and lots of energy
64  std::unordered_map<int, std::vector<unsigned> > theHPDs;
65  std::unordered_multimap<double, unsigned> theEnergies;
66  for (const auto& itrbx : _rbxs) {
67  if ((std::abs(itrbx.first) < 20 && itrbx.second.size() > 30) ||
68  (std::abs(itrbx.first) > 20 && itrbx.second.size() > 30)) {
69  const std::vector<unsigned>& rechits = itrbx.second;
70  theHPDs.clear();
71  theEnergies.clear();
72  int nSeeds0 = rechits.size();
73  for (unsigned jh = 0; jh < rechits.size(); ++jh) {
74  const reco::PFRecHit& rechit = (*input)[jh];
75  // check if rechit is a seed
76  unsigned nN = 0; // neighbours over threshold
77  bool isASeed = true;
78  auto const& neighbours4 = rechit.neighbours4();
79  for (auto k : neighbours4) {
80  auto const& neighbour = (*input)[k];
81  if (neighbour.energy() > rechit.energy()) {
82  --nSeeds0;
83  isASeed = false;
84  break;
85  } else {
86  if (neighbour.energy() > 0.4)
87  ++nN;
88  }
89  }
90  if (isASeed && !nN)
91  --nSeeds0;
92 
93  HcalDetId theHcalDetId(rechit.detId());
94  int iphi = theHcalDetId.iphi();
95  switch (rechit.layer()) {
96  case PFLayer::HCAL_BARREL1:
97  theHPDs[iphi].push_back(rechits[jh]);
98  break;
99  case PFLayer::HCAL_ENDCAP:
100  theHPDs[(iphi - 1) / 2].push_back(rechits[jh]);
101  break;
102  default:
103  break;
104  }
105  const double rhenergy = rechit.energy();
106  theEnergies.emplace(rhenergy, rechits[jh]);
107  }
108  if (nSeeds0 > 6) {
109  unsigned nHPD15 = 0;
110  for (const auto& itHPD : theHPDs) {
111  int hpdN = itHPD.first;
112  const std::vector<unsigned>& hpdHits = itHPD.second;
113  if ((std::abs(hpdN) < 100 && hpdHits.size() > 14) || (std::abs(hpdN) > 100 && hpdHits.size() > 14))
114  ++nHPD15;
115  }
116  if (nHPD15 > 1) {
117  unsigned nn = 0;
118  double threshold = 1.0;
119  for (const auto& itEn : theEnergies) {
120  ++nn;
121  if (nn < 5) {
122  mask[itEn.second] = false;
123  } else if (nn == 5) {
124  threshold = itEn.first * 5;
125  mask[itEn.second] = false;
126  } else {
127  if (itEn.first < threshold)
128  mask[itEn.second] = false;
129  }
130  }
131  }
132  }
133  }
134  }
135 
136  // loop on the HPDs
137  std::unordered_map<int, std::vector<unsigned> >::iterator neighbour1;
138  std::unordered_map<int, std::vector<unsigned> >::iterator neighbour2;
139  std::unordered_map<int, std::vector<unsigned> >::iterator neighbour0;
140  std::unordered_map<int, std::vector<unsigned> >::iterator neighbour3;
141  unsigned size1 = 0, size2 = 0;
142  for (const auto& ithpd : _hpds) {
143  const std::vector<unsigned>& rechits = ithpd.second;
144  theEnergies.clear();
145  for (const unsigned rhidx : rechits) {
146  const reco::PFRecHit& rechit = input->at(rhidx);
147  theEnergies.emplace(rechit.energy(), rhidx);
148  }
149 
150  const int thehpd = ithpd.first;
151  switch (std::abs(thehpd)) {
152  case 1:
153  neighbour1 = (thehpd > 0 ? _hpds.find(72) : _hpds.find(-72));
154  break;
155  case 72:
156  neighbour2 = (thehpd > 0 ? _hpds.find(1) : _hpds.find(-1));
157  break;
158  case 101:
159  neighbour1 = (thehpd > 0 ? _hpds.find(136) : _hpds.find(-136));
160  break;
161  case 136:
162  neighbour2 = (thehpd > 0 ? _hpds.find(101) : _hpds.find(-101));
163  break;
164  default:
165  neighbour1 = (thehpd > 0 ? _hpds.find(thehpd - 1) : _hpds.find(thehpd + 1));
166  neighbour2 = (thehpd > 0 ? _hpds.find(thehpd + 1) : _hpds.find(thehpd - 1));
167  break;
168  }
169  if (neighbour1 != _hpds.end()) {
170  const int nb1 = neighbour1->first;
171  switch (std::abs(nb1)) {
172  case 1:
173  neighbour0 = (nb1 > 0 ? _hpds.find(72) : _hpds.find(-72));
174  break;
175  case 101:
176  neighbour0 = (nb1 > 0 ? _hpds.find(136) : _hpds.find(-136));
177  break;
178  default:
179  neighbour0 = (nb1 > 0 ? _hpds.find(nb1 - 1) : _hpds.find(nb1 + 1));
180  break;
181  }
182  } else {
183  neighbour0 = _hpds.end();
184  }
185 
186  if (neighbour2 != _hpds.end()) {
187  const int nb2 = neighbour2->first;
188  switch (std::abs(nb2)) {
189  case 72:
190  neighbour3 = (nb2 > 0 ? _hpds.find(1) : _hpds.find(-1));
191  break;
192  case 136:
193  neighbour3 = (nb2 > 0 ? _hpds.find(101) : _hpds.find(-101));
194  break;
195  default:
196  neighbour3 = (nb2 > 0 ? _hpds.find(nb2 + 1) : _hpds.find(nb2 - 1));
197  break;
198  }
199  } else {
200  neighbour3 = _hpds.end();
201  }
202 
203  size1 = neighbour1 != _hpds.end() ? neighbour1->second.size() : 0;
204  size2 = neighbour2 != _hpds.end() ? neighbour2->second.size() : 0;
205  if (size1 > 10) {
206  if ((abs(neighbour1->first) > 100 && neighbour1->second.size() > 15) ||
207  (abs(neighbour1->first) < 100 && neighbour1->second.size() > 12))
208  size1 = neighbour0 != _hpds.end() ? neighbour0->second.size() : 0;
209  }
210  if (size2 > 10) {
211  if ((abs(neighbour2->first) > 100 && neighbour2->second.size() > 15) ||
212  (abs(neighbour2->first) < 100 && neighbour2->second.size() > 12))
213  size2 = neighbour3 != _hpds.end() ? neighbour3->second.size() : 0;
214  }
215  if ((std::abs(ithpd.first) > 100 && ithpd.second.size() > 15) ||
216  (std::abs(ithpd.first) < 100 && ithpd.second.size() > 12)) {
217  if ((double)(size1 + size2) / (float)ithpd.second.size() < 1.0) {
218  unsigned nn = 0;
219  double threshold = 1.0;
220  for (const auto& itEn : theEnergies) {
221  if (nn < 5) {
222  mask[itEn.second] = false;
223  } else if (nn == 5) {
224  threshold = itEn.first * 2.5;
225  mask[itEn.second] = false;
226  } else {
227  if (itEn.first < threshold)
228  mask[itEn.second] = false;
229  }
230  }
231  }
232  }
233  }
234 }
RBXAndHPDCleaner::_rbxs
std::unordered_map< int, std::vector< unsigned > > _rbxs
Definition: RBXAndHPDCleaner.h:17
mps_fire.i
i
Definition: mps_fire.py:341
reco::PFRecHit::detId
unsigned detId() const
rechit detId
Definition: PFRecHit.h:89
MessageLogger_cff.threshold
threshold
Definition: MessageLogger_cff.py:9
reco::PFRecHit::layer
PFLayer::Layer layer() const
rechit layer
Definition: PFRecHit.h:92
reco::PFRecHit
Particle flow rechit (rechit + geometry and topology information). See clustering algorithm in PFClus...
Definition: PFRecHit.h:31
funct::abs
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
TrackInfoProducer_cfi.rechits
rechits
Definition: TrackInfoProducer_cfi.py:9
reco::PFRecHit::energy
float energy() const
rechit energy
Definition: PFRecHit.h:95
HcalDetId::iphi
int iphi() const
get the cell iphi
Definition: HcalDetId.h:157
training_settings.idx
idx
Definition: training_settings.py:16
RBXAndHPDCleaner::_hpds
std::unordered_map< int, std::vector< unsigned > > _hpds
Definition: RBXAndHPDCleaner.h:17
heppy_batch.val
val
Definition: heppy_batch.py:351
reco::PFRecHit::neighbours4
Neighbours neighbours4() const
Definition: PFRecHit.h:77
RBXAndHPDCleaner& RBXAndHPDCleaner::operator= ( const RBXAndHPDCleaner &  )
delete

Referenced by RBXAndHPDCleaner().

Member Data Documentation

std::unordered_map<int, std::vector<unsigned> > RBXAndHPDCleaner::_hpds
private

Definition at line 17 of file RBXAndHPDCleaner.h.

Referenced by clean().

std::unordered_map<int, std::vector<unsigned> > RBXAndHPDCleaner::_rbxs
private

Definition at line 17 of file RBXAndHPDCleaner.h.

Referenced by clean().

Generated for CMSSW Reference Manual by doxygen 1.8.11