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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
|
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import io
import os
import platform
import sys
from warnings import warn
from setuptools import Extension, setup
from setuptools.command.build_ext import build_ext as _build_ext
from setuptools.command.sdist import sdist as _sdist
from distutils.sysconfig import get_config_var
from distutils.version import LooseVersion
def is_platform_mac():
return sys.platform == 'darwin'
# Alias ModuleNotFound for Python <= 3.5
if sys.version_info < (3, 6):
ModuleNotFoundError = ImportError
# For macOS, ensure extensions are built for macOS 10.9 when compiling on a
# 10.9 system or above, overriding distutils behaviour which is to target
# the version that Python was built for. This may be overridden by setting
# MACOSX_DEPLOYMENT_TARGET before calling setup.py
if is_platform_mac():
if 'MACOSX_DEPLOYMENT_TARGET' not in os.environ:
current_system = LooseVersion(platform.mac_ver()[0])
python_target = LooseVersion(
get_config_var('MACOSX_DEPLOYMENT_TARGET'))
if python_target < '10.9' and current_system >= '10.9':
os.environ['MACOSX_DEPLOYMENT_TARGET'] = '10.9'
try:
from Cython.Build import cythonize as _cythonize
USE_CYTHON = True
except (ImportError, ModuleNotFoundError):
USE_CYTHON = False
def cythonize(extensions, **_ignore):
# Attempt to use Cython
if USE_CYTHON:
return _cythonize(extensions)
# Cython is not available
for extension in extensions:
sources = []
for sfile in extension.sources:
path, ext = os.path.splitext(sfile)
if ext in ('.pyx', '.py'):
if extension.language == 'c++':
ext = '.cpp'
else:
ext = '.c'
sfile = path + ext
sources.append(sfile)
extension.sources[:] = sources
return extensions
EXTENSIONS = [
Extension('pyemd.emd',
sources=['pyemd/emd.pyx'],
language="c++")
]
EXT_MODULES = cythonize(EXTENSIONS)
class sdist(_sdist):
def run(self):
# Ensure the compiled Cython files in the distribution are up-to-date
if USE_CYTHON:
_cythonize(EXTENSIONS)
else:
warn('\n\n\033[91m\033[1m WARNING: '
'IF YOU A PREPARING A DISTRIBUTION: Cython is not available! '
'The cythonized `*.cpp` files may be out of date. Please '
'install Cython and run `sdist` again.'
'\033[0m\n')
_sdist.run(self)
# See https://stackoverflow.com/a/21621689/1085344
class build_ext(_build_ext):
def finalize_options(self):
_build_ext.finalize_options(self)
# Prevent numpy from thinking it is still in its setup process:
if hasattr(__builtins__, '__NUMPY_SETUP__'):
__builtins__.__NUMPY_SETUP__ = False
import numpy
self.include_dirs.append(numpy.get_include())
CMDCLASS = {
'sdist': sdist,
'build_ext': build_ext
}
with io.open('README.rst', encoding='utf-8') as f:
README = f.read()
ABOUT = {}
with open('./pyemd/__about__.py') as f:
exec(f.read(), ABOUT)
REQUIRES = [
"numpy >=1.9.0, <1.20.0; python_version<='3.6'",
"numpy >=1.9.0, <2.0.0; python_version>'3.6'"
]
setup(
name=ABOUT['__title__'],
version=ABOUT['__version__'],
description=ABOUT['__description__'],
long_description=README,
author=ABOUT['__author__'],
author_email=ABOUT['__author_email__'],
url=ABOUT['__url__'],
license=ABOUT['__license__'],
packages=['pyemd'],
install_requires=REQUIRES,
cmdclass=CMDCLASS,
setup_requires=REQUIRES,
ext_modules=EXT_MODULES,
classifiers=[
'Development Status :: 3 - Alpha',
'Intended Audience :: Developers',
'Natural Language :: English',
'License :: OSI Approved :: MIT License',
'Programming Language :: Python :: 2',
'Programming Language :: Python :: 3',
],
)
|
