setup.py

#!/usr/bin/env python
"""
setup.py

Created by Stephan Hügel on 2016-07-25
"""

import sys

import numpy
from Cython.Build import cythonize
from setuptools import Extension, setup

# # Set dynamic RPATH differently, depending on platform
ldirs = []
ddirs = []
if "linux" in sys.platform:
    # from http://stackoverflow.com/a/10252190/416626
    # the $ORIGIN trick is not perfect, though
    ldirs = ["-Wl,-rpath", "-Wl,$ORIGIN/"]
    platform_lib = "libpolylineffi.so"
if sys.platform == "darwin":
    # You must compile your binary with rpath support for this to work
    # RUSTFLAGS="-C rpath" cargo build --release
    platform_lib = "libpolylineffi.dylib"
    ldirs = ["-Wl,-rpath", "-Wl,@loader_path/"]
if sys.platform == "win32":
    ddirs = ["src/pypolyline/header.h"]
    platform_lib = "polylineffi.dll"
ddirs.append("src/pypolyline/header.h")


extension = Extension(
    name="pypolyline.cutil",
    sources=["src/pypolyline/cutil.pyx"],
    libraries=["polylineffi"],
    depends=ddirs,
    language="c",
    include_dirs=["src/pypolyline", numpy.get_include()],
    library_dirs=["src/pypolyline"],
    extra_link_args=ldirs,
)

extensions = cythonize(
    [
        extension,
    ],
    compiler_directives={"language_level": "3"},
)

setup(
    package_data={
        "pypolyline": [platform_lib],
    },
    ext_modules=[extension],
)