HcalPFClusterIsolation.cc

Go to the documentation of this file.

#include "RecoEgamma/EgammaIsolationAlgos/interface/HcalPFClusterIsolation.h"
 
#include "DataFormats/RecoCandidate/interface/RecoEcalCandidate.h"
#include "DataFormats/RecoCandidate/interface/RecoChargedCandidate.h"
#include "DataFormats/EgammaCandidates/interface/GsfElectron.h"
#include "DataFormats/EgammaCandidates/interface/Photon.h"
 
#include <DataFormats/Math/interface/deltaR.h>
 
template <typename T1>
HcalPFClusterIsolation<T1>::HcalPFClusterIsolation(double drMax,
                                                   double drVetoBarrel,
                                                   double drVetoEndcap,
                                                   double etaStripBarrel,
                                                   double etaStripEndcap,
                                                   double energyBarrel,
                                                   double energyEndcap,
                                                   bool useEt)
    : drMax_(drMax),
      drVetoBarrel_(drVetoBarrel),
      drVetoEndcap_(drVetoEndcap),
      etaStripBarrel_(etaStripBarrel),
      etaStripEndcap_(etaStripEndcap),
      energyBarrel_(energyBarrel),
      energyEndcap_(energyEndcap),
      useEt_(useEt) {}
 
template <typename T1>
HcalPFClusterIsolation<T1>::~HcalPFClusterIsolation() {}
 
template <typename T1>
double HcalPFClusterIsolation<T1>::getSum(const T1Ref candRef,
                                          const std::vector<edm::Handle<reco::PFClusterCollection>>& clusterHandles) {
  double etSum = 0.;
 
  float etaStrip = 0;
  float dRVeto = 0;
  if (fabs(candRef->eta()) < 1.479) {
    dRVeto = drVetoBarrel_;
    etaStrip = etaStripBarrel_;
  } else {
    dRVeto = drVetoEndcap_;
    etaStrip = etaStripEndcap_;
  }
 
  for (unsigned int nHandle = 0; nHandle < clusterHandles.size(); nHandle++) {
    for (unsigned i = 0; i < clusterHandles[nHandle]->size(); i++) {
      const reco::PFClusterRef pfclu(clusterHandles[nHandle], i);
 
      if (fabs(candRef->eta()) < 1.479) {
        if (fabs(pfclu->pt()) < energyBarrel_)
          continue;
      } else {
        if (fabs(pfclu->energy()) < energyEndcap_)
          continue;
      }
 
      float dEta = fabs(candRef->eta() - pfclu->eta());
      if (dEta < etaStrip)
        continue;
 
      float dR = deltaR(candRef->eta(), candRef->phi(), pfclu->eta(), pfclu->phi());
      if (dR > drMax_ || dR < dRVeto)
        continue;
 
      if (useEt_)
        etSum += pfclu->pt();
      else
        etSum += pfclu->energy();
    }
  }
 
  return etSum;
}
 
template class HcalPFClusterIsolation<reco::RecoEcalCandidate>;
template class HcalPFClusterIsolation<reco::RecoChargedCandidate>;
template class HcalPFClusterIsolation<reco::Photon>;
template class HcalPFClusterIsolation<reco::GsfElectron>;