PFDisplacedVertexCandidateFinder.h

Go to the documentation of this file.

#ifndef RecoParticleFlow_PFTracking_PFDisplacedVertexCandidateFinder_h
#define RecoParticleFlow_PFTracking_PFDisplacedVertexCandidateFinder_h
 
#include "DataFormats/ParticleFlowReco/interface/PFDisplacedVertexCandidate.h"
#include "DataFormats/ParticleFlowReco/interface/PFDisplacedVertexCandidateFwd.h"
#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "DataFormats/BeamSpot/interface/BeamSpot.h"
 
#include "FWCore/ParameterSet/interface/ParameterSet.h"
 
#include "TrackingTools/TrajectoryParametrization/interface/GlobalTrajectoryParameters.h"
#include "TrackingTools/PatternTools/interface/TwoTrackMinimumDistance.h"
 
 
class MagneticField;
 
class PFDisplacedVertexCandidateFinder {
public:
  PFDisplacedVertexCandidateFinder();
 
  ~PFDisplacedVertexCandidateFinder();
 
  typedef std::vector<bool> Mask;
 
  typedef std::list<reco::TrackBaseRef>::iterator IE;
  typedef std::list<reco::TrackBaseRef>::const_iterator IEC;
  typedef reco::PFDisplacedVertexCandidateCollection::const_iterator IBC;
 
 
  void setParameters(double dcaCut, double primaryVertexCut, double dcaPInnerHitCut, const edm::ParameterSet& ps_trk) {
    dcaCut_ = dcaCut;
    primaryVertexCut2_ = primaryVertexCut * primaryVertexCut;
    dcaPInnerHitCut2_ = dcaPInnerHitCut * dcaPInnerHitCut;
    nChi2_max_ = ps_trk.getParameter<double>("nChi2_max");
    pt_min_ = ps_trk.getParameter<double>("pt_min");
    pt_min_prim_ = ps_trk.getParameter<double>("pt_min_prim");
    dxy_ = ps_trk.getParameter<double>("dxy");
  }
 
  void setDebug(bool debug) { debug_ = debug; }
 
  void setInput(const edm::Handle<reco::TrackCollection>& trackh, const MagneticField* magField);
 
  std::unique_ptr<reco::PFDisplacedVertexCandidateCollection> transferVertexCandidates() {
    return std::move(vertexCandidates_);
  }
 
  const std::unique_ptr<reco::PFDisplacedVertexCandidateCollection>& vertexCandidates() const {
    return std::move(vertexCandidates_);
  }
 
 
  void findDisplacedVertexCandidates();
 
  void setPrimaryVertex(edm::Handle<reco::VertexCollection> mainVertexHandle,
                        edm::Handle<reco::BeamSpot> beamSpotHandle);
 
private:
 
  IE associate(IE next, IE last, reco::PFDisplacedVertexCandidate& tempVertexCandidate);
 
  void link(const reco::TrackBaseRef& el1,
            const reco::TrackBaseRef& el2,
            double& dist,
            GlobalPoint& P,
            reco::PFDisplacedVertexCandidate::VertexLinkTest& linktest);
 
  void packLinks(reco::PFDisplacedVertexCandidate& vertexCandidate);
 
 
  GlobalTrajectoryParameters getGlobalTrajectoryParameters(const reco::Track*) const;
 
  bool goodPtResolution(const reco::TrackBaseRef& trackref) const;
 
  bool isSelected(const reco::TrackBaseRef& trackref) { return goodPtResolution(trackref); }
 
  friend std::ostream& operator<<(std::ostream&, const PFDisplacedVertexCandidateFinder&);
 
 
  std::unique_ptr<reco::PFDisplacedVertexCandidateCollection> vertexCandidates_;
 
  std::list<reco::TrackBaseRef> eventTracks_;
 
  Mask trackMask_;
  std::vector<GlobalTrajectoryParameters> eventTrackTrajectories_;
 
 
  double dcaCut_;
  double primaryVertexCut2_;
  double dcaPInnerHitCut2_;
 
  double nChi2_max_;
  double pt_min_;
 
  double pt_min_prim_;
  double dxy_;
 
  unsigned vertexCandidatesSize_;
 
  // Two track minimum distance algo
  TwoTrackMinimumDistance theMinimum_;
 
  math::XYZPoint pvtx_;
 
  bool debug_;
 
  // Tracker geometry for extrapolation
  const MagneticField* magField_;
};
 
#endif