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 51 of file TFCSEnergyAndHitGAN.cxx.

52 : TFCSParametrizationBinnedChain(name, title) {
54}
TFCSParametrizationBinnedChain(const char *name=nullptr, const char *title=nullptr)

◆ ~TFCSEnergyAndHitGAN()

TFCSEnergyAndHitGAN::~TFCSEnergyAndHitGAN ( )
virtual

Definition at line 56 of file TFCSEnergyAndHitGAN.cxx.

56 {
57 if (m_input != nullptr) {
58 delete m_input;
59 }
60 if (m_graph != nullptr) {
61 delete m_graph;
62 }
63}
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 256 of file TFCSEnergyAndHitGAN.cxx.

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

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

◆ 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 73 of file TFCSEnergyAndHitGAN.cxx.

73 {
74 if (bin >= m_bin_ninit.size())
75 return 0;
76 return m_bin_ninit[bin];
77}
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 227 of file TFCSEnergyAndHitGAN.cxx.

229 {
230 return std::string(
231 Form("layer=%d", simulstate.getAuxInfo<int>("GANlayer"_FCShash)));
232}

◆ GetBinning()

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

Definition at line 143 of file TFCSEnergyAndHitGAN.cxx.

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

◆ initializeNetwork()

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

Definition at line 88 of file TFCSEnergyAndHitGAN.cxx.

89 {
90
91 // initialize all necessary constants
92 // FIXME eventually all these could be stored in the .json file
93
95 "Using FastCaloGANInputFolderName: " << FastCaloGANInputFolderName);
96 // get neural net JSON file as an std::istream object
97 int etaMax = etaMin + 5;
98
100 set_pdgid(pid);
101 if (pid == 11)
102 add_pdgid(-pid);
103 if (pid == 211)
104 add_pdgid(-pid);
105 set_eta_min(etaMin / 100.0);
106 set_eta_max(etaMax / 100.0);
107 set_eta_nominal((etaMin + etaMax) / 200.0);
108
109 std::string inputFile = std::format("{}/neural_net_{}_eta_{}_{}.json",
110 FastCaloGANInputFolderName, pid, etaMin, etaMax);
111 ATH_MSG_INFO("For pid: " << pid << " and eta " << etaMin << "-" << etaMax
112 << ", loading json file " << inputFile);
113 std::ifstream input(inputFile);
114 if (!input.is_open()) { // check: verify the file actually exists and opened
115 ATH_MSG_ERROR(std::format("Could not open json file: {}", inputFile));
116 return false;
117 }
118 std::stringstream sin;
119 sin << input.rdbuf();
120 input.close();
121 // build the graph
122 auto config = lwt::parse_json_graph(sin);
123 m_graph = new lwt::LightweightGraph(config);
124 assert(m_graph!=nullptr);
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 65 of file TFCSEnergyAndHitGAN.cxx.

65 {
66 if (get_Binning().find(calosample) == get_Binning().cend())
67 return false;
68 if (get_Binning().at(calosample).GetNbinsX() == 1)
69 return false;
70 return true;
71}
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 605 of file TFCSEnergyAndHitGAN.cxx.

605 {
607 TString opt(option);
608 bool shortprint = opt.Index("short") >= 0;
609 bool longprint = msgLvl(MSG::DEBUG) || (msgLvl(MSG::INFO) && !shortprint);
610 TString optprint = opt;
611 optprint.ReplaceAll("short", "");
612
613 if (longprint) {
614 ATH_MSG_INFO(optprint << " "
615 << "Graph=" << CxxUtils::as_const_ptr(m_graph)
616 << "; json input=" << CxxUtils::as_const_ptr(m_input)
617 << "; free mem=" << GANfreemem()
618 << "; latent space=" << m_GANLatentSize
619 << "; Binning size=" << m_Binning.size());
620 for (const auto &l : m_Binning)
621 if (is_match_calosample(l.first)) {
622 ATH_MSG_INFO(optprint << " "
623 << "layer=" << l.first
624 << " nR=" << l.second.GetNbinsX()
625 << " nalpha=" << l.second.GetNbinsY());
626 }
627 }
628
629 TString prefix = "- ";
630 for (unsigned int ichain = 0; ichain < size(); ++ichain) {
631 if (ichain == 0 && ichain != m_bin_start.front()) {
632 prefix = "> ";
633 if (longprint)
634 ATH_MSG_INFO(optprint << prefix << "Run for all bins");
635 }
636 for (unsigned int ibin = 0; ibin < get_number_of_bins(); ++ibin) {
637 if (ichain == m_bin_start[ibin]) {
638 if (ibin < get_number_of_bins() - 1)
639 if (ichain == m_bin_start[ibin + 1])
640 continue;
641 prefix = Form("%-2d", ibin);
642 if (longprint)
643 ATH_MSG_INFO(optprint << prefix << "Run for " << get_bin_text(ibin));
644 }
645 }
646 if (ichain == m_bin_start.back()) {
647 prefix = "< ";
648 if (longprint)
649 ATH_MSG_INFO(optprint << prefix << "Run for all bins");
650 }
651 chain()[ichain]->Print(opt + prefix);
652 }
653}
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 79 of file TFCSEnergyAndHitGAN.cxx.

