python callback at restart

ECP-WarpX · Jan 10, 2025 · d5aa155 · d5aa155
1 parent 17b0c3d
commit d5aa155
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/Python/pywarpx/callbacks.py b/Python/pywarpx/callbacks.py
@@ -280,6 +280,7 @@ def callfuncsinlist(self, *args, **kw):
     "loadExternalFields": {},
     "beforeInitEsolve": {},
     "afterInitEsolve": {},
+    "afterInitatRestart": {},
     "afterinit": {},
     "beforecollisions": {},
     "aftercollisions": {},
@@ -405,6 +406,13 @@ def callfromafterInitEsolve(f):
 def installafterInitEsolve(f):
     installcallback("afterInitEsolve", f)
 
+# ----------------------------------------------------------------------------
+def callfromafterInitatRestart(f):
+    installcallback("afterInitatRestart", f)
+    return f
+
+def installafterInitatRestart(f):
+    installcallback("afterInitatRestart", f)
 
 # ----------------------------------------------------------------------------
 def callfromafterinit(f):

diff --git a/Source/Initialization/WarpXInitData.cpp b/Source/Initialization/WarpXInitData.cpp
@@ -610,6 +610,9 @@ WarpX::InitData ()
             AddExternalFields(lev);
         }
     }
+    else {
+        ExecutePythonCallback("afterInitatRestart");
+    }
 
     if (restart_chkfile.empty() || write_diagnostics_on_restart) {
         // Write full diagnostics before the first iteration.