build.py

import os
from typing import Any, Dict, List

import numpy as np

# setuptools must be imported before cython. setuptools overrides the
# distutils.extension.Extension class. If imported after, the isinstance check in
# distutils.command.check_extensions_list fails and setuptools gives the erroneous
# error:
#
# "error: each element of 'ext_modules' option must be an Extension instance or 2-tuple"
#
# ref: https://github.com/cython/cython/issues/4724
from setuptools import find_packages  # isort: skip
from setuptools.command.build_ext import build_ext  # isort: skip
from setuptools.extension import Extension  # isort: skip
from Cython.Build import cythonize  # isort: skip

cwd = os.path.dirname(__file__)
package_dir = os.path.join(cwd, "pyembree")
dependencies_dir = os.path.join(package_dir, "embree")

version_file = os.path.join(package_dir, "_version.py")

with open(version_file, mode="r") as fd:
    exec(fd.read())

include = [
    np.get_include(),
    os.path.join(dependencies_dir, "include", "embree2"),
]
library = [
    os.path.join(dependencies_dir, "lib"),
    os.path.join(dependencies_dir, "bin"),
]

packages = ["pyembree"]

with open("README.rst", encoding="utf-8") as file_:
    readme = file_.read()


def build(setup_kwargs: Dict[str, Any]) -> None:
    ext_modules: List[Extension] = cythonize(
        module_list="pyembree/*.pyx",
        language_level=3,
        include_path=include,
    )

    for ext in ext_modules:
        ext.include_dirs = include
        ext.library_dirs = library
        if os.name == "nt":
            ext.libraries = [
                "embree",
                "tbb",
                "tbbmalloc",
            ]
        else:
            # It is recommended to build against tbb and tbbmalloc, which may improve
            # the library's runtime performance. However, to be a 'manylinux' wheel, these
            # must be removed for maximum portability.
            #
            # See also `ci/embree_linux.bash`
            #
            ext.libraries = [
                "embree",
                # "tbb",  # Uncomment to build against tbb
                # "tbbmalloc"  # Uncomment to build against tbb
            ]

    setup_kwargs.update(
        {
            "name": "pyembree",
            "version": __version__,
            "description": "Python wrapper for Intel Embree 2.17.7",
            "long_description": readme,
            "long_description_content_type": "text/x-rst",
            "author": "Anthony Scopatz",
            "author_email": "scopatz@gmail.com",
            "maintainer": "Adam Hendry",
            "maintainer_email": "adam.grant.hendry@gmail.com",
            "url": "https://github.com/adam-grant-hendry/pyembree",
            "ext_modules": ext_modules,
            "cmdclass": {"build_ext": build_ext},
            "zip_safe": False,
            "packages": find_packages(),
            "packages": packages,
            "python_requires": ">=3.8,<3.9",
            "classifiers": [
                "License :: OSI Approved :: BSD License",
                "Operating System :: POSIX :: Linux",
                "Operating System :: MacOS :: MacOS X",
                "Operating System :: Microsoft :: Windows :: Windows 10",
                "Programming Language :: Python :: 3",
                "Programming Language :: Python :: 3.8",
            ],
        }
    )