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

Fix a few bugs related to particle track files

Browse source
This commit is contained in:
Paul Romano 2015-06-06 22:13:28 +07:00
parent 57663dd195
commit ffcb4eec2a
3 changed files with 10 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

10
scripts/openmc-track-to-vtk
View file

@ -44,7 +44,7 @@ def main():
if not (fname.endswith('.h5') or fname.endswith('.binary')):
raise ValueError("Input file names must either end with '.h5' or"
"'.binary'.")
# Make sure that the output filename ends with '.pvtp'.
if not args.out:
args.out = 'tracks.pvtp'
@ -75,21 +75,21 @@ def main():
n_points = coords.shape[0]
for i in range(n_points):
points.InsertNextPoint(coords[i,:])
# Create VTK line and assign points to line.
line = vtk.vtkPolyLine()
line.GetPointIds().SetNumberOfIds(n_points)
for i in range(n_points):
line.GetPointIds().SetId(i, point_offset+i)
cells.InsertNextCell(line)
point_offset += n_points
data = vtk.vtkPolyData()
data.SetPoints(points)
data.SetLines(cells)
writer = vtk.vtkXMLPPolyDataWriter()
writer.SetInput(data)
writer.SetInputData(data)
writer.SetFileName(args.out)
writer.Write()

2
scripts/openmc-voxel-to-silovtk
View file

@ -62,7 +62,7 @@ def main(file_, o):
grid.GetCellData().AddArray(data)
writer = vtk.vtkXMLImageDataWriter()
writer.SetInput(grid)
writer.SetInputData(grid)
if not o.output[-4:] == ".vti": o.output += ".vti"
writer.SetFileName(o.output)
writer.Write()