From 9d00b3fa5fb00e72805488b82621192615bbf6e0 Mon Sep 17 00:00:00 2001
From: Antoine Robert <antoine.robert.01@gmail.com>
Date: Wed, 18 Sep 2024 14:36:14 +0200
Subject: [PATCH] ENH: Wrap reorder projections filter for vector and CUDA
 images

---
 .../rtkReorderProjectionsImageFilter.wrap     | 21 ++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/wrapping/rtkReorderProjectionsImageFilter.wrap b/wrapping/rtkReorderProjectionsImageFilter.wrap
index 1288647fc..a2957bf69 100644
--- a/wrapping/rtkReorderProjectionsImageFilter.wrap
+++ b/wrapping/rtkReorderProjectionsImageFilter.wrap
@@ -1,3 +1,22 @@
+if(RTK_USE_CUDA)
+  itk_wrap_include(itkCudaImage.h)
+endif()
+
 itk_wrap_class("rtk::ReorderProjectionsImageFilter" POINTER)
-  itk_wrap_image_filter("${WRAP_ITK_ALL_TYPES}" 2 3)
+    itk_wrap_image_filter("${WRAP_ITK_ALL_TYPES}" 2 3)
+
+    # Force VECTOR_COMPONENTS to contain "2;3;4;5"
+    set(vectorComponents 2 3 4 5)
+
+    foreach(nmat ${vectorComponents})
+        foreach(vt ${WRAP_ITK_VECTOR_REAL})
+
+          itk_wrap_template("I${ITKM_${vt}${nmat}}3I${ITKM_${vt}${nmat}}3" "itk::Image<${ITKT_${vt}${nmat}},3>, itk::Image<${ITKT_${vt}${nmat}},3>")
+
+        endforeach()
+
+        if(RTK_USE_CUDA)
+          itk_wrap_template("CI${ITKM_VF${nmat}}3CI${ITKM_VF${nmat}}3" "itk::CudaImage<${ITKT_VF${nmat}},3>, itk::CudaImage<${ITKT_VF${nmat}},3>")
+        endif()
+    endforeach()
 itk_end_wrap_class()