ElectronTkIsolation Class Reference

#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 *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 &)
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::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.

References setAlgosToReject(), and setDzOption().

      : 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");
  }

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.

References setAlgosToReject(), and setDzOption().

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

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.

References setAlgosToReject(), and setDzOption().

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

ElectronTkIsolation::~ElectronTkIsolation

(

)

Definition at line 39 of file ElectronTkIsolation.cc.

{}

Member Function Documentation

std::pair< int, double > ElectronTkIsolation::getIso

(

const reco::GsfElectron *

electron

)

const

Definition at line 41 of file ElectronTkIsolation.cc.

References reco::GsfElectron::gsfTrack().

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

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

std::pair< int, double > ElectronTkIsolation::getIso

(

const reco::Track *

tmpTrack

)

const

Definition at line 46 of file ElectronTkIsolation.cc.

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

                                                                                {
  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;
}

int ElectronTkIsolation::getNumberTracks

(

const reco::GsfElectron *

electron

)

const

Definition at line 100 of file ElectronTkIsolation.cc.

References getIso().

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

double ElectronTkIsolation::getPtTracks

(

const reco::GsfElectron *

electron

)

const

Definition at line 105 of file ElectronTkIsolation.cc.

References getIso().

Referenced by EgammaElectronTkIsolationProducer::produce().

{ return getIso(electron).second; }

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

private

Definition at line 107 of file ElectronTkIsolation.cc.

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

Referenced by getIso().

                                                                 {
  int algo = trk.algo();
  bool rejAlgo = std::binary_search(algosToReject_.begin(), algosToReject_.end(), algo);
  //check also the originalAlgo in case the track is reconstructed by more than one
  algo = trk.originalAlgo();
  //reject only if both algo and originalAlgo are not acceptable
  rejAlgo &= std::binary_search(algosToReject_.begin(), algosToReject_.end(), algo);
  return !rejAlgo;
}

void ElectronTkIsolation::setAlgosToReject ( )

private

Definition at line 117 of file ElectronTkIsolation.cc.

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

Referenced by ElectronTkIsolation().

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

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

inline

Definition at line 84 of file ElectronTkIsolation.h.

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

Referenced by ElectronTkIsolation().

                                       {
    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;
  }

Member Data Documentation

std::vector<int> ElectronTkIsolation::algosToReject_

private

Definition at line 113 of file ElectronTkIsolation.h.

Referenced by passAlgo(), and setAlgosToReject().

reco::TrackBase::Point ElectronTkIsolation::beamPoint_

private

Definition at line 115 of file ElectronTkIsolation.h.

Referenced by getIso().

double ElectronTkIsolation::drb_

private

Definition at line 112 of file ElectronTkIsolation.h.

Referenced by getIso().

int ElectronTkIsolation::dzOption_

private

Definition at line 117 of file ElectronTkIsolation.h.

Referenced by getIso(), and setDzOption().

double ElectronTkIsolation::extRadius_

private

Definition at line 105 of file ElectronTkIsolation.h.

double ElectronTkIsolation::intRadiusBarrel_

private

Definition at line 106 of file ElectronTkIsolation.h.

Referenced by getIso().

double ElectronTkIsolation::intRadiusEndcap_

private

Definition at line 107 of file ElectronTkIsolation.h.

Referenced by getIso().

double ElectronTkIsolation::lip_

private

Definition at line 111 of file ElectronTkIsolation.h.

Referenced by getIso().

double ElectronTkIsolation::ptLow_

private

Definition at line 110 of file ElectronTkIsolation.h.

Referenced by getIso().

double ElectronTkIsolation::stripBarrel_

private

Definition at line 108 of file ElectronTkIsolation.h.

Referenced by getIso().

double ElectronTkIsolation::stripEndcap_

private

Definition at line 109 of file ElectronTkIsolation.h.

Referenced by getIso().

const reco::TrackCollection* ElectronTkIsolation::trackCollection_

private

Definition at line 114 of file ElectronTkIsolation.h.