#include <ElectronTkIsolation.h>

Public Member Functions
	ElectronTkIsolation (double extRadius, double intRadius, double ptLow, double lip, double drb, const reco::TrackCollection *trackCollection, reco::TrackBase::Point beamPoint)

	ElectronTkIsolation (double extRadius, double intRadiusBarrel, double intRadiusEndcap, double stripBarrel, double stripEndcap, double ptLow, double lip, double drb, const reco::TrackCollection *, reco::TrackBase::Point beamPoint, const std::string &)

	ElectronTkIsolation (double extRadius, double intRadiusBarrel, double intRadiusEndcap, double stripBarrel, double stripEndcap, double ptLow, double lip, double drb, const reco::TrackCollection *trackCollection, reco::TrackBase::Point beamPoint)

std::pair< int, double >	getIso (const reco::GsfElectron *) const

std::pair< int, double >	getIso (const reco::Track *) const

int	getNumberTracks (const reco::GsfElectron *) const

double	getPtTracks (const reco::GsfElectron *) const

void	setDzOption (const std::string &s)

	~ElectronTkIsolation ()

Private Member Functions
bool	passAlgo (const reco::TrackBase &trk) const

void	setAlgosToReject ()

Private Attributes
std::vector< int >	algosToReject_

reco::TrackBase::Point	beamPoint_

double	drb_

int	dzOption_

double	extRadius_

double	intRadiusBarrel_

double	intRadiusEndcap_

double	lip_

double	ptLow_

double	stripBarrel_

double	stripEndcap_

const reco::TrackCollection *	trackCollection_

Detailed Description

Definition at line 19 of file ElectronTkIsolation.h.

Constructor & Destructor Documentation

◆ ElectronTkIsolation() [1/3]

ElectronTkIsolation::ElectronTkIsolation	(	double	extRadius,
		double	intRadius,
		double	ptLow,
		double	lip,
		double	drb,
		const reco::TrackCollection *	trackCollection,
		reco::TrackBase::Point	beamPoint
	)

inline

Definition at line 22 of file ElectronTkIsolation.h.

       : extRadius_(extRadius),
         intRadiusBarrel_(intRadius),
         intRadiusEndcap_(intRadius),
         stripBarrel_(0.0),
         stripEndcap_(0.0),
         ptLow_(ptLow),
         lip_(lip),
         drb_(drb),
         trackCollection_(trackCollection),
         beamPoint_(beamPoint) {
     setAlgosToReject();
     setDzOption("vz");
   }

References setAlgosToReject(), and setDzOption().

◆ ElectronTkIsolation() [2/3]

ElectronTkIsolation::ElectronTkIsolation	(	double	extRadius,
		double	intRadiusBarrel,
		double	intRadiusEndcap,
		double	stripBarrel,
		double	stripEndcap,
		double	ptLow,
		double	lip,
		double	drb,
		const reco::TrackCollection *	trackCollection,
		reco::TrackBase::Point	beamPoint
	)

inline

Definition at line 43 of file ElectronTkIsolation.h.

       : extRadius_(extRadius),
         intRadiusBarrel_(intRadiusBarrel),
         intRadiusEndcap_(intRadiusEndcap),
         stripBarrel_(stripBarrel),
         stripEndcap_(stripEndcap),
         ptLow_(ptLow),
         lip_(lip),
         drb_(drb),
         trackCollection_(trackCollection),
         beamPoint_(beamPoint) {
     setAlgosToReject();
     setDzOption("vz");
   }

References setAlgosToReject(), and setDzOption().

◆ ElectronTkIsolation() [3/3]

ElectronTkIsolation::ElectronTkIsolation	(	double	extRadius,
		double	intRadiusBarrel,
		double	intRadiusEndcap,
		double	stripBarrel,
		double	stripEndcap,
		double	ptLow,
		double	lip,
		double	drb,
		const reco::TrackCollection *	trackCollection,
		reco::TrackBase::Point	beamPoint,
		const std::string &	dzOptionString
	)

Definition at line 14 of file ElectronTkIsolation.cc.

     : extRadius_(extRadius),
       intRadiusBarrel_(intRadiusBarrel),
       intRadiusEndcap_(intRadiusEndcap),
       stripBarrel_(stripBarrel),
       stripEndcap_(stripEndcap),
       ptLow_(ptLow),
       lip_(lip),
       drb_(drb),
       trackCollection_(trackCollection),
       beamPoint_(beamPoint) {
   setAlgosToReject();
   setDzOption(dzOptionString);
 }

References setAlgosToReject(), and setDzOption().

◆ ~ElectronTkIsolation()

ElectronTkIsolation::~ElectronTkIsolation ( )

