TeamPGM/pyrogram
2
0
Fork 0
mirror of https://github.com/TeamPGM/pyrogram.git synced 2024-11-27 16:45:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Faster RLE codec implementation (#938)

Browse Source 
* faster pyrogram lre encode implementation

* Update file_id.py

* optimized rle decode

Co-authored-by: andrew (from workstation) <andrew-ld@protonmail.com>
Co-authored-by: Dan <14043624+delivrance@users.noreply.github.com>
This commit is contained in:
andrew-ld 2022-03-28 20:10:52 +02:00 committed by GitHub
parent 190760cf0e
commit 06ee482b23
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
pyrogram/file_id.py
View File

@ -21,6 +21,7 @@ import logging
import struct
from enum import IntEnum
from io import BytesIO
from typing import List
from pyrogram.raw.core import Bytes, String
@ -63,7 +64,7 @@ def rle_encode(s: bytes) -> bytes:
Returns:
``bytes``: The encoded bytes
"""
r: bytes = b""
r: List[int] = []
n: int = 0
for b in s:
@ -71,15 +72,15 @@ def rle_encode(s: bytes) -> bytes:
n += 1
else:
if n:
r += bytes([0, n])
r.extend((0, n))
n = 0
r += bytes([b])
r.append(b)
if n:
r += bytes([0, n])
r.extend((0, n))
return r
return bytes(r)
def rle_decode(s: bytes) -> bytes:
@ -87,12 +88,12 @@ def rle_decode(s: bytes) -> bytes:
Parameters:
s (``bytes``):
Bytes to encode
Bytes to decode
Returns:
``bytes``: The encoded bytes
``bytes``: The decoded bytes
"""
r: bytes = b""
r: List[int] = []
z: bool = False
for b in s:
@ -101,12 +102,12 @@ def rle_decode(s: bytes) -> bytes:
continue
if z:
r += b"\x00" * b
r.extend((0,) * b)
z = False
else:
r += bytes([b])
r.append(b)
return r
return bytes(r)
class FileType(IntEnum):