93 lines
3.2 KiB
Python
93 lines
3.2 KiB
Python
from __future__ import division, print_function, absolute_import
|
|
|
|
from os.path import join, dirname
|
|
import glob
|
|
|
|
|
|
def pre_build_hook(build_ext, ext):
|
|
from scipy._build_utils.compiler_helper import get_cxx_std_flag
|
|
std_flag = get_cxx_std_flag(build_ext._cxx_compiler)
|
|
if std_flag is not None:
|
|
ext.extra_compile_args.append(std_flag)
|
|
|
|
|
|
def configuration(parent_package='', top_path=None):
|
|
from numpy.distutils.misc_util import Configuration, get_numpy_include_dirs
|
|
from numpy.distutils.misc_util import get_info as get_misc_info
|
|
from scipy._build_utils.system_info import get_info as get_sys_info
|
|
from distutils.sysconfig import get_python_inc
|
|
|
|
config = Configuration('spatial', parent_package, top_path)
|
|
|
|
config.add_data_dir('tests')
|
|
|
|
# spatial.transform
|
|
config.add_subpackage('transform')
|
|
|
|
# qhull
|
|
qhull_src = sorted(glob.glob(join(dirname(__file__), 'qhull_src',
|
|
'src', '*.c')))
|
|
|
|
inc_dirs = [get_python_inc()]
|
|
if inc_dirs[0] != get_python_inc(plat_specific=1):
|
|
inc_dirs.append(get_python_inc(plat_specific=1))
|
|
inc_dirs.append(get_numpy_include_dirs())
|
|
inc_dirs.append(join(dirname(dirname(__file__)), '_lib'))
|
|
|
|
cfg = dict(get_sys_info('lapack_opt'))
|
|
cfg.setdefault('include_dirs', []).extend(inc_dirs)
|
|
config.add_extension('qhull',
|
|
sources=['qhull.c', 'qhull_misc.c'] + qhull_src,
|
|
**cfg)
|
|
|
|
# cKDTree
|
|
ckdtree_src = ['query.cxx',
|
|
'build.cxx',
|
|
'query_pairs.cxx',
|
|
'count_neighbors.cxx',
|
|
'query_ball_point.cxx',
|
|
'query_ball_tree.cxx',
|
|
'sparse_distances.cxx']
|
|
|
|
ckdtree_src = [join('ckdtree', 'src', x) for x in ckdtree_src]
|
|
|
|
ckdtree_headers = ['ckdtree_decl.h',
|
|
'coo_entries.h',
|
|
'distance_base.h',
|
|
'distance.h',
|
|
'ordered_pair.h',
|
|
'partial_sort.h',
|
|
'rectangle.h']
|
|
|
|
ckdtree_headers = [join('ckdtree', 'src', x) for x in ckdtree_headers]
|
|
|
|
ckdtree_dep = ['ckdtree.cxx'] + ckdtree_headers + ckdtree_src
|
|
ext = config.add_extension('ckdtree',
|
|
sources=['ckdtree.cxx'] + ckdtree_src,
|
|
depends=ckdtree_dep,
|
|
include_dirs=inc_dirs + [join('ckdtree', 'src')])
|
|
ext._pre_build_hook = pre_build_hook
|
|
|
|
# _distance_wrap
|
|
config.add_extension('_distance_wrap',
|
|
sources=[join('src', 'distance_wrap.c')],
|
|
depends=[join('src', 'distance_impl.h')],
|
|
include_dirs=[get_numpy_include_dirs()],
|
|
extra_info=get_misc_info("npymath"))
|
|
|
|
config.add_extension('_voronoi',
|
|
sources=['_voronoi.c'])
|
|
|
|
config.add_extension('_hausdorff',
|
|
sources=['_hausdorff.c'])
|
|
|
|
# Add license files
|
|
config.add_data_files('qhull_src/COPYING.txt')
|
|
|
|
return config
|
|
|
|
|
|
if __name__ == '__main__':
|
|
from numpy.distutils.core import setup
|
|
setup(**configuration(top_path='').todict())
|