#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

Public Member Functions inherited from edm::stream::EDProducerBase
	EDProducerBase ()

ModuleDescription const &	moduleDescription () const

	~EDProducerBase () override

Public Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

std::vector< edm::ProductResolverIndex > const &	indiciesForPutProducts (BranchType iBranchType) const

	ProducerBase ()

std::vector< edm::ProductResolverIndex > const &	putTokenIndexToProductResolverIndex () const

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...

void	resolvePutIndicies (BranchType iBranchType, ModuleToResolverIndicies const &iIndicies, std::string const &moduleLabel)

virtual	~ProducerBase () noexcept(false)

Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const

void	convertCurrentProcessAlias (std::string const &processName)
	Convert "@currentProcess" in InputTag process names to the actual current process name. More...

	EDConsumerBase ()

	EDConsumerBase (EDConsumerBase const &)=delete

	EDConsumerBase (EDConsumerBase &&)=default

ProductResolverIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

std::vector< ProductResolverIndexAndSkipBit > const &	itemsToGetFrom (BranchType iType) const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const

EDConsumerBase const &	operator= (EDConsumerBase const &)=delete

EDConsumerBase &	operator= (EDConsumerBase &&)=default

bool	registeredToConsume (ProductResolverIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)

virtual	~EDConsumerBase () noexcept(false)

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

Static Public Member Functions inherited from edm::stream::EDProducerBase
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (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

Public Types inherited from edm::stream::EDProducerBase
typedef EDProducerAdaptorBase	ModuleType

Public Types inherited from edm::ProducerBase
using	ModuleToResolverIndicies = std::unordered_multimap< std::string, std::tuple< edm::TypeID const , const char , edm::ProductResolverIndex >>

typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)

ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...

template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()

void	consumesMany (const TypeToGet &id)

template<BranchType B>
void	consumesMany (const TypeToGet &id)

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

Detailed Description

Definition at line 13 of file HIBestVertexProducer.h.

Constructor & Destructor Documentation

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

explicit

Definition at line 20 of file HIBestVertexProducer.cc.

References edm::ParameterSet::getParameter().

                             : 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 34 of file HIBestVertexProducer.cc.

35 {

36 }

Member Function Documentation

void HIBestVertexProducer::beginJob ( void )

private

Definition at line 51 of file HIBestVertexProducer.cc.

References produce().

52 {

53 }

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

static

Definition at line 39 of file HIBestVertexProducer.cc.

References edm::ConfigurationDescriptions::add(), and edm::ParameterSetDescription::add().

                                                                                        {
    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 57 of file HIBestVertexProducer.cc.

References ecalDrivenElectronSeedsParameters_cff::beamSpot, reco::BeamSpot::BeamWidthX(), reco::BeamSpot::BeamWidthY(), edm::Event::getByToken(), edm::HandleBase::isValid(), eostools::move(), reco::BeamSpot::position(), funct::pow(), edm::Handle< T >::product(), edm::Event::put(), reco::BeamSpot::sigmaZ(), reco::BeamSpot::x0(), and reco::BeamSpot::y0().

Referenced by beginJob().

 {
   
   // 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 26 of file HIBestVertexProducer.h.

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

private

Definition at line 24 of file HIBestVertexProducer.h.

edm::ParameterSet HIBestVertexProducer::theConfig

private

Definition at line 23 of file HIBestVertexProducer.h.

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

private

Definition at line 27 of file HIBestVertexProducer.h.

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

private

Definition at line 25 of file HIBestVertexProducer.h.

bool HIBestVertexProducer::theUseFinalAdaptiveVertexCollection

private

Definition at line 28 of file HIBestVertexProducer.h.

Public Member Functions

Static Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation