TrackWithVertexSelector Class Reference

#include <TrackWithVertexSelector.h>

Public Member Functions
void	init (const edm::Event &event, const edm::EventSetup &)
void	init (const edm::Event &event)
bool	operator() (const reco::Track &t) const
bool	operator() (const reco::Track &t, const edm::Event &iEvent)
bool	operator() (const reco::TrackRef &t) const
bool	operator() (const reco::TrackRef &t, const edm::Event &iEvent)
bool	testTrack (const reco::Track &t) const
bool	testTrack (const reco::TrackRef &t) const
bool	testVertices (const reco::Track &t, const reco::VertexCollection &vtxs) const
bool	testVertices (const reco::TrackRef &t, const reco::VertexCollection &vtxs) const
	TrackWithVertexSelector (const edm::ParameterSet &iConfig, edm::ConsumesCollector &&iC)
	TrackWithVertexSelector (const edm::ParameterSet &iConfig, edm::ConsumesCollector &iC)
	~TrackWithVertexSelector ()

Private Types
typedef math::XYZPoint	Point

Private Attributes
double	d0Max_
double	dzMax_
double	etaMax_
double	etaMin_
double	normalizedChi2_
double	nSigmaDtVertex_
uint32_t	numberOfLostHits_
uint32_t	numberOfValidHits_
uint32_t	numberOfValidPixelHits_
uint32_t	nVertices_
double	ptErrorCut_
double	ptMax_
double	ptMin_
std::string	quality_
double	rhoVtx_
edm::ValueMap< float > const *	timeresoscoll_ = 0
edm::EDGetTokenT < edm::ValueMap< float > >	timeResosToken_
edm::ValueMap< float > const *	timescoll_ = 0
edm::EDGetTokenT < edm::ValueMap< float > >	timesToken_
reco::VertexCollection const *	vcoll_ = 0
edm::EDGetTokenT < reco::VertexCollection >	vertexToken_
bool	vtxFallback_
double	zetaVtx_

Detailed Description

Definition at line 18 of file TrackWithVertexSelector.h.

Member Typedef Documentation

typedef math::XYZPoint TrackWithVertexSelector::Point

private

Definition at line 61 of file TrackWithVertexSelector.h.

Constructor & Destructor Documentation

TrackWithVertexSelector::TrackWithVertexSelector ( const edm::ParameterSet &  iConfig, edm::ConsumesCollector &&  iC  )

inlineexplicit

Definition at line 20 of file TrackWithVertexSelector.h.

                                                                                                   :
        TrackWithVertexSelector(iConfig, iC) {}

TrackWithVertexSelector::TrackWithVertexSelector ( const edm::ParameterSet &  iConfig, edm::ConsumesCollector &  iC  )

explicit

Definition at line 11 of file TrackWithVertexSelector.cc.

                                                                                                          :
  numberOfValidHits_(iConfig.getParameter<uint32_t>("numberOfValidHits")),
  numberOfValidPixelHits_(iConfig.getParameter<uint32_t>("numberOfValidPixelHits")),
  numberOfLostHits_(iConfig.getParameter<uint32_t>("numberOfLostHits")),
  normalizedChi2_(iConfig.getParameter<double>("normalizedChi2")),
  ptMin_(iConfig.getParameter<double>("ptMin")),
  ptMax_(iConfig.getParameter<double>("ptMax")),
  etaMin_(iConfig.getParameter<double>("etaMin")),
  etaMax_(iConfig.getParameter<double>("etaMax")),
  dzMax_(iConfig.getParameter<double>("dzMax")),
  d0Max_(iConfig.getParameter<double>("d0Max")),
  ptErrorCut_(iConfig.getParameter<double>("ptErrorCut")),
  quality_(iConfig.getParameter<std::string>("quality")),
  nVertices_(iConfig.getParameter<bool>("useVtx") ? iConfig.getParameter<uint32_t>("nVertices") : 0),
  vertexToken_(iC.consumes<reco::VertexCollection>(iConfig.getParameter<edm::InputTag>("vertexTag"))),
  timesToken_(iC.consumes<edm::ValueMap<float> >(iConfig.getParameter<edm::InputTag>("timesTag"))),
  timeResosToken_(iC.consumes<edm::ValueMap<float> >(iConfig.getParameter<edm::InputTag>("timeResosTag"))),
  vtxFallback_(iConfig.getParameter<bool>("vtxFallback")),
  zetaVtx_(iConfig.getParameter<double>("zetaVtx")),
  rhoVtx_(iConfig.getParameter<double>("rhoVtx")),
  nSigmaDtVertex_(iConfig.getParameter<double>("nSigmaDtVertex")) {
}

