#include <AdaptiveChisquarePrimaryVertexFitter.h>

Inheritance diagram for AdaptiveChisquarePrimaryVertexFitter:

Classes
struct	TrackInfo

Public Types
using	Error3 = ROOT::Math::SMatrix< double, 3 >

Public Member Functions
	AdaptiveChisquarePrimaryVertexFitter (double chicutoff=2.5, double zcutoff=1.0, double mintrkweight=0.4, bool multivertexfit=false)

std::vector< TransientVertex >	fit (const std::vector< reco::TransientTrack > &, const std::vector< TransientVertex > &, const reco::BeamSpot &, const bool) override

	~AdaptiveChisquarePrimaryVertexFitter () override=default

Public Member Functions inherited from PrimaryVertexFitterBase
	PrimaryVertexFitterBase (const edm::ParameterSet &conf)

	PrimaryVertexFitterBase ()

virtual	~PrimaryVertexFitterBase ()=default

Protected Member Functions
bool	clean ()

void	fill_trackinfo (const std::vector< reco::TransientTrack > &, const reco::BeamSpot &)

void	fill_weights (const reco::BeamSpot &, const double beta=1.)

Error3	get_inverse_beam_covariance (const reco::BeamSpot &)

TransientVertex	get_TransientVertex (const unsigned int, const std::vector< std::pair< unsigned int, float >> &, const std::vector< reco::TransientTrack > &, const float, const reco::BeamSpot &)

void	make_vtx_trk_map (const double)

TransientVertex	refit (const TransientVertex &, const reco::BeamSpot &, const bool)

void	remove_vertex (unsigned int)

double	track_in_vertex_chsq (const TrackInfo &, const double, const double, const double)

double	update (const reco::BeamSpot &, float beam_weight, const bool fill_covariances=false)

void	verify ()

std::vector< TransientVertex >	vertices (const std::vector< reco::TransientTrack > &, const std::vector< TransientVertex > &, const reco::BeamSpot &, const bool)

Protected Attributes
double	chi_cutoff_

std::vector< Error3 >	covv_

double	min_trackweight_

double	multivertexfit_

std::vector< unsigned int >	tkfirstv_

std::vector< unsigned int >	tkmap_

std::vector< double >	tkweight_

std::vector< TrackInfo >	trackinfo_

std::vector< double >	xv_

std::vector< double >	yv_

double	z_cutoff_

std::vector< double >	zv_

Detailed Description

Description: simultaneous chisquared fit of primary vertices

Definition at line 15 of file AdaptiveChisquarePrimaryVertexFitter.h.

Member Typedef Documentation

◆ Error3

using AdaptiveChisquarePrimaryVertexFitter::Error3 = ROOT::Math::SMatrix<double, 3>

Definition at line 28 of file AdaptiveChisquarePrimaryVertexFitter.h.

Constructor & Destructor Documentation

◆ AdaptiveChisquarePrimaryVertexFitter()

AdaptiveChisquarePrimaryVertexFitter::AdaptiveChisquarePrimaryVertexFitter	(	double	chicutoff = `2.5`,
		double	zcutoff = `1.0`,
		double	mintrkweight = `0.4`,
		bool	multivertexfit = `false`
	)

Definition at line 8 of file AdaptiveChisquarePrimaryVertexFitter.cc.

References chi_cutoff_, min_trackweight_, multivertexfit_, and z_cutoff_.

     : chi_cutoff_(chicutoff), z_cutoff_(zcutoff), min_trackweight_(mintrkweight), multivertexfit_(multivertexfit) {
 #ifdef PVTX_DEBUG
   edm::LogInfo("AdaptiveChisquarePrimaryVertexFitter")
       << "instantiating AdaptiveChisquarePrimaryVertexFitter with "
       << " chi cutoff =" << chi_cutoff_ << " z cutoff =" << z_cutoff_ << " min_trackweight=" << min_trackweight_
       << " multivertex mode " << multivertexfit_ << std::endl;
 #endif
 }

◆ ~AdaptiveChisquarePrimaryVertexFitter()

AdaptiveChisquarePrimaryVertexFitter::~AdaptiveChisquarePrimaryVertexFitter ( )

overridedefault

Member Function Documentation

◆ clean()

bool AdaptiveChisquarePrimaryVertexFitter::clean ( )

protected

Definition at line 177 of file AdaptiveChisquarePrimaryVertexFitter.cc.

