From 812456172e3223687e986ed27a95c6d402723120 Mon Sep 17 00:00:00 2001
From: Roelof Groenewald <40245517+roelof-groenewald@users.noreply.github.com>
Date: Fri, 22 Jul 2022 13:02:06 -0700
Subject: Add Python Callback Call when Checkpointing Signal is Received
 (#3251)

---
 Python/pywarpx/callbacks.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

(limited to 'Python/pywarpx/callbacks.py')

diff --git a/Python/pywarpx/callbacks.py b/Python/pywarpx/callbacks.py
index cb9293c5a..e9f034a36 100644
--- a/Python/pywarpx/callbacks.py
+++ b/Python/pywarpx/callbacks.py
@@ -264,6 +264,7 @@ _beforestep = CallbackFunctions('beforestep')
 _afterstep = CallbackFunctions('afterstep')
 _afterdiagnostics = CallbackFunctions('afterdiagnostics')
 _afterrestart = CallbackFunctions('afterrestart',lcallonce=1)
+_oncheckpointsignal = CallbackFunctions('oncheckpointsignal')
 _particleinjection = CallbackFunctions('particleinjection')
 _appliedfields = CallbackFunctions('appliedfields')
 
@@ -506,6 +507,20 @@ def isinstalledafterrestart(f):
     raise Exception('restart call back not implemented yet')
     return _afterrestart.isinstalledfuncinlist(f)
 
+# ----------------------------------------------------------------------------
+def oncheckpointsignal(f):
+    installoncheckpointsignal(f)
+    return f
+def installoncheckpointsignal(f):
+    "Adds a function to the list of functions called on checkpoint signal"
+    _oncheckpointsignal.installfuncinlist(f)
+def uninstalloncheckpointsignal(f):
+    "Removes the function from the list of functions called on checkpoint signal"
+    _oncheckpointsignal.uninstallfuncinlist(f)
+def isinstalledoncheckpointsignal(f):
+    "Checks if the function is called on checkpoint signal"
+    return _oncheckpointsignal.isinstalledfuncinlist(f)
+
 # ----------------------------------------------------------------------------
 def callfromparticleinjection(f):
     installparticleinjection(f)
-- 
cgit v1.2.3