setup.py

"""
Setup script for Fast JPEG Decoder
"""

from setuptools import setup, Extension
from setuptools.command.build_ext import build_ext
import sys
import os

class get_pybind_include:
    """Helper class to determine the pybind11 include path"""
    def __str__(self):
        import pybind11
        return pybind11.get_include()

ext_modules = [
    Extension(
        'fast_jpeg_decoder._fast_jpeg_decoder',
        sources=[
            'src/cpp/bitstream.cpp',
            'src/cpp/idct.cpp',
            'src/cpp/huffman.cpp',
            'src/cpp/decoder.cpp',
            'src/bindings/bindings.cpp',
        ],
        include_dirs=[
            get_pybind_include(),
            'src/cpp',
        ],
        language='c++',
        extra_compile_args=['-std=c++11', '-O3', '-Wall'],
    ),
]

setup(
    name='fast_jpeg_decoder',
    version='0.1.0',
    author='Your Name',
    description='A high-performance JPEG decoder implemented in C++',
    long_description='',
    ext_modules=ext_modules,
    packages=['fast_jpeg_decoder'],
    package_dir={'fast_jpeg_decoder': 'src/python/fast_jpeg_decoder'},
    install_requires=['numpy', 'pybind11>=2.6.0'],
    python_requires='>=3.6',
    zip_safe=False,
)