References funct::abs(), PVValHelper::dz, dqmiolumiharvest::j, dqmdumpme::k, SiStripPI::min, remove_vertex(), tkfirstv_, tkweight_, xv_, and zv_.

Referenced by vertices().

                                                  {
   /* in multi-vertex fitting, nearby vertices can fall on top of each other, 
      even when the initial seeds don't, some kind of duplicate removal is required
      the approach in this method is similar to the method applied in clustering:
      at least two tracks with a weight above a threshold (trkweight_threshold) are required.
      vertices that don't fulfill this are either insignficant or very close
      to another vertex
    */
   const double trkweight_threshold = 0.7;
   unsigned int nv = xv_.size();
   if (nv < 2)
     return false;
 
   // sum of weights per vertex
   std::vector<double> wsumhi(nv, 0);
   for (unsigned int k = 0; k < nv; k++) {
     for (unsigned int j = tkfirstv_[k]; j < tkfirstv_[k + 1]; j++) {
       if (tkweight_[j] > trkweight_threshold)
         wsumhi[k] += tkweight_[j];
     }
   }
 
   double dzmin = 0;
   unsigned int k_dzmin = 0;
   for (unsigned int k = 0; k < nv - 1; k++) {
     double dz = std::abs(zv_[k + 1] - zv_[k]);
     if ((k == 0) || (dz < dzmin)) {
       dzmin = dz;
       k_dzmin = k;
     }
   }
 
   if ((std::abs(dzmin) < 0.0200) && (std::min(wsumhi[k_dzmin], wsumhi[k_dzmin + 1]) < 0.5)) {
     if (wsumhi[k_dzmin] < wsumhi[k_dzmin + 1]) {
       remove_vertex(k_dzmin);
     } else {
       remove_vertex(k_dzmin + 1);
     }
   }
 
   return true;
 }

◆ fill_trackinfo()

void AdaptiveChisquarePrimaryVertexFitter::fill_trackinfo	(	const std::vector< reco::TransientTrack > &	tracks,
		const reco::BeamSpot &	beamSpot
	)

protected

Definition at line 34 of file AdaptiveChisquarePrimaryVertexFitter.cc.

Referenced by refit(), and vertices().

                                                                                         {
   /* fill track information used during fits into arrays, parallell to the list of input tracks */
 
   trackinfo_.clear();
   trackinfo_.reserve(tracks.size());
 
   for (auto &trk : tracks) {
     TrackInfo ti;
     // F1,F2 are the perigee parameters (3,4)
     const auto tspca = trk.stateAtBeamLine().trackStateAtPCA();  // freeTrajectoryState
     const auto tspca_pe = PerigeeConversions::ftsToPerigeeError(tspca);
     const auto momentum = tspca.momentum();
     auto const cos_phi = momentum.x() / momentum.perp();
     auto const sin_phi = momentum.y() / momentum.perp();
     auto const tan_lambda = momentum.z() / momentum.perp();
 
     // covariance matrix of (F1,F2)
     double cov11 = tspca_pe.covarianceMatrix()(3, 3);
     double cov22 = tspca_pe.covarianceMatrix()(4, 4);
     double cov12 = tspca_pe.covarianceMatrix()(3, 4);
 
     // S = cov^{-1}
     double DetV = cov11 * cov22 - cov12 * cov12;
     if (fabs(DetV) < 1.e-16) {
       edm::LogWarning("AdaptiveChisquarePrimaryVertexFitter")
           << "Warning, det(V) almost vanishes : " << DetV << " !! This should not happen!" << std::endl;
       ti.S11 = 0;
       ti.S22 = 0;
       ti.S12 = 0;
     } else {
       ti.S11 = cov22 / DetV;
       ti.S22 = cov11 / DetV;
       ti.S12 = -cov12 / DetV;
     }
     ti.b1 = tspca.position().x() * sin_phi - tspca.position().y() * cos_phi;
     ti.H1[0] = -sin_phi;
     ti.H1[1] = cos_phi;
     ti.H1[2] = 0;
     ti.b2 = tspca.position().z() - (tspca.position().x() * cos_phi + tspca.position().y() * sin_phi) * tan_lambda;
     ti.H2[0] = cos_phi * tan_lambda;
     ti.H2[1] = sin_phi * tan_lambda;
     ti.H2[2] = -1.;
 
     for (int k = 0; k < 3; k++) {
       double SH1k = (ti.S11 * ti.H1[k] + ti.S12 * ti.H2[k]);
       double SH2k = (ti.S12 * ti.H1[k] + ti.S22 * ti.H2[k]);
       ti.g[k] = ti.b1 * SH1k + ti.b2 * SH2k;
       for (int l = 0; l < 3; l++) {
         ti.C(l, k) = ti.H1[l] * SH1k + ti.H2[l] * SH2k;
       }
     }
 
     ti.zpca = tspca.position().z();
     ti.dzError = trk.track().dzError();
     trackinfo_.push_back(ti);
   }
 }

