RecoTrackSelectorBase Class Reference

#include <RecoTrackSelectorBase.h>

Inheritance diagram for RecoTrackSelectorBase:

Public Member Functions
void	init (const edm::Event &event, const edm::EventSetup &es)
bool	operator() (const reco::TrackRef &tref) const
bool	operator() (const reco::Track &t) const
	RecoTrackSelectorBase ()
	RecoTrackSelectorBase (const edm::ParameterSet &cfg, edm::ConsumesCollector &iC)

Private Attributes
std::vector < reco::TrackBase::TrackAlgorithm >	algorithm_
std::vector < reco::TrackBase::TrackAlgorithm >	algorithmMask_
edm::EDGetTokenT< reco::BeamSpot >	bsSrcToken_
double	lip_
double	maxChi2_
double	maxRapidity_
int	min3DLayer_
int	minHit_
int	minLayer_
int	minPixelHit_
double	minRapidity_
std::vector < reco::TrackBase::TrackAlgorithm >	originalAlgorithm_
double	ptMin_
std::vector < reco::TrackBase::TrackQuality >	quality_
double	tip_
bool	usePV_
reco::Track::Point	vertex_
edm::EDGetTokenT < reco::VertexCollection >	vertexToken_

Detailed Description

Definition at line 14 of file RecoTrackSelectorBase.h.

Constructor & Destructor Documentation

RecoTrackSelectorBase::RecoTrackSelectorBase ( )

inline

Definition at line 16 of file RecoTrackSelectorBase.h.

{}

RecoTrackSelectorBase::RecoTrackSelectorBase ( const edm::ParameterSet &  cfg, edm::ConsumesCollector &  iC  )

inline

Definition at line 17 of file RecoTrackSelectorBase.h.

References reco::TrackBase::algoByName(), HLT_FULL_cff::algorithm, algorithm_, algorithmMask_, edm::ConsumesCollector::consumes(), edm::ParameterSet::getParameter(), originalAlgorithm_, TrackCollections2monitor_cff::quality, quality_, reco::TrackBase::qualityByName(), AlCaHLTBitMon_QueryRunRegistry::string, usePV_, and vertexToken_.

                                                                               :
    ptMin_(cfg.getParameter<double>("ptMin")),
    minRapidity_(cfg.getParameter<double>("minRapidity")),
    maxRapidity_(cfg.getParameter<double>("maxRapidity")),
    tip_(cfg.getParameter<double>("tip")),
    lip_(cfg.getParameter<double>("lip")),
    maxChi2_(cfg.getParameter<double>("maxChi2")),
    minHit_(cfg.getParameter<int>("minHit")),
    minPixelHit_(cfg.getParameter<int>("minPixelHit")),
    minLayer_(cfg.getParameter<int>("minLayer")),
    min3DLayer_(cfg.getParameter<int>("min3DLayer")),
    usePV_(cfg.getParameter<bool>("usePV")),
    bsSrcToken_(iC.consumes<reco::BeamSpot>(cfg.getParameter<edm::InputTag>("beamSpot"))) {
      if (usePV_)
        vertexToken_ = iC.consumes<reco::VertexCollection>(cfg.getParameter<edm::InputTag>("vertexTag"));
      for(const std::string& quality: cfg.getParameter<std::vector<std::string> >("quality"))
        quality_.push_back(reco::TrackBase::qualityByName(quality));
      for(const std::string& algorithm: cfg.getParameter<std::vector<std::string> >("algorithm"))
        algorithm_.push_back(reco::TrackBase::algoByName(algorithm));
      for(const std::string& algorithm: cfg.getParameter<std::vector<std::string> >("originalAlgorithm"))
        originalAlgorithm_.push_back(reco::TrackBase::algoByName(algorithm));
      for(const std::string& algorithm: cfg.getParameter<std::vector<std::string> >("algorithmMaskContains"))
        algorithmMask_.push_back(reco::TrackBase::algoByName(algorithm));
    }

Member Function Documentation

void RecoTrackSelectorBase::init ( const edm::Event &  event, const edm::EventSetup &  es  )

inline

Definition at line 42 of file RecoTrackSelectorBase.h.

References SiPixelRawToDigiRegional_cfi::beamSpot, bsSrcToken_, usePV_, vertex_, and vertexToken_.

Referenced by RecoTrackViewRefSelector::select(), RecoTrackRefSelector::select(), and RecoTrackSelector::select().

                                                            {
     edm::Handle<reco::BeamSpot> beamSpot;
     event.getByToken(bsSrcToken_,beamSpot);
     vertex_ = beamSpot->position();
     if (!usePV_) return;
     edm::Handle<reco::VertexCollection> hVtx;
     event.getByToken(vertexToken_, hVtx);
     if (hVtx->empty()) return;
     vertex_ = (*hVtx)[0].position();
  }

bool RecoTrackSelectorBase::operator() ( const reco::TrackRef &  tref ) const

inline

Definition at line 53 of file RecoTrackSelectorBase.h.

                                                    {
    return (*this)(*tref);
  }

bool RecoTrackSelectorBase::operator() ( const reco::Track &  t ) const

inline

Definition at line 57 of file RecoTrackSelectorBase.h.

