PFDisplacedVertexFinder.h

Go to the documentation of this file.

#ifndef RecoParticleFlow_PFTracking_PFDisplacedVertexFinder_h
#define RecoParticleFlow_PFTracking_PFDisplacedVertexFinder_h
 
#include "RecoParticleFlow/PFTracking/interface/PFDisplacedVertexHelper.h"
#include "FWCore/Framework/interface/ESHandle.h"
 
#include "DataFormats/ParticleFlowReco/interface/PFDisplacedVertexCandidate.h"
#include "DataFormats/ParticleFlowReco/interface/PFDisplacedVertexCandidateFwd.h"
#include "DataFormats/ParticleFlowReco/interface/PFDisplacedVertexSeed.h"
#include "DataFormats/ParticleFlowReco/interface/PFDisplacedVertexSeedFwd.h"
#include "DataFormats/ParticleFlowReco/interface/PFDisplacedVertex.h"
#include "DataFormats/ParticleFlowReco/interface/PFDisplacedVertexFwd.h"
 
#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
#include "DataFormats/GeometryVector/interface/GlobalVector.h"
 
#include "DataFormats/TrackReco/interface/TrackFwd.h"
 
#include "RecoParticleFlow/PFTracking/interface/PFCheckHitPattern.h"
 
 
class TrackingGeometry;
class TrackerGeometry;
class MagneticField;
 
class PFDisplacedVertexFinder {
public:
  PFDisplacedVertexFinder();
 
  ~PFDisplacedVertexFinder();
 
 
  typedef reco::PFDisplacedVertexSeedCollection::iterator IDVS;
  typedef reco::PFDisplacedVertexCollection::iterator IDV;
 
  typedef std::pair<unsigned int, unsigned int> PFTrackHitInfo;
  typedef std::pair<PFTrackHitInfo, PFTrackHitInfo> PFTrackHitFullInfo;
 
  enum FitterType { F_NOTDEFINED, F_DONOTREFIT, F_KALMAN, F_ADAPTIVE };
 
 
  void setParameters(double transvSize,
                     double longSize,
                     double primaryVertexCut,
                     double tobCut,
                     double tecCut,
                     double minAdaptWeight,
                     bool switchOff2TrackVertex) {
    transvSize_ = transvSize;
    longSize_ = longSize;
    primaryVertexCut_ = primaryVertexCut;
    tobCut_ = tobCut;
    tecCut_ = tecCut;
    minAdaptWeight_ = minAdaptWeight;
    switchOff2TrackVertex_ = switchOff2TrackVertex;
  }
 
  void setDebug(bool debug) { debug_ = debug; }
 
  void setEdmParameters(const MagneticField* magField,
                        edm::ESHandle<GlobalTrackingGeometry> globTkGeomHandle,
                        const TrackerTopology* tkerTopo,
                        const TrackerGeometry* tkerGeom) {
    magField_ = magField;
    globTkGeomHandle_ = globTkGeomHandle;
    tkerTopo_ = tkerTopo;
    tkerGeom_ = tkerGeom;
  }
 
  void setTracksSelector(const edm::ParameterSet& ps) { helper_.setTracksSelector(ps); }
 
  void setVertexIdentifier(const edm::ParameterSet& ps) { helper_.setVertexIdentifier(ps); }
 
  void setPrimaryVertex(edm::Handle<reco::VertexCollection> mainVertexHandle,
                        edm::Handle<reco::BeamSpot> beamSpotHandle) {
    helper_.setPrimaryVertex(mainVertexHandle, beamSpotHandle);
  }
 
  void setAVFParameters(const edm::ParameterSet& ps) {
    sigmacut_ = ps.getParameter<double>("sigmacut");
    t_ini_ = ps.getParameter<double>("Tini");
    ratio_ = ps.getParameter<double>("ratio");
  }
 
  void setInput(const edm::Handle<reco::PFDisplacedVertexCandidateCollection>&);
 
  std::unique_ptr<reco::PFDisplacedVertexCollection> transferDisplacedVertices() {
    return std::move(displacedVertices_);
  }
 
  const std::unique_ptr<reco::PFDisplacedVertexCollection>& displacedVertices() const {
    return std::move(displacedVertices_);
  }
 
 
  void findDisplacedVertices();
 
private:
 
  void findSeedsFromCandidate(const reco::PFDisplacedVertexCandidate&, reco::PFDisplacedVertexSeedCollection&);
 
  void mergeSeeds(reco::PFDisplacedVertexSeedCollection&, std::vector<bool>& bLocked);
 
  bool fitVertexFromSeed(const reco::PFDisplacedVertexSeed&, reco::PFDisplacedVertex&);
 
  void selectAndLabelVertices(reco::PFDisplacedVertexCollection&, std::vector<bool>&);
 
  bool rejectAndLabelVertex(reco::PFDisplacedVertex& dv);
 
 
  bool isCloseTo(const reco::PFDisplacedVertexSeed&, const reco::PFDisplacedVertexSeed&) const;
 
  std::pair<float, float> getTransvLongDiff(const GlobalPoint&, const GlobalPoint&) const;
 
  reco::PFDisplacedVertex::VertexTrackType getVertexTrackType(PFTrackHitFullInfo&) const;
 
  unsigned commonTracks(const reco::PFDisplacedVertex&, const reco::PFDisplacedVertex&) const;
 
  friend std::ostream& operator<<(std::ostream&, const PFDisplacedVertexFinder&);
 
 
  reco::PFDisplacedVertexCandidateCollection const* displacedVertexCandidates_;
  std::unique_ptr<reco::PFDisplacedVertexCollection> displacedVertices_;
 
 
 
  float transvSize_;
  float longSize_;
  double primaryVertexCut_;
  double tobCut_;
  double tecCut_;
  double minAdaptWeight_;
 
  bool switchOff2TrackVertex_;
 
 
  double sigmacut_;  //= 6;
  double t_ini_;     //= 256.;
  double ratio_;     //= 0.25;
 
  bool debug_;
 
  edm::ESHandle<GlobalTrackingGeometry> globTkGeomHandle_;
 
  const TrackerTopology* tkerTopo_;
  const TrackerGeometry* tkerGeom_;
 
  const MagneticField* magField_;
 
  PFCheckHitPattern hitPattern_;
 
  PFDisplacedVertexHelper helper_;
};
 
#endif