ch-shrinkwrap/setup.py at master · python-microscopy/ch-shrinkwrap · GitHub

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
#!/usr/bin/env python

import os
import numpy
import PYME
import PYME.experimental
from setuptools import setup, Extension
from Cython.Build import cythonize

this_dir = os.path.dirname(os.path.abspath(__file__))
ch_dir = os.path.join(this_dir, 'ch_shrinkwrap')  # absolute, for include_dirs only
pyme_exp_dir = os.path.dirname(PYME.experimental.__file__)
# parent of the PYME package dir — where Cython searches for cimported .pxd files
pyme_pkg_root = os.path.dirname(os.path.dirname(PYME.__file__))

cython_include_dirs = [numpy.get_include(), ch_dir, pyme_exp_dir]

module0 = Extension('ch_shrinkwrap.membrane_mesh_utils',
                    sources=['ch_shrinkwrap/membrane_mesh_utils.c'],
                    include_dirs=[numpy.get_include()])
module1 = Extension('ch_shrinkwrap._membrane_mesh',
                    sources=['ch_shrinkwrap/_membrane_mesh.pyx'],
                    include_dirs=cython_include_dirs)
module2 = Extension('ch_shrinkwrap._skeleton_mesh',
                    sources=['ch_shrinkwrap/_skeleton_mesh.pyx'],
                    include_dirs=cython_include_dirs)
module3 = Extension('ch_shrinkwrap.conj_grad_utils',
                    sources=['ch_shrinkwrap/conj_grad_utils.c'],
                    include_dirs=[numpy.get_include()])

setup(ext_modules=cythonize([module0, module1, module2, module3],
                            include_path=[pyme_pkg_root],
                            language_level=3))