ATLAS Offline Software
Loading...
Searching...
No Matches
TFCSEnergyAndHitGAN Class Reference

#include <TFCSEnergyAndHitGAN.h>

Inheritance diagram for TFCSEnergyAndHitGAN:
Collaboration diagram for TFCSEnergyAndHitGAN:

Public Types

enum  FCSGANfreemem { kGANfreemem = BIT(17) }
 Status bit for chain persistency. More...
enum  FCSEnergyInitializationStatusBits { kOnlyScaleEnergy = }
 Status bit for energy initialization. More...
typedef std::map< std::string, std::map< std::string, double > > NetworkInputs
typedef std::map< std::string, double > NetworkOutputs
typedef std::map< int, TH2D > Binning
enum  FCSSplitChainObjects { kSplitChainObjects , kRetryChainFromStart = BIT(17) }
 Status bit for chain persistency. More...
typedef std::vector< TFCSParametrizationBase * > Chain_t
enum  FCSStatusBits { kMatchAllPDGID }
 Status bit for FCS needs. More...
typedef std::map< TFCSParametrizationBase *, Duplicate_tFindDuplicates_t
typedef std::map< std::string, FindDuplicates_tFindDuplicateClasses_t

Public Member Functions

 TFCSEnergyAndHitGAN (const char *name=nullptr, const char *title=nullptr)
virtual ~TFCSEnergyAndHitGAN ()
virtual bool is_match_Ekin_bin (int) const override
virtual 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
bool GANfreemem () const
void set_GANfreemem ()
void reset_GANfreemem ()
bool OnlyScaleEnergy () const
void set_OnlyScaleEnergy ()
void reset_OnlyScaleEnergy ()
virtual int get_bin (TFCSSimulationState &simulstate, const TFCSTruthState *, const TFCSExtrapolationState *) const override
 use the layer to be done as binning of the GAN chain
virtual const std::string get_variable_text (TFCSSimulationState &simulstate, const TFCSTruthState *, const TFCSExtrapolationState *) const override
unsigned int get_nr_of_init (unsigned int bin) const
void set_nr_of_init (unsigned int bin, unsigned int ninit)
const Binningget_Binning () const
const lwt::LightweightGraph * get_graph () const
const std::string * get_input () const
bool initializeNetwork (int pid, int etaMin, const std::string &FastCaloGANInputFolderName)
bool fillFastCaloGanNetworkInputs (TFCSSimulationState &simulstate, const TFCSTruthState *truth, NetworkInputs &inputs, double &trueEnergy) const
bool fillEnergy (TFCSSimulationState &simulstate, const TFCSTruthState *truth, const TFCSExtrapolationState *extrapol, NetworkInputs inputs) const
virtual FCSReturnCode simulate (TFCSSimulationState &simulstate, const TFCSTruthState *truth, const TFCSExtrapolationState *extrapol) const override
 Method in all derived classes to do some simulation.
virtual void Print (Option_t *option="") const override
virtual void push_before_first_bin (TFCSParametrizationBase *param)
virtual void push_back_in_bin (TFCSParametrizationBase *param, unsigned int bin)
virtual unsigned int get_number_of_bins () const
virtual const std::string get_bin_text (int bin) const
 print the range of a bin; for bin -1, print the allowed range
bool SplitChainObjects () const
void set_SplitChainObjects ()
void reset_SplitChainObjects ()
bool RetryChainFromStart () const
void set_RetryChainFromStart ()
void reset_RetryChainFromStart ()
virtual unsigned int size () const override
 Some derived classes have daughter instances of TFCSParametrizationBase objects The size() and operator[] methods give general access to these daughters.
virtual const TFCSParametrizationBaseoperator[] (unsigned int ind) const override
 Some derived classes have daughter instances of TFCSParametrizationBase objects The size() and operator[] methods give general access to these daughters.
virtual TFCSParametrizationBaseoperator[] (unsigned int ind) override
 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 ind, TFCSParametrizationBase *param) override
 Some derived classes have daughter instances of TFCSParametrizationBase objects The set_daughter method allows to change these daughters - expert use only!
const Chain_tchain () const
Chain_tchain ()
void push_back (const Chain_t::value_type &param)
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 bool operator== (const TFCSParametrizationBase &ref) const
 The == operator compares the content of instances.
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 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

void GetBinning (int pid, int etaMax, const std::string &FastCaloGANInputFolderName)
void recalc_pdgid_intersect ()
void recalc_pdgid_union ()
void recalc_Ekin_intersect ()
void recalc_eta_intersect ()
void recalc_Ekin_eta_intersect ()
void recalc_Ekin_union ()
void recalc_eta_union ()
void recalc_Ekin_eta_union ()
virtual void recalc ()
 Default is to call recalc_pdgid_intersect() and recalc_Ekin_eta_intersect()
FCSReturnCode simulate_and_retry (TFCSParametrizationBase *parametrization, TFCSSimulationState &simulstate, const TFCSTruthState *truth, const TFCSExtrapolationState *extrapol) const
bool compare (const TFCSParametrizationBase &ref) const

Protected Attributes

std::vector< unsigned int > m_bin_start
 Contains the index where the TFCSParametrizationBase* instances to run for a given bin start.

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::vector< int > m_bin_ninit
std::string * m_input = nullptr
lwt::LightweightGraph * m_graph = nullptr
Binning m_Binning
 Do not persistify.
int m_GANLatentSize = 0
Chain_t m_chain
std::vector< std::unique_ptr< TFCSParametrizationBase > > m_writtenBases
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 TFCSEnergyAndHitGAN.h.

Member Typedef Documentation

◆ Binning

typedef std::map<int, TH2D> TFCSEnergyAndHitGAN::Binning

Definition at line 23 of file TFCSEnergyAndHitGAN.h.

◆ Chain_t

Definition at line 40 of file TFCSParametrizationChain.h.

◆ FindDuplicateClasses_t

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

Definition at line 142 of file TFCSParametrizationBase.h.

◆ FindDuplicates_t

◆ NetworkInputs

typedef std::map<std::string, std::map<std::string, double> > TFCSEnergyAndHitGAN::NetworkInputs

Definition at line 21 of file TFCSEnergyAndHitGAN.h.

◆ NetworkOutputs

typedef std::map<std::string, double> TFCSEnergyAndHitGAN::NetworkOutputs

Definition at line 22 of file TFCSEnergyAndHitGAN.h.

Member Enumeration Documentation

◆ FCSEnergyInitializationStatusBits

Status bit for energy initialization.

Enumerator
kOnlyScaleEnergy 

Set this bit in the TObject bit field the simulated energy.

should only be scaled by the GAN

Definition at line 46 of file TFCSEnergyAndHitGAN.h.

46 {
48 BIT(18)
50 };
@ kOnlyScaleEnergy
Set this bit in the TObject bit field the simulated energy.

◆ FCSGANfreemem

Status bit for chain persistency.

Enumerator
kGANfreemem 

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

Definition at line 36 of file TFCSEnergyAndHitGAN.h.

36 {
37 kGANfreemem = BIT(17)
39 };
@ kGANfreemem
Set this bit in the TObject bit if the memory for m_input should be freed after reading in athena.

◆ FCSSplitChainObjects

Status bit for chain persistency.

Enumerator
kSplitChainObjects 

Set this bit in the TObject bit field if the TFCSParametrizationBase objects in the chain should be written as separate keys into the root file instead of directly writing the objects.

This is needed if the sum of all objects in the chain use >1GB of memory, which can't be handeled by TBuffer. Drawback is that identical objects will get stored as multiple instances

kRetryChainFromStart 

Definition at line 20 of file TFCSParametrizationChain.h.

20 {
22 BIT(16),
29 kRetryChainFromStart = BIT(17)
30 };
@ kSplitChainObjects
Set this bit in the TObject bit field if the TFCSParametrizationBase objects in the chain should be w...

◆ FCSStatusBits

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.

52 {
53 kMatchAllPDGID = BIT(
54 14)
55 };
@ kMatchAllPDGID
Set this bit in the TObject bit field if valid for all PDGID.

Constructor & Destructor Documentation

◆ TFCSEnergyAndHitGAN()

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

Definition at line 47 of file TFCSEnergyAndHitGAN.cxx.

48 : TFCSParametrizationBinnedChain(name, title) {
50}
TFCSParametrizationBinnedChain(const char *name=nullptr, const char *title=nullptr)

◆ ~TFCSEnergyAndHitGAN()

TFCSEnergyAndHitGAN::~TFCSEnergyAndHitGAN ( )
virtual

Definition at line 52 of file TFCSEnergyAndHitGAN.cxx.

52 {
53 if (m_input != nullptr) {
54 delete m_input;
55 }
56 if (m_graph != nullptr) {
57 delete m_graph;
58 }
59}
lwt::LightweightGraph * m_graph

Member Function Documentation

◆ add_pdgid()

void TFCSParametrization::add_pdgid ( int id)
virtualinherited

Reimplemented in TFCSEnergyBinParametrization.

Definition at line 35 of file TFCSParametrization.cxx.

35{ m_pdgid.insert(id); }

◆ chain() [1/2]

Chain_t & TFCSParametrizationChain::chain ( )
inlineinherited

Definition at line 54 of file TFCSParametrizationChain.h.

◆ chain() [2/2]

const Chain_t & TFCSParametrizationChain::chain ( ) const
inlineinherited

Definition at line 53 of file TFCSParametrizationChain.h.

53{ return m_chain; };

◆ 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.

130{};

◆ clear()

void TFCSParametrization::clear ( )
inherited

Definition at line 18 of file TFCSParametrization.cxx.

