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

package_name = 'camera_kl520'

models_paths = []
directories= glob('models/')+glob('models/*/')+glob('models/*/*/')
for directory in directories:
    models_paths.append((os.path.join('share',package_name,directory),glob(f'{directory}/*.*')))

setup(
    name=package_name,
    version='0.0.0',
    packages=find_packages(exclude=['test']),
    data_files=[
        ('share/ament_index/resource_index/packages',
            ['resource/' + package_name]),
        ('share/' + package_name, ['package.xml']),
        # get all the utils python files
        (os.path.join('share', package_name, 'utils'),glob('utils/*.py')),
        # get the firmware image and models
        (os.path.join('share', package_name, 'res','firmware','KL520'),glob('res/firmware/KL520/*.bin')),
        (os.path.join('share', package_name, 'res','images'),glob('res/images/*.*')),
        (os.path.join('share', package_name, 'res','models','KL520', 'yolov5-noupsample_w640h640_kn-model-zoo'),glob('res/models/KL520/yolov5-noupsample_w640h640_kn-model-zoo/*.nef')),
        ]+models_paths,

    install_requires=[
        'setuptools', 
        'torch', 
        'torchvision',
    ],
    zip_safe=True,
    maintainer='jack',
    maintainer_email='jack23162329@gmail.com',
    description='TODO: Package description',
    license='TODO: License declaration',
    # tests_require=['pytest'],
    entry_points={
        'console_scripts': [
            "detection_pub = camera_kl520.object_detection_pub:main",
            "detection_sub = camera_kl520.object_detection_sub:main",
            'yolo_inference_pub = camera_kl520.camera_yolo5l_pub:main',
            'yolo_inference_sub = camera_kl520.camera_yolo5l_sub:main',
            "testing_camera = camera_kl520.testing_camera:main",
            "scan_kl520 = camera_kl520.Scan_kl520:main",
            "detection_kl520_pub = camera_kl520.object_detection_kl520_pub:main",
            "detection_kl520_sub = camera_kl520.object_detection_kl520_sub:main",
            "yolo5s_kl520_pub = camera_kl520.camera_yolo5v_kl520_pub:main",
            "yolo5s_kl520_sub = camera_kl520.camera_yolo5v_kl520_sub:main",
            "yolo5s_kl720_pub = camera_kl520.camera_yolo5v_kl720_pub:main",
            "yolo5s_kl720_sub = camera_kl520.camera_yolo5v_kl720_sub:main",
        ],
    },
)