sticker-convert 2.7.12__py3-none-any.whl → 2.7.13__py3-none-any.whl

sticker_convert/converter.py

@@ -19,8 +19,8 @@ from sticker_convert.utils.media.codec_info import CodecInfo
 from sticker_convert.utils.media.format_verify import FormatVerify
 
 if TYPE_CHECKING:
-    from av.video.frame import VideoFrame  # type: ignore
-    from av.video.plane import VideoPlane  # type: ignore
+    from av.video.frame import VideoFrame
+    from av.video.plane import VideoPlane
 
 MSG_START_COMP = "[I] Start compressing {} -> {}"
 MSG_SKIP_COMP = "[S] Compatible file found, skip compress and just copy {} -> {}"
@@ -264,14 +264,21 @@ class StickerConvert:
                 self.result_size = self.size
                 self.result_step = step_current
 
-            if step_upper - step_lower > 1 and self.size_max:
+            if (
+                step_upper - step_lower > 0
+                and step_current != step_lower
+                and self.size_max
+            ):
                 if self.size <= self.size_max:
                     sign = "<"
                     step_upper = step_current
                 else:
                     sign = ">"
                     step_lower = step_current
-                step_current = int(rounding((step_lower + step_upper) / 2))
+                if step_current == step_lower + 1:
+                    step_current = step_lower
+                else:
+                    step_current = int(rounding((step_lower + step_upper) / 2))
                 self.recompress(sign)
             elif self.result:
                 return self.compress_done(self.result, self.result_step)
@@ -409,7 +416,7 @@ class StickerConvert:
 
         if suffix in (".tgs", ".lottie", ".json"):
             self._frames_import_lottie()
-        elif suffix in (".webp", ".apng", "png"):
+        elif suffix in (".webp", ".apng", ".png", ".gif"):
             # ffmpeg do not support webp decoding (yet)
             # ffmpeg could fail to decode apng if file is buggy
             self._frames_import_pillow()
@@ -431,6 +438,7 @@ class StickerConvert:
         from av.codec.context import CodecContext
         from av.container.input import InputContainer
         from av.video.codeccontext import VideoCodecContext
+        from av.video.frame import VideoFrame
 
         # Crashes when handling some webm in yuv420p and convert to rgba
         # https://github.com/PyAV-Org/PyAV/issues/1166
@@ -451,19 +459,31 @@ class StickerConvert:
 
             for packet in container.demux(container.streams.video):
                 for frame in context.decode(packet):
-                    if frame.width % 2 != 0:
-                        width = frame.width - 1
-                    else:
-                        width = frame.width
+                    width_orig = frame.width
+                    height_orig = frame.height
 
-                    if frame.height % 2 != 0:
-                        height = frame.height - 1
-                    else:
-                        height = frame.height
+                    # Need to pad frame to even dimension first
+                    if width_orig % 2 == 1 or height_orig % 2 == 1:
+                        from av.filter import Graph
+
+                        width_new = width_orig + width_orig % 2
+                        height_new = height_orig + height_orig % 2
+
+                        graph = Graph()
+                        in_src = graph.add_buffer(template=container.streams.video[0])
+                        pad = graph.add(
+                            "pad", f"{width_new}:{height_new}:0:0:color=#00000000"
+                        )
+                        in_src.link_to(pad)
+                        sink = graph.add("buffersink")
+                        pad.link_to(sink)
+                        graph.configure()
+
+                        graph.push(frame)
+                        frame = cast(VideoFrame, graph.pull())
 
                     if frame.format.name == "yuv420p":
                         rgb_array = frame.to_ndarray(format="rgb24")
