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

Merge branch 'plot_mat' into from_hdf5

Browse source
This commit is contained in:
Adam Nelson 2016-11-20 06:05:50 -05:00
commit 39dcd86e54
1 changed files with 4 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

16
openmc/plotter.py
View file

@ -124,20 +124,12 @@ def plot_xs(this, types, divisor_types=None, temperature=294., axis=None,
# grid, and then do the actual division
Enum = E[:]
E = np.union1d(Enum, Ediv)
if data_type == 'nuclide':
data_new = []
else:
data_new = np.zeros((len(types), len(E)))
data_new = np.zeros((len(types), len(E)))
for line in range(len(types)):
if data_type == 'nuclide':
data_new.append(openmc.data.Combination([data[line],
data_div[line]],
[np.divide]))
else:
data_new[line, :] = \
np.divide(np.interp(E, Enum, data[line, :]),
np.interp(E, Ediv, data_div[line, :]))
data_new[line, :] = \
np.divide(np.interp(E, Enum, data[line, :]),
np.interp(E, Ediv, data_div[line, :]))
if divisor_types[line] != 'unity':
types[line] = types[line] + ' / ' + divisor_types[line]
data = data_new