from setuptools import find_packages, setup
import os
from glob import glob

package_name = "yoso_pkg"

setup(
    name=package_name,
    version="0.0.0",
    packages=find_packages(exclude=["resource"]),
    data_files=[
        ("share/ament_index/resource_index/packages", ["resource/" + package_name]),
        ("share/" + package_name, ["package.xml"]),
        (
            os.path.join("share", package_name, "launch"),
            glob(os.path.join("launch", "*.launch.[pxy][yma]*")),
        ),
        (os.path.join("share", package_name, "config"), glob("config/*.yaml")),
        (
            os.path.join("share", package_name, "config/cityscapes"),
            glob("config/cityscapes/*.yaml"),
        ),
        (
            os.path.join("share", package_name, "config/klass-dataset"),
            glob("config/klass-dataset/*.yaml"),
        ),
    ],
    install_requires=["setuptools"],
    zip_safe=True,
    maintainer="hu.zhi",
    maintainer_email="hu.zhi@todo.todo",
    description="TODO: Package description",
    license="TODO: License declaration",
    tests_require=["pytest"],
    entry_points={
        "console_scripts": [
            "yoso_inferencer = yoso_pkg.yoso_pub:main",
            "yoso_visualization = yoso_pkg.yoso_visualization:main",
        ],
    },
)