TauGNNUtils::GNNVarCalc Class Reference

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 = std::function< bool(const xAOD::TauJet &, float &)>
using	TrackCalc = std::function< bool(const xAOD::TauJet &, const xAOD::TauTrack &, float &)>
using	ClusterCalc = std::function< bool(const xAOD::TauJet &, const xAOD::CaloVertexedTopoCluster &, float &)>

Public Member Functions
	GNNVarCalc (bool useTRT)
	~GNNVarCalc ()=default
float	compute (const std::string &name, const xAOD::TauJet &tau) const
std::vector< float >	compute (const std::string &name, const xAOD::TauJet &tau, const std::vector< const xAOD::TauTrack * > &tracks) const
std::vector< float >	compute (const std::string &name, const xAOD::TauJet &tau, const std::vector< xAOD::CaloVertexedTopoCluster > &clusters) const
void	setLevel (MSG::Level lvl)
	Change the current logging level. More...

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

Static Private Member Functions
static void	initialize_map (bool useTRT)

Private Attributes
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...

Static Private Attributes
static std::once_flag	m_init_flag
static const std::unordered_map< std::string, ScalarCalc >	m_scalar_map
static std::unordered_map< std::string, TrackCalc > m_track_map	ATLAS_THREAD_SAFE
static const std::unordered_map< std::string, ClusterCalc >	m_cluster_map

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

D. Qichen

Definition at line 275 of file TauGNNUtils.h.

Member Typedef Documentation

ClusterCalc

using TauGNNUtils::GNNVarCalc::ClusterCalc = std::function<bool(const xAOD::TauJet &, const xAOD::CaloVertexedTopoCluster &, float &)>

Definition at line 280 of file TauGNNUtils.h.

ScalarCalc

using TauGNNUtils::GNNVarCalc::ScalarCalc = std::function<bool(const xAOD::TauJet &, float &)>

Definition at line 278 of file TauGNNUtils.h.

TrackCalc

using TauGNNUtils::GNNVarCalc::TrackCalc = std::function<bool(const xAOD::TauJet &, const xAOD::TauTrack &, float &)>

Definition at line 279 of file TauGNNUtils.h.

Constructor & Destructor Documentation

GNNVarCalc()

TauGNNUtils::GNNVarCalc::GNNVarCalc

(

bool

useTRT

)

Definition at line 14 of file TauGNNUtils.cxx.

                                  :
    asg::AsgMessaging("TauGNNUtils::GNNVarCalc") {
    initialize_map(useTRT);
}

~GNNVarCalc()

TauGNNUtils::GNNVarCalc::~GNNVarCalc ( )

default

Member Function Documentation

compute() [1/3]

float TauGNNUtils::GNNVarCalc::compute	(	const std::string &	name,
		const xAOD::TauJet &	tau
	)		const

Definition at line 25 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
    float out;
    bool success = func(tau, out);
    if (!success) {
        ATH_MSG_ERROR("Error in scalar variable calculation");
        throw std::runtime_error("Error in scalar variable calculation");
    }
    return out;
}

compute() [2/3]

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

Definition at line 45 of file TauGNNUtils.cxx.

                                                                           {
    std::vector<float> out;
    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 std::runtime_error("Variable '" + name + "' not defined");
    }
 
    // Calculate variables for selected tracks
    bool success = true;
    float value;
    for (const auto *const trk : tracks) {
        success = success && func(tau, *trk, value);
        out.push_back(value);
    }
    if (!success) {
        ATH_MSG_ERROR("Error in track variable calculation");
        throw std::runtime_error("Error in track variable calculation");
    }
    return out;
}

compute() [3/3]

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

Definition at line 73 of file TauGNNUtils.cxx.

                                                                                    {
    std::vector<float> out;
    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;
    float value;
    for (const xAOD::CaloVertexedTopoCluster& cluster : clusters) {
        success = success && func(tau, cluster, value);
        out.push_back(value);
    }
    if (!success) {
        ATH_MSG_ERROR("Error in cluster variable calculation");
        throw std::runtime_error("Error in cluster variable calculation");
    }
    return out;
}

initialize_map()

void TauGNNUtils::GNNVarCalc::initialize_map ( bool  useTRT )

staticprivate

