/
setup.py
executable file
·103 lines (97 loc) · 4.34 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
#!/usr/bin/env python
from __future__ import division
#from __future__ import unicode_literals
from __future__ import print_function
from __future__ import absolute_import
import os, sys
from os.path import join
# We require Python v2.7 or newer
if sys.version_info[:2] < (2,7): raise RuntimeError("This requires Python v2.7 or newer")
# Prepare for compiling the source code
from distutils.ccompiler import get_default_compiler
import numpy
compiler_name = get_default_compiler() # TODO: this isn't the compiler that will necessarily be used, but is a good guess...
compiler_opt = {
'msvc' : ['/D_SCL_SECURE_NO_WARNINGS','/EHsc','/O2','/DNPY_NO_DEPRECATED_API=7','/bigobj','/openmp'],
# TODO: older versions of gcc need -std=c++0x instead of -std=c++11
'unix' : ['-std=c++11','-O3','-DNPY_NO_DEPRECATED_API=7','-fopenmp'], # gcc/clang (whatever is system default)
'mingw32' : ['-std=c++11','-O3','-DNPY_NO_DEPRECATED_API=7','-fopenmp'],
'cygwin' : ['-std=c++11','-O3','-DNPY_NO_DEPRECATED_API=7','-fopenmp'],
}
linker_opt = {
'msvc' : [],
'unix' : ['-fopenmp'], # gcc/clang (whatever is system default)
'mingw32' : ['-fopenmp'],
'cygwin' : ['-fopenmp'],
}
np_inc = numpy.get_include()
import pysegtools
cy_inc = join(os.path.dirname(pysegtools.__file__), 'general', 'cython') # TODO: better way to get this
src_ext = '.cpp'
def create_ext(name, dep=[], src=[], inc=[], lib=[], objs=[]):
from distutils.extension import Extension
return Extension(
name=name,
depends=dep,
sources=[join(*name.split('.'))+src_ext]+src,
define_macros=[('NPY_NO_DEPRECATED_API','7'),],
include_dirs=[np_inc,cy_inc]+inc,
library_dirs=lib,
extra_objects=objs,
extra_compile_args=compiler_opt.get(compiler_name, []),
extra_link_args=linker_opt.get(compiler_name, []),
language='c++',
)
def create_ext_mtrand(name, dep=[], src=[], inc=[], lib=[], objs=[]):
"""Create an extension that links against the mtrand library."""
import numpy.random.mtrand
from os.path import isfile, dirname, abspath
path = getattr(numpy.random.mtrand, '__file__', None)
if path is None or not isfile(path):
path = abspath(join(np_inc, '..', '..', 'random', 'mtrand'))
for ext in ('.so', '.dylib', '.pyd', '.dll'):
if os.path.isfile(path+ext): path += ext; break
else: raise ImportError('cannot find mtrand library')
return create_ext(name, dep, src, inc + [dirname(path)], lib, objs + [path])
# Find and use Cython if available
try:
from distutils.version import StrictVersion
import Cython.Build
if StrictVersion(Cython.__version__) >= StrictVersion('0.22'):
src_ext = '.pyx'
def cythonize(*args, **kwargs):
kwargs.setdefault('include_path', []).append(cy_inc)
return Cython.Build.cythonize(*args, **kwargs)
except ImportError:
def cythonize(exts, *args, **kwargs): return exts
# Finally we get to run setup
try: from setuptools import setup
except ImportError: from distutils.core import setup
import chm
setup(name='chm',
version='%s'%chm.__version__,
description='Cascaded Hierarchical Model - an image segmentation framework',
author='Jeffrey Bush',
author_email='jeff@coderforlife.com',
url='https://www.sci.utah.edu/software/chm.html',
packages=['chm', 'chm.filters'],
setup_requires=['numpy>=1.7'],
install_requires=['numpy>=1.7','scipy>=0.16','pysegtools[MATLAB]>=0.1'],
extras_require={ 'OPT': ['pyfftw>=0.10'], },
use_2to3=True, # the code *should* support Python 3 once run through 2to3 but this isn't tested
zip_safe=False,
package_data = { '': ['*.pyx', '*.pyxdep', '*.pxi', '*.pxd', '*.h', '*.txt'], }, # Make sure all Cython files are wrapped up with the code
ext_modules = cythonize([
create_ext('chm._utils'),
create_ext('chm.__stats'),
create_ext('chm.__ldnn'),
create_ext_mtrand('chm.__shuffle'),
create_ext('chm.__imresize'),
create_ext('chm.filters._correlate'),
create_ext('chm.filters._haar'),
create_ext('chm.filters._hog', dep=[join('chm','filters','HOG.h')], src=[join('chm','filters','HOG.cpp')]),
create_ext('chm.filters._sift'),
create_ext('chm.filters._frangi'),
create_ext('chm.filters._intensity'),
])
)