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
|
#!/usr/bin/env python3
#$ python setup.py build_ext --inplace
# a bit of monkeypatching ...
try:
from numpy.distutils.fcompiler import FCompiler
from numpy.distutils.unixccompiler import UnixCCompiler
try: # Python 2
meth = UnixCCompiler.runtime_library_dir_option.im_func
except AttributeError: # Python 3
meth = UnixCCompiler.runtime_library_dir_option
FCompiler.runtime_library_dir_option = meth
except Exception:
pass
def configuration(parent_package='',top_path=None):
INCLUDE_DIRS = []
LIBRARY_DIRS = []
LIBRARIES = []
# PETSc
import os
PETSC_DIR = os.environ['PETSC_DIR']
PETSC_ARCH = os.environ.get('PETSC_ARCH', '')
PETSC_LIB_EXT = os.environ.get('PETSC_LIB_EXT', '')
from os.path import join, isdir
if not PETSC_LIB_EXT:
from configparser import ConfigParser
parser = ConfigParser()
petscvariables = join(PETSC_DIR, PETSC_ARCH, 'lib/petsc/conf/petscvariables')
with open(petscvariables) as stream: # waiting for https://github.com/python/cpython/pull/2735 to avoid this step
parser.read_string("[petscvariables]\n" + stream.read())
PETSC_LIB_EXT = parser['petscvariables'].get('PETSC_LIB_EXT','')
if PETSC_ARCH and isdir(join(PETSC_DIR, PETSC_ARCH)):
INCLUDE_DIRS += [join(PETSC_DIR, PETSC_ARCH, 'include'),
join(PETSC_DIR, 'include')]
LIBRARY_DIRS += [join(PETSC_DIR, PETSC_ARCH, 'lib')]
else:
if PETSC_ARCH: pass # XXX should warn ...
INCLUDE_DIRS += [join(PETSC_DIR, 'include')]
LIBRARY_DIRS += [join(PETSC_DIR, 'lib')]
LIBRARIES += [#'petscts', 'petscsnes', 'petscksp',
#'petscdm', 'petscmat', 'petscvec',
'petsc']
if PETSC_LIB_EXT:
LIBRARIES = [lib+PETSC_LIB_EXT for lib in LIBRARIES]
# PETSc for Python
import petsc4py
INCLUDE_DIRS += [petsc4py.get_include()]
# Configuration
from numpy.distutils.misc_util import Configuration
config = Configuration('', parent_package, top_path)
config.add_extension('_Bratu3D',
sources = ['Bratu3D.i',
'Bratu3D.c'],
depends = ['Bratu3D.h'],
include_dirs=INCLUDE_DIRS + [os.curdir],
libraries=LIBRARIES,
library_dirs=LIBRARY_DIRS,
runtime_library_dirs=LIBRARY_DIRS)
return config
if __name__ == "__main__":
from numpy.distutils.core import setup
setup(**configuration(top_path='').todict())
|
