forked from wmayner/pyemd
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
107 lines (87 loc) · 2.75 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
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
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import os
from distutils.command.sdist import sdist as _sdist
from setuptools import Extension, setup
from setuptools.command.build_ext import build_ext as _build_ext
try:
from Cython.Build import cythonize
USE_CYTHON = True
except ImportError:
USE_CYTHON = False
EXTENSIONS = [
Extension('pyemd.emd',
sources=['pyemd/emd.pyx'],
language="c++")
]
class sdist(_sdist):
def run(self):
# Make sure the compiled Cython files in the distribution are
# up-to-date
from Cython.Build import cythonize
cythonize(EXTENSIONS)
_sdist.run(self)
# See http://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
}
def no_cythonize(extensions, **_ignore):
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
if USE_CYTHON:
ext_modules = cythonize(EXTENSIONS)
else:
ext_modules = no_cythonize(EXTENSIONS)
with open('README.rst', 'r') as f:
readme = f.read()
about = {}
with open('./pyemd/__about__.py') as f:
exec(f.read(), about)
requires = ['numpy >=1.9.0, <2.0.0']
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',
'Programming Language :: Python :: 3',
'Programming Language :: Python :: 3.3',
'Programming Language :: Python :: 3.4',
'Programming Language :: Python :: 3.5',
'Programming Language :: Python :: 3.6'
],
)