Phi & Force: Expose to Python

ECP-WarpX · Aug 16, 2022 · db90b99 · db90b99
1 parent 33c9558
commit db90b99
Showing 1 changed file with 18 additions and 1 deletion.
diff --git a/src/python/ImpactX.cpp b/src/python/ImpactX.cpp
@@ -130,7 +130,24 @@ void init_ImpactX(py::module& m)
             "rho",
             [](ImpactX & ix, int const lev) { return &ix.m_rho.at(lev); },
             py::arg("lev"),
-            py::return_value_policy::reference_internal
+            py::return_value_policy::reference_internal,
+            "charge density per level"
+        )
+        .def(
+            "phi",
+            [](ImpactX & ix, int const lev) { return &ix.m_phi.at(lev); },
+            py::arg("lev"),
+            py::return_value_policy::reference_internal,
+            "scalar potential per level"
+        )
+        .def(
+            "space_charge_force",
+            [](ImpactX & ix, int const lev, std::string const comp) {
+                return &ix.m_space_charge_force.at(lev).at(comp);
+            },
+            py::arg("lev"), py::arg("comp"),
+            py::return_value_policy::reference_internal,
+            "space charge force (vector: x,y,z) per level"
         )
         .def_readwrite("lattice",
             &ImpactX::m_lattice,