9j)SSKrSSKrSSKrSSKrSSKrSSKJrJrJr SSK J r SSK J r SSK JrJrJrJrJrJrJr \R*"\5r"SS5r\"5rS\S \4S jrS \\\ 4S \ 4S jr"S S5r"SS\5rg)N)heapifyheappopheappush)IOBase)Path)AnyCallableIterableIteratorMappingOptionalUnionc^\rSrSrSrS\4SjrS\4SjrS\4Sjr S\4Sjr S\ 4Sjr S r g ) _EndOfFileSymbolz Internal class for "end of file" symbol to be able to detect the end of the encoded bit stream, which does not necessarily align with byte boundaries. returncg)N_EOFselfs P/home/wildlama/miniconda3/lib/python3.13/site-packages/dahuffman/huffmancodec.py__repr___EndOfFileSymbol.__repr__scg)NTrrothers r__lt___EndOfFileSymbol.__lt__srcg)NFrrs r__gt___EndOfFileSymbol.__gt__src4URUR:H$N) __class__rs r__eq___EndOfFileSymbol.__eq__!s$..00rc,[UR5$r%)hashr&rs r__hash___EndOfFileSymbol.__hash__$sDNN##rrN)__name__ __module__ __qualname____firstlineno____doc__strrboolrr"r'intr+__static_attributes__rrrrrsC # tt1t1$#$rrdatarc[S5SR[S5[0R[U5[5$)z' Guess concat function from given data r)typejoinbytesgetlist)r6s r _guess_concatr>0s6 R"'' S 5  c$t*drpathc[U5nUR5(dURSS9 UR5(deU$)NT)parents)rexistsmkdiris_dir)r?s r ensure_dirrE:s9 :D ;;== 4  ;;=== Krc ^\rSrSrSr\S\4S\S\S\ 4Sjjr S\4S jr \ R4S \SS 4S jjrS \\\\4S\4SjrS \\\\4S\\4SjrSS \\\\4S\\S\\\\44SjjrS \\\\4S\4SjrSS\\\4S\SS 4Sjjr\S\\\4SS4Sj5rSr g ) PrefixCodecBz, Prefix code codec, using given code table. T code_tableconcatcheckcXlX lX@lU(aQ[UR[5(a/[ SURR 555(degg)a Initialize codec with given code table. :param code_table: mapping of symbol to code tuple (bitsize, value) :param concat: function to concatenate symbols :param check: whether to check the code table :param eof: "end of file" symbol (customizable for advanced usage) c3# UHFup[U[5=(a( US:=(a [U[5=(a US:v MH g7f)rN) isinstancer4).0bvs r 'PrefixCodec.__init__..WsA92FQ1c"OqAvO*Q2DOaO2sAAN)_table_concat_eofrOdictallvalues)rrIrJrKeofs r__init__PrefixCodec.__init__Gsa!   dkk400S9"kk002966 6 rrcUR$)zR Get code table :return: dictionary mapping symbol to code tuple (bitsize, value) )rUrs rget_code_tablePrefixCodec.get_code_table_s {{routNc [[[R"S/SURR 55565n[ SU55nSUSS-n[U6H!nURUR"U65 M# g)z Print code table overview )BitsCodeValueSymbolc3# UHGunup#[U5[U5SSRUS5[U5[U54v MI g7f)N0)r2binrjustrepr)rPsymbolbitsvals rrS/PrefixCodec.print_code_table..psU4G/FKT  IHQRL..tS9H L  4GsAAc3F# UHn[SU55v M g7f)c38# UHn[U5v M g7fr%)len)rPss rrS9PrefixCodec.print_code_table...}s/3a3q663sN)max)rPcols rrSrp}sC7Cs/3///7s!z{0:>%d} {1:%d} {2:>%d} {3} N) r=zip itertoolschainrUitemstuplewriteformat)rracolumnswidthstemplaterows rprint_code_tablePrefixCodec.print_code_tablefs  8948;;3D3D3F   "C7CC1F2AJ>=C IIhoos+ ,!rr6c6[URU55$)z~ Encode given data. :param data: sequence of symbols (e.g. byte string, unicode string, list, iterator) :return: byte string )r;encode_streaming)rr6s rencodePrefixCodec.encodesT**4011rc#@# SnSnUHInURUupVX%-U-nX5- nUS:dM'X#S- - nUv X'US- -- nUS-nUS:aM"MK US:aAURURupVX%-U-nX5- nUS:aX#S- - nOUSU- -nUv gg7f)z Encode given data in streaming fashion. :param data: sequence of symbols (e.g. byte string, unicode string, list, iterator) :return: generator of bytes rN)rUrW)rr6buffersizertrQrRbytes rrPrefixCodec.encode_streamingsA;;q>DAkQ&F ID!)(+ D1H#56  !) & !8;;tyy)DAkQ&F IDqy(+!d(+J s*B BA BcVU=(d UR"URU55$)z Decode given data. :param data: sequence of bytes (string, list or generator of bytes) :param concat: optional override of function to concatenate the decoded symbols :return: )rVdecode_streaming)rr6rJs rdecodePrefixCodec.decodes&&$,,&(=(=d(CDDrc#2# URR5VVVs0sH unup4X44U_M nnnnSnSnUHMnSHDn US-[X-5-nUS- nXv4U;dM$XWU4n XR:Xa gU v SnSnMF MO gs snnnf7f)z Decode given data in streaming fashion :param data: sequence of bytes (string, list or generator of bytes) :return: generator of symbols r)@ rrhrNrNN)rUr|r3rW) rr6rtrQrRlookuprrrmrms rrPrefixCodec.decode_streamings.2[[->->-@A-@ 6A1&!)-@AD2 A+dh7 >V+#&L1F* LFD3 BsBB/B'0Br?metadatac UR5nU[U5URS.nU(aX$S'[U5n[ UR 5 UR SS9n[R"XES9 SSS5 [RSR[U5R[U5[U5S95 g!,(df  NX=f) zv Persist the code table to a file. :param path: file path to persist to :param metadata: additional metadata :return: )rIr9rJrwbmode)fileNz)Saved {c} code table ({l} items) to {p!r}clp)r_r9rVrrEparentopenpickledump_loginforr-rsr2)rr?rrIr6fs rsavePrefixCodec.saves((* $Jll  ' Dz4;; YYDY !Q KK %" 7 > >t*%%ZCI ?  " !s $C  Cc j[U5nURSS9n[R"U5nSSS5 WSn[ U[ 5(deUSn[ RSRUR[U5[U5S95 U"XBSS 9$!,(df  Nx=f) zd Load a persisted PrefixCodec :param path: path to serialized PrefixCodec code table data. :return: rbrNr9rIz0Loading {c} with {l} code table items from {p!r}rrJrJ) rrrload issubclassrGrrrr-rsr2)r?rr6clsrIs rrPrefixCodec.loadsDz YYDY !Q;;q>D"6l#{++++,'  > E E,,#j/SY F  :8n55" !s B$$ B2)rVrWrUr%)!r-r.r/r0r1r=rrXr r3r\r_sysstdoutrrrr2r;r rr r4rr rrrrr staticmethodrr5rrrrGrGBsO 48tQU(0@D0.1ZZ-F-D-825eX!56252%U3x+?%@%Xc]%POS E%#./ E9A(9K E sE8# $ EU5(3-+?%@X0 sDy) S D 465d#6 66rrGcx\rSrSrSr\S\4S\\\ 4S\ \ SS4Sjj5r \S\\ \\4SS4S j5rS rg) HuffmanCodecizz Huffman coder, with code table built from given symbol frequencies or raw data, providing encoding and decoding methods. N frequenciesrJrc U=(d [[[U555nUR5VVs/sH upEXTS4/4PM nnnX1;aUR SUS4/45 [ U5 [ U5S:a[U5n[U5nUSUS-USVV V s/sHunupXIS-U 44PM sn n nUSVV V s/sHunupXIS-SU -U -44PM sn n n-4n [Xk5 [ U5S:aM[[U5S5n U"XSUS9$s snnfs sn n nfs sn n nf)z Build Huffman code table from given symbol frequencies :param frequencies: symbol to frequency mapping :param concat: function to concatenate symbols :param eof: "end of file" symbol (customizable for advanced usage) )rrrNrF)rJrKr[) r>nextiterr|appendrrsrrrX) rrrJr[rtrheaparQnrRmergedtables rfrom_frequenciesHuffmanCodec.from_frequencies sIA=d;.?)@A0;/@/@/BC/Btq[M"/BC  ! KKc6]O, -  $i!m A A!qt 45aD9D[a!!!eQZD9AB1F+1fqAAQ!|,-FGF T "$i!mWT]1%&5u#>>-D:FsD1-D7 D> r6c`[R"U5nURU[U5S9$)z Build Huffman code table from symbol sequence :param data: sequence of symbols (e.g. byte string, unicode string, list, iterator) :return: HuffmanCoder r) collectionsCounterrr>)rr6rs r from_dataHuffmanCodec.from_data3s/"))$/ ##K d8K#LLrr)r-r.r/r0r1 classmethodrrrXr r r rr2r;r rr5rrrrrs &*  %?4=)%?"%?  %?%?NMU3x#78M^MMrr)rrzloggingrrheapqrrriorpathlibrtypingrr r r r r r getLoggerr-rrrr>r2rErGrrrrrs ,,NNN"$$6U39%$@6@6F7M;7Mr