#include <VertexProducer.h>

Inheritance diagram for VertexProducer:

Public Member Functions
	VertexProducer (const edm::ParameterSet &)

	~VertexProducer () override

Public Member Functions inherited from edm::global::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

EDProducer &	operator= (const EDProducer &)=delete

bool	wantsGlobalLuminosityBlocks () const final

bool	wantsGlobalRuns () const final

bool	wantsInputProcessBlocks () const final

bool	wantsProcessBlocks () const final

bool	wantsStreamLuminosityBlocks () const final

bool	wantsStreamRuns () const final

Public Member Functions inherited from edm::global::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

std::vector< bool > const &	recordProvenanceList () 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)

TypeLabelList const &	typeLabelList () const
	used by the fwk to register the list of products of this module More...

	~ProducerBase () noexcept(false) override

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

ESResolverIndex const *	esGetTokenIndices (edm::Transition iTrans) const

std::vector< ESResolverIndex > const &	esGetTokenIndicesVector (edm::Transition iTrans) const

std::vector< ESRecordIndex > const &	esGetTokenRecordIndicesVector (edm::Transition iTrans) const

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::array< std::vector< ModuleDescription const > , NumBranchTypes > &modulesAll, std::vector< ModuleProcessName > &modulesInPreviousProcesses, 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

void	selectInputProcessBlocks (ProductRegistry const &productRegistry, ProcessBlockHelperBase const &processBlockHelperBase)

ProductResolverIndexAndSkipBit	uncheckedIndexFrom (EDGetToken) const

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

void	updateLookup (eventsetup::ESRecordsToProductResolverIndices const &)

virtual	~EDConsumerBase () noexcept(false)

Private Types
typedef std::vector< TTTrackType >	TTTrackCollectionType

typedef edm::View< TTTrackType >	TTTrackCollectionView

typedef edm::RefVector< TTTrackCollectionType >	TTTrackRefCollectionType

typedef TTTrack< Ref_Phase2TrackerDigi_ >	TTTrackType

Private Member Functions
void	produce (edm::StreamID, edm::Event &, const edm::EventSetup &) const override

Private Attributes
const edm::EDGetTokenT< TTTrackRefCollectionType >	l1TracksToken_

const std::string	outputCollectionName_

tensorflow::GraphDef *	PattRecGraph_

tensorflow::Session *	PattRecSesh_

l1tVertexFinder::AlgoSettings	settings_

tensorflow::GraphDef *	TrkWGraph_

tensorflow::Session *	TrkWSesh_

const edm::ESGetToken< TrackerTopology, TrackerTopologyRcd >	tTopoToken

Additional Inherited Members
Public Types inherited from edm::global::EDProducerBase
typedef EDProducerBase	ModuleType

Public Types inherited from edm::ProducerBase
template<typename T >
using	BranchAliasSetterT = ProductRegistryHelper::BranchAliasSetterT< T >

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

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

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

Protected Member Functions inherited from edm::ProducerBase
template<Transition Tr = Transition::Event>
auto	produces (std::string instanceName) noexcept
	declare what type of product will make and with which optional label More...

template<Transition B>
BranchAliasSetter	produces (const TypeID &id, std::string instanceName=std::string(), bool recordProvenance=true)

template<BranchType B>
BranchAliasSetter	produces (const TypeID &id, std::string instanceName=std::string(), bool recordProvenance=true)

BranchAliasSetter	produces (const TypeID &id, std::string instanceName=std::string(), bool recordProvenance=true)

template<typename ProductType , Transition B>
BranchAliasSetterT< ProductType >	produces (std::string instanceName)

template<class ProductType >
BranchAliasSetterT< ProductType >	produces ()

template<typename ProductType , BranchType B>
BranchAliasSetterT< ProductType >	produces (std::string instanceName)

template<typename ProductType , BranchType B>
BranchAliasSetterT< ProductType >	produces ()

template<class ProductType >
BranchAliasSetterT< ProductType >	produces (std::string instanceName)