18 {
19 m_pdgid.clear();
26}
static constexpr double init_Ekin_nominal
static constexpr double init_eta_max
Do not persistify!
static constexpr double init_Ekin_max
Do not persistify!
static constexpr double init_eta_nominal
Do not persistify!
static constexpr double init_eta_min
Do not persistify!
static constexpr double init_Ekin_min
Do not persistify!

◆ clear_pdgid()

void TFCSParametrization::clear_pdgid ( )
virtualinherited

Reimplemented in TFCSEnergyBinParametrization.

Definition at line 37 of file TFCSParametrization.cxx.

37{ m_pdgid.clear(); }

◆ compare()

bool TFCSParametrization::compare ( const TFCSParametrizationBase & ref) const
protectedinherited

Definition at line 78 of file TFCSParametrization.cxx.

78 {
79 if (IsA() != ref.IsA()) {
80 ATH_MSG_DEBUG("compare(): different class types "
81 << IsA()->GetName() << " != " << ref.IsA()->GetName());
82 return false;
83 }
84 std::string name(GetName());
85 if (name != ref.GetName()) {
86 ATH_MSG_DEBUG("compare(): different names " << GetName()
87 << " != " << ref.GetName());
88 return false;
89 }
90 std::string title(GetTitle());
91 if (title != ref.GetTitle()) {
92 ATH_MSG_DEBUG("compare(): different titles " << GetTitle()
93 << " != " << ref.GetTitle());
94 return false;
95 }
96 if (is_match_all_pdgid() != ref.is_match_all_pdgid() ||
97 pdgid() != ref.pdgid()) {
98 ATH_MSG_DEBUG("compare(): different pdgids");
99 return false;
100 }
101 if (Ekin_nominal() != ref.Ekin_nominal() || Ekin_min() != ref.Ekin_min() ||
102 Ekin_max() != ref.Ekin_max()) {
103 ATH_MSG_DEBUG("compare(): different Ekin range");
104 return false;
105 }
106 if (eta_nominal() != ref.eta_nominal() || eta_min() != ref.eta_min() ||
107 eta_max() != ref.eta_max()) {
108 ATH_MSG_DEBUG("compare(): different eta range");
109 return false;
110 }
111
112 return true;
113}
const boost::regex ref(r_ef)
#define ATH_MSG_DEBUG(x)
virtual bool is_match_all_pdgid() const
double Ekin_nominal() const override
double eta_nominal() const override
double Ekin_min() const override
double eta_max() const override
double Ekin_max() const override
const std::set< int > & pdgid() const override
double eta_min() const override
#define IsA
Declare the TObject style functions.

◆ Ekin_max()

double TFCSParametrization::Ekin_max ( ) const
inlineoverridevirtualinherited

Reimplemented from TFCSParametrizationBase.

Definition at line 37 of file TFCSParametrization.h.

37{ return m_Ekin_max; };

◆ Ekin_min()

double TFCSParametrization::Ekin_min ( ) const
inlineoverridevirtualinherited

Reimplemented from TFCSParametrizationBase.

Definition at line 36 of file TFCSParametrization.h.

36{ return m_Ekin_min; };

◆ Ekin_nominal()

double TFCSParametrization::Ekin_nominal ( ) const
inlineoverridevirtualinherited

Reimplemented from TFCSParametrizationBase.

Definition at line 35 of file TFCSParametrization.h.

35{ return m_Ekin_nominal; };

◆ eta_max()

double TFCSParametrization::eta_max ( ) const
inlineoverridevirtualinherited

Reimplemented from TFCSParametrizationBase.

Definition at line 40 of file TFCSParametrization.h.

40{ return m_eta_max; };

◆ eta_min()

double TFCSParametrization::eta_min ( ) const
inlineoverridevirtualinherited

Reimplemented from TFCSParametrizationBase.

Definition at line 39 of file TFCSParametrization.h.

39{ return m_eta_min; };

◆ eta_nominal()

double TFCSParametrization::eta_nominal ( ) const
inlineoverridevirtualinherited

Reimplemented from TFCSParametrizationBase.

Definition at line 38 of file TFCSParametrization.h.

38{ return m_eta_nominal; };

◆ fillEnergy()

bool TFCSEnergyAndHitGAN::fillEnergy ( TFCSSimulationState & simulstate,
const TFCSTruthState * truth,
const TFCSExtrapolationState * extrapol,
NetworkInputs inputs ) const

Definition at line 262 of file TFCSEnergyAndHitGAN.cxx.

