o e,@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 KD:\Projects\ConvertPro\env\Lib\site-packages\imageio/plugins/pillowmulti.pyrsrc@s*eZdZdZdZdZGdddejZdS) GIFFormatz(See :mod:`imageio.plugins.pillow_legacy`ZiIz Static and animated gif (Pillow)c@s2eZdZ      d ddZdd Zd d ZdS) zGIFFormat.WriterrN FcCst|}|dks |dkrtd|dvr&dttdd}td||dur2dt||_nt|t t frBd d |D|_nt||_t|}|d ksU|td krWd }t|}t |}|rcd 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_opens2    zGIFFormat.Writer._opencCs|jdSN)r,closer-r r r_closeHzGIFFormat.Writer._closecCs~t|dd}|jdkr|jddkr|dddddf}|j}t|tr2|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 r2r6rCr r r rWriters ) rDN)rr r __doc__r r rrDr r r rrs rcCs|jdddS)Nrlittle) byteorder)to_bytes)rr r rintToBinXr7rIc@sxeZdZdZ    dddZddZd 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.GifImagePluginrJ)r-filerKrLrMrNrJr r r__init__bs  zGifWriter.__init__c Cs|d}}|jr||\}}|||j|j}ddlm}||dd}|jdkr6||||j ||_ | |||||||_ |jd7_dS)Nrrr) getheaderr8r) rKgetSubRectangle converToPILrMrNrQrUr> write_headerrLrP write_imagerO) r-r@r/rBZim_rectrectim_pilrUpaletter r rr?ys   zGifWriter.add_imagecCs<||}||}|j||j||j|dSr3) getheaderAnim getAppExtr1write)r-r@Z globalPaletter.headerZappextr r rrXs    zGifWriter.write_headercCs|jdddS)N;utf-8)r1r_encoder5r r rr4szGifWriter.closec Cs|j}||}d}|r!t|dkr!||d7}|r!t|dkst|dks)J|||} |||} ||jks>|dkrS|| || |||dn || |||D]} || q_dS)N rr)r1rJr=popgetImageDescriptorgetGraphicsControlExtrPr_) r-r@r\rZr/rBr1dataZimdeslidZgraphextrr r rrYs&          zGifWriter.write_imagecCs4d}|t|jd7}|t|jd7}|d7}|S)z4Get animation header. To replace PILs getheader()[0]sGIF89arrsrIsize)r-r@bbr r rr]s zGifWriter.getheaderAnimNcCs`|durd}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. NrT,rrrl)r-r@Zxyrnr r rrhs zGifWriter.getImageDescriptorcCsP|dkrdS|dkr d}d}|dkr&d}|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. rrdris! s NETSCAPE2.0s)rI)r-r.rnr 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!r8rrbdg?rq)chrrcrIr)r-r/rBrnr r rris zGifWriter.getGraphicsControlExtc Cs|jdkr |dfS|j}t||}|jdkr|d}t|d}t|d}|jrR|jrRt|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. rrTr8rrr9)rr) r>rOr absr:sumZargwherermr) r-r@prevdiffXYZx0x1Zy0y1r r rrVs     zGifWriter.getSubRectanglecCst|d}|dvrd}|d}t||}|j||d}|S|dvr:|dkr,|d}n|d}|j||d }|Std |) 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)rrrrRGB)r}methodzInvalid value for quantizer: %r)rconvertNeuQuantquantizer)r-r@r0Z palette_sizer[Z nq_samplefacZ nqInstancer r rrW0s       zGifWriter.converToPIL)Trrrr3)rrr)r)rr r rErSr?rXr4rYr]rhr^rirVrWr r r rr+\s"  !   r+)rEloggingnumpyr Z pillow_legacyrrr getLoggerrr"rrrrIr+r r r rs B