xref: /petsc/setup.py (revision 5648181bc3a0a9c8f81b1a42bbc7583c40bf7723) ! 
1#!/usr/bin/env python
2
3"""
4PETSc for Python
5================
6
7Python bindings for PETSc libraries.
8"""
9
10
11## try:
12##     import setuptools
13## except ImportError:
14##     pass
15
16
17# -----------------------------------------------------------------------------
18# Metadata
19# -----------------------------------------------------------------------------
20
21from conf.metadata import metadata
22
23def version():
24    import os, re
25    data = open(os.path.join('src', '__init__.py')).read()
26    m = re.search(r"__version__\s*=\s*'(.*)'", data)
27    return m.groups()[0]
28
29name     = 'petsc4py'
30version  = version()
31
32url      = 'http://%(name)s.googlecode.com/' % vars()
33download = url + 'files/%(name)s-%(version)s.tar.gz' % vars()
34
35descr    = __doc__.strip().split('\n'); del descr[1:3]
36devstat  = ['Development Status :: 3 - Alpha']
37keywords = ['PETSc', 'MPI']
38
39metadata['name'] = name
40metadata['version'] = version
41metadata['description'] = descr.pop(0)
42metadata['long_description'] = '\n'.join(descr)
43metadata['keywords'] += keywords
44metadata['classifiers'] += devstat
45metadata['url'] = url
46metadata['download_url'] = download
47
48metadata['provides'] = ['petsc4py']
49metadata['requires'] = ['numpy']
50
51# -----------------------------------------------------------------------------
52# Extension modules
53# -----------------------------------------------------------------------------
54
55def get_ext_modules(Extension):
56    from os   import walk, path
57    from glob import glob
58    depends = []
59    for pth, dirs, files in walk('src'):
60        depends += glob(path.join(pth, '*.h'))
61    for pth, dirs, files in walk(path.join('src', 'source')):
62        depends += glob(path.join(pth, '*.h'))
63        depends += glob(path.join(pth, '*.c'))
64    try:
65        import numpy
66        numpy_includes = [numpy.get_include()]
67    except ImportError:
68        numpy_includes = []
69    return [Extension('petsc4py.lib.PETSc',
70                      sources=['src/PETSc.c',
71                               'src/source/libpetsc4py.c',
72                               ],
73                      include_dirs=['src/include',
74                                    'src/source',
75                                    ] + numpy_includes,
76                      depends=depends)]
77
78# -----------------------------------------------------------------------------
79# Setup
80# -----------------------------------------------------------------------------
81
82from conf.petscconf import setup, Extension
83from conf.petscconf import config, build, build_ext
84
85def run_setup():
86    import sys, os
87    if (('distribute' in sys.modules) or
88        ('setuptools' in sys.modules)):
89        metadata['install_requires'] = ['numpy']
90        if not os.environ.get('PETSC_DIR'):
91            metadata['install_requires'].append('petsc')
92    if 'setuptools' in sys.modules:
93        metadata['zip_safe'] = False
94    setup(packages     = ['petsc4py',
95                          'petsc4py.lib',],
96          package_dir  = {'petsc4py'     : 'src',
97                          'petsc4py.lib' : 'src/lib'},
98          package_data = {'petsc4py'     : ['include/petsc4py/*.h',
99                                            'include/petsc4py/*.i',
100                                            'include/petsc4py/*.pxd',
101                                            'include/petsc4py/*.pxi',
102                                            'include/petsc4py/*.pyx',],
103                          'petsc4py.lib' : ['petsc.cfg'],},
104          ext_modules  = get_ext_modules(Extension),
105          cmdclass     = {'config'     : config,
106                          'build'      : build,
107                          'build_ext'  : build_ext},
108          **metadata)
109
110def run_cython(source):
111    import sys, os
112    CYTHON_VERSION_REQUIRED = .13
113    source_c = os.path.splitext(source)[0] + '.c'
114    if (os.path.exists(source_c)):
115        return False
116    try:
117        import Cython
118    except ImportError:
119        warn = lambda msg='': sys.stderr.write(msg+'\n')
120        warn("*"*80)
121        warn()
122        warn(" You need to generate C source files with Cython!!")
123        warn(" Download and install Cython <http://www.cython.org>")
124        warn()
125        warn("*"*80)
126        raise SystemExit
127    import Cython.Compiler.Version
128    if float(Cython.Compiler.Version.version) < CYTHON_VERSION_REQUIRED:
129        warn = lambda msg='': sys.stderr.write(msg+'\n')
130        warn("*"*80)
131        warn()
132        warn(" You need to install Cython version "+str(CYTHON_VERSION_REQUIRED)+" <http://www.cython.org>")
133        warn()
134        warn("*"*80)
135        raise SystemExit
136    from distutils import log
137    from conf.cythonize import run as cythonize
138    log.info("cythonizing '%s' source" % source)
139    cythonize(source)
140    return True
141
142def main():
143    import os
144    try:
145        run_setup()
146    except:
147        done = run_cython(os.path.join('src', 'petsc4py.PETSc.pyx'))
148        if done:
149            run_setup()
150        else:
151            raise
152
153if __name__ == '__main__':
154    main()
155
156# -----------------------------------------------------------------------------
157