From 4c03377506022d262cdd73c644d1b4eb5b9ce51b Mon Sep 17 00:00:00 2001 From: Paul Romano Date: Mon, 8 Jun 2015 13:43:15 +0700 Subject: [PATCH] Check for VTK version to determine whether to call SetInput or SetInputData --- scripts/openmc-track-to-vtk | 5 ++++- scripts/openmc-voxel-to-silovtk | 5 ++++- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/scripts/openmc-track-to-vtk b/scripts/openmc-track-to-vtk index 1761d08ba6..8db57e2a3f 100755 --- a/scripts/openmc-track-to-vtk +++ b/scripts/openmc-track-to-vtk @@ -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() diff --git a/scripts/openmc-voxel-to-silovtk b/scripts/openmc-voxel-to-silovtk index 633d44cb12..eb052c75fa 100755 --- a/scripts/openmc-voxel-to-silovtk +++ b/scripts/openmc-voxel-to-silovtk @@ -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)