add afterBfieldpush callback

roelof-groenewald · Dec 6, 2023 · 7384c96 · 7384c96
1 parent 06839c9
commit 7384c96
Showing 1 changed file with 24 additions and 21 deletions.
diff --git a/Source/FieldSolver/FiniteDifferenceSolver/HybridPICModel/HybridPICModel.cpp b/Source/FieldSolver/FiniteDifferenceSolver/HybridPICModel/HybridPICModel.cpp
@@ -8,6 +8,7 @@
  */
 
 #include "HybridPICModel.H"
+#include "Python/callbacks.H"
 
 using namespace amrex;
 
@@ -565,13 +566,13 @@ void HybridPICModel::CalculateEquilibCurrent (
 }
 
 void HybridPICModel::BfieldEvolveRK (
-        amrex::Vector<std::array< std::unique_ptr<amrex::MultiFab>, 3>>& Bfield,
-        amrex::Vector<std::array< std::unique_ptr<amrex::MultiFab>, 3>>& Efield,
-        amrex::Vector<std::array< std::unique_ptr<amrex::MultiFab>, 3>> const& Jfield,
-        amrex::Vector<std::unique_ptr<amrex::MultiFab>> const& rhofield,
-        amrex::Vector<std::array< std::unique_ptr<amrex::MultiFab>, 3>> const& edge_lengths,
-        amrex::Real const dt, DtType dt_type,
-        IntVect ng, std::optional<bool> nodal_sync )
+    amrex::Vector<std::array< std::unique_ptr<amrex::MultiFab>, 3>>& Bfield,
+    amrex::Vector<std::array< std::unique_ptr<amrex::MultiFab>, 3>>& Efield,
+    amrex::Vector<std::array< std::unique_ptr<amrex::MultiFab>, 3>> const& Jfield,
+    amrex::Vector<std::unique_ptr<amrex::MultiFab>> const& rhofield,
+    amrex::Vector<std::array< std::unique_ptr<amrex::MultiFab>, 3>> const& edge_lengths,
+    amrex::Real const dt, DtType dt_type,
+    IntVect ng, std::optional<bool> nodal_sync )
 {
     auto& warpx = WarpX::GetInstance();
     for (int lev = 0; lev <= warpx.finestLevel(); ++lev)
@@ -584,13 +585,13 @@ void HybridPICModel::BfieldEvolveRK (
 }
 
 void HybridPICModel::BfieldEvolveRK (
-        amrex::Vector<std::array< std::unique_ptr<amrex::MultiFab>, 3>>& Bfield,
-        amrex::Vector<std::array< std::unique_ptr<amrex::MultiFab>, 3>>& Efield,
-        amrex::Vector<std::array< std::unique_ptr<amrex::MultiFab>, 3>> const& Jfield,
-        amrex::Vector<std::unique_ptr<amrex::MultiFab>> const& rhofield,
-        amrex::Vector<std::array< std::unique_ptr<amrex::MultiFab>, 3>> const& edge_lengths,
-        amrex::Real const dt, int lev, DtType dt_type,
-        IntVect ng, std::optional<bool> nodal_sync )
+    amrex::Vector<std::array< std::unique_ptr<amrex::MultiFab>, 3>>& Bfield,
+    amrex::Vector<std::array< std::unique_ptr<amrex::MultiFab>, 3>>& Efield,
+    amrex::Vector<std::array< std::unique_ptr<amrex::MultiFab>, 3>> const& Jfield,
+    amrex::Vector<std::unique_ptr<amrex::MultiFab>> const& rhofield,
+    amrex::Vector<std::array< std::unique_ptr<amrex::MultiFab>, 3>> const& edge_lengths,
+    amrex::Real const dt, int lev, DtType dt_type,
+    IntVect ng, std::optional<bool> nodal_sync )
 {
     // Make copies of the B-field multifabs at t = n and create multifabs for
     // each direction to store the Runge-Kutta intermediate terms. Each
@@ -697,13 +698,13 @@ void HybridPICModel::BfieldEvolveRK (
 }
 
 void HybridPICModel::BfieldPush (
-        amrex::Vector<std::array< std::unique_ptr<amrex::MultiFab>, 3>>& Bfield,
-        amrex::Vector<std::array< std::unique_ptr<amrex::MultiFab>, 3>>& Efield,
-        amrex::Vector<std::array< std::unique_ptr<amrex::MultiFab>, 3>> const& Jfield,
-        amrex::Vector<std::unique_ptr<amrex::MultiFab>> const& rhofield,
-        amrex::Vector<std::array< std::unique_ptr<amrex::MultiFab>, 3>> const& edge_lengths,
-        amrex::Real const dt, DtType dt_type,
-        IntVect ng, std::optional<bool> nodal_sync )
+    amrex::Vector<std::array< std::unique_ptr<amrex::MultiFab>, 3>>& Bfield,
+    amrex::Vector<std::array< std::unique_ptr<amrex::MultiFab>, 3>>& Efield,
+    amrex::Vector<std::array< std::unique_ptr<amrex::MultiFab>, 3>> const& Jfield,
+    amrex::Vector<std::unique_ptr<amrex::MultiFab>> const& rhofield,
+    amrex::Vector<std::array< std::unique_ptr<amrex::MultiFab>, 3>> const& edge_lengths,
+    amrex::Real const dt, DtType dt_type,
+    IntVect ng, std::optional<bool> nodal_sync )
 {
     auto& warpx = WarpX::GetInstance();
 
@@ -715,4 +716,6 @@ void HybridPICModel::BfieldPush (
     // Push forward the B-field using Faraday's law
     warpx.EvolveB(dt, dt_type);
     warpx.FillBoundaryB(ng, nodal_sync);
+
+    ExecutePythonCallback("afterBfieldpush");
 }