Tool to calculate input variables for the GNN-based tau identification. More...

#include <TauGNNUtils.h>

Inheritance diagram for TauGNNUtils::GNNVarCalc:

Collaboration diagram for TauGNNUtils::GNNVarCalc:

Public Types
using	ScalarCalc = bool(*)(const xAOD::TauJet &, double &)

using	TrackCalc = bool(*)(const xAOD::TauJet &, const xAOD::TauTrack &, double &)

using	ClusterCalc = bool(*)(const xAOD::TauJet &, const xAOD::CaloVertexedTopoCluster &, double &)

Public Member Functions
	GNNVarCalc ()

	~GNNVarCalc ()=default

bool	compute (const std::string &name, const xAOD::TauJet &tau, double &out) const

bool	compute (const std::string &name, const xAOD::TauJet &tau, const std::vector< const xAOD::TauTrack * > &tracks, std::vector< double > &out) const

bool	compute (const std::string &name, const xAOD::TauJet &tau, const std::vector< xAOD::CaloVertexedTopoCluster > &clusters, std::vector< double > &out) const

void	insert (const std::string &name, ScalarCalc func, const std::vector< std::string > &scalar_vars)

void	insert (const std::string &name, TrackCalc func, const std::vector< std::string > &track_vars)

void	insert (const std::string &name, ClusterCalc func, const std::vector< std::string > &cluster_vars)

void	setLevel (MSG::Level lvl)
	Change the current logging level. More...

Private Member Functions
void	initMessaging () const
	Initialize our message level and MessageSvc. More...

Private Attributes
std::unordered_map< std::string, ScalarCalc >	m_scalar_map

std::unordered_map< std::string, TrackCalc >	m_track_map

std::unordered_map< std::string, ClusterCalc >	m_cluster_map

std::string	m_nm
	Message source name. More...

boost::thread_specific_ptr< MsgStream >	m_msg_tls
	MsgStream instance (a std::cout like with print-out levels) More...

std::atomic< IMessageSvc * >	m_imsg { nullptr }
	MessageSvc pointer. More...

std::atomic< MSG::Level >	m_lvl { MSG::NIL }
	Current logging level. More...

std::atomic_flag m_initialized	ATLAS_THREAD_SAFE = ATOMIC_FLAG_INIT
	Messaging initialized (initMessaging) More...

Detailed Description

Tool to calculate input variables for the GNN-based tau identification.

Used to calculate input variables for (onnx)GNN-based tau identification on the fly by providing a mapping between variable names (strings) and functions to calculate these variables.

Author: C. Deutsch; W. Davey; N.M. Tamir

Definition at line 31 of file TauGNNUtils.h.

Member Typedef Documentation

◆ ClusterCalc

using TauGNNUtils::GNNVarCalc::ClusterCalc = bool (*)(const xAOD::TauJet &, const xAOD::CaloVertexedTopoCluster &, double &)

Definition at line 39 of file TauGNNUtils.h.

◆ ScalarCalc

using TauGNNUtils::GNNVarCalc::ScalarCalc = bool (*)(const xAOD::TauJet &, double &)

Definition at line 34 of file TauGNNUtils.h.

◆ TrackCalc

using TauGNNUtils::GNNVarCalc::TrackCalc = bool (*)(const xAOD::TauJet &, const xAOD::TauTrack &, double &)

Definition at line 36 of file TauGNNUtils.h.

Constructor & Destructor Documentation

◆ GNNVarCalc()

TauGNNUtils::GNNVarCalc::GNNVarCalc ( )

Definition at line 14 of file TauGNNUtils.cxx.

14 : asg::AsgMessaging("TauGNNUtils::GNNVarCalc") {

15 }

◆ ~GNNVarCalc()

TauGNNUtils::GNNVarCalc::~GNNVarCalc ( )

default

Member Function Documentation

◆ compute() [1/3]

bool TauGNNUtils::GNNVarCalc::compute	(	const std::string &	name,
		const xAOD::TauJet &	tau,
		const std::vector< const xAOD::TauTrack * > &	tracks,
		std::vector< double > &	out
	)		const