◆ fill_weights()

void AdaptiveChisquarePrimaryVertexFitter::fill_weights	(	const reco::BeamSpot &	beamspot,
		const double	beta = `1.`
	)

protected

Definition at line 141 of file AdaptiveChisquarePrimaryVertexFitter.cc.

References cms::cuda::assert(), HLT_2024v12_cff::beta, chi_cutoff_, MillePedeFileConverter_cfg::e, mps_fire::i, dqmiolumiharvest::j, dqmdumpme::k, nt, tkfirstv_, tkmap_, tkweight_, track_in_vertex_chsq(), trackinfo_, xv_, yv_, and zv_.

Referenced by refit(), and vertices().

                                                                                                  {
   // multi-vertex version
   unsigned const int nt = trackinfo_.size();
   unsigned const int nv = xv_.size();
   const double beta_over_2 = 0.5 * beta;
   const double argmax = beta_over_2 * chi_cutoff_ * chi_cutoff_ * 5;
   const double Z_cutoff = vdt::fast_exp(-beta_over_2 * chi_cutoff_ * chi_cutoff_);
 
   std::vector<double> Z_track(nt, Z_cutoff);
 
   // evaluate and cache track-vertex assignment chi**2 for all clusters and sum up Z
   for (unsigned int k = 0; k < nv; k++) {
     for (unsigned int j = tkfirstv_[k]; j < tkfirstv_[k + 1]; j++) {
       const unsigned int i = tkmap_[j];
       double arg = beta_over_2 * track_in_vertex_chsq(trackinfo_[i], xv_[k], yv_[k], zv_[k]);
       if (arg < argmax) {
         const double e = vdt::fast_exp(-arg);
         tkweight_[j] = e;  // must later be normalized by the proper Z_track[i]
         Z_track[i] += e;   // sum up exponentials to normalize
       } else {
         tkweight_[j] = 0.;
       }
     }
   }
 
   // now we have the partition function, Z_i and can evaluate assignment probabilities (aka weights)
   for (unsigned int j = 0; j < tkmap_.size(); j++) {
     const unsigned int i = tkmap_[j];
 #ifdef PVT_DEBUG
     assert((i < nt) && "tkmap out of range");
     assert((tkmap_.size() == tkweight_.size()) && "map and list not aliged");
 #endif
     tkweight_[j] /= Z_track[i];
   }
 }

◆ fit()

std::vector< TransientVertex > AdaptiveChisquarePrimaryVertexFitter::fit	(	const std::vector< reco::TransientTrack > &	tracks,
		const std::vector< TransientVertex > &	clusters,
		const reco::BeamSpot &	beamspot,
		const bool	useBeamConstraint
	)

overridevirtual

Implements PrimaryVertexFitterBase.

Definition at line 526 of file AdaptiveChisquarePrimaryVertexFitter.cc.

References bsc_activity_cfg::clusters, multivertexfit_, FSQDQM_cfi::pvs, refit(), DiMuonV_cfg::tracks, HLT_2024v12_cff::useBeamConstraint, and vertices().

Referenced by trackingPlots.Iteration::modules().

                                                                                                      {
   if (multivertexfit_) {
     return vertices(tracks, clusters, beamspot, useBeamConstraint);
 
   } else {
     // fit the clusters one-by-one using the tracklist of the clusters (ignores the "tracks" argument)
     std::vector<TransientVertex> pvs;
     pvs.reserve(clusters.size());
     for (auto &cluster : clusters) {
       if (cluster.originalTracks().size() > (useBeamConstraint ? 0 : 1)) {
         auto pv = refit(cluster, beamspot, useBeamConstraint);
         if (pv.isValid()) {
           pvs.emplace_back(pv);
         }
       }
     }
     return pvs;
   }
 }

◆ get_inverse_beam_covariance()

AdaptiveChisquarePrimaryVertexFitter::Error3 AdaptiveChisquarePrimaryVertexFitter::get_inverse_beam_covariance ( const reco::BeamSpot & beamspot )

protected