TrackWithVertexSelector::~TrackWithVertexSelector

(

)

Definition at line 34 of file TrackWithVertexSelector.cc.

{  }

Member Function Documentation

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

inline

Definition at line 25 of file TrackWithVertexSelector.h.

References init().

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

{init(event);}

void TrackWithVertexSelector::init

(

const edm::Event &

event

)

Definition at line 37 of file TrackWithVertexSelector.cc.

References edm::HandleBase::isValid(), edm::Handle< T >::product(), timeresoscoll_, timeResosToken_, timescoll_, timesToken_, vcoll_, and vertexToken_.

                                                         {
    edm::Handle<reco::VertexCollection> hVtx;
    event.getByToken(vertexToken_, hVtx);
    vcoll_ = hVtx.product();

    edm::Handle<edm::ValueMap<float> > hTimes;
    event.getByToken(timesToken_, hTimes);
    timescoll_ = hTimes.isValid() ? hTimes.product() : nullptr;

    edm::Handle<edm::ValueMap<float> > hTimeResos;
    event.getByToken(timeResosToken_, hTimeResos);
    timeresoscoll_ = hTimeResos.isValid() ? hTimeResos.product() : nullptr;
}

bool TrackWithVertexSelector::operator()

(

const reco::Track &

t

)

const

Definition at line 126 of file TrackWithVertexSelector.cc.

References nVertices_, testTrack(), testVertices(), and vcoll_.

                                                                 {
  if (!testTrack(t)) return false;
  if (nVertices_ == 0) return true;
  return testVertices(t, *vcoll_);
}

bool TrackWithVertexSelector::operator() ( const reco::Track &  t, const edm::Event &  iEvent  )

inline

Definition at line 29 of file TrackWithVertexSelector.h.

References init(), and t.

                                                                  {
         init(iEvent); return (*this)(t);   
      }

bool TrackWithVertexSelector::operator()

(

const reco::TrackRef &

t

)

const

Definition at line 132 of file TrackWithVertexSelector.cc.

References nVertices_, testTrack(), testVertices(), and vcoll_.

                                                                       {
  if (!testTrack(tref)) return false;
  if (nVertices_ == 0) return true;
  return testVertices(tref, *vcoll_);
}

bool TrackWithVertexSelector::operator() ( const reco::TrackRef &  t, const edm::Event &  iEvent  )

inline

Definition at line 34 of file TrackWithVertexSelector.h.

References init(), and t.

                                                                     {
         init(iEvent); return (*this)(t);   
      }

bool TrackWithVertexSelector::testTrack

(

const reco::Track &

t

)

const

Definition at line 51 of file TrackWithVertexSelector.cc.

