ó ;c›]c@`s‡dZddlmZmZmZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlmZejeƒZdddfZdjd„eDƒƒZe jdd kZereZeZeZeZeZn7ej ƒe!Ze"Ze!Zee#fZee j$fZe j%Z&e j'Z(d „Z)d „Z*d „Z+d „Z,de-fd„ƒYZ.de-fd„ƒYZ/de/fd„ƒYZ0de-fd„ƒYZ1e2d„Z3e2d„Z4e2d„Z5e2d„Z6e5Z7e3Z8de-fd„ƒYZ9de9fd„ƒYZ:de9fd„ƒYZ;e:e;gZ<ed krƒddl=Z=e=j>ƒndS(!s! Python implementation of the Binary Structured Data Format (BSDF). BSDF is a binary format for serializing structured (scientific) data. See http://bsdf.io for more information. This is the reference implementation, which is relatively relatively sophisticated, providing e.g. lazy loading of blobs and streamed reading/writing. A simpler Python implementation is available as ``bsdf_lite.py``. This module has no dependencies and works on Python 2.7 and 3.4+. Note: on Legacy Python (Python 2.7), non-Unicode strings are encoded as bytes. i(tabsolute_importtdivisiontprint_functionN(tBytesIOiit.cc`s|]}t|ƒVqdS(N(tstr(t.0ti((s4lib/python2.7/site-packages/imageio/plugins/_bsdf.pys 0sicC`s-|dkrtd|ƒStdd|ƒSdS(sE Encode an unsigned integer into a variable sized blob of bytes. iús Use of str.isidentifier() for Legacy Python, but slower. s^\w+$s^[0-9]N(t isinstancet string_typestretmatchtUNICODER(ts((s4lib/python2.7/site-packages/imageio/plugins/_bsdf.pyt _isidentifiermstBsdfSerializercB`szeZdZd d„Zdeeeed„Zd„Zd„Z d„Z d„Z d„Z d „Z d „Zd „ZRS( s{ Instances of this class represent a BSDF encoder/decoder. It acts as a placeholder for a set of extensions and encoding/decoding options. Use this to predefine extensions and options for high performance encoding/decoding. For general use, see the functions `save()`, `encode()`, `load()`, and `decode()`. This implementation of BSDF supports streaming lists (keep adding to a list after writing the main file), lazy loading of blobs, and in-place editing of blobs (for streams opened with a+). Options for encoding: * compression (int or str): ``0`` or "no" for no compression (default), ``1`` or "zlib" for Zlib compression (same as zip files and PNG), and ``2`` or "bz2" for Bz2 compression (more compact but slower writing). Note that some BSDF implementations (e.g. JavaScript) may not support compression. * use_checksum (bool): whether to include a checksum with binary blobs. * float64 (bool): Whether to write floats as 64 bit (default) or 32 bit. Options for decoding: * load_streaming (bool): if True, and the final object in the structure was a stream, will make it available as a stream in the decoded object. * lazy_blob (bool): if True, bytes are represented as Blob objects that can be used to lazily access the data, and also overwrite the data if the file is open in a+ mode. cK`sVi|_i|_|dkr't}nx|D]}|j|ƒq.W|j|dS(N(t _extensionst_extensions_by_clsRtstandard_extensionst add_extensiont_parse_options(tselft extensionstoptionst extension((s4lib/python2.7/site-packages/imageio/plugins/_bsdf.pyt__init__—s     icC`s©t|tƒrEidd6dd6dd6}|j|jƒ|ƒ}n|dkr`tdƒ‚n||_t|ƒ|_t|ƒ|_t|ƒ|_ t|ƒ|_ dS( Nitnoitzlibitbz2s3Compression must be 0, 1, 2, "no", "zlib", or "bz2"(iii( RRtgettlowert TypeErrort _compressiontboolt _use_checksumt_float64t_load_streamingt _lazy_blob(R%t compressiont use_checksumtfloat64tload_streamingt lazy_blobtm((s4lib/python2.7/site-packages/imageio/plugins/_bsdf.pyR$ s   cC`sYt|tƒot|tƒs-tdƒ‚n|ƒ}|j}t|tƒs]tdƒ‚nt|ƒdkst|ƒdkrtdƒ‚n||j kr³t j d|ƒn|j }|sËg}n't|t tfƒré|}n |g}x,|D]$}t|tƒsùtdƒ‚qùqùWx$|D]}||jf|j|Cannot seek in a blob that is not created by the BSDF decoder.isSeek beyond blob boundaries.N(R”RRyR©R•RR¶(R%tp((s4lib/python2.7/site-packages/imageio/plugins/_bsdf.pyRËs  cC`s2|jdkrtdƒ‚n|jjƒ|jS(sM Get the current file pointer position (relative to the blob start). s>Cannot tell in a blob that is not created by the BSDF decoder.N(R”RRyR†R¶(R%((s4lib/python2.7/site-packages/imageio/plugins/_bsdf.pyR†Øs cC`s€|jdkrtdƒ‚n|jr6tdƒ‚n|jjƒt|ƒ|jkrgtdƒ‚nt|_ |jj |ƒS(s" Write bytes to the blob. s?Cannot write in a blob that is not created by the BSDF decoder.s,Cannot arbitrarily write in compressed blob.sWrite beyond blob boundaries.N( R”RRyR6R•R†RR·RSR¬RR(R%R((s4lib/python2.7/site-packages/imageio/plugins/_bsdf.pyRRás  " cC`sq|jdkrtdƒ‚n|jr6tdƒ‚n|jjƒ||jkratdƒ‚n|jj|ƒS(s% Read n bytes from the blob. s>Cannot read in a blob that is not created by the BSDF decoder.s+Cannot arbitrarily read in compressed blob.sRead beyond blob boundaries.N(R”RRyR6R•R†R·R (R%R((s4lib/python2.7/site-packages/imageio/plugins/_bsdf.pyR ïs  cC`sÍ|jdk r|j}nA|jjƒ}|jdƒ|jj|jƒ}|jj|ƒ|jdkrt|}nU|jdkr•tj |ƒ}n4|jdkr¶t j |ƒ}nt d|jƒ‚|S(s0 Get the contents of the blob as bytes. iiisInvalid compression %iN( R§RR”R†RR R¨R6R+t decompressR,Ry(R%R§RRj((s4lib/python2.7/site-packages/imageio/plugins/_bsdf.pyRxüs   cC`s||jrx|jrx|jdƒ|jj|jƒ}|jj|j|jddƒ|jjt j |ƒj ƒƒndS(s[ Reset the blob's checksum if present. Call this after modifying the data. iiiN( R7R¬RR”R R¨R¶RµRRR²R³R´(R%R§((s4lib/python2.7/site-packages/imageio/plugins/_bsdf.pytupdate_checksums  "(RiRŽRRTR)R¦R`R«RR†RRR RxR¼(((s4lib/python2.7/site-packages/imageio/plugins/_bsdf.pyR_]s   (  cK`st||}|j|ƒS(ss Save (BSDF-encode) the given object to bytes. See `BSDFSerializer` for details on extensions and options. (RR(R€R&R'R((s4lib/python2.7/site-packages/imageio/plugins/_bsdf.pyRscK`s]t||}t|tƒrIt|dƒ}|j||ƒSWdQXn|j||ƒSdS(s Save (BSDF-encode) the given object to the given filename or file object. See` BSDFSerializer` for details on extensions and options. twbN(RRRtopenR~(R R€R&R'Rtfp((s4lib/python2.7/site-packages/imageio/plugins/_bsdf.pyR~'s cK`st||}|j|ƒS(sq Load a (BSDF-encoded) structure from bytes. See `BSDFSerializer` for details on extensions and options. (RRv(RR&R'R((s4lib/python2.7/site-packages/imageio/plugins/_bsdf.pyRv3scK`s{t||}t|tƒrj|jdƒrBtjj|ƒ}nt|dƒ}|j|ƒSWdQXn |j|ƒSdS(s Load a (BSDF-encoded) structure from the given filename or file object. See `BSDFSerializer` for details on extensions and options. s~/s~\trbN(s~/s~\( RRRt startswithtostpatht expanduserR¾Rˆ(R R&R'RR¿((s4lib/python2.7/site-packages/imageio/plugins/_bsdf.pyRˆ;sR>cB`s>eZdZdZdZd„Zd„Zd„Zd„ZRS(sE Base 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. tcC`sd|jtt|ƒƒfS(Ns(R?thextid(R%((s4lib/python2.7/site-packages/imageio/plugins/_bsdf.pyt__repr__rscC`st||jƒS(N(RRC(R%RRL((s4lib/python2.7/site-packages/imageio/plugins/_bsdf.pyRuscC`s tƒ‚dS(N(tNotImplementedError(R%RRL((s4lib/python2.7/site-packages/imageio/plugins/_bsdf.pyRxscC`s tƒ‚dS(N(RÉ(R%RRL((s4lib/python2.7/site-packages/imageio/plugins/_bsdf.pyRv{s(( RiRŽRR?RCRÈRRRv(((s4lib/python2.7/site-packages/imageio/plugins/_bsdf.pyR>Us   tComplexExtensioncB`s&eZdZeZd„Zd„ZRS(RpcC`s|j|jfS(N(trealtimag(R%RRL((s4lib/python2.7/site-packages/imageio/plugins/_bsdf.pyR„scC`st|d|dƒS(Nii(tcomplex(R%RRL((s4lib/python2.7/site-packages/imageio/plugins/_bsdf.pyRv‡s(RiRŽR?RÍRCRRv(((s4lib/python2.7/site-packages/imageio/plugins/_bsdf.pyRÊs tNDArrayExtensioncB`s2eZdZd„Zd„Zd„Zd„ZRS(tndarraycC`s.dtjkr*ddl}|j|_ndS(Ntnumpyi(tsystmodulesRÐRÏRC(R%tnp((s4lib/python2.7/site-packages/imageio/plugins/_bsdf.pyR)s cC`s+t|dƒo*t|dƒo*t|dƒS(Ntshapetdtypettobytes(Rª(R%RRL((s4lib/python2.7/site-packages/imageio/plugins/_bsdf.pyR•scC`s+td|jdt|jƒd|jƒƒS(NRÔRÕtdata(RYRÔt text_typeRÕRÖ(R%RRL((s4lib/python2.7/site-packages/imageio/plugins/_bsdf.pyR˜scC`sSyddl}Wntk r$|SX|j|dd|dƒ}|d|_|S(NiR×RÕRÔ(RÐt ImportErrort frombufferRÔ(R%RRLRÓta((s4lib/python2.7/site-packages/imageio/plugins/_bsdf.pyRv›s  (RiRŽR?R)RRRv(((s4lib/python2.7/site-packages/imageio/plugins/_bsdf.pyR΋s    t__main__(?Rt __future__RRRR,R²tloggingRÂRR‚RÑttypesR+tioRt getLoggerRiRAR„tjoinR‰t version_infoR[RRØRRWR’RUR<RFt basicConfigtunicodet basestringtlongt ClassTypeRƒRtunpackR R RRRtobjectRRaRdR_RRR~RvRˆtloadstdumpsR>RÊRÎR"tbsdf_clitmain(((s4lib/python2.7/site-packages/imageio/plugins/_bsdf.pytsf                 ÿn X    *