idSim/setup.py at master · molumitu/idSim · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
import os, sys
from setuptools import Extension, setup, find_packages
from Cython.Build import cythonize

import numpy as np

prefix = os.environ["CONDA_PREFIX"]
if sys.platform == "win32":
    include_dirs = [os.path.join(prefix, "Library", "include")]
    library_dirs = [os.path.join(prefix, "Library", "lib")]
else:
    include_dirs = [os.path.join(prefix, "include")]
    library_dirs = [os.path.join(prefix, "lib")]

extensions = [
    Extension(
        "idsim.lib",
        [
            "src/lib.pyx",
            "src/lidar.c",
            "src/line.c",
            "src/polygon.c",
            "src/vehicle.c",
            "src/waypoint.c",
        ],
        include_dirs=[
            *include_dirs,
            np.get_include(),
        ],
        library_dirs=library_dirs,
        libraries=["geos_c"],
    )
]

setup(
    packages=find_packages(),
    include_package_data=False,
    package_data={
        "idsim": ["*.pyi", "assets/**/*", "py.typed"],
        "idscene": ["*.pyi", "assets/**/*", "py.typed"],
    },
    ext_modules=cythonize(extensions),
)