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 15 of file HGCalIsoCalculator.cc.

                                       {
}

Member Function Documentation

const float HGCalIsoCalculator::getIso

(

const unsigned int

ring

)

const

Definition at line 109 of file HGCalIsoCalculator.cc.

References isoringdeposits_, and relativeConstraints::ring.

Referenced by HGCalEgammaIDHelper::getIsolationRing(), and setNLayers().

                                                                  {
    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 47 of file HGCalIsoCalculator.cc.

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

Referenced by HGCalEgammaIDHelper::computeHGCAL(), and setNLayers().

                                                                       {

    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 getIso(), MuonTCMETValueMapProducer_cff::nLayers, nlayers_, produceHGCalIso(), relativeConstraints::ring, and setRecHits().

{nlayers_=nLayers;}

void HGCalIsoCalculator::setNRings

(

const size_t

nrings

)

Definition at line 91 of file HGCalIsoCalculator.cc.

References mps_fire::i, isoringdeposits_, nlayers_, ringasso_, and mps_merge::separator.

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

                                                     {
    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(), and setNLayers().

      {
    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_, and setNRings().

Referenced by HGCalEgammaIDHelper::eventInit().

{rechittools_ = recHitTools;}

Member Data Documentation

bool HGCalIsoCalculator::debug_

private

Definition at line 84 of file HGCalIsoCalculator.h.

float HGCalIsoCalculator::dr2_

private

Definition at line 77 of file HGCalIsoCalculator.h.

Referenced by produceHGCalIso(), and setDeltaR().

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

private

Definition at line 83 of file HGCalIsoCalculator.h.

Referenced by produceHGCalIso(), and setRecHits().

std::vector<float> HGCalIsoCalculator::isoringdeposits_

private

Definition at line 74 of file HGCalIsoCalculator.h.

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

float HGCalIsoCalculator::mindr2_

private

Definition at line 77 of file HGCalIsoCalculator.h.

Referenced by produceHGCalIso(), and setMinDeltaR().

unsigned int HGCalIsoCalculator::nlayers_

private

Definition at line 85 of file HGCalIsoCalculator.h.

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

edm::Handle<HGCRecHitCollection> HGCalIsoCalculator::recHitsBH_

private

Definition at line 82 of file HGCalIsoCalculator.h.

Referenced by produceHGCalIso(), and setRecHits().

edm::Handle<HGCRecHitCollection> HGCalIsoCalculator::recHitsEE_

private

Definition at line 80 of file HGCalIsoCalculator.h.

Referenced by produceHGCalIso(), and setRecHits().

edm::Handle<HGCRecHitCollection> HGCalIsoCalculator::recHitsFH_

private

Definition at line 81 of file HGCalIsoCalculator.h.

Referenced by produceHGCalIso(), and setRecHits().

const hgcal::RecHitTools* HGCalIsoCalculator::rechittools_

private

Definition at line 79 of file HGCalIsoCalculator.h.

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

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

private

Definition at line 75 of file HGCalIsoCalculator.h.

Referenced by produceHGCalIso(), and setNRings().