sticker-convert 2.2.7__py3-none-any.whl → 2.3.1__py3-none-any.whl

sticker_convert/__init__.py

@@ -1,3 +1,3 @@
 #!/usr/bin/env python3
 '''sticker-convert'''
-__version__ = '2.2.7'
+__version__ = '2.3.1'

sticker_convert/converter.py

@@ -30,6 +30,14 @@ def get_step_value(
     else:
         return None
 
+def useful_array(plane, bytes_per_pixel=1, dtype='uint8'):
+    total_line_size = abs(plane.line_size)
+    useful_line_size = plane.width * bytes_per_pixel
+    arr = np.frombuffer(plane, np.uint8)
+    if total_line_size != useful_line_size:
+        arr = arr.reshape(-1, total_line_size)[:, 0:useful_line_size].reshape(-1)
+    return arr.view(np.dtype(dtype))
+
 class StickerConvert:
     MSG_START_COMP = '[I] Start compressing {} -> {}'
     MSG_SKIP_COMP = '[S] Compatible file found, skip compress and just copy {} -> {}'
@@ -225,30 +233,69 @@ class StickerConvert:
                 self.frames_raw.append(frame)
             return
         
-        frame_format = 'rgba'
         # Crashes when handling some webm in yuv420p and convert to rgba
         # https://github.com/PyAV-Org/PyAV/issues/1166
         metadata = iio.immeta(self.in_f, plugin='pyav', exclude_applied=False)
         context = None
         if metadata.get('video_format') == 'yuv420p':
-            if metadata.get('alpha_mode') != '1':
-                frame_format = 'rgb24'
             if metadata.get('codec') == 'vp8':
-                context = CodecContext.create('v8', 'r')
+                context = CodecContext.create('libvpx', 'r')
             elif metadata.get('codec') == 'vp9':
                 context = CodecContext.create('libvpx-vp9', 'r')
         
         with av.open(self.in_f) as container:
             if not context:
                 context = container.streams.video[0].codec_context
-            for packet in container.demux(video=0):
+
+            for packet in container.demux(container.streams.video):
                 for frame in context.decode(packet):
-                    frame = frame.to_ndarray(format=frame_format)
-                    if frame_format == 'rgb24':
-                        frame = np.dstack(
-                            (frame, np.zeros(frame.shape[:2], dtype=np.uint8)+255)
-                            )
-                    self.frames_raw.append(frame)
+                    if frame.width % 2 != 0:
+                        width = frame.width - 1
+                    else:
+                        width = frame.width
+                    if frame.height % 2 != 0:
+                        height = frame.height - 1
+                    else:
+                        height = frame.height
+                    if frame.format.name == 'yuv420p':
+                        rgb_array = frame.to_ndarray(format='rgb24')
+                        rgba_array = np.dstack(
+                            (rgb_array, np.zeros(frame.shape[:2], dtype=np.uint8) + 255)
+                        )
+                    else:
+                        # yuva420p may cause crash
+                        # https://github.com/laggykiller/sticker-convert/issues/114
+                        frame = frame.reformat(width=width, height=height, format='yuva420p', dst_colorspace=1)
+
+                        # https://stackoverflow.com/questions/72308308/converting-yuv-to-rgb-in-python-coefficients-work-with-array-dont-work-with-n
+                        y = useful_array(frame.planes[0]).reshape(height, width)
+                        u = useful_array(frame.planes[1]).reshape(height // 2, width // 2)
+                        v = useful_array(frame.planes[2]).reshape(height // 2, width // 2)
+                        a = useful_array(frame.planes[3]).reshape(height, width)
+
+                        u = u.repeat(2, axis=0).repeat(2, axis=1)
+                        v = v.repeat(2, axis=0).repeat(2, axis=1)
+
+                        y = y.reshape((y.shape[0], y.shape[1], 1))
+                        u = u.reshape((u.shape[0], u.shape[1], 1))
+                        v = v.reshape((v.shape[0], v.shape[1], 1))
+                        a = a.reshape((a.shape[0], a.shape[1], 1))
+
+                        yuv_array = np.concatenate((y, u, v), axis=2)
+
+                        yuv_array = yuv_array.astype(np.float32)
+                        yuv_array[:, :, 0] = yuv_array[:, :, 0].clip(16, 235).astype(yuv_array.dtype) - 16
+                        yuv_array[:, :, 1:] = yuv_array[:, :, 1:].clip(16, 240).astype(yuv_array.dtype) - 128
+
+                        convert = np.array([
+                            [1.164,  0.000,  1.793],
+                            [1.164, -0.213, -0.533],
+                            [1.164,  2.112,  0.000]
+                        ])
+                        rgb_array = np.matmul(yuv_array, convert.T).clip(0,255).astype('uint8')
+                        rgba_array = np.concatenate((rgb_array, a), axis=2)
+
+                        self.frames_raw.append(rgba_array)
 
     def frames_import_lottie(self):
         if self.in_f_ext == '.tgs':
@@ -313,11 +360,11 @@ class StickerConvert:
         # fps_ratio: 1 frame in new anim equal to how many frame in old anim
         # speed_ratio: How much to speed up / slow down
         fps_ratio = self.fps_orig / self.fps
-        if (self.opt_comp.duration_min and
+        if (self.opt_comp.duration_min != None and
             self.duration_orig < self.opt_comp.duration_min):
 
             speed_ratio = self.duration_orig / self.opt_comp.duration_min
-        elif (self.opt_comp.duration_max and
+        elif (self.opt_comp.duration_max != None and
               self.duration_orig > self.opt_comp.duration_max):
             
             speed_ratio = self.duration_orig / self.opt_comp.duration_max

sticker_convert/gui.py

@@ -543,11 +543,18 @@ class GUI(Window):
         # output_option_display = self.get_output_display_name()
         url = self.input_address_var.get()
 
-        # Input
-        if os.path.isdir(self.input_setdir_var.get()):
-            self.input_frame.input_setdir_entry.config(bootstyle='default')
+        if os.path.abspath(self.input_setdir_var.get()) == os.path.abspath(self.output_setdir_var.get()):
+            in_out_dir_same = True
         else:
+            in_out_dir_same = False
+
+        # Input
+        if in_out_dir_same == True:
+            self.input_frame.input_setdir_entry.config(bootstyle='danger')
+        elif not os.path.isdir(self.input_setdir_var.get()):
             self.input_frame.input_setdir_entry.config(bootstyle='warning')
+        else:
+            self.input_frame.input_setdir_entry.config(bootstyle='default')
 
         self.input_frame.address_lbl.config(text=self.input_presets[input_option_display]['address_lbls'])
         self.input_frame.address_entry.config(bootstyle='default')
@@ -574,10 +581,12 @@ class GUI(Window):
                 self.input_frame.address_tip.config(text=f'Detected URL: {download_option}')
 
         # Output
-        if os.path.isdir(self.output_setdir_var.get()):
-            self.output_frame.output_setdir_entry.config(bootstyle='default')
-        else:
+        if in_out_dir_same == True:
+            self.output_frame.output_setdir_entry.config(bootstyle='danger')
+        elif not os.path.isdir(self.output_setdir_var.get()):
             self.output_frame.output_setdir_entry.config(bootstyle='warning')
+        else:
+            self.output_frame.output_setdir_entry.config(bootstyle='default')
 
         if (MetadataHandler.check_metadata_required(output_option, 'title') and
             not MetadataHandler.check_metadata_provided(self.input_setdir_var.get(), input_option, 'title') and

sticker_convert/job.py

@@ -89,6 +89,10 @@ class Job:
         save_to_local_tip += '    If you want to upload the results by yourself,\n'
         save_to_local_tip += '    select "Save to local directory only" for output\n'
 
+        if os.path.abspath(self.opt_input.dir) == os.path.abspath(self.opt_output.dir):
+            error_msg += '\n'
+            error_msg += '[X] Input and output directories cannot be the same\n'
+
         if self.opt_input.option == 'auto':
             error_msg += '\n'
             error_msg += '[X] Unrecognized URL input source\n'

sticker_convert/utils/media/codec_info.py

@@ -5,6 +5,8 @@ import mimetypes
 from typing import Optional
 
 import imageio.v3 as iio
+import av # type: ignore
+from av.codec.context import CodecContext # type: ignore
 from rlottie_python import LottieAnimation  # type: ignore
 from PIL import Image, UnidentifiedImageError
 import mmap
@@ -66,8 +68,31 @@ class CodecInfo:
                 else:
                     fps = frames / total_duration * 1000
             else:
-                metadata = iio.immeta(file, plugin="pyav", exclude_applied=False)
-                fps = metadata.get("fps", 1)
+                # Getting fps from metadata is not reliable
+                # Example: https://github.com/laggykiller/sticker-convert/issues/114
+                metadata = iio.immeta(file, plugin='pyav', exclude_applied=False)
+                context = None
+                if metadata.get('video_format') == 'yuv420p':
+                    if metadata.get('codec') == 'vp8':
+                        context = CodecContext.create('libvpx', 'r')
+                    elif metadata.get('codec') == 'vp9':
+                        context = CodecContext.create('libvpx-vp9', 'r')
+
+                with av.open(file) as container:
+                    stream = container.streams.video[0]
+                    if not context:
+                        context = stream.codec_context
+
+                    last_frame = None
+                    for frame_count, frame in enumerate(container.decode(stream)):
+                        last_frame = frame
+                        if frame_count > 10:
+                            break
+
+                    if frame_count <= 1:
+                        fps = 1
+                    else:
+                        fps = frame_count / (last_frame.pts * last_frame.time_base.numerator / last_frame.time_base.denominator)
 
         return fps
 

{sticker_convert-2.2.7.dist-info → sticker_convert-2.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sticker-convert
-Version: 2.2.7
+Version: 2.3.1
 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>
@@ -376,7 +376,7 @@ Requires-Dist: pyoxipng ~=9.0.0
 Requires-Dist: python-telegram-bot ~=20.5
 Requires-Dist: requests ~=2.31.0
 Requires-Dist: rlottie-python ~=1.2.1
-Requires-Dist: selenium ~=4.16.0
+Requires-Dist: selenium ~=4.17.2
 Requires-Dist: signalstickers-client ~=3.3.0
 Requires-Dist: tqdm ~=4.66.1
 Requires-Dist: ttkbootstrap-fork-laggykiller ~=1.5.1

{sticker_convert-2.2.7.dist-info → sticker_convert-2.3.1.dist-info}/RECORD

@@ -1,9 +1,9 @@
-sticker_convert/__init__.py,sha256=rdbO3DFWe9RYZpMQSefq0PJivnUxBt4pCyMb9I7vgCs,66
+sticker_convert/__init__.py,sha256=fl39fVB4JgtiBbT7-wCNoXpljUvI7OAZBy3q531bmn4,66
 sticker_convert/__main__.py,sha256=HHzFzRsxxhxkCL15_sE1rBAdtprT0NfVUNR591BMvl4,628
 sticker_convert/cli.py,sha256=mDU18LdLD2yr1jvzuE8R9IpESd9ZiMTU1_DlcnY6WWw,16631
-sticker_convert/converter.py,sha256=BCbrTYqJa1wjORWrPqW3pwl5CTauXUBw0f5Se-OBP1M,17343
-sticker_convert/gui.py,sha256=Li1i0bo0mZEOdd8udzWi9Wh-08xpgECwvejMGugSBGc,27591
-sticker_convert/job.py,sha256=VEyEW3KV040PbzfPCxUpLU9Im8RVv_hQqGUVBDRz87k,17440
+sticker_convert/converter.py,sha256=nK_gcUmxR4n61BUOlEDPDXNkNF8kEJaar5NIzPYflWQ,19905
+sticker_convert/gui.py,sha256=vR1tkqMSkxkMt-GD-V5i8DYSQ-HUjkwgKl1y0iveE9g,28019
+sticker_convert/job.py,sha256=gwqHy7k6fmhimmYnXYWCNg3IJJpvPcMUteu7otljeiY,17640
 sticker_convert/job_option.py,sha256=MdBhyp4-cIHxno4SwBX2n45lkq3dK8Epy84nxvrT_PQ,10926
 sticker_convert/downloaders/download_base.py,sha256=TYC6bod0IHUApPBLltsOg9zUOMJRScUp7RMwUG8nF78,2572
 sticker_convert/downloaders/download_kakao.py,sha256=6NKUT5iHyvVX7og8kSyk80TXUM0upUQMnnDbPFCK0NQ,8034
@@ -81,12 +81,12 @@ sticker_convert/utils/files/json_manager.py,sha256=BBndjSY5kvjxurOpnDI2DRsKoLrNB
 sticker_convert/utils/files/metadata_handler.py,sha256=KIy8j89DMBW7-gpL9jqhzz96GgBvPHsRYegHXemk_mY,8346
 sticker_convert/utils/files/run_bin.py,sha256=4c8y5c_2wUfm0hbJSOGFcezpW5BVqM5_e9aliPbxvwo,1632
 sticker_convert/utils/media/apple_png_normalize.py,sha256=YGrZ8pZvgfhw8dW-0gf7XcXk_R82lJTX-oTo6SsB7uY,3719
-sticker_convert/utils/media/codec_info.py,sha256=R0t8Zh3OXxM6EMukhzF7aE3aI4yykT-7I9GTeW9Uw48,4862
+sticker_convert/utils/media/codec_info.py,sha256=vla8jbnx0w1Gp4UWj1n95qLVct8l0Th70mAEMFr0LQM,6061
 sticker_convert/utils/media/decrypt_kakao.py,sha256=gMz64vIGEIPAl4FFWuUbPMHE7vExr46ZFpzMoukvwws,1918
 sticker_convert/utils/media/format_verify.py,sha256=7WvvDSFDkyG1WlkXKqVbmPjKiyEedtqivZcCqSNziD8,6179
-sticker_convert-2.2.7.dist-info/LICENSE,sha256=gXf5dRMhNSbfLPYYTY_5hsZ1r7UU1OaKQEAQUhuIBkM,18092
-sticker_convert-2.2.7.dist-info/METADATA,sha256=aandEqGO9mWN2_CJwFkIKj2DvTnaVw0ry1DvvyqVR8o,45184
-sticker_convert-2.2.7.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-sticker_convert-2.2.7.dist-info/entry_points.txt,sha256=MNJ7XyC--ugxi5jS1nzjDLGnxCyLuaGdsVLnJhDHCqs,66
-sticker_convert-2.2.7.dist-info/top_level.txt,sha256=r9vfnB0l1ZnH5pTH5RvkobnK3Ow9m0RsncaOMAtiAtk,16
-sticker_convert-2.2.7.dist-info/RECORD,,
+sticker_convert-2.3.1.dist-info/LICENSE,sha256=gXf5dRMhNSbfLPYYTY_5hsZ1r7UU1OaKQEAQUhuIBkM,18092
+sticker_convert-2.3.1.dist-info/METADATA,sha256=tNM3JESS5RftQwEVQ4HJ0cCmwu_WBzyy2svIXDC5CEA,45184
+sticker_convert-2.3.1.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+sticker_convert-2.3.1.dist-info/entry_points.txt,sha256=MNJ7XyC--ugxi5jS1nzjDLGnxCyLuaGdsVLnJhDHCqs,66
+sticker_convert-2.3.1.dist-info/top_level.txt,sha256=r9vfnB0l1ZnH5pTH5RvkobnK3Ow9m0RsncaOMAtiAtk,16
+sticker_convert-2.3.1.dist-info/RECORD,,