PFDisplacedVertexFinder Class Reference

#include <PFDisplacedVertexFinder.h>

Public Types
enum	FitterType { F_NOTDEFINED, F_DONOTREFIT, F_KALMAN, F_ADAPTIVE }
	Fitter Type. More...
typedef reco::PFDisplacedVertexCollection::iterator	IDV
typedef reco::PFDisplacedVertexSeedCollection::iterator	IDVS
	-----— Useful Types -----— /// More...
typedef std::pair< PFTrackHitInfo, PFTrackHitInfo >	PFTrackHitFullInfo
typedef std::pair< unsigned int, unsigned int >	PFTrackHitInfo

Public Member Functions
const std::unique_ptr< reco::PFDisplacedVertexCollection > &	displacedVertices () const
void	findDisplacedVertices ()
	-----— Main function which find vertices -----— /// More...
	PFDisplacedVertexFinder ()
void	setAVFParameters (const edm::ParameterSet &ps)
void	setDebug (bool debug)
	Sets debug printout flag. More...
void	setEdmParameters (const MagneticField *magField, edm::ESHandle< GlobalTrackingGeometry > globTkGeomHandle, const TrackerTopology *tkerTopo, const TrackerGeometry *tkerGeom)
	Sets parameters for track extrapolation and hits study. More...
void	setInput (const edm::Handle< reco::PFDisplacedVertexCandidateCollection > &)
	Set input collections of tracks. More...
void	setParameters (double transvSize, double longSize, double primaryVertexCut, double tobCut, double tecCut, double minAdaptWeight, bool switchOff2TrackVertex)
	--—— Set different algo parameters --—— /// More...
void	setPrimaryVertex (edm::Handle< reco::VertexCollection > mainVertexHandle, edm::Handle< reco::BeamSpot > beamSpotHandle)
void	setTracksSelector (const edm::ParameterSet &ps)
void	setVertexIdentifier (const edm::ParameterSet &ps)
std::unique_ptr< reco::PFDisplacedVertexCollection >	transferDisplacedVertices ()
	~PFDisplacedVertexFinder ()

Private Member Functions
unsigned	commonTracks (const reco::PFDisplacedVertex &, const reco::PFDisplacedVertex &) const
void	findSeedsFromCandidate (const reco::PFDisplacedVertexCandidate &, reco::PFDisplacedVertexSeedCollection &)
	--—— Different steps of the finder algorithm --—— /// More...
bool	fitVertexFromSeed (const reco::PFDisplacedVertexSeed &, reco::PFDisplacedVertex &)
	Fit one by one the vertex points with associated tracks to get displaced vertices. More...
std::pair< float, float >	getTransvLongDiff (const GlobalPoint &, const GlobalPoint &) const
reco::PFDisplacedVertex::VertexTrackType	getVertexTrackType (PFTrackHitFullInfo &) const
bool	isCloseTo (const reco::PFDisplacedVertexSeed &, const reco::PFDisplacedVertexSeed &) const
	-----— Tools -----— /// More...
void	mergeSeeds (reco::PFDisplacedVertexSeedCollection &, std::vector< bool > &bLocked)
	Sometimes two vertex candidates can be quite close and coming from the same vertex. More...
bool	rejectAndLabelVertex (reco::PFDisplacedVertex &dv)
void	selectAndLabelVertices (reco::PFDisplacedVertexCollection &, std::vector< bool > &)
	Remove potentially fakes displaced vertices. More...

Private Attributes
bool	debug_
	If true, debug printouts activated. More...
reco::PFDisplacedVertexCandidateCollection const *	displacedVertexCandidates_
	-----— Members -----— /// More...
std::unique_ptr< reco::PFDisplacedVertexCollection >	displacedVertices_
edm::ESHandle< GlobalTrackingGeometry >	globTkGeomHandle_
	Tracker geometry for discerning hit positions. More...
PFDisplacedVertexHelper	helper_
PFCheckHitPattern	hitPattern_
float	longSize_
const MagneticField *	magField_
	to be able to extrapolate tracks f More...
double	minAdaptWeight_
double	primaryVertexCut_
double	ratio_
double	sigmacut_
	Adaptive Vertex Fitter parameters. More...
bool	switchOff2TrackVertex_
double	t_ini_
double	tecCut_
const TrackerGeometry *	tkerGeom_
const TrackerTopology *	tkerTopo_
	doc? More...
double	tobCut_
float	transvSize_
	--—— Parameters --—— /// More...

Friends
std::ostream &	operator<< (std::ostream &, const PFDisplacedVertexFinder &)

Detailed Description

Definition at line 31 of file PFDisplacedVertexFinder.h.

Member Typedef Documentation

IDV

typedef reco::PFDisplacedVertexCollection::iterator PFDisplacedVertexFinder::IDV

Definition at line 40 of file PFDisplacedVertexFinder.h.

IDVS

typedef reco::PFDisplacedVertexSeedCollection::iterator PFDisplacedVertexFinder::IDVS

-----— Useful Types -----— ///

Definition at line 39 of file PFDisplacedVertexFinder.h.

PFTrackHitFullInfo

typedef std::pair<PFTrackHitInfo, PFTrackHitInfo> PFDisplacedVertexFinder::PFTrackHitFullInfo

Definition at line 43 of file PFDisplacedVertexFinder.h.

PFTrackHitInfo

typedef std::pair<unsigned int, unsigned int> PFDisplacedVertexFinder::PFTrackHitInfo

Definition at line 42 of file PFDisplacedVertexFinder.h.

Member Enumeration Documentation

FitterType

enum PFDisplacedVertexFinder::FitterType

Fitter Type.

Enumerator
F_NOTDEFINED
F_DONOTREFIT
F_KALMAN
F_ADAPTIVE

Definition at line 46 of file PFDisplacedVertexFinder.h.

{ F_NOTDEFINED, F_DONOTREFIT, F_KALMAN, F_ADAPTIVE };

Constructor & Destructor Documentation

PFDisplacedVertexFinder()

PFDisplacedVertexFinder::PFDisplacedVertexFinder

(

)

