mir/pyrogram
2
0
Fork 0
mirror of https://github.com/pyrogram/pyrogram synced 2025-08-28 04:48:06 +00:00
Code Issues Releases Wiki Activity

Improve the RLE codec

Browse Source
This commit is contained in:
Dan 2022-03-11 12:29:05 +01:00
parent 8d7b62654c
commit d0e2235835
1 changed files with 16 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
pyrogram/file_id.py
View File

@ -63,20 +63,20 @@ def rle_encode(s: bytes) -> bytes:
Returns:
``bytes``: The encoded bytes
"""
r = b""
n = 0
r: bytes = b""
n: int = 0
for b in s:
if b == 0:
if not b:
n += 1
else:
if n > 0:
if n:
r += bytes([0, n])
n = 0
r += bytes([b])
if n > 0:
if n:
r += bytes([0, n])
return r
@ -92,17 +92,19 @@ def rle_decode(s: bytes) -> bytes:
Returns:
``bytes``: The encoded bytes
"""
r = b""
i = 0
r: bytes = b""
z: bool = False
while i < len(s):
if s[i] != 0:
r += bytes([s[i]])
for b in s:
if not b:
z = True
continue
if z:
r += b"\x00" * b
z = False
else:
r += b"\x00" * s[i + 1]
i += 1
i += 1
r += bytes([b])
return r