EleTkIsolFromCands Class Reference

#include <EleTkIsolFromCands.h>

Classes
struct	Configuration
struct	Output
struct	TrkCuts

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

Public Member Functions
	EleTkIsolFromCands (Configuration const &cfg, reco::TrackCollection const &tracks)
	EleTkIsolFromCands (Configuration const &cfg, pat::PackedCandidateCollection const &cands, PIDVeto pidVeto=PIDVeto::NONE)
Output	operator() (const reco::TrackBase &electronTrack)

Static Public Member Functions
static PIDVeto	pidVetoFromStr (const std::string &vetoStr)
static edm::ParameterSetDescription	pSetDescript ()

Private Types
using	TrackTable = edm::soa::Table< edm::soa::col::Pt, edm::soa::col::Eta, edm::soa::col::Phi, edm::soa::col::Vz >

Private Member Functions
TrackTable const &	getPreselectedTracks (bool isBarrel)

Static Private Member Functions
static bool	passAlgo (const reco::TrackBase &trk, const std::vector< reco::TrackBase::TrackAlgorithm > &algosToRej)
static bool	passPIDVeto (const int pdgId, const EleTkIsolFromCands::PIDVeto pidVeto)
static bool	passQual (const reco::TrackBase &trk, const std::vector< reco::TrackBase::TrackQuality > &quals)
static bool	passTrackPreselection (const reco::TrackBase &trk, float trkPt, const TrkCuts &cuts)
static TrackTable	preselectTracks (reco::TrackCollection const &tracks, TrkCuts const &cuts)
static TrackTable	preselectTracksFromCands (pat::PackedCandidateCollection const &cands, TrkCuts const &cuts, PIDVeto=PIDVeto::NONE)

Private Attributes
pat::PackedCandidateCollection const *	cands_ = nullptr
Configuration const &	cfg_
const PIDVeto	pidVeto_ = PIDVeto::NONE
TrackTable	preselectedTracksWithBarrelCuts_
TrackTable	preselectedTracksWithEndcapCuts_
reco::TrackCollection const *	tracks_ = nullptr
bool	tracksCachedForBarrelCuts_ = false
bool	tracksCachedForEndcapCuts_ = false

Detailed Description

Definition at line 46 of file EleTkIsolFromCands.h.

Member Typedef Documentation

using EleTkIsolFromCands::TrackTable = edm::soa::Table<edm::soa::col::Pt, edm::soa::col::Eta, edm::soa::col::Phi, edm::soa::col::Vz>

private

Definition at line 106 of file EleTkIsolFromCands.h.

Member Enumeration Documentation

enum EleTkIsolFromCands::PIDVeto

strong

Enumerator
NONE
ELES
NONELES

Definition at line 71 of file EleTkIsolFromCands.h.

                     {
    NONE = 0,
    ELES,
    NONELES,
  };

Constructor & Destructor Documentation

EleTkIsolFromCands::EleTkIsolFromCands ( Configuration const &  cfg, reco::TrackCollection const &  tracks  )

inlineexplicit

Definition at line 77 of file EleTkIsolFromCands.h.

      : cfg_{cfg}, tracks_{&tracks} {}

EleTkIsolFromCands::EleTkIsolFromCands ( Configuration const &  cfg, pat::PackedCandidateCollection const &  cands, PIDVeto  pidVeto = PIDVeto::NONE  )

inlineexplicit

Definition at line 79 of file EleTkIsolFromCands.h.

      : cfg_{cfg}, cands_{&cands}, pidVeto_{pidVeto} {}

Member Function Documentation

EleTkIsolFromCands::TrackTable const & EleTkIsolFromCands::getPreselectedTracks ( bool  isBarrel )

private

Definition at line 186 of file EleTkIsolFromCands.cc.

References EleTkIsolFromCands::Configuration::barrelCuts, cands_, cfg_, cuts, EleTkIsolFromCands::Configuration::endcapCuts, pidVeto_, preselectedTracksWithBarrelCuts_, preselectedTracksWithEndcapCuts_, preselectTracks(), preselectTracksFromCands(), tracks_, tracksCachedForBarrelCuts_, and tracksCachedForEndcapCuts_.

