CMS 3D CMS Logo

List of all members | Public Member Functions | Private Attributes
RealisticSimClusterMapper Class Reference
Inheritance diagram for RealisticSimClusterMapper:
InitialClusteringStepBase

Public Member Functions

void buildClusters (const edm::Handle< reco::PFRecHitCollection > &, const std::vector< bool > &, const std::vector< bool > &, reco::PFClusterCollection &, const HcalPFCuts *) override
 
RealisticSimClusterMapperoperator= (const RealisticSimClusterMapper &)=delete
 
 RealisticSimClusterMapper (const edm::ParameterSet &conf, edm::ConsumesCollector &cc)
 
 RealisticSimClusterMapper (const RealisticSimClusterMapper &)=delete
 
void update (const edm::EventSetup &) final
 
void updateEvent (const edm::Event &) final
 
 ~RealisticSimClusterMapper () override
 
- Public Member Functions inherited from InitialClusteringStepBase
 InitialClusteringStepBase (const edm::ParameterSet &conf, edm::ConsumesCollector &cc)
 
 InitialClusteringStepBase (const ICSB &)=delete
 
std::ostream & operator<< (std::ostream &o) const
 
ICSBoperator= (const ICSB &)=delete
 
void reset ()
 
virtual ~InitialClusteringStepBase ()=default
 

Private Attributes

const float calibMaxEta_ = 3.0
 
const float calibMinEta_ = 1.4
 
std::vector< double > egammaCalib_
 
const float exclusiveFraction_ = 0.7f
 
edm::ESGetToken< CaloGeometry, CaloGeometryRecordgeomToken_
 
std::vector< double > hadronCalib_
 
const float invisibleFraction_ = 0.3f
 
const float maxDforTimingSquared_ = 4.0f
 
const float maxDistance_ = 10.f
 
const bool maxDistanceFilter_ = false
 
const unsigned int minNHitsforTiming_ = 3
 
hgcal::RecHitTools rhtools_
 
edm::Handle< SimClusterCollectionsimClusterH_
 
edm::EDGetTokenT< SimClusterCollectionsimClusterToken_
 
const float timeOffset_
 
const bool useMCFractionsForExclEnergy_ = false
 

Additional Inherited Members

- Protected Types inherited from InitialClusteringStepBase
typedef std::tuple< std::vector< int >, std::vector< double >, std::vector< double > > I3tuple
 
- Protected Member Functions inherited from InitialClusteringStepBase
reco::PFRecHitRef makeRefhit (const edm::Handle< reco::PFRecHitCollection > &h, const unsigned i) const
 
- Protected Attributes inherited from InitialClusteringStepBase
const std::unordered_map< std::string, int > _layerMap
 
unsigned _nClustersFound
 
unsigned _nSeeds
 
std::unordered_map< int, I3tuple_thresholds
 

Detailed Description

Definition at line 24 of file RealisticSimClusterMapper.cc.

Constructor & Destructor Documentation

◆ RealisticSimClusterMapper() [1/2]

RealisticSimClusterMapper::RealisticSimClusterMapper ( const edm::ParameterSet conf,
edm::ConsumesCollector cc 
)
inline

Definition at line 26 of file RealisticSimClusterMapper.cc.

28  invisibleFraction_(conf.getParameter<double>("invisibleFraction")),
29  exclusiveFraction_(conf.getParameter<double>("exclusiveFraction")),
30  maxDistanceFilter_(conf.getParameter<bool>("maxDistanceFilter")),
31  maxDistance_(conf.getParameter<double>("maxDistance")),
32  maxDforTimingSquared_(conf.getParameter<double>("maxDforTimingSquared")),
33  timeOffset_(conf.getParameter<double>("timeOffset")),
34  minNHitsforTiming_(conf.getParameter<unsigned int>("minNHitsforTiming")),
35  useMCFractionsForExclEnergy_(conf.getParameter<bool>("useMCFractionsForExclEnergy")),
36  calibMinEta_(conf.getParameter<double>("calibMinEta")),
37  calibMaxEta_(conf.getParameter<double>("calibMaxEta")),
38  hadronCalib_(conf.getParameter<std::vector<double> >("hadronCalib")),
39  egammaCalib_(conf.getParameter<std::vector<double> >("egammaCalib")),
40  simClusterToken_(cc.consumes<SimClusterCollection>(conf.getParameter<edm::InputTag>("simClusterSrc"))),
41  geomToken_(cc.esConsumes<edm::Transition::BeginRun>()) {}
T getParameter(std::string const &) const
Definition: ParameterSet.h:307
uint32_t cc[maxCellsPerHit]
Definition: gpuFishbone.h:49
InitialClusteringStepBase(const edm::ParameterSet &conf, edm::ConsumesCollector &cc)
edm::EDGetTokenT< SimClusterCollection > simClusterToken_
std::vector< SimCluster > SimClusterCollection
edm::ESGetToken< CaloGeometry, CaloGeometryRecord > geomToken_

