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
34
35
36
37
#!/usr/bin/env python

import os
from setuptools import find_packages #, Extension

from numpy.distutils.core import setup, Extension
from numpy.distutils.misc_util import get_numpy_include_dirs
from Cython.Build import cythonize

this_dir = os.path.dirname(__file__)

def install_pyme_plugin():
    import sys
    import subprocess
    import os
    plugin_install_path = os.path.join(this_dir,
                                       'install_plugin.py')
    subprocess.Popen('%s %s' % (sys.executable, plugin_install_path),
                        shell=True)

module0 = Extension('ch_shrinkwrap.membrane_mesh_utils', sources=[os.path.join(this_dir, 'ch_shrinkwrap/membrane_mesh_utils.c')])
module1 = Extension('ch_shrinkwrap._membrane_mesh', sources=[os.path.join(this_dir, 'ch_shrinkwrap/_membrane_mesh.pyx')], include_dirs=get_numpy_include_dirs())
module2 = Extension('ch_shrinkwrap._skeleton_mesh', sources=[os.path.join(this_dir, 'ch_shrinkwrap/_skeleton_mesh.pyx')], include_dirs=get_numpy_include_dirs())
# module2 = Extension('ch_shrinkwrap._conj_grad', sources=[os.path.join(this_dir, 'ch_shrinkwrap/_conj_grad.pyx')], include_dirs=get_numpy_include_dirs())
module3 = Extension('ch_shrinkwrap.conj_grad_utils', sources=[os.path.join(this_dir, 'ch_shrinkwrap/conj_grad_utils.c')])

setup(name='ch_shrinkwrap',
      version='0.0',
      package_path=os.path.join(this_dir,'ch_shrinkwrap'),
      description='Iterative mesh fitting to SMLM data.',
      author='Zach Marin, David Baddeley',
      author_email='zach.marin@yale.edu, d.baddeley@auckland.ac.nz',
      packages=find_packages(),
      ext_modules = cythonize([module0, module1, module2, module3])
     )

install_pyme_plugin()