Merge pull request #243 from phcerdan/update_convert_itk_vtk_examples

ENH: Use itk.image_to_vtk_image if itk version is at least 5.2

Author: thewtex

src/Bridge/VtkGlue/ConvertAnRGBitkImageTovtkImageData/Code.py

@@ -19,26 +19,18 @@
 
 from distutils.version import StrictVersion as VS
 
-if VS(itk.Version.GetITKVersion()) < VS("4.10.0"):
-    print("ITK 4.10.0 is required.")
+if VS(itk.Version.GetITKVersion()) < VS("5.2.0"):
+    print("ITK 5.2.0 is required.")
     sys.exit(1)
 
 if len(sys.argv) != 2:
     print("Usage: " + sys.argv[0] + " <InputFileName>")
     sys.exit(1)
-imageFileName = sys.argv[1]
 
-Dimension = 2
-PixelComponentType = itk.UC
-PixelType = itk.RGBPixel[PixelComponentType]
-ImageType = itk.Image[PixelType, Dimension]
+imageFileName = sys.argv[1]
 
-reader = itk.ImageFileReader[ImageType].New()
-reader.SetFileName(imageFileName)
+inputImage = itk.imread(imageFileName)
 
-itkToVtkFilter = itk.ImageToVTKImageFilter[ImageType].New()
-itkToVtkFilter.SetInput(reader.GetOutput())
+vtkImage = itk.image_to_vtk_image(inputImage)
 
-itkToVtkFilter.Update()
-myvtkImageData = itkToVtkFilter.GetOutput()
-print(myvtkImageData)
+print(vtkImage)

src/Bridge/VtkGlue/ConvertAnitkImageTovtkImageData/Code.py

@@ -17,21 +17,20 @@
 import sys
 import itk
 
+from distutils.version import StrictVersion as VS
+
+if VS(itk.Version.GetITKVersion()) < VS("5.2.0"):
+    print("ITK 5.2.0 is required.")
+    sys.exit(1)
+
 if len(sys.argv) != 2:
     print("Usage: " + sys.argv[0] + " <InputFileName>")
     sys.exit(1)
-imageFileName = sys.argv[1]
 
-Dimension = 2
-PixelType = itk.UC
-ImageType = itk.Image[PixelType, Dimension]
+imageFileName = sys.argv[1]
 
-reader = itk.ImageFileReader[ImageType].New()
-reader.SetFileName(imageFileName)
+inputImage = itk.imread(imageFileName)
 
-itkToVtkFilter = itk.ImageToVTKImageFilter[ImageType].New()
-itkToVtkFilter.SetInput(reader.GetOutput())
+vtkImage = itk.image_to_vtk_image(inputImage)
 
-itkToVtkFilter.Update()
-myvtkImageData = itkToVtkFilter.GetOutput()
-print(myvtkImageData)
+print(vtkImage)