From 14e3684a6a4013cb27900d474a60a18424bf0ed8 Mon Sep 17 00:00:00 2001 From: Adam Nelson Date: Sun, 20 Nov 2016 06:05:36 -0500 Subject: [PATCH] Fixed issue with plotting nuclides with divisor types --- openmc/plotter.py | 16 ++++------------ 1 file changed, 4 insertions(+), 12 deletions(-) diff --git a/openmc/plotter.py b/openmc/plotter.py index 45783d6cf7..d6c9527bd0 100644 --- a/openmc/plotter.py +++ b/openmc/plotter.py @@ -116,20 +116,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