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

#include <TFCSEnergyAndHitGANV2.h>

Inheritance diagram for TFCSEnergyAndHitGANV2:
Collaboration diagram for TFCSEnergyAndHitGANV2:

Public Types

enum  FCSGANfreemem { kGANfreemem = BIT(17) }
 Status bit for chain persistency. More...
enum  FCSEnergyInitializationStatusBits { kOnlyScaleEnergy = }
 Status bit for energy initialization. More...
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

 TFCSEnergyAndHitGANV2 (const char *name=nullptr, const char *title=nullptr)
virtual ~TFCSEnergyAndHitGANV2 ()
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 TFCSGANXMLParameters::Binningget_Binning () const
const TFCSGANEtaSlice::ExtrapolatorWeights get_ExtrapolationWeights () const
bool initializeNetwork (int const &pid, int const &etaMin, const std::string &FastCaloGANInputFolderName)
bool fillEnergy (TFCSSimulationState &simulstate, const TFCSTruthState *truth, const TFCSExtrapolationState *extrapol) 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 test_path (const std::string &path, TFCSSimulationState *simulstate=nullptr, const TFCSTruthState *truth=nullptr, const TFCSExtrapolationState *extrapol=nullptr, const std::string &outputname="unnamed", int pid=211)
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 SetRegionAndSliceFromXML (int pid, int etaMax, 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 Member Functions

int GetAlphaBinsForRBin (const TAxis *x, int ix, int yBinNum) const

Static Private Member Functions

static int GetBinsInFours (double const &bins)

Private Attributes

std::vector< int > m_bin_ninit
TFCSGANEtaSlicem_slice = nullptr
TFCSGANXMLParameters m_param
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 20 of file TFCSEnergyAndHitGANV2.h.

Member Typedef Documentation

◆ 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

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 44 of file TFCSEnergyAndHitGANV2.h.

44 {
46 BIT(18)
48 };
@ 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 34 of file TFCSEnergyAndHitGANV2.h.

34 {
35 kGANfreemem = BIT(17)
37 };
@ 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

◆ TFCSEnergyAndHitGANV2()

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

Definition at line 35 of file TFCSEnergyAndHitGANV2.cxx.

37 : TFCSParametrizationBinnedChain(name, title) {
39}
TFCSParametrizationBinnedChain(const char *name=nullptr, const char *title=nullptr)

◆ ~TFCSEnergyAndHitGANV2()

TFCSEnergyAndHitGANV2::~TFCSEnergyAndHitGANV2 ( )
virtual

Definition at line 41 of file TFCSEnergyAndHitGANV2.cxx.

41 {
42 if (m_slice != nullptr) {
43 delete m_slice;
44 }
45}

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 TFCSEnergyAndHitGANV2::fillEnergy ( TFCSSimulationState & simulstate,
const TFCSTruthState * truth,
const TFCSExtrapolationState * extrapol ) const

Definition at line 113 of file TFCSEnergyAndHitGANV2.cxx.

115 {
116 if (!truth) {
117 ATH_MSG_ERROR("Invalid truth pointer");
118 return false;
119 }
120
121 if (!extrapol) {
122 ATH_MSG_ERROR("Invalid extrapolation pointer");
123 return false;
124 }
125
126 const int pdgId = truth->pdgid();
127 const float charge = HepPDT::ParticleID(pdgId).charge();
128
129 float Einit;
130 const float Ekin = truth->Ekin();
131 if (OnlyScaleEnergy())
132 Einit = simulstate.E();
133 else
134 Einit = Ekin;
135
136 ATH_MSG_VERBOSE("Momentum " << truth->P() << " pdgId " << truth->pdgid());
137 // check that the network exists
138 if (!m_slice->IsGanCorrectlyLoaded()) {
139 ATH_MSG_WARNING("GAN not loaded correctly.");
140 return false;
141 }
142
144 m_slice->GetNetworkOutputs(truth, extrapol, simulstate);
145 ATH_MSG_VERBOSE("network outputs size: " << outputs.size());
146
147 const TFCSGANXMLParameters::Binning &binsInLayers = m_param.GetBinning();
148 const auto ganVersion = m_param.GetGANVersion();
149 const TFCSGANEtaSlice::FitResultsPerLayer &fitResults =
150 m_slice->GetFitResults(); // used only if GAN version > 1
151
152 ATH_MSG_DEBUG("energy voxels size = " << outputs.size());
153
154 double totalEnergy = 0;
155 for (const auto & output : outputs) {
156 totalEnergy += output.second;
157 }
158 if (totalEnergy < 0) {
159 ATH_MSG_WARNING("Energy from GAN is negative, skipping particle");
160 return false;
161 }
162
163 ATH_MSG_VERBOSE("Get binning");
164
165 simulstate.set_E(0);
166
167 int vox = 0;
168 for (const auto &[layer, h] : binsInLayers) {
169 // attempt to debug intermittent ci issues described in
170 // https://its.cern.ch/jira/browse/ATLASSIM-7031
171 if (h.IsZombie()) {
172 ATH_MSG_INFO("Got truth state: ");
173 truth->Print();
174
175 ATH_MSG_INFO("Got extrapolation state: ");
176 extrapol->Print();
177
178 ATH_MSG_INFO("Got simulation state: ");
179 simulstate.Print();
180
181 ATH_MSG_INFO("Got GAN XML parameters: ");
182 m_param.Print();
183
184 ATH_MSG_ERROR("Histogram pointer for layer " << layer << " is broken");
185 return false;
186 }
187
188 const int xBinNum = h.GetNbinsX();
189 const int yBinNum = h.GetNbinsY();
190 const TAxis *x = h.GetXaxis();
191
192 // If only one bin in r means layer is empty, no value should be added
193 if (xBinNum == 1) {
194 ATH_MSG_VERBOSE(" Layer "
195 << layer
196 << " has only one bin in r, this means is it not used, "
197 "skipping (this is needed to keep correct "
198 "syncronisation of voxel and layers)");
199 // delete h;
200 continue;
201 }
202
203 ATH_MSG_VERBOSE(" Getting energy for Layer " << layer);
204
205 // First fill energies
206 for (int ix = 1; ix <= xBinNum; ++ix) {
207 double binsInAlphaInRBin = GetAlphaBinsForRBin(x, ix, yBinNum);
208 for (int iy = 1; iy <= binsInAlphaInRBin; ++iy) {
209 const double energyInVoxel = outputs.at(std::to_string(vox));
210 ATH_MSG_VERBOSE(" Vox " << vox << " energy " << energyInVoxel
211 << " binx " << ix << " biny " << iy);
212
213 if (energyInVoxel <= 0) {
214 vox++;
215 continue;
216 }
217
218 simulstate.add_E(layer, Einit * energyInVoxel);
219 vox++;
220 }
221 }
222 }
223
224 for (unsigned int ichain = m_bin_start.back(); ichain < size(); ++ichain) {
225 ATH_MSG_DEBUG("now run for all bins: " << chain()[ichain]->GetName());
226 if (simulate_and_retry(chain()[ichain], simulstate, truth, extrapol) !=
227 FCSSuccess) {
228 return FCSFatal;
229 }
230 }
231
232 vox = 0;
233 for (const auto &[layer, h] : binsInLayers) {
234 const int xBinNum = h.GetNbinsX();
235 const int yBinNum = h.GetNbinsY();
236 const TAxis *x = h.GetXaxis();
237 const TAxis *y = h.GetYaxis();
238
239 simulstate.setAuxInfo<int>("GANlayer"_FCShash, layer);
240 TFCSLateralShapeParametrizationHitBase::Hit hit;
241
242 // If only one bin in r means layer is empty, no value should be added
243 if (xBinNum == 1) {
244 ATH_MSG_VERBOSE(" Layer "
245 << layer
246 << " has only one bin in r, this means is it not used, "
247 "skipping (this is needed to keep correct "
248 "syncronisation of voxel and layers)");
249 // delete h;
250 continue;
251 }
252
253 if (get_number_of_bins() > 0) {
254 const int bin = get_bin(simulstate, truth, extrapol);
255 if (bin >= 0 && bin < (int)get_number_of_bins()) {
256 for (unsigned int ichain = m_bin_start[bin];
257 ichain < TMath::Min(m_bin_start[bin] + get_nr_of_init(bin),
258 m_bin_start[bin + 1]);
259 ++ichain) {
260 ATH_MSG_DEBUG("for " << get_variable_text(simulstate, truth, extrapol)
261 << " run init " << get_bin_text(bin) << ": "
262 << chain()[ichain]->GetName());
263 if (chain()[ichain]->InheritsFrom(
264 TFCSLateralShapeParametrizationHitBase::Class())) {
265 TFCSLateralShapeParametrizationHitBase *sim =
266 static_cast<TFCSLateralShapeParametrizationHitBase *>(chain()[ichain]);
267 if (sim->simulate_hit(hit, simulstate, truth, extrapol) !=
268 FCSSuccess) {
269 ATH_MSG_ERROR("error for "
270 << get_variable_text(simulstate, truth, extrapol)
271 << " run init " << get_bin_text(bin) << ": "
272 << chain()[ichain]->GetName());
273 return false;
274 }
275 } else {
276 ATH_MSG_ERROR("for "
277 << get_variable_text(simulstate, truth, extrapol)
278 << " run init " << get_bin_text(bin) << ": "
279 << chain()[ichain]->GetName()
280 << " does not inherit from "
281 "TFCSLateralShapeParametrizationHitBase");
282 return false;
283 }
284 }
285 } else {
286 ATH_MSG_WARNING("nothing to init for "
287 << get_variable_text(simulstate, truth, extrapol)
288 << ": " << get_bin_text(bin));
289 }
290 }
291
292 int binResolution = 5;
293 if (layer == 1 || layer == 5) {
294 binResolution = 1;
295 }
296
297 const double center_eta = hit.center_eta();
298 const double center_phi = hit.center_phi();
299 const double center_r = hit.center_r();
300 const double center_z = hit.center_z();
301
302 ATH_MSG_VERBOSE(" Layer " << layer << " Extrap eta " << center_eta
303 << " phi " << center_phi << " R " << center_r);
304
305 const float dist000 =
306 TMath::Sqrt(center_r * center_r + center_z * center_z);
307 const float eta_jakobi = TMath::Abs(2.0 * TMath::Exp(-center_eta) /
308 (1.0 + TMath::Exp(-2 * center_eta)));
309
310 int nHitsAlpha{};
311 int nHitsR{};
312
313 // Now create hits
314 for (int ix = 1; ix <= xBinNum; ++ix) {
315 const int binsInAlphaInRBin = GetAlphaBinsForRBin(x, ix, yBinNum);
316
317 // Horrible work around for variable # of bins along alpha direction
318 const int binsToMerge = yBinNum == 32 ? 32 / binsInAlphaInRBin : 1;
319 for (int iy = 1; iy <= binsInAlphaInRBin; ++iy) {
320 const double energyInVoxel = outputs.at(std::to_string(vox));
321 const int lowEdgeIndex = (iy - 1) * binsToMerge + 1;
322
323 ATH_MSG_VERBOSE(" Vox " << vox << " energy " << energyInVoxel
324 << " binx " << ix << " biny " << iy);
325
326 if (energyInVoxel <= 0) {
327 vox++;
328 continue;
329 }
330
331 if (std::abs(pdgId) == 22 || std::abs(pdgId) == 11) {
332 // maximum 10 MeV per hit, equaly distributed in alpha and r
333 int maxHitsInVoxel = energyInVoxel * truth->Ekin() / 10;
334 if (maxHitsInVoxel < 1)
335 maxHitsInVoxel = 1;
336 nHitsAlpha = std::sqrt(maxHitsInVoxel);
337 nHitsR = std::sqrt(maxHitsInVoxel);
338 } else {
339 // One hit per mm along r
340 nHitsR = x->GetBinUpEdge(ix) - x->GetBinLowEdge(ix);
341 if (yBinNum == 1) {
342 // nbins in alpha depend on circumference lenght
343 const double r = x->GetBinUpEdge(ix);
344 nHitsAlpha = ceil(2 * TMath::Pi() * r / binResolution);
345 } else {
346 // d = 2*r*sin (a/2r) this distance at the upper r must be 1mm for
347 // layer 1 or 5, 5mm otherwise.
348 const double angle = y->GetBinUpEdge(iy) - y->GetBinLowEdge(iy);
349 const double r = x->GetBinUpEdge(ix);
350 const double d = 2 * r * sin(angle / 2 * r);
351 nHitsAlpha = ceil(d / binResolution);
352 }
353
354 if (layer != 1 && layer != 5) {
355 // For layers that are not EMB1 or EMEC1 use a maximum of 10 hits
356 // per direction, a higher granularity is needed for the other
357 // layers
358 const int maxNhits = 10;
359 nHitsAlpha = std::min(maxNhits, std::max(1, nHitsAlpha));
360 nHitsR = std::min(maxNhits, std::max(1, nHitsR));
361 }
362 }
363
364 for (int ir = 0; ir < nHitsR; ++ir) {
365 double r =
366 x->GetBinLowEdge(ix) + x->GetBinWidth(ix) / (nHitsR + 1) * ir;
367
368 for (int ialpha = 1; ialpha <= nHitsAlpha; ++ialpha) {
369 if (ganVersion > 1) {
370 if (fitResults.at(layer)[ix - 1] != 0) {
371 int tries = 0;
372 double a = CLHEP::RandFlat::shoot(simulstate.randomEngine(),
373 x->GetBinLowEdge(ix),
374 x->GetBinUpEdge(ix));
375 double rand_r =
376 log((a - x->GetBinLowEdge(ix)) / (x->GetBinWidth(ix))) /
377 fitResults.at(layer)[ix - 1];
378 while ((rand_r < x->GetBinLowEdge(ix) ||
379 rand_r > x->GetBinUpEdge(ix)) &&
380 tries < 100) {
381 a = CLHEP::RandFlat::shoot(simulstate.randomEngine(),
382 x->GetBinLowEdge(ix),
383 x->GetBinUpEdge(ix));
384 rand_r =
385 log((a - x->GetBinLowEdge(ix)) / (x->GetBinWidth(ix))) /
386 fitResults.at(layer)[ix - 1];
387 tries++;
388 }
389 if (tries >= 100) {
390 ATH_MSG_VERBOSE(" Too many tries for bin ["
391 << x->GetBinLowEdge(ix) << "-"
392 << x->GetBinUpEdge(ix) << "] having slope "
393 << fitResults.at(layer)[ix - 1]
394 << " will use grid (old method)");
395 } else {
396 r = rand_r;
397 }
398 }
399 }
400
401 double alpha;
402 if (binsInAlphaInRBin == 1) {
403 alpha = CLHEP::RandFlat::shoot(simulstate.randomEngine(),
404 -TMath::Pi(), TMath::Pi());
405 } else {
406 alpha =
407 y->GetBinLowEdge(lowEdgeIndex) +
408 y->GetBinWidth(iy) * binsToMerge / (nHitsAlpha + 1) * ialpha;
409
410 if (m_param.IsSymmetrisedAlpha()) {
411 if (CLHEP::RandFlat::shoot(simulstate.randomEngine(),
412 -TMath::Pi(), TMath::Pi()) < 0) {
413 alpha = -alpha;
414 }
415 }
416 }
417
418 hit.reset();
419 hit.E() = Einit * energyInVoxel / (nHitsAlpha * nHitsR);
420
421 if (layer <= 20) {
422 float delta_eta_mm = r * cos(alpha);
423 float delta_phi_mm = r * sin(alpha);
424
425 ATH_MSG_VERBOSE("delta_eta_mm " << delta_eta_mm
426 << " delta_phi_mm "
427 << delta_phi_mm);
428
429 // Particles with negative eta are expected to have the same shape
430 // as those with positive eta after transformation: delta_eta -->
431 // -delta_eta
432 if (center_eta < 0.)
433 delta_eta_mm = -delta_eta_mm;
434 // We derive the shower shapes for electrons and positively
435 // charged hadrons. Particle with the opposite charge are expected
436 // to have the same shower shape after the transformation:
437 // delta_phi --> -delta_phi
438 if ((charge < 0. && pdgId != 11) || pdgId == -11)
439 delta_phi_mm = -delta_phi_mm;
440
441 const float delta_eta = delta_eta_mm / eta_jakobi / dist000;
442 const float delta_phi = delta_phi_mm / center_r;
443
444 hit.eta() = center_eta + delta_eta;
445 hit.phi() = TVector2::Phi_mpi_pi(center_phi + delta_phi);
446
447 ATH_MSG_VERBOSE(" Hit eta " << hit.eta() << " phi " << hit.phi()
448 << " layer " << layer);
449 } else { // FCAL is in (x,y,z)
450 const float hit_r = r * cos(alpha) + center_r;
451 float delta_phi = r * sin(alpha) / center_r;
452 // We derive the shower shapes for electrons and positively
453 // charged hadrons. Particle with the opposite charge are expected
454 // to have the same shower shape after the transformation:
455 // delta_phi --> -delta_phi
456 if ((charge < 0. && pdgId != 11) || pdgId == -11)
458 const float hit_phi =
459 TVector2::Phi_mpi_pi(center_phi + delta_phi);
460 hit.x() = hit_r * cos(hit_phi);
461 hit.y() = hit_r * sin(hit_phi);
462 hit.z() = center_z;
463 ATH_MSG_VERBOSE(" Hit x " << hit.x() << " y " << hit.y()
464 << " layer " << layer);
465 }
466
467 if (get_number_of_bins() > 0) {
468 const int bin = get_bin(simulstate, truth, extrapol);
469 if (bin >= 0 && bin < (int)get_number_of_bins()) {
470 for (unsigned int ichain =
471 m_bin_start[bin] + get_nr_of_init(bin);
472 ichain < m_bin_start[bin + 1]; ++ichain) {
474 "for " << get_variable_text(simulstate, truth, extrapol)
475 << " run " << get_bin_text(bin) << ": "
476 << chain()[ichain]->GetName());
477 if (chain()[ichain]->InheritsFrom(
478 TFCSLateralShapeParametrizationHitBase::Class())) {
479 TFCSLateralShapeParametrizationHitBase *sim =
480 static_cast<TFCSLateralShapeParametrizationHitBase
481 *>(chain()[ichain]);
482 if (sim->simulate_hit(hit, simulstate, truth, extrapol) !=
483 FCSSuccess) {
485 "error for "
486 << get_variable_text(simulstate, truth, extrapol)
487 << " run init " << get_bin_text(bin) << ": "
488 << chain()[ichain]->GetName());
489 return false;
490 }
491 } else {
493 "for " << get_variable_text(simulstate, truth, extrapol)
494 << " run init " << get_bin_text(bin) << ": "
495 << chain()[ichain]->GetName()
496 << " does not inherit from "
497 "TFCSLateralShapeParametrizationHitBase");
498 return false;
499 }
500 }
501 } else {
503 "nothing to do for "
504 << get_variable_text(simulstate, truth, extrapol) << ": "
505 << get_bin_text(bin));
506 }
507 } else {
508 ATH_MSG_WARNING("no bins defined, is this intended?");
509 }
510 }
511 }
512 vox++;
513 }
514 }
515
516 ATH_MSG_VERBOSE("Number of voxels " << vox);
517 ATH_MSG_VERBOSE("Done layer " << layer);
518 }
519
520 if (simulstate.E() > std::numeric_limits<double>::epsilon()) {
521 for (int ilayer = 0; ilayer < CaloCell_ID_FCS::MaxSample; ++ilayer) {
522 simulstate.set_Efrac(ilayer, simulstate.E(ilayer) / simulstate.E());
523 }
524 }
525
526 ATH_MSG_VERBOSE("Done particle");
527 return true;
528}
#define ATH_MSG_ERROR(x)
#define ATH_MSG_INFO(x)
#define ATH_MSG_VERBOSE(x)
#define ATH_MSG_WARNING(x)
double charge(const T &p)
Definition AtlasPID.h:997
static Double_t a
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
virtual const std::string get_variable_text(TFCSSimulationState &simulstate, const TFCSTruthState *, const TFCSExtrapolationState *) const override
int GetAlphaBinsForRBin(const TAxis *x, int ix, int yBinNum) const
TFCSGANXMLParameters m_param
std::map< int, std::vector< double > > FitResultsPerLayer
std::map< std::string, double > NetworkOutputs
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
void Print(Option_t *option="") const
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()
void Print(Option_t *option="") const
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
output
Definition merge.py:16

