WeightedMeanPrimaryVertexEstimator Class Reference

#include <WeightedMeanFitter.h>

Inheritance diagram for WeightedMeanPrimaryVertexEstimator:

Public Member Functions
std::vector< TransientVertex >	fit (const std::vector< reco::TransientTrack > &dummy, const std::vector< TransientVertex > &clusters, const reco::BeamSpot &beamSpot, const bool useBeamConstraint) override
	WeightedMeanPrimaryVertexEstimator ()=default
	~WeightedMeanPrimaryVertexEstimator () override=default
Public Member Functions inherited from PrimaryVertexFitterBase
	PrimaryVertexFitterBase (const edm::ParameterSet &conf)
	PrimaryVertexFitterBase ()
virtual	~PrimaryVertexFitterBase ()=default

Detailed Description

Definition at line 464 of file WeightedMeanFitter.h.

Constructor & Destructor Documentation

WeightedMeanPrimaryVertexEstimator()

WeightedMeanPrimaryVertexEstimator::WeightedMeanPrimaryVertexEstimator ( )

default

~WeightedMeanPrimaryVertexEstimator()

WeightedMeanPrimaryVertexEstimator::~WeightedMeanPrimaryVertexEstimator ( )

overridedefault

Member Function Documentation

fit()

std::vector<TransientVertex> WeightedMeanPrimaryVertexEstimator::fit ( const std::vector< reco::TransientTrack > &  dummy, const std::vector< TransientVertex > &  clusters, const reco::BeamSpot &  beamSpot, const bool  useBeamConstraint  )

inlineoverridevirtual

Implements PrimaryVertexFitterBase.

Definition at line 469 of file WeightedMeanFitter.h.

References pwdgSkimBPark_cfi::beamSpot, bsc_activity_cfg::clusters, submitPVResolutionJobs::err, AlCaHLTBitMon_ParallelJobs::p, SiStripOfflineCRack_cfg::p2, FSQDQM_cfi::pvs, fileCollector::seed, WeightedMeanFitter::startError, HLT_2024v13_cff::useBeamConstraint, findQualityFiles::v, WeightedMeanFitter::weightedMeanOutlierRejection(), and WeightedMeanFitter::weightedMeanOutlierRejectionBeamSpot().

Referenced by trackingPlots.Iteration::modules().

                                                                          {
    std::vector<TransientVertex> pvs;
    std::vector<TransientVertex> seed(1);

    for (auto& cluster : clusters) {
      if (cluster.originalTracks().size() > 1) {
        std::vector<reco::TransientTrack> tracklist = cluster.originalTracks();
        TransientVertex::TransientTrackToFloatMap trkWeightMap;
        std::vector<std::pair<GlobalPoint, GlobalPoint>> points;
        if (useBeamConstraint && (tracklist.size() > 1)) {
          for (const auto& itrack : tracklist) {
            GlobalPoint p = itrack.stateAtBeamLine().trackStateAtPCA().position();
            GlobalPoint err(itrack.stateAtBeamLine().transverseImpactParameter().error(),
                            itrack.stateAtBeamLine().transverseImpactParameter().error(),
                            itrack.track().dzError());
            std::pair<GlobalPoint, GlobalPoint> p2(p, err);
            points.push_back(p2);
          }

          TransientVertex v = WeightedMeanFitter::weightedMeanOutlierRejectionBeamSpot(points, tracklist, beamSpot);
          if (!v.hasTrackWeight()) {
            // if the fitter doesn't provide weights, fill dummy values
            TransientVertex::TransientTrackToFloatMap trkWeightMap;
            for (const auto& trk : v.originalTracks()) {
              trkWeightMap[trk] = 1.;
            }
            v.weightMap(trkWeightMap);
          }
          if ((v.positionError().matrix())(2, 2) != (WeightedMeanFitter::startError * WeightedMeanFitter::startError))
            pvs.push_back(v);
        } else if (!(useBeamConstraint) && (tracklist.size() > 1)) {
          for (const auto& itrack : tracklist) {
            GlobalPoint p = itrack.impactPointState().globalPosition();
            GlobalPoint err(itrack.track().dxyError(), itrack.track().dxyError(), itrack.track().dzError());
            std::pair<GlobalPoint, GlobalPoint> p2(p, err);
            points.push_back(p2);
          }

          TransientVertex v = WeightedMeanFitter::weightedMeanOutlierRejection(points, tracklist);
          if (!v.hasTrackWeight()) {
            // if the fitter doesn't provide weights, fill dummy values
            TransientVertex::TransientTrackToFloatMap trkWeightMap;
            for (const auto& trk : v.originalTracks()) {
              trkWeightMap[trk] = 1.;
            }
            v.weightMap(trkWeightMap);
          }
          if ((v.positionError().matrix())(2, 2) != (WeightedMeanFitter::startError * WeightedMeanFitter::startError))
            pvs.push_back(v);  //FIX with constants
        }
      }
    }
    return pvs;
  }