HGCalIsoCalculator Class Reference

#include <HGCalIsoCalculator.h>

Public Member Functions
const float	getIso (const unsigned int ring) const
	HGCalIsoCalculator ()
void	produceHGCalIso (const reco::CaloClusterPtr &seedCluster)
void	setDeltaR (const float dr)
void	setMinDeltaR (const float dr)
void	setNLayers (unsigned int nLayers)
void	setNRings (const size_t nrings)
void	setRecHits (edm::Handle< HGCRecHitCollection > hitsEE, edm::Handle< HGCRecHitCollection > hitsFH, edm::Handle< HGCRecHitCollection > hitsBH)
	fill - once per event More...
void	setRecHitTools (const hgcal::RecHitTools *recHitTools)
	~HGCalIsoCalculator ()

Private Attributes
bool	debug_
float	dr2_
std::vector< std::pair< float, float > >	hitEtaPhiCache_
std::vector< float >	isoringdeposits_
float	mindr2_
unsigned int	nlayers_
edm::Handle< HGCRecHitCollection >	recHitsBH_
edm::Handle< HGCRecHitCollection >	recHitsEE_
edm::Handle< HGCRecHitCollection >	recHitsFH_
const hgcal::RecHitTools *	rechittools_
std::vector< unsigned int >	ringasso_

Detailed Description

Definition at line 46 of file HGCalIsoCalculator.h.

Constructor & Destructor Documentation

HGCalIsoCalculator::HGCalIsoCalculator

(

)

Definition at line 11 of file HGCalIsoCalculator.cc.

References setNRings().

    : dr2_(0.15 * 0.15), mindr2_(0), rechittools_(nullptr), debug_(false), nlayers_(30) {
  setNRings(5);
}

HGCalIsoCalculator::~HGCalIsoCalculator

(

)

Definition at line 16 of file HGCalIsoCalculator.cc.

{}

Member Function Documentation

const float HGCalIsoCalculator::getIso

(

const unsigned int

ring

)

const

Definition at line 114 of file HGCalIsoCalculator.cc.

References isoringdeposits_, and relativeConstraints::ring.

Referenced by HGCalEgammaIDHelper::getIsolationRing().

                                                                    {
  if (ring >= isoringdeposits_.size())
    throw cms::Exception("HGCalIsoCalculator::getIso: ring index out of range");
  return isoringdeposits_[ring];
}

void HGCalIsoCalculator::produceHGCalIso

(

const reco::CaloClusterPtr &

seedCluster

)

Definition at line 48 of file HGCalIsoCalculator.cc.

References reco::deltaR2(), dr2_, Exception, hgcal::RecHitTools::getLayerWithOffset(), hitEtaPhiCache_, isoringdeposits_, phase1PixelTopology::layer, mindr2_, nlayers_, alignCSCRings::r, recHitsBH_, recHitsEE_, recHitsFH_, rechittools_, relativeConstraints::ring, and ringasso_.

Referenced by HGCalEgammaIDHelper::computeHGCAL().

                                                                       {
  if (!rechittools_)
    throw cms::Exception("HGCalIsoCalculator::produceHGCalIso: rechittools not set");

  for (auto& r : isoringdeposits_)
    r = 0;

  // make local temporaries to pass to the lambda
  // avoids recomputing every iteration
  const float seedEta = seed->eta();
  const float seedPhi = seed->phi();
  const std::vector<std::pair<DetId, float>>& seedHitsAndFractions = seed->hitsAndFractions();

  auto checkAndFill = [this, &seedEta, &seedPhi, &seedHitsAndFractions](const HGCRecHit& hit,
                                                                        std::pair<float, float> etaphiVal) {
    float deltar2 = reco::deltaR2(etaphiVal.first, etaphiVal.second, seedEta, seedPhi);
    if (deltar2 > dr2_ || deltar2 < mindr2_)
      return;

    unsigned int layer = rechittools_->getLayerWithOffset(hit.id());
    if (layer >= nlayers_)
      return;

    //do not consider hits associated to the photon cluster
    if (std::none_of(seedHitsAndFractions.begin(), seedHitsAndFractions.end(), [&hit](const auto& seedhit) {
          return hit.id() == seedhit.first;
        })) {
      const unsigned int ring = ringasso_.at(layer);
      isoringdeposits_.at(ring) += hit.energy();
    }
  };

  // The cache order is EE,FH,BH, so we should loop over them the same here
  auto itEtaPhiCache = hitEtaPhiCache_.cbegin();
  for (const auto& hit : *recHitsEE_) {
    checkAndFill(hit, *itEtaPhiCache);
    itEtaPhiCache++;
  }
  for (const auto& hit : *recHitsFH_) {
    checkAndFill(hit, *itEtaPhiCache);
    itEtaPhiCache++;
  }
  for (const auto& hit : *recHitsBH_) {
    checkAndFill(hit, *itEtaPhiCache);
    itEtaPhiCache++;
  }
}

void HGCalIsoCalculator::setDeltaR ( const float  dr )

inline

Definition at line 52 of file HGCalIsoCalculator.h.

References runTauDisplay::dr, and dr2_.

Referenced by HGCalEgammaIDHelper::HGCalEgammaIDHelper().

{ dr2_ = dr * dr; }

void HGCalIsoCalculator::setMinDeltaR ( const float  dr )

inline

Definition at line 54 of file HGCalIsoCalculator.h.

References runTauDisplay::dr, and mindr2_.

Referenced by HGCalEgammaIDHelper::HGCalEgammaIDHelper().

{ mindr2_ = dr * dr; }

