feat: added JPEG input and output format

Author: ForNeus57

Makefile

@@ -14,7 +14,7 @@ clang-format:
 	 clang-format -i $(shell find ./src/cpp/ -name '*.hpp' -o -name '*.cpp')
 
 clang-tidy:
-	clang-tidy $(shell find ./src/cpp/ -name '*.hpp' -o -name '*.cpp') -- -I$(shell python3 -c "import sysconfig; print(sysconfig.get_path('include'))") -I$(shell python3 -c "import numpy; print(numpy.get_include())")
+	clang-tidy $(shell find ./src/cpp/ -name '*.hpp' -o -name '*.cpp') -- -I$(shell python3 -c "import sysconfig; print(sysconfig.get_path('include'))") -I$(shell python3 -c "import numpy; print(numpy.get_include())") -I$(shell python3 -c "import sysconfig; print(f'{sysconfig.get_paths()['purelib']}/nvidia/nvjpeg/include')") -I$(shell python3 -c "import sysconfig; print(f'{sysconfig.get_paths()['purelib']}/nvidia/cuda_runtime/include')") -I$(shell python3 -c "import sysconfig; print(f'{sysconfig.get_paths()['purelib']}/nvidia/cuda_nvcc/include')")
 
 mypy:
 	mypy ./src/python/

pyproject.toml

@@ -1,3 +1,8 @@
 [build-system]
-requires = ["setuptools", "numpy==2.3.5", "cython"]
+requires = [
+    "setuptools",
+    "numpy==2.3.5",
+    "cython",
+    "cuda-toolkit[cudart,nvcc,nvjpeg]==12.4.1"
+]
 build-backend = "setuptools.build_meta"

setup.py

@@ -1,11 +1,45 @@
+import sysconfig
+import os
+
 from setuptools import setup, Extension
 import numpy as np
 
+
+
 def main() -> None:
+    base_nvidia_packages_dir = f'{sysconfig.get_paths()["purelib"]}/nvidia'
+
     setup(ext_modules=[
         Extension(name='app.fast',
-                  sources=['src/cpp/fast/fast.cpp'],
-                  include_dirs=[np.get_include()],
+                  sources=[
+                      'src/cpp/fast/fast.cpp'
+                  ],
+                  include_dirs=[
+                      np.get_include(),
+                      f'{base_nvidia_packages_dir}/nvjpeg/include',
+                      f'{base_nvidia_packages_dir}/cuda_runtime/include',
+                      f'{base_nvidia_packages_dir}/cuda_nvcc/include'
+                  ],
+                  library_dirs=[
+                      f'{base_nvidia_packages_dir}/nvjpeg/lib',
+                      f'{base_nvidia_packages_dir}/cuda_runtime/lib',
+                      f'{base_nvidia_packages_dir}/cuda_nvcc/lib'
+                  ],
+                  extra_link_args=[
+                      f'{base_nvidia_packages_dir}/cuda_runtime/lib/libcudart.so.12',
+                      f'{base_nvidia_packages_dir}/nvjpeg/lib/libnvjpeg.so.12',
+
+                  ],
+                  runtime_library_dirs=[
+                      f'{base_nvidia_packages_dir}/nvjpeg/lib',
+                      f'{base_nvidia_packages_dir}/cuda_runtime/lib',
+                      f'{base_nvidia_packages_dir}/cuda_nvcc/lib'
+                  ],
+                  extra_compile_args=[
+                      '-Wall',
+                      '-Wextra',
+                      '-Wno-unused-parameter'
+                  ],
                   language='c++'),
     ])