SoA Name Helpers

ax3l · Jan 23, 2025 · e1e0e1b · e1e0e1b
1 parent 0585d17
commit e1e0e1b
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/src/Particle/ParticleContainer.H b/src/Particle/ParticleContainer.H
@@ -236,6 +236,24 @@ void make_ParticleContainer_and_Iterators (py::module &m, std::string allocstr)
             "Get the names for the int SoA components"
         )
 
+        .def("has_real_comp",
+            &ParticleContainerType::HasRealComp,
+            "Check if a container has an ParticleReal component"
+        )
+        .def("has_int_comp",
+            &ParticleContainerType::HasIntComp,
+            "Check if a container has an Integer component"
+        )
+
+        .def("get_real_comp_index",
+            &ParticleContainerType::GetRealCompIndex,
+            "Get the ParticleReal SoA index of a component"
+        )
+        .def("get_int_comp_index",
+            &ParticleContainerType::GetIntCompIndex,
+            "Get the Integer SoA index of a component"
+        )
+
         .def_property_readonly("finest_level", &ParticleContainerBase::finestLevel)
 
         // ParticleContainer ( const ParticleContainer &) = delete;

diff --git a/src/Particle/StructOfArrays.H b/src/Particle/StructOfArrays.H
@@ -10,7 +10,6 @@
 #include <AMReX_GpuAllocators.H>
 #include <AMReX_StructOfArrays.H>
 
-#include <stdexcept>
 #include <sstream>