void HGCalIsoCalculator::setNLayers ( unsigned int  nLayers )

inline

Definition at line 60 of file HGCalIsoCalculator.h.

References nlayers_.

{ nlayers_ = nLayers; }

void HGCalIsoCalculator::setNRings

(

const size_t

nrings

)

Definition at line 96 of file HGCalIsoCalculator.cc.

References counter, mps_fire::i, isoringdeposits_, nlayers_, ringasso_, and edm::separator().

Referenced by HGCalEgammaIDHelper::HGCalEgammaIDHelper(), and HGCalIsoCalculator().

                                                      {
  if (nrings > nlayers_)
    throw std::logic_error("PhotonHGCalIsoCalculator::setNRings: max number of rings reached");

  ringasso_.clear();
  isoringdeposits_.clear();
  unsigned int separator = nlayers_ / nrings;
  size_t counter = 0;
  for (size_t i = 0; i < nlayers_ + 1; i++) {
    ringasso_.push_back(counter);
    //the last ring might be larger.
    if (i && !(i % separator) && (int)counter < (int)nrings - 1) {
      counter++;
    }
  }
  isoringdeposits_.resize(nrings, 0);
}

void HGCalIsoCalculator::setRecHits	(	edm::Handle< HGCRecHitCollection >	hitsEE,
		edm::Handle< HGCRecHitCollection >	hitsFH,
		edm::Handle< HGCRecHitCollection >	hitsBH
	)

fill - once per event

Definition at line 18 of file HGCalIsoCalculator.cc.

References PVValHelper::eta, Exception, hgcal::RecHitTools::getEta(), hgcal::RecHitTools::getPhi(), hgcal::RecHitTools::getPosition(), hitEtaPhiCache_, hit::id, phi, position, recHitsBH_, recHitsEE_, recHitsFH_, and rechittools_.

Referenced by HGCalEgammaIDHelper::eventInit().

                                                                           {
  recHitsEE_ = hitsEE;
  recHitsFH_ = hitsFH;
  recHitsBH_ = hitsBH;

  if (!rechittools_)
    throw cms::Exception("HGCalIsoCalculator::produceHGCalIso: rechittools not set");

  hitEtaPhiCache_.clear();
  hitEtaPhiCache_.reserve(recHitsEE_->size() + recHitsFH_->size() + recHitsBH_->size());

  // Since HGCal is not projective and the rechits don't cache any
  // eta,phi, we make our own here
  auto makeEtaPhiPair = [this](const auto& hit) {
    const GlobalPoint position = rechittools_->getPosition(hit.id());
    float eta = rechittools_->getEta(position, 0);  //assume vertex at z=0
    float phi = rechittools_->getPhi(position);
    return std::make_pair(eta, phi);
  };

  for (const auto& hit : *recHitsEE_)
    hitEtaPhiCache_.push_back(makeEtaPhiPair(hit));
  for (const auto& hit : *recHitsFH_)
    hitEtaPhiCache_.push_back(makeEtaPhiPair(hit));
  for (const auto& hit : *recHitsBH_)
    hitEtaPhiCache_.push_back(makeEtaPhiPair(hit));
}

void HGCalIsoCalculator::setRecHitTools ( const hgcal::RecHitTools *  recHitTools )

inline

Definition at line 56 of file HGCalIsoCalculator.h.

References rechittools_.

Referenced by HGCalEgammaIDHelper::eventInit().

{ rechittools_ = recHitTools; }

Member Data Documentation

bool HGCalIsoCalculator::debug_

private

Definition at line 82 of file HGCalIsoCalculator.h.

float HGCalIsoCalculator::dr2_

private

Definition at line 75 of file HGCalIsoCalculator.h.

Referenced by produceHGCalIso(), and setDeltaR().

std::vector<std::pair<float, float> > HGCalIsoCalculator::hitEtaPhiCache_

private

Definition at line 81 of file HGCalIsoCalculator.h.

Referenced by produceHGCalIso(), and setRecHits().

std::vector<float> HGCalIsoCalculator::isoringdeposits_

private

Definition at line 72 of file HGCalIsoCalculator.h.

Referenced by getIso(), produceHGCalIso(), and setNRings().

float HGCalIsoCalculator::mindr2_

private

Definition at line 75 of file HGCalIsoCalculator.h.

Referenced by produceHGCalIso(), and setMinDeltaR().

unsigned int HGCalIsoCalculator::nlayers_

private

Definition at line 83 of file HGCalIsoCalculator.h.

Referenced by produceHGCalIso(), setNLayers(), and setNRings().

edm::Handle<HGCRecHitCollection> HGCalIsoCalculator::recHitsBH_

private

Definition at line 80 of file HGCalIsoCalculator.h.

Referenced by produceHGCalIso(), and setRecHits().

edm::Handle<HGCRecHitCollection> HGCalIsoCalculator::recHitsEE_

private

Definition at line 78 of file HGCalIsoCalculator.h.

Referenced by produceHGCalIso(), and setRecHits().

edm::Handle<HGCRecHitCollection> HGCalIsoCalculator::recHitsFH_

private

Definition at line 79 of file HGCalIsoCalculator.h.

Referenced by produceHGCalIso(), and setRecHits().

const hgcal::RecHitTools* HGCalIsoCalculator::rechittools_

private

Definition at line 77 of file HGCalIsoCalculator.h.

Referenced by produceHGCalIso(), setRecHits(), and setRecHitTools().

std::vector<unsigned int> HGCalIsoCalculator::ringasso_

private

Definition at line 73 of file HGCalIsoCalculator.h.

Referenced by produceHGCalIso(), and setNRings().