diff options
| -rw-r--r-- | c_extension.c | 13 | ||||
| -rw-r--r-- | setup_c_extension.py | 18 |
2 files changed, 28 insertions, 3 deletions
diff --git a/c_extension.c b/c_extension.c index 00da503..6cfadba 100644 --- a/c_extension.c +++ b/c_extension.c @@ -42,8 +42,17 @@ POSSIBILITY OF SUCH DAMAGE. #include "helpers.h" -#include </usr/lib64/python3.6/site-packages/numpy/core/include/numpy/arrayobject.h> - +/* On Debian Linux and its derivatives (such as ubuntu), Python libraries + * installed through the package manager are kept in a non-standard directory + * 'dist-packages' instead of 'site-packages'. Why? Who knows. + * + * https://wiki.debian.org/Python#Deviations_from_upstream + */ +#ifdef DEBIAN_LINUX + #include </usr/lib/python3/dist-packages/numpy/core/include/numpy/arrayobject.h> +#else + #include </usr/lib64/python3.6/site-packages/numpy/core/include/numpy/arrayobject.h> +#endif #define DEFAULT_FRAME_RATE 10 ///< default frame rate diff --git a/setup_c_extension.py b/setup_c_extension.py index 695c88c..9bed938 100644 --- a/setup_c_extension.py +++ b/setup_c_extension.py @@ -1,8 +1,24 @@ from distutils.core import setup, Extension + +extra_compile_args = [] + +# Detect if distribution is Ubuntu or Debian based +with open('/etc/os-release') as f: + lines = filter(None, f.read().split('\n')) + + info = {} + for (key, val) in map(lambda x: x.split('='), lines): + info[key] = val + + # The ID_LIKE key is usually a space separated list of OSes upon which the distribution is based. + if any(os in ('ubuntu', 'debian') for os in info['ID_LIKE'].split()): + extra_compile_args.append('-DDEBIAN_LINUX') + + setup(name = 'o3000', version = '1.0', ext_modules = [Extension( 'o3000', sources = ['c_extension.c', 'helpers.c'], libraries = ['o3000','o3000_imgpipe','tiff'], - #extra_compile_args = ['-Wextra'] + extra_compile_args = extra_compile_args # + ['-Wextra'] )]) |