TFCSMLCalorimeterSimulator Class Reference

#include <TFCSMLCalorimeterSimulator.h>

Inheritance diagram for TFCSMLCalorimeterSimulator:

Collaboration diagram for TFCSMLCalorimeterSimulator:

Classes
struct	event_t
struct	layer_t

Public Member Functions
	TFCSMLCalorimeterSimulator ()
virtual	~TFCSMLCalorimeterSimulator ()
bool	loadSimulator (std::string &filename)
void	Print () const
VNetworkBase::NetworkOutputs	predictVoxels (TFCSSimulationState &simulstate, float eta, float energy) const
event_t	getEvent (TFCSSimulationState &simulstate, float eta, float energy) const
VNetworkBase::NetworkOutputs	predictVoxels () const
void	setInputShapes (std::vector< long unsigned int > &layer_boundaries, std::vector< long unsigned int > &used_layers)
bool	msgLvl (const MSG::Level lvl) const
	Check whether the logging system is active at the provided verbosity level. More...
MsgStream &	msg () const
	Return a stream for sending messages directly (no decoration) More...
MsgStream &	msg (const MSG::Level lvl) const
	Return a decorated starting stream for sending messages. More...
MSG::Level	level () const
	Retrieve output level. More...
virtual void	setLevel (MSG::Level lvl)
	Update outputlevel. More...

Static Public Member Functions
static std::string	startMsg (MSG::Level lvl, const std::string &file, int line)
	Make a message to decorate the start of logging. More...

Private Attributes
std::unique_ptr< VNetworkBase >	m_onnx_model = nullptr
int	m_nEvents = 1
std::vector< long unsigned int >	m_layer_boundaries = {0, 36, 200, 310, 346, 382}
std::vector< long unsigned int >	m_used_layers = {0, 1, 2, 3, 12}
long unsigned int	m_nVoxels = 382
long unsigned int	m_nLayers = 5
std::string	m_nm
	Message source name. More...

Static Private Attributes
static boost::thread_specific_ptr< MsgStream > m_msg_tls	ATLAS_THREAD_SAFE
	Do not persistify! More...

Detailed Description

Definition at line 23 of file TFCSMLCalorimeterSimulator.h.

Constructor & Destructor Documentation

TFCSMLCalorimeterSimulator()

TFCSMLCalorimeterSimulator::TFCSMLCalorimeterSimulator

(

)

Definition at line 12 of file TFCSMLCalorimeterSimulator.cxx.

{}

~TFCSMLCalorimeterSimulator()

TFCSMLCalorimeterSimulator::~TFCSMLCalorimeterSimulator ( )

virtual

Definition at line 14 of file TFCSMLCalorimeterSimulator.cxx.

{}

Member Function Documentation

getEvent()

TFCSMLCalorimeterSimulator::event_t TFCSMLCalorimeterSimulator::getEvent	(	TFCSSimulationState &	simulstate,
		float	eta,
		float	energy
	)		const

Definition at line 85 of file TFCSMLCalorimeterSimulator.cxx.

                                                                                                                                     {
 
  // Get the voxel energies
  VNetworkBase::NetworkOutputs outputs = predictVoxels(simulstate, eta, energy);
 
  // check if the output contains a nan
  // If yes: retry up to 5 times
  float first_output = outputs.begin()->second;
  bool contains_nan = std::isnan(first_output);
  if (contains_nan) {
    int retry = 0;
    while (contains_nan) {
 
      if (retry > 5) {
        ATH_MSG_WARNING("Network output contains NaN. Giving up.");
        break;
      }
 
      ATH_MSG_WARNING("Network output contains NaN. Retrying.");
      outputs = predictVoxels(simulstate, eta, energy);
      first_output = outputs.begin()->second;
      contains_nan = std::isnan(first_output);
 
      retry++;
    }
  }
 
  // Fill the event structure with the voxel energies
  std::vector<unsigned int> bin_index_vector;
  std::vector<float> E_vector;
 
  event_t event;
 
  long unsigned int layer_index = 0;
  long unsigned int layer = m_used_layers.at(layer_index);
 
  for (long unsigned int voxel_index = 0; voxel_index < m_nVoxels; ++voxel_index) {
 
 
    if (voxel_index == m_layer_boundaries[layer_index+1]) {
      layer_index = layer_index + 1;
      layer = m_used_layers.at(layer_index);
    }
 
    float voxel_energy = outputs[std::to_string(voxel_index)];
 
    if (voxel_energy > 0) {
      if (event.event_data.size() <= layer) {
        event.event_data.resize(layer+1);
      }
      event.event_data.at(layer).bin_index_vector.push_back(voxel_index - m_layer_boundaries[layer_index]);
 
      // We need energy fractions, not MeV values
      event.event_data.at(layer).E_vector.push_back(voxel_energy/energy);
    }
 
  }
 
  return event;
  
}

level()

MSG::Level ISF_FCS::MLogging::level ( ) const

inlineinherited

Retrieve output level.

Definition at line 201 of file MLogging.h.

{ return msg().level(); }

loadSimulator()

bool TFCSMLCalorimeterSimulator::loadSimulator

(

std::string &

filename

)

Definition at line 16 of file TFCSMLCalorimeterSimulator.cxx.

                                                                  {
  // Load the simulator
  try {
    m_onnx_model = TFCSNetworkFactory::create(filename);
  } catch (std::exception &e) {
    ATH_MSG_ERROR("Failed to load simulator from file " << filename << " with error " << e.what());
    return false;
  }
 
  if (m_onnx_model == nullptr) {
    ATH_MSG_ERROR("Failed to load simulator from file " << filename);
    return false;
  }
 
  return true;
}

msg() [1/2]

MsgStream & ISF_FCS::MLogging::msg ( ) const

inlineinherited

Return a stream for sending messages directly (no decoration)

Definition at line 231 of file MLogging.h.

                                      {
  MsgStream *ms = m_msg_tls.get();
  if (!ms) {
    ms = new MsgStream(Athena::getMessageSvc(), m_nm);
    m_msg_tls.reset(ms);
  }
  return *ms;
}

msg() [2/2]

MsgStream & ISF_FCS::MLogging::msg ( const MSG::Level  lvl ) const

inlineinherited

Return a decorated starting stream for sending messages.

Definition at line 240 of file MLogging.h.

                                                        {
  return msg() << lvl;
}

msgLvl()

bool ISF_FCS::MLogging::msgLvl ( const MSG::Level  lvl ) const

inlineinherited

Check whether the logging system is active at the provided verbosity level.

Definition at line 222 of file MLogging.h.

                                                     {
  if (msg().level() <= lvl) {
    msg() << lvl;
    return true;
  } else {
    return false;
  }
}

predictVoxels() [1/2]

VNetworkBase::NetworkOutputs TFCSMLCalorimeterSimulator::predictVoxels

(

)

const

Definition at line 147 of file TFCSMLCalorimeterSimulator.cxx.

                                                                           {
 
  // For testing...
  // This function sets the input dimensionality and the number of predicted layers
  // to work with the currently best photon ML simulation model. 382 voxels spanned over the
  // presampler, the three EMB layers and the first HCAL layer.
  // This allows for easier testing calls.
 
  int nEvents = 1;
  int nVoxels = 382;
  int nLayers= 5;
 
  std::vector<float> eta_vector(nEvents, 2.0);
  std::vector<float> energy_vector(nEvents, 65536.0);
  std::vector<float> z_shape_vector(nEvents*nVoxels, 0.5);
  std::vector<float> z_energy_vector(nEvents*nLayers, 0.5);
 
  VNetworkBase::NetworkInputs inputs;
 
  int i = 0;
  for (float eta : eta_vector) {
    inputs["inn_eta_in"].insert(std::pair<std::string, double>("variable_" + std::to_string(i), eta));
    i++;
  }
 
  i = 0;
  for (float energy : energy_vector) {
    inputs["inn_einc_in"].insert(std::pair<std::string, double>("variable_" + std::to_string(i), energy));
    i++;
  }
 
  i = 0;
  for (float z_shape : z_shape_vector) {
    inputs["cfm_z_shape"].insert(std::pair<std::string, double>("variable_" + std::to_string(i), z_shape));
    i++;
  }
 
  i = 0;
  for (float z_energy : z_energy_vector) {
    inputs["inn_z_energy"].insert(std::pair<std::string, double>("variable_" + std::to_string(i), z_energy));
    i++;
  }
 
  ATH_MSG_DEBUG(VNetworkBase::representNetworkInputs(inputs, 1000));
 
  VNetworkBase::NetworkOutputs outputs = m_onnx_model->compute(inputs);
 
  ATH_MSG_DEBUG(VNetworkBase::representNetworkOutputs(outputs, 1000));
 
  return outputs;
 
 
}

predictVoxels() [2/2]

VNetworkBase::NetworkOutputs TFCSMLCalorimeterSimulator::predictVoxels	(	TFCSSimulationState &	simulstate,
		float	eta,
		float	energy
	)		const

Definition at line 33 of file TFCSMLCalorimeterSimulator.cxx.

                                                                                                                                   {
 
  // Bring eta into the needed range
  eta = std::abs(eta) * 10;
 
  // Initialize the energy and eta input vectors
  std::vector<float> eta_vector(m_nEvents, eta);
  std::vector<float> energy_vector(m_nEvents, energy);
 
  // sample the z vectors according to a standard normal distribution
  std::vector<float> z_shape_vector(m_nEvents * m_nVoxels, 0.0);
  std::vector<float> z_energy_vector(m_nEvents * m_nLayers, 0.0);
  for (auto& z_shape : z_shape_vector) {
    z_shape = CLHEP::RandGauss::shoot(simulstate.randomEngine(), 0.0, 1.0);
  }
  for (auto& z_energy : z_energy_vector) {
    z_energy = CLHEP::RandGauss::shoot(simulstate.randomEngine(), 0.0, 1.0);
  }
 
  // Prepare the inputs for the network
  VNetworkBase::NetworkInputs inputs;
 
  int i = 0;
  for (float thisEta : eta_vector) {
    inputs["inn_eta_in"].insert(std::pair<std::string, double>("variable_" + std::to_string(i), thisEta));
    i++;
  }
 
  i = 0;
  for (float thisEnergy : energy_vector) {
    inputs["inn_einc_in"].insert(std::pair<std::string, double>("variable_" + std::to_string(i), thisEnergy));
    i++;
  }
 
  i = 0;
  for (float z_shape : z_shape_vector) {
    inputs["cfm_z_shape"].insert(std::pair<std::string, double>("variable_" + std::to_string(i), z_shape));
    i++;
  }
 
  i = 0;
  for (float z_energy : z_energy_vector) {
    inputs["inn_z_energy"].insert(std::pair<std::string, double>("variable_" + std::to_string(i), z_energy));
    i++;
  }
 
  // Compute the network outputs
  VNetworkBase::NetworkOutputs outputs = m_onnx_model->compute(inputs);
 
  return outputs;
}

