src/python.cc

   1 // -*- mode: C++; indent-tabs-mode: nil; c-basic-offset: 2; -*-
   2
   3 #include "python.hh"
   4 #include "openbox.hh"
   5 #include "actions.hh"
   6 #include "python.hh"
   7 #include "bindings.hh"
   8 #include "otk/display.hh"
   9
  10 extern "C" {
  11 // The initializer in openbox_wrap.cc
  12 extern void init_openbox(void);
  13 // The initializer in otk_wrap.cc
  14 extern void init_otk(void);
  15 }
  16
  17 namespace ob {
  18
  19 static PyObject *obdict = NULL;
  20
  21 // ************************************************************* //
  22 // Define some custom types which are passed to python callbacks //
  23 // ************************************************************* //
  24
  25 typedef struct {
  26   PyObject_HEAD;
  27   OBActions::ActionType action;
  28   Window window;
  29   OBWidget::WidgetType type;
  30   unsigned int state;
  31   unsigned int button;
  32   int xroot;
  33   int yroot;
  34   Time time;
  35 } ActionData;
  36
  37 typedef struct {
  38   PyObject_HEAD;
  39   Window window;
  40   unsigned int state;
  41   unsigned int key;
  42   Time time;
  43 } BindingData;
  44
  45 static void ActionDataDealloc(ActionData *self)
  46 {
  47   PyObject_Del((PyObject*)self);
  48 }
  49
  50 static void BindingDataDealloc(BindingData *self)
  51 {
  52   PyObject_Del((PyObject*)self);
  53 }
  54
  55 PyObject *ActionData_action(ActionData *self, PyObject *args)
  56 {
  57   if(!PyArg_ParseTuple(args,":action")) return NULL;
  58   return PyLong_FromLong((int)self->action);
  59 }
  60
  61 PyObject *ActionData_window(ActionData *self, PyObject *args)
  62 {
  63   if(!PyArg_ParseTuple(args,":window")) return NULL;
  64   return PyLong_FromLong(self->window);
  65 }
  66
  67 PyObject *ActionData_target(ActionData *self, PyObject *args)
  68 {
  69   if(!PyArg_ParseTuple(args,":target")) return NULL;
  70   return PyLong_FromLong((int)self->type);
  71 }
  72
  73 PyObject *ActionData_modifiers(ActionData *self, PyObject *args)
  74 {
  75   if(!PyArg_ParseTuple(args,":modifiers")) return NULL;
  76   return PyLong_FromUnsignedLong(self->state);
  77 }
  78
  79 PyObject *ActionData_button(ActionData *self, PyObject *args)
  80 {
  81   if(!PyArg_ParseTuple(args,":button")) return NULL;
  82   int b = 0;
  83   switch (self->button) {
  84   case Button5: b++;
  85   case Button4: b++;
  86   case Button3: b++;
  87   case Button2: b++;
  88   case Button1: b++;
  89   default: ;
  90   }
  91   return PyLong_FromLong(b);
  92 }
  93
  94 PyObject *ActionData_xroot(ActionData *self, PyObject *args)
  95 {
  96   if(!PyArg_ParseTuple(args,":xroot")) return NULL;
  97   return PyLong_FromLong(self->xroot);
  98 }
  99
 100 PyObject *ActionData_yroot(ActionData *self, PyObject *args)
 101 {
 102   if(!PyArg_ParseTuple(args,":yroot")) return NULL;
 103   return PyLong_FromLong(self->yroot);
 104 }
 105
 106 PyObject *ActionData_time(ActionData *self, PyObject *args)
 107 {
 108   if(!PyArg_ParseTuple(args,":time")) return NULL;
 109   return PyLong_FromLong(self->time);
 110 }
 111
 112 static PyMethodDef ActionData_methods[] = {
 113   {"action", (PyCFunction)ActionData_action, METH_VARARGS,
 114    "Return the action being executed."},
 115   {"window", (PyCFunction)ActionData_window, METH_VARARGS,
 116    "Return the client window id."},
 117   {"target", (PyCFunction)ActionData_target, METH_VARARGS,
 118    "Return the target type that the action is occuring on."},
 119   {"modifiers", (PyCFunction)ActionData_modifiers, METH_VARARGS,
 120    "Return the modifier keys state."},
 121   {"button", (PyCFunction)ActionData_button, METH_VARARGS,
 122    "Return the number of the pressed button (1-5)."},
 123   {"xroot", (PyCFunction)ActionData_xroot, METH_VARARGS,
 124    "Return the X-position of the mouse cursor on the root window."},
 125   {"yroot", (PyCFunction)ActionData_yroot, METH_VARARGS,
 126    "Return the Y-position of the mouse cursor on the root window."},
 127   {"time", (PyCFunction)ActionData_time, METH_VARARGS,
 128    "Return the time at which the event occured."},
 129   {NULL, NULL, 0, NULL}
 130 };
 131
 132 PyObject *BindingData_window(BindingData *self, PyObject *args)
 133 {
 134   if(!PyArg_ParseTuple(args,":window")) return NULL;
 135   return PyLong_FromLong(self->window);
 136 }
 137
 138 PyObject *BindingData_modifiers(BindingData *self, PyObject *args)
 139 {
 140   if(!PyArg_ParseTuple(args,":modifiers")) return NULL;
 141   return PyLong_FromUnsignedLong(self->state);
 142 }
 143
 144 PyObject *BindingData_key(BindingData *self, PyObject *args)
 145 {
 146   if(!PyArg_ParseTuple(args,":key")) return NULL;
 147   return PyString_FromString(
 148     XKeysymToString(XKeycodeToKeysym(otk::OBDisplay::display, self->key, 0)));
 149
 150 }
 151
 152 PyObject *BindingData_time(BindingData *self, PyObject *args)
 153 {
 154   if(!PyArg_ParseTuple(args,":time")) return NULL;
 155   return PyLong_FromLong(self->time);
 156 }
 157
 158 static PyMethodDef BindingData_methods[] = {
 159   {"window", (PyCFunction)BindingData_window, METH_VARARGS,
 160    "Return the client window id."},
 161   {"modifiers", (PyCFunction)BindingData_modifiers, METH_VARARGS,
 162    "Return the modifier keys state."},
 163   {"key", (PyCFunction)BindingData_key, METH_VARARGS,
 164    "Return the name of the pressed key."},
 165   {"time", (PyCFunction)BindingData_time, METH_VARARGS,
 166    "Return the time at which the event occured."},
 167   {NULL, NULL, 0, NULL}
 168 };
 169
 170 static PyObject *ActionDataGetAttr(PyObject *obj, char *name)
 171 {
 172   return Py_FindMethod(ActionData_methods, obj, name);
 173 }
 174
 175 static PyObject *BindingDataGetAttr(PyObject *obj, char *name)
 176 {
 177   return Py_FindMethod(BindingData_methods, obj, name);
 178 }
 179
 180 static PyTypeObject ActionData_Type = {
 181   PyObject_HEAD_INIT(NULL)
 182   0,
 183   "ActionData",
 184   sizeof(ActionData),
 185   0,
 186   (destructor)ActionDataDealloc,
 187   0,
 188   (getattrfunc)ActionDataGetAttr,
 189   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
 190 };
 191
 192 static PyTypeObject BindingData_Type = {
 193   PyObject_HEAD_INIT(NULL)
 194   0,
 195   "BindingData",
 196   sizeof(BindingData),
 197   0,
 198   (destructor)BindingDataDealloc,
 199   0,
 200   (getattrfunc)BindingDataGetAttr,
 201   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
 202 };
 203
 204 // **************** //
 205 // End custom types //
 206 // **************** //
 207
 208 void python_init(char *argv0)
 209 {
 210   Py_SetProgramName(argv0);
 211   Py_Initialize();
 212   init_otk();
 213   init_openbox();
 214   PyRun_SimpleString("from _otk import *; from _openbox import *;");
 215
 216   // set up access to the python global variables
 217   PyObject *obmodule = PyImport_AddModule("__main__");
 218   obdict = PyModule_GetDict(obmodule);
 219
 220   // set up the custom types
 221   ActionData_Type.ob_type = &PyType_Type;
 222   BindingData_Type.ob_type = &PyType_Type;
 223 }
 224
 225 void python_destroy()
 226 {
 227   Py_DECREF(obdict);
 228 }
 229
 230 bool python_exec(const std::string &path)
 231 {
 232   FILE *rcpyfd = fopen(path.c_str(), "r");
 233   if (!rcpyfd) {
 234     printf("failed to load python file %s\n", path.c_str());
 235     return false;
 236   }
 237   PyRun_SimpleFile(rcpyfd, const_cast<char*>(path.c_str()));
 238   fclose(rcpyfd);
 239   return true;
 240 }
 241
 242 static void call(PyObject *func, PyObject *data)
 243 {
 244   PyObject *arglist;
 245   PyObject *result;
 246
 247   arglist = Py_BuildValue("(O)", data);
 248
 249   // call the callback
 250   result = PyEval_CallObject(func, arglist);
 251   if (!result) {
 252     // an exception occured in the script, display it
 253     PyErr_Print();
 254   }
 255
 256   Py_XDECREF(result);
 257   Py_DECREF(arglist);
 258 }
 259
 260 void python_callback(PyObject *func, OBActions::ActionType action,
 261                      Window window, OBWidget::WidgetType type,
 262                      unsigned int state, unsigned int button,
 263                      int xroot, int yroot, Time time)
 264 {
 265   assert(func);
 266
 267   ActionData *data = PyObject_New(ActionData, &ActionData_Type);
 268   data->action = action;
 269   data->window = window;
 270   data->type   = type;
 271   data->state  = state;
 272   data->button = button;
 273   data->xroot  = xroot;
 274   data->yroot  = yroot;
 275   data->time   = time;
 276
 277   call(func, (PyObject*)data);
 278   Py_DECREF(data);
 279 }
 280
 281 void python_callback(PyObject *func, Window window, unsigned int state,
 282                          unsigned int key, Time time)
 283 {
 284   if (!func) return;
 285
 286   BindingData *data = PyObject_New(BindingData, &BindingData_Type);
 287   data->window = window;
 288   data->state  = state;
 289   data->key    = key;
 290   data->time   = time;
 291
 292   call(func, (PyObject*)data);
 293   Py_DECREF(data);
 294 }
 295
 296 bool python_get_string(const char *name, std::string *value)
 297 {
 298   PyObject *val = PyDict_GetItemString(obdict, const_cast<char*>(name));
 299   if (!(val && PyString_Check(val))) return false;
 300
 301   *value = PyString_AsString(val);
 302   return true;
 303 }
 304
 305
 306 }