ENH: Added ComputeInverseFFTOfImage Python script

Author: kian-weimer

src/Filtering/FFT/ComputeInverseFFTOfImage/CMakeLists.txt

@@ -23,3 +23,8 @@ install(FILES Code.cxx CMakeLists.txt
 enable_testing()
 add_test(NAME ComputeInverseFFTOfImageTest
   COMMAND ${CMAKE_RUNTIME_OUTPUT_DIRECTORY}/ComputeInverseFFTOfImage)
+
+if(ITK_WRAP_PYTHON)
+add_test(NAME ComputeInverseFFTOfImageTestPython
+  COMMAND ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/Code.py)
+endif()

src/Filtering/FFT/ComputeInverseFFTOfImage/Code.py

@@ -0,0 +1,54 @@
+#!/usr/bin/env python
+
+# Copyright NumFOCUS
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#        http://www.apache.org/licenses/LICENSE-2.0.txt
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import sys
+import itk
+
+dimension = 2
+float_image_type = itk.Image[itk.F, dimension]
+
+if len(sys.argv) < 2:
+    corner = itk.Index[dimension]()
+    corner.Fill(0)
+
+    size = itk.Size[dimension]()
+    size.Fill(200)
+
+    region = itk.ImageRegion[dimension]()
+    region.SetIndex(corner)
+    region.SetSize(size)
+
+    image = float_image_type.New(Regions=region)
+    image.Allocate()
+
+    # Make a square
+    image[40:100, 40:100] = 100
+
+else:
+    image = itk.imread(sys.argv[1], pixel_type=itk.F)
+
+# Define some types
+unsigned_char_image_type = itk.Image[itk.UC, dimension]
+
+# Compute the FFT
+image = itk.forward_fft_image_filter(image)
+
+# Compute the IFFT
+image = itk.inverse_fft_image_filter(image)
+
+image = itk.cast_image_filter(image, ttype=(float_image_type, unsigned_char_image_type))
+
+itk.imwrite(image, "ComputeInverseFFTOfImagePython.png")

src/Filtering/FFT/ComputeInverseFFTOfImage/Documentation.rst

@@ -25,6 +25,13 @@ Results
 Code
 ----
 
+Python
+......
+
+.. literalinclude:: Code.py
+   :language: python
+   :lines: 1, 16-
+
 C++
 ...