HIBestVertexProducer Class Reference

#include <HIBestVertexProducer.h>

Inheritance diagram for HIBestVertexProducer:

Public Member Functions
	HIBestVertexProducer (const edm::ParameterSet &ps)
void	produce (edm::Event &ev, const edm::EventSetup &es) override
	~HIBestVertexProducer () override
Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default
	EDProducer (const EDProducer &)=delete
bool	hasAbilityToProduceInBeginLumis () const final
bool	hasAbilityToProduceInBeginProcessBlocks () const final
bool	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () const final
bool	hasAbilityToProduceInEndProcessBlocks () const final
bool	hasAbilityToProduceInEndRuns () const final
const EDProducer &	operator= (const EDProducer &)=delete

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &descriptions)

Private Member Functions
void	beginJob ()

Private Attributes
edm::EDGetTokenT < reco::VertexCollection >	theAdaptiveVertexCollection
edm::EDGetTokenT< reco::BeamSpot >	theBeamSpotTag
edm::ParameterSet	theConfig
edm::EDGetTokenT < reco::VertexCollection >	theFinalAdaptiveVertexCollection
edm::EDGetTokenT < reco::VertexCollection >	theMedianVertexCollection
bool	theUseFinalAdaptiveVertexCollection

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer<>
using	CacheTypes = CacheContexts< T...>
using	GlobalCache = typename CacheTypes::GlobalCache
using	HasAbility = AbilityChecker< T...>
using	InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache
using	LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache
using	LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >
using	LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache
using	RunCache = typename CacheTypes::RunCache
using	RunContext = RunContextT< RunCache, GlobalCache >
using	RunSummaryCache = typename CacheTypes::RunSummaryCache

Detailed Description

Definition at line 16 of file HIBestVertexProducer.h.

Constructor & Destructor Documentation

HIBestVertexProducer::HIBestVertexProducer ( const edm::ParameterSet &  ps )

explicit

Definition at line 17 of file HIBestVertexProducer.cc.

References edm::ParameterSet::getParameter(), theFinalAdaptiveVertexCollection, and theUseFinalAdaptiveVertexCollection.

    : theConfig(ps),
      theBeamSpotTag(consumes<reco::BeamSpot>(ps.getParameter<edm::InputTag>("beamSpotLabel"))),
      theMedianVertexCollection(
          consumes<reco::VertexCollection>(ps.getParameter<edm::InputTag>("medianVertexCollection"))),
      theAdaptiveVertexCollection(
          consumes<reco::VertexCollection>(ps.getParameter<edm::InputTag>("adaptiveVertexCollection"))),
      theUseFinalAdaptiveVertexCollection(ps.getParameter<bool>("useFinalAdaptiveVertexCollection")) {
  if (theUseFinalAdaptiveVertexCollection) {
    theFinalAdaptiveVertexCollection =
        consumes<reco::VertexCollection>(ps.getParameter<edm::InputTag>("finalAdaptiveVertexCollection"));
  }
  produces<reco::VertexCollection>();
}

HIBestVertexProducer::~HIBestVertexProducer ( )

override

Definition at line 33 of file HIBestVertexProducer.cc.

{}

Member Function Documentation

void HIBestVertexProducer::beginJob ( void  )

private

Definition at line 48 of file HIBestVertexProducer.cc.

{}

void HIBestVertexProducer::fillDescriptions ( edm::ConfigurationDescriptions &  descriptions )

static

Definition at line 36 of file HIBestVertexProducer.cc.

References edm::ConfigurationDescriptions::add(), edm::ParameterSetDescription::add(), submitPVResolutionJobs::desc, and HLT_FULL_cff::InputTag.

                                                                                      {
  edm::ParameterSetDescription desc;
  desc.add<InputTag>("beamSpotLabel", edm::InputTag("offlineBeamSpot"));
  desc.add<InputTag>("adaptiveVertexCollection", edm::InputTag("hiBestAdaptiveVertex"));
  desc.add<InputTag>("medianVertexCollection", edm::InputTag("hiPixelMedianVertex"));
  desc.add<bool>("useFinalAdaptiveVertexCollection", false);
  desc.add<InputTag>("finalAdaptiveVertexCollection", edm::InputTag("hiBestOfflinePrimaryVertex"));

  descriptions.add("hiSelectedPixelVertex", desc);
}

void HIBestVertexProducer::produce ( edm::Event &  ev, const edm::EventSetup &  es  )

override

Definition at line 51 of file HIBestVertexProducer.cc.

