#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::PFDisplacedVertexCandidateCollection::iterator	IDVC

typedef reco::PFDisplacedVertexSeedCollection::iterator	IDVS

typedef std::set < reco::TrackBaseRef > ::iterator	IEset
	-----— Useful Types -----— /// More...

typedef std::pair < PFTrackHitInfo, PFTrackHitInfo >	PFTrackHitFullInfo

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

Public Member Functions
const std::auto_ptr < reco::PFDisplacedVertexCollection > &	displacedVertices () const

void	findDisplacedVertices ()
	-----— Main function which find vertices -----— /// More...

	PFDisplacedVertexFinder ()

void	setAVFParameters (edm::ParameterSet ps)

void	setDebug (bool debug)
	Sets debug printout flag. More...

void	setEdmParameters (const MagneticField *magField, edm::ESHandle< GlobalTrackingGeometry > globTkGeomHandle, edm::ESHandle< TrackerGeometry > tkerGeomHandle)
	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)
	--—— Set different algo parameters --—— /// More...

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

void	setTracksSelector (edm::ParameterSet ps)

void	setVertexIdentifier (edm::ParameterSet ps)

std::auto_ptr < reco::PFDisplacedVertexCollection >	transferDisplacedVertices ()

	~PFDisplacedVertexFinder ()

Private Member Functions
unsigned	commonTracks (const reco::PFDisplacedVertex &, const reco::PFDisplacedVertex &) const

void	findSeedsFromCandidate (reco::PFDisplacedVertexCandidate &, reco::PFDisplacedVertexSeedCollection &)
	--—— Different steps of the finder algorithm --—— /// More...

bool	fitVertexFromSeed (reco::PFDisplacedVertexSeed &, reco::PFDisplacedVertex &)
	Fit one by one the vertex points with associated tracks to get displaced vertices. More...

double	getLongDiff (const GlobalPoint &, const GlobalPoint &) const

double	getLongProj (const GlobalPoint &, const GlobalVector &) const

double	getTransvDiff (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...

std::auto_ptr < reco::PFDisplacedVertexCandidateCollection >	displacedVertexCandidates_
	-----— Members -----— /// More...

std::auto_ptr < reco::PFDisplacedVertexCollection >	displacedVertices_

edm::ESHandle < GlobalTrackingGeometry >	globTkGeomHandle_
	Tracker geometry for discerning hit positions. More...

PFDisplacedVertexHelper	helper_

PFCheckHitPattern	hitPattern_

double	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...

double	t_ini_

double	tecCut_

edm::ESHandle< TrackerGeometry >	tkerGeomHandle_
	doc? More...

double	tobCut_

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

typedef reco::PFDisplacedVertexCandidateCollection::iterator PFDisplacedVertexFinder::IDVC

Definition at line 43 of file PFDisplacedVertexFinder.h.

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

Definition at line 44 of file PFDisplacedVertexFinder.h.

typedef std::set< reco::TrackBaseRef >::iterator PFDisplacedVertexFinder::IEset

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

Definition at line 42 of file PFDisplacedVertexFinder.h.

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

Definition at line 48 of file PFDisplacedVertexFinder.h.

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

Definition at line 47 of file PFDisplacedVertexFinder.h.

Member Enumeration Documentation

enum PFDisplacedVertexFinder::FitterType

Fitter Type.

Enumerator
F_NOTDEFINED
F_DONOTREFIT
F_KALMAN
F_ADAPTIVE

Definition at line 51 of file PFDisplacedVertexFinder.h.

                   {
     F_NOTDEFINED,
     F_DONOTREFIT,
     F_KALMAN,
     F_ADAPTIVE
   };

Constructor & Destructor Documentation

PFDisplacedVertexFinder::PFDisplacedVertexFinder ( )

Definition at line 27 of file PFDisplacedVertexFinder.cc.

                                                  : 
   displacedVertexCandidates_( new PFDisplacedVertexCandidateCollection ),
   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 39 of file PFDisplacedVertexFinder.cc.

39 {}

Member Function Documentation

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

private

Definition at line 741 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;
 
 }

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

inline

Definition at line 111 of file PFDisplacedVertexFinder.h.

References displacedVertices_.

Referenced by operator<<().

111 {return displacedVertices_;}

PFDisplacedVertexFinder::displacedVertices_

std::auto_ptr< reco::PFDisplacedVertexCollection > displacedVertices_

Definition: PFDisplacedVertexFinder.h:156

void PFDisplacedVertexFinder::findDisplacedVertices ( )

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

Definition at line 65 of file PFDisplacedVertexFinder.cc.

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

                                                {
 
   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 );
 
 
   // 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(IDVC idvc = displacedVertexCandidates_->begin(); 
       idvc != displacedVertexCandidates_->end(); idvc++) {
 
     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.push_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	(	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 147 of file PFDisplacedVertexFinder.cc.

References gather_cfg::cout, reco::PFDisplacedVertexCandidate::dcaPoint(), debug_, getLongDiff(), getTransvDiff(), 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();
       double Delta_Long = getLongDiff(vertexPoint, dcaPoint);
       double Delta_Transv = getTransvDiff(vertexPoint, dcaPoint);
       if (Delta_Long > longSize_) continue;
       if (Delta_Transv > 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--;
       bNeedNewCandidate = false;
     }
 
 
     
     (*idvc_current).updateSeedPoint(dcaPoint, vertexCandidate.tref(ie1), vertexCandidate.tref(ie2));
 
 
 
   }
 
 
 }

