From 9541cc63637ced49c25116bd4d66b0b996c9f3fc Mon Sep 17 00:00:00 2001
From: Philip Cook <cookpa@pennmedicine.upenn.edu>
Date: Tue, 6 Aug 2024 21:41:02 -0400
Subject: [PATCH] ENH: antsTransform.apply_to_image on displacement field
 transforms

---
 src/antsTransform.cxx | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/src/antsTransform.cxx b/src/antsTransform.cxx
index 4fde0f1c..3c557cae 100644
--- a/src/antsTransform.cxx
+++ b/src/antsTransform.cxx
@@ -128,7 +128,7 @@ AntsTransform<TransformType> antsTransformFromDisplacementField( AntsImage<Vecto
 }
 
 template <typename TransformType, typename VectorImageType, typename PrecisionType, unsigned int Dimension>
-AntsImage<VectorImageType> antsTransformToDisplacementField( AntsTransform<TransformType> & xfrm, 
+AntsImage<VectorImageType> antsTransformToDisplacementField( AntsTransform<TransformType> & xfrm,
                                                              AntsImage<itk::Image<PrecisionType, Dimension>> & ref )
 {
   //typedef itk::Transform<PrecisionType,Dimension,Dimension>                  TransformType;
@@ -246,6 +246,16 @@ void local_antsTransform(nb::module_ &m) {
     m.def("transformImage", &transformImage<itk::Transform<double,3, 3>, itk::Image<double, 3>>);
     m.def("transformImage", &transformImage<itk::Transform<double,4, 4>, itk::Image<double, 4>>);
 
+    // displacement field transforms
+    m.def("transformImage", &transformImage<itk::DisplacementFieldTransform<float, 2>, itk::Image<unsigned char, 2>>);
+    m.def("transformImage", &transformImage<itk::DisplacementFieldTransform<float, 3>, itk::Image<unsigned char, 3>>);
+    m.def("transformImage", &transformImage<itk::DisplacementFieldTransform<float, 2>, itk::Image<unsigned int, 2>>);
+    m.def("transformImage", &transformImage<itk::DisplacementFieldTransform<float, 3>, itk::Image<unsigned int, 3>>);
+    m.def("transformImage", &transformImage<itk::DisplacementFieldTransform<float, 2>, itk::Image<float, 2>>);
+    m.def("transformImage", &transformImage<itk::DisplacementFieldTransform<float, 3>, itk::Image<float, 3>>);
+    m.def("transformImage", &transformImage<itk::DisplacementFieldTransform<float, 2>, itk::Image<double, 2>>);
+    m.def("transformImage", &transformImage<itk::DisplacementFieldTransform<float, 3>, itk::Image<double, 3>>);
+
     m.def("inverseTransform", &inverseTransform<itk::Transform<float, 2, 2>, itk::Transform<float, 2, 2>>);
     m.def("inverseTransform", &inverseTransform<itk::Transform<float, 3, 3>, itk::Transform<float, 3, 3>>);
     m.def("inverseTransform", &inverseTransform<itk::Transform<float, 4, 4>, itk::Transform<float, 4, 4>>);
@@ -286,7 +296,7 @@ void local_antsTransform(nb::module_ &m) {
     m.def("antsTransformToDisplacementField", &antsTransformToDisplacementField<itk::DisplacementFieldTransform<float,2>, itk::VectorImage<float,2>,float,2>);
     m.def("antsTransformToDisplacementField", &antsTransformToDisplacementField<itk::DisplacementFieldTransform<float,3>, itk::VectorImage<float,3>,float,3>);
 
-    
+
     nb::class_<AntsTransform<itk::DisplacementFieldTransform<float,2>>>(m, "AntsTransformDF2");
     nb::class_<AntsTransform<itk::DisplacementFieldTransform<float,3>>>(m, "AntsTransformDF3");
     nb::class_<AntsTransform<itk::Transform<float, 2, 2>>>(m, "AntsTransformF22");