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