Definition at line 327 of file AdaptiveChisquarePrimaryVertexFitter.cc.

References funct::pow().

Referenced by get_TransientVertex(), and update().

                                   {
   auto SBeam = beamspot.rotatedCovariance3D();
   if (SBeam.Invert()) {
     return SBeam;
   } else {
     edm::LogWarning("AdaptiveChisquarePrimaryVertexFitter")
         << "Warning, beam-spot covariance matrix inversion failed " << std::endl;
     Error3 S0;
     S0(0, 0) = 1. / pow(beamspot.BeamWidthX(), 2);
     S0(1, 1) = 1. / pow(beamspot.BeamWidthY(), 2);
     S0(2, 2) = 1. / pow(beamspot.sigmaZ(), 2);
     return S0;
   }
 }

◆ get_TransientVertex()

TransientVertex AdaptiveChisquarePrimaryVertexFitter::get_TransientVertex	(	const unsigned int	k,
		const std::vector< std::pair< unsigned int, float >> &	vertex_track_weights,
		const std::vector< reco::TransientTrack > &	tracks,
		const float	beam_weight,
		const reco::BeamSpot &	beamspot
	)

protected

Definition at line 343 of file AdaptiveChisquarePrimaryVertexFitter.cc.

References nano_mu_local_reco_cff::chi2, covv_, PVValHelper::dx, PVValHelper::dy, PVValHelper::dz, get_inverse_beam_covariance(), mps_fire::i, dqmdumpme::k, min_trackweight_, ntuplemaker::posError, track_in_vertex_chsq(), trackinfo_, DiMuonV_cfg::tracks, L1BJetProducer_cff::vtx, xv_, yv_, and zv_.

Referenced by refit(), and vertices().

                                   {
   const GlobalPoint pos(xv_[k], yv_[k], zv_[k]);
   const GlobalError posError(
       covv_[k](0, 0), covv_[k](1, 0), covv_[k](1, 1), covv_[k](2, 0), covv_[k](2, 1), covv_[k](2, 2));
   float chi2 = 0.;
   float vtx_ndof = -3.;
   if (beam_weight > 0) {
     // add beam-spot chi**2 and degrees of freedom
     vtx_ndof = 3 * beam_weight;
     const auto S = get_inverse_beam_covariance(beamspot);
     const double dx = xv_[k] - beamspot.x0();
     const double dy = yv_[k] - beamspot.y0();
     const double dz = zv_[k] - beamspot.z0();
     chi2 = beam_weight * (S(0, 0) * dx * dx + S(1, 1) * dy * dy + 2 * S(0, 1) * dx * dy + S(2, 2) * dz * dz +
                           2 * S(0, 2) * dx * dz + 2 * S(1, 2) * dy * dz);
   }
 
   std::vector<reco::TransientTrack> vertex_tracks;
   TransientVertex::TransientTrackToFloatMap trkWeightMap;
   for (const auto &tk : vertex_track_weights) {
     const unsigned int i = tk.first;
     const float track_weight = tk.second;
     if (track_weight >= min_trackweight_) {
       vertex_tracks.emplace_back(tracks[i]);
       trkWeightMap[tracks[i]] = track_weight;
       vtx_ndof += 2 * track_weight;
       chi2 += track_weight * track_in_vertex_chsq(trackinfo_[i], xv_[k], yv_[k], zv_[k]);
     }
   }
 
   auto vtx = TransientVertex(pos, posError, vertex_tracks, chi2, vtx_ndof);
   vtx.weightMap(trkWeightMap);
 
   return vtx;
 }

◆ make_vtx_trk_map()

void AdaptiveChisquarePrimaryVertexFitter::make_vtx_trk_map ( const double zrange_scale )

protected

Definition at line 93 of file AdaptiveChisquarePrimaryVertexFitter.cc.

References funct::abs(), mps_fire::i, dqmdumpme::k, nt, tkfirstv_, tkmap_, tkweight_, trackinfo_, xv_, yv_, z_cutoff_, OfflinePixel3DPrimaryVertices_cfi::zrange, and zv_.

