Dogcows Code - chaz/openbox/blob - 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   PyRun_SimpleString("openbox = Openbox_instance()");
 216
 217   // set up access to the python global variables
 218   PyObject *obmodule = PyImport_AddModule("__main__");
 219   obdict = PyModule_GetDict(obmodule);
 220
 221   // set up the custom types
 222   ActionData_Type.ob_type = &PyType_Type;
 223   BindingData_Type.ob_type = &PyType_Type;
 224 }
 225
 226 void python_destroy()
 227 {
 228   Py_DECREF(obdict);
 229 }
 230
 231 bool python_exec(const std::string &path)
 232 {
 233   FILE *rcpyfd = fopen(path.c_str(), "r");
 234   if (!rcpyfd) {
 235     printf("failed to load python file %s\n", path.c_str());
 236     return false;
 237   }
 238   PyRun_SimpleFile(rcpyfd, const_cast<char*>(path.c_str()));
 239   fclose(rcpyfd);
 240   return true;
 241 }
 242
 243 static void call(PyObject *func, PyObject *data)
 244 {
 245   PyObject *arglist;
 246   PyObject *result;
 247
 248   arglist = Py_BuildValue("(O)", data);
 249
 250   // call the callback
 251   result = PyEval_CallObject(func, arglist);
 252   if (!result) {
 253     // an exception occured in the script, display it
 254     PyErr_Print();
 255   }
 256
 257   Py_XDECREF(result);
 258   Py_DECREF(arglist);
 259 }
 260
 261 void python_callback(PyObject *func, OBActions::ActionType action,
 262                      Window window, OBWidget::WidgetType type,
 263                      unsigned int state, unsigned int button,
 264                      int xroot, int yroot, Time time)
 265 {
 266   assert(func);
 267
 268   ActionData *data = PyObject_New(ActionData, &ActionData_Type);
 269   data->action = action;
 270   data->window = window;
 271   data->type   = type;
 272   data->state  = state;
 273   data->button = button;
 274   data->xroot  = xroot;
 275   data->yroot  = yroot;
 276   data->time   = time;
 277
 278   call(func, (PyObject*)data);
 279   Py_DECREF(data);
 280 }
 281
 282 void python_callback(PyObject *func, Window window, unsigned int state,
 283                          unsigned int key, Time time)
 284 {
 285   if (!func) return;
 286
 287   BindingData *data = PyObject_New(BindingData, &BindingData_Type);
 288   data->window = window;
 289   data->state  = state;
 290   data->key    = key;
 291   data->time   = time;
 292
 293   call(func, (PyObject*)data);
 294   Py_DECREF(data);
 295 }
 296
 297 bool python_get_string(const char *name, std::string *value)
 298 {
 299   PyObject *val = PyDict_GetItemString(obdict, const_cast<char*>(name));
 300   if (!(val && PyString_Check(val))) return false;
 301
 302   *value = PyString_AsString(val);
 303   return true;
 304 }
 305
 306 bool python_get_stringlist(const char *name, std::vector<std::string> *value)
 307 {
 308   PyObject *val = PyDict_GetItemString(obdict, const_cast<char*>(name));
 309   if (!(val && PyList_Check(val))) return false;
 310
 311   for (int i = 0, end = PyList_Size(val); i < end; ++i) {
 312     PyObject *str = PyList_GetItem(val, i);
 313     if (PyString_Check(str))
 314       value->push_back(PyString_AsString(str));
 315   }
 316   return true;
 317 }
 318
 319 }