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 d/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/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|dvrLdttdd}td||durddt||_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 6z*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_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.immetar0disposer r r _append_dataKs  zGIFFormat.Writer._append_data)rNrrrF)rr r r3r7rCr r r rWriters )rDN)rr r __doc__r r rrDr r r rrsrcCs|jdddS)Nrlittle) byteorder)to_bytes)rr r rintToBinXsrIc@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) r2opt_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}|jdkrl||||j ||_ | |||||||_ |jd7_dS)Nrrr) getheaderr8r) rKgetSubRectangle converToPILrMrNrQrUr> write_headerrLrP write_imagerO) r.r@r0rBZim_rectrectim_pilrUpaletter r rr?ys   zGifWriter.add_imagecCs<||}||}|j||j||j|dSr4) getheaderAnim getAppExtr2write)r.r@Z globalPaletter/headerZappextr r rrXs     zGifWriter.write_headercCs|jdddS)N;utf-8)r2r_encoder6r r rr5szGifWriter.closec Cs|j}||}d}|r4t|dkr4||d7}qt|dksDJ|||} |||} ||jksn|dkr|| || |||dn|| |||D]} || qdS)Nr rr)r2rJr=popgetImageDescriptorgetGraphicsControlExtrPr_) r.r@r\rZr0rBr2dataZimdesZlidZgraphextrr 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`|dur 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. NrT,rrrj)r.r@Zxyrlr r rrgs 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. rrris! s NETSCAPE2.0s)rI)r.r/rlr 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?ro)chrrcrIr)r.r0rBrlr r rrhs 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. rrTr8rrr9)rr) r>rOr!absr:sumZargwhererkr) r.r@prevdiffXYZx0x1Zy0y1r r rrVs     zGifWriter.getSubRectanglecCst|d}|dvr:d}|d}t||}|j||d}nB|dvrp|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@r1Z palette_sizer[Z nq_samplefacZ nqInstancer r rrW0s      zGifWriter.converToPIL)Trrr)N)rpr)r)rr r rErSr?rXr5rYr]rgr^rhrVrWr r r rr,\s   !   r,)rEloggingnumpyr!Z pillow_legacyrrr getLoggerrr#r~rrrIr,r r r rs B