-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.py
70 lines (60 loc) · 2.48 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
# distutils is deprecated; to be removed for Python 3.12.
# See https://numpy.org/devdocs/reference/distutils_status_migration.html for
# migration advice.
# This setup.py file no longer uses numpy.distutils so it might be easy to
# move fully to setuptools.
import os
import setuptools # used indirectly for bdist_wheel cmd and long_description_content_type
from distutils.core import setup
from distutils.extension import Extension
import numpy as np
_ = setuptools
with open("README.md", 'r', encoding="utf-8") as readme:
long_description = readme.read()
current_dir = os.getcwd()
arrow_dir = os.path.join(current_dir, 'stochastic_arrow')
# Compile the Cython code to C for development builds:
# USE_CYTHON=1 python setup.py build_ext --inplace
# and for building source distribution packages:
# USE_CYTHON=1 python setup.py sdist
# and *not* when installing a distribution package.
# See http://docs.cython.org/en/latest/src/userguide/source_files_and_compilation.html#distributing-cython-modules
USE_CYTHON = 'USE_CYTHON' in os.environ
ext = '.pyx' if USE_CYTHON else '.c'
cython_extensions = [
Extension('stochastic_arrow.arrowhead',
sources=[
'stochastic_arrow/mersenne.c',
'stochastic_arrow/obsidian.c',
'stochastic_arrow/arrowhead'+ext,],
include_dirs=['stochastic_arrow', np.get_include()],
define_macros=[('NPY_NO_DEPRECATED_API', 'NPY_1_7_API_VERSION')],
)]
if USE_CYTHON:
from Cython.Build import cythonize
cython_extensions = cythonize(
cython_extensions,
include_path=['stochastic_arrow'],
annotate=True, # to get an HTML code listing
)
setup(
name='stochastic-arrow',
version='1.0.0',
packages=['stochastic_arrow'],
author='Ryan Spangler, John Mason, Jerry Morrison, Chris Skalnik, Travis Ahn-Horst, Sean Cheah',
author_email='ryan.spangler@gmail.com',
url='https://github.com/CovertLab/arrow',
license='MIT',
include_dirs=[arrow_dir, np.get_include()],
ext_modules=cython_extensions,
long_description=long_description,
long_description_content_type='text/markdown',
requires=['numpy (>=1.14)', 'six'],
classifiers=[
'Development Status :: 3 - Alpha',
'License :: OSI Approved :: MIT License',
'Programming Language :: Python',
'Programming Language :: Python :: 2.7',
'Programming Language :: Python :: 3',
'Topic :: Scientific/Engineering',
])