References funct::abs(), reco::TrackBase::d0(), d0Max_, reco::TrackBase::dz(), dzMax_, reco::TrackBase::eta(), etaMax_, etaMin_, reco::TrackBase::hitPattern(), bookConverter::max, reco::TrackBase::normalizedChi2(), normalizedChi2_, reco::TrackBase::numberOfLostHits(), numberOfLostHits_, reco::TrackBase::numberOfValidHits(), numberOfValidHits_, reco::HitPattern::numberOfValidPixelHits(), numberOfValidPixelHits_, reco::TrackBase::pt(), reco::TrackBase::ptError(), ptErrorCut_, ptMax_, ptMin_, reco::TrackBase::quality(), quality_, and reco::TrackBase::qualityByName().

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

                                                                {
  using std::abs;
  if ((t.numberOfValidHits() >= numberOfValidHits_) &&
      (static_cast<unsigned int>(t.hitPattern().numberOfValidPixelHits()) >= numberOfValidPixelHits_) &&
      (t.numberOfLostHits() <= numberOfLostHits_) &&
      (t.normalizedChi2()    <= normalizedChi2_) &&
      (t.ptError()/t.pt()*std::max(1.,t.normalizedChi2()) <= ptErrorCut_) &&
      (t.quality(t.qualityByName(quality_))) &&
      (t.pt()              >= ptMin_)      &&
      (t.pt()              <= ptMax_)      &&
      (abs(t.eta())   <= etaMax_)     &&
      (abs(t.eta())   >= etaMin_)     &&
      (abs(t.dz())    <= dzMax_)      &&
      (abs(t.d0())    <= d0Max_)  ) {
    return true;
  }
  return false;
}

bool TrackWithVertexSelector::testTrack

(

const reco::TrackRef &

t

)

const

Definition at line 70 of file TrackWithVertexSelector.cc.

References testTrack().

                                                                      {
  return testTrack(*tref);
}

bool TrackWithVertexSelector::testVertices	(	const reco::Track &	t,
		const reco::VertexCollection &	vtxs
	)		const

Definition at line 74 of file TrackWithVertexSelector.cc.

References funct::abs(), reco::TrackBase::dxy(), reco::TrackBase::dz(), nVertices_, convertSQLiteXML::ok, rhoVtx_, reco::TrackBase::vertex(), vtxFallback_, and zetaVtx_.

Referenced by operator()().

                                                                                                     {
  bool ok = false;
  if (vtxs.size() > 0) {
    unsigned int tested = 1;
    for (reco::VertexCollection::const_iterator it = vtxs.begin(), ed = vtxs.end();
     it != ed; ++it) {
      if ( (std::abs(t.dxy(it->position())) < rhoVtx_) &&
           (std::abs(t.dz(it->position())) < zetaVtx_)  ) {
        ok = true; break;
      }
      if (tested++ >= nVertices_) break;
    }
  } else if (vtxFallback_) {
    return ( (std::abs(t.vertex().z()) < 15.9) && (t.vertex().Rho() < 0.2) );
  }
  return ok;
}

bool TrackWithVertexSelector::testVertices	(	const reco::TrackRef &	t,
		const reco::VertexCollection &	vtxs
	)		const

Definition at line 92 of file TrackWithVertexSelector.cc.

References funct::abs(), alignCSCRings::e, edm::isNotFinite(), nSigmaDtVertex_, nVertices_, convertSQLiteXML::ok, rhoVtx_, mathSSE::sqrt(), t, timeresoscoll_, timescoll_, vtxFallback_, and zetaVtx_.

                                                                                                           {
  const auto& t = *tref;
  const bool timeAvailable = timescoll_ != nullptr && timeresoscoll_ != nullptr;
  bool ok = false;
  if (vtxs.size() > 0) {
    unsigned int tested = 1;
    for (reco::VertexCollection::const_iterator it = vtxs.begin(), ed = vtxs.end();
     it != ed; ++it) {
      const bool useTime = timeAvailable && it->t() != 0.;
      float time = useTime ? (*timescoll_)[tref] : -1.f;
      float timeReso = useTime ? (*timeresoscoll_)[tref] : -1.f;
      timeReso = ( timeReso > 1e-6 ? timeReso : fakeBeamSpotTimeWidth );

      if( edm::isNotFinite(time) ) {
    time = 0.0;
    timeReso = 2.0*fakeBeamSpotTimeWidth;
      }

      const double vtxSigmaT2 = it->tError() * it->tError();
      const double vtxTrackErr = std::sqrt( vtxSigmaT2 + timeReso*timeReso );

      if ( (std::abs(t.dxy(it->position())) < rhoVtx_) &&
           (std::abs(t.dz(it->position())) < zetaVtx_) &&
       ( !useTime || (std::abs(time - it->t())/vtxTrackErr < nSigmaDtVertex_) ) ) {
        ok = true; break;
      }
      if (tested++ >= nVertices_) break;
    }
  } else if (vtxFallback_) {
    return ( (std::abs(t.vertex().z()) < 15.9) && (t.vertex().Rho() < 0.2) );
  }
  return ok;
}

