aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--c_extension.c13
-rw-r--r--setup_c_extension.py18
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']
)])