#include "python.hh"
#include "openbox.hh"
-
-#include <vector>
-#include <algorithm>
+#include "actions.hh"
+#include "python.hh"
+#include "bindings.hh"
+#include "otk/display.hh"
+
+extern "C" {
+// The initializer in openbox_wrap.cc
+extern void init_openbox(void);
+// The initializer in otk_wrap.cc
+extern void init_otk(void);
+}
namespace ob {
-typedef std::vector<PyObject*> FunctionList;
+static PyObject *obdict = NULL;
+
+// ************************************************************* //
+// Define some custom types which are passed to python callbacks //
+// ************************************************************* //
+
+typedef struct {
+ PyObject_HEAD;
+ OBActions::ActionType action;
+ Window window;
+ OBWidget::WidgetType type;
+ unsigned int state;
+ unsigned int button;
+ int xroot;
+ int yroot;
+ Time time;
+} ActionData;
+
+typedef struct {
+ PyObject_HEAD;
+ Window window;
+ unsigned int state;
+ unsigned int key;
+ Time time;
+} BindingData;
+
+static void ActionDataDealloc(ActionData *self)
+{
+ PyObject_Del((PyObject*)self);
+}
-static FunctionList callbacks[OBActions::NUM_ACTIONS];
-static FunctionList keyfuncs;
-static FunctionList mousefuncs;
+static void BindingDataDealloc(BindingData *self)
+{
+ PyObject_Del((PyObject*)self);
+}
-bool python_register(int action, PyObject *callback)
+PyObject *ActionData_action(ActionData *self, PyObject *args)
{
- 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;
+ if(!PyArg_ParseTuple(args,":action")) return NULL;
+ return PyLong_FromLong((int)self->action);
}
-bool python_preregister(int action, PyObject *callback)
+PyObject *ActionData_window(ActionData *self, PyObject *args)
{
- 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;
+ if(!PyArg_ParseTuple(args,":window")) return NULL;
+ return PyLong_FromLong(self->window);
}
-bool python_unregister(int action, PyObject *callback)
+PyObject *ActionData_target(ActionData *self, PyObject *args)
{
- 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;
- }
+ if(!PyArg_ParseTuple(args,":target")) return NULL;
+ return PyLong_FromLong((int)self->type);
+}
- 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;
+PyObject *ActionData_modifiers(ActionData *self, PyObject *args)
+{
+ if(!PyArg_ParseTuple(args,":modifiers")) return NULL;
+ return PyLong_FromUnsignedLong(self->state);
}
-bool python_unregister_all(int action)
+PyObject *ActionData_button(ActionData *self, PyObject *args)
{
- if (action < 0 || action >= OBActions::NUM_ACTIONS) {
- PyErr_SetString(PyExc_AssertionError, "Invalid action type.");
- return false;
+ if(!PyArg_ParseTuple(args,":button")) return NULL;
+ int b = 0;
+ switch (self->button) {
+ case Button5: b++;
+ case Button4: b++;
+ case Button3: b++;
+ case Button2: b++;
+ case Button1: b++;
+ default: ;
}
+ return PyLong_FromLong(b);
+}
- while (!callbacks[action].empty()) {
- Py_XDECREF(callbacks[action].back());
- callbacks[action].pop_back();
- }
- return true;
+PyObject *ActionData_xroot(ActionData *self, PyObject *args)
+{
+ if(!PyArg_ParseTuple(args,":xroot")) return NULL;
+ return PyLong_FromLong(self->xroot);
}
-void python_callback(OBActions::ActionType action, Window window,
- OBWidget::WidgetType type, unsigned int state,
- long d1, long d2, long d3, long d4)
+PyObject *ActionData_yroot(ActionData *self, PyObject *args)
{
- PyObject *arglist;
- PyObject *result;
+ if(!PyArg_ParseTuple(args,":yroot")) return NULL;
+ return PyLong_FromLong(self->yroot);
+}
- 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();
- }
- }
+PyObject *ActionData_time(ActionData *self, PyObject *args)
+{
+ if(!PyArg_ParseTuple(args,":time")) return NULL;
+ return PyLong_FromLong(self->time);
+}
- Py_DECREF(arglist);
+static PyMethodDef ActionData_methods[] = {
+ {"action", (PyCFunction)ActionData_action, METH_VARARGS,
+ "Return the action being executed."},
+ {"window", (PyCFunction)ActionData_window, METH_VARARGS,
+ "Return the client window id."},
+ {"target", (PyCFunction)ActionData_target, METH_VARARGS,
+ "Return the target type that the action is occuring on."},
+ {"modifiers", (PyCFunction)ActionData_modifiers, METH_VARARGS,
+ "Return the modifier keys state."},
+ {"button", (PyCFunction)ActionData_button, METH_VARARGS,
+ "Return the number of the pressed button (1-5)."},
+ {"xroot", (PyCFunction)ActionData_xroot, METH_VARARGS,
+ "Return the X-position of the mouse cursor on the root window."},
+ {"yroot", (PyCFunction)ActionData_yroot, METH_VARARGS,
+ "Return the Y-position of the mouse cursor on the root window."},
+ {"time", (PyCFunction)ActionData_time, METH_VARARGS,
+ "Return the time at which the event occured."},
+ {NULL, NULL, 0, NULL}
+};
+
+PyObject *BindingData_window(BindingData *self, PyObject *args)
+{
+ if(!PyArg_ParseTuple(args,":window")) return NULL;
+ return PyLong_FromLong(self->window);
}
+PyObject *BindingData_modifiers(BindingData *self, PyObject *args)
+{
+ if(!PyArg_ParseTuple(args,":modifiers")) return NULL;
+ return PyLong_FromUnsignedLong(self->state);
+}
+PyObject *BindingData_key(BindingData *self, PyObject *args)
+{
+ if(!PyArg_ParseTuple(args,":key")) return NULL;
+ return PyString_FromString(
+ XKeysymToString(XKeycodeToKeysym(otk::OBDisplay::display, self->key, 0)));
+}
+PyObject *BindingData_time(BindingData *self, PyObject *args)
+{
+ if(!PyArg_ParseTuple(args,":time")) return NULL;
+ return PyLong_FromLong(self->time);
+}
+static PyMethodDef BindingData_methods[] = {
+ {"window", (PyCFunction)BindingData_window, METH_VARARGS,
+ "Return the client window id."},
+ {"modifiers", (PyCFunction)BindingData_modifiers, METH_VARARGS,
+ "Return the modifier keys state."},
+ {"key", (PyCFunction)BindingData_key, METH_VARARGS,
+ "Return the name of the pressed key."},
+ {"time", (PyCFunction)BindingData_time, METH_VARARGS,
+ "Return the time at which the event occured."},
+ {NULL, NULL, 0, NULL}
+};
+
+static PyObject *ActionDataGetAttr(PyObject *obj, char *name)
+{
+ return Py_FindMethod(ActionData_methods, obj, name);
+}
-bool python_bind_key(PyObject *keylist, PyObject *callback)
+static PyObject *BindingDataGetAttr(PyObject *obj, char *name)
{
- 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;
- }
+ return Py_FindMethod(BindingData_methods, obj, name);
+}
- 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));
- }
+static PyTypeObject ActionData_Type = {
+ PyObject_HEAD_INIT(NULL)
+ 0,
+ "ActionData",
+ sizeof(ActionData),
+ 0,
+ (destructor)ActionDataDealloc,
+ 0,
+ (getattrfunc)ActionDataGetAttr,
+ 0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0
+};
+
+static PyTypeObject BindingData_Type = {
+ PyObject_HEAD_INIT(NULL)
+ 0,
+ "BindingData",
+ sizeof(BindingData),
+ 0,
+ (destructor)BindingDataDealloc,
+ 0,
+ (getattrfunc)BindingDataGetAttr,
+ 0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0
+};
+
+// **************** //
+// End custom types //
+// **************** //
+
+void python_init(char *argv0)
+{
+ Py_SetProgramName(argv0);
+ Py_Initialize();
+ init_otk();
+ init_openbox();
+ PyRun_SimpleString("from _otk import *; from _openbox import *;");
+ PyRun_SimpleString("openbox = Openbox_instance()");
+
+ // set up access to the python global variables
+ PyObject *obmodule = PyImport_AddModule("__main__");
+ obdict = PyModule_GetDict(obmodule);
+
+ // set up the custom types
+ ActionData_Type.ob_type = &PyType_Type;
+ BindingData_Type.ob_type = &PyType_Type;
+}
- // the id is what the binding class can call back with so it doesnt have to
- // worry about the python function pointer
- int id = keyfuncs.size();
- if (Openbox::instance->bindings()->add_key(vectkeylist, id)) {
- Py_XINCREF(callback); // Add a reference to new callback
- keyfuncs.push_back(callback);
- return true;
- } else {
- PyErr_SetString(PyExc_AssertionError,"Unable to create binding. Invalid.");
- return false;
- }
+void python_destroy()
+{
+ Py_DECREF(obdict);
}
-bool python_unbind_key(PyObject *keylist)
+bool python_exec(const std::string &path)
{
- if (!PyList_Check(keylist)) {
- PyErr_SetString(PyExc_AssertionError, "Invalid keylist. Not a list.");
+ FILE *rcpyfd = fopen(path.c_str(), "r");
+ if (!rcpyfd) {
+ printf("failed to load python file %s\n", path.c_str());
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_key(vectkeylist)) >= 0) {
- assert(keyfuncs[id]); // shouldn't be able to remove it twice
- Py_XDECREF(keyfuncs[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.
- keyfuncs[id] = 0;
- return true;
- }
-
- return false;
+ PyRun_SimpleFile(rcpyfd, const_cast<char*>(path.c_str()));
+ fclose(rcpyfd);
+ return true;
}
-bool python_bind_mouse(const std::string &button, PyObject *callback)
+static void call(PyObject *func, PyObject *data)
{
- if (!PyCallable_Check(callback)) {
- PyErr_SetString(PyExc_AssertionError, "Invalid callback function.");
- return false;
- }
+ PyObject *arglist;
+ PyObject *result;
- // the id is what the binding class can call back with so it doesnt have to
- // worry about the python function pointer
- int id = mousefuncs.size();
- if (Openbox::instance->bindings()->add_mouse(button, id)) {
- Py_XINCREF(callback); // Add a reference to new callback
- mousefuncs.push_back(callback);
- return true;
- } else {
- PyErr_SetString(PyExc_AssertionError,"Unable to create binding. Invalid.");
- return false;
+ arglist = Py_BuildValue("(O)", data);
+
+ // call the callback
+ result = PyEval_CallObject(func, arglist);
+ if (!result) {
+ // an exception occured in the script, display it
+ PyErr_Print();
}
+
+ Py_XDECREF(result);
+ Py_DECREF(arglist);
}
-bool python_unbind_mouse(const std::string &button)
+void python_callback(PyObject *func, OBActions::ActionType action,
+ Window window, OBWidget::WidgetType type,
+ unsigned int state, unsigned int button,
+ int xroot, int yroot, Time time)
{
- int id;
- if ((id =
- Openbox::instance->bindings()->remove_mouse(button)) >= 0) {
- assert(mousefuncs[id]); // shouldn't be able to remove it twice
- Py_XDECREF(mousefuncs[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.
- mousefuncs[id] = 0;
- return true;
- }
+ assert(func);
- return false;
+ ActionData *data = PyObject_New(ActionData, &ActionData_Type);
+ data->action = action;
+ data->window = window;
+ data->type = type;
+ data->state = state;
+ data->button = button;
+ data->xroot = xroot;
+ data->yroot = yroot;
+ data->time = time;
+
+ call(func, (PyObject*)data);
+ Py_DECREF(data);
}
-bool python_unbind_all()
+void python_callback(PyObject *func, Window window, unsigned int state,
+ unsigned int key, Time time)
{
- Openbox::instance->bindings()->remove_all();
- return true;
-}
+ if (!func) return;
+ BindingData *data = PyObject_New(BindingData, &BindingData_Type);
+ data->window = window;
+ data->state = state;
+ data->key = key;
+ data->time = time;
-void python_callback_binding(int id, OBActions::ActionType action,
- Window window, unsigned int state,
- unsigned int keybutton, Time time)
+ call(func, (PyObject*)data);
+ Py_DECREF(data);
+}
+
+bool python_get_string(const char *name, std::string *value)
{
- PyObject *func;
+ PyObject *val = PyDict_GetItemString(obdict, const_cast<char*>(name));
+ if (!(val && PyString_Check(val))) return false;
- assert(action >= 0 && action < OBActions::NUM_ACTIONS);
-
- if (action == OBActions::Action_KeyPress)
- func = keyfuncs[id];
- else
- func = mousefuncs[id];
-
- if (!func) return;
-
- PyObject *arglist;
- PyObject *result;
+ *value = PyString_AsString(val);
+ return true;
+}
- arglist = Py_BuildValue("iliil", action, window, state, keybutton, time);
+bool python_get_stringlist(const char *name, std::vector<std::string> *value)
+{
+ PyObject *val = PyDict_GetItemString(obdict, const_cast<char*>(name));
+ if (!(val && PyList_Check(val))) return false;
- // call the callback
- result = PyEval_CallObject(func, arglist);
- if (result) {
- Py_DECREF(result);
- } else {
- // an exception occured in the script, display it
- PyErr_Print();
+ for (int i = 0, end = PyList_Size(val); i < end; ++i) {
+ PyObject *str = PyList_GetItem(val, i);
+ if (PyString_Check(str))
+ value->push_back(PyString_AsString(str));
}
-
- Py_DECREF(arglist);
+ return true;
}
}