Array4: New overloads for contains

ax3l · May 29, 2024 · c38ff8b · c38ff8b
1 parent 4bb33a6
commit c38ff8b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/src/Base/Array4.H b/src/Base/Array4.H
@@ -247,8 +247,10 @@ namespace pyAMReX
                 return h_data;
             }, py::return_value_policy::move)
 
-            .def("contains", &Array4<T>::contains)
-            //.def("__contains__", &Array4<T>::contains)
+            //.def("__contains__", &Array4<T>::contains)  // syntax: "other in b"
+            .def("contains", py::overload_cast<int, int, int>(&Array4<T>::contains, py::const_))
+            .def("contains", py::overload_cast<IntVect const &>(&Array4<T>::contains, py::const_))
+            .def("contains", py::overload_cast<Dim3 const &>(&Array4<T>::contains, py::const_))
 
             // getter
             .def("__getitem__", [](Array4<T> & a4, IntVect const & v){ return a4(v); })

diff --git a/src/Base/Box.cpp b/src/Base/Box.cpp
@@ -288,7 +288,8 @@ void init_Box(py::module &m) {
 
         .def("make_slab",
              &Box::makeSlab,
-             py::arg("direction"), py::arg("slab_index"))
+             py::arg("direction"), py::arg("slab_index"),
+             "Flatten the box in one direction.")
 
         // minBox
         // chop