TausRUsDataLoader Class Reference

Builds the fixed-size input tensors of the TausRUs network. More...

#include <TausRUsDataLoader.h>

Inheritance diagram for TausRUsDataLoader:

Collaboration diagram for TausRUsDataLoader:

Classes
struct	VariableConfig
struct	InputConfig
struct	Sequence
	One input node, its variables resolved to their functions. More...

Public Member Functions
	TausRUsDataLoader (const std::string &name)
StatusCode	initialize (const std::vector< InputConfig > &inputs)
	Resolve the variables of inputs to their functions.
AthInfer::InputDataMap	loadInputs (const xAOD::TauJet &tau, const xAOD::VertexContainer &vertices) const
std::vector< const xAOD::TauTrack * >	selectTracks (const xAOD::TauJet &tau) const
	The tracks and vertices the input tensors are built from, in slot order, for decoding the per-slot heads.
std::vector< const xAOD::Vertex * >	selectVertices (const xAOD::VertexContainer &vertices) const
size_t	maxTracks () const
size_t	maxVertices () const
void	setLevel (MSG::Level lvl)
	Change the current logging level.
Functions providing the same interface as AthMessaging
bool	msgLvl (const MSG::Level lvl) const
	Test the output level of the object.
MsgStream &	msg () const
	The standard message stream.
MsgStream &	msg (const MSG::Level lvl) const
	The standard message stream.

Private Types
using	ClusterFunc_t = std::function<bool(const xAOD::TauJet&, const xAOD::CaloVertexedTopoCluster&, float&)>
using	TrackFunc_t = std::function<bool(const xAOD::TauJet&, const xAOD::TauTrack&, float&)>
using	VertexFunc_t = std::function<bool(const xAOD::TauJet&, const xAOD::Vertex&, float&)>
using	ScalarFunc_t = std::function<bool(const xAOD::TauJet&, float&)>

Private Member Functions
template<class Func>
StatusCode	resolve (const InputConfig &input, const std::unordered_map< std::string, Func > &funcMap, Sequence< Func > &sequence) const
std::vector< const xAOD::CaloVertexedTopoCluster * >	selectClusters (const xAOD::TauJet &tau, std::vector< xAOD::CaloVertexedTopoCluster > &storage) const
void	initMessaging () const
	Initialize our message level and MessageSvc.

Private Attributes
Sequence< ClusterFunc_t >	m_clusters
Sequence< TrackFunc_t >	m_tracks
Sequence< VertexFunc_t >	m_vertices
Sequence< ScalarFunc_t >	m_scalars
std::string	m_nm
	Message source name.
boost::thread_specific_ptr< MsgStream >	m_msg_tls
	MsgStream instance (a std::cout like with print-out levels).
std::atomic< IMessageSvc * >	m_imsg { nullptr }
	MessageSvc pointer.
std::atomic< MSG::Level >	m_lvl { MSG::NIL }
	Current logging level.
std::atomic_flag m_initialized	ATLAS_THREAD_SAFE = ATOMIC_FLAG_INIT
	Messaging initialized (initMessaging).

Static Private Attributes
static const std::unordered_map< std::string, ClusterFunc_t >	m_clusterFuncs
static const std::unordered_map< std::string, TrackFunc_t >	m_trackFuncs
static const std::unordered_map< std::string, VertexFunc_t >	m_vertexFuncs
static const std::unordered_map< std::string, ScalarFunc_t >	m_scalarFuncs

Detailed Description

Builds the fixed-size input tensors of the TausRUs network.

The inputs are described by the model metadata: per input node, the collection it is built from, its truncation and its variables. Each variable name selects one function from the map of its collection, so a model with other inputs needs no code change unless it uses a new variable. The constituents are sorted and truncated, the remaining slots zero-padded.

Definition at line 67 of file TausRUsDataLoader.h.

Member Typedef Documentation

ClusterFunc_t

using TausRUsDataLoader::ClusterFunc_t = std::function<bool(const xAOD::TauJet&, const xAOD::CaloVertexedTopoCluster&, float&)>

