python script used to add a total source intensity tag from an expanded r2s step 2 source .h5m. Raw

total_n_convert.py

from pyne.mesh import Mesh,IMeshTag
import numpy as np
import sys


m = Mesh(sys.argv[1])

m.source_density_total = IMeshTag(size=1, dtype=float)
total = np.array(m.source_density_total[:])

for name, tag in m.tags.items():
    if "source_density_" in name:
        total += np.array(tag[:])

m.source_density_total[:] = total[:]

fout = sys.argv[1].split(".")
m.mesh.save(fout[0]+"_w_total.vtk")