53 declareInterface<IVertexFinder>(
this);
62 ATH_MSG_FATAL(
" Split vertices cannot be obtained if beam spot constraint "
63 "is true! Change settings...");
64 return StatusCode::FAILURE;
70 return StatusCode::FAILURE;
75 return StatusCode::FAILURE;
80 return StatusCode::FAILURE;
85 return StatusCode::FAILURE;
92 return StatusCode::FAILURE;
99 return StatusCode::SUCCESS;
103 struct xAODVertex_pair
118 std::pair<xAOD::VertexContainer*, xAOD::VertexAuxContainer*>
124 " Number of input tracks before track selection: " << trackTES->
size());
129 std::vector<Trk::ITrackLink*> selectedTracks;
133 bool selectionPassed;
134 for (TrackDataVecIter itr = (*trackTES).begin(); itr != (*trackTES).end();
140 static_cast<bool>(
m_trkFilter->accept(**itr, &beamPosition));
143 selectionPassed =
static_cast<bool>(
m_trkFilter->accept(**itr, &
null));
145 if (selectionPassed) {
150 selectedTracks.push_back(linkTT);
155 <<
" survived the preselection.");
157 std::pair<xAOD::VertexContainer*, xAOD::VertexAuxContainer*>
158 returnContainers =
findVertex(ctx, selectedTracks);
160 return returnContainers;
163 std::pair<xAOD::VertexContainer*, xAOD::VertexAuxContainer*>
165 const EventContext& ctx,
168 ATH_MSG_DEBUG(
" Number of input tracks before track selection: "
169 << trackParticles->
size());
174 std::vector<Trk::ITrackLink*> selectedTracks;
176 using TrackParticleDataVecIter =
179 bool selectionPassed;
180 for (TrackParticleDataVecIter itr = trackParticles->
begin();
181 itr != trackParticles->
end();
189 beamSpot->beamVtx().covariancePosition());
191 static_cast<bool>(
m_trkFilter->accept(**itr, &beamPosition));
198 vertexError.setZero();
199 null.setCovariancePosition(vertexError);
200 selectionPassed =
static_cast<bool>(
m_trkFilter->accept(**itr, &
null));
203 if (selectionPassed) {
209 selectedTracks.push_back(linkTT);
214 << selectedTracks.size()
215 <<
" survived the preselection.");
217 std::pair<xAOD::VertexContainer*, xAOD::VertexAuxContainer*>
218 returnContainers =
findVertex(ctx, selectedTracks);
220 return returnContainers;
223 std::pair<xAOD::VertexContainer*, xAOD::VertexAuxContainer*>
225 const EventContext& ctx,
226 const std::vector<Trk::ITrackLink*>& trackVector)
const
234 std::vector<Trk::ITrackLink*> origTracks = trackVector;
235 std::vector<Trk::ITrackLink*> seedTracks = trackVector;
238 std::vector<xAODVertex_pair> myxAODVertices;
243 theVertexContainer->setStore(theVertexAuxContainer);
251 <<
"), skipping vertexing and returning only dummy...");
258 beamSpot->beamVtx().covariancePosition());
259 dummyxAODVertex->
vxTrackAtVertex() = std::vector<Trk::VxTrackAtVertex>();
261 return std::make_pair(theVertexContainer, theVertexAuxContainer);
265 unsigned int seedtracknumber = seedTracks.size();
276 seedBegin = seedTracks.begin();
277 seedEnd = seedTracks.end();
279 if (seedtracknumber == 0) {
280 ATH_MSG_DEBUG(
" New iteration. No tracks available after track selection "
281 "for seeding. No finding done.");
289 std::vector<const Trk::TrackParameters*> perigeeList;
291 seedtrkAtVtxIter != seedEnd;
292 ++seedtrkAtVtxIter) {
293 perigeeList.push_back((*seedtrkAtVtxIter)->parameters());
302 beamSpot->beamVtx().covariancePosition());
304 beamSpot->beamVtx().fitQuality().chiSquared(),
305 beamSpot->beamVtx().fitQuality().doubleNumberDoF());
306 actualVertex =
m_SeedFinder->findSeed(perigeeList, &theconstraint);
310 looseConstraintCovariance.setIdentity();
311 looseConstraintCovariance = looseConstraintCovariance * 1
e+8;
320 <<
" at y: " << actualVertex.y()
321 <<
" at z: " << actualVertex.z() <<
endmsg;
324 if (actualVertex.z() == 0.) {
329 std::vector<const Trk::TrackParameters*> perigeesToFit;
330 std::vector<const Trk::TrackParameters*> perigeesToFitSplitVertex;
332 int numberOfTracks(perigeeList.size());
334 std::vector<const Trk::TrackParameters*>::const_iterator perigeeListBegin =
336 std::vector<const Trk::TrackParameters*>::const_iterator perigeeListEnd =
341 for (std::vector<const Trk::TrackParameters*>::const_iterator
342 perigeeListIter = perigeeListBegin;
343 perigeeListIter != perigeeListEnd;
345 if (numberOfTracks <= 2) {
346 perigeesToFit.push_back(*perigeeListIter);
349 perigeesToFit.push_back(*perigeeListIter);
355 perigeesToFit.push_back(*perigeeListIter);
358 perigeesToFitSplitVertex.push_back(*perigeeListIter);
364 std::unique_ptr<Trk::PlaneSurface> mySurface =
366 *perigeeListIter, &actualVertex,
distance);
368 msg(MSG::WARNING) <<
" ImpactPoint3dEstimator failed to find minimum "
369 "distance between track and vertex seed: "
375 <<
" Distance between track and seed vtx is negative: " <<
distance
384 if (myPerigee && myPerigee->covariance()) {
397 perigeesToFit.push_back(*perigeeListIter);
400 perigeesToFitSplitVertex.push_back(*perigeeListIter);
407 if (perigeesToFit.empty()) {
408 ATH_MSG_DEBUG(
" No good seed found. Exiting search for vertices...");
433 double ndfSplitVertex = 0.;
434 int ntracksSplitVertex = 0;
437 bool goodVertex = myxAODVertex !=
nullptr &&
442 msg(
MSG::DEBUG) <<
" xAOD::Vertex is pointer: " << myxAODVertex
443 <<
" ndf = " <<
ndf <<
" ntracks (with weight>0.01) "
444 << ntracks <<
" beam constraint "
455 int numberOfAddedTracks = 0;
466 std::vector<Trk::VxTrackAtVertex>* myVxTracksAtVtx =
467 (&(*vxIter)->vxTrackAtVertex());
469 if (!myVxTracksAtVtx)
473 myVxTracksAtVtx->begin();
475 myVxTracksAtVtx->end();
479 tracksIter != tracksEnd;) {
482 if ((*tracksIter).weight() > 0.01) {
488 (*tracksIter).initialPerigee();
490 if (trackPerigee ==
nullptr) {
492 return std::make_pair(theVertexContainer, theVertexAuxContainer);
495 double chi2_newvtx =
compatibility(*trackPerigee, *myxAODVertex);
496 double chi2_oldvtx =
compatibility(*trackPerigee, *(*vxIter));
500 if (chi2_newvtx < chi2_oldvtx) {
503 <<
") more compatible to new one (chi2= "
504 << chi2_newvtx <<
")");
506 perigeesToFit.push_back(trackPerigee);
509 bool isFound =
false;
519 if ((*origIter)->parameters() == trackPerigee) {
521 seedTracks.push_back(*origIter);
530 numberOfAddedTracks += 1;
538 tracksIter = myVxTracksAtVtx->erase(tracksIter);
539 tracksBegin = myVxTracksAtVtx->begin();
540 tracksEnd = myVxTracksAtVtx->end();
550 if (numberOfAddedTracks > 0) {
552 myxAODVertex =
nullptr;
564 goodVertex = myxAODVertex !=
nullptr &&
569 << myxAODVertex <<
" ndf = " <<
ndf
570 <<
" ntracks (with weight>0.01) " << ntracks
571 <<
" beam constraint "
576 ATH_MSG_DEBUG(
" Adding tracks resulted in an invalid vertex. "
588 bool goodSplitVertex =
false;
591 goodSplitVertex = myxAODSplitVertex !=
nullptr && ndfSplitVertex > 0 &&
592 ntracksSplitVertex >= 2;
595 << myxAODSplitVertex <<
" ndf = " << ndfSplitVertex
596 <<
" ntracks (with weight>0.01) " << ntracksSplitVertex);
598 if (!goodSplitVertex) {
602 myxAODSplitVertex, perigeesToFitSplitVertex, seedTracks);
609 theVertexContainer->
push_back(myxAODVertex);
613 myxAODVertex =
nullptr;
620 theVertexContainer->
push_back(myxAODVertex);
624 myxAODVertex =
nullptr;
633 beamSpot->beamVtx().covariancePosition());
635 std::vector<Trk::VxTrackAtVertex>();
638 if (goodSplitVertex) {
641 theVertexContainer->
push_back(myxAODSplitVertex);
643 if (myxAODSplitVertex) {
644 delete myxAODSplitVertex;
645 myxAODSplitVertex =
nullptr;
654 beamSpot->beamVtx().covariancePosition());
656 std::vector<Trk::VxTrackAtVertex>();
662 }
while (seedTracks.size() > 1 && iterations <
m_maxVertices);
665 ATH_MSG_DEBUG(
"Reached maximum iterations, have "<<iterations<<
" vertices");
676 if (!theVertexContainer->
empty()) {
686 primaryVtx->covariancePosition());
688 std::vector<Trk::VxTrackAtVertex>();
696 else if (theVertexContainer->
empty()) {
703 beamSpot->beamVtx().covariancePosition());
704 dummyxAODVertex->
vxTrackAtVertex() = std::vector<Trk::VxTrackAtVertex>();
709 for (
unsigned int i = 0;
i < theVertexContainer->
size() - 1;
i++) {
712 " Vtx: " <<
i <<
" x= " << (*theVertexContainer)[
i]->position().
x()
713 <<
" y= " << (*theVertexContainer)[
i]->position().
y() <<
" z= "
714 << (*theVertexContainer)[
i]->position().
z() <<
" ntracks= "
715 << (*theVertexContainer)[
i]->vxTrackAtVertex().
size()
716 <<
" chi2= " << (*theVertexContainer)[
i]->
chiSquared()
717 <<
" ndf = " << (*theVertexContainer)[
i]->numberDoF());
733 std::vector<Trk::VxTrackAtVertex>* myVxTracksAtVtx =
734 &((*vxIter)->vxTrackAtVertex());
736 if (!myVxTracksAtVtx)
740 myVxTracksAtVtx->begin();
742 myVxTracksAtVtx->end();
747 tracksIter != tracksEnd;
752 origtrkiter != origtrkend;
754 if ((*origtrkiter)->parameters() == (*tracksIter).initialPerigee()) {
759 (*tracksIter).setOrigTrack(*origtrkiter);
770 (*vxIter)->addTrackAtVertex(*linkToXAODTP, (*tracksIter).weight());
773 origTracks.erase(origtrkiter);
774 origtrkbegin = origTracks.begin();
775 origtrkend = origTracks.end();
780 ATH_MSG_ERROR(
" Cannot find vector element to fix links (step 4)! ");
786 origtrkiter != origtrkend;
788 if ((*origtrkiter) != 0) {
794 return std::make_pair(theVertexContainer, theVertexAuxContainer);
800 return StatusCode::SUCCESS;
821 myLinearizedTrack->expectedCovarianceAtPCA().block<2, 2>(0, 0);
824 (myLinearizedTrack->positionJacobian() *
825 (
vertex.covariancePosition() *
826 myLinearizedTrack->positionJacobian().transpose()))
829 weightReduced += errorVertexReduced;
831 weightReduced = weightReduced.inverse().eval();
833 myLinearizedTrack->expectedParametersAtPCA().block<2, 1>(0, 0);
834 double returnValue = trackParameters2D.dot(weightReduced * trackParameters2D);
836 delete myLinearizedTrack;
837 myLinearizedTrack =
nullptr;
844 std::vector<const Trk::TrackParameters*>& perigeesToFit,
845 std::vector<Trk::ITrackLink*>& seedTracks)
const
853 perigeesToFit.begin();
859 perigeesToFitIter != perigeesToFitEnd;
860 ++perigeesToFitIter) {
867 if ((*seedIter)->parameters() == *perigeesToFitIter) {
869 seedTracks.erase(seedIter);
870 seedBegin = seedTracks.begin();
871 seedEnd = seedTracks.end();
877 " Cannot find vector element to delete when removing BAD vertex! ");
890 std::vector<Trk::VxTrackAtVertex> myVxTracksAtVtx =
894 myVxTracksAtVtx.begin();
896 myVxTracksAtVtx.end();
899 tracksIter != tracksEnd;
901 if ((*tracksIter).weight() > 0.01) {
911 std::vector<const Trk::TrackParameters*>& perigeesToFit,
912 std::vector<Trk::ITrackLink*>& seedTracks)
const
917 std::vector<Trk::VxTrackAtVertex>* tracksAtVertex =
920 std::vector<Trk::VxTrackAtVertex>::const_iterator tracksAtVertexBegin =
921 tracksAtVertex->begin();
922 std::vector<Trk::VxTrackAtVertex>::const_iterator tracksAtVertexEnd =
923 tracksAtVertex->end();
929 perigeesToFit.begin();
933 for (std::vector<Trk::VxTrackAtVertex>::const_iterator tracksAtVertexIter =
935 tracksAtVertexIter != tracksAtVertexEnd;
936 ++tracksAtVertexIter) {
943 if ((*seedIter)->parameters() == (*tracksAtVertexIter).initialPerigee()) {
945 if ((*tracksAtVertexIter).weight() > 0.01) {
946 seedTracks.erase(seedIter);
947 seedBegin = seedTracks.begin();
948 seedEnd = seedTracks.end();
955 ATH_MSG_ERROR(
" Cannot find vector element to delete (step 1)! ");
961 perigeesToFitIter != perigeesToFitEnd;
962 ++perigeesToFitIter) {
963 if (*perigeesToFitIter == (*tracksAtVertexIter).initialPerigee()) {
965 if ((*tracksAtVertexIter).weight() > 0.01) {
966 perigeesToFit.erase(perigeesToFitIter);
967 perigeesToFitBegin = perigeesToFit.begin();
968 perigeesToFitEnd = perigeesToFit.end();
975 ATH_MSG_ERROR(
" Cannot find vector element to delete (step 2)! ");
979 std::vector<Trk::VxTrackAtVertex>* myVxTracksAtVertex =
983 myVxTracksAtVertex->begin();
985 myVxTracksAtVertex->end();
989 perigeesToFitIter != perigeesToFitEnd;
990 ++perigeesToFitIter) {
998 if (myPerigee ==
nullptr) {
1010 seedIter != seedEnd;
1012 if ((*seedIter)->parameters() == *perigeesToFitIter) {
1014 seedTracks.erase(seedIter);
1015 seedBegin = seedTracks.begin();
1016 seedEnd = seedTracks.end();
1022 ATH_MSG_ERROR(
" Cannot find vector element to delete (step 3)! ");
1028 tracksIter != tracksEnd;
1030 if ((*tracksIter).initialPerigee() == *perigeesToFitIter) {
1034 myVxTracksAtVertex->erase(tracksIter);
1035 tracksBegin = myVxTracksAtVertex->begin();
1036 tracksEnd = myVxTracksAtVertex->end();