Definition at line 30 of file PFDisplacedVertexFinder.cc.

    : displacedVertexCandidates_(nullptr),
      displacedVertices_(new PFDisplacedVertexCollection),
      transvSize_(0.0),
      longSize_(0.0),
      primaryVertexCut_(0.0),
      tobCut_(0.0),
      tecCut_(0.0),
      minAdaptWeight_(2.0),
      debug_(false) {}

~PFDisplacedVertexFinder()

PFDisplacedVertexFinder::~PFDisplacedVertexFinder

(

)

Definition at line 41 of file PFDisplacedVertexFinder.cc.

{}

Member Function Documentation

commonTracks()

unsigned PFDisplacedVertexFinder::commonTracks ( const reco::PFDisplacedVertex &  v1, const reco::PFDisplacedVertex &  v2  ) const

private

Definition at line 661 of file PFDisplacedVertexFinder.cc.

References edm::RefToBase< T >::key(), reco::Vertex::originalTrack(), and reco::Vertex::refittedTracks().

Referenced by selectAndLabelVertices().

                                                                                                             {
  vector<Track> vt1 = v1.refittedTracks();
  vector<Track> vt2 = v2.refittedTracks();

  unsigned commonTracks = 0;

  for (unsigned il1 = 0; il1 < vt1.size(); il1++) {
    unsigned il1_idx = v1.originalTrack(vt1[il1]).key();

    for (unsigned il2 = 0; il2 < vt2.size(); il2++)
      if (il1_idx == v2.originalTrack(vt2[il2]).key()) {
        commonTracks++;
        break;
      }
  }

  return commonTracks;
}

displacedVertices()

const std::unique_ptr<reco::PFDisplacedVertexCollection>& PFDisplacedVertexFinder::displacedVertices ( ) const

inline

Definition at line 104 of file PFDisplacedVertexFinder.h.

References displacedVertices_, and eostools::move().

                                                                                  {
    return std::move(displacedVertices_);
  }

findDisplacedVertices()

void PFDisplacedVertexFinder::findDisplacedVertices

(

)

-----— Main function which find vertices -----— ///

Definition at line 54 of file PFDisplacedVertexFinder.cc.

References gather_cfg::cout, debug_, displacedVertexCandidates_, displacedVertices_, findSeedsFromCandidate(), fitVertexFromSeed(), mps_fire::i, mergeSeeds(), and selectAndLabelVertices().

Referenced by PFDisplacedVertexProducer::produce().

                                                    {
  if (debug_)
    cout << "========= Start Find Displaced Vertices =========" << endl;

  // The vertexCandidates have not been passed to the event
  // So they need to be cleared is they are not empty
  if (displacedVertices_.get())
    displacedVertices_->clear();
  else
    displacedVertices_ = std::make_unique<PFDisplacedVertexCollection>();

  if (displacedVertexCandidates_ == nullptr) {
    edm::LogInfo("EmptyVertexInput")
        << "displacedVertexCandidates are not set or the setInput was called with invalid vertex";
    return;
  }

  // Prepare the collections
  PFDisplacedVertexSeedCollection tempDisplacedVertexSeeds;
  tempDisplacedVertexSeeds.reserve(4 * displacedVertexCandidates_->size());
  PFDisplacedVertexCollection tempDisplacedVertices;
  tempDisplacedVertices.reserve(4 * displacedVertexCandidates_->size());

  if (debug_)
    cout << "1) Parsing displacedVertexCandidates into displacedVertexSeeds" << endl;

  // 1) Parsing displacedVertexCandidates into displacedVertexSeeds which would
  // be later used for vertex fitting

  int i = -1;

  for (auto const& idvc : *displacedVertexCandidates_) {
    i++;
    if (debug_) {
      cout << "Analyse Vertex Candidate " << i << endl;
    }

    findSeedsFromCandidate(idvc, tempDisplacedVertexSeeds);
  }

  if (debug_)
    cout << "2) Merging Vertex Seeds" << endl;

  // 2) Some displacedVertexSeeds coming from different displacedVertexCandidates
  // may be closed enough to be merged together. bLocked is an array which keeps the
  // information about the seeds which are desabled.
  vector<bool> bLockedSeeds;
  bLockedSeeds.resize(tempDisplacedVertexSeeds.size());
  mergeSeeds(tempDisplacedVertexSeeds, bLockedSeeds);

  if (debug_)
    cout << "3) Fitting Vertices From Seeds" << endl;

  // 3) Fit displacedVertices from displacedVertexSeeds
  for (unsigned idv = 0; idv < tempDisplacedVertexSeeds.size(); idv++) {
    if (!tempDisplacedVertexSeeds[idv].isEmpty() && !bLockedSeeds[idv]) {
      PFDisplacedVertex displacedVertex;
      bLockedSeeds[idv] = fitVertexFromSeed(tempDisplacedVertexSeeds[idv], displacedVertex);
      if (!bLockedSeeds[idv])
        tempDisplacedVertices.emplace_back(displacedVertex);
    }
  }

  if (debug_)
    cout << "4) Rejecting Bad Vertices and label them" << endl;

  // 4) Reject displaced vertices which may be considered as fakes
  vector<bool> bLocked;
  bLocked.resize(tempDisplacedVertices.size());
  selectAndLabelVertices(tempDisplacedVertices, bLocked);

  if (debug_)
    cout << "5) Fill the Displaced Vertices" << endl;

  // 5) Fill the displacedVertex_ which would be transfered to the producer
  displacedVertices_->reserve(tempDisplacedVertices.size());

  for (unsigned idv = 0; idv < tempDisplacedVertices.size(); idv++)
    if (!bLocked[idv])
      displacedVertices_->push_back(tempDisplacedVertices[idv]);

  if (debug_)
    cout << "========= End Find Displaced Vertices =========" << endl;
}

findSeedsFromCandidate()

void PFDisplacedVertexFinder::findSeedsFromCandidate ( const reco::PFDisplacedVertexCandidate &  vertexCandidate, reco::PFDisplacedVertexSeedCollection &  tempDisplacedVertexSeeds  )

private

--—— Different steps of the finder algorithm --—— ///

Analyse a vertex candidate and select potential vertex point(s)

Definition at line 141 of file PFDisplacedVertexFinder.cc.

References gather_cfg::cout, reco::PFDisplacedVertexCandidate::dcaPoint(), debug_, getTransvLongDiff(), longSize_, reco::PFDisplacedVertexCandidate::r2Map(), mathSSE::sqrt(), transvSize_, reco::PFDisplacedVertexCandidate::tref(), and PV3DBase< T, PVType, FrameType >::x().