private

Definition at line 99 of file TausRUsDataLoader.h.

ScalarFunc_t

using TausRUsDataLoader::ScalarFunc_t = std::function<bool(const xAOD::TauJet&, float&)>

private

Definition at line 102 of file TausRUsDataLoader.h.

TrackFunc_t

using TausRUsDataLoader::TrackFunc_t = std::function<bool(const xAOD::TauJet&, const xAOD::TauTrack&, float&)>

private

Definition at line 100 of file TausRUsDataLoader.h.

VertexFunc_t

using TausRUsDataLoader::VertexFunc_t = std::function<bool(const xAOD::TauJet&, const xAOD::Vertex&, float&)>

private

Definition at line 101 of file TausRUsDataLoader.h.

Constructor & Destructor Documentation

TausRUsDataLoader()

TausRUsDataLoader::TausRUsDataLoader ( const std::string & name )

explicit

Definition at line 139 of file TausRUsDataLoader.cxx.

  : asg::AsgMessaging(name) {}

Member Function Documentation

initialize()

StatusCode TausRUsDataLoader::initialize

(

const std::vector< InputConfig > &

inputs

)

Resolve the variables of inputs to their functions.

Definition at line 168 of file TausRUsDataLoader.cxx.

                                                                             {
  for (const InputConfig& input : inputs) {
    if (input.collection == "clusters") {
      ATH_CHECK(resolve(input, m_clusterFuncs, m_clusters));
    } else if (input.collection == "tracks") {
      ATH_CHECK(resolve(input, m_trackFuncs, m_tracks));
    } else if (input.collection == "vertices") {
      ATH_CHECK(resolve(input, m_vertexFuncs, m_vertices));
    } else if (input.collection == "seedjet") {
      ATH_CHECK(resolve(input, m_scalarFuncs, m_scalars));
    } else {
      ATH_MSG_ERROR("Input '" << input.name << "' is built from unknown collection '"
                    << input.collection << "'");
      return StatusCode::FAILURE;
    }
  }
  return StatusCode::SUCCESS;
}

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();
  // If user did not set an explicit level, set a default
  if (m_lvl == MSG::NIL) {
    m_lvl = m_imsg ?
      static_cast<MSG::Level>( m_imsg.load()->outputLevel(m_nm) ) :
      MSG::INFO;
  }
}

loadInputs()

AthInfer::InputDataMap TausRUsDataLoader::loadInputs	(	const xAOD::TauJet &	tau,
		const xAOD::VertexContainer &	vertices ) const

Definition at line 233 of file TausRUsDataLoader.cxx.

                                                                         {
  AthInfer::InputDataMap inputData;
 
  auto add = [&inputData](const auto& sequence, std::vector<float> tensor, bool isScalar) {
    const auto nVariables = static_cast<int64_t>(sequence.funcs.size());
    std::vector<int64_t> shape = isScalar
      ? std::vector<int64_t>{1, nVariables}
      : std::vector<int64_t>{1, static_cast<int64_t>(sequence.maxObjects), nVariables};
    inputData[sequence.name] = std::make_pair(std::move(shape), std::move(tensor));
  };
 
  if (!m_clusters.name.empty()) {
    std::vector<xAOD::CaloVertexedTopoCluster> storage;
    add(m_clusters, fillTensor(m_clusters, tau, selectClusters(tau, storage)), false);
  }
  if (!m_tracks.name.empty()) {
    add(m_tracks, fillTensor(m_tracks, tau, selectTracks(tau)), false);
  }
  if (!m_vertices.name.empty()) {
    add(m_vertices, fillTensor(m_vertices, tau, selectVertices(vertices)), false);
  }
  if (!m_scalars.name.empty()) {
    std::vector<float> tensor(m_scalars.funcs.size(), 0.f);
    for (size_t iVar = 0; iVar < m_scalars.funcs.size(); ++iVar) {
      float value = 0.f;
      if (m_scalars.funcs[iVar](tau, value)) {
        tensor[iVar] = (value + m_scalars.offsets[iVar]) * m_scalars.scales[iVar];
      }
    }
    add(m_scalars, std::move(tensor), true);
  }
  return inputData;
}