Referenced by operator()().

                                                                                          {
  auto const& cuts = isBarrel ? cfg_.barrelCuts : cfg_.endcapCuts;
  auto& preselectedTracks = isBarrel ? preselectedTracksWithBarrelCuts_ : preselectedTracksWithEndcapCuts_;
  bool& tracksCached = isBarrel ? tracksCachedForBarrelCuts_ : tracksCachedForEndcapCuts_;

  if (!tracksCached) {
    preselectedTracks = tracks_ ? preselectTracks(*tracks_, cuts) : preselectTracksFromCands(*cands_, cuts, pidVeto_);
    tracksCached = true;
  }

  return preselectedTracks;
}

EleTkIsolFromCands::Output EleTkIsolFromCands::operator()

(

const reco::TrackBase &

electronTrack

)

Definition at line 102 of file EleTkIsolFromCands.cc.

References funct::abs(), EleTkIsolFromCands::Configuration::barrelCuts, cfg_, cuy::col, cuts, reco::deltaR2(), EleTkIsolFromCands::Configuration::endcapCuts, reco::TrackBase::eta(), getPreselectedTracks(), reco::TrackBase::phi(), and reco::TrackBase::vz().

                                                                                   {
  using namespace edm::soa::col;

  float ptSum = 0.;
  int nrTrks = 0;

  const float eleEta = eleTrk.eta();
  const float elePhi = eleTrk.phi();
  const float eleVz = eleTrk.vz();

  const bool isBarrelElectron = std::abs(eleEta) < 1.5;

  auto const& preselectedTracks = getPreselectedTracks(isBarrelElectron);
  auto const& cuts = isBarrelElectron ? cfg_.barrelCuts : cfg_.endcapCuts;

  for (auto const& trk : preselectedTracks) {
    const float dR2 = reco::deltaR2(eleEta, elePhi, trk.get<Eta>(), trk.get<Phi>());
    const float dEta = trk.get<Eta>() - eleEta;
    const float dZ = eleVz - trk.get<Vz>();

    if (dR2 >= cuts.minDR2 && dR2 <= cuts.maxDR2 && std::abs(dEta) >= cuts.minDEta && std::abs(dZ) < cuts.maxDZ) {
      ptSum += trk.get<Pt>();
      nrTrks++;
    }
  }
  return {nrTrks, ptSum};
}

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

staticprivate

Definition at line 181 of file EleTkIsolFromCands.cc.

References reco::TrackBase::algo().

Referenced by passTrackPreselection().

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

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

staticprivate

Definition at line 130 of file EleTkIsolFromCands.cc.

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

Referenced by preselectTracksFromCands().

                                                                                          {
  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 169 of file EleTkIsolFromCands.cc.

References reco::TrackBase::quality().

Referenced by passTrackPreselection().

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

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

  return false;
}

bool EleTkIsolFromCands::passTrackPreselection ( const reco::TrackBase &  trk, float  trkPt, const TrkCuts &  cuts  )

staticprivate

Definition at line 163 of file EleTkIsolFromCands.cc.

References EleTkIsolFromCands::TrkCuts::algosToReject, EleTkIsolFromCands::TrkCuts::allowedQualities, reco::TrackBase::hitPattern(), EleTkIsolFromCands::TrkCuts::maxDPtPt, EleTkIsolFromCands::TrkCuts::minHits, EleTkIsolFromCands::TrkCuts::minPixelHits, EleTkIsolFromCands::TrkCuts::minPt, reco::TrackBase::numberOfValidHits(), reco::HitPattern::numberOfValidPixelHits(), passAlgo(), passQual(), and reco::TrackBase::ptError().

Referenced by preselectTracks(), and preselectTracksFromCands().

                                                                                                           {
  return trackPt > cuts.minPt && trk.numberOfValidHits() >= cuts.minHits &&
         trk.hitPattern().numberOfValidPixelHits() >= cuts.minPixelHits && passQual(trk, cuts.allowedQualities) &&
         passAlgo(trk, cuts.algosToReject) && (cuts.maxDPtPt < 0 || trk.ptError() / trackPt < cuts.maxDPtPt);
}

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

static

Definition at line 150 of file EleTkIsolFromCands.cc.

References ELES, Exception, NONE, and NONELES.

Referenced by 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";
  }
}

EleTkIsolFromCands::TrackTable EleTkIsolFromCands::preselectTracks ( reco::TrackCollection const &  tracks, TrkCuts const &  cuts  )