◆ ~RealisticSimClusterMapper()

RealisticSimClusterMapper::~RealisticSimClusterMapper ( )
inlineoverride

Definition at line 43 of file RealisticSimClusterMapper.cc.

43 {}

◆ RealisticSimClusterMapper() [2/2]

RealisticSimClusterMapper::RealisticSimClusterMapper ( const RealisticSimClusterMapper )
delete

Member Function Documentation

◆ buildClusters()

void RealisticSimClusterMapper::buildClusters ( const edm::Handle< reco::PFRecHitCollection > &  input,
const std::vector< bool > &  rechitMask,
const std::vector< bool > &  seedable,
reco::PFClusterCollection output,
const HcalPFCuts hcalCuts 
)
overridevirtual

Implements InitialClusteringStepBase.

Definition at line 110 of file RealisticSimClusterMapper.cc.

References funct::abs(), reco::PFCluster::addRecHitFraction(), calibMaxEta_, calibMinEta_, RealisticHitToClusterAssociator::computeAssociation(), egammaCalib_, relativeConstraints::empty, PVValHelper::eta, exclusiveFraction_, f, RealisticHitToClusterAssociator::filterHitsByDistance(), RealisticHitToClusterAssociator::findAndMergeInvisibleClusters(), RealisticHitToClusterAssociator::findCentersOfGravity(), dqmdumpme::first, hgcalsimclustertime::ComputeClusterTime::fixSizeHighestDensity(), DetId::Forward, ForwardEmpty, HLT_2024v14_cff::fraction, spr::getEnergy(), hgcal::RecHitTools::getGeometryType(), hgcal::RecHitTools::getLayer(), hgcal::RecHitTools::getLayerWithOffset(), hgcal::RecHitTools::getPosition(), hadronCalib_, hfClusterShapes_cfi::hits, reco::CaloCluster::hitsAndFractions(), mps_fire::i, RealisticHitToClusterAssociator::init(), input, RealisticHitToClusterAssociator::insertHitEnergy(), RealisticHitToClusterAssociator::insertHitPosition(), RealisticHitToClusterAssociator::insertLayerId(), RealisticHitToClusterAssociator::insertSimClusterIdAndFraction(), invisibleFraction_, hgcal::RecHitTools::lastLayerEE(), hgcal::RecHitTools::lastLayerFH(), InitialClusteringStepBase::makeRefhit(), maxDforTimingSquared_, maxDistance_, maxDistanceFilter_, heavyionUCCDQM_cfi::nClusters, phase1PixelTopology::numberOfLayers, or, EgammaValidation_cff::pdgId, funct::pow(), RealisticHitToClusterAssociator::realisticClusters(), rhtools_, fileCollector::seed, reco::CaloCluster::setCorrectedEnergy(), reco::CaloCluster::setEnergy(), reco::CaloCluster::setSeed(), reco::PFCluster::setTime(), simClusterH_, and useMCFractionsForExclEnergy_.

