PyTurboJPEG 2.3.0__tar.gz → 2.4.0__tar.gz

{pyturbojpeg-2.3.0 → pyturbojpeg-2.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PyTurboJPEG
-Version: 2.3.0
+Version: 2.4.0
 Summary: A Python wrapper of libjpeg-turbo for decoding and encoding JPEG image.
 Home-page: https://github.com/lilohuang/PyTurboJPEG
 Author: Lilo Huang
@@ -179,6 +179,13 @@ with open('input.jpg', 'rb') as f:
     cropped_data = jpeg.crop(f.read(), 8, 8, 320, 240)
 with open('cropped_output.jpg', 'wb') as f:
     f.write(cropped_data)
+
+# Lossless Huffman table optimization (re-encodes with optimal tables,
+# identical pixels; typically smaller unless already optimized)
+with open('input.jpg', 'rb') as f:
+    optimized_data = jpeg.optimize(f.read())
+with open('optimized_output.jpg', 'wb') as f:
+    f.write(optimized_data)
 ```
 
 ### In-Place Operations

{pyturbojpeg-2.3.0 → pyturbojpeg-2.4.0}/PyTurboJPEG.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PyTurboJPEG
-Version: 2.3.0
+Version: 2.4.0
 Summary: A Python wrapper of libjpeg-turbo for decoding and encoding JPEG image.
 Home-page: https://github.com/lilohuang/PyTurboJPEG
 Author: Lilo Huang
@@ -179,6 +179,13 @@ with open('input.jpg', 'rb') as f:
     cropped_data = jpeg.crop(f.read(), 8, 8, 320, 240)
 with open('cropped_output.jpg', 'wb') as f:
     f.write(cropped_data)
+
+# Lossless Huffman table optimization (re-encodes with optimal tables,
+# identical pixels; typically smaller unless already optimized)
+with open('input.jpg', 'rb') as f:
+    optimized_data = jpeg.optimize(f.read())
+with open('optimized_output.jpg', 'wb') as f:
+    f.write(optimized_data)
 ```
 
 ### In-Place Operations

{pyturbojpeg-2.3.0 → pyturbojpeg-2.4.0}/README.md

@@ -155,6 +155,13 @@ with open('input.jpg', 'rb') as f:
     cropped_data = jpeg.crop(f.read(), 8, 8, 320, 240)
 with open('cropped_output.jpg', 'wb') as f:
     f.write(cropped_data)
+
+# Lossless Huffman table optimization (re-encodes with optimal tables,
+# identical pixels; typically smaller unless already optimized)
+with open('input.jpg', 'rb') as f:
+    optimized_data = jpeg.optimize(f.read())
+with open('optimized_output.jpg', 'wb') as f:
+    f.write(optimized_data)
 ```
 
 ### In-Place Operations

{pyturbojpeg-2.3.0 → pyturbojpeg-2.4.0}/setup.py

@@ -2,7 +2,7 @@ import io
 from setuptools import setup, find_packages
 setup(
     name='PyTurboJPEG',
-    version='2.3.0',
+    version='2.4.0',
     description='A Python wrapper of libjpeg-turbo for decoding and encoding JPEG image.',
     author='Lilo Huang',
     author_email='kuso.cc@gmail.com',

{pyturbojpeg-2.3.0 → pyturbojpeg-2.4.0}/tests/test_turbojpeg.py

@@ -513,6 +513,17 @@ class TestCropMultiple:
         assert subsample == TJSAMP_GRAY
 
 
+class TestOptimize:
+    """Test optimize function."""
+
+    def test_optimize_is_lossless(self, jpeg_instance, encoded_sample_jpeg):
+        """Test optimization returns a valid JPEG with unchanged pixels."""
+        optimized = jpeg_instance.optimize(encoded_sample_jpeg)
+        original_pixels = jpeg_instance.decode(encoded_sample_jpeg)
+        optimized_pixels = jpeg_instance.decode(optimized)
+        assert np.array_equal(original_pixels, optimized_pixels)
+
+
 class TestBufferSize:
     """Test buffer_size function."""
     

{pyturbojpeg-2.3.0 → pyturbojpeg-2.4.0}/turbojpeg.py

@@ -23,7 +23,7 @@
 # SOFTWARE.
 
 __author__ = 'Lilo Huang <kuso.cc@gmail.com>'
-__version__ = '2.3.0'
+__version__ = '2.4.0'
 
 from ctypes import *
 from ctypes.util import find_library
@@ -125,6 +125,7 @@ TJXOPT_GRAY = 8
 TJXOPT_NOOUTPUT = 16
 TJXOPT_PROGRESSIVE = 32
 TJXOPT_COPYNONE = 64
+TJXOPT_OPTIMIZE = 256  # Huffman table optimization
 
 # pixel size
 # see details in https://github.com/libjpeg-turbo/libjpeg-turbo/blob/main/src/turbojpeg.h
@@ -1351,6 +1352,43 @@ class TurboJPEG(object):
         finally:
             self.__destroy(handle)
 
+    def optimize(self, jpeg_buf, copynone=False):
+        """Losslessly optimize the Huffman tables of a jpeg image.
+
+        Re-encodes the entropy-coded data with optimal Huffman tables
+        (equivalent to ``jpegtran -optimize``) without any loss in image
+        quality. Typically reduces file size, unless the input is already
+        optimized.
+
+        Parameters
+        ----------
+        jpeg_buf: bytes
+            Input jpeg image.
+        copynone: bool
+            True = do not copy EXIF data (False by default)
+
+        Returns
+        ----------
+        bytes
+            Huffman-optimized jpeg image.
+        """
+        handle = self.__init(TJINIT_TRANSFORM)
+        try:
+            jpeg_array = np.frombuffer(jpeg_buf, dtype=np.uint8)
+            src_addr = self.__getaddr(jpeg_array)
+            status = self.__decompress_header(handle, src_addr, jpeg_array.size)
+            if status != 0:
+                self.__report_error(handle)
+            # Without TJXOPT_CROP the cropping region is ignored and the whole
+            # image is transformed, so the (zeroed) region needs no setup.
+            transforms = (TransformStruct * 1)()
+            transforms[0].op = TJXOP_NONE
+            transforms[0].options = TJXOPT_OPTIMIZE | (copynone and TJXOPT_COPYNONE)
+            return self.__do_transform(handle, src_addr, jpeg_array.size, 1, transforms)[0]
+
+        finally:
+            self.__destroy(handle)
+
     def buffer_size(self, img_array, jpeg_subsample=TJSAMP_422):
         """Get maximum number of bytes of compressed jpeg data"""
         height, width = img_array.shape[:2]