◆ 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 TFCSEnergyAndHitGANV2::GANfreemem ( ) const
inline

Definition at line 39 of file TFCSEnergyAndHitGANV2.h.

39{ return TestBit(kGANfreemem); };

◆ get_bin()

virtual int TFCSEnergyAndHitGANV2::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 55 of file TFCSEnergyAndHitGANV2.h.

56 {
57 return simulstate.getAuxInfo<int>("GANlayer"_FCShash);
58 };
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 TFCSGANXMLParameters::Binning & TFCSEnergyAndHitGANV2::get_Binning ( ) const
inline

Definition at line 66 of file TFCSEnergyAndHitGANV2.h.

66 {
67 return m_param.GetBinning();
68 };

◆ get_ExtrapolationWeights()

const TFCSGANEtaSlice::ExtrapolatorWeights TFCSEnergyAndHitGANV2::get_ExtrapolationWeights ( ) const
inline

Definition at line 69 of file TFCSEnergyAndHitGANV2.h.

69 {
70 return m_slice->GetExtrapolatorWeights();
71 };

◆ get_nr_of_init()

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

Definition at line 55 of file TFCSEnergyAndHitGANV2.cxx.

55 {
56 if (bin >= m_bin_ninit.size())
57 return 0;
58 return m_bin_ninit[bin];
59}
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 TFCSEnergyAndHitGANV2::get_variable_text ( TFCSSimulationState & simulstate,
const TFCSTruthState * ,
const TFCSExtrapolationState *  ) const
overridevirtual