Definition at line 32 of file TauGNNUtils.cxx.

                                                     {
     out.clear();
     out.reserve(tracks.size());
  
     // Retrieve calculator function
     TrackCalc func = nullptr;
     try {
         func = m_track_map.at(name);
     } catch (const std::out_of_range &e) {
         ATH_MSG_ERROR("Variable '" << name << "' not defined");
         throw;
     }
  
     // Calculate variables for selected tracks
     bool success = true;
     double value;
     for (const auto *const trk : tracks) {
         success = success && func(tau, *trk, value);
         out.push_back(value);
     }
  
     return success;
 }

◆ compute() [2/3]

bool TauGNNUtils::GNNVarCalc::compute	(	const std::string &	name,
		const xAOD::TauJet &	tau,
		const std::vector< xAOD::CaloVertexedTopoCluster > &	clusters,
		std::vector< double > &	out
	)		const

Definition at line 58 of file TauGNNUtils.cxx.

                                                     {
     out.clear();
     out.reserve(clusters.size());
  
     // Retrieve calculator function
     ClusterCalc func = nullptr;
     try {
         func = m_cluster_map.at(name);
     } catch (const std::out_of_range &e) {
         ATH_MSG_ERROR("Variable '" << name << "' not defined");
         throw;
     }
  
     // Calculate variables for selected clusters
     bool success = true;
     double value;
     for (const xAOD::CaloVertexedTopoCluster& cluster : clusters) {
         success = success && func(tau, cluster, value);
         out.push_back(value);
     }
  
     return success;
 }

◆ compute() [3/3]

bool TauGNNUtils::GNNVarCalc::compute	(	const std::string &	name,
		const xAOD::TauJet &	tau,
		double &	out
	)		const

Definition at line 17 of file TauGNNUtils.cxx.

                                          {
     // Retrieve calculator function
     ScalarCalc func = nullptr;
     try {
         func = m_scalar_map.at(name);
     } catch (const std::out_of_range &e) {
         ATH_MSG_ERROR("Variable '" << name << "' not defined");
         throw;
     }
  
     // Calculate variable
     return func(tau, out);
 }

◆ initMessaging()

void AthMessaging::initMessaging ( ) const

privateinherited

Initialize our message level and MessageSvc.

This method should only be called once.

Definition at line 39 of file AthMessaging.cxx.

 {
   m_imsg = Athena::getMessageSvc();
   m_lvl = m_imsg ?
     static_cast<MSG::Level>( m_imsg.load()->outputLevel(m_nm) ) :
     MSG::INFO;
 }

◆ insert() [1/3]

void TauGNNUtils::GNNVarCalc::insert	(	const std::string &	name,
		ClusterCalc	func,
		const std::vector< std::string > &	cluster_vars
	)

Definition at line 104 of file TauGNNUtils.cxx.

                                                                                                          {
     if (std::find(cluster_vars.begin(), cluster_vars.end(), name) == cluster_vars.end()) {
       return;
     }
     if (!func) {
         throw std::invalid_argument("Nullptr passed to GNNVarCalc::insert");
     }
     m_cluster_map[name] = func;
 }

◆ insert() [2/3]

void TauGNNUtils::GNNVarCalc::insert	(	const std::string &	name,
		ScalarCalc	func,
		const std::vector< std::string > &	scalar_vars
	)

Definition at line 84 of file TauGNNUtils.cxx.

                                                                                                        {
     if (std::find(scalar_vars.begin(), scalar_vars.end(), name) == scalar_vars.end()) {
       return;
     }
     if (!func) {
         throw std::invalid_argument("Nullptr passed to GNNVarCalc::insert");
     }
     m_scalar_map[name] = func;
 }

◆ insert() [3/3]

void TauGNNUtils::GNNVarCalc::insert	(	const std::string &	name,
		TrackCalc	func,
		const std::vector< std::string > &	track_vars
	)

Definition at line 94 of file TauGNNUtils.cxx.

                                                                                                      {
     if (std::find(track_vars.begin(), track_vars.end(), name) == track_vars.end()) {
       return;
     }
     if (!func) {
         throw std::invalid_argument("Nullptr passed to GNNVarCalc::insert");
     }
     m_track_map[name] = func;
 }

◆ msg() [1/2]

MsgStream & asg::AsgMessaging::msg ( ) const

inherited

The standard message stream.

Returns: A reference to the default message stream of this object.