bool PFDisplacedVertexFinder::fitVertexFromSeed	(	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 234 of file PFDisplacedVertexFinder.cc.

References reco::PFDisplacedVertex::addElement(), PFCheckHitPattern::analyze(), reco::PFDisplacedVertex::calcKinematics(), reco::Vertex::chi2(), ExpressReco_HICollisions_FallBack::chi2, gather_cfg::cout, debug_, reco::PFDisplacedVertexSeed::Dump(), reco::PFDisplacedVertexSeed::elements(), cmsCodeRules.cppFunctionSkipper::exception, F_ADAPTIVE, F_DONOTREFIT, F_KALMAN, F_NOTDEFINED, getVertexTrackType(), globTkGeomHandle_, helper_, hitPattern_, i, PFDisplacedVertexHelper::isTrackSelected(), TransientVertex::isValid(), magField_, minAdaptWeight_, reco::Vertex::ndof(), 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(), t_ini_, tecCut_, tkerGeomHandle_, tobCut_, TransientVertex::totalChiSquared(), reco::TransientTrack::track(), ExpressReco_HICollisions_FallBack::track, TransientVertex::trackWeight(), KalmanVertexFitter::vertex(), AdaptiveVertexFitter::vertex(), CommonMethods::weight(), 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 ----
 
   set < TrackBaseRef, PFDisplacedVertexSeed::Compare > tracksToFit = displacedVertexSeed.elements();
   GlobalPoint seedPoint = displacedVertexSeed.seedPoint();
 
   vector<TransientTrack> transTracks;
   vector<TransientTrack> transTracksRaw;
   vector<TrackBaseRef> transTracksRef;
   vector<TrackBaseRef> transTracksRefRaw;
 
   transTracks.reserve(tracksToFit.size());
   transTracksRaw.reserve(tracksToFit.size());
   transTracksRef.reserve(tracksToFit.size());
   transTracksRefRaw.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(IEset ie = tracksToFit.begin(); ie !=  tracksToFit.end(); ie++){
     TransientTrack tmpTk( *((*ie).get()), magField_, globTkGeomHandle_);
     transTracksRaw.push_back( tmpTk );
     transTracksRefRaw.push_back( *ie );
     if ( (*ie)->algo()-4 > 3 ) nStep45++;
     if ( (*ie)->algo()-4 < 0 ||(*ie)->algo()-4 > 5 ) nNotIterative++;
   }
 
   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;
 
     GlobalError globalError;
 
     theVertexAdaptiveRaw = TransientVertex(seedPoint,  globalError, transTracksRaw, 1.);
 
 
 
   } else {
 
 
 
     if (debug_) cout << "Raw fit done." << endl;
     
     AdaptiveVertexFitter theAdaptiveFitterRaw(GeometricAnnealing(sigmacut_, t_ini_, ratio_),
                           DefaultLinearizationPointFinder(),
                           KalmanVertexUpdator<5>(), 
                           KalmanVertexTrackCompatibilityEstimator<5>(), 
                           KalmanVertexSmoother() );
 
 
   
     if ( transTracksRaw.size() < 10 ){
 
       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();
 
       if (rho < primaryVertexCut_) {
     if (debug_) cout << "KFT Vertex geometrically rejected with  tracks #rho = " << rho << endl;
     return true;
       }
 
 
       theVertexAdaptiveRaw = theAdaptiveFitterRaw.vertex(transTracksRaw, theVertexAdaptiveRaw.position());
 
 
     } else {
 
 
       theVertexAdaptiveRaw = theAdaptiveFitterRaw.vertex(transTracksRaw, seedPoint);
 
     }
 
     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(tkerGeomHandle_, transTracksRefRaw[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(transTracksRefRaw[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().trackerExpectedHitsOuter().numberOfHits() << 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) 
     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 = (PFDisplacedVertex) theRecoVtx;
   displacedVertex.removeTracks();
   
   for(unsigned i = 0; i < transTracks.size();i++) {
     
     PFTrackHitFullInfo pattern = hitPattern_.analyze(tkerGeomHandle_, 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;
 
 }

double PFDisplacedVertexFinder::getLongDiff	(	const GlobalPoint &	Ref,
		const GlobalPoint &	ToProject
	)		const

private

Definition at line 688 of file PFDisplacedVertexFinder.cc.

References Basic3DVector< T >::dot(), Basic3DVector< T >::mag(), and Basic3DVector< T >::mag2().

Referenced by findSeedsFromCandidate(), and isCloseTo().

                                                                                                {
 
   Basic3DVector<double>vRef(Ref);
   Basic3DVector<double>vToProject(ToProject);
   return fabs((vRef.dot(vToProject)-vRef.mag2())/vRef.mag());
 
 }

double PFDisplacedVertexFinder::getLongProj	(	const GlobalPoint &	Ref,
		const GlobalVector &	ToProject
	)		const

private

Definition at line 697 of file PFDisplacedVertexFinder.cc.

References Basic3DVector< T >::dot(), and Basic3DVector< T >::mag().

                                                                                                 {
 
   Basic3DVector<double>vRef(Ref);
   Basic3DVector<double>vToProject(ToProject);
   return (vRef.dot(vToProject))/vRef.mag();
 
 
 }

double PFDisplacedVertexFinder::getTransvDiff	(	const GlobalPoint &	Ref,
		const GlobalPoint &	ToProject
	)		const

private

Definition at line 708 of file PFDisplacedVertexFinder.cc.

References Basic3DVector< T >::cross(), and Basic3DVector< T >::mag().

Referenced by findSeedsFromCandidate(), and isCloseTo().

                                                                                                  {
 
   Basic3DVector<double>vRef(Ref);
   Basic3DVector<double>vToProject(ToProject);
   return fabs(vRef.cross(vToProject).mag()/vRef.mag());
 
 }

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

private

Definition at line 718 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;
 }

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

private

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

Definition at line 671 of file PFDisplacedVertexFinder.cc.

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

Referenced by mergeSeeds().

                                                                                                            {
 
   bool isCloseTo = false;
 
   const GlobalPoint vP1 = dv1.seedPoint();
   const GlobalPoint vP2 = dv2.seedPoint();
 
   double Delta_Long = getLongDiff(vP1, vP2);
   double Delta_Transv = getTransvDiff(vP1, vP2);
   if (Delta_Long < longSize_ && Delta_Transv < transvSize_) isCloseTo = true;
 
   return isCloseTo;
 
 }

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 204 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] && 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 658 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();
 
 }

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

private

Remove potentially fakes displaced vertices.

Definition at line 565 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]);
   
 
 }

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

