Check for macroscopic data in materials when running Universe.plot

This commit is contained in:
Paul Romano 2022-08-02 22:11:37 -05:00
parent 8a6dc7e9f8
commit 8965dd1e66

View file

@ -334,6 +334,13 @@ class Universe(UniverseBase):
if seed is not None:
model.settings.seed = seed
# Determine whether any materials contains macroscopic data and if
# so, set energy mode accordingly
for mat in self.get_all_materials().values():
if mat._macroscopic is not None:
model.settings.energy_mode = 'multi-group'
break
# Create plot object matching passed arguments
plot = openmc.Plot()
plot.origin = origin