-
Notifications
You must be signed in to change notification settings - Fork 224
/
setup.py
130 lines (119 loc) · 5.29 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
127
128
129
130
try:
from setuptools import setup, Extension
except ImportError:
# Legacy distutils import. No longer available on Pyton > 3.12
from distutils.core import setup, Extension
from codecs import open
import os
import sys
import sysconfig
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 = "-DGITHASH="+ghash.strip()
except:
ghash_arg = "-DGITHASH=be4654593929232c50cce49e27913af43a9db2b8" #GITHASHAUTOUPDATE
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=['-Wl,-install_name,@rpath/librebound'+suffix]
if sys.platform == 'win32':
extra_compile_args=[ghash_arg, '-DLIBREBOUND', '-D_GNU_SOURCE', '-DSERVER']
else:
# Default compile args
extra_compile_args=['-fstrict-aliasing', '-std=c99','-Wno-unknown-pragmas', ghash_arg, '-DLIBREBOUND', '-D_GNU_SOURCE', '-DSERVER', '-fPIC']
# For coverage runs, turn off optimizations and turn on coverage generation
COVERAGE = os.environ.get("COVERAGE", None)
if COVERAGE:
extra_compile_args += ['-O1', '-fprofile-arcs', '-ftest-coverage' ,'-coverage']
extra_link_args += ['-fprofile-arcs', '-ftest-coverage' ,'-coverage']
else:
extra_compile_args += ['-O3']
# 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')
# Option to enable AVX512 enabled integrators (WHFast512).
AVX512 = os.environ.get("AVX512", None)
if AVX512:
extra_compile_args.append('-march=native')
extra_compile_args.append('-DAVX512')
libreboundmodule = Extension('librebound',
sources = [ 'src/rebound.c',
'src/integrator_ias15.c',
'src/integrator_whfast.c',
'src/integrator_whfast512.c',
'src/integrator_saba.c',
'src/integrator_mercurius.c',
'src/integrator_trace.c',
'src/integrator_eos.c',
'src/integrator_leapfrog.c',
'src/integrator_bs.c',
'src/integrator_janus.c',
'src/integrator_sei.c',
'src/integrator.c',
'src/gravity.c',
'src/server.c',
'src/boundary.c',
'src/display.c',
'src/collision.c',
'src/tools.c',
'src/fmemopen.c',
'src/rotations.c',
'src/derivatives.c',
'src/tree.c',
'src/particle.c',
'src/binarydiff.c',
'src/output.c',
'src/input.c',
'src/simulationarchive.c',
'src/transformations.c',
],
include_dirs = ['src'],
define_macros=[ ('LIBREBOUND', None) ],
extra_link_args=extra_link_args,
extra_compile_args=extra_compile_args,
)
here = os.path.abspath(os.path.dirname(__file__))
with open(os.path.join(here, 'README.md'), encoding='utf-8') as f:
long_description = f.read()
setup(name='rebound',
version='4.4.6',
description='An open-source multi-purpose N-body code',
long_description=long_description,
long_description_content_type="text/markdown",
url='https://github.com/hannorein/rebound/',
author='Hanno Rein',
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 symplectic wisdom-holman',
packages=['rebound', 'rebound.integrators'],
package_data = {'rebound':['rebound.h']},
install_requires=[],
tests_require=["numpy","matplotlib"],
test_suite="rebound.tests",
ext_modules = [libreboundmodule],
zip_safe=False)