maxTracks()

size_t TausRUsDataLoader::maxTracks ( ) const

inline

Definition at line 95 of file TausRUsDataLoader.h.

{ return m_tracks.maxObjects; }

maxVertices()

size_t TausRUsDataLoader::maxVertices ( ) const

inline

Definition at line 96 of file TausRUsDataLoader.h.

{ return m_vertices.maxObjects; }

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
   }

resolve()

template<class Func>

StatusCode TausRUsDataLoader::resolve ( const InputConfig & input, const std::unordered_map< std::string, Func > & funcMap, Sequence< Func > & sequence ) const

private

Definition at line 142 of file TausRUsDataLoader.cxx.

                                                                      {
  if (!sequence.name.empty()) {
    ATH_MSG_ERROR("Inputs '" << sequence.name << "' and '" << input.name
                  << "' are both built from " << input.collection);
    return StatusCode::FAILURE;
  }
  sequence.name = input.name;
  sequence.maxObjects = input.maxObjects;
  for (const VariableConfig& variable : input.variables) {
    const auto func = funcMap.find(variable.name);
    if (func == funcMap.end()) {
      ATH_MSG_ERROR("Variable '" << variable.name << "' of input '" << input.name
                    << "' is not defined for " << input.collection);
      return StatusCode::FAILURE;
    }
    sequence.funcs.push_back(func->second);
    sequence.offsets.push_back(variable.offset);
    sequence.scales.push_back(variable.scale);
  }
  ATH_MSG_INFO("TausRUs input '" << input.name << "' (" << input.collection << "): "
               << input.maxObjects << " objects x " << input.variables.size() << " variables");
  return StatusCode::SUCCESS;
}

selectClusters()

std::vector< const xAOD::CaloVertexedTopoCluster * > TausRUsDataLoader::selectClusters ( const xAOD::TauJet & tau, std::vector< xAOD::CaloVertexedTopoCluster > & storage ) const

private

Definition at line 187 of file TausRUsDataLoader.cxx.

                                                           {
  storage = tau.vertexedClusters();
 
  std::vector<const xAOD::CaloVertexedTopoCluster*> selected;
  selected.reserve(storage.size());
  for (const xAOD::CaloVertexedTopoCluster& cluster : storage) {
    selected.push_back(&cluster);
  }
  std::sort(selected.begin(), selected.end(),
            [](const xAOD::CaloVertexedTopoCluster* lhs,
               const xAOD::CaloVertexedTopoCluster* rhs) {
              return lhs->clust().e() > rhs->clust().e();
            });
  if (selected.size() > m_clusters.maxObjects) selected.resize(m_clusters.maxObjects);
  return selected;
}

selectTracks()

std::vector< const xAOD::TauTrack * > TausRUsDataLoader::selectTracks

(

const xAOD::TauJet &

tau

)

const

The tracks and vertices the input tensors are built from, in slot order, for decoding the per-slot heads.

Definition at line 206 of file TausRUsDataLoader.cxx.

                                                                                          {
  std::vector<const xAOD::TauTrack*> tracks = tau.allTracks();
  std::erase_if(tracks, [](const xAOD::TauTrack* track) {
    return track == nullptr || track->track() == nullptr;
  });
  std::sort(tracks.begin(), tracks.end(),
            [](const xAOD::TauTrack* lhs, const xAOD::TauTrack* rhs) {
              return lhs->track()->pt() > rhs->track()->pt();
            });
  if (tracks.size() > m_tracks.maxObjects) tracks.resize(m_tracks.maxObjects);
  return tracks;
}

selectVertices()

std::vector< const xAOD::Vertex * > TausRUsDataLoader::selectVertices

(

const xAOD::VertexContainer &

vertices

)

const

