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
|
#!/usr/bin/env python
from __future__ import nested_scopes
import os
import sys
import glob
def configuration(parent_package='',parent_path=None):
from scipy_distutils.core import Extension
from scipy_distutils.misc_util import get_path, default_config_dict, dot_join
from scipy_distutils.system_info import get_info,dict_append,NotFoundError
package = 'sparse'
config = default_config_dict(package,parent_package)
local_path = get_path(__name__,parent_path)
def local_join(*paths):
return os.path.join(*((local_path,)+paths))
def local_glob(*names):
return glob.glob(os.path.join(*((local_path,)+names)))
lapack_opt = get_info('lapack_opt')
if not lapack_opt:
raise NotFoundError,'no lapack/blas resources found'
if sys.platform=='win32':
superlu_defs = [('NO_TIMER',1)]
else:
superlu_defs = []
superlu_defs.append(('USE_VENDOR_BLAS',1))
superlu = ('superlu_src',{'sources':local_glob('SuperLU','SRC','*.c'),'macros':superlu_defs})
sparsekit = ('sparsekit_src',{'sources':local_glob('sparsekit','*.f')})
#SuperLU/SRC/util.h has been modifed to use these by default
#macs = [('USER_ABORT','superlu_python_module_abort'),
# ('USER_MALLOC','superlu_python_module_malloc'),
# ('USER_FREE','superlu_python_module_free')]
# Extension
sources = ['_zsuperlumodule.c','_superlu_utils.c','_superluobject.c']
ext_args = {'name':dot_join(parent_package,package,'_zsuperlu'),
'sources':map(local_join,sources),
'libraries': [superlu],
}
dict_append(ext_args,**lapack_opt)
ext = Extension(**ext_args)
config['ext_modules'].append(ext)
sources = ['_dsuperlumodule.c','_superlu_utils.c','_superluobject.c']
ext_args = {'name':dot_join(parent_package,package,'_dsuperlu'),
'sources':map(local_join,sources),
'libraries': [superlu],
}
dict_append(ext_args,**lapack_opt)
ext = Extension(**ext_args)
config['ext_modules'].append(ext)
sources = ['_csuperlumodule.c','_superlu_utils.c','_superluobject.c']
ext_args = {'name':dot_join(parent_package,package,'_csuperlu'),
'sources':map(local_join,sources),
'libraries': [superlu],
}
dict_append(ext_args,**lapack_opt)
ext = Extension(**ext_args)
config['ext_modules'].append(ext)
sources = ['_ssuperlumodule.c','_superlu_utils.c','_superluobject.c']
ext_args = {'name':dot_join(parent_package,package,'_ssuperlu'),
'sources':map(local_join,sources),
'libraries': [superlu],
}
dict_append(ext_args,**lapack_opt)
ext = Extension(**ext_args)
config['ext_modules'].append(ext)
# ext_args = {'name':dot_join(parent_package,package,'_sparsekit'),
# 'sources':[local_join('_sparsekit.pyf')],
# 'libraries' : [sparsekit]
# }
# dict_append(ext_args,**lapack_opt)
# ext = Extension(**ext_args)
# config['ext_modules'].append(ext)
sources = ['spblas.f.src','spconv.f.src','sparsetools.pyf.src']
sources = [local_join('sparsetools',x) for x in sources]
ext = Extension(dot_join(parent_package, package, 'sparsetools'), sources=sources)
config['ext_modules'].append(ext)
return config
if __name__ == '__main__':
from scipy_distutils.core import setup
setup(**configuration(parent_path=''))
|
