#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 -------- /// | |
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 -------- /// | |
PFDisplacedVertexFinder () | |
void | setAVFParameters (edm::ParameterSet ps) |
void | setDebug (bool debug) |
Sets debug printout flag. | |
void | setEdmParameters (const MagneticField *magField, edm::ESHandle< GlobalTrackingGeometry > globTkGeomHandle, edm::ESHandle< TrackerGeometry > tkerGeomHandle) |
Sets parameters for track extrapolation and hits study. | |
void | setInput (const edm::Handle< reco::PFDisplacedVertexCandidateCollection > &) |
Set input collections of tracks. | |
void | setParameters (double transvSize, double longSize, double primaryVertexCut, double tobCut, double tecCut, double minAdaptWeight) |
-------- Set different algo parameters -------- /// | |
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 -------- /// | |
bool | fitVertexFromSeed (reco::PFDisplacedVertexSeed &, reco::PFDisplacedVertex &) |
Fit one by one the vertex points with associated tracks to get displaced vertices. | |
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 -------- /// | |
void | mergeSeeds (reco::PFDisplacedVertexSeedCollection &, std::vector< bool > &bLocked) |
Sometimes two vertex candidates can be quite close and coming from the same vertex. | |
bool | rejectAndLabelVertex (reco::PFDisplacedVertex &dv) |
void | selectAndLabelVertices (reco::PFDisplacedVertexCollection &, std::vector< bool > &) |
Remove potentially fakes displaced vertices. | |
Private Attributes | |
bool | debug_ |
If true, debug printouts activated. | |
std::auto_ptr < reco::PFDisplacedVertexCandidateCollection > | displacedVertexCandidates_ |
-------- Members -------- /// | |
std::auto_ptr < reco::PFDisplacedVertexCollection > | displacedVertices_ |
edm::ESHandle < GlobalTrackingGeometry > | globTkGeomHandle_ |
Tracker geometry for discerning hit positions. | |
PFDisplacedVertexHelper | helper_ |
PFCheckHitPattern | hitPattern_ |
double | longSize_ |
const MagneticField * | magField_ |
to be able to extrapolate tracks f | |
double | minAdaptWeight_ |
double | primaryVertexCut_ |
double | ratio_ |
double | sigmacut_ |
Adaptive Vertex Fitter parameters. | |
double | t_ini_ |
double | tecCut_ |
edm::ESHandle< TrackerGeometry > | tkerGeomHandle_ |
doc? | |
double | tobCut_ |
double | transvSize_ |
-------- Parameters -------- /// | |
Friends | |
std::ostream & | operator<< (std::ostream &, const PFDisplacedVertexFinder &) |
Definition at line 32 of file PFDisplacedVertexFinder.h.
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.
Fitter Type.
Definition at line 51 of file PFDisplacedVertexFinder.h.
{ F_NOTDEFINED, F_DONOTREFIT, F_KALMAN, F_ADAPTIVE };
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.
{}
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<<().
{return displacedVertices_;}
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(), gather_cfg::cout, debug_, reco::PFDisplacedVertexSeed::Dump(), reco::PFDisplacedVertexSeed::elements(), 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(), TransientVertex::trackWeight(), KalmanVertexFitter::vertex(), AdaptiveVertexFitter::vertex(), CommonMethods::weight(), PV3DBase< T, PVType, FrameType >::x(), PV3DBase< T, PVType, FrameType >::y(), PV3DBase< T, PVType, FrameType >::z(), and 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 & | , |
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 & | , |
std::vector< bool > & | |||
) | [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] |
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 longSize_, minAdaptWeight_, primaryVertexCut_, tecCut_, tobCut_, 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] |
Definition at line 109 of file PFDisplacedVertexFinder.h.
References displacedVertices_.
{return displacedVertices_;}
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; }
bool PFDisplacedVertexFinder::debug_ [private] |
If true, debug printouts activated.
Definition at line 177 of file PFDisplacedVertexFinder.h.
Referenced by findDisplacedVertices(), findSeedsFromCandidate(), fitVertexFromSeed(), mergeSeeds(), selectAndLabelVertices(), and setDebug().
std::auto_ptr< reco::PFDisplacedVertexCandidateCollection > PFDisplacedVertexFinder::displacedVertexCandidates_ [private] |
-------- Members -------- ///
Definition at line 155 of file PFDisplacedVertexFinder.h.
Referenced by findDisplacedVertices(), and setInput().
std::auto_ptr< reco::PFDisplacedVertexCollection > PFDisplacedVertexFinder::displacedVertices_ [private] |
Definition at line 156 of file PFDisplacedVertexFinder.h.
Referenced by displacedVertices(), findDisplacedVertices(), and transferDisplacedVertices().
Tracker geometry for discerning hit positions.
Definition at line 180 of file PFDisplacedVertexFinder.h.
Referenced by fitVertexFromSeed(), and setEdmParameters().
Definition at line 191 of file PFDisplacedVertexFinder.h.
Referenced by fitVertexFromSeed(), operator<<(), rejectAndLabelVertex(), setPrimaryVertex(), setTracksSelector(), and setVertexIdentifier().
Definition at line 189 of file PFDisplacedVertexFinder.h.
Referenced by fitVertexFromSeed().
double PFDisplacedVertexFinder::longSize_ [private] |
Definition at line 163 of file PFDisplacedVertexFinder.h.
Referenced by findSeedsFromCandidate(), isCloseTo(), and setParameters().
const MagneticField* PFDisplacedVertexFinder::magField_ [private] |
to be able to extrapolate tracks f
Definition at line 186 of file PFDisplacedVertexFinder.h.
Referenced by fitVertexFromSeed(), and setEdmParameters().
double PFDisplacedVertexFinder::minAdaptWeight_ [private] |
Definition at line 167 of file PFDisplacedVertexFinder.h.
Referenced by fitVertexFromSeed(), and setParameters().
double PFDisplacedVertexFinder::primaryVertexCut_ [private] |
Definition at line 164 of file PFDisplacedVertexFinder.h.
Referenced by fitVertexFromSeed(), selectAndLabelVertices(), and setParameters().
double PFDisplacedVertexFinder::ratio_ [private] |
Definition at line 173 of file PFDisplacedVertexFinder.h.
Referenced by fitVertexFromSeed(), operator<<(), and setAVFParameters().
double PFDisplacedVertexFinder::sigmacut_ [private] |
Adaptive Vertex Fitter parameters.
Definition at line 171 of file PFDisplacedVertexFinder.h.
Referenced by fitVertexFromSeed(), operator<<(), and setAVFParameters().
double PFDisplacedVertexFinder::t_ini_ [private] |
Definition at line 172 of file PFDisplacedVertexFinder.h.
Referenced by fitVertexFromSeed(), operator<<(), and setAVFParameters().
double PFDisplacedVertexFinder::tecCut_ [private] |
Definition at line 166 of file PFDisplacedVertexFinder.h.
Referenced by fitVertexFromSeed(), selectAndLabelVertices(), and setParameters().
doc?
Definition at line 183 of file PFDisplacedVertexFinder.h.
Referenced by fitVertexFromSeed(), and setEdmParameters().
double PFDisplacedVertexFinder::tobCut_ [private] |
Definition at line 165 of file PFDisplacedVertexFinder.h.
Referenced by fitVertexFromSeed(), selectAndLabelVertices(), and setParameters().
double PFDisplacedVertexFinder::transvSize_ [private] |
-------- Parameters -------- ///
Algo parameters for the vertex finder
Definition at line 162 of file PFDisplacedVertexFinder.h.
Referenced by findSeedsFromCandidate(), isCloseTo(), and setParameters().