114  {
115  const SimClusterCollection& simClusters = *simClusterH_;
116  auto const& hits = *input;
117  RealisticHitToClusterAssociator realisticAssociator;
120  realisticAssociator.init(hits.size(), simClusters.size(), numberOfLayers + 1);
121  // for quick indexing back to hit energy
122  std::unordered_map<uint32_t, size_t> detIdToIndex(hits.size());
123  for (uint32_t i = 0; i < hits.size(); ++i) {
124  detIdToIndex[hits[i].detId()] = i;
125  auto ref = makeRefhit(input, i);
126  const auto& hitPos = rhtools_.getPosition(ref->detId());
127 
128  realisticAssociator.insertHitPosition(hitPos.x(), hitPos.y(), hitPos.z(), i);
129  realisticAssociator.insertHitEnergy(ref->energy(), i);
130  realisticAssociator.insertLayerId(rhtools_.getLayerWithOffset(ref->detId()), i);
131  }
132 
133  for (unsigned int ic = 0; ic < simClusters.size(); ++ic) {
134  const auto& sc = simClusters[ic];
135  const auto& hitsAndFractions = sc.hits_and_fractions();
136  for (const auto& hAndF : hitsAndFractions) {
137  auto itr = detIdToIndex.find(hAndF.first);
138  if (itr == detIdToIndex.end()) {
139  continue; // hit wasn't saved in reco or did not pass the SNR threshold
140  }
141  auto hitId = itr->second;
142  auto ref = makeRefhit(input, hitId);
143  float fraction = hAndF.second;
144  float associatedEnergy = fraction * ref->energy();
145  realisticAssociator.insertSimClusterIdAndFraction(ic, fraction, hitId, associatedEnergy);
146  }
147  }
148  realisticAssociator.computeAssociation(
151  realisticAssociator.findCentersOfGravity();
152  if (maxDistanceFilter_)
153  realisticAssociator.filterHitsByDistance(maxDistance_);
154 
155  const auto& realisticClusters = realisticAssociator.realisticClusters();
156  unsigned int nClusters = realisticClusters.size();
157  float bin_norm = 1. / (calibMaxEta_ - calibMinEta_);
158  float egamma_bin_norm = egammaCalib_.size() * bin_norm;
159  float hadron_bin_norm = hadronCalib_.size() * bin_norm;
160  for (unsigned ic = 0; ic < nClusters; ++ic) {
161  float highest_energy = 0.0f;
162  output.emplace_back();
163  reco::PFCluster& back = output.back();
165  float energyCorrection = 1.f;
166  float timeRealisticSC = -99.;
167  if (realisticClusters[ic].isVisible()) {
168  int pdgId = simClusters[ic].pdgId();
169  auto abseta = std::abs(simClusters[ic].eta());
170  if ((abseta >= calibMinEta_) and (abseta <= calibMaxEta_)) //protecting range
171  {
172  if ((isEGamma(pdgId) or isPi0(pdgId)) and !egammaCalib_.empty()) {
173  unsigned int etabin = std::floor(((abseta - calibMinEta_) * egamma_bin_norm));
174  energyCorrection = egammaCalib_[etabin];
175  } else if (isHadron(pdgId) and !(isPi0(pdgId)) and
176  !hadronCalib_
177  .empty()) // this function is expensive.. should we treat as hadron everything which is not egamma?
178  {
179  unsigned int etabin = std::floor(((abseta - calibMinEta_) * hadron_bin_norm));
180  energyCorrection = hadronCalib_[etabin];
181  }
182  }
183  std::vector<float> timeHits;
184  const auto& hitsIdsAndFractions = realisticClusters[ic].hitsIdsAndFractions();
185  for (const auto& idAndF : hitsIdsAndFractions) {
186  auto fraction = idAndF.second;
187  if (fraction > 0.f) {
188  auto ref = makeRefhit(input, idAndF.first);
190  const float hit_energy = fraction * ref->energy();
191  if (hit_energy > highest_energy || highest_energy == 0.0) {
192  highest_energy = hit_energy;
193  seed = ref;
194  }
195  //select hits good for timing
196  if (ref->time() > -1.) {
197  int rhLayer = rhtools_.getLayerWithOffset(ref->detId());
198  std::array<float, 3> scPosition = realisticClusters[ic].getCenterOfGravity(rhLayer);
199  float distanceSquared =
200  std::pow((ref->position().x() - scPosition[0]), 2) + std::pow((ref->position().y() - scPosition[1]), 2);
201  if (distanceSquared < maxDforTimingSquared_) {
202  timeHits.push_back(ref->time());
203  }
204  }
205  }
206  }
207  //assign time if minimum number of hits
209  timeRealisticSC = (timeEstimator.fixSizeHighestDensity(timeHits)).first;
210  }
211  if (!back.hitsAndFractions().empty()) {
212  back.setSeed(seed->detId());
213  back.setEnergy(realisticClusters[ic].getEnergy());
214  back.setCorrectedEnergy(energyCorrection * realisticClusters[ic].getEnergy()); //applying energy correction
215  back.setTime(timeRealisticSC);
216  } else {
217  back.setSeed(-1);
218  back.setEnergy(0.f);
219  }
220  }
221 }
void insertLayerId(unsigned int layerId, unsigned int hitIndex)
const std::vector< std::pair< DetId, float > > & hitsAndFractions() const
Definition: CaloCluster.h:210
reco::PFRecHitRef makeRefhit(const edm::Handle< reco::PFRecHitCollection > &h, const unsigned i) const
unsigned int getLayer(DetId::Detector type, bool nose=false) const
Definition: RecHitTools.cc:318
Particle flow cluster, see clustering algorithm in PFClusterAlgo.
Definition: PFCluster.h:42
constexpr uint32_t numberOfLayers
double getEnergy(HBHERecHitCollection::const_iterator hit, int useRaw=0, bool debug=false)
void setTime(float time, float timeError=0)
Definition: PFCluster.h:84
Fraction of a PFRecHit (rechits can be shared between several PFCluster&#39;s)
void setEnergy(double energy)
Definition: CaloCluster.h:136
int getGeometryType() const
Definition: RecHitTools.h:86
static std::string const input
Definition: EdmProvDump.cc:50
void setSeed(const DetId &id)
Definition: CaloCluster.h:146
std::pair< float, float > fixSizeHighestDensity(std::vector< float > &time, std::vector< float > weight=std::vector< float >(), unsigned int minNhits=3, float deltaT=0.210, float timeWidthBy=0.5)
void init(std::size_t numberOfHits, std::size_t numberOfSimClusters, std::size_t numberOfLayers)
unsigned int lastLayerFH() const
Definition: RecHitTools.h:77
GlobalPoint getPosition(const DetId &id) const
Definition: RecHitTools.cc:140
const std::vector< RealisticCluster > & realisticClusters() const
void setCorrectedEnergy(double cenergy)
Definition: CaloCluster.h:137
void insertSimClusterIdAndFraction(unsigned int scIdx, float fraction, unsigned int hitIndex, float associatedEnergy)
void computeAssociation(float exclusiveFraction, bool useMCFractionsForExclEnergy, unsigned int fhOffset, unsigned int bhOffset)
The Signals That Services Can Subscribe To This is based on ActivityRegistry and is current per Services can connect to the signals distributed by the ActivityRegistry in order to monitor the activity of the application Each possible callback has some defined which we here list in angle e< void, edm::EventID const &, edm::Timestamp const & > We also list in braces which AR_WATCH_USING_METHOD_ is used for those or
Definition: Activities.doc:12
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
void findAndMergeInvisibleClusters(float invisibleFraction, float exclusiveFraction)
void insertHitPosition(float x, float y, float z, unsigned int hitIndex)
double f[11][100]
void insertHitEnergy(float energy, unsigned int hitIndex)
void addRecHitFraction(const reco::PFRecHitFraction &frac)
add a given fraction of the rechit
Definition: PFCluster.cc:33
Definition: output.py:1
edm::Handle< SimClusterCollection > simClusterH_
std::vector< SimCluster > SimClusterCollection
Power< A, B >::type pow(const A &a, const B &b)
Definition: Power.h:29
unsigned int lastLayerEE(bool nose=false) const
Definition: RecHitTools.h:76
unsigned int getLayerWithOffset(const DetId &) const
Definition: RecHitTools.cc:381

◆ operator=()

RealisticSimClusterMapper& RealisticSimClusterMapper::operator= ( const RealisticSimClusterMapper )
delete

◆ update()

void RealisticSimClusterMapper::update ( const edm::EventSetup es)
finalvirtual

Reimplemented from InitialClusteringStepBase.

Definition at line 108 of file RealisticSimClusterMapper.cc.

References geomToken_, edm::EventSetup::getData(), rhtools_, and hgcal::RecHitTools::setGeometry().

Referenced by progressbar.ProgressBar::__next__(), MatrixUtil.Matrix::__setitem__(), MatrixUtil.Steps::__setitem__(), progressbar.ProgressBar::finish(), and MatrixUtil.Steps::overwrite().

T const & getData(const ESGetToken< T, R > &iToken) const noexcept(false)
Definition: EventSetup.h:119
void setGeometry(CaloGeometry const &)
Definition: RecHitTools.cc:79
edm::ESGetToken< CaloGeometry, CaloGeometryRecord > geomToken_

◆ updateEvent()

void RealisticSimClusterMapper::updateEvent ( const edm::Event ev)
finalvirtual

Reimplemented from InitialClusteringStepBase.

Definition at line 106 of file RealisticSimClusterMapper.cc.

References makeMEIFBenchmarkPlots::ev, simClusterH_, and simClusterToken_.

106 { ev.getByToken(simClusterToken_, simClusterH_); }
edm::EDGetTokenT< SimClusterCollection > simClusterToken_
edm::Handle< SimClusterCollection > simClusterH_

Member Data Documentation

◆ calibMaxEta_

const float RealisticSimClusterMapper::calibMaxEta_ = 3.0
private

Definition at line 67 of file RealisticSimClusterMapper.cc.

Referenced by buildClusters().

◆ calibMinEta_

const float RealisticSimClusterMapper::calibMinEta_ = 1.4
private

Definition at line 66 of file RealisticSimClusterMapper.cc.

Referenced by buildClusters().

◆ egammaCalib_

std::vector<double> RealisticSimClusterMapper::egammaCalib_
private

Definition at line 69 of file RealisticSimClusterMapper.cc.

Referenced by buildClusters().

◆ exclusiveFraction_

const float RealisticSimClusterMapper::exclusiveFraction_ = 0.7f
private

Definition at line 59 of file RealisticSimClusterMapper.cc.

Referenced by buildClusters().

◆ geomToken_

edm::ESGetToken<CaloGeometry, CaloGeometryRecord> RealisticSimClusterMapper::geomToken_
private

Definition at line 74 of file RealisticSimClusterMapper.cc.

Referenced by update().

◆ hadronCalib_

std::vector<double> RealisticSimClusterMapper::hadronCalib_
private

Definition at line 68 of file RealisticSimClusterMapper.cc.

Referenced by buildClusters().

◆ invisibleFraction_

const float RealisticSimClusterMapper::invisibleFraction_ = 0.3f
private

Definition at line 58 of file RealisticSimClusterMapper.cc.

Referenced by buildClusters().

◆ maxDforTimingSquared_

const float RealisticSimClusterMapper::maxDforTimingSquared_ = 4.0f
private

Definition at line 62 of file RealisticSimClusterMapper.cc.

Referenced by buildClusters().

◆ maxDistance_

const float RealisticSimClusterMapper::maxDistance_ = 10.f
private

Definition at line 61 of file RealisticSimClusterMapper.cc.

Referenced by buildClusters().

◆ maxDistanceFilter_

const bool RealisticSimClusterMapper::maxDistanceFilter_ = false
private

Definition at line 60 of file RealisticSimClusterMapper.cc.

Referenced by buildClusters().

◆ minNHitsforTiming_

const unsigned int RealisticSimClusterMapper::minNHitsforTiming_ = 3
private

Definition at line 64 of file RealisticSimClusterMapper.cc.

◆ rhtools_

hgcal::RecHitTools RealisticSimClusterMapper::rhtools_
private

Definition at line 57 of file RealisticSimClusterMapper.cc.

Referenced by buildClusters(), and update().

◆ simClusterH_

edm::Handle<SimClusterCollection> RealisticSimClusterMapper::simClusterH_
private

Definition at line 72 of file RealisticSimClusterMapper.cc.

Referenced by buildClusters(), and updateEvent().

◆ simClusterToken_

edm::EDGetTokenT<SimClusterCollection> RealisticSimClusterMapper::simClusterToken_
private

Definition at line 71 of file RealisticSimClusterMapper.cc.

Referenced by updateEvent().

◆ timeOffset_

const float RealisticSimClusterMapper::timeOffset_
private

Definition at line 63 of file RealisticSimClusterMapper.cc.

◆ useMCFractionsForExclEnergy_

const bool RealisticSimClusterMapper::useMCFractionsForExclEnergy_ = false
private

Definition at line 65 of file RealisticSimClusterMapper.cc.

Referenced by buildClusters().