STYLE: Pythonic style for ResampleAVectorImage

Author: thewtex

src/Filtering/ImageGrid/ResampleAVectorImage/CMakeLists.txt

@@ -1,37 +1,37 @@
 cmake_minimum_required(VERSION 3.10.2)
 
-set( input_image ${CMAKE_CURRENT_BINARY_DIR}/VisibleWomanHeadSlice.png )
-set( output_image Output.png )
+set(input_image ${CMAKE_CURRENT_BINARY_DIR}/VisibleWomanHeadSlice.png)
+set(output_image Output.png)
 
-project( ResampleAVectorImage )
+project(ResampleAVectorImage)
 
-find_package( ITK REQUIRED )
-include( ${ITK_USE_FILE} )
+find_package(ITK REQUIRED)
+include(${ITK_USE_FILE})
 
-add_executable( ResampleAVectorImage Code.cxx )
-target_link_libraries( ResampleAVectorImage ${ITK_LIBRARIES} )
+add_executable(ResampleAVectorImage Code.cxx)
+target_link_libraries(ResampleAVectorImage ${ITK_LIBRARIES})
 
-install( TARGETS ResampleAVectorImage
+install(TARGETS ResampleAVectorImage
   DESTINATION bin/ITKExamples/Filtering/ImageGrid/
   COMPONENT Runtime
-)
+  )
 
-install( FILES Code.cxx CMakeLists.txt Code.py
+install(FILES Code.cxx CMakeLists.txt Code.py
   DESTINATION share/ITKExamples/Code/Filtering/ImageGrid/ResampleAVectorImage
   COMPONENT Code
-)
+  )
 
 enable_testing()
 
-add_test( NAME ResampleAVectorImageTest
+add_test(NAME ResampleAVectorImageTest
   COMMAND ${CMAKE_RUNTIME_OUTPUT_DIRECTORY}/ResampleAVectorImage
   ${input_image}
   ${output_image}
-)
+  )
 
-if( ITK_WRAP_PYTHON )
-  string( REPLACE . "Python." output_image "${output_image}" )
-  add_test( NAME ResampleAVectorImageTestPython
+if(ITK_WRAP_PYTHON)
+  string(REPLACE . "Python." output_image "${output_image}")
+  add_test(NAME ResampleAVectorImageTestPython
     COMMAND ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/Code.py
       ${input_image}
       ${output_image}

src/Filtering/ImageGrid/ResampleAVectorImage/Code.py

@@ -17,49 +17,29 @@
 import sys
 import itk
 
-from distutils.version import StrictVersion as VS
-if VS(itk.Version.GetITKVersion()) < VS("4.7.0"):
-    print("ITK 4.7.0 is required (see example documentation).")
-    sys.exit(1)
-
 if len(sys.argv) != 3:
-    print("Usage: " + sys.argv[0] + " [inputImage] [outputImage]")
+    print("Usage: " + sys.argv[0] + " [input_filename] [output_filename]")
     sys.exit(1)
 
-inputImage = sys.argv[1]
-outputImage = sys.argv[2]
-
-Dimension = 2
+input_filename = sys.argv[1]
+output_filename = sys.argv[2]
 
 PixelType = itk.RGBPixel[itk.UC]
-ImageType = itk.Image[PixelType, Dimension]
-
-ReaderType = itk.ImageFileReader[ImageType]
-reader = ReaderType.New()
-reader.SetFileName(inputImage)
-
-filterType = itk.ResampleImageFilter[ImageType, ImageType]
-resampleImageFilter = filterType.New()
-
-interpolatorType = itk.LinearInterpolateImageFunction[ImageType, itk.D]
-interpolator = interpolatorType.New()
-
-resampleImageFilter.SetInterpolator(interpolator)
 
-transformType = itk.IdentityTransform[itk.D, Dimension]
-transform = transformType.New()
+input_image = itk.imread(input_filename, pixel_type=PixelType)
 
-resampleImageFilter.SetTransform(transform)
+ImageType = type(input_image)
+interpolator = itk.LinearInterpolateImageFunction[ImageType, itk.D].New()
 
-resampleImageFilter.SetDefaultPixelValue([50, 50, 50])
-resampleImageFilter.SetOutputSpacing([0.5, 0.5])
-resampleImageFilter.SetOutputOrigin([30, 40])
+Dimension = input_image.GetImageDimension()
+transform = itk.IdentityTransform[itk.D, Dimension].New()
 
-resampleImageFilter.SetSize([300, 300])
-resampleImageFilter.SetInput(reader.GetOutput())
+output_image = itk.resample_image_filter(input_image,
+        interpolator=interpolator,
+        transform=transform,
+        default_pixel_value=[50, 50, 50],
+        output_spacing=[0.5, 0.5],
+        output_origin=[30, 40],
+        size=[300, 300])
 
-WriterType = itk.ImageFileWriter[ImageType]
-writer = WriterType.New()
-writer.SetFileName(outputImage)
-writer.SetInput(resampleImageFilter.GetOutput())
-writer.Update()
+itk.imwrite(output_image, output_filename)