setup.py

from setuptools import setup, find_packages
from setuptools.extension import Extension
import subprocess
import os
import numpy as np

here = os.path.abspath(os.path.dirname(__file__))


about = {}
with open(os.path.join(here, "src", "flowty", "__version__.py"), "r") as f:
    exec(f.read(), about)


def get_cli_output(cmd):
    if isinstance(cmd, str):
        cmd = cmd.split()
    return subprocess.check_output(cmd).decode("utf-8")


# libs are of the form -llibname so we drop the first 2 chars to get
# libname only.
opencv_libs = get_cli_output("pkg-config --libs opencv4").split()
opencv_cflags = get_cli_output("pkg-config --cflags opencv4").split()


def cython_extension(
    pyx_file,
    language="c++",
    name=None,
    include_dirs=None,
    extra_compile_args=None,
    extra_link_args=None,
):
    if include_dirs is None:
        include_dirs = []
    if extra_compile_args is None:
        extra_compile_args = []
    if extra_link_args is None:
        extra_link_args = []
    if name is None:
        name = pyx_file[len("src/"):-len(".pyx")].replace("/", ".")
    return Extension(
        name,
        sources=[pyx_file],
        language="c++",
        libraries=["stdc++"],
        include_dirs=[np.get_include(), *include_dirs],
        extra_compile_args=opencv_cflags + extra_compile_args,
        extra_link_args=opencv_libs + extra_link_args,
    )


extensions = [
    cython_extension(path, extra_compile_args=['-std=c++11'])
    for path in [
        "src/flowty/cv/core.pyx",
        "src/flowty/cv/videoio.pyx",
        "src/flowty/cv/imgcodecs.pyx",
        "src/flowty/cv/cuda.pyx",
        "src/flowty/cv/optflow.pyx",
        "src/flowty/cv/cuda_optflow.pyx",
    ]
]

docs_require = ["sphinx"]
tests_require = ["pytest", "imageio", "cython"]

if __name__ == '__main__':
    setup(
        name=about["__title__"],
        description=about["__description__"],
        version=about["__version__"],
        ext_modules=extensions,
        packages=find_packages('src') + ['flowty.algorithms'],
        package_dir={'': 'src'},
        install_requires=["numpy", "tqdm"],
        extras_require={
            "docs": docs_require,
            "test": tests_require,
            "dev": tests_require + docs_require,
        },
        # Include package data specified in MANIFEST.in
        include_package_data=True,
        classifiers=[
            # How mature is this project? Common values
            # are
            #   3 - Alpha
            #   4 - Beta
            #   5 - Production/Stable
            "Development Status :: 3 - Alpha",
            "Intended Audience :: Developers",
            "Programming Language :: Python :: 3.5",
            "Programming Language :: Python :: 3.6",
            "Programming Language :: Python :: 3.7",
        ],
        entry_points={
            'console_scripts': ['flowty=flowty.flowty:main']
        },
        keywords=[
            "computer-vision",
            "optical-flow",
            "optical",
            "flow",
            "computer",
            "vision",
            "opencv",
            "cython",
            "cuda",
            "gpu",
            "tvl1",
            "farneback",
            "brox",
            "flowty",
        ],
        author=about["__author__"],
        author_email=about["__author_email__"],
        license=about["__license__"],
        url="http://github.com/willprice/flowty",
        project_urls={
            "Bug Tracker": "https://github.com/willprice/flowty/issues",
            "Documentation": "https://flowty.readthedocs.io",
            "Source Code": "http://github.com/willprice/flowty",
        },
        zip_safe=False
    )