inline

Definition at line 98 of file PFDisplacedVertexFinder.h.

References edm::ParameterSet::getParameter(), ratio_, 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 74 of file PFDisplacedVertexFinder.h.

References debug, and debug_.

74 {debug_ = debug;}

PFDisplacedVertexFinder::debug_

bool debug_

If true, debug printouts activated.

Definition: PFDisplacedVertexFinder.h:177

debug

#define debug

Definition: MEtoEDMFormat.h:34

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

inline

Sets parameters for track extrapolation and hits study.

Definition at line 77 of file PFDisplacedVertexFinder.h.

References globTkGeomHandle_, magField_, and tkerGeomHandle_.

                                                          { 
     magField_ = magField; 
     globTkGeomHandle_ = globTkGeomHandle;
     tkerGeomHandle_ = tkerGeomHandle; 
   }

void PFDisplacedVertexFinder::setInput ( const edm::Handle< reco::PFDisplacedVertexCandidateCollection > & displacedVertexCandidates )

Set input collections of tracks.

Definition at line 42 of file PFDisplacedVertexFinder.cc.

References displacedVertexCandidates_, i, and edm::HandleBase::isValid().

                                                                                                         {
 
   if( displacedVertexCandidates_.get() ) {
     displacedVertexCandidates_->clear();
   }
   else 
     displacedVertexCandidates_.reset( new PFDisplacedVertexCandidateCollection );
 
 
   if(displacedVertexCandidates.isValid()) {
     for(unsigned i=0;i<displacedVertexCandidates->size(); i++) {
       PFDisplacedVertexCandidateRef dvcref( displacedVertexCandidates, i); 
       displacedVertexCandidates_->push_back( (*dvcref));
     }
   }
 
 }

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

inline

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

Sets algo parameters for the vertex finder

Definition at line 62 of file PFDisplacedVertexFinder.h.

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

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

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

inline

Definition at line 93 of file PFDisplacedVertexFinder.h.

References helper_, and PFDisplacedVertexHelper::setPrimaryVertex().

                                                      {
     helper_.setPrimaryVertex(mainVertexHandle, beamSpotHandle);
   }

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

inline

Definition at line 85 of file PFDisplacedVertexFinder.h.

References helper_, and PFDisplacedVertexHelper::setTracksSelector().

                                             {
     helper_.setTracksSelector(ps);
   }

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

inline

Definition at line 89 of file PFDisplacedVertexFinder.h.

References helper_, and PFDisplacedVertexHelper::setVertexIdentifier().

                                               {
     helper_.setVertexIdentifier(ps);
   }

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

inline

Returns: auto_ptr to collection of DisplacedVertices

Definition at line 109 of file PFDisplacedVertexFinder.h.

References displacedVertices_.

109 {return displacedVertices_;}

PFDisplacedVertexFinder::displacedVertices_

std::auto_ptr< reco::PFDisplacedVertexCollection > displacedVertices_

Definition: PFDisplacedVertexFinder.h:156

Friends And Related Function Documentation

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

friend

Definition at line 760 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::auto_ptr< reco::PFDisplacedVertexCollection >& displacedVertices_
     = 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;
 }