a d@sdZddlmZmZmZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlmZeeZdZdddeDZe jdd kZereZeZeZeZeZn&e e!Ze"Ze!Zee#fZee j$fZe j%Z&e j'Z(d d Z)d d Z*ddZ+ddZ,Gddde-Z.Gddde-Z/Gddde/Z0Gddde-Z1d)ddZ2d*ddZ3d+ddZ4d,d d!Z5e4Z6e2Z7Gd"d#d#e-Z8Gd$d%d%e8Z9Gd&d'd'e8Z:e9e:gZ;ed(krddl0rcCs"|dkrtd|Stdd|SdS)z?Encode an unsigned integer into a variable sized blob of bytes.|d}||d|t t |||nzt|t t fr||d|t t ||D]}| |||dqnn.t|tr$||d|t t ||D]d\}}tr|sJnt|sJ|d} |t t | || | |||dqnt|tr`||d|t||j|jd} | |nXt|tr||d|||n.t|tr|jdkrtdn2t|tr||d|tdddntdt |dkrtd|||||j |jn|dur(td||j !|j"d} | durz|j#D](\} } | $||rL| | jf} qzqLd} | dur| \}}| ||||||nd}t||j"j%dS) zMain encoder function.NvTyFniihhi __class__r1r,__name__)r6rvaluestreamsr&rr'vkeyZname_bblobexrOcZext_id2Zextension_encodetrrrros                    zBsdfSerializer._encodecsd}|}|stn2||krLtddd}|d}nd}|dkr`d}n|dkrpd}n|d krd }n|d krtd d d}n|dkrtddd}n|dkrtddd}nd|dkrtddd}nB|dkrRtddd}|dkr>tddd}|d}n|dkrPtddd}|dkr|dk}tddd}jrt|r|nd}|jj nR|rfddt |D}n2g}z|  qWntyYn0n6|dkr6tddd}fddt |D}n|dkrt }tddd}|dkrtddd}t |D]d} tddd} | dkrtddd} | dksJ| d}  || <qnB|d kr8j r"tdf}ntd f} | }n td!||durj|d} | durt| |}ntd"||S)#zMain decoder function.rrrr!NrZr[Tr\Fr]zrrz*BsdfSerializer._decode..csg|]}qSrrr rrrr~rrgrhzParse error %rz'BSDF warning: no extension found for %r)rr?EOFErrorrdecoderErzr|ror}ranger{rprFru get_bytes RuntimeErrorr1r>rQrR)r6rcharrrr&rZn_sclosedr Zn_namerOrr9rrrr}Ds                 zBsdfSerializer._decodecCst}||||S)zSave the given object to bytes.)rsavegetvalue)r6obrrrrr"s zBsdfSerializer.encodecCsx|d|tdtd|tdtdg}||||dt|dkrt|d}|j|krttddS)z0Write the given object to the given file object.BSDFrrrNz8The stream object must be the last object to be encoded.) rkstructpackVERSIONror$ _start_postellry)r6rrrstreamrrrrs  zBsdfSerializer.savecCst|}||S)z@Load the data structure that is BSDF-encoded in the given bytes.)rload)r6r'rrrrrszBsdfSerializer.decodecCs|d}|dkrtd|td|dd}td|dd}d||f}|tdkrrd}t|t|f|tdkrd }t|t|f||S) z6Load a BSDF-encoded object from the given file object.rrz'This does not look like a BSDF file: %rrrrz%i.%izLReading file with different major version (%s) from the implementation (%s).zWBSDF warning: reading file with higher minor version (%s) than the implementation (%s).)rrrr __version__rQrRr})r6rZf4 major_version minor_versionZ file_versionrrrrrs     zBsdfSerializer.load)N)rFTFF)r __module__ __qualname____doc__r:r5r4rYror}r"rrrrrrrr0ts   + [Zr0c@s.eZdZdZd ddZddZeddZd S) rwzBase class for streams.ricCsRd|_d|_t|tr"||_d}n|dkr0d|_|dvstd|j|j|dS)z8Seek to the given position (relative to the blob start).Nz>Cannot seek in a blob that is not created by the BSDF decoder.rzSeek beyond blob boundaries.)rrrrrr)r6prrrrs  z Blob.seekcCs"|jdurtd|j|jS)zCGet the current file pointer position (relative to the blob start).Nz>Cannot tell in a blob that is not created by the BSDF decoder.)rrrrrrrrrs  z Blob.tellcCsR|jdurtd|jr td|jt||jkr@tdd|_|j|S)zWrite bytes to the blob.Nz?Cannot write in a blob that is not created by the BSDF decoder.z,Cannot arbitrarily write in compressed blob.zWrite beyond blob boundaries.T) rrrGrrr$rrrk)r6r'rrrrks z Blob.writecCsH|jdurtd|jr td|j||jkrCannot read in a blob that is not created by the BSDF decoder.z+Cannot arbitrarily read in compressed blob.zRead beyond blob boundaries.)rrrGrrrr)r6rrrrrs z Blob.readcCs|jdur|j}n.|j}|d|j|j}|j||jdkrP|}n:|jdkrft|}n$|jdkr|t |}nt d|j|S)z&Get the contents of the blob as bytes.NrrrzInvalid compression %i) rrrrrrrGr< decompressr=r)r6rr rrrrrs         zBlob.get_bytescCsZ|jrV|jrV|d|j|j}|j|j|jdd|jt | dS)zZReset the blob's checksum if present. Call this after modifying the data. rrrN) rHrrrrrrrrkrrr)r6rrrrupdate_checksums   zBlob.update_checksumN)rrF)rrrrr:rrvrrrrkrrrrrrrruNs '   rucKst|fi|}||S)zrSave (BSDF-encode) the given object to bytes. See `BSDFSerializer` for details on extensions and options. )r0r")rr7r8r.rrrr"sr"cKsbt|fi|}t|trRt|d}|||WdS1sF0Yn |||SdS)zSave (BSDF-encode) the given object to the given filename or file object. See` BSDFSerializer` for details on extensions and options. wbN)r0r)r*openr)rrr7r8r.fprrrrs   ,rcKst|fi|}||S)zpLoad a (BSDF-encoded) structure from bytes. See `BSDFSerializer` for details on extensions and options. )r0r)r'r7r8r.rrrrsrcKstt|fi|}t|trf|dr0tj|}t|d}||WdS1sZ0Yn ||SdS)zLoad a (BSDF-encoded) structure from the given filename or file object. See `BSDFSerializer` for details on extensions and options. )z~/z~\rbN) r0r)r* startswithospath expanduserrr)rr7r8r.rrrrr$s    *rc@s8eZdZdZdZdZddZddZdd Zd d Z d S) rNaCBase class to implement BSDF extensions for special data types. Extension classes are provided to the BSDF serializer, which instantiates the class. That way, the extension can be somewhat dynamic: e.g. the NDArrayExtension exposes the ndarray class only when numpy is imported. A extension instance must have two attributes. These can be attributes of the class, or of the instance set in ``__init__()``: * name (str): the name by which encoded values will be identified. * cls (type): the type (or list of types) to match values with. This is optional, but it makes the encoder select extensions faster. Further, it needs 3 methods: * `match(serializer, value) -> bool`: return whether the extension can convert the given value. The default is ``isinstance(value, self.cls)``. * `encode(serializer, value) -> encoded_value`: the function to encode a value to more basic data types. * `decode(serializer, encoded_value) -> value`: the function to decode an encoded value back to its intended representation. rcCsd|jtt|fS)Nz)rOhexidrrrr__repr__[szExtension.__repr__cCs t||jSr )r)rSr6r.rrrrr,^szExtension.matchcCs tdSr NotImplementedErrorrrrrr"aszExtension.encodecCs tdSr rrrrrrdszExtension.decodeN) rrrrrOrSrr,r"rrrrrrN>srNc@s$eZdZdZeZddZddZdS)ComplexExtensionrcCs |j|jfSr )realimagrrrrr"lszComplexExtension.encodecCst|d|dS)Nrr)complexrrrrroszComplexExtension.decodeN)rrrrOrrSr"rrrrrrhsrc@s0eZdZdZddZddZddZdd Zd S) NDArrayExtensionndarraycCsdtjvrddl}|j|_dS)Nnumpyr)sysmodulesrrrS)r6nprrrr:vs zNDArrayExtension.__init__cCst|dot|dot|dS)Nshapedtypetobytes)rrrrrr,|szNDArrayExtension.matchcCst|jt|j|dS)N)rrdata)rpr text_typerrrrrrr"szNDArrayExtension.encodecCsHz ddl}Wnty"|YS0|j|d|dd}|d|_|S)Nrrr)rr)r ImportErrorZ frombufferr)r6r.rrarrrrs    zNDArrayExtension.decodeN)rrrrOr:r,r"rrrrrrss r__main__)N)N)N)N)>r __future__rrrr=rloggingrr+rrtypesr<ior getLoggerrrQrjoinr version_inforrr rr*rnrrlrLrV basicConfigunicode basestringlongZ ClassTyperrunpackrrr r(r/objectr0rwrzrur"rrrloadsdumpsrNrrr3Zbsdf_climainrrrrsj     gW;    *