Reimplemented from TFCSParametrizationBinnedChain.

Definition at line 106 of file TFCSEnergyAndHitGANV2.cxx.

108 {
109 return std::string(
110 Form("layer=%d", simulstate.getAuxInfo<int>("GANlayer"_FCShash)));
111}

◆ GetAlphaBinsForRBin()

int TFCSEnergyAndHitGANV2::GetAlphaBinsForRBin ( const TAxis * x,
int ix,
int yBinNum ) const
private

Definition at line 706 of file TFCSEnergyAndHitGANV2.cxx.

707 {
708 double binsInAlphaInRBin = yBinNum;
709 if (yBinNum == 32) {
710 ATH_MSG_DEBUG("yBinNum is special value 32");
711 const double widthX = x->GetBinWidth(ix);
712 const double radious = x->GetBinCenter(ix);
713 double circumference = radious * 2 * TMath::Pi();
714 if (m_param.IsSymmetrisedAlpha()) {
715 circumference = radious * TMath::Pi();
716 }
717
718 const double bins = circumference / widthX;
719 binsInAlphaInRBin = GetBinsInFours(bins);
720 ATH_MSG_DEBUG("Bin in alpha: " << binsInAlphaInRBin << " for r bin: " << ix
721 << " (" << x->GetBinLowEdge(ix) << "-"
722 << x->GetBinUpEdge(ix) << ")");
723 }
724 return binsInAlphaInRBin;
725}
static const std::vector< std::string > bins
static int GetBinsInFours(double const &bins)

