BoundaryConditions: array ctors

m-a-d-n-e-s-s · Dec 6, 2024 · 4d1174b · 4d1174b
1 parent 23ec4de
commit 4d1174b
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/madness/mra/bc.h b/src/madness/mra/bc.h
@@ -81,6 +81,20 @@ template <std::size_t NDIM> class BoundaryConditions {
       bc[i] = code;
   }
 
+  /// Constructor sets boundary condition per axis
+  template <std::size_t N = NDIM, typename = std::enable_if_t<N!=0>>
+  BoundaryConditions(const std::array<BCType, N>& bc_per_axis) {
+    for (std::size_t i = 0; i < NDIM * 2; ++i)
+      bc[i] = bc_per_axis[i / 2];
+  }
+
+  /// Constructor sets boundary condition per boundary
+  template <std::size_t N = NDIM, typename = std::enable_if_t<N!=0>>
+  BoundaryConditions(const std::array<BCType, N*2>& bc_per_boundary) {
+    for (std::size_t i = 0; i < NDIM * 2; ++i)
+      bc[i] = bc_per_boundary[i];
+  }
+
   /// Copy constructor is deep
   BoundaryConditions(const BoundaryConditions<NDIM> &other) { *this = other; }