Referenced by findDisplacedVertices().

                                                                                                                {
  const PFDisplacedVertexCandidate::DistMap r2Map = vertexCandidate.r2Map();
  bool bNeedNewCandidate = false;

  tempDisplacedVertexSeeds.push_back(PFDisplacedVertexSeed());

  IDVS idvc_current;

  for (PFDisplacedVertexCandidate::DistMap::const_iterator imap = r2Map.begin(); imap != r2Map.end(); imap++) {
    unsigned ie1 = (*imap).second.first;
    unsigned ie2 = (*imap).second.second;

    if (debug_)
      cout << "ie1 = " << ie1 << " ie2 = " << ie2 << " radius = " << sqrt((*imap).first) << endl;

    GlobalPoint dcaPoint = vertexCandidate.dcaPoint(ie1, ie2);
    if (fabs(dcaPoint.x()) > 1e9)
      continue;

    bNeedNewCandidate = true;
    for (idvc_current = tempDisplacedVertexSeeds.begin(); idvc_current != tempDisplacedVertexSeeds.end();
         idvc_current++) {
      if ((*idvc_current).isEmpty()) {
        bNeedNewCandidate = false;
        break;
      }
      const GlobalPoint vertexPoint = (*idvc_current).seedPoint();
      std::pair<float, float> diffs = getTransvLongDiff(vertexPoint, dcaPoint);
      if (diffs.second > longSize_)
        continue;
      if (diffs.first > transvSize_)
        continue;
      bNeedNewCandidate = false;
      break;
    }
    if (bNeedNewCandidate) {
      if (debug_)
        cout << "create new displaced vertex" << endl;
      tempDisplacedVertexSeeds.push_back(PFDisplacedVertexSeed());
      idvc_current = tempDisplacedVertexSeeds.end();
      idvc_current--;
    }

    (*idvc_current).updateSeedPoint(dcaPoint, vertexCandidate.tref(ie1), vertexCandidate.tref(ie2));
  }
}

fitVertexFromSeed()

bool PFDisplacedVertexFinder::fitVertexFromSeed ( const reco::PFDisplacedVertexSeed &  displacedVertexSeed, reco::PFDisplacedVertex &  displacedVertex  )

private

Fit one by one the vertex points with associated tracks to get displaced vertices.

This prefit procedure allow to reduce the Warning rate from Adaptive Vertex fitter It reject also many fake tracks

Definition at line 209 of file PFDisplacedVertexFinder.cc.

References reco::PFDisplacedVertex::addElement(), PFCheckHitPattern::analyze(), reco::PFDisplacedVertex::calcKinematics(), hltPixelTracks_cff::chi2, reco::Vertex::chi2(), gather_cfg::cout, debug_, reco::PFDisplacedVertexSeed::Dump(), reco::PFDisplacedVertexSeed::elements(), cppFunctionSkipper::exception, F_ADAPTIVE, F_DONOTREFIT, F_KALMAN, F_NOTDEFINED, getVertexTrackType(), globTkGeomHandle_, helper_, PFTrackAlgoTools::highQuality(), hitPattern_, mps_fire::i, PFDisplacedVertexHelper::isTrackSelected(), TransientVertex::isValid(), magField_, minAdaptWeight_, reco::Vertex::ndof(), PFTrackAlgoTools::nonIterative(), topSingleLeptonDQM_PU_cfi::pattern, TransientVertex::position(), PFDisplacedVertexHelper::primaryVertex(), primaryVertexCut_, ratio_, TransientVertex::refittedTrack(), reco::Vertex::removeTracks(), rho, reco::PFDisplacedVertexSeed::seedPoint(), reco::PFDisplacedVertex::setPrimaryDirection(), sigmacut_, mathSSE::sqrt(), PFTrackAlgoTools::step45(), switchOff2TrackVertex_, t_ini_, tecCut_, tkerGeom_, tkerTopo_, tobCut_, TransientVertex::totalChiSquared(), reco::TransientTrack::track(), HLT_2022v12_cff::track, TransientVertex::trackWeight(), KalmanVertexFitter::vertex(), AdaptiveVertexFitter::vertex(), extraflags_cff::vtx, PV3DBase< T, PVType, FrameType >::x(), PV3DBase< T, PVType, FrameType >::y(), z, and PV3DBase< T, PVType, FrameType >::z().