Member Data Documentation

double TrackWithVertexSelector::d0Max_

private

Definition at line 48 of file TrackWithVertexSelector.h.

Referenced by testTrack().

double TrackWithVertexSelector::dzMax_

private

Definition at line 48 of file TrackWithVertexSelector.h.

Referenced by testTrack().

double TrackWithVertexSelector::etaMax_

private

Definition at line 47 of file TrackWithVertexSelector.h.

Referenced by testTrack().

double TrackWithVertexSelector::etaMin_

private

Definition at line 47 of file TrackWithVertexSelector.h.

Referenced by testTrack().

double TrackWithVertexSelector::normalizedChi2_

private

Definition at line 46 of file TrackWithVertexSelector.h.

Referenced by testTrack().

double TrackWithVertexSelector::nSigmaDtVertex_

private

Definition at line 56 of file TrackWithVertexSelector.h.

Referenced by testVertices().

uint32_t TrackWithVertexSelector::numberOfLostHits_

private

Definition at line 45 of file TrackWithVertexSelector.h.

Referenced by testTrack().

uint32_t TrackWithVertexSelector::numberOfValidHits_

private

Definition at line 43 of file TrackWithVertexSelector.h.

Referenced by testTrack().

uint32_t TrackWithVertexSelector::numberOfValidPixelHits_

private

Definition at line 44 of file TrackWithVertexSelector.h.

Referenced by testTrack().

uint32_t TrackWithVertexSelector::nVertices_

private

Definition at line 52 of file TrackWithVertexSelector.h.

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

double TrackWithVertexSelector::ptErrorCut_

private

Definition at line 49 of file TrackWithVertexSelector.h.

Referenced by testTrack().

double TrackWithVertexSelector::ptMax_

private

Definition at line 47 of file TrackWithVertexSelector.h.

Referenced by testTrack().

double TrackWithVertexSelector::ptMin_

private

Definition at line 47 of file TrackWithVertexSelector.h.

Referenced by testTrack().

std::string TrackWithVertexSelector::quality_

private

Definition at line 50 of file TrackWithVertexSelector.h.

Referenced by testTrack().

double TrackWithVertexSelector::rhoVtx_

private

Definition at line 56 of file TrackWithVertexSelector.h.

Referenced by testVertices().

edm::ValueMap<float> const* TrackWithVertexSelector::timeresoscoll_ = 0

private

Definition at line 60 of file TrackWithVertexSelector.h.

Referenced by init(), and testVertices().

edm::EDGetTokenT<edm::ValueMap<float> > TrackWithVertexSelector::timeResosToken_

private

Definition at line 54 of file TrackWithVertexSelector.h.

Referenced by init().

edm::ValueMap<float> const* TrackWithVertexSelector::timescoll_ = 0

private

Definition at line 59 of file TrackWithVertexSelector.h.

Referenced by init(), and testVertices().

edm::EDGetTokenT<edm::ValueMap<float> > TrackWithVertexSelector::timesToken_

private

Definition at line 54 of file TrackWithVertexSelector.h.

Referenced by init().

reco::VertexCollection const* TrackWithVertexSelector::vcoll_ = 0

private

Definition at line 58 of file TrackWithVertexSelector.h.

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

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

private

Definition at line 53 of file TrackWithVertexSelector.h.

Referenced by init().

bool TrackWithVertexSelector::vtxFallback_

private

Definition at line 55 of file TrackWithVertexSelector.h.

Referenced by testVertices().

double TrackWithVertexSelector::zetaVtx_

private

Definition at line 56 of file TrackWithVertexSelector.h.

Referenced by testVertices().