EleTkIsolFromCands Class Reference

#include <EleTkIsolFromCands.h>

Classes
struct	TrkCuts

Public Types
enum	PIDVeto { PIDVeto::NONE = 0, PIDVeto::ELES, PIDVeto::NONELES }

Public Member Functions
std::pair< int, double >	calIsol (const reco::TrackBase &trk, const pat::PackedCandidateCollection &cands, const PIDVeto=PIDVeto::NONE) const
std::pair< int, double >	calIsol (const double eleEta, const double elePhi, const double eleVZ, const pat::PackedCandidateCollection &cands, const PIDVeto=PIDVeto::NONE) const
std::pair< int, double >	calIsol (const reco::TrackBase &trk, const reco::TrackCollection &tracks) const
std::pair< int, double >	calIsol (const double eleEta, const double elePhi, const double eleVZ, const reco::TrackCollection &tracks) const
template<typename... Args>
double	calIsolPt (Args &&...args) const
	EleTkIsolFromCands (const edm::ParameterSet &para)
	EleTkIsolFromCands (const EleTkIsolFromCands &)=default
EleTkIsolFromCands &	operator= (const EleTkIsolFromCands &)=default
	~EleTkIsolFromCands ()=default

Static Public Member Functions
static bool	passPIDVeto (const int pdgId, const EleTkIsolFromCands::PIDVeto pidVeto)
static PIDVeto	pidVetoFromStr (const std::string &vetoStr)
static edm::ParameterSetDescription	pSetDescript ()

Static Private Member Functions
static bool	passAlgo (const reco::TrackBase &trk, const std::vector< reco::TrackBase::TrackAlgorithm > &algosToRej)
static bool	passQual (const reco::TrackBase &trk, const std::vector< reco::TrackBase::TrackQuality > &quals)
static bool	passTrkSel (const reco::TrackBase &trk, const double trkPt, const TrkCuts &cuts, const double eleEta, const double elePhi, const double eleVZ)

Private Attributes
TrkCuts	barrelCuts_
TrkCuts	endcapCuts_

Detailed Description

Definition at line 44 of file EleTkIsolFromCands.h.

Member Enumeration Documentation

enum EleTkIsolFromCands::PIDVeto

strong

Enumerator
NONE
ELES
NONELES

Definition at line 46 of file EleTkIsolFromCands.h.

                     {
    NONE = 0,
    ELES,
    NONELES,
  };

Constructor & Destructor Documentation

EleTkIsolFromCands::EleTkIsolFromCands ( const edm::ParameterSet &  para )

explicit

Definition at line 43 of file EleTkIsolFromCands.cc.

    : barrelCuts_(para.getParameter<edm::ParameterSet>("barrelCuts")),
      endcapCuts_(para.getParameter<edm::ParameterSet>("endcapCuts")) {}

EleTkIsolFromCands::EleTkIsolFromCands ( const EleTkIsolFromCands &  )

default

EleTkIsolFromCands::~EleTkIsolFromCands ( )

default

Member Function Documentation

std::pair< int, double > EleTkIsolFromCands::calIsol	(	const reco::TrackBase &	trk,
		const pat::PackedCandidateCollection &	cands,
		const PIDVeto	pidVeto = PIDVeto::NONE
	)		const

Definition at line 54 of file EleTkIsolFromCands.cc.

References HLT_2018_cff::cands, reco::TrackBase::eta(), reco::TrackBase::phi(), and reco::TrackBase::vz().

Referenced by calIsol(), and calIsolPt().

                                                                                {
  return calIsol(eleTrk.eta(), eleTrk.phi(), eleTrk.vz(), cands, pidVeto);
}

std::pair< int, double > EleTkIsolFromCands::calIsol	(	const double	eleEta,
		const double	elePhi,
		const double	eleVZ,
		const pat::PackedCandidateCollection &	cands,
		const PIDVeto	pidVeto = PIDVeto::NONE
	)		const

Definition at line 60 of file EleTkIsolFromCands.cc.