template<typename ProductType , Transition B>
BranchAliasSetterT< ProductType >	produces ()

template<Transition Tr = Transition::Event>
auto	produces () noexcept

ProducesCollector	producesCollector ()

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

template<BranchType B = InEvent>
EDConsumerBaseAdaptor< B >	consumes (edm::InputTag tag) noexcept

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 ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes ()

template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag const &tag)

template<Transition Tr = Transition::Event>
constexpr auto	esConsumes ()

template<Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag tag)

template<Transition Tr = Transition::Event>
ESGetTokenGeneric	esConsumes (eventsetup::EventSetupRecordKey const &iRecord, eventsetup::DataKey const &iKey)
	Used with EventSetupRecord::doGet. More...

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)

void	resetItemsToGetFrom (BranchType iType)

Detailed Description

Definition at line 32 of file VertexProducer.h.

Member Typedef Documentation

◆ TTTrackCollectionType

typedef std::vector<TTTrackType> VertexProducer::TTTrackCollectionType

private

Definition at line 39 of file VertexProducer.h.

◆ TTTrackCollectionView

typedef edm::View<TTTrackType> VertexProducer::TTTrackCollectionView

private

Definition at line 41 of file VertexProducer.h.

◆ TTTrackRefCollectionType

typedef edm::RefVector<TTTrackCollectionType> VertexProducer::TTTrackRefCollectionType

private

Definition at line 40 of file VertexProducer.h.

◆ TTTrackType

typedef TTTrack<Ref_Phase2TrackerDigi_> VertexProducer::TTTrackType

private

Definition at line 38 of file VertexProducer.h.

Constructor & Destructor Documentation

◆ VertexProducer()

VertexProducer::VertexProducer ( const edm::ParameterSet & iConfig )

explicit

Definition at line 18 of file VertexProducer.cc.

References tensorflow::createSession(), l1tVertexFinder::AlgoSettings::debug(), tensorflow::loadGraphDef(), outputCollectionName_, PattRecGraph_, PattRecSesh_, settings_, TrkWGraph_, TrkWSesh_, l1tVertexFinder::AlgoSettings::vx_algo(), l1tVertexFinder::AlgoSettings::vx_pattrec_graph(), and l1tVertexFinder::AlgoSettings::vx_trkw_graph().

     : l1TracksToken_(consumes<TTTrackRefCollectionType>(iConfig.getParameter<edm::InputTag>("l1TracksInputTag"))),
       tTopoToken(esConsumes<TrackerTopology, TrackerTopologyRcd>()),
       outputCollectionName_(iConfig.getParameter<std::string>("l1VertexCollectionName")),
       settings_(AlgoSettings(iConfig)) {
   // Get configuration parameters
 
   switch (settings_.vx_algo()) {
     case Algorithm::fastHisto:
       edm::LogInfo("VertexProducer") << "VertexProducer::Finding vertices using the fastHisto binning algorithm";
       break;
     case Algorithm::fastHistoEmulation:
       edm::LogInfo("VertexProducer")
           << "VertexProducer::Finding vertices using the emulation version of the fastHisto binning algorithm";
       break;
     case Algorithm::fastHistoLooseAssociation:
       edm::LogInfo("VertexProducer")
           << "VertexProducer::Finding vertices using the fastHistoLooseAssociation binning algorithm";
       break;
     case Algorithm::GapClustering:
       edm::LogInfo("VertexProducer") << "VertexProducer::Finding vertices using a gap clustering algorithm";
       break;
     case Algorithm::agglomerativeHierarchical:
       edm::LogInfo("VertexProducer") << "VertexProducer::Finding vertices using a Simple Merge Clustering algorithm";
       break;
     case Algorithm::DBSCAN:
       edm::LogInfo("VertexProducer") << "VertexProducer::Finding vertices using a DBSCAN algorithm";
       break;
     case Algorithm::PVR:
       edm::LogInfo("VertexProducer") << "VertexProducer::Finding vertices using a PVR algorithm";
       break;
     case Algorithm::adaptiveVertexReconstruction:
       edm::LogInfo("VertexProducer")
           << "VertexProducer::Finding vertices using an AdaptiveVertexReconstruction algorithm";
       break;
     case Algorithm::HPV:
       edm::LogInfo("VertexProducer") << "VertexProducer::Finding vertices using a Highest Pt Vertex algorithm";
       break;
     case Algorithm::Kmeans:
       edm::LogInfo("VertexProducer") << "VertexProducer::Finding vertices using a kmeans algorithm";
       break;
     case Algorithm::NNEmulation:
       edm::LogInfo("VertexProducer") << "VertexProducer::Finding vertices using the Neural Network Emulation";
       break;
   }
 
   //--- Define EDM output to be written to file (if required)
   if (settings_.vx_algo() == Algorithm::fastHistoEmulation || settings_.vx_algo() == Algorithm::NNEmulation) {
     produces<l1t::VertexWordCollection>(outputCollectionName_ + "Emulation");
   } else {
     produces<l1t::VertexCollection>(outputCollectionName_);
   }
 
   if (settings_.vx_algo() == Algorithm::NNEmulation) {
     // load graphs, create a new session and add the graphDef
     if (settings_.debug() > 1) {
       edm::LogInfo("VertexProducer") << "loading TrkWeight graph from " << settings_.vx_trkw_graph() << std::endl;
       edm::LogInfo("VertexProducer") << "loading PatternRec graph from " << settings_.vx_pattrec_graph() << std::endl;
     }
     TrkWGraph_ = tensorflow::loadGraphDef(settings_.vx_trkw_graph());
     TrkWSesh_ = tensorflow::createSession(TrkWGraph_);
     PattRecGraph_ = tensorflow::loadGraphDef(settings_.vx_pattrec_graph());
     PattRecSesh_ = tensorflow::createSession(PattRecGraph_);
   }
 }