265 {
266 const int pdgId = truth->pdgid();
267 const float charge = HepPDT::ParticleID(pdgId).charge();
268
269 float Einit;
270 const float Ekin = truth->Ekin();
271 if (OnlyScaleEnergy())
272 Einit = simulstate.E();
273 else
274 Einit = Ekin;
275
276 ATH_MSG_VERBOSE("Momentum " << truth->P() << " pdgId " << truth->pdgid());
277
278 // compute the network output values
279 // ATH_MSG_VERBOSE("neural network input = "<<inputs);
280 ATH_MSG_VERBOSE("input size " << inputs["node_0"].size());
281 NetworkOutputs outputs = m_graph->compute(inputs);
282 // ATH_MSG_VERBOSE("neural network output = "<<outputs);
283
284 ATH_MSG_VERBOSE("neural network output size = " << outputs.size());
285
286 const Binning &binsInLayers = m_Binning;
287 ATH_MSG_VERBOSE("Get binning");
288
289 simulstate.set_E(0);
290
291 int vox = 0;
292 for (const auto& element : binsInLayers) {
293 int layer = element.first;
294
295 const TH2D *h = &element.second;
296 int xBinNum = h->GetNbinsX();
297 // If only one bin in r means layer is empty, no value should be added
298 if (xBinNum == 1) {
299 ATH_MSG_VERBOSE(" Layer "
300 << layer
301 << " has only one bin in r, this means is it not used, "
302 "skipping (this is needed to keep correct "
303 "syncronisation of voxel and layers)");
304 // delete h;
305 continue;
306 }
307
308 ATH_MSG_VERBOSE(" Getting energy for Layer " << layer);
309
310 int yBinNum = h->GetNbinsY();
311
312 // First fill energies
313 for (int iy = 1; iy <= yBinNum; ++iy) {
314 for (int ix = 1; ix <= xBinNum; ++ix) {
315 double energyInVoxel = outputs["out_" + std::to_string(vox)];
316 ATH_MSG_VERBOSE(" Vox " << vox << " energy " << energyInVoxel
317 << " binx " << ix << " biny " << iy);
318
319 if (energyInVoxel == 0) {
320 vox++;
321 continue;
322 }
323
324 simulstate.add_E(layer, Einit * energyInVoxel);
325 vox++;
326 }
327 }
328 }
329
330 for (unsigned int ichain = m_bin_start.back(); ichain < size(); ++ichain) {
331 ATH_MSG_DEBUG("now run for all bins: " << chain()[ichain]->GetName());
332 if (simulate_and_retry(chain()[ichain], simulstate, truth, extrapol) !=
333 FCSSuccess) {
334 return FCSFatal;
335 }
336 }
337
338 vox = 0;
339 for (const auto& element : binsInLayers) {
340 int layer = element.first;
341 simulstate.setAuxInfo<int>("GANlayer"_FCShash, layer);
342 TFCSLateralShapeParametrizationHitBase::Hit hit;
343
344 const TH2D *h = &element.second;
345 int xBinNum = h->GetNbinsX();
346 // If only one bin in r means layer is empty, no value should be added
347 if (xBinNum == 1) {
348 ATH_MSG_VERBOSE(" Layer "
349 << layer
350 << " has only one bin in r, this means is it not used, "
351 "skipping (this is needed to keep correct "
352 "syncronisation of voxel and layers)");
353 // delete h;
354 continue;
355 }
356
357 if (get_number_of_bins() > 0) {
358 const int bin = get_bin(simulstate, truth, extrapol);
359 if (bin >= 0 && bin < (int)get_number_of_bins()) {
360 for (unsigned int ichain = m_bin_start[bin];
361 ichain < TMath::Min(m_bin_start[bin] + get_nr_of_init(bin),
362 m_bin_start[bin + 1]);
363 ++ichain) {
364 ATH_MSG_DEBUG("for " << get_variable_text(simulstate, truth, extrapol)
365 << " run init " << get_bin_text(bin) << ": "
366 << chain()[ichain]->GetName());
367 if (chain()[ichain]->InheritsFrom(
368 TFCSLateralShapeParametrizationHitBase::Class())) {
369 TFCSLateralShapeParametrizationHitBase *sim =
370 static_cast<TFCSLateralShapeParametrizationHitBase *>(chain()[ichain]);
371 if (sim->simulate_hit(hit, simulstate, truth, extrapol) !=
372 FCSSuccess) {
373 ATH_MSG_ERROR("error for "
374 << get_variable_text(simulstate, truth, extrapol)
375 << " run init " << get_bin_text(bin) << ": "
376 << chain()[ichain]->GetName());
377 return false;
378 }
379 } else {
380 ATH_MSG_ERROR("for "
381 << get_variable_text(simulstate, truth, extrapol)
382 << " run init " << get_bin_text(bin) << ": "
383 << chain()[ichain]->GetName()
384 << " does not inherit from "
385 "TFCSLateralShapeParametrizationHitBase");
386 return false;
387 }
388 }
389 } else {
390 ATH_MSG_WARNING("nothing to init for "
391 << get_variable_text(simulstate, truth, extrapol)
392 << ": " << get_bin_text(bin));
393 }
394 }
395
396 int binResolution = 5;
397 if (layer == 1 || layer == 5) {
398 binResolution = 1;
399 }
400
401 const double center_eta = hit.center_eta();
402 const double center_phi = hit.center_phi();
403 const double center_r = hit.center_r();
404 const double center_z = hit.center_z();
405
406 ATH_MSG_VERBOSE(" Layer " << layer << " Extrap eta " << center_eta
407 << " phi " << center_phi << " R " << center_r);
408
409 const float dist000 =
410 TMath::Sqrt(center_r * center_r + center_z * center_z);
411 const float eta_jakobi = TMath::Abs(2.0 * TMath::Exp(-center_eta) /
412 (1.0 + TMath::Exp(-2 * center_eta)));
413
414 int nHitsAlpha;
415 int nHitsR;
416
417 int yBinNum = h->GetNbinsY();
418
419 // Now create hits
420 for (int iy = 1; iy <= yBinNum; ++iy) {
421 for (int ix = 1; ix <= xBinNum; ++ix) {
422
423 double energyInVoxel = outputs["out_" + std::to_string(vox)];
424 ATH_MSG_VERBOSE(" Vox " << vox << " energy " << energyInVoxel
425 << " binx " << ix << " biny " << iy);
426
427 if (energyInVoxel == 0) {
428 vox++;
429 continue;
430 }
431
432 const TAxis *x = h->GetXaxis();
433 nHitsR = x->GetBinUpEdge(ix) - x->GetBinLowEdge(ix);
434 if (yBinNum == 1) {
435 // nbins in alpha depend on circumference lenght
436 double r = x->GetBinUpEdge(ix);
437 nHitsAlpha = ceil(2 * TMath::Pi() * r / binResolution);
438 } else {
439 // d = 2*r*sin (a/2r) this distance at the upper r must be 1mm for
440 // layer 1 or 5, 5mm otherwise.
441 const TAxis *y = h->GetYaxis();
442 double angle = y->GetBinUpEdge(iy) - y->GetBinLowEdge(iy);
443 double r = x->GetBinUpEdge(ix);
444 double d = 2 * r * sin(angle / 2 * r);
445 nHitsAlpha = ceil(d / binResolution);
446 }
447
448 nHitsAlpha = std::min(10, std::max(1, nHitsAlpha));
449 nHitsR = std::min(10, std::max(1, nHitsR));
450
451 for (int ir = 0; ir < nHitsR; ++ir) {
452 const TAxis *x = h->GetXaxis();
453 double r =
454 x->GetBinLowEdge(ix) + x->GetBinWidth(ix) / (nHitsR + 1) * ir;
455
456 for (int ialpha = 0; ialpha < nHitsAlpha; ++ialpha) {
457 double alpha;
458 if (yBinNum == 1) {
459 alpha = CLHEP::RandFlat::shoot(simulstate.randomEngine(), -M_PI,
460 M_PI);
461 } else {
462 const TAxis *y = h->GetYaxis();
463 alpha = y->GetBinLowEdge(iy) +
464 y->GetBinWidth(iy) / (nHitsAlpha + 1) * ialpha;
465 }
466
467 hit.reset();
468 hit.E() = Einit * energyInVoxel / (nHitsAlpha * nHitsR);
469
470 if (layer <= 20) {
471 float delta_eta_mm = r * cos(alpha);
472 float delta_phi_mm = r * sin(alpha);
473
474 ATH_MSG_VERBOSE("delta_eta_mm " << delta_eta_mm
475 << " delta_phi_mm "
476 << delta_phi_mm);
477
478 // Particles with negative eta are expected to have the same shape
479 // as those with positive eta after transformation: delta_eta -->
480 // -delta_eta
481 if (center_eta < 0.)
482 delta_eta_mm = -delta_eta_mm;
483 // We derive the shower shapes for electrons and positively charged hadrons.
484 // Particle with the opposite charge are expected to have the same shower shape
485 // after the transformation: delta_phi --> -delta_phi
486 if ((charge < 0. && pdgId!=11) || pdgId==-11)
487 delta_phi_mm = -delta_phi_mm;
488
489 const float delta_eta = delta_eta_mm / eta_jakobi / dist000;
490 const float delta_phi = delta_phi_mm / center_r;
491
492 hit.eta() = center_eta + delta_eta;
493 hit.phi() = center_phi + delta_phi;
494
495 ATH_MSG_VERBOSE(" Hit eta " << hit.eta() << " phi " << hit.phi()
496 << " layer " << layer);
497 } else { // FCAL is in (x,y,z)
498 const float hit_r = r * cos(alpha) + center_r;
499 float delta_phi = r * sin(alpha) / center_r;
500
501 // We derive the shower shapes for electrons and positively charged hadrons.
502 // Particle with the opposite charge are expected to have the same shower shape
503 // after the transformation: delta_phi --> -delta_phi
504 if ((charge < 0. && pdgId!=11) || pdgId==-11)
506 const float hit_phi = delta_phi + center_phi;
507 hit.x() = hit_r * cos(hit_phi);
508 hit.y() = hit_r * sin(hit_phi);
509 hit.z() = center_z;
510 ATH_MSG_VERBOSE(" Hit x " << hit.x() << " y " << hit.y()
511 << " layer " << layer);
512 }
513
514 if (get_number_of_bins() > 0) {
515 const int bin = get_bin(simulstate, truth, extrapol);
516 if (bin >= 0 && bin < (int)get_number_of_bins()) {
517 for (unsigned int ichain =
518 m_bin_start[bin] + get_nr_of_init(bin);
519 ichain < m_bin_start[bin + 1]; ++ichain) {
521 "for " << get_variable_text(simulstate, truth, extrapol)
522 << " run " << get_bin_text(bin) << ": "
523 << chain()[ichain]->GetName());
524 if (chain()[ichain]->InheritsFrom(
525 TFCSLateralShapeParametrizationHitBase::Class())) {
526 TFCSLateralShapeParametrizationHitBase *sim =
527 static_cast<TFCSLateralShapeParametrizationHitBase
528 *>(chain()[ichain]);
529 if (sim->simulate_hit(hit, simulstate, truth, extrapol) !=
530 FCSSuccess) {
532 "error for "
533 << get_variable_text(simulstate, truth, extrapol)
534 << " run init " << get_bin_text(bin) << ": "
535 << chain()[ichain]->GetName());
536 return false;
537 }
538 } else {
540 "for " << get_variable_text(simulstate, truth, extrapol)
541 << " run init " << get_bin_text(bin) << ": "
542 << chain()[ichain]->GetName()
543 << " does not inherit from "
544 "TFCSLateralShapeParametrizationHitBase");
545 return false;
546 }
547 }
548 } else {
550 "nothing to do for "
551 << get_variable_text(simulstate, truth, extrapol) << ": "
552 << get_bin_text(bin));
553 }
554 } else {
555 ATH_MSG_WARNING("no bins defined, is this intended?");
556 }
557 }
558 }
559 vox++;
560 }
561 }
562
563 ATH_MSG_VERBOSE("Number of voxels " << vox);
564
565 // delete h;
566 ATH_MSG_VERBOSE("Done layer " << layer);
567 }
568 if (simulstate.E() > std::numeric_limits<double>::epsilon()) {
569 for (int ilayer = 0; ilayer < CaloCell_ID_FCS::MaxSample; ++ilayer) {
570 simulstate.set_Efrac(ilayer, simulstate.E(ilayer) / simulstate.E());
571 }
572 }
573
574 ATH_MSG_VERBOSE("Done particle");
575
576 return true;
577}
#define M_PI
#define ATH_MSG_ERROR(x)
#define ATH_MSG_VERBOSE(x)
#define ATH_MSG_WARNING(x)
double charge(const T &p)
Definition AtlasPID.h:997
double angle(const GeoTrf::Vector2D &a, const GeoTrf::Vector2D &b)
#define y
#define x
unsigned int get_nr_of_init(unsigned int bin) const
virtual int get_bin(TFCSSimulationState &simulstate, const TFCSTruthState *, const TFCSExtrapolationState *) const override
use the layer to be done as binning of the GAN chain
Binning m_Binning
Do not persistify.
std::map< std::string, double > NetworkOutputs
virtual const std::string get_variable_text(TFCSSimulationState &simulstate, const TFCSTruthState *, const TFCSExtrapolationState *) const override
std::map< int, TH2D > Binning
virtual FCSReturnCode simulate_hit(Hit &hit, TFCSSimulationState &simulstate, const TFCSTruthState *truth, const TFCSExtrapolationState *extrapol)
simulated one hit position with some energy.
virtual unsigned int size() const
Some derived classes have daughter instances of TFCSParametrizationBase objects The size() and operat...
virtual unsigned int get_number_of_bins() const
std::vector< unsigned int > m_bin_start
Contains the index where the TFCSParametrizationBase* instances to run for a given bin start.
virtual const std::string get_bin_text(int bin) const
print the range of a bin; for bin -1, print the allowed range
FCSReturnCode simulate_and_retry(TFCSParametrizationBase *parametrization, TFCSSimulationState &simulstate, const TFCSTruthState *truth, const TFCSExtrapolationState *extrapol) const
const Chain_t & chain() const
virtual unsigned int size() const override
Some derived classes have daughter instances of TFCSParametrizationBase objects The size() and operat...
void set_E(int sample, double Esample)
void add_E(int sample, double Esample)
void set_Efrac(int sample, double Efracsample)
void setAuxInfo(std::uint32_t index, const T &val)
CLHEP::HepRandomEngine * randomEngine()
int pdgid() const
double Ekin() const
int ir
counter of the current depth
Definition fastadd.cxx:49
int r
Definition globals.cxx:22
@ layer
Definition HitInfo.h:79
delta_phi(phi1, phi2)
Definition eFEXNTuple.py:14

