o h@sdZddlmZmZmZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlmZeeZdZdddeDZe jdd kZer[eZeZeZeZeZne 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|dkr td|Stdd|S)z?Encode an unsigned integer into a variable sized blob of bytes.rz*BsdfSerializer._decode..crrrr rrrr~rrjrkzParse error %rz'BSDF warning: no extension found for %r)rr>EOFErrorrdecoderDr}rrrrranger~rsrErx get_bytes RuntimeErrorr0r=rRrS)r5rcharrrr%rn_sclosedr n_namerPrr8rrrrDs               zBsdfSerializer._decodecCst}||||S)zSave the given object to bytes.)rsavegetvalue)r5obrrrrr!s zBsdfSerializer.encodecCs||d|tdtd|tdtdg}||||dt|dkr:|d}|j|krCannot seek in a blob that is not created by the BSDF decoder.rzSeek beyond blob boundaries.)rrrrrr)r5prrrrs  z Blob.seekcCs"|jdur td|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|jdur td|jrtd|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#rrrn)r5r&rrrrns  z Blob.writecCsH|jdur td|jrtd|j||jkrtd|j|S)zRead n bytes from the blob.Nz>Cannot 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|jdur |j}n|j}|d|j|j}|j||jdkr)|}|S|jdkr5t|}|S|jdkrAt |}|St d|j)z&Get the contents of the blob as bytes.NrrrzInvalid compression %i) rrrrrrrFr; decompressr<r)r5rr rrrrrs       zBlob.get_bytescCsb|jr-|jr/|d|j|j}|j|j|jdd|jt | dSdSdS)zZReset the blob's checksum if present. Call this after modifying the data. rrrN) rGrrrrrrrrnrrr)r5rrrrupdate_checksums  zBlob.update_checksumN)rrF)rrrrr9rryrrrrnrrrrrrrrxNs '   rxcKt|fi|}||S)zrSave (BSDF-encode) the given object to bytes. See `BSDFSerializer` for details on extensions and options. )r/r!)rr6r7r-rrrr! r!cKs`t|fi|}t|tr*t|d}|||WdS1s#wYdS|||S)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    $ rcKr)zpLoad a (BSDF-encoded) structure from bytes. See `BSDFSerializer` for details on extensions and options. )r/r)r&r6r7r-rrrrrrcKsrt|fi|}t|tr4|drtj|}t|d }||WdS1s-wYdS||S)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) rOaCBase 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)rPhexidrrrr__repr__[szExtension.__repr__cCs t||jSr )r(rTr5r-rrrrr+^ zExtension.matchcCtr NotImplementedErrorrrrrr!azExtension.encodecCrr rrrrrrdrzExtension.decodeN) rrrrrPrTrr+r!rrrrrrO>s rOc@s$eZdZdZeZddZddZdS)ComplexExtensionrcCs |j|jfSr )realimagrrrrr!lrzComplexExtension.encodecCst|d|dS)Nrr)complexrrrrroszComplexExtension.decodeN)rrrrPrrTr!rrrrrrhs  rc@s0eZdZdZddZddZddZdd Zd S) NDArrayExtensionndarraycCs"dtjvrddl}|j|_dSdS)Nnumpyr)sysmodulesrrrT)r5nprrrr9vs  zNDArrayExtension.__init__cCst|dot|dot|dS)Nshapedtypetobytes)rrrrrr+|szNDArrayExtension.matchcCst|jt|j|dS)N)rrdata)rsr text_typerrrrrrr!szNDArrayExtension.encodecCsHzddl}Wn ty|YSw|j|d|dd}|d|_|S)Nrr r)rr)r ImportError frombufferr)r5r-rrarrrrs   zNDArrayExtension.decodeN)rrrrPr9r+r!rrrrrrss  r__main__r )>r __future__rrrr<rloggingrr*rrtypesr;ior getLoggerrrRrjoinr version_inforur r r)rqrrorMrW basicConfigunicode basestringlong ClassTyperrunpackrrrr'r.objectr/rzr}rxr!rrrloadsdumpsrOrrr2bsdf_climainrrrrsn      gW ;   *