U a+d,@spdZddlZddlZddlmZmZmZee Z dZ GdddeZ GdddeZ d d ZGd d d ZdS) z" PIL formats for multiple images. N) PillowFormatndarray_to_pil image_as_uintc@seZdZdZdZdS) TIFFFormatizTIFF format (Pillow)N)__name__ __module__ __qualname___modes _descriptionr r ?/tmp/pip-unpacked-wheel-xbmu82vq/imageio/plugins/pillowmulti.pyrsrc@s*eZdZdZdZdZGdddejZdS) GIFFormatz(See :mod:`imageio.plugins.pillow_legacy`ZiIz Static and animated gif (Pillow)c@s&eZdZd ddZdd Zd d ZdS) zGIFFormat.WriterrN FcCst|}|dks|dkr td|dkrLdttdd}td||dkrddt||_n*t|t t frd d |D|_n t||_t|}|d ks|td krd }t|}t |}|rd nd|_ |j }t||||t||_dS)Nrz!GIF quantize param must be 2..256)r @rrg+?zDWarning: palettesize (%r) modified to a factor of two between 2-256.g?cSsg|] }t|qSr )float).0dr r r 6sz*GIFFormat.Writer._open..rinfr)int ValueErrornplog2loggerwarningr _duration isinstancelisttuplebool_disposerequestZget_file GifWriter_writer)selfloopdurationZfpsZ palettesize quantizerZ subrectanglesfpr r r_opens:   zGIFFormat.Writer._opencCs|jdSN)r,closer-r r r_closeHszGIFFormat.Writer._closecCs~t|dd}|jdkr:|jddkr:|dddddf}|j}t|trd|tt|d|jj }|j }|j |||dS)Nr)Zbitdepthrr) rndimshaper$r%r&minlenr,_countr) add_image)r-immetar/disposer r r _append_dataKs  zGIFFormat.Writer._append_data)rNrrrF)rr r r2r6rBr r r rWriters )rCN)rr r __doc__r r rrCr r r rrsrcCs|jdddS)Nrlittle) byteorder)to_bytes)rr r rintToBinXsrHc@speZdZdZdddZddZd d Zd d Zd dZddZ dddZ ddZ d ddZ ddZ d!ddZdS)"r+zClass that for helping write the animated GIF file. This is based on code from images2gif.py (part of visvis). The version here is modified to allow streamed writing. TrrcCsF||_||_||_||_||_d|_d|_d|_ddlm }||_ dS)Nr)getdata) r1opt_subrectangleopt_loop opt_quantizeropt_palette_size_previous_image_global_paletter=PIL.GifImagePluginrI)r-filerJrKrLrMrIr r r__init__bs zGifWriter.__init__c Cs|d}}|jr||\}}|||j|j}ddlm}||dd}|jdkrl||||j ||_ | |||||||_ |jd7_dS)Nrrr) getheaderr7r) rJgetSubRectangle converToPILrLrMrPrTr= write_headerrKrO write_imagerN) r-r?r/rAZim_rectrectim_pilrTpaletter r rr>ys   zGifWriter.add_imagecCs<||}||}|j||j||j|dSr3) getheaderAnim getAppExtr1write)r-r?Z globalPaletter.headerZappextr r rrWs     zGifWriter.write_headercCs|jdddS)N;utf-8)r1r^encoder5r r rr4szGifWriter.closec Cs|j}||}d}|r4t|dkr4||d7}qt|dksDt|||} |||} ||jksn|dkr|| || |||dn|| |||D]} || qdS)N rr) r1rIr<popAssertionErrorgetImageDescriptorgetGraphicsControlExtrOr^) r-r?r[rYr/rAr1dataZimdesZlidZgraphextrr r rrXs"         zGifWriter.write_imagecCs4d}|t|jd7}|t|jd7}|d7}|S)z4Get animation header. To replace PILs getheader()[0]sGIF89arrsrHsize)r-r?bbr r rr\s zGifWriter.getheaderAnimNcCs`|dkr d}d}|t|d7}|t|d7}|t|jd7}|t|jd7}|d7}|S)aUsed for the local color table properties per image. Otherwise global color table applies to all frames irrespective of whether additional colors comes in play that require a redefined palette. Still a maximum of 256 color per frame, obviously. Written by Ant1 on 2010-08-22 Modified by Alex Robinson in Janurari 2011 to implement subrectangles. NrS,rrrk)r-r?Zxyrmr r rrhs zGifWriter.getImageDescriptorcCsP|dkr dS|dkrd}d}|dkrLd}|d7}|d7}|t|7}|d7}|S) z|Application extension. This part specifies the amount of loops. If loop is 0 or inf, it goes on infinitely. rrcris! s NETSCAPE2.0s)rH)r-r.rmr r rr]s zGifWriter.getAppExt皙?rcCsJd}|t|d@d>d7}|tt|dd7}|d7}|d7}|S)abGraphics Control Extension. A sort of header at the start of each image. Specifies duration and transparancy. Dispose ------- * 0 - No disposal specified. * 1 - Do not dispose. The graphic is to be left in place. * 2 - Restore to background color. The area used by the graphic must be restored to the background color. * 3 - Restore to previous. The decoder is required to restore the area overwritten by the graphic with what was there prior to rendering the graphic. * 4-7 -To be defined. s!r7rradg?rp)chrrbrHr)r-r/rArmr r rris zGifWriter.getGraphicsControlExtc Cs|jdkr|dfS|j}t||}|jdkr:|d}t|d}t|d}|jr|jrt|dt|dd}}t|dt|dd}} nd\}}d\}} ||| ||f||ffS)arCalculate the minimal rectangle that need updating. Returns a two-element tuple containing the cropped image and an x-y tuple. Calculating the subrectangles takes extra time, obviously. However, if the image sizes were reduced, the actual writing of the GIF goes faster. In some cases applying this method produces a GIF faster. rrSr7rrr8)rr) r=rNr absr9sumZargwhererlr) r-r?prevZdiffXYZx0x1Zy0y1r r rrUs     zGifWriter.getSubRectanglecCst|d}|dkr:d}|d}t||}|j||d}nB|dkrp|dkrV|d}n |d}|j||d }n td ||S) zConvert image to Paletted PIL image. PIL used to not do a very good job at quantization, but I guess this has improved a lot (at least in Pillow). I don't think we need neuqant (and we can add it later if we really want). Zgif)ZnqZneuquantrZRGBA)colors)rrrrZRGB)r{methodzInvalid value for quantizer: %r)rconvertNeuQuantquantizer)r-r?r0Z palette_sizerZZ nq_samplefacZ nqInstancer r rrV0s      zGifWriter.converToPIL)Trrr)N)rqr)r)rr r rDrRr>rWr4rXr\rhr]rirUrVr r r rr+\s   !   r+)rDloggingZnumpyr Z pillow_legacyrrr getLoggerrr"r~rrrHr+r r r rs B