Functionality to set device properties eg. INPUT_PROP_DIRECT or INPUT_PROP_POINTER

Author: quoing

.gitignore

@@ -13,6 +13,7 @@ tags
 TAGS
 evdev/*.so
 evdev/ecodes.c
+evdev/iprops.c
 docs/_build
 .#*
 __pycache__

evdev/device.py

@@ -223,6 +223,11 @@ def capabilities(self, verbose=False, absinfo=True):
         else:
             return self._capabilities(absinfo)
 
+    def props(self):
+        props = _input.ioctl_EVIOCGPROP(self.fd)
+
+        return props
+
     def leds(self, verbose=False):
         '''
         Return currently set LED keys.

evdev/geniprops.py

@@ -0,0 +1,116 @@
+# -*- coding: utf-8; -*-
+
+'''
+Generate a Python extension module with the constants defined in linux/input.h.
+'''
+
+from __future__ import print_function
+import os, sys, re
+
+
+#-----------------------------------------------------------------------------
+# The default header file locations to try.
+headers = [
+    '/usr/include/linux/input.h',
+    '/usr/include/linux/input-event-codes.h',
+]
+
+if sys.argv[1:]:
+    headers = sys.argv[1:]
+
+
+#-----------------------------------------------------------------------------
+macro_regex = r'#define +((?:INPUT_PROP)_\w+).*'
+macro_regex = re.compile(macro_regex)
+
+uname = list(os.uname()); del uname[1]
+uname = ' '.join(uname)
+
+
+#-----------------------------------------------------------------------------
+template = r'''
+#include <Python.h>
+#ifdef __FreeBSD__
+#include <dev/evdev/input.h>
+#else
+#include <linux/input.h>
+#include <linux/uinput.h>
+#endif
+
+/* Automatically generated by evdev.geniprops */
+/* Generated on %s */
+
+#define MODULE_NAME "iprops"
+#define MODULE_HELP "linux/input.h macros"
+
+static PyMethodDef MethodTable[] = {
+    { NULL, NULL, 0, NULL}
+};
+
+#if PY_MAJOR_VERSION >= 3
+static struct PyModuleDef moduledef = {
+    PyModuleDef_HEAD_INIT,
+    MODULE_NAME,
+    MODULE_HELP,
+    -1,          /* m_size */
+    MethodTable, /* m_methods */
+    NULL,        /* m_reload */
+    NULL,        /* m_traverse */
+    NULL,        /* m_clear */
+    NULL,        /* m_free */
+};
+#endif
+
+static PyObject *
+moduleinit(void)
+{
+
+#if PY_MAJOR_VERSION >= 3
+    PyObject* m = PyModule_Create(&moduledef);
+#else
+    PyObject* m = Py_InitModule3(MODULE_NAME, MethodTable, MODULE_HELP);
+#endif
+
+    if (m == NULL) return NULL;
+
+%s
+
+    return m;
+}
+
+#if PY_MAJOR_VERSION >= 3
+PyMODINIT_FUNC
+PyInit__iprops(void)
+{
+    return moduleinit();
+}
+#else
+PyMODINIT_FUNC
+init_iprops(void)
+{
+    moduleinit();
+}
+#endif
+'''
+
+def parse_header(header):
+    for line in open(header):
+        macro = macro_regex.search(line)
+        if macro:
+            yield '    PyModule_AddIntMacro(m, %s);' % macro.group(1)
+
+all_macros = []
+for header in headers:
+    try:
+        fh = open(header)
+    except (IOError, OSError):
+        continue
+    all_macros += parse_header(header)
+
+if not all_macros:
+    print('no input macros found in: %s' % ' '.join(headers), file=sys.stderr)
+    sys.exit(1)
+
+
+macros = os.linesep.join(all_macros)
+print(template % (uname, macros))

evdev/input.c

@@ -454,6 +454,33 @@ erase_effect(PyObject *self, PyObject *args)
     return Py_None;
 }
 
+static PyObject *
+ioctl_EVIOCGPROP(PyObject *self, PyObject *args)
+{
+    int fd, ret;
+
+    ret = PyArg_ParseTuple(args, "i", &fd);
+    if (!ret) return NULL;
+
+    char bytes[(INPUT_PROP_MAX+7)/8];
+    memset(bytes, 0, sizeof bytes);
+
+    ret = ioctl(fd, EVIOCGPROP(sizeof(bytes)), &bytes);
+
+    if (ret == -1)
+        return NULL;
+
+    PyObject* res = PyList_New(0);
+    for (int i=0; i<INPUT_PROP_MAX; i++) {
+        if (test_bit(bytes, i)) {
+            PyList_Append(res, Py_BuildValue("i", i));
+        }
+    }
+
+    return res;
+}
+
+
 
 static PyMethodDef MethodTable[] = {
     { "ioctl_devinfo",        ioctl_devinfo,        METH_VARARGS, "fetch input device info" },
@@ -468,6 +495,7 @@ static PyMethodDef MethodTable[] = {
     { "device_read_many",     device_read_many,     METH_VARARGS, "read all available input events from a device" },
     { "upload_effect",        upload_effect,        METH_VARARGS, "" },
     { "erase_effect",         erase_effect,         METH_VARARGS, "" },
+    { "ioctl_EVIOCGPROP",     ioctl_EVIOCGPROP,     METH_VARARGS, "get device properties"},
 
     { NULL, NULL, 0, NULL}
 };

evdev/uinput.c

@@ -77,6 +77,26 @@ uinput_set_phys(PyObject *self, PyObject *args)
         return NULL;
 }
 
+static PyObject *
+uinput_set_prop(PyObject *self, PyObject *args)
+{
+    int fd;
+    uint16_t prop;
+
+    int ret = PyArg_ParseTuple(args, "ih", &fd, &prop);
+    if (!ret) return NULL;
+
+    if (ioctl(fd, UI_SET_PROPBIT, prop) < 0)
+        goto on_err;
+
+    Py_RETURN_NONE;
+
+    on_err:
+        _uinput_close(fd);
+        PyErr_SetFromErrno(PyExc_IOError);
+        return NULL;
+}
+
 
 // Different kernel versions have different device setup methods. You can read
 // more about it here:
@@ -339,6 +359,9 @@ static PyMethodDef MethodTable[] = {
     { "set_phys", uinput_set_phys, METH_VARARGS,
       "Set physical path"},
 
+    { "set_prop", uinput_set_prop, METH_VARARGS,
+      "Set device property"},
+
     { NULL, NULL, 0, NULL}
 };
 

evdev/uinput.py

@@ -77,7 +77,7 @@ def __init__(self,
                  events=None,
                  name='py-evdev-uinput',
                  vendor=0x1, product=0x1, version=0x1, bustype=0x3,
-                 devnode='/dev/uinput', phys='py-evdev-uinput'):
+                 devnode='/dev/uinput', phys='py-evdev-uinput', props=[]):
         '''
         Arguments
         ---------
