Fix GIL for OpenMM::CustomCVForce::getCollectiveVariableValues (#3463)

5e687c30 · Raimondas Galvelis · GitHub · 4358f071 · 5e687c30
Unverified Commit 5e687c30 authored Feb 15, 2022 by Raimondas Galvelis Committed by GitHub Feb 15, 2022
Show 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
@@ -65,3 +65,17 @@
    }
    PyEval_RestoreThread(_savePythonThreadState);
 }
+
+%exception OpenMM::CustomCVForce::getCollectiveVariableValues {
+    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);
+}
\ No newline at end of file