TFCSPredictExtrapWeights Class Reference

#include <TFCSPredictExtrapWeights.h>

Inheritance diagram for TFCSPredictExtrapWeights:

Collaboration diagram for TFCSPredictExtrapWeights:

Public Types
enum	FCSfreemem { kfreemem = BIT(17) }
enum	TFCSPredictExtrapWeightsStatusBits { kUseHardcodedWeight = BIT(15) }
enum	FCSStatusBits { kMatchAllPDGID }
	Status bit for FCS needs. More...
typedef std::map< TFCSParametrizationBase *, Duplicate_t >	FindDuplicates_t
typedef std::map< std::string, FindDuplicates_t >	FindDuplicateClasses_t

Public Member Functions
	TFCSPredictExtrapWeights (const char *name=nullptr, const char *title=nullptr)
virtual	~TFCSPredictExtrapWeights ()
virtual bool	operator== (const TFCSParametrizationBase &ref) const override
	The == operator compares the content of instances.
virtual FCSReturnCode	simulate (TFCSSimulationState &simulstate, const TFCSTruthState *truth, const TFCSExtrapolationState *extrapol) const override
	Method in all derived classes to do some simulation.
virtual FCSReturnCode	simulate_hit (Hit &hit, TFCSSimulationState &simulstate, const TFCSTruthState *truth, const TFCSExtrapolationState *extrapol) override
	simulated one hit position with some energy.
bool	freemem () const
void	set_freemem ()
bool	initializeNetwork (int pid, const std::string &etaBin, const std::string &FastCaloNNInputFolderName)
bool	getNormInputs (const std::string &etaBin, const std::string &FastCaloTXTInputFolderName)
std::map< std::string, double >	prepareInputs (TFCSSimulationState &simulstate, const float truthE) const
void	Print (Option_t *option="") const override
bool	UseHardcodedWeight () const
void	set_UseHardcodedWeight ()
void	reset_UseHardcodedWeight ()
virtual double	get_sigma2_fluctuation (TFCSSimulationState &simulstate, const TFCSTruthState *truth, const TFCSExtrapolationState *extrapol) const
	Give the effective size sigma^2 of the fluctuations that should be generated by the amount of generated hits.
virtual int	get_number_of_hits (TFCSSimulationState &simulstate, const TFCSTruthState *truth, const TFCSExtrapolationState *extrapol) const
	Call get_number_of_hits() only once per shower simulation, as it could be calculated with random numbers and give different results each time.
virtual float	get_E_hit (TFCSSimulationState &simulstate, const TFCSTruthState *truth, const TFCSExtrapolationState *extrapol) const
	Get hit energy from layer energy and number of hits.
virtual float	getMinWeight () const
	Get minimum and maximum value of weight for hit energy reweighting.
virtual float	getMaxWeight () const
bool	is_match_Ekin_bin (int bin) const override
bool	is_match_calosample (int calosample) const override
virtual bool	is_match_all_Ekin_bin () const override
virtual bool	is_match_all_calosample () const override
int	Ekin_bin () const
void	set_Ekin_bin (int bin)
int	calosample () const
void	set_calosample (int cs)
virtual void	set_pdgid_Ekin_eta_Ekin_bin_calosample (const TFCSLateralShapeParametrization &ref)
void	clear ()
virtual bool	is_match_pdgid (int id) const override
virtual bool	is_match_Ekin (float Ekin) const override
virtual bool	is_match_eta (float eta) const override
virtual bool	is_match_all_Ekin () const override
virtual bool	is_match_all_eta () const override
const std::set< int > &	pdgid () const override
double	Ekin_nominal () const override
double	Ekin_min () const override
double	Ekin_max () const override
double	eta_nominal () const override
double	eta_min () const override
double	eta_max () const override
virtual void	set_pdgid (int id)
virtual void	set_pdgid (const std::set< int > &ids)
virtual void	add_pdgid (int id)
virtual void	clear_pdgid ()
virtual void	set_Ekin_nominal (double min)
virtual void	set_Ekin_min (double min)
virtual void	set_Ekin_max (double max)
virtual void	set_eta_nominal (double min)
virtual void	set_eta_min (double min)
virtual void	set_eta_max (double max)
virtual void	set_Ekin (const TFCSParametrizationBase &ref)
virtual void	set_eta (const TFCSParametrizationBase &ref)
virtual void	set_Ekin_eta (const TFCSParametrizationBase &ref)
virtual void	set_pdgid_Ekin_eta (const TFCSParametrizationBase &ref)
virtual bool	is_match_all_pdgid () const
virtual void	set_match_all_pdgid ()
virtual void	reset_match_all_pdgid ()
virtual void	set_geometry (ICaloGeometry *geo)
	Method to set the geometry access pointer.
virtual unsigned int	size () const
	Some derived classes have daughter instances of TFCSParametrizationBase objects The size() and operator[] methods give general access to these daughters.
virtual const TFCSParametrizationBase *	operator[] (unsigned int) const
	Some derived classes have daughter instances of TFCSParametrizationBase objects The size() and operator[] methods give general access to these daughters.
virtual TFCSParametrizationBase *	operator[] (unsigned int)
	Some derived classes have daughter instances of TFCSParametrizationBase objects The size() and operator[] methods give general access to these daughters.
virtual void	set_daughter (unsigned int, TFCSParametrizationBase *)
	Some derived classes have daughter instances of TFCSParametrizationBase objects The set_daughter method allows to change these daughters - expert use only!
virtual void	CleanAuxInfo (TFCSSimulationState &) const
	Method in all derived classes to delete objects stored in the simulstate AuxInfo.
void	FindDuplicates (FindDuplicateClasses_t &dup)
void	RemoveDuplicates ()
void	RemoveNameTitle ()
bool	msgLvl (const MSG::Level lvl) const
	Check whether the logging system is active at the provided verbosity level.
MsgStream &	msg () const
	Return a stream for sending messages directly (no decoration).
MsgStream &	msg (const MSG::Level lvl) const
	Return a decorated starting stream for sending messages.
MSG::Level	level () const
	Retrieve output level.
virtual void	setLevel (MSG::Level lvl)
	Update outputlevel.

Static Public Member Functions
static void	test_path (std::string &net_path, std::string const &norm_path, TFCSSimulationState *simulstate=nullptr, const TFCSTruthState *truth=nullptr, const TFCSExtrapolationState *extrapol=nullptr)
static void	unit_test (TFCSSimulationState *simulstate=nullptr, const TFCSTruthState *truth=nullptr, const TFCSExtrapolationState *extrapol=nullptr)
static std::string	startMsg (MSG::Level lvl, const std::string &file, int line)
	Make a message to decorate the start of logging.

Protected Member Functions
bool	compare (const TFCSParametrizationBase &ref) const

Static Protected Attributes
static constexpr double	init_Ekin_nominal = 0
static constexpr double	init_Ekin_min = 0
	Do not persistify!
static constexpr double	init_Ekin_max = 14000000
	Do not persistify!
static constexpr double	init_eta_nominal = 0
	Do not persistify!
static constexpr double	init_eta_min = -100
	Do not persistify!
static constexpr double	init_eta_max = 100
	Do not persistify!

