toolkit/setup.py at main · synaptics-synap/toolkit · 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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
from setuptools import setup, find_packages, Extension
import os, shutil, re
from pathlib import Path
from datetime import datetime

# Get the current pysynap version
def get_version():
    with open('pysynap/__init__.py', 'r') as init_file:
        init_py = init_file.read()
    version_match = re.search(r"^version = ['\"]([^'\"]*)['\"]", init_py, re.M)
    if version_match:
        version = version_match.group(1)
        snapshot = datetime.now().strftime("%Y%m%d%H%M%S")
        return f"{version}rc{snapshot}"
    raise RuntimeError("Unable to find version string.")

# Get the required dependecy list
def get_requirements(directory):
    requirements = []
    for root, dirs, files in os.walk(directory):
        for file in files:
            if file.endswith("requirements.txt"):
                with open(os.path.join(root, file), "r") as f:
                    for line in f:
                        line = line.strip()
                        # Skip lines starting with --extra-index-url
                        if not line.startswith('--extra-index-url'):
                            requirements.append(line)
    return requirements

requirements = get_requirements('.')

# Define the extension module
extensions = [
    Extension(
        name="pysynap.prebuilts",
        sources=[],
        libraries=["ebg_utils", "ovxlib"],
        library_dirs=["pysynap/prebuilts/lib/x86_64-linux-gcc"],
    ),
]

# Ensure the scripts are available in the package directory to be included in the wheel
shutil.copyfile('synap_convert.py', 'pysynap/scripts/synap_convert.py')
shutil.copyfile('image_from_raw.py', 'pysynap/scripts/image_from_raw.py')
shutil.copyfile('image_to_raw.py', 'pysynap/scripts/image_to_raw.py')
shutil.copyfile('image_od.py', 'pysynap/scripts/image_od.py')
shutil.copyfile('export_yolo.py', 'pysynap/scripts/export_yolo.py')

setup(
    name='synap',
    version=get_version(),
    author='Synaptics',
    author_email='meetdineshbhai.patel@synaptics.com',
    description='Synaptics AI Toolkit',
    packages=find_packages(exclude=['**/.*', '**/.git']),
    python_requires='>=3.10, <3.13',
    install_requires=requirements,
    package_dir={
        'pysynap': 'pysynap'
    },
    package_data={
        'pysynap': ['*', 'utils/*.sh'],
    },
    include_package_data=True,
    ext_modules=extensions,
    classifiers=[ 'Operating System :: POSIX :: Linux','Programming Language :: Python :: 3.10','Programming Language :: Python :: 3.12',],
    entry_points={
        'console_scripts': [
            'synap_convert=pysynap.scripts.synap_convert:main',
            'synap_image_from_raw=pysynap.scripts.image_from_raw:main',
            'synap_image_to_raw=pysynap.scripts.image_to_raw:main',
            'synap_image_od=pysynap.scripts.image_od:main',
            'synap_export_yolo=pysynap.scripts.export_yolo:main',
            'synap_export_clean=pysynap.tools.utils.cleanup:main',
            'synap_copy=pysynap.tools.copy:main',
            'synap_profile=pysynap.tools.profile:main',
            'synap=pysynap.scripts.synap_help:main',
        ],
    },
)