-
Notifications
You must be signed in to change notification settings - Fork 5
/
Copy pathsetup.py
57 lines (42 loc) · 1.07 KB
/
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
44
45
46
47
48
49
50
51
52
import os
import setuptools
import sys
import numpy as np
class NumpyImport:
def __repr__(self):
import numpy as np
return np.get_include()
__fspath__ = __repr__
join = os.path.join
fpzipdir = 'fpzip-1.3.0'
sources = [
join(fpzipdir, 'src', x) for x in (
'error.cpp', 'rcdecoder.cpp', 'rcencoder.cpp',
'rcqsmodel.cpp', 'write.cpp', 'read.cpp',
)
]
sources += [ 'fpzip.pyx' ]
extra_compile_args = [
'-DFPZIP_FP=FPZIP_FP_FAST',
'-DFPZIP_BLOCK_SIZE=0x1000',
'-DWITH_UNION',
]
if sys.platform == 'darwin':
extra_compile_args += [ '-stdlib=libc++', '-mmacosx-version-min=10.9' ]
if sys.platform == 'win32':
extra_compile_args += [ '/std:c++11', '/O2' ]
else:
extra_compile_args += [ '-O3', '-std=c++11' ]
setuptools.setup(
setup_requires=['pbr', 'numpy','cython'],
python_requires=">=3.8,<4.0",
ext_modules=[
setuptools.Extension(
'fpzip',
sources=sources,
language='c++',
include_dirs=[ join(fpzipdir, 'include'), str(NumpyImport()) ],
extra_compile_args=extra_compile_args,
)
],
pbr=True)