PrimaryVertexProducerAlgorithm Class Reference

#include <RecoVertex/PrimaryVertexProducerAlgorithm/src/PrimaryVertexProducerAlgorithm.cc>

Inheritance diagram for PrimaryVertexProducerAlgorithm:

Classes
struct	algo

Public Member Functions
PrimaryVertexProducerAlgorithm *	clone () const override
edm::ParameterSet	config () const
	PrimaryVertexProducerAlgorithm (const edm::ParameterSet &)
std::vector< TransientVertex >	vertices (const std::vector< reco::TransientTrack > &tracks) const override
virtual std::vector< TransientVertex >	vertices (const std::vector< reco::TransientTrack > &tracks, const reco::BeamSpot &beamSpot, const std::string &label="") const
	~PrimaryVertexProducerAlgorithm () override
Public Member Functions inherited from VertexReconstructor
	VertexReconstructor ()
virtual std::vector< TransientVertex >	vertices (const std::vector< reco::TransientTrack > &t, const reco::BeamSpot &) const
virtual std::vector< TransientVertex >	vertices (const std::vector< reco::TransientTrack > &primaries, const std::vector< reco::TransientTrack > &tracks, const reco::BeamSpot &spot) const
virtual	~VertexReconstructor ()

Public Attributes
edm::InputTag	beamSpotLabel
edm::InputTag	trackLabel

Private Attributes
std::vector< algo >	algorithms
bool	fVerbose
edm::ParameterSet	theConfig
TrackClusterizerInZ *	theTrackClusterizer
TrackFilterForPVFindingBase *	theTrackFilter

Detailed Description

Description: allow redoing the primary vertex reconstruction from a list of tracks, considered obsolete

Implementation: <Notes on="" implementation>="">

Definition at line 57 of file PrimaryVertexProducerAlgorithm.h.

Constructor & Destructor Documentation

PrimaryVertexProducerAlgorithm::PrimaryVertexProducerAlgorithm ( const edm::ParameterSet &  conf )

explicit

Definition at line 19 of file PrimaryVertexProducerAlgorithm.cc.

References qcdUeDQM_cfi::algorithm, algorithms, beamSpotLabel, edm::ParameterSet::exists(), PrimaryVertexProducerAlgorithm::algo::fitter, fVerbose, edm::ParameterSet::getParameter(), edm::ParameterSet::getUntrackedParameter(), PrimaryVertexProducerAlgorithm::algo::label, PrimaryVertexProducerAlgorithm::algo::minNdof, AlCaHLTBitMon_QueryRunRegistry::string, theTrackClusterizer, theTrackFilter, trackLabel, PrimaryVertexProducerAlgorithm::algo::useBeamConstraint, HLT_2018_cff::vertexCollections, and PrimaryVertexProducerAlgorithm::algo::vertexSelector.