-                        cast("np.ndarray[Any, Any]", rgb_array)
                         rgba_array = np.dstack(
                             (
                                 rgb_array,
@@ -475,13 +495,13 @@ class StickerConvert:
                         # Not safe to directly call frame.to_ndarray(format="rgba")
                         # https://github.com/laggykiller/sticker-convert/issues/114
                         frame = frame.reformat(
-                            width=width,
-                            height=height,
                             format="yuva420p",
                             dst_colorspace=1,
                         )
                         rgba_array = yuva_to_rgba(frame)
 
+                    # Remove pixels that was added to make dimensions even
+                    rgba_array = rgba_array[0:width_orig, 0:height_orig]
                     self.frames_raw.append(rgba_array)
 
     def _frames_import_lottie(self) -> None:
@@ -727,10 +747,10 @@ class StickerConvert:
         if 0 in alpha:
             extra_kwargs["transparency"] = 0
             extra_kwargs["disposal"] = 2
-            im_out = [self.quantize(Image.fromarray(i)) for i in frames_processed]
+            im_out = [self.quantize(Image.fromarray(i)) for i in frames_processed]  # type: ignore
         else:
             im_out = [
-                self.quantize(Image.fromarray(i).convert("RGB")).convert("RGB")
+                self.quantize(Image.fromarray(i).convert("RGB")).convert("RGB")  # type: ignore
                 for i in frames_processed
             ]
 

sticker_convert/version.py

@@ -1,3 +1,3 @@
 #!/usr/bin/env python3
 
-__version__ = "2.7.12"
+__version__ = "2.7.13"

{sticker_convert-2.7.12.dist-info → sticker_convert-2.7.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sticker-convert
-Version: 2.7.12
+Version: 2.7.13
 Summary: Convert (animated) stickers to/from WhatsApp, Telegram, Signal, Line, Kakao, iMessage. Written in Python.
 Author-email: laggykiller <chaudominic2@gmail.com>
 Maintainer-email: laggykiller <chaudominic2@gmail.com>

{sticker_convert-2.7.12.dist-info → sticker_convert-2.7.13.dist-info}/RECORD

@@ -1,12 +1,12 @@
 sticker_convert/__init__.py,sha256=iQnv6UOOA69c3soAn7ZOnAIubTIQSUxtq1Uhh8xRWvU,102
 sticker_convert/__main__.py,sha256=6RJauR-SCSSTT3TU7FFB6B6PVwsCxO2xZXtmZ3jc2Is,463
 sticker_convert/cli.py,sha256=kyfhvMoHq42uLZsYYLrr6b30xsNE93GVZlbGYPFjC2I,18385
-sticker_convert/converter.py,sha256=-BxUT-VCw-VbfbG7ZZxNTI4FIJcU3Pq2rzw0VuitiVI,32904
+sticker_convert/converter.py,sha256=XqJyHPWaZCZEbUK_-0CUYmumeUzgOS4lg7zTPH9JYA8,33821
 sticker_convert/definitions.py,sha256=ZhP2ALCEud-w9ZZD4c3TDG9eHGPZyaAL7zPUsJAbjtE,2073
 sticker_convert/gui.py,sha256=TqdgFbHBRYgcXWWrsfxLUJ8Zu9WeE11vYyZMX6nalik,30599
 sticker_convert/job.py,sha256=J4e7dZ48t5EhM3fG-xF1BEQ10WYljx3wWamKXAJjCa8,26000
 sticker_convert/job_option.py,sha256=1YVhyTfu2cWz3qpAKbdIM11jbL0CJz0ksOYAeg8v6dc,7649
-sticker_convert/version.py,sha256=itLv7GvtbPZ961xSemMjepArEGEJR3yg6PpGH0MuovQ,47
+sticker_convert/version.py,sha256=jOLhRKwMOcmLYOtzpH73qVOpP1okliTWPuuIgfNT8xw,47
 sticker_convert/downloaders/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sticker_convert/downloaders/download_base.py,sha256=5R7c8kwahAflOOYrtSQPnBVQ4T-DsprPUMP7G9wcJX4,2824
 sticker_convert/downloaders/download_kakao.py,sha256=RYrebTxEjKjXAr9xw18r0dMW0dFjSqZxzi8dpaq56TY,8581
@@ -92,9 +92,9 @@ sticker_convert/utils/media/apple_png_normalize.py,sha256=LbrQhc7LlYX4I9ek4XJsZE
 sticker_convert/utils/media/codec_info.py,sha256=JHHlfCLSpepYFv6TSL1XqwUJoVOpOPFxurRtduplPGY,12856
 sticker_convert/utils/media/decrypt_kakao.py,sha256=4wq9ZDRnFkx1WmFZnyEogBofiLGsWQM_X69HlA36578,1947
 sticker_convert/utils/media/format_verify.py,sha256=Xf94jyqk_6M9IlFGMy0wYIgQKn_yg00nD4XW0CgAbew,5732
-sticker_convert-2.7.12.dist-info/LICENSE,sha256=gXf5dRMhNSbfLPYYTY_5hsZ1r7UU1OaKQEAQUhuIBkM,18092
-sticker_convert-2.7.12.dist-info/METADATA,sha256=tJ1IQvfsDyRiDDQSkHech1YhfhlB9U9HPeUEk5a15eY,49133
-sticker_convert-2.7.12.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-sticker_convert-2.7.12.dist-info/entry_points.txt,sha256=MNJ7XyC--ugxi5jS1nzjDLGnxCyLuaGdsVLnJhDHCqs,66
-sticker_convert-2.7.12.dist-info/top_level.txt,sha256=r9vfnB0l1ZnH5pTH5RvkobnK3Ow9m0RsncaOMAtiAtk,16
-sticker_convert-2.7.12.dist-info/RECORD,,
+sticker_convert-2.7.13.dist-info/LICENSE,sha256=gXf5dRMhNSbfLPYYTY_5hsZ1r7UU1OaKQEAQUhuIBkM,18092
+sticker_convert-2.7.13.dist-info/METADATA,sha256=dwERdM02NVAJqK7mBwQjAaWMipbyJVr-IRIzSSKJY98,49133
+sticker_convert-2.7.13.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+sticker_convert-2.7.13.dist-info/entry_points.txt,sha256=MNJ7XyC--ugxi5jS1nzjDLGnxCyLuaGdsVLnJhDHCqs,66
+sticker_convert-2.7.13.dist-info/top_level.txt,sha256=r9vfnB0l1ZnH5pTH5RvkobnK3Ow9m0RsncaOMAtiAtk,16
+sticker_convert-2.7.13.dist-info/RECORD,,