pyksc/setup.py at master · flaviovdf/pyksc · 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
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
#!/usr/bin/env python
# -*- coding: utf-8
from __future__ import division, print_function
'''Setup script'''

import glob
import numpy
import os
import sys

from distutils.core import setup
from distutils.extension import Extension
from Cython.Distutils import build_ext

SOURCE = 'src/'
os.chdir(SOURCE)

import platform
if platform.system() == 'Darwin':
    os.environ["CC"] = "gcc-6"
    os.environ["CXX"] = "gcc-6"

if sys.version_info[:2] < (2, 7):
    print('Requires Python version 2.7 or later (%d.%d detected).' %
          sys.version_info[:2])
    sys.exit(-1)

def get_packages():
    '''Appends all packages (based on recursive sub dirs)'''

    packages  = ['pyksc']

    for package in packages:
        base = os.path.join(package, '**/')
        sub_dirs = glob.glob(base)
        while len(sub_dirs) != 0:
            for sub_dir in sub_dirs:
                package_name = sub_dir.replace('/', '.')
                if package_name.endswith('.'):
                    package_name = package_name[:-1]

                packages.append(package_name)

            base = os.path.join(base, '**/')
            sub_dirs = glob.glob(base)

    return packages

def get_extensions():
    '''Get's all .pyx and.pxd files'''

    extensions = []
    for base in ['pyksc']:
        pyx_files = glob.glob(os.path.join(base, '*.pyx'))

        for pyx in pyx_files:
            pxd = pyx.replace('pyx', 'pxd')
            module = pyx.replace('.pyx', '').replace('/', '.')

            if os.path.exists(pxd):
                ext_files = [pyx, pxd]
            else:
                ext_files = [pyx]

            extension = Extension(module, ext_files,
                                  include_dirs=[numpy.get_include()],
                                  libraries=['blas'],
                                  extra_compile_args=['-fopenmp',
                                   '-msse', '-msse2', '-mfpmath=sse'],
                                  extra_link_args=['-fopenmp'])

            extensions.append(extension)

    return extensions

if __name__ == "__main__":
    packages = get_packages()
    extensions = get_extensions()

    setup(
        cmdclass = {'build_ext': build_ext},
        name             = 'pyksc',
        packages         = packages,
        ext_modules      = extensions
      )