XjgUddlZddlZddlmZmZddlmZmZedZdZ e e d<dgZ dZ Gd d ZGd d ZGd deeZdS)N)CallableIterable)GenericTypeVarT_ENCODING_VERSIONAppendingByteSerializercReZdZd dZdeddfdZdeddfdZdeddfd Z defd Z dS) BytesWriterreturnNc8tt|_dSN) bytearrayCHECKSUM_DIGEST_SIZE_dataselfs TC:\Projects\DSCons\.venv\Lib\site-packages\torch\utils\_appending_byte_serializer.py__init__zBytesWriter.__init__s344 icf|j|ddddSNbigF byteordersigned)rextendto_bytes)rrs r write_uint64zBytesWriter.write_uint64s0 !**Q%*FFGGGGGrsc~tj|d}||dSNzutf-8)base64 b64encodeencode write_bytes)rr$payloads r write_strzBytesWriter.write_strs8"188G#4#455 !!!!!rbc~|t||j|dSr)r#lenrr!)rr-s rr*zBytesWriter.write_bytes!s8 #a&&!!! !rctj|jtdddd}t |tkrt d||jdt<t|jS)Nr rFr,Computed checksum digest has unexpected sizer)zlibcrc32rrr"r/AssertionErrorbytes)rdigests rr"zBytesWriter.to_bytes%sDJ';'<'<=>>GG uH   v;;. . . !OPP P-3 1))*TZ   rrN) __name__ __module__ __qualname__rintr#strr,r5r*r"rrr r s5555HcHdHHHH"3"4""""Ut!%!!!!!!rr cNeZdZdeddfdZdefdZdefdZde fdZ defdZ dS) BytesReaderdatarNct|tkrtdtj|tdddd}t|tkrtd|dt|kr"t d|dtd |||_t|_dS) Nz+Input data is too short to contain checksumr rFrr1rz>Bytes object is corrupted, checksum does not match. Expected: z, Got: ) r/rr4r2r3r" RuntimeErrorr_i)rr@r6s rrzBytesReader.__init__0s t99+ + + !NOO OD!5!6!6788AA uB   v;;. . . !OPP P && '6 1 1O!!$8"89OODJOO   &rc<t|j|jkSr)r/rrCrs r is_finishedzBytesReader.is_finishedBs4:$'))rct|j|j|jdzdd}|xjdz c_|Sr)r; from_bytesrrC)rresults r read_uint64zBytesReader.read_uint64EsK Jtw1, -u    1  rcttj|dSr&)r' b64decode read_bytesdecoders rread_strzBytesReader.read_strLs+ 1 12299'BBBrc|}|j|j|j|z}|xj|z c_|Sr)rIrrC)rsizerHs rrLzBytesReader.read_bytesOs@!!DGdgn45 4 r) r8r9r:r5rboolrEr;rIr<rNrLr=rrr?r?/s'U't''''$*T****SC#CCCCErr?ceZdZUdZeeegdfed<eed<deeegdfddfdZddZ d eddfd Z d e eddfd Z de fd Zed e deegefdeefdZdS)r z Provides efficient serialization and deserialization of list of bytes Note that this does not provide any guarantees around byte order N _serialize_fn_writer serialize_fnrc<||_|dSr)rSclear)rrUs rrz AppendingByteSerializer.__init__ds * rcjt|_|jtdSr)r rTr#r rs rrWzAppendingByteSerializer.clearls*"}}  !!"344444rr@c<||j|dSr)rSrT)rr@s rappendzAppendingByteSerializer.appendqs  4<.....relemsc:|D]}||dSr)rZ)rr[elems rr!zAppendingByteSerializer.extendts0  D KK      rc4|jSr)rTr"rs rr"z AppendingByteSerializer.to_bytesxs|$$&&&rdeserialize_fnc6t|}|tkr$td|g}|s2||||2|S)NzVEncoding version mismatch in AppendingByteSerializer.to_list, got )r?rIr r4rErZ)r@r_readerrHs rto_listzAppendingByteSerializer.to_list{sT""     #4 4 4 0++--00  $$&& 2 MM..00 1 1 1$$&& 2 rr7)r8r9r:__doc__rr r__annotations__rrWrZrr!r5r" staticmethodr?listrbr=rrr r [s= [!,d23333  Q/56  5555 /1/////HQKD'%'''' e +9I0J tTUw   \   r)r'r2collections.abcrrtypingrrrr r;rd__all__rr r?r r=rrrjs ........######## GCLL3 $ %!!!!!!!!4$$$$$$$$X,,,,,gaj,,,,,r