crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-28 22:26:08 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

Passing mesh memo through to source from_xml method

Browse source
This commit is contained in:
Patrick Shriwise 2023-01-12 23:51:12 -06:00
parent 55138aa144
commit 4d06ed5640
1 changed files with 15 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

25
openmc/settings.py
View file

@ -978,8 +978,8 @@ class Settings:
for source in self.source:
root.append(source.to_xml_element())
if isinstance(source.space, MeshSpatial):
path = f"./mesh[@id='{source.space.mesh.id}']"
if root.find(path) is None:
path = f"./mesh[@id='{source.space.mesh.id}']"
if root.find(path) is None:
root.append(source.space.mesh.to_xml_element())
def _create_volume_calcs_subelement(self, root):
@ -1326,17 +1326,22 @@ class Settings:
threshold = float(get_text(elem, 'threshold'))
self.keff_trigger = {'type': trigger, 'threshold': threshold}
def _source_from_xml_element(self, root):
def _source_from_xml_element(self, root, meshes=None):
for elem in root.findall('source'):
src = Source.from_xml_element(elem)
if isinstance(src.space, MeshSpatial):
mesh_id = int(get_text(elem, 'mesh'))
path = f"./mesh[@id='{mesh_id}']"
mesh_elem = root.find(path)
if mesh_elem is not None:
src.space.mesh = MeshBase.from_xml_element(mesh_elem)
else:
raise RuntimeError('No mesh was specified for the mesh source')
if mesh_id not in meshes:
path = f"./mesh[@id='{mesh_id}']"
mesh_elem = root.find(path)
if mesh_elem is not None:
mesh = MeshBase.from_xml_element(mesh_elem)
meshes[mesh.id] = mesh
try:
src.space.mesh = meshes[mesh_id]
except KeyError as e:
raise e(f'Mesh with ID {mesh_id} was not found.')
# add newly constructed source object to the list
self.source.append(src)
def _volume_calcs_from_xml_element(self, root):
@ -1710,7 +1715,7 @@ class Settings:
settings._rel_max_lost_particles_from_xml_element(elem)
settings._generations_per_batch_from_xml_element(elem)
settings._keff_trigger_from_xml_element(elem)
settings._source_from_xml_element(elem)
settings._source_from_xml_element(elem, meshes)
settings._volume_calcs_from_xml_element(elem)
settings._output_from_xml_element(elem)
settings._statepoint_from_xml_element(elem)