setup.py
4.34 KB
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
import os
from os.path import join
from scipy._build_utils import numpy_nodepr_api
def configuration(parent_package='',top_path=None):
from numpy.distutils.misc_util import Configuration
from scipy._build_utils.system_info import get_info
from scipy._build_utils import (uses_blas64, blas_ilp64_pre_build_hook,
combine_dict, get_f2py_int64_options)
config = Configuration('integrate', parent_package, top_path)
if uses_blas64():
lapack_opt = get_info('lapack_ilp64_opt', 2)
pre_build_hook = blas_ilp64_pre_build_hook(lapack_opt)
f2py_options = get_f2py_int64_options()
else:
lapack_opt = get_info('lapack_opt')
pre_build_hook = None
f2py_options = None
mach_src = [join('mach','*.f')]
quadpack_src = [join('quadpack', '*.f')]
lsoda_src = [join('odepack', fn) for fn in [
'blkdta000.f', 'bnorm.f', 'cfode.f',
'ewset.f', 'fnorm.f', 'intdy.f',
'lsoda.f', 'prja.f', 'solsy.f', 'srcma.f',
'stoda.f', 'vmnorm.f', 'xerrwv.f', 'xsetf.f',
'xsetun.f']]
vode_src = [join('odepack', 'vode.f'), join('odepack', 'zvode.f')]
dop_src = [join('dop','*.f')]
quadpack_test_src = [join('tests','_test_multivariate.c')]
odeint_banded_test_src = [join('tests', 'banded5x5.f')]
config.add_library('mach', sources=mach_src, config_fc={'noopt': (__file__, 1)},
_pre_build_hook=pre_build_hook)
config.add_library('quadpack', sources=quadpack_src, _pre_build_hook=pre_build_hook)
config.add_library('lsoda', sources=lsoda_src, _pre_build_hook=pre_build_hook)
config.add_library('vode', sources=vode_src, _pre_build_hook=pre_build_hook)
config.add_library('dop', sources=dop_src, _pre_build_hook=pre_build_hook)
# Extensions
# quadpack:
include_dirs = [join(os.path.dirname(__file__), '..', '_lib', 'src')]
cfg = combine_dict(lapack_opt,
include_dirs=include_dirs,
libraries=['quadpack', 'mach'])
config.add_extension('_quadpack',
sources=['_quadpackmodule.c'],
depends=(['__quadpack.h']
+ quadpack_src + mach_src),
**cfg)
# odepack/lsoda-odeint
cfg = combine_dict(lapack_opt, numpy_nodepr_api,
libraries=['lsoda', 'mach'])
config.add_extension('_odepack',
sources=['_odepackmodule.c'],
depends=(lsoda_src + mach_src),
**cfg)
# vode
cfg = combine_dict(lapack_opt,
libraries=['vode'])
ext = config.add_extension('vode',
sources=['vode.pyf'],
depends=vode_src,
f2py_options=f2py_options,
**cfg)
ext._pre_build_hook = pre_build_hook
# lsoda
cfg = combine_dict(lapack_opt,
libraries=['lsoda', 'mach'])
ext = config.add_extension('lsoda',
sources=['lsoda.pyf'],
depends=(lsoda_src + mach_src),
f2py_options=f2py_options,
**cfg)
ext._pre_build_hook = pre_build_hook
# dop
ext = config.add_extension('_dop',
sources=['dop.pyf'],
libraries=['dop'],
depends=dop_src,
f2py_options=f2py_options)
ext._pre_build_hook = pre_build_hook
config.add_extension('_test_multivariate',
sources=quadpack_test_src)
# Fortran+f2py extension module for testing odeint.
cfg = combine_dict(lapack_opt,
libraries=['lsoda', 'mach'])
ext = config.add_extension('_test_odeint_banded',
sources=odeint_banded_test_src,
depends=(lsoda_src + mach_src),
f2py_options=f2py_options,
**cfg)
ext._pre_build_hook = pre_build_hook
config.add_subpackage('_ivp')
config.add_data_dir('tests')
return config
if __name__ == '__main__':
from numpy.distutils.core import setup
setup(**configuration(top_path='').todict())