-
Notifications
You must be signed in to change notification settings - Fork 36
/
Copy pathsetup.py
43 lines (35 loc) · 889 Bytes
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
import setuptools
import sys
class NumpyImport:
def __repr__(self):
import numpy as np
return np.get_include()
__fspath__ = __repr__
# NOTE: If edt.cpp does not exist:
# cython -3 --fast-fail -v --cplus edt.pyx
extra_compile_args = []
if sys.platform == 'win32':
extra_compile_args += [
'/std:c++17', '/O2'
]
else:
extra_compile_args += [
'-std=c++17', '-O3', '-ffast-math', '-pthread'
]
if sys.platform == 'darwin':
extra_compile_args += [ '-stdlib=libc++', '-mmacosx-version-min=10.9' ]
setuptools.setup(
setup_requires=['pbr', 'cython'],
python_requires=">=3.8,<4",
ext_modules=[
setuptools.Extension(
'edt',
sources=[ 'src/edt.pyx' ],
language='c++',
include_dirs=[ 'src', str(NumpyImport()) ],
extra_compile_args=extra_compile_args,
),
],
long_description_content_type='text/markdown',
pbr=True
)