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 ('setuptools' in sys.modules and 88 not os.environ.get('PETSC_DIR')): 89 metadata['requires'].append('petsc') 90 if 'setuptools' in sys.modules: 91 metadata['zip_safe'] = False 92 setup(packages = ['petsc4py', 93 'petsc4py.lib',], 94 package_dir = {'petsc4py' : 'src', 95 'petsc4py.lib' : 'src/lib'}, 96 package_data = {'petsc4py' : ['include/petsc4py/*.h', 97 'include/petsc4py/*.i', 98 'include/petsc4py/*.pxd', 99 'include/petsc4py/*.pxi', 100 'include/petsc4py/*.pyx',], 101 'petsc4py.lib' : ['petsc.cfg'],}, 102 ext_modules = get_ext_modules(Extension), 103 cmdclass = {'config' : config, 104 'build' : build, 105 'build_ext' : build_ext}, 106 **metadata) 107 108def run_cython(source): 109 import sys, os 110 source_c = os.path.splitext(source)[0] + '.c' 111 if (os.path.exists(source_c)): 112 return False 113 try: 114 import Cython 115 except ImportError: 116 warn = lambda msg='': sys.stderr.write(msg+'\n') 117 warn("*"*80) 118 warn() 119 warn(" You need to generate C source files with Cython!!") 120 warn(" Download and install Cython <http://www.cython.org>") 121 warn() 122 warn("*"*80) 123 raise SystemExit 124 from distutils import log 125 from conf.cythonize import run as cythonize 126 log.info("cythonizing '%s' source" % source) 127 cythonize(source) 128 return True 129 130def main(): 131 import os 132 try: 133 run_setup() 134 except: 135 done = run_cython(os.path.join('src', 'petsc4py.PETSc.pyx')) 136 if done: 137 run_setup() 138 else: 139 raise 140 141if __name__ == '__main__': 142 main() 143 144# ----------------------------------------------------------------------------- 145