Definition at line 39 of file ElectronTkIsolation.cc.

39 {}

Member Function Documentation

◆ getIso() [1/2]

std::pair< int, double > ElectronTkIsolation::getIso ( const reco::GsfElectron * electron ) const

Definition at line 41 of file ElectronTkIsolation.cc.

                                                                                       {
   return getIso(&(*(electron->gsfTrack())));
 }

References HPSPFTauProducerPuppi_cfi::electron.

Referenced by getNumberTracks(), getPtTracks(), and EgammaHLTElectronTrackIsolationProducers::produce().

◆ getIso() [2/2]

std::pair< int, double > ElectronTkIsolation::getIso ( const reco::Track * tmpTrack ) const

Definition at line 46 of file ElectronTkIsolation.cc.

                                                                                 {
   int counter = 0;
   double ptSum = 0.;
   //Take the electron track
   math::XYZVector tmpElectronMomentumAtVtx = (*tmpTrack).momentum();
   double tmpElectronEtaAtVertex = (*tmpTrack).eta();
  
   for (reco::TrackCollection::const_iterator itrTr = (*trackCollection_).begin(); itrTr != (*trackCollection_).end();
        ++itrTr) {
     double this_pt = (*itrTr).pt();
     if (this_pt < ptLow_)
       continue;
  
     double dzCut = 0;
     switch (dzOption_) {
       case egammaisolation::EgammaTrackSelector::dz:
         dzCut = fabs((*itrTr).dz() - (*tmpTrack).dz());
         break;
       case egammaisolation::EgammaTrackSelector::vz:
         dzCut = fabs((*itrTr).vz() - (*tmpTrack).vz());
         break;
       case egammaisolation::EgammaTrackSelector::bs:
         dzCut = fabs((*itrTr).dz(beamPoint_) - (*tmpTrack).dz(beamPoint_));
         break;
       case egammaisolation::EgammaTrackSelector::vtx:
         dzCut = fabs((*itrTr).dz(tmpTrack->vertex()));
         break;
       default:
         dzCut = fabs((*itrTr).vz() - (*tmpTrack).vz());
         break;
     }
     if (dzCut > lip_)
       continue;
     if (fabs((*itrTr).dxy(beamPoint_)) > drb_)
       continue;
     double dr = ROOT::Math::VectorUtil::DeltaR(itrTr->momentum(), tmpElectronMomentumAtVtx);
     double deta = (*itrTr).eta() - tmpElectronEtaAtVertex;
     bool isBarrel = std::abs(tmpElectronEtaAtVertex) < 1.479;
     double intRadius = isBarrel ? intRadiusBarrel_ : intRadiusEndcap_;
     double strip = isBarrel ? stripBarrel_ : stripEndcap_;
     if (dr < extRadius_ && dr >= intRadius && std::abs(deta) >= strip && passAlgo(*itrTr)) {
       ++counter;
       ptSum += this_pt;
     }
  
   }  //end loop over tracks
  
   std::pair<int, double> retval;
   retval.first = counter;
   retval.second = ptSum;
  
   return retval;
 }

References funct::abs(), beamPoint_, egammaisolation::EgammaTrackSelector::bs, counter, electronAnalyzer_cfi::DeltaR, flavorHistoryFilter_cfi::dr, drb_, egammaisolation::EgammaTrackSelector::dz, TrackSplittingMonitor_cfi::dzCut, dzOption_, electronHcalTowerIsolationLcone_cfi::intRadius, intRadiusBarrel_, intRadiusEndcap_, PixelPluginsPhase0_cfi::isBarrel, lip_, passAlgo(), ptLow_, digitizers_cfi::strip, stripBarrel_, stripEndcap_, reco::TrackBase::vertex(), egammaisolation::EgammaTrackSelector::vtx, and egammaisolation::EgammaTrackSelector::vz.

◆ getNumberTracks()

int ElectronTkIsolation::getNumberTracks ( const reco::GsfElectron * electron ) const

Definition at line 100 of file ElectronTkIsolation.cc.

                                                                               {
   //counter for the tracks in the isolation cone
   return getIso(electron).first;
 }

References HPSPFTauProducerPuppi_cfi::electron, and getIso().

◆ getPtTracks()

double ElectronTkIsolation::getPtTracks ( const reco::GsfElectron * electron ) const

Definition at line 105 of file ElectronTkIsolation.cc.

105 { return getIso(electron).second; }

References HPSPFTauProducerPuppi_cfi::electron, and getIso().

Referenced by EgammaElectronTkIsolationProducer::produce().

◆ passAlgo()

bool ElectronTkIsolation::passAlgo ( const reco::TrackBase & trk ) const

private