Referenced by findDisplacedVertices().

                                                                                    {
  if (debug_)
    cout << "== Start vertexing procedure ==" << endl;

  // ---- Prepare transient track list ----

  auto const& tracksToFit = displacedVertexSeed.elements();
  const GlobalPoint& seedPoint = displacedVertexSeed.seedPoint();

  vector<TransientTrack> transTracks;
  vector<TransientTrack> transTracksRaw;
  vector<TrackBaseRef> transTracksRef;

  transTracks.reserve(tracksToFit.size());
  transTracksRaw.reserve(tracksToFit.size());
  transTracksRef.reserve(tracksToFit.size());

  TransientVertex theVertexAdaptiveRaw;
  TransientVertex theRecoVertex;

  // ---- 1) Clean for potentially poor seeds ------- //
  // --------------------------------------------- //

  if (tracksToFit.size() < 2) {
    if (debug_)
      cout << "Only one to Fit Track" << endl;
    return true;
  }

  double rho = sqrt(seedPoint.x() * seedPoint.x() + seedPoint.y() * seedPoint.y());
  double z = seedPoint.z();

  if (rho > tobCut_ || fabs(z) > tecCut_) {
    if (debug_)
      cout << "Seed Point out of the tracker rho = " << rho << " z = " << z << " nTracks = " << tracksToFit.size()
           << endl;
    return true;
  }

  if (debug_)
    displacedVertexSeed.Dump();

  int nStep45 = 0;
  int nNotIterative = 0;

  // Fill vectors of TransientTracks and TrackRefs after applying preselection cuts.
  for (auto const& ie : tracksToFit) {
    TransientTrack tmpTk(*(ie.get()), magField_, globTkGeomHandle_);
    transTracksRaw.emplace_back(tmpTk);
    bool nonIt = PFTrackAlgoTools::nonIterative((ie)->algo());
    bool step45 = PFTrackAlgoTools::step45((ie)->algo());
    bool highQ = PFTrackAlgoTools::highQuality((ie)->algo());
    if (step45)
      nStep45++;
    else if (nonIt)
      nNotIterative++;
    else if (!highQ) {
      nNotIterative++;
      nStep45++;
    }
  }

  if (rho > 25 && nStep45 + nNotIterative < 1) {
    if (debug_)
      cout << "Seed point at rho > 25 cm but no step 4-5 tracks" << endl;
    return true;
  }

  // ----------------------------------------------- //
  // ---- PRESELECT GOOD TRACKS FOR FINAL VERTEX --- //
  // ----------------------------------------------- //

  // 1)If only two track are found do not prefit

  if (transTracksRaw.size() == 2) {
    if (debug_)
      cout << "No raw fit done" << endl;
    if (switchOff2TrackVertex_) {
      if (debug_)
        cout << "Due to probably high pile-up conditions 2 track vertices switched off" << endl;
      return true;
    }
    GlobalError globalError;

    theVertexAdaptiveRaw = TransientVertex(seedPoint, globalError, transTracksRaw, 1.);

  } else {  //branch with transTracksRaw.size of at least 3

    if (debug_)
      cout << "Raw fit done." << endl;

    AdaptiveVertexFitter theAdaptiveFitterRaw(GeometricAnnealing(sigmacut_, t_ini_, ratio_),
                                              DefaultLinearizationPointFinder(),
                                              KalmanVertexUpdator<5>(),
                                              KalmanVertexTrackCompatibilityEstimator<5>(),
                                              KalmanVertexSmoother());

    if (transTracksRaw.size() == 3) {
      theVertexAdaptiveRaw = theAdaptiveFitterRaw.vertex(transTracksRaw, seedPoint);

    } else if (transTracksRaw.size() < 1000) {

      if (debug_)
        cout << "First test with KFT" << endl;

      KalmanVertexFitter theKalmanFitter(true);
      theVertexAdaptiveRaw = theKalmanFitter.vertex(transTracksRaw, seedPoint);

      if (!theVertexAdaptiveRaw.isValid() || theVertexAdaptiveRaw.totalChiSquared() < 0.) {
        if (debug_)
          cout << "Prefit failed : valid? " << theVertexAdaptiveRaw.isValid()
               << " totalChi2 = " << theVertexAdaptiveRaw.totalChiSquared() << endl;
        return true;
      }

      if (debug_)
        cout << "We use KFT instead of seed point to set up a point for AVF "
             << " x = " << theVertexAdaptiveRaw.position().x() << " y = " << theVertexAdaptiveRaw.position().y()
             << " z = " << theVertexAdaptiveRaw.position().z() << endl;

      // To save time: reject the Displaced vertex if it is too close to the beam pipe.
      // Frequently it is very big vertices, with some dosens of tracks

      Vertex vtx = theVertexAdaptiveRaw;
      rho = vtx.position().rho();

      //   cout << "primary vertex cut  = " << primaryVertexCut_ << endl;

      if (rho < primaryVertexCut_ || rho > 100) {
        if (debug_)
          cout << "KFT Vertex geometrically rejected with  tracks #rho = " << rho << endl;
        return true;
      }

      //     cout << "primary vertex cut  = " << primaryVertexCut_ << " rho = " << rho << endl;

      theVertexAdaptiveRaw = theAdaptiveFitterRaw.vertex(transTracksRaw, theVertexAdaptiveRaw.position());

    } else {
      edm::LogWarning("TooManyPFDVCandidates") << "gave up vertex reco for " << transTracksRaw.size() << " tracks";
    }

    if (!theVertexAdaptiveRaw.isValid() || theVertexAdaptiveRaw.totalChiSquared() < 0.) {
      if (debug_)
        cout << "Fit failed : valid? " << theVertexAdaptiveRaw.isValid()
             << " totalChi2 = " << theVertexAdaptiveRaw.totalChiSquared() << endl;
      return true;
    }

    // To save time: reject the Displaced vertex if it is too close to the beam pipe.
    // Frequently it is very big vertices, with some dosens of tracks

    Vertex vtx = theVertexAdaptiveRaw;
    rho = vtx.position().rho();

    if (rho < primaryVertexCut_) {
      if (debug_)
        cout << "Vertex "
             << " geometrically rejected with " << transTracksRaw.size() << " tracks #rho = " << rho << endl;
      return true;
    }
  }

  // ---- Remove tracks with small weight or
  //      big first (last) hit_to_vertex distance
  //      and then refit                          ---- //

  for (unsigned i = 0; i < transTracksRaw.size(); i++) {
    if (debug_)
      cout << "Raw Weight track " << i << " = " << theVertexAdaptiveRaw.trackWeight(transTracksRaw[i]) << endl;

    if (theVertexAdaptiveRaw.trackWeight(transTracksRaw[i]) > minAdaptWeight_) {
      PFTrackHitFullInfo pattern = hitPattern_.analyze(tkerTopo_, tkerGeom_, tracksToFit[i], theVertexAdaptiveRaw);

      PFDisplacedVertex::VertexTrackType vertexTrackType = getVertexTrackType(pattern);

      if (vertexTrackType != PFDisplacedVertex::T_NOT_FROM_VERTEX) {
        bool bGoodTrack = helper_.isTrackSelected(transTracksRaw[i].track(), vertexTrackType);

        if (bGoodTrack) {
          transTracks.push_back(transTracksRaw[i]);
          transTracksRef.push_back(tracksToFit[i]);
        } else {
          if (debug_)
            cout << "Track rejected nChi2 = " << transTracksRaw[i].track().normalizedChi2()
                 << " pt = " << transTracksRaw[i].track().pt()
                 << " dxy (wrt (0,0,0)) = " << transTracksRaw[i].track().dxy()
                 << " nHits = " << transTracksRaw[i].track().numberOfValidHits() << " nOuterHits = "
                 << transTracksRaw[i].track().hitPattern().numberOfLostHits(HitPattern::MISSING_OUTER_HITS) << endl;
        }
      } else {
        if (debug_) {
          cout << "Remove track because too far away from the vertex:" << endl;
        }
      }
    }
  }

  if (debug_)
    cout << "All Tracks " << transTracksRaw.size() << " with good weight " << transTracks.size() << endl;

  // ---- Refit ---- //
  FitterType vtxFitter = F_NOTDEFINED;

  if (transTracks.size() < 2)
    return true;
  else if (transTracks.size() == 2) {
    if (switchOff2TrackVertex_) {
      if (debug_)
        cout << "Due to probably high pile-up conditions 2 track vertices switched off" << endl;
      return true;
    }
    vtxFitter = F_KALMAN;
  } else if (transTracks.size() > 2 && transTracksRaw.size() > transTracks.size())
    vtxFitter = F_ADAPTIVE;
  else if (transTracks.size() > 2 && transTracksRaw.size() == transTracks.size())
    vtxFitter = F_DONOTREFIT;
  else
    return true;

  if (debug_)
    cout << "Vertex Fitter " << vtxFitter << endl;

  if (vtxFitter == F_KALMAN) {
    KalmanVertexFitter theKalmanFitter(true);
    theRecoVertex = theKalmanFitter.vertex(transTracks, seedPoint);

  } else if (vtxFitter == F_ADAPTIVE) {
    AdaptiveVertexFitter theAdaptiveFitter(GeometricAnnealing(sigmacut_, t_ini_, ratio_),
                                           DefaultLinearizationPointFinder(),
                                           KalmanVertexUpdator<5>(),
                                           KalmanVertexTrackCompatibilityEstimator<5>(),
                                           KalmanVertexSmoother());

    theRecoVertex = theAdaptiveFitter.vertex(transTracks, seedPoint);

  } else if (vtxFitter == F_DONOTREFIT) {
    theRecoVertex = theVertexAdaptiveRaw;
  } else {
    return true;
  }

  // ---- Check if the fitted vertex is valid ---- //

  if (!theRecoVertex.isValid() || theRecoVertex.totalChiSquared() < 0.) {
    if (debug_)
      cout << "Refit failed : valid? " << theRecoVertex.isValid() << " totalChi2 = " << theRecoVertex.totalChiSquared()
           << endl;
    return true;
  }

  // ---- Create vertex ---- //

  Vertex theRecoVtx = theRecoVertex;

  double chi2 = theRecoVtx.chi2();
  double ndf = theRecoVtx.ndof();

  if (chi2 > TMath::ChisquareQuantile(0.95, ndf)) {
    if (debug_)
      cout << "Rejected because of chi2 = " << chi2 << " ndf = " << ndf
           << " confid. level: " << TMath::ChisquareQuantile(0.95, ndf) << endl;
    return true;
  }

  // ---- Create and fill vector of refitted TransientTracks ---- //

  // -----------------------------------------------//
  // ---- Prepare and Fill the Displaced Vertex ----//
  // -----------------------------------------------//

  displacedVertex = theRecoVtx;
  displacedVertex.removeTracks();

  for (unsigned i = 0; i < transTracks.size(); i++) {
    PFTrackHitFullInfo pattern = hitPattern_.analyze(tkerTopo_, tkerGeom_, transTracksRef[i], theRecoVertex);

    PFDisplacedVertex::VertexTrackType vertexTrackType = getVertexTrackType(pattern);

    Track refittedTrack;
    float weight = theRecoVertex.trackWeight(transTracks[i]);

    if (weight < minAdaptWeight_)
      continue;

    try {
      refittedTrack = theRecoVertex.refittedTrack(transTracks[i]).track();
    } catch (cms::Exception& exception) {
      continue;
    }

    if (debug_) {
      cout << "Vertex Track Type = " << vertexTrackType << endl;

      cout << "nHitBeforeVertex = " << pattern.first.first << " nHitAfterVertex = " << pattern.second.first
           << " nMissHitBeforeVertex = " << pattern.first.second << " nMissHitAfterVertex = " << pattern.second.second
           << " Weight = " << weight << endl;
    }

    displacedVertex.addElement(transTracksRef[i], refittedTrack, pattern, vertexTrackType, weight);
  }

  displacedVertex.setPrimaryDirection(helper_.primaryVertex());
  displacedVertex.calcKinematics();

  if (debug_)
    cout << "== End vertexing procedure ==" << endl;

  return false;
}

