B %Z\}-@sdZddlmZmZmZddlZddlZddlm Z ddl m Z ej ddkZ e rXeZneZdd ZGd d d ZGd d d eZGdddeZGddde ZeddddZe edS)z BSDF plugin. )absolute_importprint_functiondivisionN)formats)FormatcCspddlm}Gddd|j}Gddd|j}Gddd|}Gd d d |}|||g}|j|f|}||fS) N)_bsdfc@s*eZdZdZdZejZddZddZ dS)z-get_bsdf_serializer..NDArrayExtensionzC Copy of BSDF's NDArrayExtension but deal with lazy blobs. ndarraycSst|jt|j|dS)N)shapedtypedata)dictr text_typer tobytes)selfsvr3lib/python3.7/site-packages/imageio/plugins/bsdf.pyencode"sz4get_bsdf_serializer..NDArrayExtension.encodecSs|S)Nr)rrrrrrdecode%sz4get_bsdf_serializer..NDArrayExtension.decodeN) __name__ __module__ __qualname____doc__namenpr clsrrrrrrNDArrayExtensions r c@s eZdZdZddZddZdS)z+get_bsdf_serializer..ImageExtensionzH We implement two extensions that trigger on the Image classes. cSst|j|jdS)N)arraymeta)rr!r")rrrrrrr,sz2get_bsdf_serializer..ImageExtension.encodecSst|d|dS)Nr!r")Image)rrrrrrr/sz2get_bsdf_serializer..ImageExtension.decodeN)rrrrrrrrrrImageExtension(sr$c@seZdZdZeZdS)z-get_bsdf_serializer..Image2DExtensionZimage2dN)rrrrImage2DrrrrrImage2DExtension2sr&c@seZdZdZeZdS)z-get_bsdf_serializer..Image3DExtensionZimage3dN)rrrrImage3DrrrrrImage3DExtension7sr()r Z ExtensionZBsdfSerializer)optionsbsdfr r$r&r(ZextsZ serializerrrrget_bsdf_serializers    r,c@s(eZdZdZddZddZddZdS) r#z Class in which we wrap the array and meta data. By using an extension we can make BSDF trigger on these classes and thus encode the images. as actual images. cCs||_||_dS)N)r!r")rr!r"rrr__init__HszImage.__init__cCsTt|jtjsN|j}|d}t|ts.|}tj||dd|_|d|j_|jS)Nrr )r r ) isinstancer!rr bytesZ get_bytesZ frombufferr )rrZblobrrr get_arrayLs  zImage.get_arraycCs|jS)N)r")rrrrget_metaVszImage.get_metaN)rrrrr-r0r1rrrrr#Bs r#c@s eZdZdS)r%N)rrrrrrrr%Zsr%c@s eZdZdS)r'N)rrrrrrrr'^sr'c@sDeZdZdZddZddZGdddejZGdd d ejZd S) BsdfFormata The BSDF format enables reading and writing of image data in the BSDF serialization format. This format allows storage of images, volumes, and series thereof. Data can be of any numeric data type, and can optionally be compressed. Each image/volume can have associated meta data, which can consist of any data type supported by BSDF. By default, image data is lazily loaded; the actual image data is not read until it is requested. This allows storing multiple images in a single file and still have fast access to individual images. Alternatively, a series of images can be read in streaming mode, reading images as they are read (e.g. from http). BSDF is a simple generic binary format. It is easy to extend and there are standard extension definitions for 2D and 3D image data. Read more at http://bsdf.io. Parameters for reading ---------------------- random_access : bool Whether individual images in the file can be read in random order. Defaults to True for normal files, and to False when reading from HTTP. If False, the file is read in "streaming mode", allowing reading files as they are read, but without support for "rewinding". Note that setting this to True when reading from HTTP, the whole file is read upon opening it (since lazy loading is not possible over HTTP). Parameters for saving --------------------- compression : {0, 1, 2} Use ``0`` or "no" for no compression, ``1`` or "zlib" for Zlib compression (same as zip files and PNG), and ``2`` or "bz2" for Bz2 compression (more compact but slower). Default 1 (zlib). Note that some BSDF implementations may not support compression (e.g. JavaScript). cCs(|jd|jdkr$|jdr$dSdS)Nr ?sBSDFT)modemodes firstbytes startswith)rrequestrrr _can_reads zBsdfFormat._can_readcCs(|jd|jdkr$|j|jkr$dSdS)Nr r3T)r4r5 extension extensions)rr8rrr _can_writes zBsdfFormat._can_writec@s6eZdZd ddZddZddZdd Zd d ZdS) zBsdfFormat.ReaderNcCs>|jjdddkstd|jjdddksX|jjdddksX|jjddd ksXi}|jjd r|dkrvd nt|}d |d <| |d <n&|dkrdnt|}||d <| |d <|j}t|\}|_|j ||_ t |j t rd|j krd|j krt |j d|j d|_ t |j t t|jfs:tddS)NsBSDFzNot a BSDF files Mimage2Ds Mimage3Dl)zhttp://zhttps://FZ lazy_blobZload_streamingTr"r!z8BSDF file does not look seem to have an image container.)r8r6AssertionErrorfilenamer7boolget_filer, _serializerload_streamr.rr#list ListStream RuntimeError)rZ random_accessr*Zrafiler+rrr_opens,      zBsdfFormat.Reader._opencCsdS)Nr)rrrr_closeszBsdfFormat.Reader._closecCs@t|jtrdSt|jtr&t|jS|jjdkr8tjS|jjS)Nr r)r.rHr#rIlencountrinf)rrrr _get_lengths    zBsdfFormat.Reader._get_lengthcCs |dks||kr(td||ft|jtr<|j}nRt|jtrT|j|}n:||jjkrhtdx||jjkr|jqjW|j}t|trd|krd|krt|d|d}t|tr| | fSt |}t |dkr|n|ddd}t d |dS) NrzImage index %i not in [0 %i].zIBSDF file is being read in streaming mode, thus does not allow rewinding.r"r!z...zBSDF file contains non-image )Z get_length IndexErrorr.rHr#rIindexnextrr0r1reprrOrK)rrVZimage_obrrrr _get_datas,        zBsdfFormat.Reader._get_datacCsiS)Nr)rrVrrr_get_meta_datasz BsdfFormat.Reader._get_meta_data)N)rrrrMrNrRrZr[rrrrReaders  , %r\c@s.eZdZd ddZddZddZdd Zd S) zBsdfFormat.Writerr cCs\d|i}t|\}|_|jjddkr4d|_d|_n$|j}||_|j||jdS)N compressionr ZivF) r,rFr8r4rH_writtenrErJsave)rr]r*r+rLrrrrMs  zBsdfFormat.Writer._opencCs|jdk r|jddS)NF)rHclose)rrrrrNs zBsdfFormat.Writer._closecCs$d}|jjddkrd}n@|jjddkr0d}n*d}|jdksV|jdkrZ|jddkrZd}|dksft|dkr|jdks|jdkr|jddkstn&|jdks|jdkr|jddkst|dkrt||}n t||}|jdkr|jrtd d |_|j }|j ||n |j |dS) Nr ZiIrZvVrr=)rrz"Cannot write singleton image twiceT) r8r4ndimr rBr%r'rHr^rErFr_append)rZimr"rbobrLrrr _append_datas*" (&    zBsdfFormat.Writer._append_datacCs tddS)Nz2The BSDF format only supports per-image meta data.)rK)rr"rrr set_meta_data.szBsdfFormat.Writer.set_meta_dataN)r )rrrrMrNrerfrrrrWriters  rgN) rrrrr9r<rr\rgrrrrr2bs $cr2r+z1Format based on the Binary Structured Data Formatz.bsdfZiIvV)rZ __future__rrrsysZnumpyrr)rZcorer version_infoZPY3strrZunicoder,r#r%r'r2formatZ add_formatrrrrs*  *Q