Referenced by vertices().

                                                                                {
   unsigned const int nv = xv_.size();
   unsigned const int nt = trackinfo_.size();
 
 #ifdef PVTX_DEBUG
   if (nv < 1) {
     edm::LogWarning("AdaptiveChisquarePrimaryVertexFit") << " empty vertex list with " << nt << " tracks" << std::endl;
     return;
   }
 #endif
 
   // parallel lists for track to vertex mapping, tracks are not sorted
   tkmap_.clear();     // index in trackinfo_
   tkweight_.clear();  // weight in vertex
   tkfirstv_.clear();  // each vertex k owns a section of those list : tkfirstv_[k] .. tkfirstv_[k+1]-1
 
   if (nv == 1) {
     // always accept all tracks for a single vertex fit
     tkfirstv_.push_back(0);
     tkfirstv_.push_back(nt);
     tkweight_.assign(nt, 0.);
     tkmap_.reserve(nt);
     for (unsigned int i = 0; i < nt; i++) {
       tkmap_.emplace_back(i);
     }
     return;
   }
 
   // n > 1
   tkmap_.reserve(nv * 100);
   tkweight_.reserve(nv * 100);
   for (unsigned int k = 0; k < nv; k++) {
     tkfirstv_.emplace_back(tkmap_.size());
     for (unsigned int i = 0; i < nt; i++) {
       auto &ti = trackinfo_[i];
       const double zrange = zrange_scale * ti.dzError;
       if (std::abs(zv_[k] - ti.zpca) < z_cutoff_) {
         const double dztrk = ti.b2 + xv_[k] * ti.H2[0] + yv_[k] * ti.H2[1] - zv_[k];
         if (std::abs(dztrk) < zrange) {
           tkmap_.emplace_back(i);
           tkweight_.emplace_back(0.);
         }
       }
     }
   }
   tkfirstv_.emplace_back(tkmap_.size());  // extra entry, simplifies loops, every vertex has a "successor" now
 }

◆ refit()

TransientVertex AdaptiveChisquarePrimaryVertexFitter::refit	(	const TransientVertex &	cluster,
		const reco::BeamSpot &	beamspot,
		const bool	useBeamConstraint
	)

protected

Definition at line 474 of file AdaptiveChisquarePrimaryVertexFitter.cc.

References HLTObjectsMonitor_cfi::beamspot, covv_, dumpMFGeometry_cfg::delta, fill_trackinfo(), fill_weights(), get_TransientVertex(), mps_fire::i, nt, TransientVertex::originalTracks(), TransientVertex::position(), tkfirstv_, tkmap_, tkweight_, update(), HLT_2024v12_cff::useBeamConstraint, xv_, yv_, PV3DBase< T, PVType, FrameType >::z(), and zv_.

Referenced by fit().

                                                                                           {
   // fit a single vertex using all tracks in the tracklist
   const unsigned int nt = cluster.originalTracks().size();
   const int max_iterations = 50;
 
   // initialize, vectors with size=1 here to avoid code duplication from the multivertex case in update()
   const double zclu = cluster.position().z();
   xv_ = {beamspot.x(zclu)};
   yv_ = {beamspot.y(zclu)};
   zv_ = {zclu};
   tkfirstv_ = {0, nt};
   covv_.clear();
 
   fill_trackinfo(cluster.originalTracks(), beamspot);
   tkweight_.assign(nt, 0.);
   tkmap_.clear();
   tkmap_.reserve(nt);
   for (unsigned int i = 0; i < nt; i++) {
     tkmap_.emplace_back(i);  // trivial map for single vertex fits
   }
 
   float beam_weight = useBeamConstraint ? 1. : 0.;
 
   double delta = 0;
   unsigned int nit = 0;
   while ((nit == 0) || ((delta > 0.0001) && (nit < max_iterations))) {
     fill_weights(beamspot);
     delta = update(beamspot, beam_weight, false);
     nit++;
   }
 
   if ((nit >= max_iterations) && (delta > 0.1)) {
     edm::LogWarning("AdaptiveChisquarePrimaryVertexFitter")
         << "single vertex fit, iteration limit reached " << nit << "  last delta = " << delta << std::endl
         << "    nt = " << cluster.originalTracks().size() << std::endl;
   }
 
   // fill the covariance matrices
   update(beamspot, beam_weight, true);
 
   // put the result into a transient vertex
   std::vector<std::pair<unsigned int, float>> vertex_track_weights;
   for (unsigned int i = 0; i < nt; i++) {
     vertex_track_weights.emplace_back(i, tkweight_[i]);
   }
 
   return get_TransientVertex(0, vertex_track_weights, cluster.originalTracks(), beam_weight, beamspot);
 }

◆ remove_vertex()

void AdaptiveChisquarePrimaryVertexFitter::remove_vertex ( unsigned int k )

protected