Definition at line 219 of file TausRUsDataLoader.cxx.

                                               {
  std::vector<const xAOD::Vertex*> selected;
  selected.reserve(std::min(vertices.size(), m_vertices.maxObjects));
  for (const xAOD::Vertex* vertex : vertices) {
    if (selected.size() == m_vertices.maxObjects) break;
    if (!vertex) continue;
    const int type = vertex->vertexType();
    if (type == 0 || type == -99) continue;
    selected.push_back(vertex);
  }
  return selected;
}

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_clusterFuncs

const std::unordered_map<std::string, ClusterFunc_t> TausRUsDataLoader::m_clusterFuncs

inlinestaticprivate

Initial value:

                                                      = {
  {"dEtaRaw",        TausRUsClusterVars::dEtaRaw},
  {"dPhiRaw",        TausRUsClusterVars::dPhiRaw},
  {"etaRaw",         TausRUsClusterVars::etaRaw},
  {"phiRaw",         TausRUsClusterVars::phiRaw},
  {"log_etRaw",      TausRUsClusterVars::log_etRaw},
  {"log_eRaw",       TausRUsClusterVars::log_eRaw},
  {"FIRST_ENG_DENS", TauClusterVars::FIRST_ENG_DENS},
  {"SECOND_R",       TauClusterVars::SECOND_R},
  {"EM_PROBABILITY", TauClusterVars::EM_PROBABILITY},
  {"SECOND_LAMBDA",  TauClusterVars::SECOND_LAMBDA},
  {"CENTER_LAMBDA",  TauClusterVars::CENTER_LAMBDA},
  {"CENTER_MAG",     TauClusterVars::CENTER_MAG},
}

Definition at line 128 of file TausRUsDataLoader.h.

                                                                                {
    {"dEtaRaw",        TausRUsClusterVars::dEtaRaw},
    {"dPhiRaw",        TausRUsClusterVars::dPhiRaw},
    {"etaRaw",         TausRUsClusterVars::etaRaw},
    {"phiRaw",         TausRUsClusterVars::phiRaw},
    {"log_etRaw",      TausRUsClusterVars::log_etRaw},
    {"log_eRaw",       TausRUsClusterVars::log_eRaw},
    {"FIRST_ENG_DENS", TauClusterVars::FIRST_ENG_DENS},
    {"SECOND_R",       TauClusterVars::SECOND_R},
    {"EM_PROBABILITY", TauClusterVars::EM_PROBABILITY},
    {"SECOND_LAMBDA",  TauClusterVars::SECOND_LAMBDA},
    {"CENTER_LAMBDA",  TauClusterVars::CENTER_LAMBDA},
    {"CENTER_MAG",     TauClusterVars::CENTER_MAG},
  };

m_clusters

Sequence<ClusterFunc_t> TausRUsDataLoader::m_clusters

private

Definition at line 123 of file TausRUsDataLoader.h.

m_imsg

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

mutableprivateinherited

MessageSvc pointer.

Definition at line 135 of file AthMessaging.h.

{ nullptr };

m_lvl

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

mutableprivateinherited

Current logging level.

Definition at line 138 of file AthMessaging.h.

{ MSG::NIL };

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_scalarFuncs

const std::unordered_map<std::string, ScalarFunc_t> TausRUsDataLoader::m_scalarFuncs

inlinestaticprivate

Initial value:

                                                     = {
  {"ptJetSeed",  TauScalarVars::ptJetSeed},
  {"etaJetSeed", TauScalarVars::etaJetSeed},
  {"phiJetSeed", TauScalarVars::phiJetSeed},
  {"mJetSeed",   TauScalarVars::mJetSeed},
}

Definition at line 170 of file TausRUsDataLoader.h.

                                                                              {
    {"ptJetSeed",  TauScalarVars::ptJetSeed},
    {"etaJetSeed", TauScalarVars::etaJetSeed},
    {"phiJetSeed", TauScalarVars::phiJetSeed},
    {"mJetSeed",   TauScalarVars::mJetSeed},
  };

m_scalars

Sequence<ScalarFunc_t> TausRUsDataLoader::m_scalars

private

Definition at line 126 of file TausRUsDataLoader.h.