References beam_dqm_sourceclient-live_cfg::beamSpot, reco::BeamSpot::BeamWidthX(), reco::BeamSpot::BeamWidthY(), submitPVValidationJobs::err, edm::Event::getByToken(), edm::HandleBase::isValid(), eostools::move(), reco::BeamSpot::position(), funct::pow(), edm::Handle< T >::product(), edm::Event::put(), reco::BeamSpot::sigmaZ(), theAdaptiveVertexCollection, theBeamSpotTag, theFinalAdaptiveVertexCollection, theMedianVertexCollection, theUseFinalAdaptiveVertexCollection, reco::BeamSpot::x0(), and reco::BeamSpot::y0().

                                                                        {
  // 1. use best adaptive vertex preferentially
  // 2. use median vertex in case where adaptive algorithm failed
  // 3. use beamspot if netither vertexing method succeeds

  // New vertex collection
  auto newVertexCollection = std::make_unique<reco::VertexCollection>();

  //** Get precise adaptive vertex **/
  edm::Handle<reco::VertexCollection> vc1;
  ev.getByToken(theAdaptiveVertexCollection, vc1);
  const reco::VertexCollection* vertices1 = vc1.product();

  if (vertices1->empty())
    LogError("HeavyIonVertexing") << "adaptive vertex collection is empty!" << endl;

  //** Final vertex collection if needed **//
  const double maxZError = 3.0;  //any vtx with error > this number is considered no good
  bool hasFinalVertex = false;
  if (theUseFinalAdaptiveVertexCollection) {
    edm::Handle<reco::VertexCollection> vc0;
    ev.getByToken(theFinalAdaptiveVertexCollection, vc0);
    const reco::VertexCollection* vertices0 = vc0.product();
    if (vertices0->empty())
      LogInfo("HeavyIonVertexing") << "final adaptive vertex collection is empty!" << endl;

    //if using final vertex and has a good vertex, use it
    if (vertices0->begin()->zError() < maxZError) {
      hasFinalVertex = true;
      auto const& vertex0 = vertices0->front();
      newVertexCollection->push_back(vertex0);
      LogInfo("HeavyIonVertexing") << "adaptive vertex:\n vz = (" << vertex0.x() << ", " << vertex0.y() << ", "
                                   << vertex0.z() << ")"
                                   << "\n error = (" << vertex0.xError() << ", " << vertex0.yError() << ", "
                                   << vertex0.zError() << ")" << endl;
    }
  }

  //otherwise use the pixel track adaptive vertex if it is good
  if (!hasFinalVertex) {
    if (vertices1->begin()->zError() < maxZError) {
      reco::VertexCollection::const_iterator vertex1 = vertices1->begin();
      newVertexCollection->push_back(*vertex1);

      LogInfo("HeavyIonVertexing") << "adaptive vertex:\n vz = (" << vertex1->x() << ", " << vertex1->y() << ", "
                                   << vertex1->z() << ")"
                                   << "\n error = (" << vertex1->xError() << ", " << vertex1->yError() << ", "
                                   << vertex1->zError() << ")" << endl;
    } else {
      //** Get fast median vertex **/
      edm::Handle<reco::VertexCollection> vc2;
      ev.getByToken(theMedianVertexCollection, vc2);
      const reco::VertexCollection* vertices2 = vc2.product();

      //** Get beam spot position and error **/
      reco::BeamSpot beamSpot;
      edm::Handle<reco::BeamSpot> beamSpotHandle;
      ev.getByToken(theBeamSpotTag, beamSpotHandle);

      if (beamSpotHandle.isValid())
        beamSpot = *beamSpotHandle;
      else
        LogError("HeavyIonVertexing") << "no beamspot found " << endl;

      if (!vertices2->empty()) {
        reco::VertexCollection::const_iterator vertex2 = vertices2->begin();
        reco::Vertex::Error err;
        err(0, 0) = pow(beamSpot.BeamWidthX(), 2);
        err(1, 1) = pow(beamSpot.BeamWidthY(), 2);
        err(2, 2) = pow(vertex2->zError(), 2);
        reco::Vertex newVertex(reco::Vertex::Point(beamSpot.x0(), beamSpot.y0(), vertex2->z()), err, 0, 1, 1);
        newVertexCollection->push_back(newVertex);

        LogInfo("HeavyIonVertexing") << "median vertex + beamspot: \n position = (" << newVertex.x() << ", "
                                     << newVertex.y() << ", " << newVertex.z() << ")"
                                     << "\n error = (" << newVertex.xError() << ", " << newVertex.yError() << ", "
                                     << newVertex.zError() << ")" << endl;

      } else {
        reco::Vertex::Error err;
        err(0, 0) = pow(beamSpot.BeamWidthX(), 2);
        err(1, 1) = pow(beamSpot.BeamWidthY(), 2);
        err(2, 2) = pow(beamSpot.sigmaZ(), 2);
        reco::Vertex newVertex(beamSpot.position(), err, 0, 0, 1);
        newVertexCollection->push_back(newVertex);

        LogInfo("HeavyIonVertexing") << "beam spot: \n position = (" << newVertex.x() << ", " << newVertex.y() << ", "
                                     << newVertex.z() << ")"
                                     << "\n error = (" << newVertex.xError() << ", " << newVertex.yError() << ", "
                                     << newVertex.zError() << ")" << endl;
      }
    }
  }

  // put new vertex collection into event
  ev.put(std::move(newVertexCollection));
}

Member Data Documentation

edm::EDGetTokenT<reco::VertexCollection> HIBestVertexProducer::theAdaptiveVertexCollection

private

Definition at line 28 of file HIBestVertexProducer.h.

Referenced by produce().

edm::EDGetTokenT<reco::BeamSpot> HIBestVertexProducer::theBeamSpotTag

private

Definition at line 26 of file HIBestVertexProducer.h.

Referenced by produce().

edm::ParameterSet HIBestVertexProducer::theConfig

private

Definition at line 25 of file HIBestVertexProducer.h.

edm::EDGetTokenT<reco::VertexCollection> HIBestVertexProducer::theFinalAdaptiveVertexCollection

private

Definition at line 29 of file HIBestVertexProducer.h.

Referenced by HIBestVertexProducer(), and produce().

edm::EDGetTokenT<reco::VertexCollection> HIBestVertexProducer::theMedianVertexCollection

private

Definition at line 27 of file HIBestVertexProducer.h.

Referenced by produce().

bool HIBestVertexProducer::theUseFinalAdaptiveVertexCollection

private

Definition at line 30 of file HIBestVertexProducer.h.

Referenced by HIBestVertexProducer(), and produce().