expose WarpXParticleContainer::sumParticleCharge to Python (#2790)

2 files changed, 10 insertions, 0 deletions

diff --git a/Source/Python/WarpXWrappers.H b/Source/Python/WarpXWrappers.H
index 9c9f63133..817c1de73 100644
--- a/Source/Python/WarpXWrappers.H
+++ b/Source/Python/WarpXWrappers.H
@@ -94,6 +94,8 @@ extern "C" {
     void warpx_addRealComp(
         const char* char_species_name, const char* char_comp_name, bool comm);
 
+    amrex::Real warpx_sumParticleCharge(const char* char_species_name, const bool local);
+
     int warpx_getParticleBoundaryBufferSize(const char* species_name, int boundary);
 
     int** warpx_getParticleBoundaryBufferScrapedSteps(
diff --git a/Source/Python/WarpXWrappers.cpp b/Source/Python/WarpXWrappers.cpp
index 6703bb665..63ba4b713 100644
--- a/Source/Python/WarpXWrappers.cpp
+++ b/Source/Python/WarpXWrappers.cpp
@@ -516,6 +516,14 @@ namespace
         mypc.defineAllParticleTiles();
     }
 
+    amrex::Real warpx_sumParticleCharge(const char* char_species_name, const bool local)
+    {
+        auto & mypc = WarpX::GetInstance().GetPartContainer();
+        const std::string species_name(char_species_name);
+        auto & myspc = mypc.GetParticleContainerFromName(species_name);
+        return myspc.sumParticleCharge(local);
+    }
+
     int warpx_getParticleBoundaryBufferSize(const char* species_name, int boundary)
     {
         const std::string name(species_name);