crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-27 13:45:36 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add Python API for settings of surface source writing

Browse source
This commit is contained in:
YoungHui Park 2020-09-23 00:58:31 -05:00
parent fd6e516a71
commit 32be23eeda
1 changed files with 43 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

43
openmc/settings.py
View file

@ -143,6 +143,11 @@ class Settings:
Options for writing state points. Acceptable keys are:
:batches: list of batches at which to write source
surf_src_write : dict
Options for writing surface source points. Acceptable keys are:
:max_surf_banks: Maximum number of particles to be banked on surfaces
per process (int)
survival_biasing : bool
Indicate whether survival biasing is to be used
tabular_legendre : dict
@ -229,6 +234,8 @@ class Settings:
self._statepoint = {}
self._sourcepoint = {}
self._surf_src_write = {}
self._no_reduce = None
self._verbosity = None
@ -356,6 +363,10 @@ class Settings:
def statepoint(self):
return self._statepoint
@property
def surf_src_write(self):
return self._surf_src_write
@property
def no_reduce(self):
return self._no_reduce
@ -567,6 +578,19 @@ class Settings:
"statepoint options.".format(key))
self._statepoint = statepoint
@surf_src_write.setter
def surf_src_write(self, surf_src_write):
cv.check_type('surface source writing options', surf_src_write, Mapping)
for key, value in surf_src_write.items():
cv.check_value('surface source writing key', key,
('max_surf_banks'))
if key == 'max_surf_banks':
cv.check_type('maximum particle banks on surfaces per process',
value, Iterable, Integral)
cv.check_greater_than('maximum particle banks on surfaces per process',
value, 0)
self._surf_src_write = surf_src_write
@confidence_intervals.setter
def confidence_intervals(self, confidence_intervals):
cv.check_type('confidence interval', confidence_intervals, bool)
@ -889,6 +913,13 @@ class Settings:
subelement = ET.SubElement(element, "overwrite_latest")
subelement.text = str(self._sourcepoint['overwrite']).lower()
def _create_surf_src_write_subelement(self, root):
if self._surf_src_write:
element = ET.SubElement(root, "surf_src_write")
if 'max_surf_banks' in self._surf_src_write:
subelement = ET.SubElement(element, "max_surf_banks")
subelement.text = str(self._surf_src_write['max_surf_banks'])
def _create_confidence_intervals(self, root):
if self._confidence_intervals is not None:
element = ET.SubElement(root, "confidence_intervals")
@ -1151,6 +1182,16 @@ class Settings:
value = [int(x) for x in value.split()]
self.sourcepoint[key] = value
def _surf_src_write_from_xml_element(self, root):
elem = root.find('surf_src_write')
if elem is not None:
for key in ('max_surf_banks'):
value = get_text(elem, key)
if value is not None:
if key in ('max_surf_banks'):
value = int(value)
self.surf_src_write[key] = value
def _confidence_intervals_from_xml_element(self, root):
text = get_text(root, 'confidence_intervals')
if text is not None:
@ -1349,6 +1390,7 @@ class Settings:
self._create_output_subelement(root_element)
self._create_statepoint_subelement(root_element)
self._create_sourcepoint_subelement(root_element)
self._create_surf_src_write_subelement(root_element)
self._create_confidence_intervals(root_element)
self._create_electron_treatment_subelement(root_element)
self._create_energy_mode_subelement(root_element)
@ -1422,6 +1464,7 @@ class Settings:
settings._output_from_xml_element(root)
settings._statepoint_from_xml_element(root)
settings._sourcepoint_from_xml_element(root)
settings._surf_src_write_from_xml_element(root)
settings._confidence_intervals_from_xml_element(root)
settings._electron_treatment_from_xml_element(root)
settings._energy_mode_from_xml_element(root)