getTransvLongDiff()

std::pair< float, float > PFDisplacedVertexFinder::getTransvLongDiff ( const GlobalPoint &  Ref, const GlobalPoint &  ToProject  ) const

private

Definition at line 630 of file PFDisplacedVertexFinder.cc.

References PV3DBase< T, PVType, FrameType >::basicVector(), Basic3DVector< T >::mag2(), and mathSSE::sqrt().

Referenced by findSeedsFromCandidate(), and isCloseTo().

                                                                                                       {
  const auto& vRef = Ref.basicVector();
  const auto& vToProject = ToProject.basicVector();
  float vRefMag2 = vRef.mag2();
  float oneOverMag = 1.0f / sqrt(vRefMag2);

  return std::make_pair(fabs(vRef.cross(vToProject).mag() * oneOverMag),
                        fabs((vRef.dot(vToProject) - vRefMag2) * oneOverMag));
}

getVertexTrackType()

reco::PFDisplacedVertex::VertexTrackType PFDisplacedVertexFinder::getVertexTrackType ( PFTrackHitFullInfo &  pairTrackHitInfo ) const

private

Definition at line 641 of file PFDisplacedVertexFinder.cc.

Referenced by fitVertexFromSeed().

                                                {
  unsigned int nHitBeforeVertex = pairTrackHitInfo.first.first;
  unsigned int nHitAfterVertex = pairTrackHitInfo.second.first;

  unsigned int nMissHitBeforeVertex = pairTrackHitInfo.first.second;
  unsigned int nMissHitAfterVertex = pairTrackHitInfo.second.second;

  // For the moment those definitions are given apriori a more general study would be useful to refine those criteria

  if (nHitBeforeVertex <= 1 && nHitAfterVertex >= 3 && nMissHitAfterVertex <= 1)
    return PFDisplacedVertex::T_FROM_VERTEX;
  else if (nHitBeforeVertex >= 3 && nHitAfterVertex <= 1 && nMissHitBeforeVertex <= 1)
    return PFDisplacedVertex::T_TO_VERTEX;
  else if ((nHitBeforeVertex >= 2 && nHitAfterVertex >= 3) || (nHitBeforeVertex >= 3 && nHitAfterVertex >= 2))
    return PFDisplacedVertex::T_MERGED;
  else
    return PFDisplacedVertex::T_NOT_FROM_VERTEX;
}