Private Attributes
std::string *	m_input = nullptr
std::vector< int > *	m_relevantLayers = nullptr
lwt::LightweightNeuralNetwork *	m_nn = nullptr
std::vector< int > *	m_normLayers
	Do not persistify.
std::vector< float > *	m_normMeans
std::vector< float > *	m_normStdDevs
int	m_Ekin_bin
int	m_calosample
std::set< int >	m_pdgid
double	m_Ekin_nominal
double	m_Ekin_min
double	m_Ekin_max
double	m_eta_nominal
double	m_eta_min
double	m_eta_max
std::string	m_nm
	Message source name.

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

Detailed Description

Definition at line 18 of file TFCSPredictExtrapWeights.h.

Member Typedef Documentation

FindDuplicateClasses_t

typedef std::map<std::string, FindDuplicates_t> TFCSParametrizationBase::FindDuplicateClasses_t

inherited

Definition at line 142 of file TFCSParametrizationBase.h.

FindDuplicates_t

typedef std::map<TFCSParametrizationBase *, Duplicate_t> TFCSParametrizationBase::FindDuplicates_t

inherited

Definition at line 141 of file TFCSParametrizationBase.h.

Member Enumeration Documentation

FCSfreemem

enum TFCSPredictExtrapWeights::FCSfreemem

Enumerator
kfreemem	Set this bit in the TObject bit if the memory for m_input should be freed after reading in athena.

Definition at line 38 of file TFCSPredictExtrapWeights.h.

                  {
    kfreemem = BIT(17) 
  };

FCSStatusBits

enum TFCSParametrizationBase::FCSStatusBits

inherited

Status bit for FCS needs.

Enumerator
kMatchAllPDGID	Set this bit in the TObject bit field if valid for all PDGID.

Definition at line 52 of file TFCSParametrizationBase.h.

                     {
    kMatchAllPDGID = BIT(
        14) 
  };

TFCSPredictExtrapWeightsStatusBits

enum TFCSPredictExtrapWeights::TFCSPredictExtrapWeightsStatusBits

Enumerator
kUseHardcodedWeight

Definition at line 70 of file TFCSPredictExtrapWeights.h.

{ kUseHardcodedWeight = BIT(15) };

Constructor & Destructor Documentation

TFCSPredictExtrapWeights()

TFCSPredictExtrapWeights::TFCSPredictExtrapWeights	(	const char *	name = nullptr,
		const char *	title = nullptr )

Definition at line 36 of file TFCSPredictExtrapWeights.cxx.

    : TFCSLateralShapeParametrizationHitBase(name, title) {
  set_freemem();
  set_UseHardcodedWeight();
}

~TFCSPredictExtrapWeights()

TFCSPredictExtrapWeights::~TFCSPredictExtrapWeights ( )

virtual

Definition at line 44 of file TFCSPredictExtrapWeights.cxx.

                                                    {
  if (m_input != nullptr) {
    delete m_input;
  }
  if (m_relevantLayers != nullptr) {
    delete m_relevantLayers;
  }
  if (m_normLayers != nullptr) {
    delete m_normLayers;
  }
  if (m_normMeans != nullptr) {
    delete m_normMeans;
  }
  if (m_normStdDevs != nullptr) {
    delete m_normStdDevs;
  }
  if (m_nn != nullptr) {
    delete m_nn;
  }
}

Member Function Documentation

add_pdgid()

void TFCSParametrization::add_pdgid ( int id )

virtualinherited

Reimplemented in TFCSEnergyBinParametrization.

Definition at line 35 of file TFCSParametrization.cxx.

{ m_pdgid.insert(id); }

calosample()

int TFCSLateralShapeParametrization::calosample ( ) const

inlineinherited

Definition at line 34 of file TFCSLateralShapeParametrization.h.

{ return m_calosample; };

CleanAuxInfo()

virtual void TFCSParametrizationBase::CleanAuxInfo ( TFCSSimulationState & ) const

inlinevirtualinherited

Method in all derived classes to delete objects stored in the simulstate AuxInfo.

Definition at line 130 of file TFCSParametrizationBase.h.

{};

clear()

void TFCSParametrization::clear ( )

inherited

Definition at line 18 of file TFCSParametrization.cxx.

                                {
  m_pdgid.clear();
  m_Ekin_nominal = init_Ekin_nominal;
  m_Ekin_min = init_Ekin_min;
  m_Ekin_max = init_Ekin_max;
  m_eta_nominal = init_eta_nominal;
  m_eta_min = init_eta_min;
  m_eta_max = init_eta_max;
}

clear_pdgid()

void TFCSParametrization::clear_pdgid ( )

virtualinherited

Reimplemented in TFCSEnergyBinParametrization.

Definition at line 37 of file TFCSParametrization.cxx.

{ m_pdgid.clear(); }

compare()

bool TFCSLateralShapeParametrization::compare ( const TFCSParametrizationBase & ref ) const

protectedinherited

Definition at line 32 of file TFCSLateralShapeParametrization.cxx.

                                              {
  if (IsA() != ref.IsA()) {
    ATH_MSG_DEBUG("compare(): different class types "
                  << IsA()->GetName() << " != " << ref.IsA()->GetName());
    return false;
  }
  const TFCSLateralShapeParametrization &ref_typed =
      static_cast<const TFCSLateralShapeParametrization &>(ref);
  if (Ekin_bin() != ref_typed.Ekin_bin()) {
    ATH_MSG_DEBUG("compare(): different Ekin bin");
    return false;
  }
  if (calosample() != ref_typed.calosample()) {
    ATH_MSG_DEBUG("compare(): different calosample");
    return false;
  }
 
  return true;
}

Ekin_bin()

int TFCSLateralShapeParametrization::Ekin_bin ( ) const

inlineinherited

Definition at line 31 of file TFCSLateralShapeParametrization.h.

{ return m_Ekin_bin; };

Ekin_max()

double TFCSParametrization::Ekin_max ( ) const

inlineoverridevirtualinherited

Reimplemented from TFCSParametrizationBase.

Definition at line 37 of file TFCSParametrization.h.

{ return m_Ekin_max; };

Ekin_min()

double TFCSParametrization::Ekin_min ( ) const

inlineoverridevirtualinherited

Reimplemented from TFCSParametrizationBase.

Definition at line 36 of file TFCSParametrization.h.

{ return m_Ekin_min; };

Ekin_nominal()

double TFCSParametrization::Ekin_nominal ( ) const

inlineoverridevirtualinherited

Reimplemented from TFCSParametrizationBase.

Definition at line 35 of file TFCSParametrization.h.

{ return m_Ekin_nominal; };

eta_max()

double TFCSParametrization::eta_max ( ) const

inlineoverridevirtualinherited

Reimplemented from TFCSParametrizationBase.

Definition at line 40 of file TFCSParametrization.h.

{ return m_eta_max; };

eta_min()

double TFCSParametrization::eta_min ( ) const

inlineoverridevirtualinherited

Reimplemented from TFCSParametrizationBase.

Definition at line 39 of file TFCSParametrization.h.

{ return m_eta_min; };

eta_nominal()

double TFCSParametrization::eta_nominal ( ) const

inlineoverridevirtualinherited

Reimplemented from TFCSParametrizationBase.

Definition at line 38 of file TFCSParametrization.h.

{ return m_eta_nominal; };

FindDuplicates()

