TrackWithVertexSelector.h

Go to the documentation of this file.

#ifndef CommonTools_RecoAlgos_TrackWithVertexSelector_H
#define CommonTools_RecoAlgos_TrackWithVertexSelector_H
 
// Original Author:  Giovanni Petrucciani
//         Created:  Fri May 25 10:06:02 CEST 2007
// $Id: TrackWithVertexSelector.h,v 1.4 2010/04/07 08:56:18 gpetrucc Exp $
 
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "DataFormats/Common/interface/ValueMap.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
 
class TrackWithVertexSelector {
public:
  explicit TrackWithVertexSelector(const edm::ParameterSet &iConfig, edm::ConsumesCollector &&iC)
      : TrackWithVertexSelector(iConfig, iC) {}
  explicit TrackWithVertexSelector(const edm::ParameterSet &iConfig, edm::ConsumesCollector &iC);
  ~TrackWithVertexSelector();
 
  void init(const edm::Event &event, const edm::EventSetup &) { init(event); }
  void init(const edm::Event &event);
 
  bool operator()(const reco::Track &t) const;
  bool operator()(const reco::Track &t, const edm::Event &iEvent) {
    init(iEvent);
    return (*this)(t);
  }
 
  bool operator()(const reco::TrackRef &t) const;
  bool operator()(const reco::TrackRef &t, const edm::Event &iEvent) {
    init(iEvent);
    return (*this)(t);
  }
  bool testTrack(const reco::Track &t) const;
  bool testVertices(const reco::Track &t, const reco::VertexCollection &vtxs) const;
 
  bool testTrack(const reco::TrackRef &t) const;
  bool testVertices(const reco::TrackRef &t, const reco::VertexCollection &vtxs) const;
 
private:
  uint32_t numberOfValidHits_;
  uint32_t numberOfValidPixelHits_;
  uint32_t numberOfLostHits_;
  double normalizedChi2_;
  double ptMin_, ptMax_, etaMin_, etaMax_;
  double dzMax_, d0Max_;
  double ptErrorCut_;
  std::string quality_;
 
  uint32_t nVertices_;
  edm::EDGetTokenT<reco::VertexCollection> vertexToken_;
  edm::EDGetTokenT<edm::ValueMap<float> > timesToken_, timeResosToken_;
  bool vtxFallback_;
  double zetaVtx_, rhoVtx_, nSigmaDtVertex_;
 
  reco::VertexCollection const *vcoll_ = nullptr;
  edm::ValueMap<float> const *timescoll_ = nullptr;
  edm::ValueMap<float> const *timeresoscoll_ = nullptr;
  typedef math::XYZPoint Point;
};
 
#endif