isCloseTo()

bool PFDisplacedVertexFinder::isCloseTo ( const reco::PFDisplacedVertexSeed &  dv1, const reco::PFDisplacedVertexSeed &  dv2  ) const

private

-----— Tools -----— ///

Definition at line 616 of file PFDisplacedVertexFinder.cc.

References getTransvLongDiff(), longSize_, reco::PFDisplacedVertexSeed::seedPoint(), and transvSize_.

Referenced by mergeSeeds().

                                                                                                                {
  const GlobalPoint& vP1 = dv1.seedPoint();
  const GlobalPoint& vP2 = dv2.seedPoint();

  std::pair<float, float> diffs = getTransvLongDiff(vP1, vP2);
  if (diffs.second > longSize_)
    return false;
  if (diffs.first > transvSize_)
    return false;
  //  if (Delta_Long < longSize_ && Delta_Transv < transvSize_) isCloseTo = true;

  return true;
}

mergeSeeds()

void PFDisplacedVertexFinder::mergeSeeds ( reco::PFDisplacedVertexSeedCollection &  tempDisplacedVertexSeeds, std::vector< bool > &  bLocked  )

private

Sometimes two vertex candidates can be quite close and coming from the same vertex.

Definition at line 189 of file PFDisplacedVertexFinder.cc.

References gather_cfg::cout, debug_, and isCloseTo().

Referenced by findDisplacedVertices().

                                                                {
  // loop over displaced vertex candidates
  // and merge them if they are close to each other
  for (unsigned idv_mother = 0; idv_mother < tempDisplacedVertexSeeds.size(); idv_mother++) {
    if (!bLocked[idv_mother]) {
      for (unsigned idv_daughter = idv_mother + 1; idv_daughter < tempDisplacedVertexSeeds.size(); idv_daughter++) {
        if (!bLocked[idv_daughter]) {
          if (isCloseTo(tempDisplacedVertexSeeds[idv_mother], tempDisplacedVertexSeeds[idv_daughter])) {
            tempDisplacedVertexSeeds[idv_mother].mergeWith(tempDisplacedVertexSeeds[idv_daughter]);
            bLocked[idv_daughter] = true;
            if (debug_)
              cout << "Seeds " << idv_mother << " and " << idv_daughter << " merged" << endl;
          }
        }
      }
    }
  }
}

rejectAndLabelVertex()

bool PFDisplacedVertexFinder::rejectAndLabelVertex ( reco::PFDisplacedVertex &  dv )

private

Definition at line 607 of file PFDisplacedVertexFinder.cc.

References helper_, PFDisplacedVertexHelper::identifyVertex(), reco::PFDisplacedVertex::isFake(), and reco::PFDisplacedVertex::setVertexType().

Referenced by selectAndLabelVertices().

                                                                        {
  PFDisplacedVertex::VertexType vertexType = helper_.identifyVertex(dv);
  dv.setVertexType(vertexType);

  return dv.isFake();
}

selectAndLabelVertices()

void PFDisplacedVertexFinder::selectAndLabelVertices ( reco::PFDisplacedVertexCollection &  tempDisplacedVertices, std::vector< bool > &  bLocked  )

private

Remove potentially fakes displaced vertices.

Definition at line 523 of file PFDisplacedVertexFinder.cc.

References commonTracks(), gather_cfg::cout, debug_, primaryVertexCut_, rejectAndLabelVertex(), rho, tecCut_, tobCut_, and z.

Referenced by findDisplacedVertices().

                                                                            {
  if (debug_)
    cout << " 4.1) Reject vertices " << endl;

  for (unsigned idv = 0; idv < tempDisplacedVertices.size(); idv++) {
    // ---- We accept a vertex only if it is not in TOB in the barrel
    //      and not in TEC in the end caps
    //      and not too much before the first pixel layer            ---- //

    const float rho = tempDisplacedVertices[idv].position().rho();
    const float z = tempDisplacedVertices[idv].position().z();

    if (rho > tobCut_ || rho < primaryVertexCut_ || fabs(z) > tecCut_) {
      if (debug_)
        cout << "Vertex " << idv << " geometrically rejected #rho = " << rho << " z = " << z << endl;

      bLocked[idv] = true;

      continue;
    }

    unsigned nPrimary = tempDisplacedVertices[idv].nPrimaryTracks();
    unsigned nMerged = tempDisplacedVertices[idv].nMergedTracks();
    unsigned nSecondary = tempDisplacedVertices[idv].nSecondaryTracks();

    if (nPrimary + nMerged > 1) {
      bLocked[idv] = true;
      if (debug_)
        cout << "Vertex " << idv << " rejected because two primary or merged tracks" << endl;
    }

    if (nPrimary + nMerged + nSecondary < 2) {
      bLocked[idv] = true;
      if (debug_)
        cout << "Vertex " << idv << " rejected because only one track related to the vertex" << endl;
    }
  }

  if (debug_)
    cout << " 4.2) Check for common vertices" << endl;

  // ---- Among the remaining vertex we shall remove one
  //      of those which have two common tracks          ---- //

  for (unsigned idv_mother = 0; idv_mother < tempDisplacedVertices.size(); idv_mother++) {
    for (unsigned idv_daughter = idv_mother + 1; idv_daughter < tempDisplacedVertices.size(); idv_daughter++) {
      if (!bLocked[idv_daughter] && !bLocked[idv_mother]) {
        const unsigned commonTrks =
            commonTracks(tempDisplacedVertices[idv_daughter], tempDisplacedVertices[idv_mother]);

        if (commonTrks > 1) {
          if (debug_)
            cout << "Vertices " << idv_daughter << " and " << idv_mother << " has many common tracks" << endl;

          // We keep the vertex vertex which contains the most of the tracks

          const int mother_size = tempDisplacedVertices[idv_mother].nTracks();
          const int daughter_size = tempDisplacedVertices[idv_daughter].nTracks();

          if (mother_size > daughter_size)
            bLocked[idv_daughter] = true;
          else if (mother_size < daughter_size)
            bLocked[idv_mother] = true;
          else {
            // If they have the same size we keep the vertex with the best normalised chi2

            const float mother_normChi2 = tempDisplacedVertices[idv_mother].normalizedChi2();
            const float daughter_normChi2 = tempDisplacedVertices[idv_daughter].normalizedChi2();
            if (mother_normChi2 < daughter_normChi2)
              bLocked[idv_daughter] = true;
            else
              bLocked[idv_mother] = true;
          }
        }
      }
    }
  }

  for (unsigned idv = 0; idv < tempDisplacedVertices.size(); idv++)
    if (!bLocked[idv])
      bLocked[idv] = rejectAndLabelVertex(tempDisplacedVertices[idv]);
}

