1 | from __future__ import division, print_function |
---|
2 | |
---|
3 | import os |
---|
4 | import sys |
---|
5 | |
---|
6 | from os.path import join |
---|
7 | |
---|
8 | def configuration(parent_package='',top_path=None): |
---|
9 | |
---|
10 | from numpy.distutils.misc_util import Configuration |
---|
11 | from numpy.distutils.system_info import get_info |
---|
12 | |
---|
13 | config = Configuration('pymetis', parent_package, top_path) |
---|
14 | |
---|
15 | config.add_data_dir('tests') |
---|
16 | |
---|
17 | if parent_package is '': |
---|
18 | anuga_dir = '..' |
---|
19 | else: |
---|
20 | anuga_dir = '.' |
---|
21 | |
---|
22 | METIS_DIR = 'metis-4.0' |
---|
23 | |
---|
24 | metis_src = [join(METIS_DIR,'*.c')] |
---|
25 | metis_headers = [join(METIS_DIR,'*.h')] |
---|
26 | include_dir = METIS_DIR |
---|
27 | |
---|
28 | #print(metis_headers) |
---|
29 | #print(metis_src) |
---|
30 | #print(include_dir) |
---|
31 | |
---|
32 | config.add_include_dirs([include_dir]) |
---|
33 | |
---|
34 | config.add_library('metis', sources=metis_src) |
---|
35 | |
---|
36 | |
---|
37 | src_files = ['metis_ext.c', 'metis_bridge.c'] |
---|
38 | if sys.platform == 'win32': |
---|
39 | src_files = src_files + ['random.c'] |
---|
40 | |
---|
41 | |
---|
42 | |
---|
43 | config.add_extension('metis_ext', |
---|
44 | sources=src_files, |
---|
45 | include_dirs = [include_dir], |
---|
46 | depends=(metis_src + metis_headers), |
---|
47 | extra_compile_args=['-I'+include_dir], |
---|
48 | libraries = ['metis', 'm']) |
---|
49 | |
---|
50 | return config |
---|
51 | |
---|
52 | if __name__ == '__main__': |
---|
53 | from numpy.distutils.core import setup |
---|
54 | setup(configuration=configuration) |
---|