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

45 : TFCSParametrizationBinnedChain(name, title) {
47}
TFCSParametrizationBinnedChain(const char *name=nullptr, const char *title=nullptr)

◆ ~TFCSEnergyAndHitGAN()

TFCSEnergyAndHitGAN::~TFCSEnergyAndHitGAN ( )
virtual

Definition at line 49 of file TFCSEnergyAndHitGAN.cxx.

49 {
50 if (m_input != nullptr) {
51 delete m_input;
52 }
53 if (m_graph != nullptr) {
54 delete m_graph;
55 }
56}
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 237 of file TFCSEnergyAndHitGAN.cxx.

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

217 {
218 // fill randomize latent space
219 // FIXME: should this really be momentum
220 trueEnergy = truth->P();
221 double randUniformZ = 0.;
222 // FIXME: make dependency on input particle eta, pdgid and energy
223 for (int i = 0; i < m_GANLatentSize; i++) {
224 randUniformZ = CLHEP::RandFlat::shoot(simulstate.randomEngine(), -1., 1.);
225 inputs["node_0"].insert(
226 std::pair<std::string, double>(std::to_string(i), randUniformZ));
227 }
228
229 // ATH_MSG_INFO( "Check label: " <<trueEnergy <<" "<<std::pow(2,22)<<"
230 // "<<trueEnergy/std::pow(2,22));
231 inputs["node_1"].insert(
232 std::pair<std::string, double>("0", trueEnergy / (std::pow(2, 22))));
233
234 return true;
235}

◆ 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)
ClassName()
Default constructor.

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

66 {
67 if (bin >= m_bin_ninit.size())
68 return 0;
69 return m_bin_ninit[bin];
70}
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 208 of file TFCSEnergyAndHitGAN.cxx.

210 {
211 return std::string(
212 Form("layer=%d", simulstate.getAuxInfo<int>("GANlayer"_FCShash)));
213}

◆ GetBinning()

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

Definition at line 136 of file TFCSEnergyAndHitGAN.cxx.

137 {
138 const std::string xmlFullFileName = std::format("{}/binning.xml", FastCaloGANInputFolderName);
139 ATH_MSG_DEBUG("Opening XML file in " << xmlFullFileName);
140
141 std::vector<Binning> AllBinning;
142 std::vector<int> EtaMaxList;
143
144 XMLCoreParser p;
145 std::unique_ptr<XMLCoreNode> doc = p.parse (xmlFullFileName);
146 for (const XMLCoreNode* bin : doc->get_children ("Bins/Bin")) {
147 int nodePid = bin->get_int_attrib ("pid");
148 int nodeEtaMax = bin->get_int_attrib ("etaMax");
149
150 Binning binsInLayer;
151 bool correctentry = true;
152 if (nodePid != pid)
153 correctentry = false;
154
155 for (const XMLCoreNode* nodeLayer : bin->get_children ("Layer")) {
156 std::vector<double> edges;
157 std::string s = nodeLayer->get_attrib ("r_edges");
158 std::istringstream ss(s);
159 std::string token;
160
161 while (std::getline(ss, token, ',')) {
162 edges.push_back(atof(token.c_str()));
163 }
164
165 int binsInAlpha = nodeLayer->get_int_attrib ("n_bin_alpha");
166 int layer = nodeLayer->get_int_attrib ("id");
167
168 if (correctentry)
169 ATH_MSG_DEBUG("nodepid=" << nodePid << " nodeEtaMax="
170 << nodeEtaMax << " Layer: " << layer
171 << " binsInAlpha: " << binsInAlpha
172 << " edges: " << s);
173
174 std::string name = "hist_pid_" + std::to_string(nodePid) +
175 "_etaSliceNumber_" +
176 std::to_string(EtaMaxList.size()) + "_layer_" +
177 std::to_string(layer);
178 int xBins = edges.size() - 1;
179 if (xBins == 0) {
180 xBins = 1; // remove warning
181 edges.push_back(0);
182 edges.push_back(1);
183 }
184 binsInLayer[layer] =
185 TH2D(name.c_str(), name.c_str(), xBins, &edges[0],
186 binsInAlpha, -TMath::Pi(), TMath::Pi());
187 binsInLayer[layer].SetDirectory(nullptr);
188 }
189
190 if (!correctentry)
191 continue;
192 AllBinning.push_back(std::move(binsInLayer));
193 EtaMaxList.push_back(nodeEtaMax);
194 }
195
196 int index = 0;
197 for (int etaMax : EtaMaxList) {
198 if (etaMid < etaMax) {
199 m_Binning = AllBinning[index];
200 break;
201 }
202 index++;
203 }
204 ATH_MSG_DEBUG("Done XML file");
205}
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 81 of file TFCSEnergyAndHitGAN.cxx.

82 {
83
84 // initialize all necessary constants
85 // FIXME eventually all these could be stored in the .json file
86
88 "Using FastCaloGANInputFolderName: " << FastCaloGANInputFolderName);
89 // get neural net JSON file as an std::istream object
90 int etaMax = etaMin + 5;
91
93 set_pdgid(pid);
94 if (pid == 11)
95 add_pdgid(-pid);
96 if (pid == 211)
97 add_pdgid(-pid);
98 set_eta_min(etaMin / 100.0);
99 set_eta_max(etaMax / 100.0);
100 set_eta_nominal((etaMin + etaMax) / 200.0);
101
102 std::string inputFile = std::format("{}/neural_net_{}_eta_{}_{}.json",
103 FastCaloGANInputFolderName, pid, etaMin, etaMax);
104 ATH_MSG_INFO("For pid: " << pid << " and eta " << etaMin << "-" << etaMax
105 << ", loading json file " << inputFile);
106 std::ifstream input(inputFile);
107 if (!input.is_open()) { // check: verify the file actually exists and opened
108 ATH_MSG_ERROR(std::format("Could not open json file: {}", inputFile));
109 return false;
110 }
111 std::stringstream sin;
112 sin << input.rdbuf();
113 input.close();
114 // build the graph
115 auto config = lwt::parse_json_graph(sin);
116 m_graph = new lwt::LightweightGraph(config);
117 assert(m_graph!=nullptr);
118 if (m_input != nullptr) {
119 delete m_input;
120 }
121 m_input = new std::string(sin.str());
122
123 m_GANLatentSize = 50;
124
125 // Get all Binning histograms to store in memory
126 GetBinning(pid, (etaMin + etaMax) / 2, FastCaloGANInputFolderName);
127
128 if (m_GANLatentSize == 0) {
129 ATH_MSG_ERROR("m_GANLatentSize uninitialized!");
130 return false;
131 }
132
133 return true;
134}
#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 58 of file TFCSEnergyAndHitGAN.cxx.

58 {
59 if (get_Binning().find(calosample) == get_Binning().cend())
60 return false;
61 if (get_Binning().at(calosample).GetNbinsX() == 1)
62 return false;
63 return true;
64}
const Binning & get_Binning() const
std::string find(const std::string &s)
return a remapped string
Definition hcg.cxx:140
const_iterator cend() const
Iterator at the end of the map.
mapped_type at(key_type key) const
Look up an element in the map.

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

586 {
588 TString opt(option);
589 bool shortprint = opt.Index("short") >= 0;
590 bool longprint = msgLvl(MSG::DEBUG) || (msgLvl(MSG::INFO) && !shortprint);
591 TString optprint = opt;
592 optprint.ReplaceAll("short", "");
593
594 if (longprint) {
595 ATH_MSG_INFO(optprint << " "
596 << "Graph=" << CxxUtils::as_const_ptr(m_graph)
597 << "; json input=" << CxxUtils::as_const_ptr(m_input)
598 << "; free mem=" << GANfreemem()
599 << "; latent space=" << m_GANLatentSize
600 << "; Binning size=" << m_Binning.size());
601 for (const auto &l : m_Binning)
602 if (is_match_calosample(l.first)) {
603 ATH_MSG_INFO(optprint << " "
604 << "layer=" << l.first
605 << " nR=" << l.second.GetNbinsX()
606 << " nalpha=" << l.second.GetNbinsY());
607 }
608 }
609
610 TString prefix = "- ";
611 for (unsigned int ichain = 0; ichain < size(); ++ichain) {
612 if (ichain == 0 && ichain != m_bin_start.front()) {
613 prefix = "> ";
614 if (longprint)
615 ATH_MSG_INFO(optprint << prefix << "Run for all bins");
616 }
617 for (unsigned int ibin = 0; ibin < get_number_of_bins(); ++ibin) {
618 if (ichain == m_bin_start[ibin]) {
619 if (ibin < get_number_of_bins() - 1)
620 if (ichain == m_bin_start[ibin + 1])
621 continue;
622 prefix = Form("%-2d", ibin);
623 if (longprint)
624 ATH_MSG_INFO(optprint << prefix << "Run for " << get_bin_text(ibin));
625 }
626 }
627 if (ichain == m_bin_start.back()) {
628 prefix = "< ";
629 if (longprint)
630 ATH_MSG_INFO(optprint << prefix << "Run for all bins");
631 }
632 chain()[ichain]->Print(opt + prefix);
633 }
634}
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 72 of file TFCSEnergyAndHitGAN.cxx.

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

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

557 {
558 for (unsigned int ichain = 0; ichain < m_bin_start[0]; ++ichain) {
559 ATH_MSG_DEBUG("now run for all bins: " << chain()[ichain]->GetName());
560 if (simulate_and_retry(chain()[ichain], simulstate, truth, extrapol) !=
561 FCSSuccess) {
562 return FCSFatal;
563 }
564 }
565
566 // Compute all inputs to the network
568 double trueEnergy;
569 ATH_MSG_VERBOSE("Get Inputs");
570 if (!fillFastCaloGanNetworkInputs(simulstate, truth, inputs, trueEnergy)) {
571 ATH_MSG_WARNING("Could not initialize network ");
572 // bail out but do not stop the job
573 return FCSSuccess;
574 }
575
576 ATH_MSG_VERBOSE("Fill Energies");
577 if (!fillEnergy(simulstate, truth, extrapol, std::move(inputs))) {
578 ATH_MSG_WARNING("Could not fill energies ");
579 // bail out but do not stop the job
580 return FCSSuccess;
581 }
582
583 return FCSSuccess;
584}
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 663 of file TFCSEnergyAndHitGAN.cxx.

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