void TFCSParametrizationBase::FindDuplicates ( FindDuplicateClasses_t & dup )

inherited

Definition at line 90 of file TFCSParametrizationBase.cxx.

                                        {
 
  for (unsigned int i = 0; i < size(); ++i)
    if ((*this)[i]) {
      TFCSParametrizationBase *param = (*this)[i];
      FindDuplicates_t &dup = dupclasses[param->GetName()];
      // If param is already in the duplication list, skip over
      auto checkexist = dup.find(param);
      if (checkexist != dup.end()) {
        ATH_MSG_WARNING(" [TFCSParametrizationBase::FindDuplicates] "
                        "DUPLICATE POINTER DETECTED");
 
        ATH_MSG_WARNING(" - Pointer           : " << param);
        ATH_MSG_WARNING(" - Name              : " << param->GetName());
        ATH_MSG_WARNING(" - Class             : " << param->ClassName());
        ATH_MSG_WARNING(" - Occurs in parent  : " << this
                          << " (" << this->ClassName() << ")"
                          << " index=" << i);
 
        if (checkexist->second.replace) {
          TFCSParametrizationBase *refparam = checkexist->second.replace;
          ATH_MSG_DEBUG("Found duplicate pointer: "
                        << refparam << "=" << refparam->GetName()
                        << ", duplicate is " << param << "=" << param->GetName()
                        << " index " << i << " of " << this);
          dup[refparam].mother.push_back(this);
          dup[refparam].index.push_back(i);
        }
        continue;
      }
      // Add param to duplication list
      dup[param] = Duplicate_t();
      for (auto &ref : dup) {
        TFCSParametrizationBase *refparam = ref.first;
        // skip itself, as this just added above
        if (param == refparam)
          continue;
        // skip nullptr reference
        if (refparam == nullptr)
          continue;
        // skip reference that is itself going to get replaced
        if (ref.second.replace)
          continue;
        // Check for objects with identical content
        if (*param == *refparam) {
          ATH_MSG_DEBUG("Found duplicate: "
                       << refparam << "=" << refparam->GetName()
                        << ", duplicate is " << param << "=" << param->GetName()
                        << " index " << i << " of " << this);
 
          dup[param].replace = refparam;
          dup[refparam].mother.push_back(this);
          dup[refparam].index.push_back(i);
          break;
        }
      }
      // Continue for child objects in param
      param->FindDuplicates(dupclasses);
    }
}

freemem()

bool TFCSPredictExtrapWeights::freemem ( ) const

inline

Definition at line 42 of file TFCSPredictExtrapWeights.h.

{ return TestBit(kfreemem); };

get_E_hit()

float TFCSLateralShapeParametrizationHitBase::get_E_hit ( TFCSSimulationState & simulstate, const TFCSTruthState * truth, const TFCSExtrapolationState * extrapol ) const

virtualinherited

Get hit energy from layer energy and number of hits.

Definition at line 31 of file TFCSLateralShapeParametrizationHitBase.cxx.

                                                  {
  const int nhits = get_number_of_hits(simulstate, truth, extrapol);
  const int sample = calosample();
  if (nhits <= 0 || sample < 0)
    return -1.;
  else
    return simulstate.E(sample) / nhits;
}

get_number_of_hits()

int TFCSLateralShapeParametrizationHitBase::get_number_of_hits ( TFCSSimulationState & simulstate, const TFCSTruthState * truth, const TFCSExtrapolationState * extrapol ) const

virtualinherited

Call get_number_of_hits() only once per shower simulation, as it could be calculated with random numbers and give different results each time.

Return a value of -1 if this instance can't determine

Reimplemented in TFCS2DFunctionLateralShapeParametrization, TFCSBinnedShowerBase, TFCSFlatLateralShapeParametrization, TFCSHistoLateralShapeParametrization, and TFCSLateralShapeParametrizationHitNumberFromE.

Definition at line 25 of file TFCSLateralShapeParametrizationHitBase.cxx.

                                            {
  return -1;
}

get_sigma2_fluctuation()

double TFCSLateralShapeParametrizationHitBase::get_sigma2_fluctuation ( TFCSSimulationState & simulstate, const TFCSTruthState * truth, const TFCSExtrapolationState * extrapol ) const

virtualinherited

Give the effective size sigma^2 of the fluctuations that should be generated by the amount of generated hits.

Return a value <0 if this instance can't determine

Reimplemented in TFCS2DFunctionLateralShapeParametrization, TFCSHistoLateralShapeParametrization, and TFCSLateralShapeParametrizationHitNumberFromE.

Definition at line 19 of file TFCSLateralShapeParametrizationHitBase.cxx.

                                            {
  return -1;
}

getMaxWeight()

float TFCSLateralShapeParametrizationHitBase::getMaxWeight ( ) const

virtualinherited

Reimplemented in TFCSHistoLateralShapeWeight.

Definition at line 46 of file TFCSLateralShapeParametrizationHitBase.cxx.

                                                                 {
  return -1.;
}

getMinWeight()

float TFCSLateralShapeParametrizationHitBase::getMinWeight ( ) const

virtualinherited

Get minimum and maximum value of weight for hit energy reweighting.

Reimplemented in TFCSHistoLateralShapeWeight.

Definition at line 42 of file TFCSLateralShapeParametrizationHitBase.cxx.

                                                                 {
  return -1.;
}

getNormInputs()

bool TFCSPredictExtrapWeights::getNormInputs	(	const std::string &	etaBin,
		const std::string &	FastCaloTXTInputFolderName )

Definition at line 87 of file TFCSPredictExtrapWeights.cxx.

                                                                          {
  ATH_MSG_DEBUG(" Getting normalization inputs... ");
 
  // Open corresponding TXT file and extract mean/std dev for each variable
  if (m_normLayers != nullptr) {
    m_normLayers->clear();
  } else {
    m_normLayers = new std::vector<int>();
  }
  if (m_normMeans != nullptr) {
    m_normMeans->clear();
  } else {
    m_normMeans = new std::vector<float>();
  }
  if (m_normStdDevs != nullptr) {
    m_normStdDevs->clear();
  } else {
    m_normStdDevs = new std::vector<float>();
  }
  std::string inputFileName = FastCaloTXTInputFolderName +
                              "MeanStdDevEnergyFractions_eta_" + etaBin +
                              ".txt";
  ATH_MSG_DEBUG(" Opening " << inputFileName);
  std::ifstream inputTXT(inputFileName);
  if (inputTXT.is_open()) {
    std::string line;
    while (getline(inputTXT, line)) {
      std::stringstream ss(line);
      unsigned int counter = 0;
      while (ss.good()) {
        std::string substr;
        getline(ss, substr, ' ');
        if (counter == 0) { // Get index (#layer or -1 if var == etrue)
          if (substr != "etrue") {
            int index = std::stoi(substr.substr(substr.find('_') + 1));
            m_normLayers->push_back(index);
          } else { // etrue
            m_normLayers->push_back(-1);
          }
        } else if (counter == 1) {
          m_normMeans->push_back(std::stof(substr));
        } else if (counter == 2) {
          m_normStdDevs->push_back(std::stof(substr));
        }
        counter++;
      }
    }
    inputTXT.close();
  } else {
    ATH_MSG_ERROR(" Unable to open file " << inputFileName);
    return false;
  }
 
  return true;
}

initializeNetwork()

bool TFCSPredictExtrapWeights::initializeNetwork	(	int	pid,
		const std::string &	etaBin,
		const std::string &	FastCaloNNInputFolderName )

Definition at line 287 of file TFCSPredictExtrapWeights.cxx.

                                                {
 
  ATH_MSG_INFO(
      "Using FastCaloNNInputFolderName: " << FastCaloNNInputFolderName);
  set_pdgid(pid);
 
  std::string inputFileName =
      FastCaloNNInputFolderName + "NN_" + etaBin + ".json";
  ATH_MSG_DEBUG("Will read JSON file: " << inputFileName);
  if (inputFileName.empty()) {
    ATH_MSG_ERROR("Could not find json file " << inputFileName);
    return false;
  } else {
    ATH_MSG_INFO("For pid: " << pid << " and etaBin" << etaBin
                             << ", loading json file " << inputFileName);
    std::ifstream input(inputFileName);
    std::stringstream sin;
    sin << input.rdbuf();
    input.close();
    auto config = lwt::parse_json(sin);
    m_nn = new lwt::LightweightNeuralNetwork(config.inputs, config.layers,
                                             config.outputs);
    if (m_nn == nullptr) {
      ATH_MSG_ERROR("Could not create LightWeightNeuralNetwork from "
                    << inputFileName);
      return false;
    }
    if (m_input != nullptr) {
      delete m_input;
    }
    m_input = new std::string(sin.str());
    // Extract relevant layers from the outputs
    m_relevantLayers = new std::vector<int>();
    for (auto name : config.outputs) {
      int layer = std::stoi(
          name.erase(0, 13)); // remove "extrapWeight_" and convert to int
      m_relevantLayers->push_back(layer);
    }
  }
  return true;
}

is_match_all_calosample()

virtual bool TFCSLateralShapeParametrization::is_match_all_calosample ( ) const

inlineoverridevirtualinherited

Reimplemented from TFCSParametrization.

Definition at line 29 of file TFCSLateralShapeParametrization.h.

{ return false; };

is_match_all_Ekin()

virtual bool TFCSParametrization::is_match_all_Ekin ( ) const

inlineoverridevirtualinherited

Reimplemented from TFCSParametrizationBase.

Definition at line 25 of file TFCSParametrization.h.

                                                  {
    return Ekin_min() == init_Ekin_min && Ekin_max() == init_Ekin_max;
  };

is_match_all_Ekin_bin()

virtual bool TFCSLateralShapeParametrization::is_match_all_Ekin_bin ( ) const

inlineoverridevirtualinherited

Reimplemented from TFCSParametrization.

Definition at line 24 of file TFCSLateralShapeParametrization.h.

                                                      {
    if (Ekin_bin() == -1)
      return true;
    return false;
  };

is_match_all_eta()

virtual bool TFCSParametrization::is_match_all_eta ( ) const

inlineoverridevirtualinherited

Reimplemented from TFCSParametrizationBase.

Definition at line 28 of file TFCSParametrization.h.

                                                 {
    return eta_min() == init_eta_min && eta_max() == init_eta_max;
  };

is_match_all_pdgid()

virtual bool TFCSParametrizationBase::is_match_all_pdgid ( ) const

inlinevirtualinherited

Definition at line 66 of file TFCSParametrizationBase.h.

{ return TestBit(kMatchAllPDGID); };

is_match_calosample()

bool TFCSLateralShapeParametrization::is_match_calosample ( int calosample ) const

inlineoverridevirtualinherited

Reimplemented from TFCSParametrizationBase.

Definition at line 20 of file TFCSLateralShapeParametrization.h.

                                                          {
    return calosample == m_calosample;
  };

is_match_Ekin()

virtual bool TFCSParametrization::is_match_Ekin ( float Ekin ) const

inlineoverridevirtualinherited

Reimplemented from TFCSParametrizationBase.

Definition at line 18 of file TFCSParametrization.h.

                                                        {
    return (Ekin >= m_Ekin_min) && (Ekin < m_Ekin_max);
  };

is_match_Ekin_bin()

bool TFCSLateralShapeParametrization::is_match_Ekin_bin ( int bin ) const

inlineoverridevirtualinherited

Reimplemented from TFCSParametrizationBase.

Definition at line 15 of file TFCSLateralShapeParametrization.h.

                                                 {
    if (Ekin_bin() == -1)
      return true;
    return bin == Ekin_bin();
  };

is_match_eta()

virtual bool TFCSParametrization::is_match_eta ( float eta ) const

inlineoverridevirtualinherited

Reimplemented from TFCSParametrizationBase.

Definition at line 21 of file TFCSParametrization.h.

                                                      {
    return (eta >= m_eta_min) && (eta < m_eta_max);
  };

is_match_pdgid()

virtual bool TFCSParametrization::is_match_pdgid ( int id ) const

inlineoverridevirtualinherited

Reimplemented from TFCSParametrizationBase.

Definition at line 15 of file TFCSParametrization.h.

                                                     {
    return TestBit(kMatchAllPDGID) || m_pdgid.find(id) != m_pdgid.end();
  };

level()

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

inlineinherited

Retrieve output level.

Definition at line 201 of file MLogging.h.

{ return msg().level(); }

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

operator==()

bool TFCSPredictExtrapWeights::operator== ( const TFCSParametrizationBase & ref ) const

overridevirtual

The == operator compares the content of instances.

The implementation in the base class only returns true for a comparison with itself

Reimplemented from TFCSParametrizationBase.

Definition at line 65 of file TFCSPredictExtrapWeights.cxx.

                                              {
  if (IsA() != ref.IsA()) {
    ATH_MSG_DEBUG("operator==: different class types "
                  << IsA()->GetName() << " != " << ref.IsA()->GetName());
    return false;
  }
  const TFCSPredictExtrapWeights &ref_typed =
      static_cast<const TFCSPredictExtrapWeights &>(ref);
 
  if (TFCSParametrizationBase::compare(ref))
    return true;
  if (!TFCSParametrization::compare(ref))
    return false;
  if (!TFCSLateralShapeParametrization::compare(ref))
    return false;
 
  return (m_input->compare(*ref_typed.m_input) == 0);
}

operator[]() [1/2]

virtual TFCSParametrizationBase * TFCSParametrizationBase::operator[] ( unsigned int )

inlinevirtualinherited

Some derived classes have daughter instances of TFCSParametrizationBase objects The size() and operator[] methods give general access to these daughters.

Reimplemented in TFCSLateralShapeParametrizationHitChain, and TFCSParametrizationChain.

Definition at line 106 of file TFCSParametrizationBase.h.

                                                              {
    return nullptr;
  };

operator[]() [2/2]

virtual const TFCSParametrizationBase * TFCSParametrizationBase::operator[] ( unsigned int ) const

inlinevirtualinherited

Some derived classes have daughter instances of TFCSParametrizationBase objects The size() and operator[] methods give general access to these daughters.

Reimplemented in TFCSLateralShapeParametrizationHitChain, and TFCSParametrizationChain.

Definition at line 99 of file TFCSParametrizationBase.h.

                                   {
    return nullptr;
  };

pdgid()

const std::set< int > & TFCSParametrization::pdgid ( ) const