References ecalcalib_dqm_sourceclient-live_cfg::algo, reco::TrackBase::algo(), reco::TrackBase::algoMask(), algorithm_, algorithmMask_, reco::TrackBase::dsz(), reco::TrackBase::dxy(), reco::TrackBase::eta(), spr::find(), reco::TrackBase::hitPattern(), i, lip_, maxChi2_, maxRapidity_, min3DLayer_, minHit_, minLayer_, minPixelHit_, minRapidity_, reco::TrackBase::normalizedChi2(), reco::HitPattern::numberOfValidHits(), reco::HitPattern::numberOfValidPixelHits(), reco::HitPattern::numberOfValidStripLayersWithMonoAndStereo(), reco::TrackBase::originalAlgo(), originalAlgorithm_, reco::HitPattern::pixelLayersWithMeasurement(), reco::TrackBase::pt(), ptMin_, reco::TrackBase::quality(), quality_, tip_, reco::HitPattern::trackerLayersWithMeasurement(), and vertex_.

                                              {
    bool quality_ok = true;
    if (quality_.size()!=0) {
      quality_ok = false;
      for (unsigned int i = 0; i<quality_.size();++i) {
    if (t.quality(quality_[i])){
      quality_ok = true;
      break;
    }
      }
    }

    bool algo_ok = true;
    if (algorithm_.size()!=0) {
      if (std::find(algorithm_.begin(),algorithm_.end(),t.algo())==algorithm_.end()) algo_ok = false;
    }
    if (!originalAlgorithm_.empty() && algo_ok) {
      if (std::find(originalAlgorithm_.begin(), originalAlgorithm_.end(), t.originalAlgo()) == originalAlgorithm_.end()) algo_ok = false;
    }
    if(!algorithmMask_.empty() && algo_ok) {
      if(std::find_if(algorithmMask_.begin(), algorithmMask_.end(), [&](reco::TrackBase::TrackAlgorithm algo) -> bool { // for some reason I have to either explicitly give the return type, or use static_cast<bool>()
            return t.algoMask()[algo];
          }) == algorithmMask_.end()) algo_ok = false;
    }
    return
      (
       (algo_ok & quality_ok) &&
       t.hitPattern().numberOfValidHits() >= minHit_ &&
       t.hitPattern().numberOfValidPixelHits() >= minPixelHit_ &&
       t.hitPattern().trackerLayersWithMeasurement() >= minLayer_ &&
       t.hitPattern().pixelLayersWithMeasurement() +
       t.hitPattern().numberOfValidStripLayersWithMonoAndStereo() >= min3DLayer_ &&
       fabs(t.pt()) >= ptMin_ &&
       t.eta() >= minRapidity_ && t.eta() <= maxRapidity_ &&
       fabs(t.dxy(vertex_)) <= tip_ &&
       fabs(t.dsz(vertex_)) <= lip_  &&
       t.normalizedChi2()<=maxChi2_
      );
  }

Member Data Documentation

std::vector<reco::TrackBase::TrackAlgorithm> RecoTrackSelectorBase::algorithm_

private

Definition at line 115 of file RecoTrackSelectorBase.h.

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

std::vector<reco::TrackBase::TrackAlgorithm> RecoTrackSelectorBase::algorithmMask_

private

Definition at line 117 of file RecoTrackSelectorBase.h.

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

edm::EDGetTokenT<reco::BeamSpot> RecoTrackSelectorBase::bsSrcToken_

private

Definition at line 111 of file RecoTrackSelectorBase.h.

Referenced by init().

double RecoTrackSelectorBase::lip_

private

Definition at line 103 of file RecoTrackSelectorBase.h.

Referenced by operator()().

double RecoTrackSelectorBase::maxChi2_

private

Definition at line 104 of file RecoTrackSelectorBase.h.

Referenced by operator()().

double RecoTrackSelectorBase::maxRapidity_

private

Definition at line 101 of file RecoTrackSelectorBase.h.

Referenced by operator()().

int RecoTrackSelectorBase::min3DLayer_

private

Definition at line 108 of file RecoTrackSelectorBase.h.

Referenced by operator()().

int RecoTrackSelectorBase::minHit_

private

Definition at line 105 of file RecoTrackSelectorBase.h.

Referenced by operator()().

int RecoTrackSelectorBase::minLayer_

private

Definition at line 107 of file RecoTrackSelectorBase.h.

Referenced by operator()().

int RecoTrackSelectorBase::minPixelHit_

private

Definition at line 106 of file RecoTrackSelectorBase.h.

Referenced by operator()().

double RecoTrackSelectorBase::minRapidity_

private

Definition at line 100 of file RecoTrackSelectorBase.h.

Referenced by operator()().

std::vector<reco::TrackBase::TrackAlgorithm> RecoTrackSelectorBase::originalAlgorithm_

private

Definition at line 116 of file RecoTrackSelectorBase.h.

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

double RecoTrackSelectorBase::ptMin_

private

Definition at line 99 of file RecoTrackSelectorBase.h.

Referenced by operator()().

std::vector<reco::TrackBase::TrackQuality> RecoTrackSelectorBase::quality_

private

Definition at line 114 of file RecoTrackSelectorBase.h.

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

double RecoTrackSelectorBase::tip_

private

Definition at line 102 of file RecoTrackSelectorBase.h.

Referenced by operator()().

bool RecoTrackSelectorBase::usePV_

private

Definition at line 109 of file RecoTrackSelectorBase.h.

Referenced by init(), and RecoTrackSelectorBase().

reco::Track::Point RecoTrackSelectorBase::vertex_

private

Definition at line 119 of file RecoTrackSelectorBase.h.

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

edm::EDGetTokenT<reco::VertexCollection> RecoTrackSelectorBase::vertexToken_

private

Definition at line 112 of file RecoTrackSelectorBase.h.

Referenced by init(), and RecoTrackSelectorBase().