setup.py

from __future__ import absolute_import, division, print_function, unicode_literals
from distutils.core import setup
from Cython.Build import cythonize
import numpy
from distutils.extension import Extension
from subprocess import check_output
from distutils import sysconfig
import re

extra_compile_args = ["-std=c++11"]

# Super hacky way of determining if clang or gcc is being used
CC = sysconfig.get_config_vars().get("CC", "gcc").split(" ")[0]
out = check_output([CC, "--version"])
if re.search("apple *llvm", str(out.lower())):
    extra_compile_args.append("-stdlib=libc++")

extensions = [
    Extension(
        "hype.graph_dataset",
        ["hype/graph_dataset.pyx"],
        include_dirs=[numpy.get_include()],
        extra_compile_args=extra_compile_args,
        language="c++",
    ),
    Extension(
        "hype.adjacency_matrix_dataset",
        ["hype/adjacency_matrix_dataset.pyx"],
        include_dirs=[numpy.get_include()],
        extra_compile_args=extra_compile_args,
        language="c++",
    ),
]


setup(ext_modules=cythonize(extensions))