#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 32 of file PFDisplacedVertexFinder.h.

Member Typedef Documentation

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

Definition at line 43 of file PFDisplacedVertexFinder.h.

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

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

Definition at line 42 of file PFDisplacedVertexFinder.h.

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

Definition at line 46 of file PFDisplacedVertexFinder.h.

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

Definition at line 45 of file PFDisplacedVertexFinder.h.

Member Enumeration Documentation

enum PFDisplacedVertexFinder::FitterType

Fitter Type.

Enumerator
F_NOTDEFINED
F_DONOTREFIT
F_KALMAN
F_ADAPTIVE

Definition at line 49 of file PFDisplacedVertexFinder.h.

                   {
     F_NOTDEFINED,
     F_DONOTREFIT,
     F_KALMAN,
     F_ADAPTIVE
   };

Constructor & Destructor Documentation

PFDisplacedVertexFinder::PFDisplacedVertexFinder ( )

Definition at line 28 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 ( )

Definition at line 40 of file PFDisplacedVertexFinder.cc.

40 {}

Member Function Documentation

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

private

Definition at line 739 of file PFDisplacedVertexFinder.cc.

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

Referenced by displacedVertices(), and 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;
 
 }

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

inline

Definition at line 112 of file PFDisplacedVertexFinder.h.

References commonTracks(), displacedVertices_, findDisplacedVertices(), findSeedsFromCandidate(), fitVertexFromSeed(), getTransvLongDiff(), getVertexTrackType(), isCloseTo(), mergeSeeds(), eostools::move(), operator<<, rejectAndLabelVertex(), and selectAndLabelVertices().

Referenced by operator<<().

112 {return std::move(displacedVertices_);}

PFDisplacedVertexFinder::displacedVertices_

std::unique_ptr< reco::PFDisplacedVertexCollection > displacedVertices_

Definition: PFDisplacedVertexFinder.h:155

eostools.move

def move(src, dest)

Definition: eostools.py:511

void PFDisplacedVertexFinder::findDisplacedVertices ( )

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

Definition at line 58 of file PFDisplacedVertexFinder.cc.

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

Referenced by displacedVertices().

                                                {
 
   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_.reset( new 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;
 
 
 }

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 143 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 displacedVertices(), and 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));
 
 
 
   }
 
 
 }

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 229 of file PFDisplacedVertexFinder.cc.

References reco::PFDisplacedVertex::addElement(), patPFMETCorrections_cff::algo, PFCheckHitPattern::analyze(), reco::PFDisplacedVertex::calcKinematics(), vertices_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(), listBenchmarks::pattern, reco::Vertex::position(), 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(), HiIsolationCommonParameters_cff::track, reco::TransientTrack::track(), TransientVertex::trackWeight(), KalmanVertexFitter::vertex(), AdaptiveVertexFitter::vertex(), badGlobalMuonTaggersAOD_cff::vtx, PV3DBase< T, PVType, FrameType >::x(), PV3DBase< T, PVType, FrameType >::y(), z, and PV3DBase< T, PVType, FrameType >::z().

Referenced by displacedVertices(), and 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;
 
 }

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

private

Definition at line 704 of file PFDisplacedVertexFinder.cc.

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

Referenced by displacedVertices(), 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));
 }

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

private

Definition at line 716 of file PFDisplacedVertexFinder.cc.

Referenced by displacedVertices(), and 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;
 }

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

private

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

Definition at line 688 of file PFDisplacedVertexFinder.cc.

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

Referenced by displacedVertices(), and 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;
 
 }

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 198 of file PFDisplacedVertexFinder.cc.

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

Referenced by displacedVertices(), and 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; 
       }
     }
       }
     }
   }
 
 }

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

private

Definition at line 675 of file PFDisplacedVertexFinder.cc.

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

Referenced by displacedVertices(), and selectAndLabelVertices().

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

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

private

Remove potentially fakes displaced vertices.

Definition at line 582 of file PFDisplacedVertexFinder.cc.

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

Referenced by displacedVertices(), and 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]);
   
 
 }

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

inline

Definition at line 99 of file PFDisplacedVertexFinder.h.

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

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

void PFDisplacedVertexFinder::setDebug ( bool debug )

inline

Sets debug printout flag.

Definition at line 73 of file PFDisplacedVertexFinder.h.

References debug, and debug_.

73 {debug_ = debug;}

PFDisplacedVertexFinder::debug_

bool debug_

If true, debug printouts activated.

Definition: PFDisplacedVertexFinder.h:178

debug

#define debug

Definition: HDRShower.cc:19

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 76 of file PFDisplacedVertexFinder.h.

References globTkGeomHandle_, magField_, tkerGeom_, and tkerTopo_.

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

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 setAVFParameters().

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

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 60 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_.

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

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

inline

Definition at line 94 of file PFDisplacedVertexFinder.h.

References helper_, and PFDisplacedVertexHelper::setPrimaryVertex().

                                                      {
     helper_.setPrimaryVertex(mainVertexHandle, beamSpotHandle);
   }

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

inline

Definition at line 86 of file PFDisplacedVertexFinder.h.

References helper_, and PFDisplacedVertexHelper::setTracksSelector().

                                                    {
     helper_.setTracksSelector(ps);
   }

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

inline

Definition at line 90 of file PFDisplacedVertexFinder.h.

References helper_, and PFDisplacedVertexHelper::setVertexIdentifier().

                                                      {
     helper_.setVertexIdentifier(ps);
   }

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

inline

Returns: unique_ptr to collection of DisplacedVertices

Definition at line 110 of file PFDisplacedVertexFinder.h.

References displacedVertices_, and eostools::move().

110 {return std::move(displacedVertices_);}

PFDisplacedVertexFinder::displacedVertices_

std::unique_ptr< reco::PFDisplacedVertexCollection > displacedVertices_

Definition: PFDisplacedVertexFinder.h:155

eostools.move

def move(src, dest)

Definition: eostools.py:511

Friends And Related Function Documentation

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

friend

Definition at line 758 of file PFDisplacedVertexFinder.cc.

Referenced by displacedVertices().

                                                                           {
 
   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;
 }