from distutils.core import setup
from distutils.extension import Extension
from Cython.Build import cythonize
import numpy

all_extensions = Extension(
    "*",
    ["pyisda/*.pyx"],
    include_dirs=["c_layer", numpy.get_include()],
    libraries=["cds", "farmhash", "lz4"],
    language="c++",
    extra_compile_args=["-fopenmp"],
    extra_link_args=["-fopenmp", "-Wl,--strip-all"],
)

c_extension = Extension(
    "pyisda.optim",
    include_dirs=["c_layer", numpy.get_include()],
    sources=["pyisda/optim.pyx", "c_layer/cdsbootstrap.c"],
    libraries=["cds"],
    language="c++",
)

all_extensions = cythonize(
    [c_extension, all_extensions],
    nthreads=4,
    compiler_directives={"embedsignature": True,
        "language_level": 3,
        "c_string_type": "unicode",
        "c_string_encoding": "ascii"},
)

setup(
    name="pyisda",
    version="0.3.21",
    author="Guillaume Horel",
    ext_modules=all_extensions,
    packages=["pyisda"],
)