Definition at line 220 of file AdaptiveChisquarePrimaryVertexFitter.cc.

References covv_, dqmdumpme::k, tkfirstv_, tkmap_, tkweight_, xv_, yv_, and zv_.

Referenced by clean().

                                                                        {
   // remove a vertex or rather merge it with it's neighbour
   // used for multi-vertex fits only
   unsigned int nv = xv_.size();
   if (nv < 2)
     return;
 
   // 1) remove the vertex from the vertex list
   xv_.erase(xv_.begin() + k);
   yv_.erase(yv_.begin() + k);
   zv_.erase(zv_.begin() + k);
   covv_.erase(covv_.begin() + k);
 
   // 2) adjust the track-map map
   // 2a) remove the map entries that belong the deleted vertex
   const unsigned int num_erased_map_entries = tkfirstv_[k + 1] - tkfirstv_[k];
   tkmap_.erase(tkmap_.begin() + tkfirstv_[k], tkmap_.begin() + tkfirstv_[k + 1]);
   tkweight_.erase(tkweight_.begin() + tkfirstv_[k], tkweight_.begin() + tkfirstv_[k + 1]);
   // 2b) adjust pointers for the following vertices, including the dummy entry behind the last (now [nv-1])
   for (unsigned int k1 = k + 1; k1 < nv + 1; k1++) {
     tkfirstv_[k1] -= num_erased_map_entries;
   }
   // 2c) erase the pointer of the removed vertex
   tkfirstv_.erase(tkfirstv_.begin() + k);
 }

◆ track_in_vertex_chsq()

double AdaptiveChisquarePrimaryVertexFitter::track_in_vertex_chsq	(	const TrackInfo &	ti,
		const double	xvtx,
		const double	yvtx,
		const double	zvtx
	)

protected

Definition at line 21 of file AdaptiveChisquarePrimaryVertexFitter.cc.

References cms::cuda::assert(), AdaptiveChisquarePrimaryVertexFitter::TrackInfo::b1, AdaptiveChisquarePrimaryVertexFitter::TrackInfo::b2, AdaptiveChisquarePrimaryVertexFitter::TrackInfo::H1, AdaptiveChisquarePrimaryVertexFitter::TrackInfo::H2, AdaptiveChisquarePrimaryVertexFitter::TrackInfo::S11, AdaptiveChisquarePrimaryVertexFitter::TrackInfo::S12, and AdaptiveChisquarePrimaryVertexFitter::TrackInfo::S22.

Referenced by fill_weights(), and get_TransientVertex().

                                                                                      {
   double F1 = ti.b1 + xvtx * ti.H1[0] + yvtx * ti.H1[1];         // using H1[2]=0
   double F2 = ti.b2 + xvtx * ti.H2[0] + yvtx * ti.H2[1] - zvtx;  // using H2[2]=-1
   double chsq = F1 * F1 * ti.S11 + F2 * F2 * ti.S22 + 2. * F1 * F2 * ti.S12;
 #ifdef PVTX_DEBUG
   assert((chsq >= 0) && " negative chi**2");
 #endif
   return chsq;
 }

◆ update()

double AdaptiveChisquarePrimaryVertexFitter::update	(	const reco::BeamSpot &	beamspot,
		float	beam_weight,
		const bool	fill_covariances = `false`
	)

protected

Definition at line 246 of file AdaptiveChisquarePrimaryVertexFitter.cc.

References funct::abs(), HltBtagPostValidation_cff::c, covv_, MillePedeFileConverter_cfg::e, get_inverse_beam_covariance(), mps_fire::i, dqmiolumiharvest::j, dqmdumpme::k, MainPageGenerator::l, SiStripPI::max, nt, tkfirstv_, tkmap_, tkweight_, trackinfo_, w(), xv_, yv_, and zv_.

