aboutsummaryrefslogtreecommitdiffstats
path: root/setup_c_extension.py
diff options
context:
space:
mode:
Diffstat (limited to 'setup_c_extension.py')
-rw-r--r--setup_c_extension.py22
1 files changed, 7 insertions, 15 deletions
diff --git a/setup_c_extension.py b/setup_c_extension.py
index 9bed938..e01837f 100644
--- a/setup_c_extension.py
+++ b/setup_c_extension.py
@@ -1,24 +1,16 @@
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')
+# Find where numpy is installed to get the header
+import pathlib
+import numpy
+numpy_headers_path = pathlib.Path(numpy.__file__).parent.joinpath("core/include")
setup(name = 'o3000', version = '1.0',
ext_modules = [Extension(
'o3000',
sources = ['c_extension.c', 'helpers.c'],
- libraries = ['o3000','o3000_imgpipe','tiff'],
- extra_compile_args = extra_compile_args # + ['-Wextra']
+ libraries = ['o3000', 'o3000_imgpipe', 'tiff'],
+ include_dirs = [str(numpy_headers_path)],
+ # extra_compile_args = ['-Wextra']
)])