79 {
80 if (bin >= m_bin_ninit.size()) {
81 m_bin_ninit.resize(bin + 1, 0);
82 m_bin_ninit.shrink_to_fit();
83 }
84 m_bin_ninit[bin] = ninit;
85}

◆ 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 574 of file TFCSEnergyAndHitGAN.cxx.

576 {
577 for (unsigned int ichain = 0; ichain < m_bin_start[0]; ++ichain) {
578 ATH_MSG_DEBUG("now run for all bins: " << chain()[ichain]->GetName());
579 if (simulate_and_retry(chain()[ichain], simulstate, truth, extrapol) !=
580 FCSSuccess) {
581 return FCSFatal;
582 }
583 }
584
585 // Compute all inputs to the network
587 double trueEnergy;
588 ATH_MSG_VERBOSE("Get Inputs");
589 if (!fillFastCaloGanNetworkInputs(simulstate, truth, inputs, trueEnergy)) {
590 ATH_MSG_WARNING("Could not initialize network ");
591 // bail out but do not stop the job
592 return FCSSuccess;
593 }
594
595 ATH_MSG_VERBOSE("Fill Energies");
596 if (!fillEnergy(simulstate, truth, extrapol, std::move(inputs))) {
597 ATH_MSG_WARNING("Could not fill energies ");
598 // bail out but do not stop the job
599 return FCSSuccess;
600 }
601
602 return FCSSuccess;
603}
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 682 of file TFCSEnergyAndHitGAN.cxx.

684 {
685 if (!simulstate) {
686 simulstate = new TFCSSimulationState();
687#if defined(__FastCaloSimStandAlone__)
688 simulstate->setRandomEngine(new CLHEP::TRandomEngine());
689#else
690 simulstate->setRandomEngine(new CLHEP::RanluxEngine());
691#endif
692 }
693 if (!truth) {
694 TFCSTruthState *t = new TFCSTruthState();
695 t->SetPtEtaPhiM(20000, 0.225, 0, 130);
696 t->set_pdgid(211);
697 truth = t;
698 }
699 if (!extrapol) {
700 TFCSExtrapolationState *e = new TFCSExtrapolationState();
701 e->set_IDCaloBoundary_eta(truth->Eta());
702 for (int i = 0; i < 24; ++i) {
703 e->set_eta(i, TFCSExtrapolationState::SUBPOS_ENT, truth->Eta());
704 e->set_eta(i, TFCSExtrapolationState::SUBPOS_EXT, truth->Eta());
705 e->set_eta(i, TFCSExtrapolationState::SUBPOS_MID, truth->Eta());
709 e->set_r(i, TFCSExtrapolationState::SUBPOS_ENT, 1500 + i * 10);
710 e->set_r(i, TFCSExtrapolationState::SUBPOS_EXT, 1510 + i * 10);
711 e->set_r(i, TFCSExtrapolationState::SUBPOS_MID, 1505 + i * 10);
712 e->set_z(i, TFCSExtrapolationState::SUBPOS_ENT, 3500 + i * 10);
713 e->set_z(i, TFCSExtrapolationState::SUBPOS_EXT, 3510 + i * 10);
714 e->set_z(i, TFCSExtrapolationState::SUBPOS_MID, 3505 + i * 10);
715 }
716 extrapol = e;
717 }
718
719 TFCSEnergyAndHitGAN GAN("GAN", "GAN");
720 GAN.setLevel(MSG::VERBOSE);
721 static constexpr int pid = 211;
722 int etaMin = 20;
723 int etaMax = etaMin + 5;
724 GAN.initializeNetwork(
725 pid, etaMin,
726 "/eos/atlas/atlascerngroupdisk/proj-simul/VoxalisationOutputs/nominal/"
727 "GAN_michele_normE_MaxE/input_for_service_new");
728 for (int i = 0; i < 24; ++i)
729 if (GAN.is_match_calosample(i)) {
730 TFCSCenterPositionCalculation *c = new TFCSCenterPositionCalculation(
731 Form("center%d", i), Form("center layer %d", i));
732 c->set_calosample(i);
733 c->setExtrapWeight(0.5);
734 c->setLevel(MSG::VERBOSE);
735 c->set_pdgid(pid);
736 if (pid == 11)
737 //pid was set to 211
738 //coverity[DEADCODE]
739 c->add_pdgid(-pid);
740 if (pid == 211)
741 c->add_pdgid(-pid);
742 c->set_eta_min(etaMin / 100.0);
743 c->set_eta_max(etaMax / 100.0);
744 c->set_eta_nominal((etaMin + etaMax) / 200.0);
745
746 GAN.push_back_in_bin(c, i);
747 GAN.set_nr_of_init(i, 1);
748 }
749
750 GAN.Print();
751
752 TFile *fGAN = TFile::Open("FCSGANtest.root", "recreate");
753 GAN.Write();
754 fGAN->ls();
755 fGAN->Close();
756 delete fGAN;
757 fGAN = TFile::Open("FCSGANtest.root");
758 TFCSEnergyAndHitGAN *GAN2 = (TFCSEnergyAndHitGAN *)(fGAN->Get("GAN"));
759 GAN2->Print();
760
761 GAN2->setLevel(MSG::DEBUG);
762 GAN2->simulate(*simulstate, truth, extrapol);
763 simulstate->Print();
764 delete fGAN;
765}
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: