mirror of
https://github.com/wassname/scikit-image.git
synced 2026-07-06 05:16:40 +08:00
sccolbert
64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
#!/usr/bin/env python
|
|
|
|
import os
|
|
import shutil
|
|
|
|
def configuration(parent_package='', top_path=None):
|
|
from numpy.distutils.misc_util import Configuration, get_numpy_include_dirs
|
|
|
|
config = Configuration('opencv', parent_package, top_path)
|
|
|
|
config.add_data_dir('tests')
|
|
|
|
# since distutils/cython has problems, we'll check to see if cython is
|
|
# installed and use that to rebuild the .c files, if not, we'll just build
|
|
# directly from the included .c files
|
|
|
|
cython_files = ['opencv_backend.pyx', 'opencv_cv.pyx']
|
|
|
|
try:
|
|
import Cython
|
|
for pyxfile in cython_files:
|
|
# make a backup of the good c files
|
|
c_file = pyxfile.rstrip('pyx') + 'c'
|
|
src = c_file
|
|
dst = c_file + '.bak'
|
|
shutil.copy(src, dst)
|
|
|
|
# run cython compiler
|
|
os.system('cython ' + pyxfile)
|
|
|
|
# if the file is small, cython compilation failed
|
|
size = os.path.getsize(c_file)
|
|
if size < 100:
|
|
print 'Cython compilation failed. Restoring from backup.'
|
|
# restore from backup
|
|
shutil.copy(dst, src)
|
|
|
|
except ImportError:
|
|
# if cython is not found, we just build from the include .c files
|
|
pass
|
|
|
|
for pyxfile in cython_files:
|
|
c_file = pyxfile.rstrip('pyx') + 'c'
|
|
config.add_extension(pyxfile.rstrip('.pyx'),
|
|
sources=[c_file],
|
|
include_dirs=[get_numpy_include_dirs()])
|
|
|
|
return config
|
|
|
|
if __name__ == '__main__':
|
|
from numpy.distutils.core import setup
|
|
setup(maintainer = 'Scikits.Image Developers',
|
|
author = 'Steven C. Colbert',
|
|
maintainer_email = 'scikits-image@googlegroups.com',
|
|
description = 'OpenCV wrapper for NumPy arrays',
|
|
url = 'http://stefanv.github.com/scikits.image/',
|
|
license = 'SciPy License (BSD Style)',
|
|
**(configuration(top_path='').todict())
|
|
)
|
|
|
|
|
|
|
|
|
|
|