Referenced by clone().

                                                                                          : theConfig(conf) {
  fVerbose = conf.getUntrackedParameter<bool>("verbose", false);
  trackLabel = conf.getParameter<edm::InputTag>("TrackLabel");
  beamSpotLabel = conf.getParameter<edm::InputTag>("beamSpotLabel");

  // select and configure the track selection
  std::string trackSelectionAlgorithm =
      conf.getParameter<edm::ParameterSet>("TkFilterParameters").getParameter<std::string>("algorithm");
  if (trackSelectionAlgorithm == "filter") {
    theTrackFilter = new TrackFilterForPVFinding(conf.getParameter<edm::ParameterSet>("TkFilterParameters"));
  } else if (trackSelectionAlgorithm == "filterWithThreshold") {
    theTrackFilter = new HITrackFilterForPVFinding(conf.getParameter<edm::ParameterSet>("TkFilterParameters"));
  } else {
    throw VertexException("PrimaryVertexProducerAlgorithm: unknown track selection algorithm: " +
                          trackSelectionAlgorithm);
  }

  // select and configure the track clusterizer
  std::string clusteringAlgorithm =
      conf.getParameter<edm::ParameterSet>("TkClusParameters").getParameter<std::string>("algorithm");
  if (clusteringAlgorithm == "gap") {
    theTrackClusterizer = new GapClusterizerInZ(
        conf.getParameter<edm::ParameterSet>("TkClusParameters").getParameter<edm::ParameterSet>("TkGapClusParameters"));
  } else if (clusteringAlgorithm == "DA") {
    theTrackClusterizer = new DAClusterizerInZ(
        conf.getParameter<edm::ParameterSet>("TkClusParameters").getParameter<edm::ParameterSet>("TkDAClusParameters"));
  }
  // provide the vectorized version of the clusterizer, if supported by the build
  else if (clusteringAlgorithm == "DA_vect") {
    theTrackClusterizer = new DAClusterizerInZ_vect(
        conf.getParameter<edm::ParameterSet>("TkClusParameters").getParameter<edm::ParameterSet>("TkDAClusParameters"));
  }

  else {
    throw VertexException("PrimaryVertexProducerAlgorithm: unknown clustering algorithm: " + clusteringAlgorithm);
  }

  // select and configure the vertex fitters
  if (conf.exists("vertexCollections")) {
    std::vector<edm::ParameterSet> vertexCollections =
        conf.getParameter<std::vector<edm::ParameterSet> >("vertexCollections");

    for (std::vector<edm::ParameterSet>::const_iterator algoconf = vertexCollections.begin();
         algoconf != vertexCollections.end();
         algoconf++) {
      algo algorithm;
      std::string fitterAlgorithm = algoconf->getParameter<std::string>("algorithm");
      if (fitterAlgorithm == "KalmanVertexFitter") {
        algorithm.fitter = new KalmanVertexFitter();
      } else if (fitterAlgorithm == "AdaptiveVertexFitter") {
        algorithm.fitter = new AdaptiveVertexFitter();
      } else {
        throw VertexException("PrimaryVertexProducerAlgorithm: unknown algorithm: " + fitterAlgorithm);
      }
      algorithm.label = algoconf->getParameter<std::string>("label");
      algorithm.minNdof = algoconf->getParameter<double>("minNdof");
      algorithm.useBeamConstraint = algoconf->getParameter<bool>("useBeamConstraint");
      algorithm.vertexSelector =
          new VertexCompatibleWithBeam(VertexDistanceXY(), algoconf->getParameter<double>("maxDistanceToBeam"));
      algorithms.push_back(algorithm);
    }
  } else {
    edm::LogWarning("MisConfiguration")
        << "this module's configuration has changed, please update to have a vertexCollections=cms.VPSet parameter.";

    algo algorithm;
    std::string fitterAlgorithm = conf.getParameter<std::string>("algorithm");
    if (fitterAlgorithm == "KalmanVertexFitter") {
      algorithm.fitter = new KalmanVertexFitter();
    } else if (fitterAlgorithm == "AdaptiveVertexFitter") {
      algorithm.fitter = new AdaptiveVertexFitter();
    } else {
      throw VertexException("PrimaryVertexProducerAlgorithm: unknown algorithm: " + fitterAlgorithm);
    }
    algorithm.label = "";
    algorithm.minNdof = conf.getParameter<double>("minNdof");
    algorithm.useBeamConstraint = conf.getParameter<bool>("useBeamConstraint");

    algorithm.vertexSelector = new VertexCompatibleWithBeam(
        VertexDistanceXY(),
        conf.getParameter<edm::ParameterSet>("PVSelParameters").getParameter<double>("maxDistanceToBeam"));

    algorithms.push_back(algorithm);
  }
}

PrimaryVertexProducerAlgorithm::~PrimaryVertexProducerAlgorithm ( )

override

Definition at line 105 of file PrimaryVertexProducerAlgorithm.cc.

References qcdUeDQM_cfi::algorithm, algorithms, theTrackClusterizer, and theTrackFilter.

                                                                {
  if (theTrackFilter)
    delete theTrackFilter;
  if (theTrackClusterizer)
    delete theTrackClusterizer;
  for (std::vector<algo>::const_iterator algorithm = algorithms.begin(); algorithm != algorithms.end(); algorithm++) {
    if (algorithm->fitter)
      delete algorithm->fitter;
    if (algorithm->vertexSelector)
      delete algorithm->vertexSelector;
  }
}

Member Function Documentation

PrimaryVertexProducerAlgorithm* PrimaryVertexProducerAlgorithm::clone ( void  ) const

inlineoverridevirtual

Clone method

Implements VertexReconstructor.

Definition at line 70 of file PrimaryVertexProducerAlgorithm.h.

References PrimaryVertexProducerAlgorithm().

{ return new PrimaryVertexProducerAlgorithm(*this); }

edm::ParameterSet PrimaryVertexProducerAlgorithm::config ( void  ) const

inline

Definition at line 73 of file PrimaryVertexProducerAlgorithm.h.

References theConfig.

{ return theConfig; }

std::vector< TransientVertex > PrimaryVertexProducerAlgorithm::vertices ( const std::vector< reco::TransientTrack > &  ) const

overridevirtual

Reconstruct vertices

Implements VertexReconstructor.

Definition at line 123 of file PrimaryVertexProducerAlgorithm.cc.

                                                       {
  throw VertexException("PrimaryVertexProducerAlgorithm: cannot make a Primary Vertex without a beam spot");

  return std::vector<TransientVertex>();
}

std::vector< TransientVertex > PrimaryVertexProducerAlgorithm::vertices ( const std::vector< reco::TransientTrack > &  tracks, const reco::BeamSpot &  beamSpot, const std::string &  label = ""  ) const

virtual

Definition at line 130 of file PrimaryVertexProducerAlgorithm.cc.