◆ GetBinsInFours()

int TFCSEnergyAndHitGANV2::GetBinsInFours ( double const & bins)
staticprivate

Definition at line 695 of file TFCSEnergyAndHitGANV2.cxx.

695 {
696 if (bins < 4)
697 return 4;
698 else if (bins < 8)
699 return 8;
700 else if (bins < 16)
701 return 16;
702 else
703 return 32;
704}

◆ initializeNetwork()

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

Definition at line 71 of file TFCSEnergyAndHitGANV2.cxx.

73 {
74
75 // initialize all necessary constants
76 // FIXME eventually all these could be stored in the .json file
77
79 "Using FastCaloGANInputFolderName: " << FastCaloGANInputFolderName);
80 // get neural net JSON file as an std::istream object
81 const int etaMax = etaMin + 5;
82
84 set_pdgid(pid);
85 if (pid == 11)
86 add_pdgid(-pid);
87 if (pid == 211)
88 add_pdgid(-pid);
89 set_eta_min(etaMin / 100.0);
90 set_eta_max(etaMax / 100.0);
91 set_eta_nominal((etaMin + etaMax) / 200.0);
92
93 int pidForXml = pid;
94 if (pid != 22 && pid != 11) {
95 pidForXml = 211;
96 }
97
98 const int etaMid = (etaMin + etaMax) / 2;
99 m_param.InitialiseFromXML(pidForXml, etaMid, FastCaloGANInputFolderName);
100 m_param.Print();
101 m_slice = new TFCSGANEtaSlice(pid, etaMin, etaMax, m_param);
102 m_slice->Print();
103 return m_slice->LoadGAN();
104}
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 TFCSEnergyAndHitGANV2::is_match_all_calosample ( ) const
inlineoverridevirtual

