FlavorTagDiscriminants::NNSharingSvc Class Reference

#include <NNSharingSvc.h>

Inheritance diagram for FlavorTagDiscriminants::NNSharingSvc:

Collaboration diagram for FlavorTagDiscriminants::NNSharingSvc:

Public Member Functions
	NNSharingSvc (const std::string &name, ISvcLocator *svc)
virtual std::shared_ptr< const GNN >	get (const std::string &nn_name, const GNNOptions &opts) override
virtual void	print () const
	Print the state of the service. More...
template<typename T >
void	declareServiceInterface ()
	add the given interface to the list of interfaces More...
virtual StatusCode	queryInterface (const InterfaceID &riid, void **ppvi)
	query interface for gaudi More...
MsgStream &	msg () const
MsgStream &	msg (const MSG::Level lvl) const
bool	msgLvl (const MSG::Level lvl) const
	DeclareInterfaceID (FlavorTagDiscriminants::INNSharingSvc, 1, 0)

Private Types
using	val_t = std::shared_ptr< const GNN >

Private Attributes
std::unordered_map< detail::NNKey, val_t, detail::NNHasher >	m_gnns
std::unordered_map< std::string, val_t >	m_base_gnns
std::vector< std::pair< const InterfaceID &(*)(), void *(*)(AsgService *)> >	m_interfaces
	list of interfaces we have More...
virtual StatusCode	initialize ()
	set up/tear down functions More...
virtual StatusCode	finalize ()

Detailed Description

Definition at line 31 of file NNSharingSvc.h.

Member Typedef Documentation

val_t

using FlavorTagDiscriminants::NNSharingSvc::val_t = std::shared_ptr<const GNN>

private

Definition at line 39 of file NNSharingSvc.h.

Constructor & Destructor Documentation

NNSharingSvc()

FlavorTagDiscriminants::NNSharingSvc::NNSharingSvc	(	const std::string &	name,
		ISvcLocator *	svc
	)

Definition at line 22 of file NNSharingSvc.cxx.

                                                                   :
    AsgService(name, svc)
  {
    declareServiceInterface<INNSharingSvc>();
  }

Member Function Documentation

DeclareInterfaceID()

FlavorTagDiscriminants::INNSharingSvc::DeclareInterfaceID ( FlavorTagDiscriminants::INNSharingSvc  , 1  , 0    )

inherited

declareServiceInterface()

template<typename T >

void asg::AsgService::declareServiceInterface

inherited

add the given interface to the list of interfaces

Definition at line 76 of file AsgService.h.

  {
#ifndef XAOD_STANDALONE
    m_interfaces.emplace_back (T::interfaceID, [] (AsgService *self) -> void* {return dynamic_cast<T*>(self);});
#endif
  }

finalize()

StatusCode asg::AsgService::finalize ( )

virtualinherited

Reimplemented in TrackAnalysisDefinitionSvc, CP::SystematicsSvc, PlotsDefinitionSvc, and AthOnnx::OnnxRuntimeSvc.

Definition at line 40 of file AsgService.cxx.

  {
    return StatusCode::SUCCESS;
  }

get()

std::shared_ptr< const GNN > FlavorTagDiscriminants::NNSharingSvc::get ( const std::string &  nn_name, const GNNOptions &  opts  )

overridevirtual

Implements FlavorTagDiscriminants::INNSharingSvc.

Definition at line 27 of file NNSharingSvc.cxx.

                            {
    detail::NNKey key{nn_name, opts};
    if (m_gnns.count(key)) {
      ATH_MSG_INFO("getting " << nn_name << " from cached NNs");
      return m_gnns.at(key);
    } else if (m_base_gnns.count(nn_name) ) {
      ATH_MSG_INFO("adapting " << nn_name << " from cached NNs, new opts");
      auto nn = std::make_shared<const GNN>(*m_base_gnns.at(nn_name), opts);
      m_gnns[key] = nn;
      return nn;
    }
    ATH_MSG_INFO("building " << nn_name << " from onnx file");
    auto nn = std::make_shared<const GNN>(nn_name, opts);
    m_base_gnns[nn_name] = nn;
    m_gnns[key] = nn;
    return nn;
  }

initialize()

StatusCode asg::AsgService::initialize ( )

virtualinherited

set up/tear down functions

Reimplemented in TrackAnalysisDefinitionSvc, CP::SelectionNameSvc, CP::SystematicsSvc, PlotsDefinitionSvc, asg::UnitTestService1, AthOnnx::OnnxRuntimeSvc, and egammaMVASvc.

Definition at line 32 of file AsgService.cxx.

  {
    return StatusCode::SUCCESS;
  }

msg() [1/2]

MsgStream& AthCommonMsg< Service >::msg ( ) const

inlineinherited

Definition at line 24 of file AthCommonMsg.h.

                                {
    return this->msgStream();
  }

msg() [2/2]

MsgStream& AthCommonMsg< Service >::msg ( const MSG::Level  lvl ) const

inlineinherited

Definition at line 27 of file AthCommonMsg.h.

                                                  {
    return this->msgStream(lvl);
  }

msgLvl()

bool AthCommonMsg< Service >::msgLvl ( const MSG::Level  lvl ) const

inlineinherited

Definition at line 30 of file AthCommonMsg.h.

                                               {
    return this->msgLevel(lvl);
  }

print()

void asg::AsgService::print ( ) const

virtualinherited

Print the state of the service.

Implements asg::IAsgService.

Definition at line 48 of file AsgService.cxx.

  {
    ATH_MSG_INFO( "AsgService " << name() << " @ " << this );
    return;
  }

queryInterface()

StatusCode asg::AsgService::queryInterface ( const InterfaceID &  riid, void **  ppvi  )

virtualinherited

query interface for gaudi

Definition at line 57 of file AsgService.cxx.

  {
    for (const auto& interface : m_interfaces)
    {
      if (riid == interface.first())
      {
        *ppvi = interface.second (this);
        addRef();
        return StatusCode::SUCCESS;
      }
    }
    return AsgServiceBase::queryInterface (riid, ppvi);
  }

Member Data Documentation

m_base_gnns

std::unordered_map<std::string, val_t> FlavorTagDiscriminants::NNSharingSvc::m_base_gnns

private

Definition at line 41 of file NNSharingSvc.h.

m_gnns

std::unordered_map<detail::NNKey, val_t, detail::NNHasher> FlavorTagDiscriminants::NNSharingSvc::m_gnns

private

Definition at line 40 of file NNSharingSvc.h.

m_interfaces

std::vector<std::pair<const InterfaceID& (*)(),void *(*)(AsgService*)> > asg::AsgService::m_interfaces

privateinherited

list of interfaces we have

Definition at line 68 of file AsgService.h.

---

The documentation for this class was generated from the following files:

• NNSharingSvc.h
• NNSharingSvc.cxx