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

changed if else layout

Browse source
This commit is contained in:
Jonathan Shimwell 2020-10-08 18:23:07 +01:00
parent 336b09b358
commit acde663c90
1 changed files with 8 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

12
openmc/plotter.py
View file

@ -405,13 +405,15 @@ def _calculate_cexs_nuclide(this, types, temperature=294., sab_name=None,
ops.append((np.add,) * (len(tmp_mts) - 2) + (np.multiply,))
else:
ops.append((np.add,) * (len(tmp_mts) - 1))
if line in openmc.data.REACTION_NUMBER:
openmc.data.REACTION_NUMBER[line]
tmp_mts = nuc.get_reaction_components(line)
elif line in openmc.data.REACTION_NUMBER.keys():
mt_number = openmc.data.REACTION_NUMBER[line]
cv.check_type('MT in types', mt_number, Integral)
cv.check_greater_than('MT in types', mt_number, 0)
tmp_mts = nuc.get_reaction_components(mt_number)
mts.append(tmp_mts)
ops.append((np.add,) * (len(tmp_mts) - 1))
yields.append(False)
else:
elif isinstance(line, int):
# Not a built-in type, we have to parse it ourselves
cv.check_type('MT in types', line, Integral)
cv.check_greater_than('MT in types', line, 0)
@ -419,6 +421,8 @@ def _calculate_cexs_nuclide(this, types, temperature=294., sab_name=None,
mts.append(tmp_mts)
ops.append((np.add,) * (len(tmp_mts) - 1))
yields.append(False)
else:
raise TypeError("Invalid type", line)
for i, mt_set in enumerate(mts):
# Get the reaction xs data from the nuclide