Reimplemented from TFCSParametrization.

Definition at line 31 of file TFCSEnergyAndHitGANV2.h.

31{ 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 TFCSEnergyAndHitGANV2::is_match_all_Ekin_bin ( ) const
inlineoverridevirtual

Reimplemented from TFCSParametrization.

Definition at line 30 of file TFCSEnergyAndHitGANV2.h.

30{ 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 TFCSEnergyAndHitGANV2::is_match_calosample ( int calosample) const
overridevirtual

Reimplemented from TFCSParametrizationChain.

Definition at line 47 of file TFCSEnergyAndHitGANV2.cxx.

47 {
48 if (get_Binning().find(calosample) == get_Binning().cend())
49 return false;
50 if (get_Binning().at(calosample).GetNbinsX() == 1)
51 return false;
52 return true;
53}
const TFCSGANXMLParameters::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 TFCSEnergyAndHitGANV2::is_match_Ekin_bin ( int ) const
inlineoverridevirtual

Reimplemented from TFCSParametrizationChain.

Definition at line 26 of file TFCSEnergyAndHitGANV2.h.

26 {
27 return true;
28 };

◆ 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 TFCSEnergyAndHitGANV2::OnlyScaleEnergy ( ) const
inline

Definition at line 50 of file TFCSEnergyAndHitGANV2.h.

50{ 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 TFCSEnergyAndHitGANV2::Print ( Option_t * option = "") const
overridevirtual

Definition at line 551 of file TFCSEnergyAndHitGANV2.cxx.

551 {
553 TString opt(option);
554 const bool shortprint = opt.Index("short") >= 0;
555 const bool longprint =
556 msgLvl(MSG::DEBUG) || (msgLvl(MSG::INFO) && !shortprint);
557 TString optprint = opt;
558 optprint.ReplaceAll("short", "");
559
560 TString prefix = "- ";
561 for (unsigned int ichain = 0; ichain < size(); ++ichain) {
562 if (ichain == 0 && ichain != m_bin_start.front()) {
563 prefix = "> ";
564 if (longprint)
565 ATH_MSG_INFO(optprint << prefix << "Run for all bins");
566 }
567 for (unsigned int ibin = 0; ibin < get_number_of_bins(); ++ibin) {
568 if (ichain == m_bin_start[ibin]) {
569 if (ibin < get_number_of_bins() - 1)
570 if (ichain == m_bin_start[ibin + 1])
571 continue;
572 prefix = Form("%-2d", ibin);
573 if (longprint)
574 ATH_MSG_INFO(optprint << prefix << "Run for " << get_bin_text(ibin));
575 }
576 }
577 if (ichain == m_bin_start.back()) {
578 prefix = "< ";
579 if (longprint)
580 ATH_MSG_INFO(optprint << prefix << "Run for all bins");
581 }
582 chain()[ichain]->Print(opt + prefix);
583 }
584}
bool msgLvl(const MSG::Level lvl) const
Check whether the logging system is active at the provided verbosity level.
Definition MLogging.h:222
void Print(Option_t *option="") const
Print object information.
virtual unsigned int size() const override
Some derived classes have daughter instances of TFCSParametrizationBase objects The size() and operat...

◆ 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
str index
Definition DeMoScan.py:362

◆ 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 TFCSEnergyAndHitGANV2::reset_GANfreemem ( )
inline

Definition at line 41 of file TFCSEnergyAndHitGANV2.h.

41{ 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 TFCSEnergyAndHitGANV2::reset_OnlyScaleEnergy ( )
inline

Definition at line 52 of file TFCSEnergyAndHitGANV2.h.

52{ 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 TFCSEnergyAndHitGANV2::set_GANfreemem ( )
inline

Definition at line 40 of file TFCSEnergyAndHitGANV2.h.

40{ 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 TFCSEnergyAndHitGANV2::set_nr_of_init ( unsigned int bin,
unsigned int ninit )

Definition at line 61 of file TFCSEnergyAndHitGANV2.cxx.

62 {
63 if (bin >= m_bin_ninit.size()) {
64 m_bin_ninit.resize(bin + 1, 0);
65 m_bin_ninit.shrink_to_fit();
66 }
67 m_bin_ninit[bin] = ninit;
68}

◆ set_OnlyScaleEnergy()

void TFCSEnergyAndHitGANV2::set_OnlyScaleEnergy ( )
inline

Definition at line 51 of file TFCSEnergyAndHitGANV2.h.

51{ 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}

◆ SetRegionAndSliceFromXML()

void TFCSEnergyAndHitGANV2::SetRegionAndSliceFromXML ( int pid,
int etaMax,
std::string FastCaloGANInputFolderName )
protected

◆ simulate()

FCSReturnCode TFCSEnergyAndHitGANV2::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 530 of file TFCSEnergyAndHitGANV2.cxx.

532 {
533 for (unsigned int ichain = 0; ichain < m_bin_start[0]; ++ichain) {
534 ATH_MSG_DEBUG("now run for all bins: " << chain()[ichain]->GetName());
535 if (simulate_and_retry(chain()[ichain], simulstate, truth, extrapol) !=
536 FCSSuccess) {
537 return FCSFatal;
538 }
539 }
540
541 ATH_MSG_VERBOSE("Fill Energies");
542 if (!fillEnergy(simulstate, truth, extrapol)) {
543 ATH_MSG_WARNING("Could not fill energies ");
544 // bail out but do not stop the job
545 return FCSSuccess;
546 }
547
548 return FCSSuccess;
549}
bool fillEnergy(TFCSSimulationState &simulstate, const TFCSTruthState *truth, const TFCSExtrapolationState *extrapol) 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}
TFile * file

◆ test_path()

void TFCSEnergyAndHitGANV2::test_path ( const std::string & path,
TFCSSimulationState * simulstate = nullptr,
const TFCSTruthState * truth = nullptr,
const TFCSExtrapolationState * extrapol = nullptr,
const std::string & outputname = "unnamed",
int pid = 211 )
static

Definition at line 604 of file TFCSEnergyAndHitGANV2.cxx.

608 {
609 ISF_FCS::MLogging logger;
610 ATH_MSG_NOCLASS(logger, "Running test on " << path << std::endl);
611 if (!simulstate) {
612 simulstate = new TFCSSimulationState();
613#if defined(__FastCaloSimStandAlone__)
614 simulstate->setRandomEngine(new CLHEP::TRandomEngine());
615#else
616 simulstate->setRandomEngine(new CLHEP::RanluxEngine());
617#endif
618 }
619 if (!truth) {
620 ATH_MSG_NOCLASS(logger, "New particle");
621 TFCSTruthState *t = new TFCSTruthState();
622 t->SetPtEtaPhiM(65536, 0, 0, ParticleConstants::chargedPionMassInMeV);
623 t->set_pdgid(pid);
624 truth = t;
625 }
626 if (!extrapol) {
627 TFCSExtrapolationState *e = new TFCSExtrapolationState();
628 e->set_IDCaloBoundary_eta(truth->Eta());
629 for (int i = 0; i < 24; ++i) {
630 e->set_eta(i, TFCSExtrapolationState::SUBPOS_ENT, truth->Eta());
631 e->set_eta(i, TFCSExtrapolationState::SUBPOS_EXT, truth->Eta());
632 e->set_eta(i, TFCSExtrapolationState::SUBPOS_MID, truth->Eta());
636 e->set_r(i, TFCSExtrapolationState::SUBPOS_ENT, 1500 + i * 10);
637 e->set_r(i, TFCSExtrapolationState::SUBPOS_EXT, 1510 + i * 10);
638 e->set_r(i, TFCSExtrapolationState::SUBPOS_MID, 1505 + i * 10);
639 e->set_z(i, TFCSExtrapolationState::SUBPOS_ENT, 3500 + i * 10);
640 e->set_z(i, TFCSExtrapolationState::SUBPOS_EXT, 3510 + i * 10);
641 e->set_z(i, TFCSExtrapolationState::SUBPOS_MID, 3505 + i * 10);
642 }
643 extrapol = e;
644 }
645
646 TFCSEnergyAndHitGANV2 GAN("GAN", "GAN");
647 GAN.setLevel(MSG::INFO);
648 const int etaMin = 20;
649 const int etaMax = etaMin + 5;
650 ATH_MSG_NOCLASS(logger, "Initialize Networks");
651 GAN.initializeNetwork(pid, etaMin, path);
652 for (int i = 0; i < 24; ++i)
653 if (GAN.is_match_calosample(i)) {
654 TFCSCenterPositionCalculation *c = new TFCSCenterPositionCalculation(
655 Form("center%d", i), Form("center layer %d", i));
656 c->set_calosample(i);
657 c->setExtrapWeight(0.5);
658 c->setLevel(MSG::INFO);
659 c->set_pdgid(pid);
660 if (pid == 11)
661 c->add_pdgid(-pid);
662 if (pid == 211)
663 c->add_pdgid(-pid);
664 c->set_eta_min(etaMin / 100.0);
665 c->set_eta_max(etaMax / 100.0);
666 c->set_eta_nominal((etaMin + etaMax) / 200.0);
667
668 GAN.push_back_in_bin(c, i);
669 GAN.set_nr_of_init(i, 1);
670 }
671
672 GAN.Print();
673
674 ATH_MSG_NOCLASS(logger, "Writing GAN to " << outputname);
675 const std::string outname = "FCSGANtest_" + outputname + ".root";
676 TFile *fGAN = TFile::Open(outname.c_str(), "recreate");
677 fGAN->cd();
678 // GAN.Write();
679 fGAN->WriteObjectAny(&GAN, "TFCSEnergyAndHitGANV2", "GAN");
680
681 fGAN->ls();
682 fGAN->Close();
683
684 ATH_MSG_NOCLASS(logger, "Open " << outname);
685 fGAN = TFile::Open(outname.c_str());
686 TFCSEnergyAndHitGANV2 *GAN2 = (TFCSEnergyAndHitGANV2 *)(fGAN->Get("GAN"));
687 GAN2->setLevel(MSG::INFO);
688 GAN2->Print();
689
690 ATH_MSG_NOCLASS(logger, "Before running GAN2->simulate()");
691 GAN2->simulate(*simulstate, truth, extrapol);
692 simulstate->Print();
693}
#define ATH_MSG_NOCLASS(logger_name, x)
Definition MLogging.h:52
virtual void setLevel(MSG::Level lvl)
Update outputlevel.
Definition MLogging.cxx:105
virtual void Print(Option_t *option="") const override
TFCSEnergyAndHitGANV2(const char *name=nullptr, const char *title=nullptr)
virtual FCSReturnCode simulate(TFCSSimulationState &simulstate, const TFCSTruthState *truth, const TFCSExtrapolationState *extrapol) const override
Method in all derived classes to do some simulation.
void setRandomEngine(CLHEP::HepRandomEngine *engine)
static Root::TMsgLogger logger("iLumiCalc")
constexpr double chargedPionMassInMeV
the mass of the charged pion (in MeV)

◆ unit_test()

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

Definition at line 586 of file TFCSEnergyAndHitGANV2.cxx.

588 {
589 ISF_FCS::MLogging logger;
590 ATH_MSG_NOCLASS(logger, "Start lwtnn test" << std::endl);
591 std::string path =
592 "/eos/atlas/atlascerngroupdisk/proj-simul/AF3_Run3/"
593 "InputsToBigParamFiles/FastCaloGANWeightsVer02/";
594 test_path(path, simulstate, truth, extrapol, "lwtnn");
595
596 ATH_MSG_NOCLASS(logger, "Start onnx test" << std::endl);
597 path =
598 "/eos/atlas/atlascerngroupdisk/proj-simul/AF3_Run3/"
599 "InputsToBigParamFiles/FastCaloGANWeightsONNXVer08/";
600 test_path(path, simulstate, truth, extrapol, "onnx");
601 ATH_MSG_NOCLASS(logger, "Finish all tests" << std::endl);
602}
static void test_path(const std::string &path, TFCSSimulationState *simulstate=nullptr, const TFCSTruthState *truth=nullptr, const TFCSExtrapolationState *extrapol=nullptr, const std::string &outputname="unnamed", int pid=211)
path
python interpreter configuration --------------------------------------—
Definition athena.py:128

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> TFCSEnergyAndHitGANV2::m_bin_ninit
private

Definition at line 102 of file TFCSEnergyAndHitGANV2.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_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_nm

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

Message source name.

Definition at line 211 of file MLogging.h.

◆ m_param

TFCSGANXMLParameters TFCSEnergyAndHitGANV2::m_param
private

Definition at line 110 of file TFCSEnergyAndHitGANV2.h.

◆ m_pdgid

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

Definition at line 63 of file TFCSParametrization.h.

◆ m_slice

TFCSGANEtaSlice* TFCSEnergyAndHitGANV2::m_slice = nullptr
private

Definition at line 109 of file TFCSEnergyAndHitGANV2.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: