X-Git-Url: https://git.dogcows.com/gitweb?a=blobdiff_plain;f=src%2Fpython.cc;h=dbe35f5d0f3a0159cdc0f14da9c28fef5dffd01e;hb=1161a90a70b21d3064a9dee62c72dd4be3025ada;hp=2c2f9ae3b90c6c467ae87e6d6c651f11e7c48777;hpb=68194ce957db36ead19a39fdc7636a220befafe9;p=chaz%2Fopenbox diff --git a/src/python.cc b/src/python.cc index 2c2f9ae3..dbe35f5d 100644 --- a/src/python.cc +++ b/src/python.cc @@ -1,45 +1,243 @@ // -*- mode: C++; indent-tabs-mode: nil; c-basic-offset: 2; -*- -#ifdef HAVE_CONFIG_H -# include "../config.h" -#endif - #include "python.hh" -#include "client.hh" #include "openbox.hh" +#include "otk/display.hh" + +#include +#include namespace ob { -extern "C" { +typedef std::vector FunctionList; + +static FunctionList callbacks[OBActions::NUM_ACTIONS]; +static FunctionList bindfuncs; -static PyObject *shit(PyObject *self, PyObject *args) +bool python_register(int action, PyObject *callback) { - if (!PyArg_ParseTuple(args, ":shit")) - return NULL; + if (action < 0 || action >= OBActions::NUM_ACTIONS || + action == OBActions::Action_KeyPress) { + 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; +} - printf("SHIT CALLED!@!\n"); +bool python_preregister(int action, PyObject *callback) +{ + if (action < 0 || action >= OBActions::NUM_ACTIONS || + action == OBActions::Action_KeyPress) { + 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].insert(callbacks[action].begin(), callback); + } + return true; +} - return Py_None; +bool python_unregister(int action, PyObject *callback) +{ + if (action < 0 || action >= OBActions::NUM_ACTIONS || + action == OBActions::Action_KeyPress) { + 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; +} + +bool python_unregister_all(int action) +{ + if (action < 0 || action >= OBActions::NUM_ACTIONS) { + PyErr_SetString(PyExc_AssertionError, "Invalid action type."); + return false; + } + + while (!callbacks[action].empty()) { + Py_XDECREF(callbacks[action].back()); + callbacks[action].pop_back(); + } + return true; +} + +void python_callback(OBActions::ActionType action, Window window, + OBWidget::WidgetType type, unsigned int state, + long d1, long d2, long d3, long d4) +{ + PyObject *arglist; + PyObject *result; + + assert(action >= 0 && action < OBActions::NUM_ACTIONS); + + if (d4 != LONG_MIN) + arglist = Py_BuildValue("iliillll", action, window, type, state, + d1, d2, d3, d4); + else if (d3 != LONG_MIN) + arglist = Py_BuildValue("iliilll", action, window, type, state, + d1, d2, d3); + else if (d2 != LONG_MIN) + arglist = Py_BuildValue("iliill", action, window, type, state, d1, d2); + else if (d1 != LONG_MIN) + arglist = Py_BuildValue("iliil", action, window, type, state, d1); + else + arglist = Py_BuildValue("ilii", action, window, type, state); + + 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); } - -static PyMethodDef OBMethods[] = { - {"shit", shit, METH_VARARGS, - "Do some shit, yo!"}, -/* {"get_client_dict", get_client_dict, METH_VARARGS, - "Get the list of all clients"},*/ - {NULL, NULL, 0, NULL} -}; -void initopenbox() + +bool python_bind(PyObject *keylist, PyObject *callback) +{ + if (!PyList_Check(keylist)) { + PyErr_SetString(PyExc_AssertionError, "Invalid keylist. Not a list."); + return false; + } + if (!PyCallable_Check(callback)) { + PyErr_SetString(PyExc_AssertionError, "Invalid callback function."); + return false; + } + + OBBindings::StringVect vectkeylist; + for (int i = 0, end = PyList_Size(keylist); i < end; ++i) { + PyObject *str = PyList_GetItem(keylist, i); + if (!PyString_Check(str)) { + PyErr_SetString(PyExc_AssertionError, + "Invalid keylist. It must contain only strings."); + return false; + } + vectkeylist.push_back(PyString_AsString(str)); + } + + // the id is what the binding class can call back with so it doesnt have to + // worry about the python function pointer + int id = bindfuncs.size(); + if (Openbox::instance->bindings()->add(vectkeylist, id)) { + Py_XINCREF(callback); // Add a reference to new callback + bindfuncs.push_back(callback); + return true; + } else { + PyErr_SetString(PyExc_AssertionError,"Unable to create binding. Invalid."); + return false; + } +} + +bool python_unbind(PyObject *keylist) { - PyClient_Type.ob_type = &PyType_Type; + if (!PyList_Check(keylist)) { + PyErr_SetString(PyExc_AssertionError, "Invalid keylist. Not a list."); + return false; + } + + OBBindings::StringVect vectkeylist; + for (int i = 0, end = PyList_Size(keylist); i < end; ++i) { + PyObject *str = PyList_GetItem(keylist, i); + if (!PyString_Check(str)) { + PyErr_SetString(PyExc_AssertionError, + "Invalid keylist. It must contain only strings."); + return false; + } + vectkeylist.push_back(PyString_AsString(str)); + } + + int id; + if ((id = + Openbox::instance->bindings()->remove(vectkeylist)) >= 0) { + assert(bindfuncs[id]); // shouldn't be able to remove it twice + Py_XDECREF(bindfuncs[id]); // Dispose of previous callback + // important note: we don't erase the item from the list cuz that would + // ruin all the id's that are in use. simply nullify it. + bindfuncs[id] = 0; + return true; + } - Py_InitModule("openbox", OBMethods); + return false; +} + +void python_set_reset_key(const std::string &key) +{ + Openbox::instance->bindings()->setResetKey(key); } + +void python_unbind_all() +{ + Openbox::instance->bindings()->remove_all(); +} + + +void python_callback_binding(int id, Window window, unsigned int state, + unsigned int keybutton, Time time) +{ + if (!bindfuncs[id]) return; // the key was unbound + + PyObject *arglist; + PyObject *result; + + arglist = Py_BuildValue("lisl", window, state, + XKeysymToString( + XKeycodeToKeysym(otk::OBDisplay::display, + keybutton, 0)), + time); + + // call the callback + result = PyEval_CallObject(bindfuncs[id], arglist); + if (result) { + Py_DECREF(result); + } else { + // an exception occured in the script, display it + PyErr_Print(); + } + + Py_DECREF(arglist); } }