◆ ~VertexProducer()

VertexProducer::~VertexProducer ( )

inlineoverride

Definition at line 35 of file VertexProducer.h.

35 {}

Member Function Documentation

◆ produce()

void VertexProducer::produce	(	edm::StreamID	,
		edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)		const

overrideprivatevirtual

Implements edm::global::EDProducerBase.

Definition at line 84 of file VertexProducer.cc.

                                                                                              {
   edm::Handle<TTTrackRefCollectionType> l1TracksHandle;
   iEvent.getByToken(l1TracksToken_, l1TracksHandle);
 
   std::vector<l1tVertexFinder::L1Track> l1Tracks;
   l1Tracks.reserve(l1TracksHandle->size());
   if (settings_.debug() > 1) {
     edm::LogInfo("VertexProducer") << "produce::Processing " << l1TracksHandle->size() << " tracks";
   }
   for (const auto& track : *l1TracksHandle) {
     auto l1track = L1Track(edm::refToPtr(track));
     if (l1track.pt() >= settings_.vx_TrackMinPt()) {
       l1Tracks.push_back(l1track);
     } else {
       if (settings_.debug() > 2) {
         edm::LogInfo("VertexProducer") << "produce::Removing track with too low of a pt (" << l1track.pt() << ")\n"
                                        << "         word = " << l1track.getTTTrackPtr()->getTrackWord().to_string(2);
       }
     }
   }
 
   if (settings_.debug() > 1) {
     edm::LogInfo("VertexProducer") << "produce::Processing " << l1Tracks.size() << " tracks after minimum pt cut of"
                                    << settings_.vx_TrackMinPt() << " GeV";
   }
 
   VertexFinder vf(l1Tracks, settings_);
 
   switch (settings_.vx_algo()) {
     case Algorithm::fastHisto: {
       const TrackerTopology& tTopo = iSetup.getData(tTopoToken);
       vf.fastHisto(&tTopo);
       break;
     }
     case Algorithm::fastHistoEmulation:
       vf.fastHistoEmulation();
       break;
     case Algorithm::fastHistoLooseAssociation:
       vf.fastHistoLooseAssociation();
       break;
     case Algorithm::GapClustering:
       vf.GapClustering();
       break;
     case Algorithm::agglomerativeHierarchical:
       vf.agglomerativeHierarchicalClustering();
       break;
     case Algorithm::DBSCAN:
       vf.DBSCAN();
       break;
     case Algorithm::PVR:
       vf.PVR();
       break;
     case Algorithm::adaptiveVertexReconstruction:
       vf.adaptiveVertexReconstruction();
       break;
     case Algorithm::HPV:
       vf.HPV();
       break;
     case Algorithm::Kmeans:
       vf.Kmeans();
       break;
     case Algorithm::NNEmulation:
       vf.NNVtxEmulation(TrkWSesh_, PattRecSesh_);
       break;
   }
 
   vf.sortVerticesInPt();
   vf.findPrimaryVertex();
 
   // //=== Store output EDM track and hardware stub collections.
   if (settings_.vx_algo() == Algorithm::fastHistoEmulation || settings_.vx_algo() == Algorithm::NNEmulation) {
     std::unique_ptr<l1t::VertexWordCollection> product_emulation =
         std::make_unique<l1t::VertexWordCollection>(vf.verticesEmulation().begin(), vf.verticesEmulation().end());
     iEvent.put(std::move(product_emulation), outputCollectionName_ + "Emulation");
   } else {
     std::unique_ptr<l1t::VertexCollection> product(new std::vector<l1t::Vertex>());
     for (const auto& vtx : vf.vertices()) {
       product->emplace_back(vtx.vertex());
     }
     iEvent.put(std::move(product), outputCollectionName_);
   }
 }

