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_ = nullptr
edm::EDGetTokenT< edm::ValueMap< float > >	timeResosToken_
edm::ValueMap< float > const *	timescoll_ = nullptr
edm::EDGetTokenT< edm::ValueMap< float > >	timesToken_
reco::VertexCollection const *	vcoll_ = nullptr
edm::EDGetTokenT< reco::VertexCollection >	vertexToken_
bool	vtxFallback_
double	zetaVtx_

Detailed Description

Definition at line 17 of file TrackWithVertexSelector.h.

Member Typedef Documentation

Point

typedef math::XYZPoint TrackWithVertexSelector::Point

private

Definition at line 63 of file TrackWithVertexSelector.h.

Constructor & Destructor Documentation

TrackWithVertexSelector() [1/2]

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

inlineexplicit

Definition at line 19 of file TrackWithVertexSelector.h.

      : TrackWithVertexSelector(iConfig, iC) {}

TrackWithVertexSelector() [2/2]

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::~TrackWithVertexSelector

(

)

Definition at line 33 of file TrackWithVertexSelector.cc.

{}

Member Function Documentation

init() [1/2]

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

inline

Definition at line 24 of file TrackWithVertexSelector.h.

References init().

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

{ init(event); }

init() [2/2]

void TrackWithVertexSelector::init

(

const edm::Event &

event

)

Definition at line 35 of file TrackWithVertexSelector.cc.

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

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

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

operator()() [1/4]

bool TrackWithVertexSelector::operator()

(

const reco::Track &

t

)

const

Definition at line 116 of file TrackWithVertexSelector.cc.

References nVertices_, submitPVValidationJobs::t, testTrack(), testVertices(), and vcoll_.

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

operator()() [2/4]

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

inline

Definition at line 28 of file TrackWithVertexSelector.h.

References iEvent, init(), and submitPVValidationJobs::t.

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

operator()() [3/4]

bool TrackWithVertexSelector::operator()

(

const reco::TrackRef &

t

)

const

Definition at line 124 of file TrackWithVertexSelector.cc.

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

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

operator()() [4/4]

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

inline

Definition at line 34 of file TrackWithVertexSelector.h.

References iEvent, init(), and submitPVValidationJobs::t.

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

testTrack() [1/2]

bool TrackWithVertexSelector::testTrack

(

const reco::Track &

t

)

const

Definition at line 49 of file TrackWithVertexSelector.cc.

References funct::abs(), d0Max_, dzMax_, etaMax_, etaMin_, SiStripPI::max, normalizedChi2_, numberOfLostHits_, numberOfValidHits_, numberOfValidPixelHits_, ptErrorCut_, ptMax_, ptMin_, quality_, and submitPVValidationJobs::t.

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

testTrack() [2/2]

bool TrackWithVertexSelector::testTrack

(

const reco::TrackRef &

t

)

const

Definition at line 62 of file TrackWithVertexSelector.cc.

References testTrack().

{ return testTrack(*tref); }

testVertices() [1/2]

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

Definition at line 64 of file TrackWithVertexSelector.cc.

References funct::abs(), nVertices_, convertSQLiteXML::ok, rhoVtx_, submitPVValidationJobs::t, vtxFallback_, and zetaVtx_.

Referenced by operator()().

                                                                                                     {
  bool ok = false;
  if (!vtxs.empty()) {
    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;
}

testVertices() [2/2]

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

Definition at line 82 of file TrackWithVertexSelector.cc.

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

                                                                                                           {
  const auto &t = *tref;
  const bool timeAvailable = timescoll_ != nullptr && timeresoscoll_ != nullptr;
  bool ok = false;
  if (!vtxs.empty()) {
    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

d0Max_

double TrackWithVertexSelector::d0Max_

private

Definition at line 50 of file TrackWithVertexSelector.h.

Referenced by testTrack().

dzMax_

double TrackWithVertexSelector::dzMax_

private

Definition at line 50 of file TrackWithVertexSelector.h.

Referenced by testTrack().

etaMax_

double TrackWithVertexSelector::etaMax_

private

Definition at line 49 of file TrackWithVertexSelector.h.

Referenced by testTrack().

etaMin_

double TrackWithVertexSelector::etaMin_

private

Definition at line 49 of file TrackWithVertexSelector.h.

Referenced by testTrack().

normalizedChi2_

double TrackWithVertexSelector::normalizedChi2_

private

Definition at line 48 of file TrackWithVertexSelector.h.

Referenced by testTrack().

nSigmaDtVertex_

double TrackWithVertexSelector::nSigmaDtVertex_

private

Definition at line 58 of file TrackWithVertexSelector.h.

Referenced by testVertices().

numberOfLostHits_

uint32_t TrackWithVertexSelector::numberOfLostHits_

private

Definition at line 47 of file TrackWithVertexSelector.h.

Referenced by testTrack().

numberOfValidHits_

uint32_t TrackWithVertexSelector::numberOfValidHits_

private

Definition at line 45 of file TrackWithVertexSelector.h.

Referenced by testTrack().

numberOfValidPixelHits_

uint32_t TrackWithVertexSelector::numberOfValidPixelHits_

private

Definition at line 46 of file TrackWithVertexSelector.h.

Referenced by testTrack().

nVertices_

uint32_t TrackWithVertexSelector::nVertices_

private

Definition at line 54 of file TrackWithVertexSelector.h.

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

ptErrorCut_

double TrackWithVertexSelector::ptErrorCut_

private

Definition at line 51 of file TrackWithVertexSelector.h.

Referenced by testTrack().

ptMax_

double TrackWithVertexSelector::ptMax_

private

Definition at line 49 of file TrackWithVertexSelector.h.

Referenced by testTrack().

ptMin_

double TrackWithVertexSelector::ptMin_

private

Definition at line 49 of file TrackWithVertexSelector.h.

Referenced by testTrack().

quality_

std::string TrackWithVertexSelector::quality_

private

Definition at line 52 of file TrackWithVertexSelector.h.

Referenced by testTrack().

rhoVtx_

double TrackWithVertexSelector::rhoVtx_

private

Definition at line 58 of file TrackWithVertexSelector.h.

Referenced by testVertices().

timeresoscoll_

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

private

Definition at line 62 of file TrackWithVertexSelector.h.

Referenced by init(), and testVertices().

timeResosToken_

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

private

Definition at line 56 of file TrackWithVertexSelector.h.

Referenced by init().

timescoll_

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

private

Definition at line 61 of file TrackWithVertexSelector.h.

Referenced by init(), and testVertices().

timesToken_

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

private

Definition at line 56 of file TrackWithVertexSelector.h.

Referenced by init().

vcoll_

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

private

Definition at line 60 of file TrackWithVertexSelector.h.

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

vertexToken_

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

private

Definition at line 55 of file TrackWithVertexSelector.h.

Referenced by init().

vtxFallback_

bool TrackWithVertexSelector::vtxFallback_

private

Definition at line 57 of file TrackWithVertexSelector.h.

Referenced by testVertices().

zetaVtx_

double TrackWithVertexSelector::zetaVtx_

private

Definition at line 58 of file TrackWithVertexSelector.h.

Referenced by testVertices().