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
bool	hasAbilityToProduceInBeginLumis () const final
bool	hasAbilityToProduceInBeginProcessBlocks () const final
bool	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () const final
bool	hasAbilityToProduceInEndProcessBlocks () const final
bool	hasAbilityToProduceInEndRuns () const final

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<>
typedef CacheContexts< T... >	CacheTypes
typedef CacheTypes::GlobalCache	GlobalCache
typedef AbilityChecker< T... >	HasAbility
typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache
typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext
typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache
typedef CacheTypes::RunCache	RunCache
typedef RunContextT< RunCache, GlobalCache >	RunContext
typedef CacheTypes::RunSummaryCache	RunSummaryCache

Detailed Description

Definition at line 16 of file HIBestVertexProducer.h.

Constructor & Destructor Documentation

HIBestVertexProducer()

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

explicit

Definition at line 18 of file HIBestVertexProducer.cc.

    : 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>();
}

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

~HIBestVertexProducer()

HIBestVertexProducer::~HIBestVertexProducer ( )

override

Definition at line 34 of file HIBestVertexProducer.cc.

{}

Member Function Documentation

beginJob()

void HIBestVertexProducer::beginJob ( void  )

private

Definition at line 49 of file HIBestVertexProducer.cc.

{}

fillDescriptions()

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

static

Definition at line 37 of file HIBestVertexProducer.cc.

                                                                                      {
  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);
}

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

produce()

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

override

Definition at line 52 of file HIBestVertexProducer.cc.

                                                                        {
  // 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));
}

References pwdgSkimBPark_cfi::beamSpot, submitPVResolutionJobs::err, ev, edm::HandleBase::isValid(), eostools::move(), funct::pow(), edm::Handle< T >::product(), theAdaptiveVertexCollection, theBeamSpotTag, theFinalAdaptiveVertexCollection, theMedianVertexCollection, and theUseFinalAdaptiveVertexCollection.

Member Data Documentation

theAdaptiveVertexCollection

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

private

Definition at line 28 of file HIBestVertexProducer.h.

Referenced by produce().

theBeamSpotTag

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

private

Definition at line 26 of file HIBestVertexProducer.h.

Referenced by produce().

theConfig

edm::ParameterSet HIBestVertexProducer::theConfig

private

Definition at line 25 of file HIBestVertexProducer.h.

theFinalAdaptiveVertexCollection

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

private

Definition at line 29 of file HIBestVertexProducer.h.

Referenced by HIBestVertexProducer(), and produce().

theMedianVertexCollection

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

private

Definition at line 27 of file HIBestVertexProducer.h.

Referenced by produce().

theUseFinalAdaptiveVertexCollection

bool HIBestVertexProducer::theUseFinalAdaptiveVertexCollection

private

Definition at line 30 of file HIBestVertexProducer.h.

Referenced by HIBestVertexProducer(), and produce().