References funct::abs(), barrelCuts_, L1TMuonDQMOffline_cfi::cuts, endcapCuts_, passPIDVeto(), passTrkSel(), and reco::TrackBase::pt().

                                                                                {
  double ptSum = 0.;
  int nrTrks = 0;

  const TrkCuts& cuts = std::abs(eleEta) < 1.5 ? barrelCuts_ : endcapCuts_;

  for (auto& cand : cands) {
    if (cand.hasTrackDetails() && cand.charge() != 0 && passPIDVeto(cand.pdgId(), pidVeto)) {
      const reco::Track& trk = cand.pseudoTrack();
      if (passTrkSel(trk, trk.pt(), cuts, eleEta, elePhi, eleVZ)) {
        ptSum += trk.pt();
        nrTrks++;
      }
    }
  }
  return {nrTrks, ptSum};
}

std::pair< int, double > EleTkIsolFromCands::calIsol	(	const reco::TrackBase &	trk,
		const reco::TrackCollection &	tracks
	)		const

Definition at line 82 of file EleTkIsolFromCands.cc.

References calIsol(), reco::TrackBase::eta(), reco::TrackBase::phi(), PDWG_EXOHSCP_cff::tracks, and reco::TrackBase::vz().

                                                                                            {
  return calIsol(eleTrk.eta(), eleTrk.phi(), eleTrk.vz(), tracks);
}

std::pair< int, double > EleTkIsolFromCands::calIsol	(	const double	eleEta,
		const double	elePhi,
		const double	eleVZ,
		const reco::TrackCollection &	tracks
	)		const

Definition at line 87 of file EleTkIsolFromCands.cc.

References funct::abs(), barrelCuts_, L1TMuonDQMOffline_cfi::cuts, endcapCuts_, and passTrkSel().

                                                                                            {
  double ptSum = 0.;
  int nrTrks = 0;

  const TrkCuts& cuts = std::abs(eleEta) < 1.5 ? barrelCuts_ : endcapCuts_;

  for (auto& trk : tracks) {
    if (passTrkSel(trk, trk.pt(), cuts, eleEta, elePhi, eleVZ)) {
      ptSum += trk.pt();
      nrTrks++;
    }
  }
  return {nrTrks, ptSum};
}

template<typename... Args>

double EleTkIsolFromCands::calIsolPt ( Args &&...  args ) const

inline

Definition at line 95 of file EleTkIsolFromCands.h.

References writedatasetfile::args, calIsol(), L1TMuonDQMOffline_cfi::cuts, passAlgo(), passPIDVeto(), passQual(), passTrkSel(), EgammaValidation_cff::pdgId, pidVetoFromStr(), and AlCaHLTBitMon_QueryRunRegistry::string.

Referenced by ElectronHEEPIDValueMapProducer::calTrkIso(), and GsfElectronAlgo::createElectron().

                                         {
    return calIsol(std::forward<Args>(args)...).second;
  }

EleTkIsolFromCands& EleTkIsolFromCands::operator= ( const EleTkIsolFromCands &  )

default

bool EleTkIsolFromCands::passAlgo ( const reco::TrackBase &  trk, const std::vector< reco::TrackBase::TrackAlgorithm > &  algosToRej  )

staticprivate

Definition at line 167 of file EleTkIsolFromCands.cc.

References reco::TrackBase::algo().

Referenced by calIsolPt(), and passTrkSel().

                                                                                            {
  return algosToRej.empty() || !std::binary_search(algosToRej.begin(), algosToRej.end(), trk.algo());
}

bool EleTkIsolFromCands::passPIDVeto ( const int  pdgId, const EleTkIsolFromCands::PIDVeto  pidVeto  )

static

Definition at line 105 of file EleTkIsolFromCands.cc.

References funct::abs(), ELES, Exception, NONE, NONELES, and PbPb_ZMuSkimMuonDPG_cff::veto.

Referenced by calIsol(), and calIsolPt().

                                                                                          {
  int pidAbs = std::abs(pdgId);
  switch (veto) {
    case PIDVeto::NONE:
      return true;
    case PIDVeto::ELES:
      if (pidAbs == 11)
        return false;
      else
        return true;
    case PIDVeto::NONELES:
      if (pidAbs == 11)
        return true;
      else
        return false;
  }
  throw cms::Exception("CodeError") << "invalid PIDVeto " << static_cast<int>(veto) << ", "
                                    << "this is likely due to some static casting of invalid ints somewhere";
}

bool EleTkIsolFromCands::passQual ( const reco::TrackBase &  trk, const std::vector< reco::TrackBase::TrackQuality > &  quals  )