inlineoverridevirtualinherited

Reimplemented from TFCSParametrizationBase.

Definition at line 34 of file TFCSParametrization.h.

{ return m_pdgid; };

prepareInputs()

std::map< std::string, double > TFCSPredictExtrapWeights::prepareInputs	(	TFCSSimulationState &	simulstate,
		const float	truthE ) const

Definition at line 147 of file TFCSPredictExtrapWeights.cxx.

                                                                  {
  std::map<std::string, double> inputVariables;
  for (int ilayer = 0; ilayer < CaloCell_ID_FCS::MaxSample; ++ilayer) {
    if (std::find(m_relevantLayers->cbegin(), m_relevantLayers->cend(),
                  ilayer) != m_relevantLayers->cend()) {
      const std::string layer = std::to_string(ilayer);
      // Find index
      auto itr =
          std::find(m_normLayers->cbegin(), m_normLayers->cend(), ilayer);
      if (itr != m_normLayers->cend()) {
        const int index = std::distance(m_normLayers->cbegin(), itr);
        inputVariables["ef_" + layer] =
            (simulstate.Efrac(ilayer) - std::as_const(m_normMeans)->at(index)) /
            std::as_const(m_normStdDevs)->at(index);
      } else {
        ATH_MSG_ERROR("Normalization information not found for layer "
                      << ilayer);
      }
    }
  }
  // Find index for truth energy
  auto itr = std::find(m_normLayers->cbegin(), m_normLayers->cend(), -1);
  int index = std::distance(m_normLayers->cbegin(), itr);
  inputVariables["etrue"] = (truthE - std::as_const(m_normMeans)->at(index)) /
                            std::as_const(m_normStdDevs)->at(index);
  if (is_match_pdgid(22)) {
    inputVariables["pdgId"] = 1; // one hot enconding
  } else if (is_match_pdgid(11) || is_match_pdgid(-11)) {
    inputVariables["pdgId"] = 0; // one hot enconding
  }
  return inputVariables;
}

Print()

void TFCSPredictExtrapWeights::Print ( Option_t * option = "" ) const

override

Definition at line 500 of file TFCSPredictExtrapWeights.cxx.

                                                           {
  TString opt(option);
  bool shortprint = opt.Index("short") >= 0;
  bool longprint = msgLvl(MSG::DEBUG) || (msgLvl(MSG::INFO) && !shortprint);
  TString optprint = opt;
  optprint.ReplaceAll("short", "");
  TFCSLateralShapeParametrizationHitBase::Print(option);
 
  if (longprint)
    ATH_MSG_INFO(optprint << "  m_input (TFCSPredictExtrapWeights): "
                          << CxxUtils::as_const_ptr(m_input));
  if (longprint)
    ATH_MSG_INFO(optprint << "  Address of m_nn: " << (void *)m_nn);
}

RemoveDuplicates()

void TFCSParametrizationBase::RemoveDuplicates ( )

inherited

Definition at line 152 of file TFCSParametrizationBase.cxx.

                                               {
  FindDuplicateClasses_t dupclasses;
  FindDuplicates(dupclasses);
 
  std::set<TFCSParametrizationBase *> dellist;
  for (auto &dupiter : dupclasses) {
    FindDuplicates_t &dup = dupiter.second;
    for (auto onedup : dup) {
      if (onedup.second.mother.empty())
        continue;
      TFCSParametrizationBase *ref = onedup.first;
      ATH_MSG_DEBUG("Main object " << ref << "=" << ref->GetName());
      for (unsigned int i = 0; i < onedup.second.mother.size(); ++i) {
        int index = onedup.second.index[i];
        TFCSParametrizationBase *mother = onedup.second.mother[i];
        TFCSParametrizationBase *delparam = mother->operator[](index);
        unsigned int delcount = dup[delparam].mother.size();
        if (delcount == 0) {
          ATH_MSG_DEBUG("  - Delete object "
                        << delparam << "=" << delparam->GetName() << " index "
                        << index << " of " << mother << ", has " << delcount
                        << " other replacements attached. Deleting");
          mother->set_daughter(index, ref);
          dellist.insert(delparam);
        } else {
          ATH_MSG_WARNING("  - Delete object "
                          << delparam << "=" << delparam->GetName() << " index "
                          << index << " of " << mother << ", has " << delcount
                          << " other replacements attached. Skipping");
        }
      }
    }
  }
 
  ATH_MSG_INFO("RERUNNING DUPLICATE FINDING");
  FindDuplicateClasses_t dupclasses2;
  FindDuplicates(dupclasses2);
 
  std::map<std::string, int> ndel;
  for (auto *delparam : dellist) {
    FindDuplicates_t &dup2 = dupclasses2[delparam->GetName()];
    bool present = dup2.find(delparam) != dup2.end();
    if (present) {
      ATH_MSG_WARNING("- Delete object " << delparam << "="
                                         << delparam->GetName()
                                         << " still referenced somewhere!");
    } else {
      ATH_MSG_DEBUG("- Delete object " << delparam << "="
                                       << delparam->GetName());
      ++ndel[delparam->ClassName()];
      delete delparam;
    }
  }
  for (auto &del : ndel)
    ATH_MSG_INFO("Deleted " << del.second << " duplicate objects of class "
                            << del.first);
}

RemoveNameTitle()

void TFCSParametrizationBase::RemoveNameTitle ( )

inherited

Definition at line 210 of file TFCSParametrizationBase.cxx.

                                              {
  for (unsigned int i = 0; i < size(); ++i)
    if ((*this)[i]) {
      TFCSParametrizationBase *param = (*this)[i];
      param->SetName("");
      param->SetTitle("");
 
      // Continue for child objects in param
      param->RemoveNameTitle();
    }
}

reset_match_all_pdgid()

virtual void TFCSParametrizationBase::reset_match_all_pdgid ( )

inlinevirtualinherited

Definition at line 84 of file TFCSParametrizationBase.h.

{ ResetBit(kMatchAllPDGID); };

reset_UseHardcodedWeight()

void TFCSPredictExtrapWeights::reset_UseHardcodedWeight ( )

inline

Definition at line 73 of file TFCSPredictExtrapWeights.h.

{ ResetBit(kUseHardcodedWeight); };

set_calosample()

void TFCSLateralShapeParametrization::set_calosample ( int cs )

inherited

Definition at line 21 of file TFCSLateralShapeParametrization.cxx.

                                                           {
  m_calosample = cs;
}

set_daughter()

virtual void TFCSParametrizationBase::set_daughter ( unsigned int , TFCSParametrizationBase *  )

inlinevirtualinherited

Some derived classes have daughter instances of TFCSParametrizationBase objects The set_daughter method allows to change these daughters - expert use only!

The original element at this position is not deleted

Reimplemented in TFCSLateralShapeParametrizationHitChain, and TFCSParametrizationChain.

Definition at line 113 of file TFCSParametrizationBase.h.

                                                        {};

set_Ekin()

void TFCSParametrization::set_Ekin ( const TFCSParametrizationBase & ref )

virtualinherited

Definition at line 55 of file TFCSParametrization.cxx.

                                                                     {
  set_Ekin_nominal(ref.Ekin_nominal());
  set_Ekin_min(ref.Ekin_min());
  set_Ekin_max(ref.Ekin_max());
}

