U 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(krddl0srcCs"|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|stnt|st|d} |t t | || | |||dqnt|tr`||d|t||j|jd} | |nXt|tr||d|||n.t|tr|jdkrtdn2t|tr||d|tdddntdt |dkrtd|||||j |j n|dk r(td||j!"|j#d} | dkrz|j$D](\} } | %||rL| | jf} qzqLd} | dk r| \}}| ||||||nd}t||j#j&dS) zMain encoder function.NvTyFniihhitddd}|d}n|dkrRtddd}|dkr|dk}tddd}jrt|r|nd}|jj nT|rfddt |D}n4g}z|  qWntk rYnXn6|dkr8tddd}fddt |D}n|dkrt }tddd}|dkrtddd}t |D]d} tddd} | dkrtddd} | dkst | d}  || <qnB|d kr:jr$tdf}ntd f} | }n td!||dk rj|d} | dk rv| |}ntd"||S)#zMain decoder function.rrrr NrXrYTrZFr[zrsz*BsdfSerializer._decode..csg|]}qSrrr rrrr~srerfzParse error %rz'BSDF warning: no extension found for %r)rr>EOFErrorrdecoderDryr{rmr|rangerzrnrrrErt get_bytes RuntimeErrorr0r=rOrP)r5rcharrrr%rZn_sclosedr Zn_namerMrr8rrrr|Ds                 zBsdfSerializer._decodecCst}||||S)zSave the given object to bytes.)rsavegetvalue)r5obrrrrr!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.) ristructpackVERSIONrmr# _start_postellrx)r5rrrstreamrrrrs  zBsdfSerializer.savecCst|}||S)z@Load the data structure that is BSDF-encoded in the given bytes.)rload)r5r&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__rOrPr|)r5rZf4 major_version minor_versionZ file_versionrrrrrs     zBsdfSerializer.load)N)rFTFF)r~ __module__ __qualname____doc__r9r4r3rWrmr|r!rrrrrrrr/ts   + [Zr/c@s.eZdZdZd ddZddZeddZd S) rvzBase class for streams.rgcCsRd|_d|_t|tr"||_d}n|dkr0d|_|dkstd|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)r5prrrrs  z Blob.seekcCs"|jdkrtd|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|jdkrtd|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) rrrFrrr#rrri)r5r&rrrris z Blob.writecCsH|jdkrtd|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.)rrrFrrrr)r5rrrrrs z Blob.readcCs|jdk r|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) rrrrrrrFr; decompressr<r)r5rr 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) rGrrrrrrrrirrr)r5rrrrupdate_checksums   zBlob.update_checksumN)rrF)r~rrrr9rrurrrrirrrrrrrrtNs '   rtcKst|f|}||S)zrSave (BSDF-encode) the given object to bytes. See `BSDFSerializer` for details on extensions and options. )r/r!)rr6r7r-rrrr!s r!c KsRt|f|}t|trBt|d}|||W5QRSQRXn |||SdS)zSave (BSDF-encode) the given object to the given filename or file object. See` BSDFSerializer` for details on extensions and options. wbN)r/r(r)openr)rrr6r7r-fprrrrs     rcKst|f|}||S)zpLoad a (BSDF-encoded) structure from bytes. See `BSDFSerializer` for details on extensions and options. )r/r)r&r6r7r-rrrrs rc Ksdt|f|}t|trV|dr,tj|}t|d}||W5QRSQRXn ||SdS)zLoad a (BSDF-encoded) structure from the given filename or file object. See `BSDFSerializer` for details on extensions and options. )z~/z~\rbN) r/r(r) startswithospath expanduserrr)rr6r7r-rrrrr$s     rc@s8eZdZdZdZdZddZddZdd Zd d Z d S) rLaDBase 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 attribiutes 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)rMhexidrrrr__repr__[szExtension.__repr__cCs t||jSr )r(rQr5r-rrrrr+^szExtension.matchcCs tdSr NotImplementedErrorrrrrr!aszExtension.encodecCs tdSr rrrrrrdszExtension.decodeN) r~rrrrMrQrr+r!rrrrrrL>srLc@s$eZdZdZeZddZddZdS)ComplexExtensionrcCs |j|jfSr )realimagrrrrr!lszComplexExtension.encodecCst|d|dS)Nrr)complexrrrrroszComplexExtension.decodeN)r~rrrMrrQr!rrrrrrhsrc@s0eZdZdZddZddZddZdd Zd S) NDArrayExtensionndarraycCsdtjkrddl}|j|_dS)Nnumpyr)sysmodulesrrrQ)r5nprrrr9vs zNDArrayExtension.__init__cCst|dot|dot|dS)Nshapedtypetobytes)rrrrrr+|szNDArrayExtension.matchcCst|jt|j|dS)N)rrdata)rnr text_typerrrrrrr!szNDArrayExtension.encodecCsJz ddl}Wntk r$|YSX|j|d|dd}|d|_|S)Nrrr)rr)r ImportErrorZ frombufferr)r5r-rrarrrrs   zNDArrayExtension.decodeN)r~rrrMr9r+r!rrrrrrss r__main__)N)N)N)N)>r __future__rrrr<rloggingrr*rrtypesr;ior getLoggerr~rOrjoinr version_inforpr rr)rlrrjrJrT basicConfigunicode basestringlongZ ClassTyperrunpackrrrr'r.objectr/rvryrtr!rrrloadsdumpsrLrrr2Zbsdf_climainrrrrsj     gW;    *