forked from hannorein/rebound
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
123 lines (112 loc) · 4.95 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
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=a548a279c439c6ccdbb05e4ded8d5a284dad5df0" #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', '-O3','-std=c99','-Wno-unknown-pragmas', ghash_arg, '-DLIBREBOUND', '-D_GNU_SOURCE', '-DSERVER', '-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')
# 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.3',
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)