◆ fillFastCaloGanNetworkInputs()

bool TFCSEnergyAndHitGAN::fillFastCaloGanNetworkInputs ( TFCSSimulationState & simulstate,
const TFCSTruthState * truth,
NetworkInputs & inputs,
double & trueEnergy ) const

Definition at line 240 of file TFCSEnergyAndHitGAN.cxx.

242 {
243 // fill randomize latent space
244 // FIXME: should this really be momentum
245 trueEnergy = truth->P();
246 double randUniformZ = 0.;
247 // FIXME: make dependency on input particle eta, pdgid and energy
248 for (int i = 0; i < m_GANLatentSize; i++) {
249 randUniformZ = CLHEP::RandFlat::shoot(simulstate.randomEngine(), -1., 1.);
250 inputs["node_0"].insert(
251 std::pair<std::string, double>(std::to_string(i), randUniformZ));
252 }
253
254 // ATH_MSG_INFO( "Check label: " <<trueEnergy <<" "<<std::pow(2,22)<<"
255 // "<<trueEnergy/std::pow(2,22));
256 inputs["node_1"].insert(
257 std::pair<std::string, double>("0", trueEnergy / (std::pow(2, 22))));
258
259 return true;
260}

◆ FindDuplicates()

void TFCSParametrizationBase::FindDuplicates ( FindDuplicateClasses_t & dup)
inherited

Definition at line 90 of file TFCSParametrizationBase.cxx.

91 {
92
93 for (unsigned int i = 0; i < size(); ++i)
94 if ((*this)[i]) {
95 TFCSParametrizationBase *param = (*this)[i];
96 FindDuplicates_t &dup = dupclasses[param->GetName()];
97 // If param is already in the duplication list, skip over
98 auto checkexist = dup.find(param);
99 if (checkexist != dup.end()) {
100 ATH_MSG_WARNING(" [TFCSParametrizationBase::FindDuplicates] "
101 "DUPLICATE POINTER DETECTED");
102
103 ATH_MSG_WARNING(" - Pointer : " << param);
104 ATH_MSG_WARNING(" - Name : " << param->GetName());
105 ATH_MSG_WARNING(" - Class : " << param->ClassName());
106 ATH_MSG_WARNING(" - Occurs in parent : " << this
107 << " (" << this->ClassName() << ")"
108 << " index=" << i);
109
110 if (checkexist->second.replace) {
111 TFCSParametrizationBase *refparam = checkexist->second.replace;
112 ATH_MSG_DEBUG("Found duplicate pointer: "
113 << refparam << "=" << refparam->GetName()
114 << ", duplicate is " << param << "=" << param->GetName()
115 << " index " << i << " of " << this);
116 dup[refparam].mother.push_back(this);
117 dup[refparam].index.push_back(i);
118 }
119 continue;
120 }
121 // Add param to duplication list
122 dup[param] = Duplicate_t();
123 for (auto &ref : dup) {
124 TFCSParametrizationBase *refparam = ref.first;
125 // skip itself, as this just added above
126 if (param == refparam)
127 continue;
128 // skip nullptr reference
129 if (refparam == nullptr)
130 continue;
131 // skip reference that is itself going to get replaced
132 if (ref.second.replace)
133 continue;
134 // Check for objects with identical content
135 if (*param == *refparam) {
136 ATH_MSG_DEBUG("Found duplicate: "
137 << refparam << "=" << refparam->GetName()
138 << ", duplicate is " << param << "=" << param->GetName()
139 << " index " << i << " of " << this);
140
141 dup[param].replace = refparam;
142 dup[refparam].mother.push_back(this);
143 dup[refparam].index.push_back(i);
144 break;
145 }
146 }
147 // Continue for child objects in param
148 param->FindDuplicates(dupclasses);
149 }
150}
void FindDuplicates(FindDuplicateClasses_t &dup)
std::map< TFCSParametrizationBase *, Duplicate_t > FindDuplicates_t
TFCSParametrizationBase(const char *name=nullptr, const char *title=nullptr)

◆ GANfreemem()

bool TFCSEnergyAndHitGAN::GANfreemem ( ) const
inline

Definition at line 41 of file TFCSEnergyAndHitGAN.h.

41{ return TestBit(kGANfreemem); };

◆ get_bin()

virtual int TFCSEnergyAndHitGAN::get_bin ( TFCSSimulationState & simulstate,
const TFCSTruthState * ,
const TFCSExtrapolationState *  ) const
inlineoverridevirtual

use the layer to be done as binning of the GAN chain

Reimplemented from TFCSParametrizationBinnedChain.

Definition at line 57 of file TFCSEnergyAndHitGAN.h.

58 {
59 return simulstate.getAuxInfo<int>("GANlayer"_FCShash);
60 };
const T getAuxInfo(std::uint32_t index) const

◆ get_bin_text()

const std::string TFCSParametrizationBinnedChain::get_bin_text ( int bin) const
virtualinherited

print the range of a bin; for bin -1, print the allowed range

Reimplemented in TFCSParametrizationAbsEtaSelectChain, TFCSParametrizationEkinSelectChain, and TFCSParametrizationEtaSelectChain.

Definition at line 60 of file TFCSParametrizationBinnedChain.cxx.

60 {
61 return std::string(Form("bin %d", bin));
62}

◆ get_Binning()

const Binning & TFCSEnergyAndHitGAN::get_Binning ( ) const
inline

Definition at line 68 of file TFCSEnergyAndHitGAN.h.

68{ return m_Binning; };

◆ get_graph()

const lwt::LightweightGraph * TFCSEnergyAndHitGAN::get_graph ( ) const
inline

Definition at line 69 of file TFCSEnergyAndHitGAN.h.

69{ return m_graph; };

◆ get_input()

const std::string * TFCSEnergyAndHitGAN::get_input ( ) const
inline

Definition at line 70 of file TFCSEnergyAndHitGAN.h.

70{ return m_input; };

◆ get_nr_of_init()

unsigned int TFCSEnergyAndHitGAN::get_nr_of_init ( unsigned int bin) const

Definition at line 69 of file TFCSEnergyAndHitGAN.cxx.

69 {
70 if (bin >= m_bin_ninit.size())
71 return 0;
72 return m_bin_ninit[bin];
73}
std::vector< int > m_bin_ninit

◆ get_number_of_bins()

virtual unsigned int TFCSParametrizationBinnedChain::get_number_of_bins ( ) const
inlinevirtualinherited

Definition at line 22 of file TFCSParametrizationBinnedChain.h.

22 {
23 return m_bin_start.size() - 1;
24 };

◆ get_variable_text()

const std::string TFCSEnergyAndHitGAN::get_variable_text ( TFCSSimulationState & simulstate,
const TFCSTruthState * ,
const TFCSExtrapolationState *  ) const
overridevirtual

Reimplemented from TFCSParametrizationBinnedChain.

Definition at line 233 of file TFCSEnergyAndHitGAN.cxx.

235 {
236 return std::string(
237 Form("layer=%d", simulstate.getAuxInfo<int>("GANlayer"_FCShash)));
238}

◆ GetBinning()

void TFCSEnergyAndHitGAN::GetBinning ( int pid,
int etaMax,
const std::string & FastCaloGANInputFolderName )
protected

Definition at line 143 of file TFCSEnergyAndHitGAN.cxx.

144 {
145 std::string xmlFullFileName = FastCaloGANInputFolderName + "/binning.xml";
146 ATH_MSG_DEBUG("Opening XML file in " << xmlFullFileName);
147
148 std::vector<Binning> AllBinning;
149 std::vector<int> EtaMaxList;
150
151 xmlDocPtr doc = xmlParseFile(xmlFullFileName.c_str());
152 for (xmlNodePtr nodeRoot = doc->children; nodeRoot != nullptr;
153 nodeRoot = nodeRoot->next) {
154 if (xmlStrEqual(nodeRoot->name, BAD_CAST "Bins")) {
155 for (xmlNodePtr nodeBin = nodeRoot->children; nodeBin != nullptr;
156 nodeBin = nodeBin->next) {
157 if (xmlStrEqual(nodeBin->name, BAD_CAST "Bin")) {
158 int nodePid = atof((const char *)xmlGetProp(nodeBin, BAD_CAST "pid"));
159 // int nodeEtaMin = atof( (const char*) xmlGetProp( nodeBin, BAD_CAST
160 // "etaMin" ) );
161 int nodeEtaMax =
162 atof((const char *)xmlGetProp(nodeBin, BAD_CAST "etaMax"));
163
164 Binning binsInLayer;
165 bool correctentry = true;
166 if (nodePid != pid)
167 correctentry = false;
168
169 for (xmlNodePtr nodeLayer = nodeBin->children; nodeLayer != nullptr;
170 nodeLayer = nodeLayer->next) {
171 if (xmlStrEqual(nodeLayer->name, BAD_CAST "Layer")) {
172 std::vector<double> edges;
173 std::string s(
174 (const char *)xmlGetProp(nodeLayer, BAD_CAST "r_edges"));
175
176 std::istringstream ss(s);
177 std::string token;
178
179 while (std::getline(ss, token, ',')) {
180 edges.push_back(atof(token.c_str()));
181 }
182
183 int binsInAlpha = atof(
184 (const char *)xmlGetProp(nodeLayer, BAD_CAST "n_bin_alpha"));
185 int layer =
186 atof((const char *)xmlGetProp(nodeLayer, BAD_CAST "id"));
187
188 if (correctentry)
189 ATH_MSG_DEBUG("nodepid=" << nodePid << " nodeEtaMax="
190 << nodeEtaMax << " Layer: " << layer
191 << " binsInAlpha: " << binsInAlpha
192 << " edges: " << s);
193
194 std::string name = "hist_pid_" + std::to_string(nodePid) +
195 "_etaSliceNumber_" +
196 std::to_string(EtaMaxList.size()) + "_layer_" +
197 std::to_string(layer);
198 int xBins = edges.size() - 1;
199 if (xBins == 0) {
200 xBins = 1; // remove warning
201 edges.push_back(0);
202 edges.push_back(1);
203 }
204 binsInLayer[layer] =
205 TH2D(name.c_str(), name.c_str(), xBins, &edges[0],
206 binsInAlpha, -TMath::Pi(), TMath::Pi());
207 binsInLayer[layer].SetDirectory(nullptr);
208 }
209 }
210
211 if (!correctentry)
212 continue;
213 AllBinning.push_back(std::move(binsInLayer));
214 EtaMaxList.push_back(nodeEtaMax);
215 }
216 }
217 }
218 }
219
220 int index = 0;
221 for (int etaMax : EtaMaxList) {
222 if (etaMid < etaMax) {
223 m_Binning = AllBinning[index];
224 break;
225 }
226 index++;
227 }
228 xmlFreeDoc(doc);
229 ATH_MSG_DEBUG("Done XML file");
230}
static Double_t ss
double atof(std::string_view str)
Converts a string into a double / float.
str index
Definition DeMoScan.py:362

