Release GIL when calling getState() (#3061)

5c1d1022 · Peter Eastman · GitHub · c9c27264 · 5c1d1022
Unverified Commit 5c1d1022 authored Mar 12, 2021 by Peter Eastman Committed by GitHub Mar 12, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 14 additions and 0 deletions

wrappers/python/src/swig_doxygen/swig_lib/python/exceptions.i ...pers/python/src/swig_doxygen/swig_lib/python/exceptions.i +14 -0

No files found.
--- a/wrappers/python/src/swig_doxygen/swig_lib/python/exceptions.i
+++ b/wrappers/python/src/swig_doxygen/swig_lib/python/exceptions.i
@@ -10,6 +10,20 @@
    }
 }
+%exception *::getState {
+    PyThreadState* _savePythonThreadState = PyEval_SaveThread();
+    try {
+        $action
+    } catch (std::exception &e) {
+        PyEval_RestoreThread(_savePythonThreadState);
+        PyObject* mm = PyImport_AddModule("openmm");
+        PyObject* openmm_exception = PyObject_GetAttrString(mm, "OpenMMException");
+        PyErr_SetString(openmm_exception, const_cast<char*>(e.what()));
+        return NULL;
+    }
+    PyEval_RestoreThread(_savePythonThreadState);
+}
 %exception *::step {
    PyThreadState* _savePythonThreadState = PyEval_SaveThread();
    try {