Referenced by progressbar.ProgressBar::__next__(), MatrixUtil.Matrix::__setitem__(), MatrixUtil.Steps::__setitem__(), progressbar.ProgressBar::finish(), MatrixUtil.Steps::overwrite(), refit(), and vertices().

                                                                                  {
   double rho_vtx = 0;
   double delta_z = 0;
   double delta_x = 0;
   double delta_y = 0;
   unsigned const int nt = trackinfo_.size();
   unsigned const int nv = xv_.size();
   if (fill_covariances) {
     covv_.clear();
   }
 
   // initial value for S, 0 or inverse of the beamspot covariance matrix
   Error3 S0;
   double c_beam[3] = {0, 0, 0};
   if (beam_weight > 0) {
     S0 = get_inverse_beam_covariance(beamspot);
     for (unsigned int j = 0; j < 3; j++) {
       c_beam[j] = -(S0(j, 0) * beamspot.x0() + S0(j, 1) * beamspot.y0() + S0(j, 2) * beamspot.z0());
     }
   }
 
   for (unsigned int k = 0; k < nv; k++) {
     rho_vtx = 0;
     Error3 S(S0);
     // sum track contributions
     double c[3] = {c_beam[0], c_beam[1], c_beam[2]};
     for (unsigned int j = tkfirstv_[k]; j < tkfirstv_[k + 1]; j++) {
       const unsigned int i = tkmap_[j];
       const auto w = tkweight_[j];
       rho_vtx += w;
       S += w * trackinfo_[i].C;
       for (unsigned int l = 0; l < 3; l++) {
         c[l] += w * trackinfo_[i].g[l];
       }
     }
 
 #ifdef PVTX_DEBUG
     if ((fabs(S(1, 2) - S(2, 1)) > 1e-3) || (fabs(S(0, 2) - S(2, 0)) > 1e-3) || (fabs(S(0, 1) - S(1, 0)) > 1e-3) ||
         (S(0, 0) <= 0) || (S(0, 0) <= 0) || (S(0, 0) <= 0)) {
       edm::LogWarning("AdaptiveChisquarePrimaryVertexFitter") << "update()  bad S-matrix   S=" << std::endl
                                                               << S << std::endl;
       edm::LogWarning("AdaptiveChisquarePrimaryVertexFitter")
           << "n-vertex = " << nv << "  n-track = " << nt << std::endl;
     }
 #endif
 
     const auto xold = xv_[k];
     const auto yold = yv_[k];
     const auto zold = zv_[k];
 
     if (S.Invert()) {
       xv_[k] = -(S(0, 0) * c[0] + S(0, 1) * c[1] + S(0, 2) * c[2]);
       yv_[k] = -(S(1, 0) * c[0] + S(1, 1) * c[1] + S(1, 2) * c[2]);
       zv_[k] = -(S(2, 0) * c[0] + S(2, 1) * c[1] + S(2, 2) * c[2]);
       if (fill_covariances) {
         covv_.emplace_back(S);
       }
     } else {
       edm::LogWarning("AdaptiveChisquarePrimaryVertexFitter") << "update()   Matrix inversion failed" << S << std::endl;
       if (fill_covariances) {
         Error3 covv_dummy;
         covv_dummy(0, 0) = 100.;
         covv_dummy(1, 1) = 100.;
         covv_dummy(2, 2) = 100.;
         covv_.emplace_back(covv_dummy);
       }
     }
 
     if ((nt > 1) && (rho_vtx > 1.0)) {
       delta_x = std::max(delta_x, std::abs(xv_[k] - xold));
       delta_y = std::max(delta_y, std::abs(yv_[k] - yold));
       delta_z = std::max(delta_z, std::abs(zv_[k] - zold));
     }
 
   }  // vertex loop
 
   return std::max(delta_z, std::max(delta_x, delta_y));
 }

◆ verify()

void AdaptiveChisquarePrimaryVertexFitter::verify ( )

inlineprotected

Definition at line 31 of file AdaptiveChisquarePrimaryVertexFitter.h.

References cms::cuda::assert(), mps_fire::i, dqmiolumiharvest::j, dqmdumpme::k, nt, tkfirstv_, tkmap_, tkweight_, trackinfo_, xv_, yv_, and zv_.

                 {  // DEBUG only
     unsigned int nt = trackinfo_.size();
     unsigned int nv = xv_.size();
     assert((yv_.size() == nv) && "yv size");
     assert((zv_.size() == nv) && "zv size");
     assert((tkfirstv_.size() == (nv + 1)) && "tkfirstv size");
     assert((tkmap_.size() == tkweight_.size()) && "tkmapsize <> tkweightssize");
     for (unsigned int k = 0; k < nv; k++) {
       assert((tkfirstv_[k] < tkweight_.size()) && "tkfirst[k]");
       assert((tkfirstv_[k + 1] <= tkweight_.size()) && "tkfirst[k+1]");
       assert((tkfirstv_[k] <= tkfirstv_[k + 1]) && "tkfirst[k/k+1]");
       for (unsigned int j = tkfirstv_[k]; j < tkfirstv_[k + 1]; k++) {
         assert((j < tkmap_.size()) && "illegal tkfirst entry");
         unsigned int i = tkmap_[j];
         assert((i < nt) && "illegal tkmap entry");
         assert((tkweight_[i] >= 0) && "negative tkweight or nan");
         assert((tkweight_[i] <= 1) && "tkweight > 1 or nan");
       }
     }
   };

