ENH: Wrap TransformToDisplacementField for Python

InsightSoftwareConsortium · Aug 24, 2021 · f87c9bd · f87c9bd
1 parent 71e04b4
commit f87c9bd
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 5 deletions.
diff --git a/Modules/Filtering/DisplacementField/include/itkTransformToDisplacementFieldFilter.hxx b/Modules/Filtering/DisplacementField/include/itkTransformToDisplacementFieldFilter.hxx
@@ -186,9 +186,10 @@ TransformToDisplacementFieldFilter<TOutputImage, TParametersValueType>::Nonlinea
 
   // Define a few variables that will be used to translate from an input pixel
   // to an output pixel
-  PointType outputPoint;      // Coordinates of output pixel
-  PointType transformedPoint; // Coordinates of transformed pixel
-  PixelType displacement;     // the difference
+  PointType outputPoint;       // Coordinates of output pixel
+  PointType transformedPoint;  // Coordinates of transformed pixel
+  PointType displacementPoint; // the difference
+  PixelType displacementPixel; // the difference, cast to pixel type
 
 
   TotalProgressReporter progress(this, output->GetRequestedRegion().GetNumberOfPixels());
@@ -205,8 +206,13 @@ TransformToDisplacementFieldFilter<TOutputImage, TParametersValueType>::Nonlinea
       // Compute corresponding input pixel position
       transformedPoint = transform->TransformPoint(outputPoint);
 
-      displacement = transformedPoint - outputPoint;
-      outIt.Set(displacement);
+      displacementPoint = transformedPoint - outputPoint;
+      // Cast PointType -> PixelType
+      for (IndexValueType idx = 0; idx < ImageDimension; ++idx)
+      {
+        displacementPixel[idx] = static_cast<typename PixelType::ValueType>(displacementPoint[idx]);
+      }
+      outIt.Set(displacementPixel);
       ++outIt;
     }
     outIt.NextLine();

diff --git a/Modules/Filtering/DisplacementField/wrapping/itkTransformToDisplacementFieldFilter.wrap b/Modules/Filtering/DisplacementField/wrapping/itkTransformToDisplacementFieldFilter.wrap
@@ -0,0 +1,10 @@
+UNIQUE(transform_param_types "${WRAP_ITK_REAL};D")
+itk_wrap_class("itk::TransformToDisplacementFieldFilter" POINTER)
+  foreach(vt ${WRAP_ITK_VECTOR})
+    foreach(d ${ITK_WRAP_IMAGE_DIMS})
+      foreach(t ${transform_param_types})
+        itk_wrap_template("${ITKM_I${vt}${d}${d}}${ITKM_${t}}" "${ITKT_I${vt}${d}${d}}, ${ITKT_${t}}")
+      endforeach()
+    endforeach()
+  endforeach()
+itk_end_wrap_class()