module ptwm.codecs

class Codec(Protocol):

def decode(self, blob: bytes | memoryview, original_len: int) -> bytes

def encode(self, data: bytes | memoryview) -> bytes

name: str

class CodecId(IntEnum):

ArithmeticO0 = 18

ArithmeticO0Adaptive = 19

ArithmeticO1 = 20

ContextMixingLite = 21

Fpc = 6

HuffLlm5Bit = 22

Huffman = 1

Identity = 0

NeuralPredictor = 24

Order1Arithmetic = 23

Order1ScaleAC = 17

PerGroupCodebook = 16

Rans = 3

Tans = 7

Zstd = 4

ZstdDict = 5

def decode_planes(blobs: Sequence[bytes | memoryview], original_lens: Sequence[int], codec: Codec | str) -> list[bytes]

def encode_planes(planes: Sequence[Plane | bytes | memoryview], codec: Codec | str) -> list[bytes]

def get_codec(name: str) -> Codec

class HuffmanCodec:

def __init__(self, name: str = 'huffman') -> None

def decode(self, blob: bytes | memoryview, original_len: int) -> bytes

def encode(self, data: bytes | memoryview) -> bytes

name: str = 'huffman'

class IdentityCodec:

def __init__(self, name: str = 'identity') -> None

def decode(self, blob: bytes | memoryview, original_len: int) -> bytes

def encode(self, data: bytes | memoryview) -> bytes

name: str = 'identity'

class RansCodec:

def __init__(self, name: str = 'rans') -> None

def decode(self, blob: bytes | memoryview, original_len: int) -> bytes

def encode(self, data: bytes | memoryview) -> bytes

name: str = 'rans'

class ZstdCodec:

def __init__(self, level: int = 3) -> None

def decode(self, blob: bytes | memoryview, original_len: int) -> bytes

def encode(self, data: bytes | memoryview) -> bytes

level: int = 3

name: str