forked from amusecode/amuse
/
setup.py
202 lines (176 loc) · 6.91 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
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
from ez_setup import use_setuptools
use_setuptools()
from distutils.command.build import build
from distutils.command.clean import clean
from distutils.command.install import install
from distutils.util import convert_path
from distutils.cmd import Command
from distutils.extension import Extension
from setuptools import setup
import sys
import os
import fnmatch
import re
import glob
if sys.hexversion > 0x03000000:
from support3.generate_main import generate_main
from support3.build_latex import build_latex
from support3.setup_codes import BuildCodes, CleanCodes, DistCleanCodes, BuildOneCode, BuildLibraries
from support3.setup_codes import ConfigureCodes
from support3.setup_codes import GenerateInstallIni
from support3.run_tests import run_tests
from distutils.command.build_py import build_py_2to3
from os import walk as py_walk
def walk(top, callback, args):
for root, dirs, files in py_walk(top):
callback(args, root, files)
else:
from support.generate_main import generate_main
from support.build_latex import build_latex
from support.setup_codes import BuildCodes, CleanCodes, DistCleanCodes, BuildOneCode, BuildLibraries
from support.setup_codes import ConfigureCodes
from support.setup_codes import GenerateInstallIni
from support.run_tests import run_tests
from os.path import walk
#include_dirs.append(sysconfig.get_python_inc())
extensions = []
class Clean(clean):
def run(self):
for cmd_name in self.get_sub_commands():
self.run_command(cmd_name)
class Install(install):
def run(self):
for cmd_name in self.get_sub_commands():
self.run_command(cmd_name)
install.run(self)
mapping_from_command_name_to_command_class = {
'build_latex':build_latex,
'build_codes':BuildCodes,
'build_libraries':BuildLibraries,
'build_code':BuildOneCode,
'configure_codes':ConfigureCodes,
'clean_codes':CleanCodes,
'dist_clean':DistCleanCodes,
'clean_python':clean,
'clean': Clean,
'tests':run_tests,
'generate_main': generate_main,
'generate_install_ini': GenerateInstallIni,
'install':install
}
if sys.hexversion > 0x03000000:
mapping_from_command_name_to_command_class['build_py'] = build_py_2to3
build.sub_commands.append(('configure_codes', None))
build.sub_commands.append(('build_codes', None))
Clean.sub_commands.append(('clean_codes',None))
Clean.sub_commands.append(('clean_python',None))
Install.sub_commands.insert(0, ('generate_install_ini',None) )
def find_packages(where='.', exclude=()):
"""Return a list all Python packages found within directory 'where'
'where' should be supplied as a "cross-platform" (i.e. URL-style) path; it
will be converted to the appropriate local path syntax. 'exclude' is a
sequence of package names to exclude; '*' can be used as a wildcard in the
names, such that 'foo.*' will exclude all subpackages of 'foo' (but not
'foo' itself).
"""
out = []
stack=[(convert_path(where), '')]
while stack:
where,prefix = stack.pop(0)
for name in os.listdir(where):
fn = os.path.join(where,name)
if ('.' not in name and os.path.isdir(fn) and
os.path.isfile(os.path.join(fn,'__init__.py'))
):
out.append(prefix+name); stack.append((fn,prefix+name+'.'))
for pat in list(exclude)+['ez_setup', 'distribute_setup']:
from fnmatch import fnmatchcase
out = [item for item in out if not fnmatchcase(item,pat)]
return out
def find_data_files(srcdir, destdir, *wildcards, **kw):
"""
get a list of all files under the srcdir matching wildcards,
returned in a format to be used for install_data
"""
def walk_helper(arg, dirname, files):
if '.svn' in dirname:
return
names = []
lst, wildcards, dirnameconverter, destdir = arg
for wc in wildcards:
wc_name = os.path.normpath(os.path.join(dirname, wc))
for f in files:
filename = os.path.normpath(os.path.join(dirname, f))
if fnmatch.fnmatch(filename, wc_name) and not os.path.isdir(filename):
names.append(filename)
if names:
destdirname = dirnameconverter.sub(destdir, dirname)
lst.append( (destdirname, names ) )
file_list = []
recursive = kw.get('recursive', True)
converter = re.compile('^({0})'.format(srcdir))
if recursive:
walk(srcdir, walk_helper, (file_list, wildcards, converter, destdir))
else:
walk_helper((file_list, wildcards, converter, destdir),
srcdir,
[os.path.basename(f) for f in glob.glob(os.path.join(srcdir, '*'))])
return file_list
all_data_files = find_data_files('data', 'share/amuse/data', '*', recursive = True)
all_data_files.extend(find_data_files('support', 'share/amuse/support', '*', recursive = False))
all_data_files.extend(find_data_files('support3', 'share/amuse/support3', '*', recursive = False))
all_data_files.extend(find_data_files('lib', 'share/amuse/lib', '*.h', '*.a', '*.mod', '*.inc', '*.so', '*.dylib', recursive = True))
all_data_files.append(('share/amuse',['./config.mk','./build.py']))
packages = find_packages('src')
packages.extend(['amuse.test.suite.' + x for x in find_packages('test')])
packages.extend(['amuse.examples.' + x for x in find_packages('examples')])
package_data = {
'amuse.rfi.tools': ['*.template'],
'amuse.test.suite.core_tests': [
'*.txt', '*.dyn', '*.ini',
'*.nemo',
'*.dat', 'gadget_snapshot'
],
'amuse.test.suite.codes_tests': [
'*.txt', 'test_sphray_data*'
],
'amuse.test.suite.ticket_tests': [
'*.out'
],
'amuse': [
'*rc'
]
}
setup(
name = 'amuse',
version = "11",
cmdclass = mapping_from_command_name_to_command_class,
ext_modules = extensions,
package_dir = {'': 'src', 'amuse.test.suite' :'test', 'amuse.examples' : 'examples'},
packages = packages,
package_data = package_data,
data_files = all_data_files,
classifiers = [
'Development Status :: 4 - Beta',
'Environment :: Console',
'Intended Audience :: End Users/Desktop',
'Intended Audience :: Developers',
'Intended Audience :: Science/Research',
'License :: GNU General Public License (GPL)',
'Operating System :: MacOS :: MacOS X',
'Operating System :: POSIX',
'Programming Language :: Python :: 2.6',
'Programming Language :: C',
'Programming Language :: C++',
'Programming Language :: Fortran',
'Topic :: Scientific/Engineering :: Astronomy',
],
url = 'http://www.amusecode.org/',
author_email = 'info@amusecode.org',
author = 'The Amuse Team',
install_requires = [
'docutils>=0.6',
'numpy>=1.2.2',
'nose>=0.11.1'
]
)