Definition at line 107 of file ElectronTkIsolation.cc.

                                                                  {
   int algo = trk.algo();
   bool rejAlgo = std::binary_search(algosToReject_.begin(), algosToReject_.end(), algo);
   return rejAlgo == false;
 }

References reco::TrackBase::algo(), and algosToReject_.

Referenced by getIso().

◆ setAlgosToReject()

void ElectronTkIsolation::setAlgosToReject ( )

private

Definition at line 113 of file ElectronTkIsolation.cc.

                                            {
   algosToReject_ = {reco::TrackBase::jetCoreRegionalStep};
   std::sort(algosToReject_.begin(), algosToReject_.end());
 }

References algosToReject_, reco::TrackBase::jetCoreRegionalStep, and jetUpdater_cfi::sort.

Referenced by ElectronTkIsolation().

◆ setDzOption()

void ElectronTkIsolation::setDzOption ( const std::string & s )

inline

Definition at line 84 of file ElectronTkIsolation.h.

                                        {
     if (!s.compare("dz"))
       dzOption_ = egammaisolation::EgammaTrackSelector::dz;
     else if (!s.compare("vz"))
       dzOption_ = egammaisolation::EgammaTrackSelector::vz;
     else if (!s.compare("bs"))
       dzOption_ = egammaisolation::EgammaTrackSelector::bs;
     else if (!s.compare("vtx"))
       dzOption_ = egammaisolation::EgammaTrackSelector::vtx;
     else
       dzOption_ = egammaisolation::EgammaTrackSelector::dz;
   }

References egammaisolation::EgammaTrackSelector::bs, egammaisolation::EgammaTrackSelector::dz, dzOption_, alignCSCRings::s, egammaisolation::EgammaTrackSelector::vtx, and egammaisolation::EgammaTrackSelector::vz.

Referenced by ElectronTkIsolation().

Member Data Documentation

◆ algosToReject_

std::vector<int> ElectronTkIsolation::algosToReject_

private

Definition at line 113 of file ElectronTkIsolation.h.

Referenced by passAlgo(), and setAlgosToReject().

◆ beamPoint_

reco::TrackBase::Point ElectronTkIsolation::beamPoint_

private

Definition at line 115 of file ElectronTkIsolation.h.

Referenced by getIso().

◆ drb_

double ElectronTkIsolation::drb_

private

Definition at line 112 of file ElectronTkIsolation.h.

Referenced by getIso().

◆ dzOption_

int ElectronTkIsolation::dzOption_

private

Definition at line 117 of file ElectronTkIsolation.h.

Referenced by getIso(), and setDzOption().

◆ extRadius_

double ElectronTkIsolation::extRadius_

private

Definition at line 105 of file ElectronTkIsolation.h.

◆ intRadiusBarrel_

double ElectronTkIsolation::intRadiusBarrel_

private

Definition at line 106 of file ElectronTkIsolation.h.

Referenced by getIso().

◆ intRadiusEndcap_

double ElectronTkIsolation::intRadiusEndcap_

private

Definition at line 107 of file ElectronTkIsolation.h.

Referenced by getIso().

◆ lip_

double ElectronTkIsolation::lip_

private

Definition at line 111 of file ElectronTkIsolation.h.

Referenced by getIso().

◆ ptLow_

double ElectronTkIsolation::ptLow_

private

Definition at line 110 of file ElectronTkIsolation.h.

Referenced by getIso().

◆ stripBarrel_

double ElectronTkIsolation::stripBarrel_

private

Definition at line 108 of file ElectronTkIsolation.h.

Referenced by getIso().

◆ stripEndcap_

double ElectronTkIsolation::stripEndcap_

private

Definition at line 109 of file ElectronTkIsolation.h.

Referenced by getIso().

◆ trackCollection_

const reco::TrackCollection* ElectronTkIsolation::trackCollection_

private

Definition at line 114 of file ElectronTkIsolation.h.

Public Member Functions

Private Member Functions

Private Attributes

Detailed Description

Constructor & Destructor Documentation

◆ ElectronTkIsolation() [1/3]

◆ ElectronTkIsolation() [2/3]

◆ ElectronTkIsolation() [3/3]

◆ ~ElectronTkIsolation()

Member Function Documentation

◆ getIso() [1/2]

◆ getIso() [2/2]

◆ getNumberTracks()

◆ getPtTracks()

◆ passAlgo()

◆ setAlgosToReject()

◆ setDzOption()

Member Data Documentation

◆ algosToReject_

◆ beamPoint_

◆ drb_

◆ dzOption_

◆ extRadius_

◆ intRadiusBarrel_

◆ intRadiusEndcap_

◆ lip_

◆ ptLow_

◆ stripBarrel_

◆ stripEndcap_

◆ trackCollection_