ENH: Add float type to StrainImageFilter wrappings

KitwareMedical · Sep 2, 2021 · ac8b58a · ac8b58a
1 parent 4597dd5
commit ac8b58a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/wrapping/itkStrainImageFilter.wrap b/wrapping/itkStrainImageFilter.wrap
@@ -1,8 +1,10 @@
+UNIQUE(types "D;${WRAP_ITK_REAL}")
+
 itk_wrap_include("itkImageToImageFilter.h")
 itk_wrap_class("itk::ImageToImageFilter" POINTER)
   foreach(d ${ITK_WRAP_IMAGE_DIMS})
     set(vector_dim ${d}) # Wrap only vector dimensions which are the same as image dimensions
-    foreach(p D)
+    foreach(p ${types})
       itk_wrap_template(
         "${ITKM_IV${p}${vector_dim}${d}}${ITKM_ISSRT${p}${d}${d}}"
         "${ITKT_IV${p}${vector_dim}${d}}, ${ITKT_ISSRT${p}${d}${d}}")
@@ -16,7 +18,7 @@ itk_end_wrap_class()
 itk_wrap_class("itk::StrainImageFilter" POINTER)
   foreach(d ${ITK_WRAP_IMAGE_DIMS})
     set(vector_dim ${d}) # Wrap only vector dimensions which are the same as image dimensions
-    foreach(p D)
+    foreach(p ${types})
       itk_wrap_template(
         "${ITKM_IV${p}${vector_dim}${d}}${ITKM_${p}}${ITKM_${p}}"
         "${ITKT_IV${p}${vector_dim}${d}}, ${ITKT_${p}}, ${ITKT_${p}}")