◆ vertices()

std::vector< TransientVertex > AdaptiveChisquarePrimaryVertexFitter::vertices	(	const std::vector< reco::TransientTrack > &	tracks,
		const std::vector< TransientVertex > &	clusters,
		const reco::BeamSpot &	beamspot,
		const bool	useBeamConstraint
	)

protected

Definition at line 384 of file AdaptiveChisquarePrimaryVertexFitter.cc.

References clean(), bsc_activity_cfg::clusters, covv_, dumpMFGeometry_cfg::delta, fill_trackinfo(), fill_weights(), get_TransientVertex(), mps_fire::i, dqmiolumiharvest::j, dqmdumpme::k, make_vtx_trk_map(), FSQDQM_cfi::pvs, tkfirstv_, tkmap_, tkweight_, trackinfo_, DiMuonV_cfg::tracks, update(), HLT_2024v12_cff::useBeamConstraint, xv_, yv_, and zv_.

Referenced by fit().

                                   {
   // simultaneous fit of all vertices in the input list
 
   const int max_iterations = 50;
 
   // initialize the vertices
   const unsigned int nv = clusters.size();
   xv_.clear();
   xv_.reserve(nv);
   yv_.clear();
   yv_.reserve(nv);
   zv_.clear();
   zv_.reserve(nv);
   tkfirstv_.clear();
   tkfirstv_.reserve(nv + 1);
   covv_.clear();
   covv_.reserve(nv);
 
   // seeds
   for (auto &clu : clusters) {
     const double zclu = clu.position().z();
     xv_.emplace_back(beamspot.x(zclu));
     yv_.emplace_back(beamspot.y(zclu));
     zv_.emplace_back(zclu);
   }
 
   fill_trackinfo(tracks, beamspot);
 
   make_vtx_trk_map(5.);  // use tracks within 5 sigma windows (if that is less than z_cutoff_)
 
   float beam_weight = useBeamConstraint ? 1. : 0.;
 
   double delta = 0;
   unsigned int nit = 0;
   while ((nit == 0) || ((delta > 0.0001) && (nit < max_iterations))) {
     fill_weights(beamspot);
     delta = update(beamspot, beam_weight, false);
     nit++;
   }
   if ((nit >= max_iterations) && (delta > 0.01)) {
     edm::LogWarning("AdaptiveChisquarePrimaryVertexFitter")
         << "iteration limit reached " << nit << "  last delta = " << delta << std::endl
         << " nv = " << nv << "    nt = " << tracks.size() << std::endl;
   }
 
   // may need to remove collapsed vertices
   nit = 0;
   while ((xv_.size() > 1) && (nit < max_iterations) && (clean())) {
     fill_weights(beamspot);
     update(beamspot, beam_weight, false);
     nit++;
   }
 
   // fill the covariance matrices
   update(beamspot, beam_weight, true);
 
   // assign tracks to vertices
   std::vector<unsigned int> track_to_vertex(trackinfo_.size(), nv);
   // for each track identify the vertex that wants it most
   std::vector<double> maxweight(trackinfo_.size(), -1.);
   for (unsigned int k = 0; k < nv; k++) {
     for (unsigned int j = tkfirstv_[k]; j < tkfirstv_[k + 1]; j++) {
       const unsigned int i = tkmap_[j];
       if (tkweight_[j] > maxweight[i]) {
         maxweight[i] = tkweight_[j];
         track_to_vertex[i] = k;
       }
     }
   }
 
   // fill the fit result into transient vertices
   std::vector<TransientVertex> pvs;
   for (unsigned int k = 0; k < xv_.size(); k++) {
     std::vector<std::pair<unsigned int, float>> vertex_tracks_weights;
     for (unsigned int j = tkfirstv_[k]; j < tkfirstv_[k + 1]; j++) {
       unsigned int i = tkmap_[j];
       if (track_to_vertex[i] == k) {
         vertex_tracks_weights.emplace_back(tkmap_[j], tkweight_[j]);
       }
     }
     pvs.emplace_back(get_TransientVertex(k, vertex_tracks_weights, tracks, beam_weight, beamspot));
   }
 
   return pvs;
 }