setup.py 3.83 KB
Newer Older
1
from distutils.version import LooseVersion
Roel Aaij's avatar
Roel Aaij committed
2
from setuptools import setup, Extension
3
from setuptools.command.build_ext import build_ext
Roel Aaij's avatar
Roel Aaij committed
4
from setuptools.command.test import test as test_ext
5
6
7
8
9
import sys
import os
import re
import platform
import subprocess
Roel Aaij's avatar
Roel Aaij committed
10
import sysconfig
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

__version__ = '0.0.1'


class CMakeExtension(Extension):
    def __init__(self, name, sourcedir=''):
        Extension.__init__(self, name, sources=[])
        self.sourcedir = os.path.abspath(sourcedir)


class CMakeBuild(build_ext):
    def run(self):
        try:
            out = subprocess.check_output(['cmake', '--version'])
        except OSError:
            raise RuntimeError(
                "CMake must be installed to build the following extensions: " +
                ", ".join(e.name for e in self.extensions))

        if platform.system() == "Windows":
            cmake_version = LooseVersion(re.search(r'version\s*([\d.]+)',
                                                   out.decode()).group(1))
            if cmake_version < '3.1.0':
                raise RuntimeError("CMake >= 3.1.0 is required on Windows")

        for ext in self.extensions:
            self.build_extension(ext)

    def build_extension(self, ext):
        extdir = os.path.abspath(
            os.path.dirname(self.get_ext_fullpath(ext.name)))
        cmake_args = ['-DCMAKE_LIBRARY_OUTPUT_DIRECTORY=' + extdir,
                      '-DPYTHON_EXECUTABLE=' + sys.executable]

        cfg = 'Debug' if self.debug else 'Release'
        build_args = ['--config', cfg]

        if platform.system() == "Windows":
            cmake_args += ['-DCMAKE_LIBRARY_OUTPUT_DIRECTORY_{}={}'.format(
                cfg.upper(),
                extdir)]
            if sys.maxsize > 2**32:
                cmake_args += ['-A', 'x64']
            build_args += ['--', '/m']
        else:
            cmake_args += ['-DCMAKE_BUILD_TYPE=' + cfg]
Roel Aaij's avatar
Roel Aaij committed
57
            build_args += ['--', '-j1']
58
59
60
61
62
63
64
65
66
67
68
69
70
71

        env = os.environ.copy()
        env['CXXFLAGS'] = '{} -DVERSION_INFO=\\"{}\\"'.format(
            env.get('CXXFLAGS', ''),
            self.distribution.get_version())
        if not os.path.exists(self.build_temp):
            os.makedirs(self.build_temp)
        subprocess.check_call(['cmake', ext.sourcedir] + cmake_args,
                              cwd=self.build_temp, env=env)
        subprocess.check_call(['cmake', '--build', '.'] + build_args,
                              cwd=self.build_temp)
        print()  # Add an empty line for cleaner output


Roel Aaij's avatar
Roel Aaij committed
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
class CatchTestCommand(test_ext):
    """
    A custom test runner to execute both Python unittest tests and C++ Catch-
    lib tests.
    """

    def distutils_dir_name(self, dname):
        """Returns the name of a distutils build directory"""
        dir_name = "{dirname}.{platform}-{version[0]}.{version[1]}"
        return dir_name.format(dirname=dname,
                               platform=sysconfig.get_platform(),
                               version=sys.version_info)

    def run(self):
        # Run Python tests
        super(CatchTestCommand, self).run()
        print("\nPython tests complete, now running C++ tests...\n")
        # Run catch tests
        subprocess.call(['ctest -V'],
                        cwd=os.path.join('build',
                                         self.distutils_dir_name('temp')),
                        shell=True)


96
97
98
99
100
101
102
103
104
105
ext_modules = [
    CMakeExtension('k40gen')
]


setup(
    name='k40gen',
    version=__version__,
    author='Roel Aaij',
    author_email='roelaaij@nikhef.nl',
Roel Aaij's avatar
Roel Aaij committed
106
    url='https://sikkel.nikhef.nl/roelaaij/k40gen',
107
108
109
    description='standalone background generator for KM3NeT',
    long_description='',
    ext_modules=ext_modules,
Roel Aaij's avatar
Roel Aaij committed
110
111
112
    setup_requires=['pytest-runner', 'numpy', 'cmake'],
    install_requires=['numpy'],
    tests_require=["pytest"],
Roel Aaij's avatar
Roel Aaij committed
113
    cmdclass=dict(build_ext=CMakeBuild, test=CatchTestCommand),
114
115
    zip_safe=False,
)