staticprivate

Definition at line 155 of file EleTkIsolFromCands.cc.

References reco::TrackBase::quality().

Referenced by calIsolPt(), and passTrkSel().

                                                                                                             {
  if (quals.empty())
    return true;

  for (auto qual : quals) {
    if (trk.quality(qual))
      return true;
  }

  return false;
}

bool EleTkIsolFromCands::passTrkSel ( const reco::TrackBase &  trk, const double  trkPt, const TrkCuts &  cuts, const double  eleEta, const double  elePhi, const double  eleVZ  )

staticprivate

Definition at line 138 of file EleTkIsolFromCands.cc.

References funct::abs(), EleTkIsolFromCands::TrkCuts::algosToReject, EleTkIsolFromCands::TrkCuts::allowedQualities, reco::deltaR2(), HLT_2018_cff::dEta, reco::TrackBase::eta(), reco::TrackBase::hitPattern(), EleTkIsolFromCands::TrkCuts::maxDPtPt, EleTkIsolFromCands::TrkCuts::maxDR2, EleTkIsolFromCands::TrkCuts::maxDZ, EleTkIsolFromCands::TrkCuts::minDEta, EleTkIsolFromCands::TrkCuts::minDR2, EleTkIsolFromCands::TrkCuts::minHits, EleTkIsolFromCands::TrkCuts::minPixelHits, EleTkIsolFromCands::TrkCuts::minPt, reco::HitPattern::numberOfValidHits(), reco::HitPattern::numberOfValidPixelHits(), passAlgo(), passQual(), reco::TrackBase::phi(), reco::TrackBase::ptError(), and reco::TrackBase::vz().

Referenced by calIsol(), and calIsolPt().

                                                        {
  const float dR2 = reco::deltaR2(eleEta, elePhi, trk.eta(), trk.phi());
  const float dEta = trk.eta() - eleEta;
  const float dZ = eleVZ - trk.vz();

  return dR2 >= cuts.minDR2 && dR2 <= cuts.maxDR2 && std::abs(dEta) >= cuts.minDEta && std::abs(dZ) < cuts.maxDZ &&
         trk.hitPattern().numberOfValidHits() >= cuts.minHits &&
         trk.hitPattern().numberOfValidPixelHits() >= cuts.minPixelHits &&
         (trk.ptError() / trkPt < cuts.maxDPtPt || cuts.maxDPtPt < 0) && passQual(trk, cuts.allowedQualities) &&
         passAlgo(trk, cuts.algosToReject) && trkPt > cuts.minPt;
}

EleTkIsolFromCands::PIDVeto EleTkIsolFromCands::pidVetoFromStr ( const std::string &  vetoStr )

static

Definition at line 125 of file EleTkIsolFromCands.cc.

References ELES, Exception, NONE, and NONELES.

Referenced by calIsolPt(), and ElectronHEEPIDValueMapProducer::ElectronHEEPIDValueMapProducer().

                                                                                     {
  if (vetoStr == "NONE")
    return PIDVeto::NONE;
  else if (vetoStr == "ELES")
    return PIDVeto::ELES;
  else if (vetoStr == "NONELES")
    return PIDVeto::NONELES;
  else {
    throw cms::Exception("CodeError") << "unrecognised string " << vetoStr
                                      << ", either a typo or this function needs to be updated";
  }
}

edm::ParameterSetDescription EleTkIsolFromCands::pSetDescript ( )

static

Definition at line 47 of file EleTkIsolFromCands.cc.

References edm::ParameterSetDescription::add(), and EleTkIsolFromCands::TrkCuts::pSetDescript().

Referenced by GsfElectronBaseProducer::fillDescriptions(), and ElectronHEEPIDValueMapProducer::fillDescriptions().

                                                            {
  edm::ParameterSetDescription desc;
  desc.add("barrelCuts", TrkCuts::pSetDescript());
  desc.add("endcapCuts", TrkCuts::pSetDescript());
  return desc;
}

Member Data Documentation

TrkCuts EleTkIsolFromCands::barrelCuts_

private

Definition at line 68 of file EleTkIsolFromCands.h.

Referenced by calIsol().

TrkCuts EleTkIsolFromCands::endcapCuts_

private

Definition at line 68 of file EleTkIsolFromCands.h.

Referenced by calIsol().