◆ initializeNetwork()

bool TFCSEnergyAndHitGAN::initializeNetwork ( int pid,
int etaMin,
const std::string & FastCaloGANInputFolderName )

Definition at line 84 of file TFCSEnergyAndHitGAN.cxx.

85 {
86
87 // initialize all necessary constants
88 // FIXME eventually all these could be stored in the .json file
89
91 "Using FastCaloGANInputFolderName: " << FastCaloGANInputFolderName);
92 // get neural net JSON file as an std::istream object
93 int etaMax = etaMin + 5;
94
96 set_pdgid(pid);
97 if (pid == 11)
98 add_pdgid(-pid);
99 if (pid == 211)
100 add_pdgid(-pid);
101 set_eta_min(etaMin / 100.0);
102 set_eta_max(etaMax / 100.0);
103 set_eta_nominal((etaMin + etaMax) / 200.0);
104
105 std::string inputFile =
106 FastCaloGANInputFolderName + "/neural_net_" + std::to_string(pid) +
107 "_eta_" + std::to_string(etaMin) + "_" + std::to_string(etaMax) + ".json";
108 if (inputFile.empty()) {
109 ATH_MSG_ERROR("Could not find json file " << inputFile);
110 return false;
111 } else {
112 ATH_MSG_INFO("For pid: " << pid << " and eta " << etaMin << "-" << etaMax
113 << ", loading json file " << inputFile);
114 std::ifstream input(inputFile);
115 std::stringstream sin;
116 sin << input.rdbuf();
117 input.close();
118 // build the graph
119 auto config = lwt::parse_json_graph(sin);
120 m_graph = new lwt::LightweightGraph(config);
121 if (m_graph == nullptr) {
122 ATH_MSG_ERROR("Could not create LightWeightGraph from " << inputFile);
123 return false;
124 }
125 if (m_input != nullptr) {
126 delete m_input;
127 }
128 m_input = new std::string(sin.str());
129 }
130 m_GANLatentSize = 50;
131
132 // Get all Binning histograms to store in memory
133 GetBinning(pid, (etaMin + etaMax) / 2, FastCaloGANInputFolderName);
134
135 if (m_GANLatentSize == 0) {
136 ATH_MSG_ERROR("m_GANLatentSize uninitialized!");
137 return false;
138 }
139
140 return true;
141}
#define ATH_MSG_INFO(x)
void GetBinning(int pid, int etaMax, const std::string &FastCaloGANInputFolderName)
virtual void set_eta_max(double max)
virtual void add_pdgid(int id)
virtual void set_eta_nominal(double min)
virtual void set_pdgid(int id)
virtual void set_eta_min(double min)

◆ is_match_all_calosample()

virtual bool TFCSEnergyAndHitGAN::is_match_all_calosample ( ) const
inlineoverridevirtual

Reimplemented from TFCSParametrization.

Definition at line 33 of file TFCSEnergyAndHitGAN.h.

