classylss/setup.py at master · bccp/classylss · 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
import os
import shutil
from glob import glob
from setuptools import setup, Extension, find_packages
from setuptools.command.build_ext import build_ext
from Cython.Build import cythonize
import numpy

# base directory of package
package_basedir = os.path.abspath(os.path.dirname(__file__))

def find_version(path, name='version'):
    import re
    s = open(path, 'rt').read()
    version_match = re.search(r"^%s = ['\"]([^'\"]*)['\"]" % name, s, re.M)
    if version_match:
        return version_match.group(1)
    raise RuntimeError("Unable to find version string.")

CLASS_VERSION = find_version("classylss/version.py", name='class_version')

def build_CLASS(prefix):
    args = (package_basedir, package_basedir, CLASS_VERSION, os.path.abspath(prefix), "./")
    command = 'sh %s/depends/install_class.sh %s %s %s %s' % args
    ret = os.system(command)
    if ret != 0:
        raise ValueError("Could not build CLASS v%s" % CLASS_VERSION)

class BuildCLib(build_ext):
    """ Custom command to build external C libraries before building extensions """
    def run(self):
        build_CLASS(self.build_temp)
        # copy data files from temp to classlssy package directory
        shutil.rmtree(os.path.join(self.build_lib, 'classylss', 'data'), ignore_errors=True)
        shutil.copytree(os.path.join(self.build_temp, 'data'), os.path.join(self.build_lib, 'classylss', 'data'))
        super().run()

    def build_extensions(self):
        libclass_path = glob(os.path.join(self.build_temp, '*', 'libclass.a'))
        if not libclass_path:
            raise FileNotFoundError("libclass.a not found in expected build directories.")

        for ext in self.extensions:
            ext.extra_objects = libclass_path
            ext.include_dirs.insert(0, os.path.join(self.build_temp, 'include'))
            ext.library_dirs.insert(0, os.path.join(self.build_temp, 'lib'))

        super().build_extensions()

classy_ext = Extension('classylss.binding',
        sources= ['classylss/binding.pyx',],
        include_dirs = [numpy.get_include(),],
        extra_link_args = ['-g', '-fPIC'],
        extra_compile_args = ['-g'],
        language= 'c++',
        libraries = ['class', 'm'],
        )

setup(
    name='classylss',
    version=find_version("classylss/version.py"),
    author='Nick Hand, Yu Feng, Simeon Bird',
    author_email='spb@ucr.edu',
    description="More Extensive Python binding of the CLASS CMB Boltzmann code",
    license='GPL3',
    url="http://github.com/nickhand/classylss",
    install_requires=['numpy', 'Cython'],
    extras_require={'tests': ['pytest', 'astropy', 'scipy']},
    ext_modules=cythonize(classy_ext, compiler_directives={'language_level' : "3"}, include_path=[numpy.get_include(), "classylss/"]),
    cmdclass={
        'build_ext': BuildCLib,
        # 'clean': CustomClean
    },
    # package_data = {'classylss': ['*.pxd', '*.pyx']},
    packages=find_packages()
)