Definition at line 19 of file TauGNNUtils.cxx.

                                           {
    std::call_once(m_init_flag, [useTRT]() {
        if(!useTRT) m_track_map["eProbabilityHT"] = Variables::Track::eProbabilityHT_noTRT;
    });
}

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

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::unordered_map<std::string, TrackCalc> m_track_map TauGNNUtils::GNNVarCalc::ATLAS_THREAD_SAFE

inlinestaticprivate

Initial value:

= {
        {"pt_log",                    Variables::Track::pt_log},
        {"trackPt",                   Variables::Track::trackPt},
        {"trackEta",                  Variables::Track::trackEta},
        {"trackPhi",                  Variables::Track::trackPhi},
        {"pt_tau_log",                Variables::Track::pt_tau_log},
        {"pt_jetseed_log",            Variables::Track::pt_jetseed_log},
        {"d0_abs_log",                Variables::Track::d0_abs_log},
        {"z0sinThetaTJVA_abs_log",    Variables::Track::z0sinThetaTJVA_abs_log},
        {"z0sinthetaTJVA",            Variables::Track::z0sinthetaTJVA},
        {"z0sinthetaSigTJVA",         Variables::Track::z0sinthetaSigTJVA},
        {"d0TJVA",                    Variables::Track::d0TJVA},
        {"d0SigTJVA",                 Variables::Track::d0SigTJVA},
        {"dEta",                      Variables::Track::dEta},
        {"dEtaJetSeedAxis",           Variables::Track::dEtaJetSeedAxis},
        {"dPhi",                      Variables::Track::dPhi},
        {"dPhiJetSeedAxis",           Variables::Track::dPhiJetSeedAxis},
        {"nInnermostPixelHits",       Variables::Track::nInnermostPixelHits},
        {"numberOfInnermostPixelLayerHits", Variables::Track::numberOfInnermostPixelLayerHits},
        {"nPixelHits",                Variables::Track::nPixelHits},
        {"nSCTHits",                  Variables::Track::nSCTHits},
        {"nIBLHitsAndExp",            Variables::Track::nIBLHitsAndExp},
        {"nPixelHitsPlusDeadSensors", Variables::Track::nPixelHitsPlusDeadSensors},
        {"nSCTHitsPlusDeadSensors",   Variables::Track::nSCTHitsPlusDeadSensors},
        {"eProbabilityHT",            Variables::Track::eProbabilityHT}
    }

Definition at line 321 of file TauGNNUtils.h.

m_cluster_map

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

inlinestaticprivate

Initial value:

= {
        {"dEta",                      Variables::Cluster::dEta},
        {"dPhi",                      Variables::Cluster::dPhi},
        {"SECOND_R",                  Variables::Cluster::SECOND_R},
        {"SECOND_LAMBDA",             Variables::Cluster::SECOND_LAMBDA},
        {"CENTER_LAMBDA",             Variables::Cluster::CENTER_LAMBDA},
        {"et",                        Variables::Cluster::et}
    }

Definition at line 348 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_init_flag

std::once_flag TauGNNUtils::GNNVarCalc::m_init_flag

inlinestaticprivate

Definition at line 301 of file TauGNNUtils.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

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

inlinestaticprivate

Initial value:

= {
        {"isolFrac",                  Variables::Scalar::isolFrac},
        {"centFrac",                  Variables::Scalar::centFrac},
        {"etOverPtLeadTrk",           Variables::Scalar::etOverPtLeadTrk},
        {"innerTrkAvgDist",           Variables::Scalar::innerTrkAvgDist},
        {"absipSigLeadTrk",           Variables::Scalar::absipSigLeadTrk},
        {"SumPtTrkFrac",              Variables::Scalar::SumPtTrkFrac},
        {"sumEMCellEtOverLeadTrkPt",  Variables::Scalar::sumEMCellEtOverLeadTrkPt},
        {"EMPOverTrkSysP",            Variables::Scalar::EMPOverTrkSysP},
        {"ptRatioEflowApprox",        Variables::Scalar::ptRatioEflowApprox},
        {"mEflowApprox",              Variables::Scalar::mEflowApprox},
        {"dRmax",                     Variables::Scalar::dRmax},
        {"trFlightPathSig",           Variables::Scalar::trFlightPathSig},
        {"massTrkSys",                Variables::Scalar::massTrkSys},
        {"pt",                        Variables::Scalar::pt}
    }

Definition at line 304 of file TauGNNUtils.h.

---

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

• TauGNNUtils.h
• TauGNNUtils.cxx