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

Check for VTK version to determine whether to call SetInput or SetInputData

Browse source
This commit is contained in:
Paul Romano 2015-06-08 13:43:15 +07:00
parent 99abca5007
commit 4c03377506
2 changed files with 8 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -89,7 +89,10 @@ def main():
data.SetLines(cells)
writer = vtk.vtkXMLPPolyDataWriter()
writer.SetInputData(data)
if vtk.vtkVersion.GetVTKMajorVersion() > 5:
writer.SetInputData(data)
else:
writer.SetInput(data)
writer.SetFileName(args.out)
writer.Write()

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

@ -62,7 +62,10 @@ def main(file_, o):
grid.GetCellData().AddArray(data)
writer = vtk.vtkXMLImageDataWriter()
writer.SetInputData(grid)
if vtk.vtkVersion.GetVTKMajorVersion() > 5:
writer.SetInputData(grid)
else:
writer.SetInput(grid)
if not o.output.endswith(".vti"):
o.output += ".vti"
writer.SetFileName(o.output)