forked from dtamayo/reboundx
-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathsetup.py
116 lines (102 loc) · 5.7 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
108
109
110
111
112
113
114
115
116
from codecs import open
import os
import inspect
import sys
from distutils import sysconfig
from distutils.sysconfig import get_python_lib
try:
from setuptools import setup, Extension
from setuptools.command.build_ext import build_ext as _build_ext
except ImportError:
print("Installing REBOUNDx requires setuptools. Do 'pip install setuptools'.")
sys.exit(1)
suffix = sysconfig.get_config_var('EXT_SUFFIX')
if suffix is None:
suffix = ".so"
# Try to get git hash
try:
import subprocess
ghash = subprocess.check_output(["git", "rev-parse", "HEAD"]).decode("ascii")
ghash_arg = "-DREBXGITHASH="+ghash
except:
ghash_arg = "-DREBXGITHASH=f6f6c159ec82f3298938f5b2899db2896afaaf1a" #GITHASHAUTOUPDATE
class build_ext(_build_ext):
def finalize_options(self):
_build_ext.finalize_options(self)
try:
import rebound
except ImportError:
print("REBOUNDx did not automatically install REBOUND. Please let me know if this happens ([email protected]), and try first installing REBOUND (http://rebound.readthedocs.org/en/latest/python_quickstart.html")
sys.exit(1)
try:
version = rebound.__version__ # Added in 2.12.1
except AttributeError:
print("REBOUNDx did not automatically install a recent enough version of REBOUND. Please let me know if this happens ([email protected]), and try upgrading REBOUND. See 5.3 in http://rebound.readthedocs.org/en/latest/python_quickstart.html")
sys.exit(1)
rebdir = os.path.dirname(inspect.getfile(rebound))
# get site-packages dir to add to paths in case reb & rebx installed simul in tmp dir
rebdirsp = get_python_lib()+'/'#[p for p in sys.path if p.endswith('site-packages')][0]+'/'
self.include_dirs.append(rebdir)
sources = [ 'src/modify_mass.c', 'src/integrator_euler.c', 'src/modify_orbits_forces.c', 'src/integrator_rk2.c', 'src/track_min_distance.c', 'src/tides_precession.c', 'src/rebxtools.c', 'src/ephemeris_forces.c', 'src/gravitational_harmonics.c', 'src/gr_potential.c', 'src/core.c', 'src/integrator_rk4.c', 'src/input.c', 'src/central_force.c', 'src/gr.c', 'src/modify_orbits_direct.c', 'src/gr_full.c', 'src/steppers.c', 'src/integrate_force.c', 'src/output.c', 'src/radiation_forces.c', 'src/integrator_implicit_midpoint.c', 'src/linkedlist.c'],
self.library_dirs.append(rebdir+'/../')
self.library_dirs.append(rebdirsp)
for ext in self.extensions:
ext.runtime_library_dirs.append(rebdir+'/../')
ext.extra_link_args.append('-Wl,-rpath,'+rebdir+'/../')
ext.runtime_library_dirs.append(rebdirsp)
ext.extra_link_args.append('-Wl,-rpath,'+rebdirsp)
from distutils.version import LooseVersion
extra_link_args=[]
if sys.platform == 'darwin':
from distutils import sysconfig
vars = sysconfig.get_config_vars()
vars['LDSHARED'] = vars['LDSHARED'].replace('-bundle', '-shared')
extra_link_args.append('-Wl,-install_name,@rpath/libreboundx'+suffix)
libreboundxmodule = Extension('libreboundx',
sources = [ 'src/modify_mass.c', 'src/integrator_euler.c', 'src/modify_orbits_forces.c', 'src/integrator_rk2.c', 'src/track_min_distance.c', 'src/tides_precession.c', 'src/rebxtools.c', 'src/ephemeris_forces.c', 'src/gravitational_harmonics.c', 'src/gr_potential.c', 'src/core.c', 'src/integrator_rk4.c', 'src/input.c', 'src/central_force.c', 'src/gr.c', 'src/modify_orbits_direct.c', 'src/gr_full.c', 'src/steppers.c', 'src/integrate_force.c', 'src/output.c', 'src/radiation_forces.c', 'src/integrator_implicit_midpoint.c', 'src/linkedlist.c'],
include_dirs = ['src'],
library_dirs = [],
runtime_library_dirs = ["."],
libraries=['rebound'+suffix[:-3]], #take off .so from the suffix
define_macros=[ ('LIBREBOUNDX', None) ],
extra_compile_args=['-fstrict-aliasing', '-O3','-std=c99', '-fPIC', '-Wpointer-arith', ghash_arg],
extra_link_args=extra_link_args,
)
here = os.path.abspath(os.path.dirname(__file__))
with open(os.path.join(here, 'README.rst'), encoding='utf-8') as f:
long_description = f.read()
setup(name='reboundx',
version='3.0.4',
description='A library for including additional forces in REBOUND',
long_description=long_description,
url='http://github.com/dtamayo/reboundx',
author='Daniel Tamayo',
author_email='[email protected]',
license='GPL',
classifiers=[
# How mature is this project? Common values are
# 3 - Alpha
# 4 - Beta
# 5 - Production/Stable
'Development Status :: 5 - Production/Stable',
# Indicate who your project is intended for
'Intended Audience :: Science/Research',
'Intended Audience :: Developers',
'Topic :: Software Development :: Build Tools',
'Topic :: Scientific/Engineering :: Astronomy',
# Pick your license as you wish (should match "license" above)
'License :: OSI Approved :: GNU General Public License v3 or later (GPLv3+)',
# Specify the Python versions you support here. In particular, ensure
# that you indicate whether you support Python 2, Python 3 or both.
'Programming Language :: Python :: 2',
'Programming Language :: Python :: 3',
],
keywords='astronomy astrophysics nbody integrator',
packages=['reboundx'],
cmdclass={'build_ext':build_ext},
setup_requires=['rebound>=3.10.0', 'numpy'],
install_requires=['rebound>=3.10.0', 'numpy'],
tests_require=["numpy","matplotlib"],
test_suite="reboundx.test",
ext_modules = [libreboundxmodule],
zip_safe=False)