Use Kokkos View in Stencil class

include/Stencil/stencil.h

@@ -1,8 +1,11 @@
 #pragma once
 
-#include <array>
 #include <initializer_list>
 
+#include <Kokkos_Core.hpp>
+
+#include "../include/LinearAlgebra/Vector/vector.h"
+
 enum class StencilPosition
 {
     TopLeft,
@@ -32,10 +35,29 @@ enum class StencilPosition
 class Stencil
 {
 public:
-    Stencil(std::initializer_list<int> init);
-    int operator[](StencilPosition type) const;
+    Stencil(std::initializer_list<int> init)
+        : values_("stencil_values", 9)
+        , stencil_size_(0)
+    {
+        int i = 0;
+        for (int v : init) {
+            values_(i++) = v;
+            if (v != -1)
+                stencil_size_++;
+        }
+    }
+
+    KOKKOS_INLINE_FUNCTION int operator[](StencilPosition type) const
+    {
+        return values_(static_cast<int>(type));
+    }
+
+    KOKKOS_INLINE_FUNCTION int stencil_size() const
+    {
+        return stencil_size_;
+    }
 
 private:
-    std::array<int, 9> values_;
+    AllocatableVector<int> values_;
     int stencil_size_;
 };

src/CMakeLists.txt

@@ -40,11 +40,6 @@ set(LEVEL_SOURCES
     ${CMAKE_CURRENT_SOURCE_DIR}/Level/levelCache.cpp
 )
 
-# file(GLOB_RECURSE STENCIL_SOURCES ${CMAKE_CURRENT_SOURCE_DIR}/Stencil/*.cpp)
-set(STENCIL_SOURCES
-    ${CMAKE_CURRENT_SOURCE_DIR}/Stencil/stencil.cpp
-)
-
 # file(GLOB_RECURSE INTERPOLATION_SOURCES ${CMAKE_CURRENT_SOURCE_DIR}/Interpolation/*.cpp)
 set(INTERPOLATION_SOURCES
     ${CMAKE_CURRENT_SOURCE_DIR}/Interpolation/extrapolated_prolongation.cpp
@@ -171,7 +166,6 @@ add_library(GMGPolarLib STATIC
     ${GMG_POLAR_SOURCES}
     ${MULTIGRID_METHODS_SOURCES}
     ${LEVEL_SOURCES}
-    ${STENCIL_SOURCES}
     ${INTERPOLATION_SOURCES}
     ${DIRECT_SOLVER_SOURCES}
     ${RESIDUAL_SOURCES}