setAVFParameters()

void PFDisplacedVertexFinder::setAVFParameters ( const edm::ParameterSet &  ps )

inline

Definition at line 90 of file PFDisplacedVertexFinder.h.

References edm::ParameterSet::getParameter(), ratio_, sigmacut_, and t_ini_.

Referenced by PFDisplacedVertexProducer::PFDisplacedVertexProducer().

                                                   {
    sigmacut_ = ps.getParameter<double>("sigmacut");
    t_ini_ = ps.getParameter<double>("Tini");
    ratio_ = ps.getParameter<double>("ratio");
  }

setDebug()

void PFDisplacedVertexFinder::setDebug ( bool  debug )

inline

Sets debug printout flag.

Definition at line 68 of file PFDisplacedVertexFinder.h.

References debug, and debug_.

Referenced by PFDisplacedVertexProducer::PFDisplacedVertexProducer().

{ debug_ = debug; }

setEdmParameters()

void PFDisplacedVertexFinder::setEdmParameters ( const MagneticField *  magField, edm::ESHandle< GlobalTrackingGeometry >  globTkGeomHandle, const TrackerTopology *  tkerTopo, const TrackerGeometry *  tkerGeom  )

inline

Sets parameters for track extrapolation and hits study.

Definition at line 71 of file PFDisplacedVertexFinder.h.

References globTkGeomHandle_, magField_, tkerGeom_, and tkerTopo_.

Referenced by PFDisplacedVertexProducer::produce().

                                                         {
    magField_ = magField;
    globTkGeomHandle_ = globTkGeomHandle;
    tkerTopo_ = tkerTopo;
    tkerGeom_ = tkerGeom;
  }

setInput()

void PFDisplacedVertexFinder::setInput

(

const edm::Handle< reco::PFDisplacedVertexCandidateCollection > &

displacedVertexCandidates

)

Set input collections of tracks.

Definition at line 43 of file PFDisplacedVertexFinder.cc.

References displacedVertexCandidates_, edm::HandleBase::isValid(), and edm::Handle< T >::product().

Referenced by PFDisplacedVertexProducer::produce().

                                                                                          {
  if (displacedVertexCandidates.isValid()) {
    displacedVertexCandidates_ = displacedVertexCandidates.product();
  } else {
    displacedVertexCandidates_ = nullptr;
  }
}

setParameters()

void PFDisplacedVertexFinder::setParameters ( double  transvSize, double  longSize, double  primaryVertexCut, double  tobCut, double  tecCut, double  minAdaptWeight, bool  switchOff2TrackVertex  )

inline

--—— Set different algo parameters --—— ///

Sets algo parameters for the vertex finder

Definition at line 51 of file PFDisplacedVertexFinder.h.

References particleFlowDisplacedVertex_cfi::longSize, longSize_, particleFlowDisplacedVertex_cfi::minAdaptWeight, minAdaptWeight_, particleFlowDisplacedVertex_cfi::primaryVertexCut, primaryVertexCut_, particleFlowDisplacedVertex_cfi::switchOff2TrackVertex, switchOff2TrackVertex_, particleFlowDisplacedVertex_cfi::tecCut, tecCut_, particleFlowDisplacedVertex_cfi::tobCut, tobCut_, particleFlowDisplacedVertex_cfi::transvSize, and transvSize_.

Referenced by PFDisplacedVertexProducer::PFDisplacedVertexProducer().

                                                 {
    transvSize_ = transvSize;
    longSize_ = longSize;
    primaryVertexCut_ = primaryVertexCut;
    tobCut_ = tobCut;
    tecCut_ = tecCut;
    minAdaptWeight_ = minAdaptWeight;
    switchOff2TrackVertex_ = switchOff2TrackVertex;
  }

setPrimaryVertex()

void PFDisplacedVertexFinder::setPrimaryVertex ( edm::Handle< reco::VertexCollection >  mainVertexHandle, edm::Handle< reco::BeamSpot >  beamSpotHandle  )

inline

Definition at line 85 of file PFDisplacedVertexFinder.h.

References helper_, and PFDisplacedVertexHelper::setPrimaryVertex().

Referenced by PFDisplacedVertexProducer::produce().

                                                                {
    helper_.setPrimaryVertex(mainVertexHandle, beamSpotHandle);
  }

setTracksSelector()

void PFDisplacedVertexFinder::setTracksSelector ( const edm::ParameterSet &  ps )

inline

Definition at line 81 of file PFDisplacedVertexFinder.h.

References helper_, and PFDisplacedVertexHelper::setTracksSelector().

Referenced by PFDisplacedVertexProducer::PFDisplacedVertexProducer().

{ helper_.setTracksSelector(ps); }

setVertexIdentifier()

void PFDisplacedVertexFinder::setVertexIdentifier ( const edm::ParameterSet &  ps )

inline

Definition at line 83 of file PFDisplacedVertexFinder.h.

References helper_, and PFDisplacedVertexHelper::setVertexIdentifier().

Referenced by PFDisplacedVertexProducer::PFDisplacedVertexProducer().

{ helper_.setVertexIdentifier(ps); }

transferDisplacedVertices()

std::unique_ptr<reco::PFDisplacedVertexCollection> PFDisplacedVertexFinder::transferDisplacedVertices ( )

inline

Returns

unique_ptr to collection of DisplacedVertices

Definition at line 100 of file PFDisplacedVertexFinder.h.

References displacedVertices_, and eostools::move().

Referenced by PFDisplacedVertexProducer::produce().

                                                                             {
    return std::move(displacedVertices_);
  }

