|
| 1 | +#!/usr/bin/env python |
| 2 | +# -*- coding: utf-8; -*- |
| 3 | + |
| 4 | +''' |
| 5 | +Generate a Python extension module that exports macros from |
| 6 | +/usr/include/linux/input.h |
| 7 | +''' |
| 8 | + |
| 9 | +import os, sys, re |
| 10 | + |
| 11 | + |
| 12 | +template = r''' |
| 13 | +#include <Python.h> |
| 14 | +#include <linux/input.h> |
| 15 | +
|
| 16 | +/* Automatically generated by evdev.genecodes */ |
| 17 | +/* Generated on %s */ |
| 18 | +
|
| 19 | +#define MODULE_NAME "_ecodes" |
| 20 | +#define MODULE_HELP "linux/input.h macros" |
| 21 | +
|
| 22 | +static PyMethodDef MethodTable[] = { |
| 23 | + { NULL, NULL, 0, NULL} |
| 24 | +}; |
| 25 | +
|
| 26 | +#if PY_MAJOR_VERSION >= 3 |
| 27 | +static struct PyModuleDef moduledef = { |
| 28 | + PyModuleDef_HEAD_INIT, |
| 29 | + MODULE_NAME, |
| 30 | + MODULE_HELP, |
| 31 | + -1, /* m_size */ |
| 32 | + MethodTable, /* m_methods */ |
| 33 | + NULL, /* m_reload */ |
| 34 | + NULL, /* m_traverse */ |
| 35 | + NULL, /* m_clear */ |
| 36 | + NULL, /* m_free */ |
| 37 | +}; |
| 38 | +#endif |
| 39 | +
|
| 40 | +static PyObject * |
| 41 | +moduleinit(void) |
| 42 | +{ |
| 43 | +
|
| 44 | +#if PY_MAJOR_VERSION >= 3 |
| 45 | + PyObject* m = PyModule_Create(&moduledef); |
| 46 | +#else |
| 47 | + PyObject* m = Py_InitModule3(MODULE_NAME, MethodTable, MODULE_HELP); |
| 48 | +#endif |
| 49 | +
|
| 50 | + if (m == NULL) return NULL; |
| 51 | +
|
| 52 | +%s |
| 53 | +
|
| 54 | + return m; |
| 55 | +} |
| 56 | +
|
| 57 | +#if PY_MAJOR_VERSION >= 3 |
| 58 | +PyMODINIT_FUNC |
| 59 | +PyInit__ecodes(void) |
| 60 | +{ |
| 61 | + return moduleinit(); |
| 62 | +} |
| 63 | +#else |
| 64 | +PyMODINIT_FUNC |
| 65 | +init_ecodes(void) |
| 66 | +{ |
| 67 | + moduleinit(); |
| 68 | +} |
| 69 | +#endif |
| 70 | +''' |
| 71 | + |
| 72 | +header = '/usr/include/linux/input.h' if len(sys.argv) == 1 else sys.argv[1] |
| 73 | +regex = r'#define +((?:KEY|ABS|REL|SW|MSC|LED|BTN|REP|SND|ID|EV|BUS|SYN|FF)_\w+)' |
| 74 | +regex = re.compile(regex) |
| 75 | + |
| 76 | +if not os.path.exists(header): |
| 77 | + print('no such file: %s' % header) |
| 78 | + sys.exit(1) |
| 79 | + |
| 80 | +def getmacros(): |
| 81 | + for line in open(header): |
| 82 | + macro = regex.search(line) |
| 83 | + if macro: |
| 84 | + yield ' PyModule_AddIntMacro(m, %s);' % macro.group(1) |
| 85 | + |
| 86 | +uname = list(os.uname()); del uname[1] |
| 87 | +uname = ' '.join(uname) |
| 88 | + |
| 89 | +macros = os.linesep.join(getmacros()) |
| 90 | +print(template % (uname, macros)) |
0 commit comments