33{ 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.

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

◆ is_match_all_Ekin_bin()

virtual bool TFCSEnergyAndHitGAN::is_match_all_Ekin_bin ( ) const
inlineoverridevirtual

Reimplemented from TFCSParametrization.

Definition at line 32 of file TFCSEnergyAndHitGAN.h.

32{ return true; };

◆ is_match_all_eta()

virtual bool TFCSParametrization::is_match_all_eta ( ) const
inlineoverridevirtualinherited

Reimplemented from TFCSParametrizationBase.

Definition at line 28 of file TFCSParametrization.h.

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

◆ is_match_all_pdgid()

virtual bool TFCSParametrizationBase::is_match_all_pdgid ( ) const
inlinevirtualinherited

Definition at line 66 of file TFCSParametrizationBase.h.

66{ return TestBit(kMatchAllPDGID); };

◆ is_match_calosample()

bool TFCSEnergyAndHitGAN::is_match_calosample ( int calosample) const
overridevirtual

Reimplemented from TFCSParametrizationChain.

Definition at line 61 of file TFCSEnergyAndHitGAN.cxx.

61 {
62 if (get_Binning().find(calosample) == get_Binning().cend())
63 return false;
64 if (get_Binning().at(calosample).GetNbinsX() == 1)
65 return false;
66 return true;
67}
const Binning & get_Binning() const
std::string find(const std::string &s)
return a remapped string
Definition hcg.cxx:138

◆ is_match_Ekin()

virtual bool TFCSParametrization::is_match_Ekin ( float Ekin) const
inlineoverridevirtualinherited

Reimplemented from TFCSParametrizationBase.

Definition at line 18 of file TFCSParametrization.h.

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

◆ is_match_Ekin_bin()

virtual bool TFCSEnergyAndHitGAN::is_match_Ekin_bin ( int ) const
inlineoverridevirtual

Reimplemented from TFCSParametrizationChain.

Definition at line 28 of file TFCSEnergyAndHitGAN.h.

28 {
29 return true;
30 };

◆ is_match_eta()

virtual bool TFCSParametrization::is_match_eta ( float eta) const
inlineoverridevirtualinherited

Reimplemented from TFCSParametrizationBase.

Definition at line 21 of file TFCSParametrization.h.

21 {
22 return (eta >= m_eta_min) && (eta < m_eta_max);
23 };
Scalar eta() const
pseudorapidity method

◆ is_match_pdgid()

virtual bool TFCSParametrization::is_match_pdgid ( int id) const
inlineoverridevirtualinherited

Reimplemented from TFCSParametrizationBase.

Definition at line 15 of file TFCSParametrization.h.

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

◆ level()

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

Retrieve output level.

Definition at line 201 of file MLogging.h.

201{ return msg().level(); }
MsgStream & msg() const
Return a stream for sending messages directly (no decoration)
Definition MLogging.h:231

◆ 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.

231 {
232 MsgStream *ms = m_msg_tls.get();
233 if (!ms) {
234 ms = new MsgStream(Athena::getMessageSvc(), m_nm);
235 m_msg_tls.reset(ms);
236 }
237 return *ms;
238}
std::string m_nm
Message source name.
Definition MLogging.h:211
IMessageSvc * getMessageSvc(bool quiet=false)

◆ 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.

240 {
241 return msg() << lvl;
242}

◆ 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.

222 {
223 if (msg().level() <= lvl) {
224 msg() << lvl;
225 return true;
226 } else {
227 return false;
228 }
229}
MSG::Level level() const
Retrieve output level.
Definition MLogging.h:201

◆ OnlyScaleEnergy()

bool TFCSEnergyAndHitGAN::OnlyScaleEnergy ( ) const
inline

Definition at line 52 of file TFCSEnergyAndHitGAN.h.

52{ return TestBit(kOnlyScaleEnergy); };

◆ operator==()

virtual bool TFCSParametrizationBase::operator== ( const TFCSParametrizationBase & ref) const
inlinevirtualinherited

The == operator compares the content of instances.

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

Reimplemented in TFCSHitCellMapping, TFCSHitCellMappingWiggle, TFCSLateralShapeParametrizationHitNumberFromE, and TFCSPredictExtrapWeights.

Definition at line 119 of file TFCSParametrizationBase.h.

119 {
120 return compare(ref);
121 };
bool compare(const TFCSParametrizationBase &ref) const
Do not persistify!

◆ operator[]() [1/2]

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

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

Reimplemented from TFCSParametrizationBase.

Definition at line 43 of file TFCSParametrizationChain.h.

43 {
44 return m_chain[ind];
45 };

◆ operator[]() [2/2]

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

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

Reimplemented from TFCSParametrizationBase.

Definition at line 46 of file TFCSParametrizationChain.h.

46 {
47 return m_chain[ind];
48 };

◆ pdgid()

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

Reimplemented from TFCSParametrizationBase.

Definition at line 34 of file TFCSParametrization.h.

34{ return m_pdgid; };

◆ Print()

void TFCSEnergyAndHitGAN::Print ( Option_t * option = "") const
overridevirtual

Definition at line 611 of file TFCSEnergyAndHitGAN.cxx.

611 {
613 TString opt(option);
614 bool shortprint = opt.Index("short") >= 0;
615 bool longprint = msgLvl(MSG::DEBUG) || (msgLvl(MSG::INFO) && !shortprint);
616 TString optprint = opt;
617 optprint.ReplaceAll("short", "");
618
619 if (longprint) {
620 ATH_MSG_INFO(optprint << " "
621 << "Graph=" << CxxUtils::as_const_ptr(m_graph)
622 << "; json input=" << CxxUtils::as_const_ptr(m_input)
623 << "; free mem=" << GANfreemem()
624 << "; latent space=" << m_GANLatentSize
625 << "; Binning size=" << m_Binning.size());
626 for (const auto &l : m_Binning)
627 if (is_match_calosample(l.first)) {
628 ATH_MSG_INFO(optprint << " "
629 << "layer=" << l.first
630 << " nR=" << l.second.GetNbinsX()
631 << " nalpha=" << l.second.GetNbinsY());
632 }
633 }
634
635 TString prefix = "- ";
636 for (unsigned int ichain = 0; ichain < size(); ++ichain) {
637 if (ichain == 0 && ichain != m_bin_start.front()) {
638 prefix = "> ";
639 if (longprint)
640 ATH_MSG_INFO(optprint << prefix << "Run for all bins");
641 }
642 for (unsigned int ibin = 0; ibin < get_number_of_bins(); ++ibin) {
643 if (ichain == m_bin_start[ibin]) {
644 if (ibin < get_number_of_bins() - 1)
645 if (ichain == m_bin_start[ibin + 1])
646 continue;
647 prefix = Form("%-2d", ibin);
648 if (longprint)
649 ATH_MSG_INFO(optprint << prefix << "Run for " << get_bin_text(ibin));
650 }
651 }
652 if (ichain == m_bin_start.back()) {
653 prefix = "< ";
654 if (longprint)
655 ATH_MSG_INFO(optprint << prefix << "Run for all bins");
656 }
657 chain()[ichain]->Print(opt + prefix);
658 }
659}
bool msgLvl(const MSG::Level lvl) const
Check whether the logging system is active at the provided verbosity level.
Definition MLogging.h:222
virtual bool is_match_calosample(int calosample) const override
void Print(Option_t *option="") const
Print object information.
const T * as_const_ptr(const T *p)
Helper for getting a const version of a pointer.
l
Printing final latex table to .tex output file.

◆ push_back()

void TFCSParametrizationChain::push_back ( const Chain_t::value_type & param)
inlineinherited

Definition at line 55 of file TFCSParametrizationChain.h.

55 {
56 m_chain.push_back(param);
57 recalc();
58 };
virtual void recalc()
Default is to call recalc_pdgid_intersect() and recalc_Ekin_eta_intersect()

◆ push_back_in_bin()

void TFCSParametrizationBinnedChain::push_back_in_bin ( TFCSParametrizationBase * param,
unsigned int bin )
virtualinherited

Reimplemented in TFCSParametrizationEkinSelectChain, TFCSParametrizationEtaSelectChain, and TFCSParametrizationFloatSelectChain.

Definition at line 31 of file TFCSParametrizationBinnedChain.cxx.

32 {
33 if (m_bin_start.size() <= bin + 1) {
34 m_bin_start.resize(bin + 2, m_bin_start.back());
35 m_bin_start.shrink_to_fit();
36 }
37 if (m_bin_start[bin + 1] == size()) {
38 chain().push_back(param);
39 } else {
40 Chain_t::iterator it(&chain()[m_bin_start[bin + 1]]);
41 chain().insert(it, param);
42 }
43 for (unsigned int ibin = bin + 1; ibin < m_bin_start.size(); ++ibin)
44 m_bin_start[ibin]++;
45 recalc();
46}

◆ push_before_first_bin()

void TFCSParametrizationBinnedChain::push_before_first_bin ( TFCSParametrizationBase * param)
virtualinherited

Definition at line 18 of file TFCSParametrizationBinnedChain.cxx.

19 {
20 if (m_bin_start[0] == size()) {
21 chain().push_back(param);
22 } else {
23 Chain_t::iterator it(&chain()[m_bin_start[0]]);
24 chain().insert(it, param);
25 }
26 for (auto &ele : m_bin_start)
27 ele++;
28 recalc();
29}

◆ recalc()

void TFCSParametrizationChain::recalc ( )
protectedvirtualinherited

◆ recalc_Ekin_eta_intersect()

void TFCSParametrizationChain::recalc_Ekin_eta_intersect ( )
protectedinherited

◆ recalc_Ekin_eta_union()

void TFCSParametrizationChain::recalc_Ekin_eta_union ( )
protectedinherited

◆ recalc_Ekin_intersect()

void TFCSParametrizationChain::recalc_Ekin_intersect ( )
protectedinherited

Definition at line 45 of file TFCSParametrizationChain.cxx.

45 {
46 set_Ekin(*m_chain[0]);
47
48 for (const auto &param : m_chain) {
49 if (param->Ekin_min() > Ekin_min())
50 set_Ekin_min(param->Ekin_min());
51 if (param->Ekin_max() < Ekin_max())
52 set_Ekin_max(param->Ekin_max());
53 if (Ekin_nominal() < Ekin_min() || Ekin_nominal() > Ekin_max())
54 set_Ekin_nominal(param->Ekin_nominal());
55 }
56
57 if (Ekin_nominal() < Ekin_min() || Ekin_nominal() > Ekin_max())
58 set_Ekin_nominal(0.5 * (Ekin_min() + Ekin_max()));
59}
virtual void set_Ekin_max(double max)
virtual void set_Ekin_min(double min)
virtual void set_Ekin(const TFCSParametrizationBase &ref)
virtual void set_Ekin_nominal(double min)

◆ recalc_Ekin_union()

void TFCSParametrizationChain::recalc_Ekin_union ( )
protectedinherited

Definition at line 82 of file TFCSParametrizationChain.cxx.

82 {
83 set_Ekin(*m_chain[0]);
84
85 for (const auto &param : m_chain) {
86 if (param->Ekin_min() < Ekin_min())
87 set_Ekin_min(param->Ekin_min());
88 if (param->Ekin_max() > Ekin_max())
89 set_Ekin_max(param->Ekin_max());
90 if (Ekin_nominal() < Ekin_min() || Ekin_nominal() > Ekin_max())
91 set_Ekin_nominal(param->Ekin_nominal());
92 }
93
94 if (Ekin_nominal() < Ekin_min() || Ekin_nominal() > Ekin_max())
95 set_Ekin_nominal(0.5 * (Ekin_min() + Ekin_max()));
96}

◆ recalc_eta_intersect()

void TFCSParametrizationChain::recalc_eta_intersect ( )
protectedinherited

Definition at line 61 of file TFCSParametrizationChain.cxx.

61 {
62 set_eta(*m_chain[0]);
63
64 for (const auto &param : m_chain) {
65 if (param->eta_min() > eta_min())
66 set_eta_min(param->eta_min());
67 if (param->eta_max() < eta_max())
68 set_eta_max(param->eta_max());
69 if (eta_nominal() < eta_min() || eta_nominal() > eta_max())
70 set_eta_nominal(param->eta_nominal());
71 }
72
73 if (eta_nominal() < eta_min() || eta_nominal() > eta_max())
74 set_eta_nominal(0.5 * (eta_min() + eta_max()));
75}
virtual void set_eta(const TFCSParametrizationBase &ref)

◆ recalc_eta_union()

void TFCSParametrizationChain::recalc_eta_union ( )
protectedinherited

Definition at line 98 of file TFCSParametrizationChain.cxx.

98 {
99 set_eta(*m_chain[0]);
100
101 for (const auto &param : m_chain) {
102 if (param->eta_min() < eta_min())
103 set_eta_min(param->eta_min());
104 if (param->eta_max() > eta_max())
105 set_eta_max(param->eta_max());
106 if (eta_nominal() < eta_min() || eta_nominal() > eta_max())
107 set_eta_nominal(param->eta_nominal());
108 }
109
110 if (eta_nominal() < eta_min() || eta_nominal() > eta_max())
111 set_eta_nominal(0.5 * (eta_min() + eta_max()));
112}

◆ recalc_pdgid_intersect()

void TFCSParametrizationChain::recalc_pdgid_intersect ( )
protectedinherited

Definition at line 20 of file TFCSParametrizationChain.cxx.

20 {
21 set_pdgid(m_chain[0]->pdgid());
22
23 for (const auto &param : m_chain) {
24 std::set<int> tmp;
25
26 std::set_intersection(pdgid().begin(), pdgid().end(),
27 param->pdgid().begin(), param->pdgid().end(),
28 std::inserter(tmp, tmp.begin()));
29 set_pdgid(tmp);
30 }
31}

◆ recalc_pdgid_union()

void TFCSParametrizationChain::recalc_pdgid_union ( )
protectedinherited

Definition at line 33 of file TFCSParametrizationChain.cxx.

33 {
34 set_pdgid(chain()[0]->pdgid());
35
36 for (const auto &param : chain()) {
37 std::set<int> tmp;
38
39 std::set_union(pdgid().begin(), pdgid().end(), param->pdgid().begin(),
40 param->pdgid().end(), std::inserter(tmp, tmp.begin()));
41 set_pdgid(tmp);
42 }
43}

◆ RemoveDuplicates()

void TFCSParametrizationBase::RemoveDuplicates ( )
inherited

Definition at line 152 of file TFCSParametrizationBase.cxx.

152 {
153 FindDuplicateClasses_t dupclasses;
154 FindDuplicates(dupclasses);
155
156 std::set<TFCSParametrizationBase *> dellist;
157 for (auto &dupiter : dupclasses) {
158 FindDuplicates_t &dup = dupiter.second;
159 for (auto onedup : dup) {
160 if (onedup.second.mother.empty())
161 continue;
162 TFCSParametrizationBase *ref = onedup.first;
163 ATH_MSG_DEBUG("Main object " << ref << "=" << ref->GetName());
164 for (unsigned int i = 0; i < onedup.second.mother.size(); ++i) {
165 int index = onedup.second.index[i];
166 TFCSParametrizationBase *mother = onedup.second.mother[i];
167 TFCSParametrizationBase *delparam = mother->operator[](index);
168 unsigned int delcount = dup[delparam].mother.size();
169 if (delcount == 0) {
170 ATH_MSG_DEBUG(" - Delete object "
171 << delparam << "=" << delparam->GetName() << " index "
172 << index << " of " << mother << ", has " << delcount
173 << " other replacements attached. Deleting");
174 mother->set_daughter(index, ref);
175 dellist.insert(delparam);
176 } else {
177 ATH_MSG_WARNING(" - Delete object "
178 << delparam << "=" << delparam->GetName() << " index "
179 << index << " of " << mother << ", has " << delcount
180 << " other replacements attached. Skipping");
181 }
182 }
183 }
184 }
185
186 ATH_MSG_INFO("RERUNNING DUPLICATE FINDING");
187 FindDuplicateClasses_t dupclasses2;
188 FindDuplicates(dupclasses2);
189
190 std::map<std::string, int> ndel;
191 for (auto *delparam : dellist) {
192 FindDuplicates_t &dup2 = dupclasses2[delparam->GetName()];
193 bool present = dup2.find(delparam) != dup2.end();
194 if (present) {
195 ATH_MSG_WARNING("- Delete object " << delparam << "="
196 << delparam->GetName()
197 << " still referenced somewhere!");
198 } else {
199 ATH_MSG_DEBUG("- Delete object " << delparam << "="
200 << delparam->GetName());
201 ++ndel[delparam->ClassName()];
202 delete delparam;
203 }
204 }
205 for (auto &del : ndel)
206 ATH_MSG_INFO("Deleted " << del.second << " duplicate objects of class "
207 << del.first);
208}
virtual void set_daughter(unsigned int, TFCSParametrizationBase *)
Some derived classes have daughter instances of TFCSParametrizationBase objects The set_daughter meth...
std::map< std::string, FindDuplicates_t > FindDuplicateClasses_t

◆ RemoveNameTitle()

void TFCSParametrizationBase::RemoveNameTitle ( )
inherited

Definition at line 210 of file TFCSParametrizationBase.cxx.

210 {
211 for (unsigned int i = 0; i < size(); ++i)
212 if ((*this)[i]) {
213 TFCSParametrizationBase *param = (*this)[i];
214 param->SetName("");
215 param->SetTitle("");
216
217 // Continue for child objects in param
218 param->RemoveNameTitle();
219 }
220}

◆ reset_GANfreemem()

void TFCSEnergyAndHitGAN::reset_GANfreemem ( )
inline

Definition at line 43 of file TFCSEnergyAndHitGAN.h.

43{ ResetBit(kGANfreemem); };

◆ reset_match_all_pdgid()

virtual void TFCSParametrizationBase::reset_match_all_pdgid ( )
inlinevirtualinherited

Definition at line 84 of file TFCSParametrizationBase.h.

84{ ResetBit(kMatchAllPDGID); };

◆ reset_OnlyScaleEnergy()

void TFCSEnergyAndHitGAN::reset_OnlyScaleEnergy ( )
inline

Definition at line 54 of file TFCSEnergyAndHitGAN.h.

54{ ResetBit(kOnlyScaleEnergy); };

◆ reset_RetryChainFromStart()

void TFCSParametrizationChain::reset_RetryChainFromStart ( )
inlineinherited

Definition at line 38 of file TFCSParametrizationChain.h.

38{ ResetBit(kRetryChainFromStart); };

◆ reset_SplitChainObjects()

void TFCSParametrizationChain::reset_SplitChainObjects ( )
inlineinherited

Definition at line 34 of file TFCSParametrizationChain.h.

34{ ResetBit(kSplitChainObjects); };

◆ RetryChainFromStart()

bool TFCSParametrizationChain::RetryChainFromStart ( ) const
inlineinherited

Definition at line 36 of file TFCSParametrizationChain.h.

36{ return TestBit(kRetryChainFromStart); };

◆ set_daughter()

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

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 from TFCSParametrizationBase.

Definition at line 49 of file TFCSParametrizationChain.h.

50 {
51 m_chain.at(ind) = param;
52 };

◆ set_Ekin()

void TFCSParametrization::set_Ekin ( const TFCSParametrizationBase & ref)
virtualinherited

Definition at line 55 of file TFCSParametrization.cxx.

55 {
56 set_Ekin_nominal(ref.Ekin_nominal());
57 set_Ekin_min(ref.Ekin_min());
58 set_Ekin_max(ref.Ekin_max());
59}

◆ set_Ekin_eta()

void TFCSParametrization::set_Ekin_eta ( const TFCSParametrizationBase & ref)
virtualinherited

Definition at line 67 of file TFCSParametrization.cxx.

67 {
69 set_eta(ref);
70}

◆ set_Ekin_max()

void TFCSParametrization::set_Ekin_max ( double max)
virtualinherited

Definition at line 45 of file TFCSParametrization.cxx.

45{ m_Ekin_max = max; }
#define max(a, b)
Definition cfImp.cxx:41

◆ set_Ekin_min()

void TFCSParametrization::set_Ekin_min ( double min)
virtualinherited

Definition at line 43 of file TFCSParametrization.cxx.

43{ m_Ekin_min = min; }
#define min(a, b)
Definition cfImp.cxx:40

◆ set_Ekin_nominal()

void TFCSParametrization::set_Ekin_nominal ( double min)
virtualinherited

Definition at line 39 of file TFCSParametrization.cxx.

39 {
40 m_Ekin_nominal = nominal;
41}

◆ set_eta()

void TFCSParametrization::set_eta ( const TFCSParametrizationBase & ref)
virtualinherited

Definition at line 61 of file TFCSParametrization.cxx.

61 {
62 set_eta_nominal(ref.eta_nominal());
63 set_eta_min(ref.eta_min());
64 set_eta_max(ref.eta_max());
65}

◆ set_eta_max()

void TFCSParametrization::set_eta_max ( double max)
virtualinherited

Definition at line 53 of file TFCSParametrization.cxx.

53{ m_eta_max = max; }

◆ set_eta_min()

void TFCSParametrization::set_eta_min ( double min)
virtualinherited

Definition at line 51 of file TFCSParametrization.cxx.

51{ m_eta_min = min; }

◆ set_eta_nominal()

void TFCSParametrization::set_eta_nominal ( double min)
virtualinherited

Definition at line 47 of file TFCSParametrization.cxx.

47 {
48 m_eta_nominal = nominal;
49}

◆ set_GANfreemem()

void TFCSEnergyAndHitGAN::set_GANfreemem ( )
inline

Definition at line 42 of file TFCSEnergyAndHitGAN.h.

42{ SetBit(kGANfreemem); };

◆ 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.

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

◆ set_match_all_pdgid()

virtual void TFCSParametrizationBase::set_match_all_pdgid ( )
inlinevirtualinherited

Definition at line 83 of file TFCSParametrizationBase.h.

83{ SetBit(kMatchAllPDGID); };

◆ set_nr_of_init()

void TFCSEnergyAndHitGAN::set_nr_of_init ( unsigned int bin,
unsigned int ninit )

Definition at line 75 of file TFCSEnergyAndHitGAN.cxx.

75 {
76 if (bin >= m_bin_ninit.size()) {
77 m_bin_ninit.resize(bin + 1, 0);
78 m_bin_ninit.shrink_to_fit();
79 }
80 m_bin_ninit[bin] = ninit;
81}

◆ set_OnlyScaleEnergy()

void TFCSEnergyAndHitGAN::set_OnlyScaleEnergy ( )
inline

Definition at line 53 of file TFCSEnergyAndHitGAN.h.

53{ SetBit(kOnlyScaleEnergy); };

◆ 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.

◆ set_pdgid() [2/2]

void TFCSParametrization::set_pdgid ( int id)
virtualinherited

Reimplemented in TFCSEnergyBinParametrization.

Definition at line 28 of file TFCSParametrization.cxx.

28 {
29 m_pdgid.clear();
30 m_pdgid.insert(id);
31}

◆ set_pdgid_Ekin_eta()

void TFCSParametrization::set_pdgid_Ekin_eta ( const TFCSParametrizationBase & ref)
virtualinherited

Definition at line 72 of file TFCSParametrization.cxx.

73 {
75 set_pdgid(ref.pdgid());
76}
virtual void set_Ekin_eta(const TFCSParametrizationBase &ref)

◆ set_RetryChainFromStart()

void TFCSParametrizationChain::set_RetryChainFromStart ( )
inlineinherited

Definition at line 37 of file TFCSParametrizationChain.h.

37{ SetBit(kRetryChainFromStart); };

◆ set_SplitChainObjects()

void TFCSParametrizationChain::set_SplitChainObjects ( )
inlineinherited

Definition at line 33 of file TFCSParametrizationChain.h.

33{ SetBit(kSplitChainObjects); };

◆ setLevel()

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

Update outputlevel.

Definition at line 105 of file MLogging.cxx.

105 {
106 lvl = (lvl >= MSG::NUM_LEVELS) ? MSG::ALWAYS
107 : (lvl < MSG::NIL) ? MSG::NIL
108 : lvl;
109 msg().setLevel(lvl);
110}

◆ simulate()

FCSReturnCode TFCSEnergyAndHitGAN::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 TFCSParametrizationBinnedChain.

Definition at line 580 of file TFCSEnergyAndHitGAN.cxx.

582 {
583 for (unsigned int ichain = 0; ichain < m_bin_start[0]; ++ichain) {
584 ATH_MSG_DEBUG("now run for all bins: " << chain()[ichain]->GetName());
585 if (simulate_and_retry(chain()[ichain], simulstate, truth, extrapol) !=
586 FCSSuccess) {
587 return FCSFatal;
588 }
589 }
590
591 // Compute all inputs to the network
593 double trueEnergy;
594 ATH_MSG_VERBOSE("Get Inputs");
595 if (!fillFastCaloGanNetworkInputs(simulstate, truth, inputs, trueEnergy)) {
596 ATH_MSG_WARNING("Could not initialize network ");
597 // bail out but do not stop the job
598 return FCSSuccess;
599 }
600
601 ATH_MSG_VERBOSE("Fill Energies");
602 if (!fillEnergy(simulstate, truth, extrapol, std::move(inputs))) {
603 ATH_MSG_WARNING("Could not fill energies ");
604 // bail out but do not stop the job
605 return FCSSuccess;
606 }
607
608 return FCSSuccess;
609}
bool fillEnergy(TFCSSimulationState &simulstate, const TFCSTruthState *truth, const TFCSExtrapolationState *extrapol, NetworkInputs inputs) const
std::map< std::string, std::map< std::string, double > > NetworkInputs
bool fillFastCaloGanNetworkInputs(TFCSSimulationState &simulstate, const TFCSTruthState *truth, NetworkInputs &inputs, double &trueEnergy) const

◆ simulate_and_retry()

FCSReturnCode TFCSParametrizationChain::simulate_and_retry ( TFCSParametrizationBase * parametrization,
TFCSSimulationState & simulstate,
const TFCSTruthState * truth,
const TFCSExtrapolationState * extrapol ) const
protectedinherited

◆ size()

virtual unsigned int TFCSParametrizationChain::size ( ) const
inlineoverridevirtualinherited

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

Reimplemented from TFCSParametrizationBase.

Definition at line 41 of file TFCSParametrizationChain.h.

41{ return m_chain.size(); };

◆ SplitChainObjects()

bool TFCSParametrizationChain::SplitChainObjects ( ) const
inlineinherited

Definition at line 32 of file TFCSParametrizationChain.h.

32{ return TestBit(kSplitChainObjects); };

◆ 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.

116 {
117 int col1_len = 20;
118 int col2_len = 5;
119 int col3_len = 10;
120 auto last_slash = file.find_last_of('/');
121 int path_len = last_slash == std::string::npos ? 0 : last_slash;
122 int trim_point = path_len;
123 int total_len = file.length();
124 if (total_len - path_len > col1_len)
125 trim_point = total_len - col1_len;
126 std::string trimmed_name = file.substr(trim_point);
127 const char *LevelNames[MSG::NUM_LEVELS] = {
128 "NIL", "VERBOSE", "DEBUG", "INFO", "WARNING", "ERROR", "FATAL", "ALWAYS"};
129 std::string level = LevelNames[lvl];
130 std::string level_string = std::string("(") + level + ") ";
131 std::stringstream output;
132 output << std::setw(col1_len) << std::right << trimmed_name << ":"
133 << std::setw(col2_len) << std::left << line << std::setw(col3_len)
134 << std::right << level_string;
135 return output.str();
136}
output
Definition merge.py:16
TFile * file

◆ unit_test()

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

Definition at line 688 of file TFCSEnergyAndHitGAN.cxx.

690 {
691 if (!simulstate) {
692 simulstate = new TFCSSimulationState();
693#if defined(__FastCaloSimStandAlone__)
694 simulstate->setRandomEngine(new CLHEP::TRandomEngine());
695#else
696 simulstate->setRandomEngine(new CLHEP::RanluxEngine());
697#endif
698 }
699 if (!truth) {
700 TFCSTruthState *t = new TFCSTruthState();
701 t->SetPtEtaPhiM(20000, 0.225, 0, 130);
702 t->set_pdgid(211);
703 truth = t;
704 }
705 if (!extrapol) {
706 TFCSExtrapolationState *e = new TFCSExtrapolationState();
707 e->set_IDCaloBoundary_eta(truth->Eta());
708 for (int i = 0; i < 24; ++i) {
709 e->set_eta(i, TFCSExtrapolationState::SUBPOS_ENT, truth->Eta());
710 e->set_eta(i, TFCSExtrapolationState::SUBPOS_EXT, truth->Eta());
711 e->set_eta(i, TFCSExtrapolationState::SUBPOS_MID, truth->Eta());
715 e->set_r(i, TFCSExtrapolationState::SUBPOS_ENT, 1500 + i * 10);
716 e->set_r(i, TFCSExtrapolationState::SUBPOS_EXT, 1510 + i * 10);
717 e->set_r(i, TFCSExtrapolationState::SUBPOS_MID, 1505 + i * 10);
718 e->set_z(i, TFCSExtrapolationState::SUBPOS_ENT, 3500 + i * 10);
719 e->set_z(i, TFCSExtrapolationState::SUBPOS_EXT, 3510 + i * 10);
720 e->set_z(i, TFCSExtrapolationState::SUBPOS_MID, 3505 + i * 10);
721 }
722 extrapol = e;
723 }
724
725 TFCSEnergyAndHitGAN GAN("GAN", "GAN");
726 GAN.setLevel(MSG::VERBOSE);
727 int pid = 211;
728 int etaMin = 20;
729 int etaMax = etaMin + 5;
730 GAN.initializeNetwork(
731 pid, etaMin,
732 "/eos/atlas/atlascerngroupdisk/proj-simul/VoxalisationOutputs/nominal/"
733 "GAN_michele_normE_MaxE/input_for_service_new");
734 for (int i = 0; i < 24; ++i)
735 if (GAN.is_match_calosample(i)) {
736 TFCSCenterPositionCalculation *c = new TFCSCenterPositionCalculation(
737 Form("center%d", i), Form("center layer %d", i));
738 c->set_calosample(i);
739 c->setExtrapWeight(0.5);
740 c->setLevel(MSG::VERBOSE);
741 c->set_pdgid(pid);
742 if (pid == 11)
743 c->add_pdgid(-pid);
744 if (pid == 211)
745 c->add_pdgid(-pid);
746 c->set_eta_min(etaMin / 100.0);
747 c->set_eta_max(etaMax / 100.0);
748 c->set_eta_nominal((etaMin + etaMax) / 200.0);
749
750 GAN.push_back_in_bin(c, i);
751 GAN.set_nr_of_init(i, 1);
752 }
753
754 GAN.Print();
755
756 TFile *fGAN = TFile::Open("FCSGANtest.root", "recreate");
757 GAN.Write();
758 fGAN->ls();
759 fGAN->Close();
760
761 fGAN = TFile::Open("FCSGANtest.root");
762 TFCSEnergyAndHitGAN *GAN2 = (TFCSEnergyAndHitGAN *)(fGAN->Get("GAN"));
763 GAN2->Print();
764
765 GAN2->setLevel(MSG::DEBUG);
766 GAN2->simulate(*simulstate, truth, extrapol);
767 simulstate->Print();
768}
virtual void setLevel(MSG::Level lvl)
Update outputlevel.
Definition MLogging.cxx:105
virtual FCSReturnCode simulate(TFCSSimulationState &simulstate, const TFCSTruthState *truth, const TFCSExtrapolationState *extrapol) const override
Method in all derived classes to do some simulation.
virtual void Print(Option_t *option="") const override
TFCSEnergyAndHitGAN(const char *name=nullptr, const char *title=nullptr)
void Print(Option_t *option="") const
void setRandomEngine(CLHEP::HepRandomEngine *engine)

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_bin_ninit

std::vector<int> TFCSEnergyAndHitGAN::m_bin_ninit
private

Definition at line 97 of file TFCSEnergyAndHitGAN.h.

◆ m_bin_start

std::vector<unsigned int> TFCSParametrizationBinnedChain::m_bin_start
protectedinherited

Contains the index where the TFCSParametrizationBase* instances to run for a given bin start.

The last entry of the vector correponds to the index from where on TFCSParametrizationBase* objects should be run again for all bins. This way one can loop over some instances for all bins, then only specific ones for one bin and at the end again over some for all bins

Definition at line 53 of file TFCSParametrizationBinnedChain.h.

◆ m_Binning

Binning TFCSEnergyAndHitGAN::m_Binning
private

Do not persistify.

Definition at line 106 of file TFCSEnergyAndHitGAN.h.

◆ m_chain

Chain_t TFCSParametrizationChain::m_chain
privateinherited

Definition at line 99 of file TFCSParametrizationChain.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_GANLatentSize

int TFCSEnergyAndHitGAN::m_GANLatentSize = 0
private

Definition at line 110 of file TFCSEnergyAndHitGAN.h.

◆ m_graph

lwt::LightweightGraph* TFCSEnergyAndHitGAN::m_graph = nullptr
private

Definition at line 104 of file TFCSEnergyAndHitGAN.h.

◆ m_input

std::string* TFCSEnergyAndHitGAN::m_input = nullptr
private

Definition at line 103 of file TFCSEnergyAndHitGAN.h.

◆ m_nm

std::string ISF_FCS::MLogging::m_nm
privateinherited

Message source name.

Definition at line 211 of file MLogging.h.

◆ m_pdgid

std::set<int> TFCSParametrization::m_pdgid
privateinherited

Definition at line 63 of file TFCSParametrization.h.

◆ m_writtenBases

std::vector<std::unique_ptr<TFCSParametrizationBase> > TFCSParametrizationChain::m_writtenBases
privateinherited

Definition at line 102 of file TFCSParametrizationChain.h.


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