add device.leds and _input.get_sw_led_snd

Author: gvalkov

evdev/device.py

@@ -141,7 +141,7 @@ def capabilities(self, verbose=False, absinfo=True):
           { ('EV_KEY', 1) : [('BTN_MOUSE', 272), ('BTN_RIGHT', 273), ('BTN_MIDDLE', 273)],
             ('EV_REL', 2) : [('REL_X', 0), ('REL_Y', 0), ('REL_HWHEEL', 6), ('REL_WHEEL', 8)] }
 
-        Unknown codes or types will be resolved to '?'.
+        Unknown codes or types will be resolved to ``'?'``.
 
         If ``absinfo`` is ``True``, the list of capabilities will also
         include absolute axis information (``absmin``, ``absmax``,
@@ -162,8 +162,26 @@ def capabilities(self, verbose=False, absinfo=True):
         else:
             return self._capabilities(absinfo)
 
+    def leds(self, verbose=False):
+        '''
+        Returns currently set LED keys. Example::
+
+          [0, 1, 8, 9]
+
+        If ``verbose`` is ``True``, event codes will be resolved to
+        their names. Unknown codes will be resolved to ``'?'``. Example::
+
+          [('LED_NUML', 0), ('LED_CAPSL', 1), ('LED_MISC', 8), ('LED_MAIL', 9)]
+
+        '''
+        leds = _input.get_sw_led_snd(self.fd, ecodes.EV_LED)
+        if verbose:
+            return [(ecodes.LED[l] if l in ecodes.LED else '?', l) for l in leds]
+
+        return leds
+
     def __eq__(self, o):
-        ''' Two devices are considered equal if their :data:`info` attributes are equal. '''
+        '''Two devices are considered equal if their :data:`info` attributes are equal.'''
         return self.info == o.info
 
     def __str__(self):

evdev/input.c

@@ -293,6 +293,44 @@ ioctl_EVIOCGRAB(PyObject *self, PyObject *args)
 }
 
 
+// todo: this function needs a better name
+static PyObject *
+get_sw_led_snd(PyObject *self, PyObject *args)
+{
+    int i, max, fd, evtype, ret;
+    PyObject* res = PyList_New(0);
+
+    ret = PyArg_ParseTuple(args, "ii", &fd, &evtype);
+    if (!ret) return NULL;
+
+    if (evtype == EV_LED)
+        max = LED_MAX;
+    else if (evtype == EV_SW)
+        max = SW_MAX;
+    else if (evtype == EV_SND)
+        max = SND_MAX;
+
+    char* bits = (char*) malloc(max/8);
+    memset(bits, 0, sizeof(bits));
+
+    if (evtype == EV_LED)
+        ret = ioctl(fd, EVIOCGLED(sizeof(bits)), bits);
+    else if (evtype == EV_SW)
+        ret = ioctl(fd, EVIOCGSW(sizeof(bits)), bits);
+    else if (evtype == EV_SND)
+        ret = ioctl(fd, EVIOCGSND(sizeof(bits)), bits);
+
+    for (i=0 ; i<max ; i++) {
+        if (test_bit(bits, i)) {
+            PyList_Append(res, Py_BuildValue("i", i));
+        }
+    }
+
+    free(bits);
+    return res;
+}
+
+
 static PyMethodDef MethodTable[] = {
     { "unpack",               event_unpack,         METH_VARARGS, "unpack a single input event" },
     { "ioctl_devinfo",        ioctl_devinfo,        METH_VARARGS, "fetch input device info" },
@@ -301,6 +339,7 @@ static PyMethodDef MethodTable[] = {
     { "ioctl_EVIOCSREP",      ioctl_EVIOCSREP,      METH_VARARGS},
     { "ioctl_EVIOCGVERSION",  ioctl_EVIOCGVERSION,  METH_VARARGS},
     { "ioctl_EVIOCGRAB",      ioctl_EVIOCGRAB,      METH_VARARGS},
+    { "get_sw_led_snd",       get_sw_led_snd,       METH_VARARGS},
     { "device_read",          device_read,          METH_VARARGS, "read an input event from a device" },
     { "device_read_many",     device_read_many,     METH_VARARGS, "read all available input events from a device" },