staticprivate

Definition at line 50 of file EleTkIsolFromCands.cc.

References PVValHelper::eta, eostools::move(), passTrackPreselection(), phi, and DiDispStaMuonMonitor_cfi::pt.

Referenced by getPreselectedTracks().

                                                                                        {
  std::vector<float> pt;
  std::vector<float> eta;
  std::vector<float> phi;
  std::vector<float> vz;
  pt.reserve(tracks.size());
  eta.reserve(tracks.size());
  phi.reserve(tracks.size());
  vz.reserve(tracks.size());

  for (auto const& trk : tracks) {
    auto trkPt = trk.pt();
    if (passTrackPreselection(trk, trkPt, cuts)) {
      pt.emplace_back(trkPt);
      eta.emplace_back(trk.eta());
      phi.emplace_back(trk.phi());
      vz.emplace_back(trk.vz());
    }
  }

  return {pt, std::move(eta), std::move(phi), std::move(vz)};
}

EleTkIsolFromCands::TrackTable EleTkIsolFromCands::preselectTracksFromCands ( pat::PackedCandidateCollection const &  cands, TrkCuts const &  cuts, PIDVeto  pidVeto = PIDVeto::NONE  )

staticprivate

Definition at line 74 of file EleTkIsolFromCands.cc.

References PVValHelper::eta, reco::TrackBase::eta(), eostools::move(), passPIDVeto(), passTrackPreselection(), phi, reco::TrackBase::phi(), DiDispStaMuonMonitor_cfi::pt, reco::TrackBase::pt(), and reco::TrackBase::vz().

Referenced by getPreselectedTracks().

                                                                                             {
  std::vector<float> pt;
  std::vector<float> eta;
  std::vector<float> phi;
  std::vector<float> vz;
  pt.reserve(cands.size());
  eta.reserve(cands.size());
  phi.reserve(cands.size());
  vz.reserve(cands.size());

  for (auto const& cand : cands) {
    if (cand.hasTrackDetails() && cand.charge() != 0 && passPIDVeto(cand.pdgId(), pidVeto)) {
      const reco::Track& trk = cand.pseudoTrack();
      float trkPt = trk.pt();
      if (passTrackPreselection(trk, trkPt, cuts)) {
        pt.emplace_back(trkPt);
        eta.emplace_back(trk.eta());
        phi.emplace_back(trk.phi());
        vz.emplace_back(trk.vz());
      }
    }
  }

  return {pt, std::move(eta), std::move(phi), std::move(vz)};
}

edm::ParameterSetDescription EleTkIsolFromCands::pSetDescript ( )

static

Definition at line 43 of file EleTkIsolFromCands.cc.

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

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

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

Member Data Documentation

pat::PackedCandidateCollection const* EleTkIsolFromCands::cands_ = nullptr

private

Definition at line 127 of file EleTkIsolFromCands.h.

Referenced by getPreselectedTracks().

Configuration const& EleTkIsolFromCands::cfg_

private

Definition at line 123 of file EleTkIsolFromCands.h.

Referenced by getPreselectedTracks(), and operator()().

const PIDVeto EleTkIsolFromCands::pidVeto_ = PIDVeto::NONE

private

Definition at line 128 of file EleTkIsolFromCands.h.

Referenced by getPreselectedTracks().

TrackTable EleTkIsolFromCands::preselectedTracksWithBarrelCuts_

private

Definition at line 129 of file EleTkIsolFromCands.h.

Referenced by getPreselectedTracks().

TrackTable EleTkIsolFromCands::preselectedTracksWithEndcapCuts_

private

Definition at line 130 of file EleTkIsolFromCands.h.

Referenced by getPreselectedTracks().

reco::TrackCollection const* EleTkIsolFromCands::tracks_ = nullptr

private

Definition at line 126 of file EleTkIsolFromCands.h.

Referenced by getPreselectedTracks().

bool EleTkIsolFromCands::tracksCachedForBarrelCuts_ = false

private

Definition at line 131 of file EleTkIsolFromCands.h.

Referenced by getPreselectedTracks().

bool EleTkIsolFromCands::tracksCachedForEndcapCuts_ = false

private

Definition at line 132 of file EleTkIsolFromCands.h.

Referenced by getPreselectedTracks().