Print()

void TFCSMLCalorimeterSimulator::Print

(

)

const

Definition at line 202 of file TFCSMLCalorimeterSimulator.cxx.

                                             {
  ATH_MSG_INFO("ONNX AICalorimeterSimulator");
}

setInputShapes()

void TFCSMLCalorimeterSimulator::setInputShapes ( std::vector< long unsigned int > &  layer_boundaries, std::vector< long unsigned int > &  used_layers  )

inline

Definition at line 45 of file TFCSMLCalorimeterSimulator.h.

                                                                                                                 {
    m_layer_boundaries = layer_boundaries;
    m_used_layers = used_layers;
    m_nVoxels = layer_boundaries.back();
    m_nLayers = used_layers.size();
  };

setLevel()

void ISF_FCS::MLogging::setLevel ( MSG::Level  lvl )

virtualinherited

Update outputlevel.

Definition at line 105 of file MLogging.cxx.

                                    {
  lvl = (lvl >= MSG::NUM_LEVELS) ? MSG::ALWAYS
        : (lvl < MSG::NIL)       ? MSG::NIL
                                 : lvl;
  msg().setLevel(lvl);
}

startMsg()

std::string ISF_FCS::MLogging::startMsg ( MSG::Level  lvl, const std::string &  file, int  line  )

staticinherited

Make a message to decorate the start of logging.

Print a message for the start of logging.

Definition at line 116 of file MLogging.cxx.

                                                                          {
  int col1_len = 20;
  int col2_len = 5;
  int col3_len = 10;
  auto last_slash = file.find_last_of('/');
  int path_len = last_slash == std::string::npos ? 0 : last_slash;
  int trim_point = path_len;
  int total_len = file.length();
  if (total_len - path_len > col1_len)
    trim_point = total_len - col1_len;
  std::string trimmed_name = file.substr(trim_point);
  const char *LevelNames[MSG::NUM_LEVELS] = {
      "NIL", "VERBOSE", "DEBUG", "INFO", "WARNING", "ERROR", "FATAL", "ALWAYS"};
  std::string level = LevelNames[lvl];
  std::string level_string = std::string("(") + level + ") ";
  std::stringstream output;
  output << std::setw(col1_len) << std::right << trimmed_name << ":"
         << std::setw(col2_len) << std::left << line << std::setw(col3_len)
         << std::right << level_string;
  return output.str();
}

Member Data Documentation

ATLAS_THREAD_SAFE

boost::thread_specific_ptr<MsgStream> m_msg_tls ISF_FCS::MLogging::ATLAS_THREAD_SAFE

inlinestaticprivateinherited

Do not persistify!

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

Definition at line 215 of file MLogging.h.

m_layer_boundaries

std::vector<long unsigned int> TFCSMLCalorimeterSimulator::m_layer_boundaries = {0, 36, 200, 310, 346, 382}

private

Definition at line 60 of file TFCSMLCalorimeterSimulator.h.

m_nEvents

int TFCSMLCalorimeterSimulator::m_nEvents = 1

private

Definition at line 56 of file TFCSMLCalorimeterSimulator.h.

m_nLayers

long unsigned int TFCSMLCalorimeterSimulator::m_nLayers = 5

private

Definition at line 63 of file TFCSMLCalorimeterSimulator.h.

m_nm

std::string ISF_FCS::MLogging::m_nm

privateinherited

Message source name.

Definition at line 211 of file MLogging.h.

m_nVoxels

long unsigned int TFCSMLCalorimeterSimulator::m_nVoxels = 382

private

Definition at line 62 of file TFCSMLCalorimeterSimulator.h.

m_onnx_model

std::unique_ptr<VNetworkBase> TFCSMLCalorimeterSimulator::m_onnx_model = nullptr

private

Definition at line 54 of file TFCSMLCalorimeterSimulator.h.

m_used_layers

std::vector<long unsigned int> TFCSMLCalorimeterSimulator::m_used_layers = {0, 1, 2, 3, 12}

private

Definition at line 61 of file TFCSMLCalorimeterSimulator.h.

---

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

• TFCSMLCalorimeterSimulator.h
• TFCSMLCalorimeterSimulator.cxx