Friends And Related Function Documentation

operator<<

std::ostream& operator<< ( std::ostream &  out, const PFDisplacedVertexFinder &  a  )

friend

Definition at line 680 of file PFDisplacedVertexFinder.cc.

                                                                          {
  if (!out)
    return out;
  out << setprecision(3) << setw(5) << endl;
  out << "" << endl;
  out << " ====================================== " << endl;
  out << " ====== Displaced Vertex Finder ======= " << endl;
  out << " ====================================== " << endl;
  out << " " << endl;

  a.helper_.Dump();
  out << "" << endl
      << " Adaptive Vertex Fitter parameters are :" << endl
      << " sigmacut = " << a.sigmacut_ << " T_ini = " << a.t_ini_ << " ratio = " << a.ratio_ << endl
      << endl;

  const std::unique_ptr<reco::PFDisplacedVertexCollection>& displacedVertices_ = std::move(a.displacedVertices());

  if (!displacedVertices_.get()) {
    out << "displacedVertex already transfered" << endl;
  } else {
    out << "Number of displacedVertices found : " << displacedVertices_->size() << endl << endl;

    int i = -1;

    for (PFDisplacedVertexFinder::IDV idv = displacedVertices_->begin(); idv != displacedVertices_->end(); idv++) {
      i++;
      out << i << " ";
      idv->Dump();
      out << "" << endl;
    }
  }

  return out;
}

Member Data Documentation

debug_

bool PFDisplacedVertexFinder::debug_

private

If true, debug printouts activated.

Definition at line 166 of file PFDisplacedVertexFinder.h.

Referenced by findDisplacedVertices(), findSeedsFromCandidate(), fitVertexFromSeed(), mergeSeeds(), selectAndLabelVertices(), and setDebug().

displacedVertexCandidates_

reco::PFDisplacedVertexCandidateCollection const* PFDisplacedVertexFinder::displacedVertexCandidates_

private

-----— Members -----— ///

Definition at line 143 of file PFDisplacedVertexFinder.h.

Referenced by findDisplacedVertices(), and setInput().

displacedVertices_

std::unique_ptr<reco::PFDisplacedVertexCollection> PFDisplacedVertexFinder::displacedVertices_

private

Definition at line 144 of file PFDisplacedVertexFinder.h.

Referenced by displacedVertices(), findDisplacedVertices(), and transferDisplacedVertices().

globTkGeomHandle_

edm::ESHandle<GlobalTrackingGeometry> PFDisplacedVertexFinder::globTkGeomHandle_

private

Tracker geometry for discerning hit positions.

Definition at line 169 of file PFDisplacedVertexFinder.h.

Referenced by fitVertexFromSeed(), and setEdmParameters().

helper_

PFDisplacedVertexHelper PFDisplacedVertexFinder::helper_

private

Definition at line 180 of file PFDisplacedVertexFinder.h.

Referenced by fitVertexFromSeed(), rejectAndLabelVertex(), setPrimaryVertex(), setTracksSelector(), and setVertexIdentifier().

hitPattern_

PFCheckHitPattern PFDisplacedVertexFinder::hitPattern_

private

Definition at line 178 of file PFDisplacedVertexFinder.h.

Referenced by fitVertexFromSeed().

longSize_

float PFDisplacedVertexFinder::longSize_

private

Definition at line 151 of file PFDisplacedVertexFinder.h.

Referenced by findSeedsFromCandidate(), isCloseTo(), and setParameters().

magField_

const MagneticField* PFDisplacedVertexFinder::magField_

private

to be able to extrapolate tracks f

Definition at line 176 of file PFDisplacedVertexFinder.h.

Referenced by fitVertexFromSeed(), and setEdmParameters().

minAdaptWeight_

double PFDisplacedVertexFinder::minAdaptWeight_

private

Definition at line 155 of file PFDisplacedVertexFinder.h.

Referenced by fitVertexFromSeed(), and setParameters().

primaryVertexCut_

double PFDisplacedVertexFinder::primaryVertexCut_

private

Definition at line 152 of file PFDisplacedVertexFinder.h.

Referenced by fitVertexFromSeed(), selectAndLabelVertices(), and setParameters().

ratio_

double PFDisplacedVertexFinder::ratio_

private

Definition at line 163 of file PFDisplacedVertexFinder.h.

Referenced by fitVertexFromSeed(), and setAVFParameters().

sigmacut_

double PFDisplacedVertexFinder::sigmacut_

private

Adaptive Vertex Fitter parameters.

Definition at line 161 of file PFDisplacedVertexFinder.h.

Referenced by fitVertexFromSeed(), and setAVFParameters().

switchOff2TrackVertex_

bool PFDisplacedVertexFinder::switchOff2TrackVertex_

private

Definition at line 157 of file PFDisplacedVertexFinder.h.

Referenced by fitVertexFromSeed(), and setParameters().

t_ini_

double PFDisplacedVertexFinder::t_ini_

private

Definition at line 162 of file PFDisplacedVertexFinder.h.

Referenced by fitVertexFromSeed(), and setAVFParameters().

tecCut_

double PFDisplacedVertexFinder::tecCut_

private

Definition at line 154 of file PFDisplacedVertexFinder.h.

Referenced by fitVertexFromSeed(), selectAndLabelVertices(), and setParameters().

tkerGeom_

const TrackerGeometry* PFDisplacedVertexFinder::tkerGeom_

private

Definition at line 173 of file PFDisplacedVertexFinder.h.

Referenced by fitVertexFromSeed(), and setEdmParameters().

tkerTopo_

const TrackerTopology* PFDisplacedVertexFinder::tkerTopo_

private

doc?

Definition at line 172 of file PFDisplacedVertexFinder.h.

Referenced by fitVertexFromSeed(), and setEdmParameters().

tobCut_

double PFDisplacedVertexFinder::tobCut_

private

Definition at line 153 of file PFDisplacedVertexFinder.h.

Referenced by fitVertexFromSeed(), selectAndLabelVertices(), and setParameters().

transvSize_

float PFDisplacedVertexFinder::transvSize_

private

--—— Parameters --—— ///

Algo parameters for the vertex finder

Definition at line 150 of file PFDisplacedVertexFinder.h.

Referenced by findSeedsFromCandidate(), isCloseTo(), and setParameters().