diff options
Diffstat (limited to 'src/python.cc')
| -rw-r--r-- | src/python.cc | 82 |
1 files changed, 82 insertions, 0 deletions
diff --git a/src/python.cc b/src/python.cc new file mode 100644 index 00000000..a00bc245 --- /dev/null +++ b/src/python.cc @@ -0,0 +1,82 @@ +// -*- mode: C++; indent-tabs-mode: nil; c-basic-offset: 2; -*- + +#include "python.hh" + +#include <vector> +#include <algorithm> + +namespace ob { + +typedef std::vector<PyObject*> FunctionList; + +static FunctionList callbacks[OBActions::NUM_ACTIONS]; + +bool python_register(int action, PyObject *callback) +{ + if (action < 0 || action >= OBActions::NUM_ACTIONS) { + PyErr_SetString(PyExc_AssertionError, "Invalid action type."); + return false; + } + if (!PyCallable_Check(callback)) { + PyErr_SetString(PyExc_AssertionError, "Invalid callback function."); + return false; + } + + FunctionList::iterator it = std::find(callbacks[action].begin(), + callbacks[action].end(), + callback); + if (it == callbacks[action].end()) { // not already in there + Py_XINCREF(callback); // Add a reference to new callback + callbacks[action].push_back(callback); + } + return true; +} + +bool python_unregister(int action, PyObject *callback) +{ + if (action < 0 || action >= OBActions::NUM_ACTIONS) { + PyErr_SetString(PyExc_AssertionError, "Invalid action type."); + return false; + } + if (!PyCallable_Check(callback)) { + PyErr_SetString(PyExc_AssertionError, "Invalid callback function."); + return false; + } + + FunctionList::iterator it = std::find(callbacks[action].begin(), + callbacks[action].end(), + callback); + if (it != callbacks[action].end()) { // its been registered before + Py_XDECREF(*it); // Dispose of previous callback + callbacks[action].erase(it); + } + return true; +} + +void python_callback(OBActions::ActionType action, Window window, + OBWidget::WidgetType type, unsigned int state, + long d1, long d2) +{ + PyObject *arglist; + PyObject *result; + + assert(action >= 0 && action < OBActions::NUM_ACTIONS); + + arglist = Py_BuildValue("iliill", action, window, type, state, d1, d2); + + FunctionList::iterator it, end = callbacks[action].end(); + for (it = callbacks[action].begin(); it != end; ++it) { + // call the callback + result = PyEval_CallObject(*it, arglist); + if (result) { + Py_DECREF(result); + } else { + // an exception occured in the script, display it + PyErr_Print(); + } + } + + Py_DECREF(arglist); +} + +} |