Member Data Documentation

◆ l1TracksToken_

const edm::EDGetTokenT<TTTrackRefCollectionType> VertexProducer::l1TracksToken_

private

Definition at line 46 of file VertexProducer.h.

Referenced by produce().

◆ outputCollectionName_

const std::string VertexProducer::outputCollectionName_

private

Definition at line 48 of file VertexProducer.h.

Referenced by produce(), and VertexProducer().

◆ PattRecGraph_

tensorflow::GraphDef* VertexProducer::PattRecGraph_

private

Definition at line 52 of file VertexProducer.h.

Referenced by VertexProducer().

◆ PattRecSesh_

tensorflow::Session* VertexProducer::PattRecSesh_

private

Definition at line 53 of file VertexProducer.h.

Referenced by produce(), and VertexProducer().

◆ settings_

l1tVertexFinder::AlgoSettings VertexProducer::settings_

private

Definition at line 55 of file VertexProducer.h.

Referenced by produce(), and VertexProducer().

◆ TrkWGraph_

tensorflow::GraphDef* VertexProducer::TrkWGraph_

private

Definition at line 50 of file VertexProducer.h.

Referenced by VertexProducer().

◆ TrkWSesh_

tensorflow::Session* VertexProducer::TrkWSesh_

private

Definition at line 51 of file VertexProducer.h.

Referenced by produce(), and VertexProducer().

◆ tTopoToken

const edm::ESGetToken<TrackerTopology, TrackerTopologyRcd> VertexProducer::tTopoToken

private

Definition at line 47 of file VertexProducer.h.

Referenced by produce().

Public Member Functions

Private Types

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Member Typedef Documentation

◆ TTTrackCollectionType

◆ TTTrackCollectionView

◆ TTTrackRefCollectionType

◆ TTTrackType

Constructor & Destructor Documentation

◆ VertexProducer()

◆ ~VertexProducer()

Member Function Documentation

◆ produce()

Member Data Documentation

◆ l1TracksToken_

◆ outputCollectionName_

◆ PattRecGraph_

◆ PattRecSesh_

◆ settings_

◆ TrkWGraph_

◆ TrkWSesh_

◆ tTopoToken