set_Ekin_bin()

void TFCSLateralShapeParametrization::set_Ekin_bin ( int bin )

inherited

Definition at line 17 of file TFCSLateralShapeParametrization.cxx.

                                                          {
  m_Ekin_bin = bin;
}

set_Ekin_eta()

void TFCSParametrization::set_Ekin_eta ( const TFCSParametrizationBase & ref )

virtualinherited

Definition at line 67 of file TFCSParametrization.cxx.

                                                                         {
  set_Ekin(ref);
  set_eta(ref);
}

set_Ekin_max()

void TFCSParametrization::set_Ekin_max ( double max )

virtualinherited

Definition at line 45 of file TFCSParametrization.cxx.

{ m_Ekin_max = max; }

set_Ekin_min()

void TFCSParametrization::set_Ekin_min ( double min )

virtualinherited

Definition at line 43 of file TFCSParametrization.cxx.

{ m_Ekin_min = min; }

set_Ekin_nominal()

void TFCSParametrization::set_Ekin_nominal ( double min )

virtualinherited

Definition at line 39 of file TFCSParametrization.cxx.

                                                         {
  m_Ekin_nominal = nominal;
}

set_eta()

void TFCSParametrization::set_eta ( const TFCSParametrizationBase & ref )

virtualinherited

Definition at line 61 of file TFCSParametrization.cxx.

                                                                    {
  set_eta_nominal(ref.eta_nominal());
  set_eta_min(ref.eta_min());
  set_eta_max(ref.eta_max());
}

set_eta_max()

void TFCSParametrization::set_eta_max ( double max )

virtualinherited

Definition at line 53 of file TFCSParametrization.cxx.

{ m_eta_max = max; }

set_eta_min()

void TFCSParametrization::set_eta_min ( double min )

virtualinherited

Definition at line 51 of file TFCSParametrization.cxx.

{ m_eta_min = min; }

set_eta_nominal()

void TFCSParametrization::set_eta_nominal ( double min )

virtualinherited

Definition at line 47 of file TFCSParametrization.cxx.

                                                        {
  m_eta_nominal = nominal;
}

set_freemem()

void TFCSPredictExtrapWeights::set_freemem ( )

inline

Definition at line 43 of file TFCSPredictExtrapWeights.h.

{ SetBit(kfreemem); };

set_geometry()

void TFCSParametrizationBase::set_geometry ( ICaloGeometry * geo )

virtualinherited

Method to set the geometry access pointer.

Loops over daughter objects if present

Reimplemented in TFCSBinnedShowerBase, TFCSHistoLateralShapeParametrization, TFCSHitCellMapping, TFCSPhiModulationCorrection, and TFCSVoxelHistoLateralCovarianceFluctuations.

Definition at line 24 of file TFCSParametrizationBase.cxx.

                                                             {
  for (unsigned int i = 0; i < size(); ++i)
    (*this)[i]->set_geometry(geo);
}

set_match_all_pdgid()

virtual void TFCSParametrizationBase::set_match_all_pdgid ( )

inlinevirtualinherited

Definition at line 83 of file TFCSParametrizationBase.h.

{ SetBit(kMatchAllPDGID); };

set_pdgid() [1/2]

void TFCSParametrization::set_pdgid ( const std::set< int > & ids )

virtualinherited

Reimplemented in TFCSEnergyBinParametrization.

Definition at line 33 of file TFCSParametrization.cxx.

{ m_pdgid = ids; }

set_pdgid() [2/2]

void TFCSParametrization::set_pdgid ( int id )

virtualinherited

Reimplemented in TFCSEnergyBinParametrization.

Definition at line 28 of file TFCSParametrization.cxx.

                                          {
  m_pdgid.clear();
  m_pdgid.insert(id);
}

set_pdgid_Ekin_eta()

void TFCSParametrization::set_pdgid_Ekin_eta ( const TFCSParametrizationBase & ref )

virtualinherited

Definition at line 72 of file TFCSParametrization.cxx.

                                        {
  set_Ekin_eta(ref);
  set_pdgid(ref.pdgid());
}

set_pdgid_Ekin_eta_Ekin_bin_calosample()

void TFCSLateralShapeParametrization::set_pdgid_Ekin_eta_Ekin_bin_calosample ( const TFCSLateralShapeParametrization & ref )

virtualinherited

Definition at line 25 of file TFCSLateralShapeParametrization.cxx.

                                                {
  set_calosample(ref.calosample());
  set_Ekin_bin(ref.Ekin_bin());
  set_pdgid_Ekin_eta(ref);
}

set_UseHardcodedWeight()

void TFCSPredictExtrapWeights::set_UseHardcodedWeight ( )

inline

Definition at line 72 of file TFCSPredictExtrapWeights.h.

{ SetBit(kUseHardcodedWeight); };

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

simulate()

FCSReturnCode TFCSPredictExtrapWeights::simulate ( TFCSSimulationState & simulstate, const TFCSTruthState * truth, const TFCSExtrapolationState * extrapol ) const

overridevirtual

Method in all derived classes to do some simulation.

Result should be returned in simulstate.

Simulate all energies in calo layers for energy parametrizations. Simulate cells for shape simulation.

Reimplemented from TFCSParametrizationBase.

Definition at line 183 of file TFCSPredictExtrapWeights.cxx.

                                            {
 
  // Get inputs to Neural Network
  std::map<std::string, double> inputVariables =
      prepareInputs(simulstate, truth->E() * 0.001);
 
  // Get predicted extrapolation weights
  auto outputs = m_nn->compute(inputVariables);
  for (int ilayer = 0; ilayer < CaloCell_ID_FCS::MaxSample; ++ilayer) {
    if (std::find(m_relevantLayers->cbegin(), m_relevantLayers->cend(),
                  ilayer) != m_relevantLayers->cend()) {
      ATH_MSG_DEBUG("TFCSPredictExtrapWeights::simulate: layer: "
                    << ilayer << " weight: "
                    << outputs["extrapWeight_" + std::to_string(ilayer)]);
      float weight = outputs["extrapWeight_" + std::to_string(ilayer)];
      // Protections
      if (weight < 0) {
        weight = 0;
      } else if (weight > 1) {
        weight = 1;
      }
      simulstate.setAuxInfo<float>(ilayer, weight);
    } else { // use weight=0.5 for non-relevant layers
      ATH_MSG_DEBUG(
          "Setting weight=0.5 for layer = " << std::to_string(ilayer));
      simulstate.setAuxInfo<float>(ilayer, float(0.5));
    }
  }
  return FCSSuccess;
}

simulate_hit()

FCSReturnCode TFCSPredictExtrapWeights::simulate_hit ( Hit & hit, TFCSSimulationState & simulstate, const TFCSTruthState * truth, const TFCSExtrapolationState * extrapol )

overridevirtual

simulated one hit position with some energy.

As last step in TFCSLateralShapeParametrizationHitChain::simulate, the hit should be mapped into a cell and this cell recorded in simulstate. All hits/cells should be resacled such that their final sum is simulstate->E(sample)

Reimplemented from TFCSLateralShapeParametrizationHitBase.

