setup

Author: zeng-qinghui

.gitignore

@@ -1,2 +1,6 @@
 out
-.vscode
+.vscode
+*.pyc
+dist
+build
+nvjpeg.egg-info

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2020-2021 Jason Dsouza
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

Makefile

@@ -8,7 +8,7 @@ CUDA_PATH=/usr/local/cuda
 endif
 
 ifndef PYTHON_VERSION
-PYTHON_VERSION=3.6
+PYTHON_VERSION=$(shell python3 -c "import sys; print('%d.%d' % (sys.version_info.major, sys.version_info.minor,))")
 endif
 
 ifndef PYTHON_BIN
@@ -39,11 +39,13 @@ out/nvjpeg-test.o: out nvjpeg-python.c
 out/nvjpeg-test: out/nvjpeg-test.o
 	gcc -o out/nvjpeg-test out/nvjpeg-test.o -L${CUDA_PATH}/lib64 -lnvjpeg -lcudart -L${PYTHON_LIB_PATH} -lpython${PYTHON_VERSION}m ${CFLAGS}
 
-out/nvjpeg.o: out nvjpeg-python.c
-	gcc -fPIC -o out/nvjpeg.o -c nvjpeg-python.c -I${CUDA_PATH}/include -I${PYTHON_INCLUDE_PATH} ${CFLAGS}
+out/nvjpeg-python.o: out nvjpeg-python.c
+	gcc -fPIC -o out/nvjpeg-python.o -c nvjpeg-python.c -I${CUDA_PATH}/include -I${PYTHON_INCLUDE_PATH} ${CFLAGS}
 
-out/${PYTHON_LIB_NAME}: out/nvjpeg.o
-	gcc --shared -fPIC -o out/${PYTHON_LIB_NAME} out/nvjpeg.o -L${CUDA_PATH}/lib64 -lnvjpeg -lcudart -L${PYTHON_LIB_PATH} -lpython${PYTHON_VERSION}m ${CFLAGS}
+out/${PYTHON_LIB_NAME}: out/nvjpeg-python.o
+	gcc --shared -fPIC -o out/${PYTHON_LIB_NAME} out/nvjpeg-python.o -L${CUDA_PATH}/lib64 -lnvjpeg -lcudart -L${PYTHON_LIB_PATH} -lpython${PYTHON_VERSION}m ${CFLAGS}
+	mkdir -p nvjpeg/lib
+	cp -f out/${PYTHON_LIB_NAME} nvjpeg/lib/${PYTHON_LIB_NAME}
 
 clean:
 	rm -Rf out

README.md

@@ -0,0 +1,47 @@
+NvJpeg - Python
+---------------------------
+
+## Require
+* nvjpeg
+* cuda >= 10.2
+* numpy >= 1.7
+* python >= 3.6
+* gcc >= 7.5
+
+## Build
+```shell
+make
+```
+
+## Install
+```shell
+make install
+```
+
+## Usage
+```python
+#!/usr/bin/env python3
+
+from nvjpeg import NvJpeg 
+
+# read file
+fp = open("input-image.jpg", "rb")
+jpegData = fp.read()
+fp.close()
+
+# decode
+nj = NvJpeg()
+img_np = nj.decode(jpegData)
+
+# use opencv show numpy image data
+cv2.imshow("Demo", img_np)
+cv2.waitKey(0)
+
+# encode numpy image data
+jpg = nj.encode(img_np)
+
+# write file
+fp = open("output-image.jpg", "wb")
+fp.write(jpg)
+fp.close()
+```

nvjpeg-python.c

@@ -234,9 +234,9 @@ int NvJpegPython_test(char* inputJpegFilePath, char* outputRawPath, char* output
 
 int main(int args, char** argv){
     NvJpegPython_test(
-        "./test/test.jpg", 
-        "./test/out/c-test.bgr",
-        "./test/out/c-test.jpg"
+        "./tests/test.jpg", 
+        "./tests/out/c-test.bgr",
+        "./tests/out/c-test.jpg"
     );
     return 0;
 }

nvjpeg/__init__.py

@@ -0,0 +1,18 @@
+import sys
+import os
+import numpy as np
+import cv2
+
+sys.path.append(os.path.join(os.path.dirname(__file__), "lib"))
+
+from nvjpeg import NvJpeg as _NvJpeg
+
+class NvJpeg:
+    def __init__():
+        self._handle = _NvJpeg()
+
+    def encode(self, numpy_array, quality=70):
+        return self._handle.encode(numpy_array, quality)
+
+    def decode(self, jpegData):
+        return self._handle.decode(jpegData)

nvjpeg/lib/.gitignore

@@ -0,0 +1 @@
+*

test/out/.gitkeep renamed to nvjpeg/lib/.gitkeep

@@ -0,0 +1,3 @@
+#!/usr/bin/env bash
+cd `cd $(dirname $0) && pwd | xargs dirname`
+make

scripts/build.sh

@@ -0,0 +1,41 @@
+import setuptools
+import sys
+import os
+
+with open("README.md", "r", encoding="utf-8") as fh:
+    long_description = fh.read()
+
+setuptools.setup(
+    name="nvjpeg",
+    version="0.0.1",
+    author="Usingnet",
+    author_email="zengqinghui@usingnet.com",
+    license="MIT",
+    description="nvjpeg for python",
+    long_description=long_description,
+    long_description_content_type="text/markdown",
+    url="https://github.com/UsingNet/nvjpeg-python",
+    packages=setuptools.find_packages(),
+    classifiers=[
+        "Development Status :: 4 - Beta"
+        "Programming Language :: Python :: 3 :: Only",
+        "License :: OSI Approved :: MIT License",
+        "Operating System :: POSIX :: Linux",
+        "Environment :: GPU :: NVIDIA CUDA :: 10.2",
+        "Environment :: GPU :: NVIDIA CUDA :: 11.0",
+    ],
+    keywords="nvjpeg jpeg encode decode",
+    python_requires=">=3.6",
+    project_urls={
+        'Source': 'https://github.com/UsingNet/nvjpeg-python',
+        'Tracker': 'https://github.com/UsingNet/nvjpeg-python/issues',
+    },
+    data_files=[
+        ('src', ['nvjpeg-python.c', 'Makefile'])
+    ],
+    scripts=[
+        'scripts/build.sh'
+    ],
+    install_requires=['make', 'gcc']
+)
+

setup.py

@@ -0,0 +1 @@
+*

test/test.py

@@ -0,0 +1,25 @@
+#!/usr/bin/env python3
+
+import sys
+import os
+import numpy as np
+import cv2
+
+sys.path.append(os.path.join(os.path.dirname(__file__), "../"))
+
+from nvjpeg import NvJpeg 
+
+fp = open(os.path.join(os.path.dirname(__file__), "test.jpg"), "rb")
+img = fp.read()
+fp.close()
+
+nj = NvJpeg()
+img_np = nj.decode(img)
+
+cv2.imshow("Demo", img_np)
+cv2.waitKey(0)
+
+jpg = nj.encode(img_np)
+fp = open(os.path.join(os.path.dirname(__file__), "out", "python-test.jpg"), "wb")
+fp.write(jpg)
+fp.close()