ENH: Adding Python Wrapping for SyNImageRegistrationMethod

Needed for ThinShellDemons Mesh Registration.
InsightSoftwareConsortium · Dec 27, 2021 · 653ab12 · 653ab12
1 parent 2b56a14
commit 653ab12
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/Modules/Registration/RegistrationMethodsv4/wrapping/itkImageRegistrationMethodv4.wrap b/Modules/Registration/RegistrationMethodsv4/wrapping/itkImageRegistrationMethodv4.wrap
@@ -22,6 +22,8 @@ itk_wrap_class("itk::ImageRegistrationMethodv4" POINTER)
                         "itk::Image< ${ITKT_${t}}, ${d} >, itk::Image< ${ITKT_${t}}, ${d} >")
       itk_wrap_template("REGv4${ITKM_${t}}${d}${ITKM_${t}}${d}T${ITKM_D}${d}${ITKM_${t}}${d}M${ITKM_D}${d}"
                         "itk::Image< ${ITKT_${t}}, ${d} >, itk::Image< ${ITKT_${t}}, ${d} >, itk::Transform< ${ITKT_D}, ${d}, ${d} >, itk::Image< ${ITKT_${t}}, ${d} >, itk::Mesh< ${ITKT_D}, ${d} >")
+      itk_wrap_template("REGv4${ITKM_${t}}${d}${ITKM_${t}}${d}DFT${ITKM_D}${d}${ITKM_${t}}${d}M${ITKM_D}${d}"
+                        "itk::Image< ${ITKT_${t}}, ${d} >, itk::Image< ${ITKT_${t}}, ${d} >, itk::DisplacementFieldTransform< ${ITKT_D}, ${d}>, itk::Image< ${ITKT_${t}}, ${d} >, itk::Mesh< ${ITKT_D}, ${d} >")
       itk_wrap_template("REGv4${ITKM_${t}}${d}${ITKM_${t}}${d}DFT${ITKM_D}${d}"
                         "itk::Image< ${ITKT_${t}}, ${d} >, itk::Image< ${ITKT_${t}}, ${d} >, itk::DisplacementFieldTransform< ${ITKT_D}, ${d} >")
       itk_wrap_template("REGv4${ITKM_${t}}${d}${ITKM_${t}}${d}BSOUDFT${ITKM_D}${d}"

diff --git a/Modules/Registration/RegistrationMethodsv4/wrapping/itkSyNImageRegistrationMethod.wrap b/Modules/Registration/RegistrationMethodsv4/wrapping/itkSyNImageRegistrationMethod.wrap
@@ -1,4 +1,5 @@
 itk_wrap_include("itkBSplineSmoothingOnUpdateDisplacementFieldTransform.h")
+itk_wrap_include("itkMesh.h")
 
 itk_wrap_class("itk::SyNImageRegistrationMethod" POINTER)
   foreach(d ${ITK_WRAP_DIMS})
@@ -7,6 +8,8 @@ itk_wrap_class("itk::SyNImageRegistrationMethod" POINTER)
                         "itk::Image< ${ITKT_${t}}, ${d} >, itk::Image< ${ITKT_${t}}, ${d} >")
       itk_wrap_template("${ITKM_${t}}${d}${ITKM_${t}}${d}BSOUDFT${ITKM_D}${d}"
                         "itk::Image< ${ITKT_${t}}, ${d} >, itk::Image< ${ITKT_${t}}, ${d} >, itk::BSplineSmoothingOnUpdateDisplacementFieldTransform< ${ITKT_D}, ${d} >")
+      itk_wrap_template("${ITKM_${t}}${d}${ITKM_${t}}${d}DFT${ITKM_D}${d}M${ITKM_D}${d}"
+                        "itk::Image< ${ITKT_${t}}, ${d} >, itk::Image< ${ITKT_${t}}, ${d} >, itk::DisplacementFieldTransform< ${ITKT_D}, ${d} >, itk::Image< ${ITKT_${t}}, ${d} >, itk::Mesh< ${ITKT_D}, ${d} >")
     endforeach()
   endforeach()
 itk_end_wrap_class()