Definition at line 49 of file AsgMessaging.cxx.

                                       {
 #ifndef XAOD_STANDALONE
       return ::AthMessaging::msg();
 #else // not XAOD_STANDALONE
       return m_msg;
 #endif // not XAOD_STANDALONE
    }

◆ msg() [2/2]

MsgStream & asg::AsgMessaging::msg ( const MSG::Level lvl ) const

inherited

The standard message stream.

Parameters

lvl	The message level to set the stream to

Returns: A reference to the default message stream, set to level "lvl"

Definition at line 57 of file AsgMessaging.cxx.

                                                           {
 #ifndef XAOD_STANDALONE
       return ::AthMessaging::msg( lvl );
 #else // not XAOD_STANDALONE
       m_msg << lvl;
       return m_msg;
 #endif // not XAOD_STANDALONE
    }

◆ msgLvl()

bool asg::AsgMessaging::msgLvl ( const MSG::Level lvl ) const

inherited

Test the output level of the object.

Parameters

lvl	The message level to test against

Returns: boolean Indicting if messages at given level will be printed; true If messages at level "lvl" will be printed

Definition at line 41 of file AsgMessaging.cxx.

                                                        {
 #ifndef XAOD_STANDALONE
       return ::AthMessaging::msgLvl( lvl );
 #else // not XAOD_STANDALONE
       return m_msg.msgLevel( lvl );
 #endif // not XAOD_STANDALONE
    }

◆ setLevel()

void AthMessaging::setLevel ( MSG::Level lvl )

inherited

Change the current logging level.

Use this rather than msg().setLevel() for proper operation with MT.

Definition at line 28 of file AthMessaging.cxx.

 {
   m_lvl = lvl;
 }

Member Data Documentation

◆ ATLAS_THREAD_SAFE

std::atomic_flag m_initialized AthMessaging::ATLAS_THREAD_SAFE = ATOMIC_FLAG_INIT

mutableprivateinherited

Messaging initialized (initMessaging)

Definition at line 141 of file AthMessaging.h.

◆ m_cluster_map

std::unordered_map<std::string, ClusterCalc> TauGNNUtils::GNNVarCalc::m_cluster_map

private

Definition at line 70 of file TauGNNUtils.h.

◆ m_imsg

std::atomic<IMessageSvc*> AthMessaging::m_imsg { nullptr }

mutableprivateinherited

MessageSvc pointer.

Definition at line 135 of file AthMessaging.h.

◆ m_lvl

std::atomic<MSG::Level> AthMessaging::m_lvl { MSG::NIL }

mutableprivateinherited

Current logging level.

Definition at line 138 of file AthMessaging.h.

◆ m_msg_tls

boost::thread_specific_ptr<MsgStream> AthMessaging::m_msg_tls

mutableprivateinherited

MsgStream instance (a std::cout like with print-out levels)

Definition at line 132 of file AthMessaging.h.

◆ m_nm

std::string AthMessaging::m_nm

privateinherited

Message source name.

Definition at line 129 of file AthMessaging.h.

◆ m_scalar_map

std::unordered_map<std::string, ScalarCalc> TauGNNUtils::GNNVarCalc::m_scalar_map

private

Definition at line 68 of file TauGNNUtils.h.

◆ m_track_map

std::unordered_map<std::string, TrackCalc> TauGNNUtils::GNNVarCalc::m_track_map

private

Definition at line 69 of file TauGNNUtils.h.

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

Public Types

Public Member Functions

Private Member Functions

Private Attributes

Detailed Description

Member Typedef Documentation

◆ ClusterCalc

◆ ScalarCalc

◆ TrackCalc

Constructor & Destructor Documentation

◆ GNNVarCalc()

◆ ~GNNVarCalc()

Member Function Documentation

◆ compute() [1/3]

◆ compute() [2/3]

◆ compute() [3/3]

◆ initMessaging()

◆ insert() [1/3]

◆ insert() [2/3]

◆ insert() [3/3]

◆ msg() [1/2]

◆ msg() [2/2]

◆ msgLvl()

◆ setLevel()

Member Data Documentation

◆ ATLAS_THREAD_SAFE

◆ m_cluster_map

◆ m_imsg

◆ m_lvl

◆ m_msg_tls

◆ m_nm

◆ m_scalar_map

◆ m_track_map