ENH: Add ComputeMeanSquareBetweenTwoImages Python example

Author: tbirdso

src/Registration/Common/ComputeMeanSquareBetweenTwoImages/CMakeLists.txt

@@ -14,7 +14,7 @@ install(TARGETS ComputeMeanSquareBetweenTwoImages
   COMPONENT Runtime
   )
 
-install(FILES Code.cxx CMakeLists.txt
+install(FILES Code.cxx Code.py CMakeLists.txt
   DESTINATION share/ITKSphinxExamples/Code/Registration/Common/ComputeMeanSquareBetweenTwoImages/
   COMPONENT Code
   )
@@ -25,3 +25,11 @@ add_test(NAME ComputeMeanSquareBetweenTwoImagesTest
   COMMAND ${CMAKE_RUNTIME_OUTPUT_DIRECTORY}/ComputeMeanSquareBetweenTwoImages
     Yinyang.png
     Gourds.png)
+    
+    
+if(ITK_WRAP_PYTHON)
+  add_test(NAME ComputeMeanSquareBetweenTwoImagesTestPython
+    COMMAND ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/Code.py
+	Yinyang.png
+	Gourds.png)
+endif()

src/Registration/Common/ComputeMeanSquareBetweenTwoImages/Code.py

@@ -0,0 +1,49 @@
+#!/usr/bin/env python3
+
+# =========================================================================
+#
+#   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
+
+if len(sys.argv) < 3:
+    raise Exception(f"Usage: {sys.argv[0]} imageFile1 imageFile2")
+
+fixed_image = itk.imread(sys.argv[1], itk.F)
+moving_image = itk.imread(sys.argv[2], itk.F)
+
+metric = itk.MeanSquaresImageToImageMetric[type(fixed_image), type(moving_image)].New()
+transform = itk.TranslationTransform[itk.D, fixed_image.GetImageDimension()].New()
+interpolator = itk.LinearInterpolateImageFunction[type(fixed_image), itk.D].New()
+
+metric.SetFixedImage(fixed_image)
+metric.SetMovingImage(moving_image)
+metric.SetFixedImageRegion(fixed_image.GetLargestPossibleRegion())
+metric.SetTransform(transform)
+metric.SetInterpolator(interpolator)
+metric.Initialize()
+
+params = itk.OptimizerParameters[itk.D]()
+params.SetSize(2)
+
+for x in range(-10, 15, 5):
+    params.SetElement(0, x)
+    for y in range(-10, 15, 5):
+        params.SetElement(1, y)
+        print(f"{list(params)}: {metric.GetValue(params):.1f}")

src/Registration/Common/ComputeMeanSquareBetweenTwoImages/Documentation.rst

@@ -64,6 +64,12 @@ C++
 .. literalinclude:: Code.cxx
    :lines: 18-
 
+Python
+...
+
+.. literalinclude:: Code.py
+   :lines: 21-
+
 Classes demonstrated
 --------------------