@@ -132,6 +132,10 @@ def __init__(self,
         # Set phys name
         _uinput.set_phys(self.fd, phys)
 
+        # Set properties
+        for prop in props:
+            _uinput.set_prop(self.fd, prop)
+
         for etype, code in prepared_events:
             _uinput.enable(self.fd, etype, code)
 

setup.py

@@ -40,6 +40,7 @@
 input_c  = Extension('evdev._input',  sources=['evdev/input.c'],  extra_compile_args=cflags)
 uinput_c = Extension('evdev._uinput', sources=['evdev/uinput.c'], extra_compile_args=cflags)
 ecodes_c = Extension('evdev._ecodes', sources=['evdev/ecodes.c'], extra_compile_args=cflags)
+iprops_c = Extension('evdev.iprops', sources=['evdev/iprops.c'], extra_compile_args=cflags)
 
 #-----------------------------------------------------------------------------
 kw = {
@@ -57,7 +58,7 @@
     'classifiers':          classifiers,
 
     'packages':             ['evdev'],
-    'ext_modules':          [input_c, uinput_c, ecodes_c],
+    'ext_modules':          [input_c, uinput_c, ecodes_c, iprops_c],
     'include_package_data': False,
     'zip_safe':             True,
     'cmdclass':             {},
@@ -106,6 +107,48 @@ def create_ecodes(headers=None):
     check_call(cmd, cwd="%s/evdev" % here, shell=True)
 
 
+#-----------------------------------------------------------------------------
+def create_iprops(headers=None):
+    if not headers:
+        headers = [
+            '/usr/include/linux/input.h',
+            '/usr/include/linux/input-event-codes.h',
+            '/usr/include/linux/uinput.h',
+        ]
+
+    headers = [header for header in headers if os.path.isfile(header)]
+    if not headers:
+        msg = '''\
+        The 'linux/input.h' and 'linux/input-event-codes.h' include files
+        are missing. You will have to install the kernel header files in
+        order to continue:
+
+            yum install kernel-headers-$(uname -r)
+            apt-get install linux-headers-$(uname -r)
+            emerge sys-kernel/linux-headers
+            pacman -S kernel-headers
+
+        In case they are installed in a non-standard location, you may use
+        the '--evdev-headers' option to specify one or more colon-separated
+        paths. For example:
+
+            python setup.py \\
+              build \\
+              build_iprops --evdev-headers path/input.h:path/input-event-codes.h \\
+              build_ext --include-dirs  path/ \\
+              install
+        '''
+
+        sys.stderr.write(textwrap.dedent(msg))
+        sys.exit(1)
+
+    from subprocess import check_call
+
+    print('writing iprops.c (using %s)' % ' '.join(headers))
+    cmd = '%s geniprops.py %s > iprops.c' % (sys.executable, ' '.join(headers))
+    check_call(cmd, cwd="%s/evdev" % here, shell=True)
+
+
 #-----------------------------------------------------------------------------
 class build_ecodes(Command):
     description = 'generate ecodes.c'
@@ -125,6 +168,24 @@ def run(self):
         create_ecodes(self.evdev_headers)
 
 
+class build_iprops(Command):
+    description = 'generate iprops.c'
+
+    user_options = [
+        ('evdev-headers=', None, 'colon-separated paths to input subsystem headers'),
+    ]
+
+    def initialize_options(self):
+        self.evdev_headers = None
+
+    def finalize_options(self):
+        if self.evdev_headers:
+            self.evdev_headers = self.evdev_headers.split(':')
+
+    def run(self):
+        create_iprops(self.evdev_headers)
+
+
 class build_ext(_build_ext.build_ext):
     def has_ecodes(self):
         ecodes_path = os.path.join(here, 'evdev/ecodes.c')
@@ -133,17 +194,25 @@ def has_ecodes(self):
             print('ecodes.c already exists ... skipping build_ecodes')
         return not res
 
+    def has_iprops(self):
+        iprops_path = os.path.join(here, 'evdev/iprops.c')
+        res = os.path.exists(iprops_path)
+        if res:
+            print('iprops.c already exists ... skipping build_iprops')
+        return not res
+
     def run(self):
         for cmd_name in self.get_sub_commands():
             self.run_command(cmd_name)
         _build_ext.build_ext.run(self)
 
-    sub_commands =  [('build_ecodes', has_ecodes)] + _build_ext.build_ext.sub_commands
+    sub_commands =  [('build_ecodes', has_ecodes), ('build_iprops', has_iprops)] + _build_ext.build_ext.sub_commands
 
 
 #-----------------------------------------------------------------------------
 kw['cmdclass']['build_ext'] = build_ext
 kw['cmdclass']['build_ecodes'] = build_ecodes
+kw['cmdclass']['build_iprops'] = build_iprops
 
 
 #-----------------------------------------------------------------------------