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