CMS 3D CMS Logo

All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Modules Pages
EgammaHLTNxNClusterProducer.cc
Go to the documentation of this file.
1 
15 
20 
23 
24 #include <vector>
25 #include <memory>
26 #include <ctime>
27 
34 
43 
44 #include "TVector3.h"
45 
47 public:
50 
51  void produce(edm::Event &, const edm::EventSetup &) override;
52  static void fillDescriptions(edm::ConfigurationDescriptions &descriptions);
53 
54 private:
55  void makeNxNClusters(edm::Event &evt,
56  const edm::EventSetup &es,
59 
60  bool checkStatusOfEcalRecHit(const EcalChannelStatus &channelStatus, const EcalRecHit &rh);
61 
62  //std::map<std::string,double> providedParameters;
63 
64  const bool doBarrel_;
65  const bool doEndcaps_;
68  const int clusEtaSize_;
69  const int clusPhiSize_;
72  const double clusSeedThr_;
73  const double clusSeedThrEndCap_;
74 
75  const bool useRecoFlag_;
77  const bool useDBStatus_;
79 
80  const int maxNumberofSeeds_;
82 
83  const int debug_;
84 
85  PositionCalc posCalculator_; // position calculation algorithm
86 };
87 
89  : doBarrel_(ps.getParameter<bool>("doBarrel")),
90  doEndcaps_(ps.getParameter<bool>("doEndcaps")),
91  barrelHitProducer_(consumes<EcalRecHitCollection>(ps.getParameter<edm::InputTag>("barrelHitProducer"))),
92  endcapHitProducer_(consumes<EcalRecHitCollection>(ps.getParameter<edm::InputTag>("endcapHitProducer"))),
93  clusEtaSize_(ps.getParameter<int>("clusEtaSize")),
94  clusPhiSize_(ps.getParameter<int>("clusPhiSize")),
95  barrelClusterCollection_(ps.getParameter<std::string>("barrelClusterCollection")),
96  endcapClusterCollection_(ps.getParameter<std::string>("endcapClusterCollection")),
97  clusSeedThr_(ps.getParameter<double>("clusSeedThr")),
98  clusSeedThrEndCap_(ps.getParameter<double>("clusSeedThrEndCap")),
99  useRecoFlag_(ps.getParameter<bool>("useRecoFlag")),
100  flagLevelRecHitsToUse_(ps.getParameter<int>("flagLevelRecHitsToUse")),
101  useDBStatus_(ps.getParameter<bool>("useDBStatus")),
102  statusLevelRecHitsToUse_(ps.getParameter<int>("statusLevelRecHitsToUse")),
103  maxNumberofSeeds_(ps.getParameter<int>("maxNumberofSeeds")),
104  maxNumberofClusters_(ps.getParameter<int>("maxNumberofClusters")),
105  debug_(ps.getParameter<int>("debugLevel")),
106  posCalculator_(PositionCalc(ps.getParameter<edm::ParameterSet>("posCalcParameters"))) {
107  produces<reco::BasicClusterCollection>(barrelClusterCollection_);
108  produces<reco::BasicClusterCollection>(endcapClusterCollection_);
109 }
110 
112 
115  desc.add<bool>(("doBarrel"), true);
116  desc.add<bool>(("doEndcaps"), true);
117  desc.add<edm::InputTag>(("barrelHitProducer"), edm::InputTag("hltEcalRegionalPi0EtaRecHit", "EcalRecHitsEB"));
118  desc.add<edm::InputTag>(("endcapHitProducer"), edm::InputTag("hltEcalRegionalPi0EtaRecHit", "EcalRecHitsEE"));
119  desc.add<int>(("clusEtaSize"), 3);
120  desc.add<int>(("clusPhiSize"), 3);
121  desc.add<std::string>(("barrelClusterCollection"), "Simple3x3ClustersBarrel");
122  desc.add<std::string>(("endcapClusterCollection"), "Simple3x3ClustersEndcap");
123  desc.add<double>(("clusSeedThr"), 0.5);
124  desc.add<double>(("clusSeedThrEndCap"), 1.0);
125  desc.add<bool>(("useRecoFlag"), false);
126  desc.add<int>(("flagLevelRecHitsToUse"), 1);
127  desc.add<bool>(("useDBStatus"), true);
128  desc.add<int>(("statusLevelRecHitsToUse"), 1);
129 
130  edm::ParameterSetDescription posCalcPSET;
131  posCalcPSET.add<double>("T0_barl", 7.4);
132  posCalcPSET.add<double>("T0_endc", 3.1);
133  posCalcPSET.add<double>("T0_endcPresh", 1.2);
134  posCalcPSET.add<double>("W0", 4.2);
135  posCalcPSET.add<double>("X0", 0.89);
136  posCalcPSET.add<bool>("LogWeighted", true);
137  desc.add<edm::ParameterSetDescription>("posCalcParameters", posCalcPSET);
138 
139  desc.add<int>(("maxNumberofSeeds"), 1000);
140  desc.add<int>(("maxNumberofClusters"), 200);
141  desc.add<int>(("debugLevel"), 0);
142  descriptions.add(("hltEgammaHLTNxNClusterProducer"), desc);
143 }
144 
146  if (doBarrel_) {
147  edm::Handle<EcalRecHitCollection> barrelRecHitsHandle;
148  evt.getByToken(barrelHitProducer_, barrelRecHitsHandle);
149  if (!barrelRecHitsHandle.isValid()) {
150  LogDebug("") << "EgammaHLTNxNClusterProducer Error! can't get product eb hit!" << std::endl;
151  }
152 
153  const EcalRecHitCollection *hits_eb = barrelRecHitsHandle.product();
154  if (debug_ >= 2)
155  LogDebug("") << "EgammaHLTNxNClusterProducer nEBrechits: " << evt.id().run() << " event " << evt.id().event()
156  << " " << hits_eb->size() << std::endl;
157 
159  }
160 
161  if (doEndcaps_) {
162  edm::Handle<EcalRecHitCollection> endcapRecHitsHandle;
163  evt.getByToken(endcapHitProducer_, endcapRecHitsHandle);
164  if (!endcapRecHitsHandle.isValid()) {
165  LogDebug("") << "EgammaHLTNxNClusterProducer Error! can't get product ee hit!" << std::endl;
166  }
167 
168  const EcalRecHitCollection *hits_ee = endcapRecHitsHandle.product();
169  if (debug_ >= 2)
170  LogDebug("") << "EgammaHLTNxNClusterProducer nEErechits: " << evt.id().run() << " event " << evt.id().event()
171  << " " << hits_ee->size() << std::endl;
173  }
174 }
175 
177  const EcalRecHit &rh) {
178  if (useRecoFlag_) {
179  int flag = rh.recoFlag();
180  if (flagLevelRecHitsToUse_ == 0) {
181  if (flag != 0)
182  return false;
183  } else if (flagLevelRecHitsToUse_ == 1) {
184  if (flag != 0 && flag != 4)
185  return false;
186  } else if (flagLevelRecHitsToUse_ == 2) {
187  if (flag != 0 && flag != 4 && flag != 6 && flag != 7)
188  return false;
189  }
190  }
191  if (useDBStatus_) {
192  int status = int(channelStatus[rh.id().rawId()].getStatusCode());
194  return false;
195  }
196 
197  return true;
198 }
199 
201  const edm::EventSetup &es,
202  const EcalRecHitCollection *hits,
206  if (useDBStatus_)
207  es.get<EcalChannelStatusRcd>().get(csHandle);
208  const EcalChannelStatus &channelStatus = *csHandle;
209 
210  std::vector<EcalRecHit> seeds;
211 
212  double clusterSeedThreshold;
214  clusterSeedThreshold = clusSeedThr_;
215  } else {
216  clusterSeedThreshold = clusSeedThrEndCap_;
217  }
218 
219  for (EcalRecHitCollection::const_iterator itt = hits->begin(); itt != hits->end(); itt++) {
220  double energy = itt->energy();
221  if (!checkStatusOfEcalRecHit(channelStatus, *itt))
222  continue;
223  if (energy > clusterSeedThreshold)
224  seeds.push_back(*itt);
225 
226  if (int(seeds.size()) > maxNumberofSeeds_) { //too many seeds, like beam splash events
227  seeds.clear();
228  break;
229  }
230  }
231 
232  // get the geometry and topology from the event setup:
233  edm::ESHandle<CaloGeometry> geoHandle;
234  es.get<CaloGeometryRecord>().get(geoHandle);
235 
236  const CaloSubdetectorGeometry *geometry_p;
237  std::unique_ptr<CaloSubdetectorTopology> topology_p;
239  geometry_p = geoHandle->getSubdetectorGeometry(DetId::Ecal, EcalBarrel);
240  topology_p = std::make_unique<EcalBarrelTopology>(*geoHandle);
241  } else {
242  geometry_p = geoHandle->getSubdetectorGeometry(DetId::Ecal, EcalEndcap);
243  topology_p = std::make_unique<EcalEndcapTopology>(*geoHandle);
244  }
245 
246  const CaloSubdetectorGeometry *geometryES_p;
247  geometryES_p = geoHandle->getSubdetectorGeometry(DetId::Ecal, EcalPreshower);
248 
249  std::vector<reco::BasicCluster> clusters;
250  std::vector<DetId> usedXtals;
251 
252  // sort seed according to Energy
253  sort(seeds.begin(), seeds.end(), [](auto const &x, auto const &y) { return (x.energy() > y.energy()); });
254 
255  for (std::vector<EcalRecHit>::iterator itseed = seeds.begin(); itseed != seeds.end(); itseed++) {
256  DetId seed_id = itseed->id();
257 
258  std::vector<DetId>::iterator itdet = find(usedXtals.begin(), usedXtals.end(), seed_id);
259  if (itdet != usedXtals.end())
260  continue;
261 
262  std::vector<DetId> clus_v = topology_p->getWindow(seed_id, clusEtaSize_, clusPhiSize_);
263  std::vector<std::pair<DetId, float> > clus_used;
264 
265  float clus_energy = 0;
266 
267  for (std::vector<DetId>::iterator det = clus_v.begin(); det != clus_v.end(); det++) {
268  DetId detid = *det;
269 
270  //not yet used
271  std::vector<DetId>::iterator itdet = find(usedXtals.begin(), usedXtals.end(), detid);
272  if (itdet != usedXtals.end())
273  continue;
274  //inside the collection
276  if (hit == hits->end())
277  continue;
278 
279  if (!checkStatusOfEcalRecHit(channelStatus, *hit))
280  continue;
281 
282  usedXtals.push_back(detid);
283  clus_used.push_back(std::pair<DetId, float>(detid, 1.));
284  clus_energy += hit->energy();
285 
286  }
287 
288  if (clus_energy <= 0)
289  continue;
290 
291  math::XYZPoint clus_pos = posCalculator_.Calculate_Location(clus_used, hits, geometry_p, geometryES_p);
292 
293  if (debug_ >= 2)
294  LogDebug("") << "nxn_cluster in run " << evt.id().run() << " event " << evt.id().event()
295  << " energy: " << clus_energy << " eta: " << clus_pos.Eta() << " phi: " << clus_pos.Phi()
296  << " nRecHits: " << clus_used.size() << std::endl;
297 
298  clusters.push_back(reco::BasicCluster(
299  clus_energy, clus_pos, reco::CaloID(detector), clus_used, reco::CaloCluster::island, seed_id));
300  if (int(clusters.size()) > maxNumberofClusters_) {
301  clusters.clear();
302  break;
303  }
304  }
305 
306  //Create empty output collections
307  auto clusters_p = std::make_unique<reco::BasicClusterCollection>();
308  clusters_p->assign(clusters.begin(), clusters.end());
310  if (debug_ >= 1)
311  LogDebug("") << "nxnclusterProducer: " << clusters_p->size() << " made in barrel" << std::endl;
312  evt.put(std::move(clusters_p), barrelClusterCollection_);
313  } else {
314  if (debug_ >= 1)
315  LogDebug("") << "nxnclusterProducer: " << clusters_p->size() << " made in endcap" << std::endl;
316  evt.put(std::move(clusters_p), endcapClusterCollection_);
317  }
318 }
319 
ConfigurationDescriptions.h
DDAxes::y
EcalRecHit
Definition: EcalRecHit.h:15
electrons_cff.bool
bool
Definition: electrons_cff.py:372
PositionCalc.h
edm::ParameterSetDescription::add
ParameterDescriptionBase * add(U const &iLabel, T const &value)
Definition: ParameterSetDescription.h:95
edm::SortedCollection< EcalRecHit >::const_iterator
std::vector< EcalRecHit >::const_iterator const_iterator
Definition: SortedCollection.h:80
MessageLogger.h
edm::Handle::product
T const * product() const
Definition: Handle.h:70
hfClusterShapes_cfi.hits
hits
Definition: hfClusterShapes_cfi.py:5
EcalRecHit::id
DetId id() const
get the id
Definition: EcalRecHit.h:77
reco::CaloID::Detectors
Detectors
Definition: CaloID.h:19
EgammaHLTNxNClusterProducer::barrelHitProducer_
const edm::EDGetTokenT< EcalRecHitCollection > barrelHitProducer_
Definition: EgammaHLTNxNClusterProducer.cc:66
mps_update.status
status
Definition: mps_update.py:69
BasicCluster.h
edm::EDGetTokenT
Definition: EDGetToken.h:33
CaloGeometryRecord
Definition: CaloGeometryRecord.h:30
reco::CaloID::DET_ECAL_ENDCAP
Definition: CaloID.h:21
edm
HLT enums.
Definition: AlignableModifier.h:19
EgammaHLTNxNClusterProducer::clusEtaSize_
const int clusEtaSize_
Definition: EgammaHLTNxNClusterProducer.cc:68
EBDetId.h
edm::ParameterSetDescription
Definition: ParameterSetDescription.h:52
EcalBarrelTopology.h
CaloGeometry::getSubdetectorGeometry
const CaloSubdetectorGeometry * getSubdetectorGeometry(const DetId &id) const
access the subdetector geometry for the given subdetector directly
Definition: CaloGeometry.cc:34
CaloID.h
EDProducer.h
edm::SortedCollection< EcalRecHit >
edm::SortedCollection::size
size_type size() const
Definition: SortedCollection.h:215
EgammaHLTNxNClusterProducer::clusSeedThr_
const double clusSeedThr_
Definition: EgammaHLTNxNClusterProducer.cc:72
EgammaHLTNxNClusterProducer::doEndcaps_
const bool doEndcaps_
Definition: EgammaHLTNxNClusterProducer.cc:65
DDAxes::x
EgammaHLTNxNClusterProducer::flagLevelRecHitsToUse_
const int flagLevelRecHitsToUse_
Definition: EgammaHLTNxNClusterProducer.cc:76
spr::find
void find(edm::Handle< EcalRecHitCollection > &hits, DetId thisDet, std::vector< EcalRecHitCollection::const_iterator > &hit, bool debug=false)
Definition: FindCaloHit.cc:19
EcalCondObjectContainer< EcalChannelStatusCode >
edm::Handle
Definition: AssociativeIterator.h:50
EcalBarrel
Definition: EcalSubdetector.h:10
EcalRecHitCollections.h
BasicClusterFwd.h
EgammaHLTNxNClusterProducer::doBarrel_
const bool doBarrel_
Definition: EgammaHLTNxNClusterProducer.cc:64
EgammaHLTNxNClusterProducer::produce
void produce(edm::Event &, const edm::EventSetup &) override
Definition: EgammaHLTNxNClusterProducer.cc:145
EgammaHLTNxNClusterProducer::clusPhiSize_
const int clusPhiSize_
Definition: EgammaHLTNxNClusterProducer.cc:69
EgammaHLTNxNClusterProducer
Definition: EgammaHLTNxNClusterProducer.cc:46
EgammaHLTNxNClusterProducer::maxNumberofSeeds_
const int maxNumberofSeeds_
Definition: EgammaHLTNxNClusterProducer.cc:80
DetId
Definition: DetId.h:17
MakerMacros.h
edm::EventSetup::get
T get() const
Definition: EventSetup.h:73
DEFINE_FWK_MODULE
#define DEFINE_FWK_MODULE(type)
Definition: MakerMacros.h:16
EgammaHLTNxNClusterProducer::checkStatusOfEcalRecHit
bool checkStatusOfEcalRecHit(const EcalChannelStatus &channelStatus, const EcalRecHit &rh)
Definition: EgammaHLTNxNClusterProducer.cc:176
edm::ConfigurationDescriptions::add
void add(std::string const &label, ParameterSetDescription const &psetDescription)
Definition: ConfigurationDescriptions.cc:57
EgammaHLTNxNClusterProducer::useDBStatus_
const bool useDBStatus_
Definition: EgammaHLTNxNClusterProducer.cc:77
PositionCalc::Calculate_Location
math::XYZPoint Calculate_Location(const HitsAndFractions &iDetIds, const edm::SortedCollection< HitType > *iRecHits, const CaloSubdetectorGeometry *iSubGeom, const CaloSubdetectorGeometry *iESGeom=nullptr)
Definition: PositionCalc.h:65
reco::CaloCluster
Definition: CaloCluster.h:31
EgammaHLTNxNClusterProducer::debug_
const int debug_
Definition: EgammaHLTNxNClusterProducer.cc:83
edm::ESHandle
Definition: DTSurvey.h:22
HCALHighEnergyHPDFilter_cfi.energy
energy
Definition: HCALHighEnergyHPDFilter_cfi.py:5
edm::Event::getByToken
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:528
ParameterSetDescription.h
edm::EventID::run
RunNumber_t run() const
Definition: EventID.h:38
CaloGeometryRecord.h
edm::ConfigurationDescriptions
Definition: ConfigurationDescriptions.h:28
AlCaHLTBitMon_QueryRunRegistry.string
string
Definition: AlCaHLTBitMon_QueryRunRegistry.py:256
EcalEndcap
Definition: EcalSubdetector.h:10
bsc_activity_cfg.clusters
clusters
Definition: bsc_activity_cfg.py:36
InitialStep_cff.seeds
seeds
Definition: InitialStep_cff.py:232
CaloSubdetectorGeometry.h
EgammaHLTNxNClusterProducer::fillDescriptions
static void fillDescriptions(edm::ConfigurationDescriptions &descriptions)
Definition: EgammaHLTNxNClusterProducer.cc:113
HLT_2018_cff.InputTag
InputTag
Definition: HLT_2018_cff.py:79016
LogDebug
#define LogDebug(id)
Definition: MessageLogger.h:670
edm::ParameterSet
Definition: ParameterSet.h:36
math::XYZPoint
XYZPointD XYZPoint
point in space with cartesian internal representation
Definition: Point3D.h:12
Event.h
ParameterSet
Definition: Functions.h:16
EcalChannelStatusRcd
Definition: EcalChannelStatusRcd.h:5
EcalEndcapTopology.h
CaloTopologyRecord.h
PositionCalc
Definition: PositionCalc.h:29
edm::EventID::event
EventNumber_t event() const
Definition: EventID.h:40
createfilelist.int
int
Definition: createfilelist.py:10
EgammaHLTNxNClusterProducer::endcapClusterCollection_
const std::string endcapClusterCollection_
Definition: EgammaHLTNxNClusterProducer.cc:71
edm::Event::put
OrphanHandle< PROD > put(std::unique_ptr< PROD > product)
Put a new product.
Definition: Event.h:132
edm::stream::EDProducer
Definition: EDProducer.h:38
EcalRecHit::recoFlag
Flags recoFlag() const
DEPRECATED provided for temporary backward compatibility.
Definition: EcalRecHit.h:207
edm::EventSetup
Definition: EventSetup.h:57
DetId::Ecal
Definition: DetId.h:27
get
#define get
EgammaHLTNxNClusterProducer::endcapHitProducer_
const edm::EDGetTokenT< EcalRecHitCollection > endcapHitProducer_
Definition: EgammaHLTNxNClusterProducer.cc:67
reco::CaloID
Definition: CaloID.h:17
reco::CaloCluster::island
Definition: CaloCluster.h:34
EcalRecHit.h
CaloTopology.h
EcalPreshower
Definition: EcalSubdetector.h:10
EgammaHLTNxNClusterProducer::~EgammaHLTNxNClusterProducer
~EgammaHLTNxNClusterProducer() override
Definition: EgammaHLTNxNClusterProducer.cc:111
CaloCellGeometry.h
eostools.move
def move(src, dest)
Definition: eostools.py:511
std
Definition: JetResolutionObject.h:76
DetId::rawId
constexpr uint32_t rawId() const
get the raw id
Definition: DetId.h:57
EgammaHLTNxNClusterProducer::clusSeedThrEndCap_
const double clusSeedThrEndCap_
Definition: EgammaHLTNxNClusterProducer.cc:73
Frameworkfwd.h
EgammaHLTNxNClusterProducer::useRecoFlag_
const bool useRecoFlag_
Definition: EgammaHLTNxNClusterProducer.cc:75
CaloGeometry.h
edm::EventBase::id
edm::EventID id() const
Definition: EventBase.h:59
EgammaHLTNxNClusterProducer::statusLevelRecHitsToUse_
const int statusLevelRecHitsToUse_
Definition: EgammaHLTNxNClusterProducer.cc:78
EgammaHLTNxNClusterProducer::makeNxNClusters
void makeNxNClusters(edm::Event &evt, const edm::EventSetup &es, const EcalRecHitCollection *hits, const reco::CaloID::Detectors detector)
Definition: EgammaHLTNxNClusterProducer.cc:200
EgammaHLTNxNClusterProducer::barrelClusterCollection_
const std::string barrelClusterCollection_
Definition: EgammaHLTNxNClusterProducer.cc:70
EventSetup.h
CaloSubdetectorGeometry
Definition: CaloSubdetectorGeometry.h:22
EgammaHLTNxNClusterProducer::posCalculator_
PositionCalc posCalculator_
Definition: EgammaHLTNxNClusterProducer.cc:85
hgcalTestNeighbor_cfi.detector
detector
Definition: hgcalTestNeighbor_cfi.py:6
ParameterSet.h
edm::HandleBase::isValid
bool isValid() const
Definition: HandleBase.h:70
edm::Event
Definition: Event.h:73
EgammaHLTNxNClusterProducer::EgammaHLTNxNClusterProducer
EgammaHLTNxNClusterProducer(const edm::ParameterSet &ps)
Definition: EgammaHLTNxNClusterProducer.cc:88
EcalChannelStatus.h
edm::InputTag
Definition: InputTag.h:15
BasicClusterShapeAssociation.h
EgammaHLTNxNClusterProducer::maxNumberofClusters_
const int maxNumberofClusters_
Definition: EgammaHLTNxNClusterProducer.cc:81
EcalChannelStatusRcd.h
hit
Definition: SiStripHitEffFromCalibTree.cc:88
reco::CaloID::DET_ECAL_BARREL
Definition: CaloID.h:20
RemoveAddSevLevel.flag
flag
Definition: RemoveAddSevLevel.py:116