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 "CommonTools/Utils/interface/KinematicTables.h"
#include "FWCore/MessageLogger/interface/MessageLogger.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& crossing_point,
            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_;

  edm::soa::PtEtaPhiTable track_table_;
};

#endif