Definition at line 217 of file TFCSPredictExtrapWeights.cxx.

                                            {
 
  const int cs = calosample();
 
  // Get corresponding predicted extrapolation weight from simulstate
  float extrapWeight;
  if (simulstate.hasAuxInfo(cs)) {
    extrapWeight = simulstate.getAuxInfo<float>(cs);
  } else { // missing AuxInfo
    ATH_MSG_FATAL(
        "Simulstate is not decorated with extrapolation weights for cs = "
        << std::to_string(cs));
    return FCSFatal;
  }
 
  double eta = (1. - extrapWeight) * extrapol->eta(cs, SUBPOS_ENT) +
               extrapWeight * extrapol->eta(cs, SUBPOS_EXT);
  double phi = (1. - extrapWeight) * extrapol->phi(cs, SUBPOS_ENT) +
               extrapWeight * extrapol->phi(cs, SUBPOS_EXT);
  float extrapWeight_for_r_z = extrapWeight;
  if (UseHardcodedWeight()) {
    extrapWeight_for_r_z = 0.5;
    ATH_MSG_DEBUG(
        "Will use extrapWeight=0.5 for r and z when constructing a hit");
  } else {
    ATH_MSG_DEBUG("Will use predicted extrapWeight also for r and z when "
                  "constructing a hit");
  }
  double r = (1. - extrapWeight_for_r_z) * extrapol->r(cs, SUBPOS_ENT) +
             extrapWeight_for_r_z * extrapol->r(cs, SUBPOS_EXT);
  double z = (1. - extrapWeight_for_r_z) * extrapol->z(cs, SUBPOS_ENT) +
             extrapWeight_for_r_z * extrapol->z(cs, SUBPOS_EXT);
 
  if (!std::isfinite(r) || !std::isfinite(z) || !std::isfinite(eta) ||
      !std::isfinite(phi)) {
    ATH_MSG_WARNING("Extrapolator contains NaN or infinite number.\nSetting "
                    "center position to calo boundary.");
    ATH_MSG_WARNING("Before fix: center_r: "
                    << r << " center_z: " << z << " center_phi: " << phi
                    << " center_eta: " << eta << " weight: " << extrapWeight
                    << " cs: " << cs);
    // If extrapolator fails we can set position to calo boundary
    r = extrapol->IDCaloBoundary_r();
    z = extrapol->IDCaloBoundary_z();
    eta = extrapol->IDCaloBoundary_eta();
    phi = extrapol->IDCaloBoundary_phi();
    ATH_MSG_WARNING("After fix: center_r: "
                    << r << " center_z: " << z << " center_phi: " << phi
                    << " center_eta: " << eta << " weight: " << extrapWeight
                    << " cs: " << cs);
  }
 
  hit.setCenter_r(r);
  hit.setCenter_z(z);
  hit.setCenter_eta(eta);
  hit.setCenter_phi(phi);
 
  ATH_MSG_DEBUG("TFCSPredictExtrapWeights: center_r: "
                << hit.center_r() << " center_z: " << hit.center_z()
                << " center_phi: " << hit.center_phi()
                << " center_eta: " << hit.center_eta()
                << " weight: " << extrapWeight << " cs: " << cs);
 
  return FCSSuccess;
}

size()

virtual unsigned int TFCSParametrizationBase::size ( ) const

inlinevirtualinherited

Some derived classes have daughter instances of TFCSParametrizationBase objects The size() and operator[] methods give general access to these daughters.

Reimplemented in TFCSLateralShapeParametrizationHitChain, and TFCSParametrizationChain.

Definition at line 93 of file TFCSParametrizationBase.h.

{ return 0; };

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

test_path()

void TFCSPredictExtrapWeights::test_path ( std::string & net_path, std::string const & norm_path, TFCSSimulationState * simulstate = nullptr, const TFCSTruthState * truth = nullptr, const TFCSExtrapolationState * extrapol = nullptr )

static

Definition at line 379 of file TFCSPredictExtrapWeights.cxx.

                                            {
  ISF_FCS::MLogging logger;
  ATH_MSG_NOCLASS(logger, "Testing net path ..."
                              << net_path.substr(net_path.length() - 20)
                              << " and norm path ..."
                              << norm_path.substr(norm_path.length() - 20));
  if (!simulstate) {
    simulstate = new TFCSSimulationState();
#if defined(__FastCaloSimStandAlone__)
    simulstate->setRandomEngine(new CLHEP::TRandomEngine());
#else
    simulstate->setRandomEngine(new CLHEP::RanluxEngine());
#endif
  }
  if (!truth) {
    TFCSTruthState *t = new TFCSTruthState();
    t->SetPtEtaPhiM(524288000, 0, 0, 130); // 524288 GeV
    t->set_pdgid(22);                      // photon
    truth = t;
  }
  if (!extrapol) {
    TFCSExtrapolationState *e = new TFCSExtrapolationState();
    e->set_IDCaloBoundary_eta(truth->Eta());
    for (int i = 0; i < 24; ++i) {
      e->set_eta(i, TFCSExtrapolationState::SUBPOS_ENT, truth->Eta());
      e->set_eta(i, TFCSExtrapolationState::SUBPOS_EXT, truth->Eta());
      e->set_eta(i, TFCSExtrapolationState::SUBPOS_MID, truth->Eta());
      e->set_phi(i, TFCSExtrapolationState::SUBPOS_ENT, 0);
      e->set_phi(i, TFCSExtrapolationState::SUBPOS_EXT, 0);
      e->set_phi(i, TFCSExtrapolationState::SUBPOS_MID, 0);
      e->set_r(i, TFCSExtrapolationState::SUBPOS_ENT, 1500 + i * 10);
      e->set_r(i, TFCSExtrapolationState::SUBPOS_EXT, 1510 + i * 10);
      e->set_r(i, TFCSExtrapolationState::SUBPOS_MID, 1505 + i * 10);
      e->set_z(i, TFCSExtrapolationState::SUBPOS_ENT, 3500 + i * 10);
      e->set_z(i, TFCSExtrapolationState::SUBPOS_EXT, 3510 + i * 10);
      e->set_z(i, TFCSExtrapolationState::SUBPOS_MID, 3505 + i * 10);
    }
    extrapol = e;
  }
 
  // Set energy in layers which then will be retrieved in simulate_hit()
  simulstate->set_E(0, 1028.77124023);
  simulstate->set_E(1, 68199.0625);
  simulstate->set_E(2, 438270.78125);
  simulstate->set_E(3, 3024.02929688);
  simulstate->set_E(12, 1330.10131836);
  simulstate->set_E(1028.77124023 + 68199.0625 + 438270.78125 + 3024.02929688 +
                    1330.10131836);
  simulstate->set_Efrac(0, simulstate->E(0) / simulstate->E());
  simulstate->set_Efrac(1, simulstate->E(1) / simulstate->E());
  simulstate->set_Efrac(2, simulstate->E(2) / simulstate->E());
  simulstate->set_Efrac(3, simulstate->E(3) / simulstate->E());
  simulstate->set_Efrac(12, simulstate->E(12) / simulstate->E());
 
  const int pdgId = truth->pdgid();
  const float Ekin = truth->Ekin();
  const float eta = truth->Eta();
 
  ATH_MSG_NOCLASS(logger, "True energy " << Ekin << " pdgId " << pdgId
                                         << " eta " << eta);
 
  // Find eta bin
  const int Eta = eta * 10;
  std::string etaBin = "";
  for (int i = 0; i <= 25; ++i) {
    int etaTmp = i * 5;
    if (Eta >= etaTmp && Eta < (etaTmp + 5)) {
      etaBin = std::to_string(i * 5) + "_" + std::to_string((i + 1) * 5);
    }
  }
 
  ATH_MSG_NOCLASS(logger, "etaBin = " << etaBin);
 
  TFCSPredictExtrapWeights NN("NN", "NN");
  NN.setLevel(MSG::INFO);
  const int pid = truth->pdgid();
  NN.initializeNetwork(pid, etaBin, net_path);
  NN.getNormInputs(etaBin, norm_path);
 
  // Get extrapWeights and save them as AuxInfo in simulstate
 
  // Get inputs to Neural Network
  std::map<std::string, double> inputVariables =
      NN.prepareInputs(*simulstate, truth->E() * 0.001);
 
  // Get predicted extrapolation weights
  ATH_MSG_NOCLASS(logger, "computing with m_nn");
  auto outputs = NN.m_nn->compute(inputVariables);
  const std::vector<int> layers = {0, 1, 2, 3, 12};
  for (int ilayer : layers) {
    simulstate->setAuxInfo<float>(
        ilayer, outputs["extrapWeight_" + std::to_string(ilayer)]);
  }
 
  // Simulate
  const int layer = 0;
  NN.set_calosample(layer);
  TFCSLateralShapeParametrizationHitBase::Hit hit;
  NN.simulate_hit(hit, *simulstate, truth, extrapol);
 
  // Write
  TFile *fNN = new TFile("FCSNNtest.root", "RECREATE");
  NN.Write();
  fNN->ls();
  fNN->Close();
  delete fNN;
 
  // Open
  fNN = TFile::Open("FCSNNtest.root");
  TFCSPredictExtrapWeights *NN2 = (TFCSPredictExtrapWeights *)(fNN->Get("NN"));
 
  NN2->setLevel(MSG::INFO);
  NN2->simulate_hit(hit, *simulstate, truth, extrapol);
  //simulstate->Print();
 
  return;
}