References qcdUeDQM_cfi::algorithm, algorithms, TrackClusterizerInZ::clusterize(), bsc_activity_cfg::clusters, gather_cfg::cout, GlobalErrorBase< T, ErrorWeightType >::cxx(), GlobalErrorBase< T, ErrorWeightType >::cyy(), GlobalErrorBase< T, ErrorWeightType >::czz(), TransientVertex::degreesOfFreedom(), VertexState::error(), fVerbose, TransientVertex::isValid(), GlobalErrorBase< T, ErrorWeightType >::matrix(), TransientVertex::position(), FSQDQM_cfi::pvs, TrackFilterForPVFindingBase::select(), theTrackClusterizer, theTrackFilter, findQualityFiles::v, PV3DBase< T, PVType, FrameType >::x(), PV3DBase< T, PVType, FrameType >::y(), and PV3DBase< T, PVType, FrameType >::z().

                                                                                                    {
  bool validBS = true;
  VertexState beamVertexState(beamSpot);
  if ((beamVertexState.error().cxx() <= 0.) || (beamVertexState.error().cyy() <= 0.) ||
      (beamVertexState.error().czz() <= 0.)) {
    validBS = false;
    edm::LogError("UnusableBeamSpot") << "Beamspot with invalid errors " << beamVertexState.error().matrix();
  }

  //   // get RECO tracks from the event
  //   // `tks` can be used as a ptr to a reco::TrackCollection
  //   edm::Handle<reco::TrackCollection> tks;
  //   iEvent.getByLabel(trackLabel, tks);

  // select tracks
  std::vector<reco::TransientTrack> seltks = theTrackFilter->select(t_tks);

  // clusterize tracks in Z
  std::vector<std::vector<reco::TransientTrack> > clusters = theTrackClusterizer->clusterize(seltks);
  if (fVerbose) {
    std::cout << " clustering returned  " << clusters.size() << " clusters  from " << seltks.size()
              << " selected tracks" << std::endl;
  }

  // vertex fits
  for (std::vector<algo>::const_iterator algorithm = algorithms.begin(); algorithm != algorithms.end(); algorithm++) {
    if (!(algorithm->label == label))
      continue;

    //std::auto_ptr<reco::VertexCollection> result(new reco::VertexCollection);
    // reco::VertexCollection vColl;

    std::vector<TransientVertex> pvs;
    for (std::vector<std::vector<reco::TransientTrack> >::const_iterator iclus = clusters.begin();
         iclus != clusters.end();
         iclus++) {
      TransientVertex v;
      if (algorithm->useBeamConstraint && validBS && ((*iclus).size() > 1)) {
        v = algorithm->fitter->vertex(*iclus, beamSpot);

      } else if (!(algorithm->useBeamConstraint) && ((*iclus).size() > 1)) {
        v = algorithm->fitter->vertex(*iclus);

      }  // else: no fit ==> v.isValid()=False

      if (fVerbose) {
        if (v.isValid())
          std::cout << "x,y,z=" << v.position().x() << " " << v.position().y() << " " << v.position().z() << std::endl;
        else
          std::cout << "Invalid fitted vertex\n";
      }

      if (v.isValid() && (v.degreesOfFreedom() >= algorithm->minNdof) &&
          (!validBS || (*(algorithm->vertexSelector))(v, beamVertexState)))
        pvs.push_back(v);
    }  // end of cluster loop

    if (fVerbose) {
      std::cout << "PrimaryVertexProducerAlgorithm::vertices  candidates =" << pvs.size() << std::endl;
    }

    // sort vertices by pt**2  vertex (aka signal vertex tagging)
    if (pvs.size() > 1) {
      sort(pvs.begin(), pvs.end(), VertexHigherPtSquared());
    }

    return pvs;
  }

  std::vector<TransientVertex> dummy;
  return dummy;  //avoid compiler warning, should never be here
}

Member Data Documentation

std::vector<algo> PrimaryVertexProducerAlgorithm::algorithms

private

Definition at line 92 of file PrimaryVertexProducerAlgorithm.h.

Referenced by PrimaryVertexProducerAlgorithm(), vertices(), and ~PrimaryVertexProducerAlgorithm().

edm::InputTag PrimaryVertexProducerAlgorithm::beamSpotLabel

Definition at line 75 of file PrimaryVertexProducerAlgorithm.h.

Referenced by PrimaryVertexProducerAlgorithm().

bool PrimaryVertexProducerAlgorithm::fVerbose

private

Definition at line 95 of file PrimaryVertexProducerAlgorithm.h.

Referenced by PrimaryVertexProducerAlgorithm(), and vertices().

edm::ParameterSet PrimaryVertexProducerAlgorithm::theConfig

private

Definition at line 94 of file PrimaryVertexProducerAlgorithm.h.

Referenced by config().

TrackClusterizerInZ* PrimaryVertexProducerAlgorithm::theTrackClusterizer

private

Definition at line 81 of file PrimaryVertexProducerAlgorithm.h.

Referenced by PrimaryVertexProducerAlgorithm(), vertices(), and ~PrimaryVertexProducerAlgorithm().

TrackFilterForPVFindingBase* PrimaryVertexProducerAlgorithm::theTrackFilter

private

Definition at line 80 of file PrimaryVertexProducerAlgorithm.h.

Referenced by PrimaryVertexProducerAlgorithm(), vertices(), and ~PrimaryVertexProducerAlgorithm().

edm::InputTag PrimaryVertexProducerAlgorithm::trackLabel

Definition at line 74 of file PrimaryVertexProducerAlgorithm.h.

Referenced by PrimaryVertexProducerAlgorithm().