-
Notifications
You must be signed in to change notification settings - Fork 59
/
setup.py
126 lines (111 loc) · 6.41 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
117
118
119
120
121
122
123
124
125
126
from codecs import open
import os
import inspect
import sys
import sysconfig
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.strip()
except:
ghash_arg = "-DREBXGITHASH=5a8389b9c1181ddaea318ade8f0b3f7bebeab699" #GITHASHAUTOUPDATE
class build_ext(_build_ext):
def finalize_options(self):
_build_ext.finalize_options(self)
if "PYODIDE" in os.environ:
return None
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 (https://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 https://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 = sysconfig.get_path('platlib')+'/'
print("***", rebdir, "***", rebdirsp, "***")
self.include_dirs.append(rebdir)
sources = [ 'src/modify_mass.c', 'src/integrator_euler.c', 'src/modify_orbits_forces.c', 'src/lense_thirring.c', 'src/integrator_rk2.c', 'src/track_min_distance.c', 'src/tides_spin.c', 'src/gas_dynamical_friction.c', 'src/rebxtools.c', 'src/inner_disk_edge.c', 'src/gravitational_harmonics.c', 'src/gr_potential.c', 'src/core.c', 'src/integrator_rk4.c', 'src/input.c', 'src/central_force.c', 'src/stochastic_forces.c', 'src/gr.c', 'src/modify_orbits_direct.c', 'src/tides_constant_time_lag.c', 'src/yarkovsky_effect.c', 'src/gr_full.c', 'src/steppers.c', 'src/integrate_force.c', 'src/interpolation.c', 'src/type_I_migration.c', 'src/output.c', 'src/radiation_forces.c', 'src/integrator_implicit_midpoint.c', 'src/exponential_migration.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)
extra_link_args=[]
if sys.platform == 'darwin':
config_vars = sysconfig.get_config_vars()
config_vars['LDSHARED'] = config_vars['LDSHARED'].replace('-bundle', '-shared')
extra_link_args.append('-Wl,-install_name,@rpath/libreboundx'+suffix)
if sys.platform == 'win32':
extra_compile_args=[ghash_arg, '-DLIBREBOUNDX', '-D_GNU_SOURCE']
else:
# Default compile args
extra_compile_args=['-fstrict-aliasing', '-O3','-std=c99','-Wno-unknown-pragmas', ghash_arg, '-DLIBREBOUNDX', '-D_GNU_SOURCE', '-fPIC']
# Option to disable FMA in CLANG.
FFP_CONTRACT_OFF = os.environ.get("FFP_CONTRACT_OFF", None)
if FFP_CONTRACT_OFF:
extra_compile_args.append('-ffp-contract=off')
libreboundxmodule = Extension('libreboundx',
sources = [ 'src/modify_mass.c', 'src/integrator_euler.c', 'src/modify_orbits_forces.c', 'src/lense_thirring.c', 'src/integrator_rk2.c', 'src/track_min_distance.c', 'src/tides_spin.c', 'src/gas_dynamical_friction.c', 'src/rebxtools.c', 'src/inner_disk_edge.c', 'src/gravitational_harmonics.c', 'src/gr_potential.c', 'src/core.c', 'src/integrator_rk4.c', 'src/input.c', 'src/central_force.c', 'src/stochastic_forces.c', 'src/gr.c', 'src/modify_orbits_direct.c', 'src/tides_constant_time_lag.c', 'src/yarkovsky_effect.c', 'src/gr_full.c', 'src/steppers.c', 'src/integrate_force.c', 'src/interpolation.c', 'src/type_I_migration.c', 'src/output.c', 'src/radiation_forces.c', 'src/integrator_implicit_midpoint.c', 'src/exponential_migration.c', 'src/linkedlist.c'],
include_dirs = ['src'],
library_dirs = [],
runtime_library_dirs = ["."],
libraries=['rebound'+suffix[:suffix.rfind('.')]],
define_macros=[ ('LIBREBOUNDX', None) ],
extra_compile_args=extra_compile_args,
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='4.2.1',
description='A library for including additional forces in REBOUND',
long_description=long_description,
url='https://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>=4.0.0'],
install_requires=['rebound>=4.0.0'],
tests_require=['rebound>=4.0.0','numpy'],
test_suite="reboundx.test",
ext_modules = [libreboundxmodule],
zip_safe=False)