B %Z\ah@sdZddlmZmZmZddlZddlZddlZddl m Z ddl m Z m Z ddlmZmZeeZd ZGd d d e ZGd d d eZGdddeZddZddZddZdddZd ddZddlmZmZdZ e!eeeedZ"ddZ#e#dS)!z+ Plugin that wraps the the Pillow library. )absolute_importprint_functiondivisionN)formats)Format image_as_uint)pillow_formats pillow_docsa Parameters for reading ---------------------- pilmode : str From the Pillow documentation: * 'L' (8-bit pixels, grayscale) * 'P' (8-bit pixels, mapped to any other mode using a color palette) * 'RGB' (3x8-bit pixels, true color) * 'RGBA' (4x8-bit pixels, true color with transparency mask) * 'CMYK' (4x8-bit pixels, color separation) * 'YCbCr' (3x8-bit pixels, color video format) * 'I' (32-bit signed integer pixels) * 'F' (32-bit floating point pixels) PIL also provides limited support for a few special modes, including 'LA' ('L' with alpha), 'RGBX' (true color with padding) and 'RGBa' (true color with premultiplied alpha). When translating a color image to grayscale (mode 'L', 'I' or 'F'), the library uses the ITU-R 601-2 luma transform:: L = R * 299/1000 + G * 587/1000 + B * 114/1000 as_gray : bool If True, the image is converted using mode 'F'. When `mode` is not None and `as_gray` is True, the image is first converted according to `mode`, and the result is then "flattened" using mode 'F'. csxeZdZdZdZdZdZdZfddZe dd Z d d Z d d Z ddZ GdddejZGdddejZZS) PillowFormatz/ Base format class for Pillow formats. FNics tt|j||t|_dS)N)superr __init__ threadingRLock_lock)selfargskwargs) __class__5lib/python3.7/site-packages/imageio/plugins/pillow.pyrEszPillowFormat.__init__cCs|jS)z The PIL plugin id. ) _plugin_id)rrrr plugin_idJszPillowFormat.plugin_idc Cs|jX|jsBd|_ddl}t|ds.tdddlm}||_n|jdkrTtd|j}WdQRX|jdkrx| n| |S)NTr __version__z/Imageio Pillow plugin requires Pillow, not PIL!)Imagez*Imageio Pillow plugin requires Pillow lib.)PNGJPEGZBMPGIFZPPM) r_pillow_importedPILhasattr ImportErrorr_Image RuntimeErrorrZpreinitZinit)rr"rrrr _init_pillowPs      zPillowFormat._init_pillowcCsT|}|jd|jdkrP|j|jkrP|j|j\}}|rP|jrP||jrPdSdS)Nr ?T)r'modemodesrOPENZ firstbytes)rrequestrfactoryacceptrrr _can_readgs zPillowFormat._can_readcCs<|}|jd|jdkr8|j|jkr8|j|jkr8dSdS)Nr r(T)r'r)r* extension extensionsrSAVE)rr,rrrr _can_writeps   zPillowFormat._can_writec@sFeZdZdddZddZddZd d Zd d Zd dZddZ dS)zPillowFormat.ReaderNFcCs|j}y|j|jj\}}Wn$tk rDtd|jjYnX||_||jd|_ t |drv| |j j |j j r|j j jr|j j j|j j _t|j t|t|j d|_|dk r||jd<d|_t |j dr|j j|_dS)NzFormat %s cannot read images.r_decompression_bomb_check)as_grayis_grayr)r n_frames)formatr'r+rKeyErrorr&name _get_file_fp_imr#r4sizepalettedirtyrawmode rawmode_saved pil_try_readdict_palette_is_grayscale_kwargs_lengthr7)rpilmoder5rr-r.rrr_openxs&      zPillowFormat.Reader._opencCsd|_|jS)NF) _we_own_fpr,get_file)rrrrr;szPillowFormat.Reader._get_filecCst|j|jr|jdS)N)save_pillow_closer=rJr<close)rrrr_closes zPillowFormat.Reader._closecCs|jS)N)rG)rrrr _get_lengthszPillowFormat.Reader._get_lengthcCs6y|j|Wn tk r0td|YnXdS)NzCould not seek to index %i)r=seekEOFError IndexError)rindexrrr_seekszPillowFormat.Reader._seekcCs||jkrtd||jf|j}||kr:||n x||krX|d7}||q %ir r) rGrRr=tellrTr?r@rArBgetdata pil_get_framerFinfo)rrSr imrrr _get_datas    zPillowFormat.Reader._get_datacCs|dks|dkst|jjS)Nr)rRr=rX)rrSrrr_get_meta_datasz"PillowFormat.Reader._get_meta_data)NF) __name__ __module__ __qualname__rIr;rNrOrTrZr[rrrrReaderws r_c@s,eZdZddZddZddZddZd S) zPillowFormat.WritercCs`|j}y|j|jj|_Wn$tk rBtd|jjYnX|j |_ i|_ d|_ dS)NzFormat %s cannot write images.F) r8r'r2rZ _save_funcr9r&r:r,rKr<_meta_written)rrrrrrIs  zPillowFormat.Writer._opencCsdS)Nr)rrrrrNszPillowFormat.Writer._closecCs|jrtd|jj|jdkrD|jddkrD|dddddf}d|_|j|t||jj }d|jkrv| }|j |j fd|jj i|jt |dS) Nz&Format %s only supports single images.r rTbitsr8)rar&r8r:ndimshaper`updatendarray_to_pilrquantizeZsaver<rL)rrYmetaZimgrrr _append_datas  z PillowFormat.Writer._append_datacCs|j|dS)N)r`rg)rrjrrr set_meta_datasz!PillowFormat.Writer.set_meta_dataN)r\r]r^rIrNrkrlrrrrWriters rm)r\r]r^__doc__r!r%_modes _descriptionrpropertyrr'r/r3rr_rm __classcell__rr)rrr ;s   Gr c@s4eZdZdZGdddejZGdddejZdS) PNGFormata A PNG format based on Pillow. This format supports grayscale, RGB and RGBA images. Parameters for reading ---------------------- ignoregamma : bool Avoid gamma correction. Default True. pilmode : str From the Pillow documentation: * 'L' (8-bit pixels, grayscale) * 'P' (8-bit pixels, mapped to any other mode using a color palette) * 'RGB' (3x8-bit pixels, true color) * 'RGBA' (4x8-bit pixels, true color with transparency mask) * 'CMYK' (4x8-bit pixels, color separation) * 'YCbCr' (3x8-bit pixels, color video format) * 'I' (32-bit signed integer pixels) * 'F' (32-bit floating point pixels) PIL also provides limited support for a few special modes, including 'LA' ('L' with alpha), 'RGBX' (true color with padding) and 'RGBa' (true color with premultiplied alpha). When translating a color image to grayscale (mode 'L', 'I' or 'F'), the library uses the ITU-R 601-2 luma transform:: L = R * 299/1000 + G * 587/1000 + B * 114/1000 as_gray : bool If True, the image is converted using mode 'F'. When `mode` is not None and `as_gray` is True, the image is first converted according to `mode`, and the result is then "flattened" using mode 'F'. Parameters for saving --------------------- optimize : bool If present and true, instructs the PNG writer to make the output file as small as possible. This includes extra processing in order to find optimal encoder settings. transparency: This option controls what color image to mark as transparent. dpi: tuple of two scalars The desired dpi in each direction. pnginfo: PIL.PngImagePlugin.PngInfo Object containing text tags. compress_level: int ZLIB compression level, a number between 0 and 9: 1 gives best speed, 9 gives best compression, 0 gives no compression at all. Default is 9. When ``optimize`` option is True ``compress_level`` has no effect (it is set to 9 regardless of a value passed). compression: int Compatibility with the freeimage PNG format. If given, it overrides compress_level. icc_profile: The ICC Profile to include in the saved file. bits (experimental): int This option controls how many bits to store. If omitted, the PNG writer uses 8 bits (256 colors). quantize: Compatibility with the freeimage PNG format. If given, it overrides bits. In this case, given as a number between 1-256. dictionary (experimental): dict Set the ZLIB encoder dictionary. c@seZdZdddZddZdS) zPNGFormat.ReaderNFTcCstjj|||dS)N)rHr5)r r_rI)rrHr5 ignoregammarrrrI$szPNGFormat.Reader._openc Cstj||\}}|jjddsyt|d}Wnttfk rJYn>Xt|j t j kr^dnd}d}|||||d|dd<||fS)NrtTgammaig?gW| ||}||fS)NZexifr)TAGS EXIF_MAIN) r r_rZZ PIL.ExifTagsrr=Z_getexifitemsrw_rotate)rrSrYrXrtagvalueZdecodedrrrrZs   zJPEGFormat.Reader._get_datacCs|jjddry|dd}Wntk r4YnVX|dkr>|dkrRt|d}|dkrft|d }|d krxt|}|d krt|}|S) z Use Orientation information from EXIF meta data to orient the image correctly. Similar code as in FreeImage plugin. rTrZ Orientation)r r)rbr)rb)r)rrrr)r,rrwr9r{Zrot90Zfliplr)rrYrjZorirrrrs    zJPEGFormat.Reader._rotate)NFT)r\r]r^rIr;rZrrrrrr_s  r_c@seZdZdddZddZdS) zJPEGFormat.WriterKFcKslt|}|dks|dkr tdtdtd|}||d<t||d<t||d<tj||j |dS)Nr dz)JPEG quality should be between 1 and 100._quality progressiver) intryminmaxboolr rmrIr`rg)rrrrrrrrrIs   zJPEGFormat.Writer._opencCs@|jdkr |jddkr tdt|dd}tj|||dS)Nrbrcrz$JPEG does not support alpha channel.r)r)rerfIOErrorrr rmrk)rrYrjrrrrks  zJPEGFormat.Writer._append_dataN)rFF)r\r]r^rIrkrrrrrms rmN)r\r]r^rnr r_rmrrrrris:7rcCs(t|dr$tt|dddr$|dS)NrMfp)r#getattrrM)rYrrrrLs rLc Csdy|dWnNtk r^}z0d}|d7}t|}d|j||f}t|Wdd}~XYnXdS)Nrz8http://pillow.readthedocs.io/en/latest/installation.htmlz#external-librarieszACould not load "%s" Reason: "%s" Please see documentation at: %s)rVrrrry)rYeZsiteZpillow_error_messageZ error_messagerrrrCsrCcCsP|jdkrdSt|d}|\}}|||d}tt|dS)NPF)rbr r)r)r{ZasarrayZ getpaletteZreshapeZ getextremaZallcloseZdiff)Z pil_imager?startstopZ valid_paletterrrrEs   rEc Cs|dkrt|}|}|dk r4||jkr0||}n|r<n|jdkrX|rX|d}n|jdkr|jdddk r|d}q|jjdkrt|jdtj }t |jdr|jj |j_ |jj r|jj n|jj}t |}d |f|_|jdd ks|jdd krL|d d krLt|dddd fd t|jd|jf}|dr|jdd kr~|dddddgfn|dddddd gf}t|tj }y ||}Wn tk r|d}YnXn |d}n.d|jkr|d}n|jdkr|d}|r |d}n$t|tjsD|jdkrD|d}|jdr|j} |jdrjdnd}d|jkr|dd}t||}| ddd |_n4|jdkr|jdkr|dkrd}tj||d }|S)!z is_gray: Whether the image *is* gray (by inspecting its palette). as_gray: Whether the resulting image must be converted to gaey. mode: The mode to convert to. NrLrRGBA)RGBrr rBrcrbrXrvrZBGRrAZCMYKrF1zI;16Bz>u2zendswithrtobytescopyr8) rYr6r5r)rzframepZ nchannelsZframe_palettedrfrrrrW!sb     .2 >          "rWcCs ddlm}|jdkr8t|dd}ddd|jd }nv|d krd }d }|jjd kr^t|}q|dkr|dkr| t j }d}}qt|dd}nt|dd}d}d}|d kr| }|jd kr| ||jj}||d|n"|jd|jdf}||||}|S|||SdS)Nr)rrbr)rrr)rbrr)ZpngrzI;16rfrrrrawr )r"rrerrfrzZkindrrZastyper{rrnewTZ frombytesZ fromarray)ZarrZ format_strrr)Z mode_baseZ array_bufferrYZ image_shaperrrrhs2          rh) GIFFormat TIFFFormatZMPEG)rrr ZTIFFcCsxxrtD]j\}}}|tkrqt|t}|jp.|}||d|||j}||_|tksX|jsft |t |_t |qWdS)Nz-PIL) r IGNORE_FORMATSSPECIAL_FORMATSrwr rprorrnr GENERIC_DOCSrZ add_format)idZsummaryZextZ FormatClsr8rrrregister_pillow_formatss  r)NNNN)N)$rnZ __future__rrrZloggingrZnumpyr{rrZcorerrZ pillow_infor r Z getLoggerr\rrr rsrrLrCrErWrhZ pillowmultirrrrDrrrrrrs2  #&   a .