unit_test()

void TFCSPredictExtrapWeights::unit_test ( TFCSSimulationState * simulstate = nullptr, const TFCSTruthState * truth = nullptr, const TFCSExtrapolationState * extrapol = nullptr )

static

Definition at line 363 of file TFCSPredictExtrapWeights.cxx.

                                            {
  const std::string this_file = __FILE__;
  const std::string parent_dir = this_file.substr(0, this_file.find("/src/"));
  const std::string norm_path = parent_dir + "/share/NormPredExtrapSample/";
  std::string net_path = "/cvmfs/atlas-nightlies.cern.ch/repo/data/data-art/"
                         "FastCaloSim/LWTNNPredExtrapSample/";
  test_path(net_path, norm_path, simulstate, truth, extrapol);
  //net_path = "/cvmfs/atlas-nightlies.cern.ch/repo/data/data-art/FastCaloSim/"
  //           "ONNXPredExtrapSample/";
  //test_path(net_path, norm_path, simulstate, truth, extrapol);
}

UseHardcodedWeight()

bool TFCSPredictExtrapWeights::UseHardcodedWeight ( ) const

inline

Definition at line 71 of file TFCSPredictExtrapWeights.h.

{ return TestBit(kUseHardcodedWeight); };

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.

init_Ekin_max

double TFCSParametrizationBase::init_Ekin_max = 14000000

staticconstexprprotectedinherited

Do not persistify!

Definition at line 155 of file TFCSParametrizationBase.h.

init_Ekin_min

double TFCSParametrizationBase::init_Ekin_min = 0

staticconstexprprotectedinherited

Do not persistify!

Definition at line 154 of file TFCSParametrizationBase.h.

init_Ekin_nominal

double TFCSParametrizationBase::init_Ekin_nominal = 0

staticconstexprprotectedinherited

Definition at line 153 of file TFCSParametrizationBase.h.

init_eta_max

double TFCSParametrizationBase::init_eta_max = 100

staticconstexprprotectedinherited

Do not persistify!

Definition at line 158 of file TFCSParametrizationBase.h.

init_eta_min

double TFCSParametrizationBase::init_eta_min = -100

staticconstexprprotectedinherited

Do not persistify!

Definition at line 157 of file TFCSParametrizationBase.h.

init_eta_nominal

double TFCSParametrizationBase::init_eta_nominal = 0

staticconstexprprotectedinherited

Do not persistify!

Definition at line 156 of file TFCSParametrizationBase.h.

m_calosample

int TFCSLateralShapeParametrization::m_calosample

privateinherited

Definition at line 47 of file TFCSLateralShapeParametrization.h.

m_Ekin_bin

int TFCSLateralShapeParametrization::m_Ekin_bin

privateinherited

Definition at line 46 of file TFCSLateralShapeParametrization.h.

m_Ekin_max

double TFCSParametrization::m_Ekin_max

privateinherited

Definition at line 64 of file TFCSParametrization.h.

m_Ekin_min

double TFCSParametrization::m_Ekin_min

privateinherited

Definition at line 64 of file TFCSParametrization.h.

m_Ekin_nominal

double TFCSParametrization::m_Ekin_nominal

privateinherited

Definition at line 64 of file TFCSParametrization.h.

m_eta_max

double TFCSParametrization::m_eta_max

privateinherited

Definition at line 65 of file TFCSParametrization.h.

m_eta_min

double TFCSParametrization::m_eta_min

privateinherited

Definition at line 65 of file TFCSParametrization.h.

m_eta_nominal

double TFCSParametrization::m_eta_nominal

privateinherited

Definition at line 65 of file TFCSParametrization.h.

m_input

std::string* TFCSPredictExtrapWeights::m_input = nullptr

private

Definition at line 80 of file TFCSPredictExtrapWeights.h.

m_nm

std::string ISF_FCS::MLogging::m_nm

privateinherited

Message source name.

Definition at line 211 of file MLogging.h.

m_nn

lwt::LightweightNeuralNetwork* TFCSPredictExtrapWeights::m_nn = nullptr

private

Definition at line 82 of file TFCSPredictExtrapWeights.h.

m_normLayers

std::vector<int>* TFCSPredictExtrapWeights::m_normLayers

private

Initial value:

                     =
nullptr

Do not persistify.

Definition at line 83 of file TFCSPredictExtrapWeights.h.

m_normMeans

std::vector<float>* TFCSPredictExtrapWeights::m_normMeans

private

Initial value:

                    =
nullptr

Definition at line 85 of file TFCSPredictExtrapWeights.h.

m_normStdDevs

std::vector<float>* TFCSPredictExtrapWeights::m_normStdDevs

private

Initial value:

                      =
nullptr

Definition at line 88 of file TFCSPredictExtrapWeights.h.

m_pdgid

std::set<int> TFCSParametrization::m_pdgid

privateinherited

Definition at line 63 of file TFCSParametrization.h.

m_relevantLayers

std::vector<int>* TFCSPredictExtrapWeights::m_relevantLayers = nullptr

private

Definition at line 81 of file TFCSPredictExtrapWeights.h.

---

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

• TFCSPredictExtrapWeights.h
• TFCSPredictExtrapWeights.cxx