o 9ήc ,@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 B/tmp/pip-target-vg8gfxp4/lib/python/imageio/plugins/pillowmulti.pyrsrc@s*eZdZdZdZdZGdddejZdS) GIFFormatz(See :mod:`imageio.plugins.pillow_legacy`iIz 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 7sz*GIFFormat.Writer._open..rinfr)int ValueErrornplog2loggerwarningr _duration isinstancelisttuplebool_disposerequestget_file GifWriter_writer)selfloopdurationfps palettesize quantizer subrectanglesfpr r r_opens2    zGIFFormat.Writer._opencCs|jdSN)r.closer/r r r_closeIzGIFFormat.Writer._closecCs~t|dd}|jdkr|jddkr|dddddf}|j}t|tr2|tt|d|jj }|j }|j |||dS)Nr)bitdepthrr) rndimshaper%r&r'minlenr._countr* add_image)r/immetar1disposer r r _append_dataLs  zGIFFormat.Writer._append_data)rNrrrF)rr r r7r;rIr r r rWriters * rJN)rr r __doc__r r rrJr r r rrs rcCs|jdddS)Nrlittle) byteorder)to_bytes)rr r rintToBinYr<rOc@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) r6opt_subrectangleopt_loop opt_quantizeropt_palette_size_previous_image_global_paletterDPIL.GifImagePluginrP)r/filerQrRrSrTrPr r r__init__cs  zGifWriter.__init__c Cs|d}}|jr||\}}|||j|j}ddlm}||dd}|jdkr6||||j ||_ | |||||||_ |jd7_dS)Nrrr) getheaderr>r) rQgetSubRectangle converToPILrSrTrWr[rD write_headerrRrV write_imagerU) r/rFr1rHim_rectrectim_pilr[paletter r rrEzs   zGifWriter.add_imagecCs<||}||}|j||j||j|dSr8) getheaderAnim getAppExtr6write)r/rF globalPaletter0headerappextr r rr^s    zGifWriter.write_headercCs|jdddS)N;utf-8)r6rfencoder:r r rr9szGifWriter.closec Cs|j}||}d}|r!t|dkr!||d7}|r!t|dkst|dks)J|||} |||} ||jks>|dkrS|| || |||dn || |||D]} || q_dS)N rr)r6rPrCpopgetImageDescriptorgetGraphicsControlExtrVrf) r/rFrcrar1rHr6dataimdeslidgraphextrr r rr_s&          zGifWriter.write_imagecCs4d}|t|jd7}|t|jd7}|d7}|S)z4Get animation header. To replace PILs getheader()[0]sGIF89arrsrOsize)r/rFbbr r rrds 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. NrZ,rrrw)r/rFxyryr r rrqs 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. rrmris! s NETSCAPE2.0s)rO)r/r0ryr r rres 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!r>rrkdg?r})chrrlrOr)r/r1rHryr r rrrs 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. rrZr>rrr?)rr) rDrUr!absr@sumargwhererxr) r/rFprevdiffXYx0x1y0y1r r rr\s     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). gif)nqneuquantrRGBA)colors)rrrrRGB)rmethodzInvalid value for quantizer: %r)rconvertNeuQuantquantizer )r/rFr4 palette_sizerb nq_samplefac nqInstancer r rr]3s       zGifWriter.converToPIL)Trrrr8)r~r)r)rr r rKrYrEr^r9r_rdrqrerrr\r]r r r rr-]s"  "   r-)rKloggingnumpyr! pillow_legacyrrr getLoggerrr#rrrrOr-r r r rs C