m_trackFuncs

const std::unordered_map<std::string, TrackFunc_t> TausRUsDataLoader::m_trackFuncs

inlinestaticprivate

Initial value:

                                                    = {
  {"dEta",                            TauTrackVars::dEta},
  {"dPhi",                            TauTrackVars::dPhi},
  {"log_pt",                          TausRUsTrackVars::log_pt},
  {"log_e",                           TausRUsTrackVars::log_e},
  {"trackEta",                        TauTrackVars::trackEta},
  {"trackPhi",                        TauTrackVars::trackPhi},
  {"charge",                          TauTrackVars::charge},
  {"qOverP",                          TauTrackVars::qOverP},
  {"d0_old",                          TauTrackVars::d0_old},
  {"z0",                              TausRUsTrackVars::z0},
  {"z0sinthetaTJVA",                  TauTrackVars::z0sinthetaTJVA},
  {"numberOfTRTHits",                 TauTrackVars::numberOfTRTHits},
  {"numberOfTRTHighThresholdHits",    TauTrackVars::numberOfTRTHighThresholdHits},
  {"numberOfSCTHits",                 TauTrackVars::numberOfSCTHits},
  {"numberOfPixelHits",               TauTrackVars::numberOfPixelHits},
  {"numberOfInnermostPixelLayerHits", TauTrackVars::numberOfInnermostPixelLayerHits},
  {"eProbabilityNN_trackParticle",    TausRUsTrackVars::eProbabilityNN_trackParticle},
}

Definition at line 143 of file TausRUsDataLoader.h.

                                                                            {
    {"dEta",                            TauTrackVars::dEta},
    {"dPhi",                            TauTrackVars::dPhi},
    {"log_pt",                          TausRUsTrackVars::log_pt},
    {"log_e",                           TausRUsTrackVars::log_e},
    {"trackEta",                        TauTrackVars::trackEta},
    {"trackPhi",                        TauTrackVars::trackPhi},
    {"charge",                          TauTrackVars::charge},
    {"qOverP",                          TauTrackVars::qOverP},
    {"d0_old",                          TauTrackVars::d0_old},
    {"z0",                              TausRUsTrackVars::z0},
    {"z0sinthetaTJVA",                  TauTrackVars::z0sinthetaTJVA},
    {"numberOfTRTHits",                 TauTrackVars::numberOfTRTHits},
    {"numberOfTRTHighThresholdHits",    TauTrackVars::numberOfTRTHighThresholdHits},
    {"numberOfSCTHits",                 TauTrackVars::numberOfSCTHits},
    {"numberOfPixelHits",               TauTrackVars::numberOfPixelHits},
    {"numberOfInnermostPixelLayerHits", TauTrackVars::numberOfInnermostPixelLayerHits},
    {"eProbabilityNN_trackParticle",    TausRUsTrackVars::eProbabilityNN_trackParticle},
  };

m_tracks

Sequence<TrackFunc_t> TausRUsDataLoader::m_tracks

private

Definition at line 124 of file TausRUsDataLoader.h.

m_vertexFuncs

const std::unordered_map<std::string, VertexFunc_t> TausRUsDataLoader::m_vertexFuncs

inlinestaticprivate

Initial value:

                                                     = {
  {"sumPt2", TausRUsVertexVars::sumPt2},
  {"x",      TausRUsVertexVars::x},
  {"y",      TausRUsVertexVars::y},
  {"z",      TausRUsVertexVars::z},
}

Definition at line 163 of file TausRUsDataLoader.h.

                                                                              {
    {"sumPt2", TausRUsVertexVars::sumPt2},
    {"x",      TausRUsVertexVars::x},
    {"y",      TausRUsVertexVars::y},
    {"z",      TausRUsVertexVars::z},
  };

m_vertices

Sequence<VertexFunc_t> TausRUsDataLoader::m_vertices

private

Definition at line 125 of file TausRUsDataLoader.h.

---

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

• TausRUsDataLoader.h
• TausRUsDataLoader.cxx