Add KDOP<2,4> and KDOP<2,8>

arborx · May 15, 2024 · fa77eb4 · fa77eb4
1 parent 8aae58e
commit fa77eb4
Showing 1 changed file with 32 additions and 0 deletions.
diff --git a/src/geometry/ArborX_KDOP.hpp b/src/geometry/ArborX_KDOP.hpp
@@ -29,6 +29,38 @@ namespace Details
 template <int DIM, int k, typename Coordinate>
 struct KDOP_Directions;
 
+template <typename Coordinate>
+struct KDOP_Directions<2, 4, Coordinate>
+{
+  static constexpr int n_directions = 2;
+  static KOKKOS_FUNCTION auto const &directions()
+  {
+    using Direction = Vector<2, Coordinate>;
+    static constexpr Kokkos::Array<Direction, n_directions> directions = {
+        Direction{1, 0},
+        Direction{0, 1},
+    };
+    return directions;
+  }
+};
+
+template <typename Coordinate>
+struct KDOP_Directions<2, 8, Coordinate>
+{
+  static constexpr int n_directions = 4;
+  static KOKKOS_FUNCTION auto const &directions()
+  {
+    using Direction = Vector<2, Coordinate>;
+    static constexpr Kokkos::Array<Direction, n_directions> directions = {
+        Direction{1, 0},
+        Direction{0, 1},
+        Direction{1, 1},
+        Direction{1, -1},
+    };
+    return directions;
+  }
+};
+
 template <typename Coordinate>
 struct KDOP_Directions<3, 6, Coordinate>
 {