B KpŽ[÷Eã%@sdZddlmZmZmZddlZddlmZddl m Z m Z ddl m Z mZmZmZGd d „d e ƒZGd d „d eƒZGd d„deƒZGdd„deƒZeeeddddœZddiZddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8g%Zd9d:„Zd;d<„ZeƒdS)=z¾ Plugin that wraps the freeimage lib. The wrapper for Freeimage is part of the core of imageio, but it's functionality is exposed via the plugin system (therefore this plugin is very thin). é)Úabsolute_importÚprint_functionÚdivisionNé)Úformats)ÚFormatÚ image_as_uinté)ÚfiÚdownloadÚIO_FLAGSÚFNAME_PER_PLATFORMc@sTeZdZdZdZedd„ƒZdd„Zdd„ZGd d „d e j ƒZ Gd d „d e j ƒZ d S)ÚFreeimageFormata{ This is the default format used for FreeImage. Each Freeimage format has the 'flags' keyword argument. See the Freeimage documentation for more information. The freeimage plugin requires a `freeimage` binary. If this binary not available on the system, it can be downloaded manually from by either - the command line script ``imageio_download_bin freeimage`` - the Python method ``imageio.plugins.freeimage.download()`` Parameters for reading ---------------------- flags : int A freeimage-specific option. In most cases we provide explicit parameters for influencing image reading. Parameters for saving ---------------------- flags : int A freeimage-specific option. In most cases we provide explicit parameters for influencing image saving. ÚicCs|jS)N)Ú_fif)Úself©rú8lib/python3.7/site-packages/imageio/plugins/freeimage.pyÚfif0szFreeimageFormat.fifcCsZt ¡rVt|dƒsFyt |jd|j¡|_Wntk rDd|_YnX|j|jkrVdSdS)NrÚréÿÿÿÿT) r Úhas_libÚhasattrÚgetFIFÚfilenameZ firstbytesrÚ Exceptionr)rÚrequestrrrÚ _can_read4s   zFreeimageFormat._can_readcCsVt ¡rRt|dƒsByt |jd¡|_Wntk r@d|_YnX|j|jkrRdSdS)NrÚwrT)r rrrrrrr)rrrrrÚ _can_write?s   zFreeimageFormat._can_writec@s6eZdZdd„Zd dd„Zdd„Zdd „Zd d „Zd S)zFreeimageFormat.ReadercCsdS)Nr r)rrrrÚ _get_lengthMsz"FreeimageFormat.Reader._get_lengthrcCs.t |jj|jj|¡|_|j |j ¡¡dS)N) r Ú create_bitmaprrÚformatrÚ_bmZload_from_filenameÚget_local_filename)rÚflagsrrrÚ_openPszFreeimageFormat.Reader._opencCs|j ¡dS)N)r#Úclose)rrrrÚ_closeTszFreeimageFormat.Reader._closecCs$|dkrtdƒ‚|j ¡|j ¡fS)Nrz+This format only supports singleton images.)Ú IndexErrorr#Zget_image_dataÚ get_meta_data)rÚindexrrrÚ _get_dataWsz FreeimageFormat.Reader._get_datacCs |dks|dkstƒ‚|j ¡S)Nr)r)r#r*)rr+rrrÚ_get_meta_data\sz%FreeimageFormat.Reader._get_meta_dataN)r)Ú__name__Ú __module__Ú __qualname__r r&r(r,r-rrrrÚReaderLs  r1c@s.eZdZd dd„Zdd„Zdd„Zdd „Zd S) zFreeimageFormat.WriterrcCs||_d|_d|_i|_dS)NF)Ú_flagsr#Ú_is_setÚ_meta)rr%rrrr&dszFreeimageFormat.Writer._opencCs.|j |j¡|j |j ¡¡|j ¡dS)N)r#Z set_meta_datar4Zsave_to_filenamerr$r')rrrrr(jszFreeimageFormat.Writer._closecCsŠ|jsd|_ntdƒ‚|jdkrD|jddkrD|dd…dd…df}|jdkrtt |jj|j j |j ¡|_|j  |¡|j  |¡||_dS)NTz1Singleton image; can only append image data once.érr r)r3Ú RuntimeErrorÚndimÚshaper#r r!rrr"rr2ÚallocateZset_image_datar4)rÚimÚmetarrrÚ _append_dataqs   z#FreeimageFormat.Writer._append_datacCs ||_dS)N)r4)rr;rrrÚ_set_meta_dataˆsz%FreeimageFormat.Writer._set_meta_dataN)r)r.r/r0r&r(r<r=rrrrÚWritercs r>N) r.r/r0Ú__doc__Ú_modesÚpropertyrrrrr1r>rrrrrs   rc@s"eZdZdZGdd„dejƒZdS)ÚFreeimageBmpFormata» A BMP format based on the Freeimage library. This format supports grayscale, RGB and RGBA images. The freeimage plugin requires a `freeimage` binary. If this binary not available on the system, it can be downloaded manually from by either - the command line script ``imageio_download_bin freeimage`` - the Python method ``imageio.plugins.freeimage.download()`` Parameters for saving --------------------- compression : bool Whether to compress the bitmap using RLE when saving. Default False. It seems this does not always work, but who cares, you should use PNG anyway. c@seZdZddd„Zdd„ZdS) zFreeimageBmpFormat.WriterrFcCs0t|ƒ}|r|tjO}n |tjO}tj ||¡S)N)Úintr Z BMP_SAVE_RLEZ BMP_DEFAULTrr>r&)rr%Ú compressionrrrr&¨s   zFreeimageBmpFormat.Writer._opencCst|dd}tj |||¡S)Né)Úbitdepth)rrr>r<)rr:r;rrrr<²s z&FreeimageBmpFormat.Writer._append_dataN)rF)r.r/r0r&r<rrrrr>§s r>N)r.r/r0r?rr>rrrrrB’srBc@s4eZdZdZGdd„dejƒZGdd„dejƒZdS)ÚFreeimagePngFormataf A PNG format based on the Freeimage library. This format supports grayscale, RGB and RGBA images. The freeimage plugin requires a `freeimage` binary. If this binary not available on the system, it can be downloaded manually from by either - the command line script ``imageio_download_bin freeimage`` - the Python method ``imageio.plugins.freeimage.download()`` Parameters for reading ---------------------- ignoregamma : bool Avoid gamma correction. Default True. Parameters for saving --------------------- compression : {0, 1, 6, 9} The compression factor. Higher factors result in more compression at the cost of speed. Note that PNG compression is always lossless. Default 9. quantize : int If specified, turn the given RGB or RGBA image in a paletted image for more efficient storage. The value should be between 2 and 256. If the value of 0 the image is not quantized. interlaced : bool Save using Adam7 interlacing. Default False. c@seZdZddd„ZdS)zFreeimagePngFormat.ReaderrTcCs$t|ƒ}|r|tjO}tj ||¡S)N)rCr ZPNG_IGNOREGAMMArr1r&)rr%Z ignoregammarrrr&×s zFreeimagePngFormat.Reader._openN)rT)r.r/r0r&rrrrr1Ösr1c@seZdZd dd„Zdd„ZdS) zFreeimagePngFormat.Writerré FcCshtjtjtjtjdœ}t|ƒ}|r,|tjO}y|||O}Wntk rXtdƒ‚YnXt j   ||¡S)N)rr érHz&Png compression must be 0, 1, 6, or 9.) r ZPNG_Z_NO_COMPRESSIONZPNG_Z_BEST_SPEEDZPNG_Z_DEFAULT_COMPRESSIONZPNG_Z_BEST_COMPRESSIONrCZPNG_INTERLACEDÚKeyErrorÚ ValueErrorrr>r&)rr%rDÚquantizeZ interlacedZcompression_maprrrr&âs  zFreeimagePngFormat.Writer._opencCs°t|jƒdkrt|dd}n t|dd}tj |||¡t|jj  dd¡ƒ}|sRnZ|j dkrj|j ddkstt d ƒ‚n8|d ks„|d krŽt d ƒ‚n|j  d |¡}|j  ¡||_ dS)NZuint16é)rFrErLFr5rzCan only quantize RGB imagesréz!PNG quantize param must be 2..256r)ÚstrZdtyperrr>r<rCrÚkwargsÚgetr7r8rKr#rLr')rr:r;ÚqZbmrrrr<ôs    z&FreeimagePngFormat.Writer._append_dataN)rrHrF)r.r/r0r&r<rrrrr>ás r>N)r.r/r0r?rr1r>rrrrrG·s rGc@s4eZdZdZGdd„dejƒZGdd„dejƒZdS)ÚFreeimageJpegFormataS A JPEG format based on the Freeimage library. This format supports grayscale and RGB images. The freeimage plugin requires a `freeimage` binary. If this binary not available on the system, it can be downloaded manually from by either - the command line script ``imageio_download_bin freeimage`` - the Python method ``imageio.plugins.freeimage.download()`` Parameters for reading ---------------------- exifrotate : bool Automatically rotate the image according to the exif flag. Default True. If 2 is given, do the rotation in Python instead of freeimage. quickread : bool Read the image more quickly, at the expense of quality. Default False. Parameters for saving --------------------- quality : scalar The compression factor of the saved image (1..100), higher numbers result in higher quality but larger file size. Default 75. progressive : bool Save as a progressive JPEG file (e.g. for images on the web). Default False. optimize : bool On saving, compute optimal Huffman coding tables (can reduce a few percent of file size). Default False. baseline : bool Save basic JPEG, without metadata or any markers. Default False. c@s&eZdZd dd„Zdd„Zdd „Zd S) zFreeimageJpegFormat.ReaderrTFcCs:t|ƒ}|r|dkr|tjO}|s,|tjO}tj ||¡S)Nr)rCr ZJPEG_EXIFROTATEZ JPEG_ACCURATErr1r&)rr%Ú exifrotateZ quickreadrrrr&/s    z FreeimageJpegFormat.Reader._opencCs&tj ||¡\}}| ||¡}||fS)N)rr1r,Ú_rotate)rr+r:r;rrrr,9s z$FreeimageJpegFormat.Reader._get_datacCs’|jj dd¡dkrŽy|dd}Wntk r8YnVX|dkrB|dkrVt |d¡}|dkrjt |d ¡}|d kr|t |¡}|d krŽt |¡}|S) aP Use Orientation information from EXIF meta data to orient the image correctly. Freeimage is also supposed to support that, and I am pretty sure it once did, but now it does not, so let's just do it in Python. Edit: and now it works again, just leave in place as a fallback. rTNrZ EXIF_MAINZ Orientation)r r)r5é)érIr5)érE)rrVrWrX)rrPrQrJÚnpZrot90Zfliplr)rr:r;ZorirrrrU>s    z"FreeimageJpegFormat.Reader._rotateN)rTF)r.r/r0r&r,rUrrrrr1.s r1c@seZdZd dd„Zdd„ZdS) zFreeimageJpegFormat.WriterréKFcCsht|ƒ}|dks|dkr tdƒ‚t|ƒ}||O}|r>|tjO}|rL|tjO}|rZ|tjO}tj ||¡S)Nr édz)JPEG quality should be between 1 and 100.) rCrKr ZJPEG_PROGRESSIVEZ JPEG_OPTIMIZEZ JPEG_BASELINErr>r&)rr%ZqualityZ progressiveÚoptimizeZbaselinerrrr&[s   z FreeimageJpegFormat.Writer._opencCs<|jdkr |jddkr tdƒ‚t|dd}tj |||¡S)Nr5rrVz$JPEG does not support alpha channel.rE)rF)r7r8ÚIOErrorrrr>r<)rr:r;rrrr<ns z'FreeimageJpegFormat.Writer._append_dataN)rrZFFF)r.r/r0r&r<rrrrr>Zs r>N)r.r/r0r?rr1r>rrrrrSs$,rS)ZjpegÚpngÚbmpÚgifÚicoÚmngÚTIFFZFI_TIFF)ZBMPrzWindows or OS/2 Bitmapr_)ZCUTézDr. HaloZcut)ZDDSézDirectX SurfaceZdds)ZEXRéz ILM OpenEXRZexr)ZG3ézRaw fax format CCITT G.3Zg3)ZGIFézGraphics Interchange Formatr`)ZHDRézHigh Dynamic Range ImageZhdr)ZICOr z Windows Iconra)ZIFFrWzIFF Interleaved Bitmapziff,lbm)ZJ2KézJPEG-2000 codestreamzj2k,j2c)ZJNGr5zJPEG Network GraphicsZjng)ZJP2ézJPEG-2000 File FormatZjp2)ZJPEGrzJPEG - JFIF Compliantzjpg,jif,jpeg,jpe)zJPEG-XRé$zJPEG XR image formatz jxr,wdp,hdp)ZKOALArVzC64 Koala GraphicsZkoa)ZMNGrIzMultiple-image Network Graphicsrb)ZPBMrXzPortable Bitmap (ASCII)Úpbm)ZPBMRAWrEzPortable Bitmap (RAW)rm)ZPCDrHz Kodak PhotoCDZpcd)ZPCXé zZsoft PaintbrushZpcx)ZPFMé zPortable floatmapZpfm)ZPGMé zPortable Greymap (ASCII)Úpgm)ZPGMRAWé zPortable Greymap (RAW)rq)ZPICTé!zMacintosh PICTz pct,pict,pic)ZPNGé zPortable Network Graphicsr^)ZPPMézPortable Pixelmap (ASCII)Úppm)ZPPMRAWézPortable Pixelmap (RAW)rv)ZPSDézAdobe PhotoshopZpsd)ZRASrMzSun Raster ImageZras)ZRAWé"zRAW camera imagez§3fr,arw,bay,bmq,cap,cine,cr2,crw,cs1,dc2,dcr,drf,dsc,dng,erf,fff,ia,iiq,k25,kc2,kdc,mdc,mef,mos,mrw,nef,nrw,orf,pef,ptx,pxn,qtk,raf,raw,rdc,rw2,rwl,rwz,sr2,srf,srw,sti)ZSGIézSGI Image Formatzsgi,rgb,rgba,bw)ZTARGAézTruevision Targaz tga,targa)rcézTagged Image File Formatztif,tiff)ZWBMPézWireless Bitmapz wap,wbmp,wbm)ZWebPé#zGoogle WebP image formatZwebp)ZXBMézX11 Bitmap FormatZxbm)ZXPMézX11 Pixmap FormatZxpmcCsRxLtD]D\}}}}t | ¡t¡}|r||d|||jƒ}||_t |¡qWdS)Nz-FI) Ú fiformatsÚSPECIAL_CLASSESrQÚlowerrr@rrÚ add_format)ÚnamerÚdesÚextÚ FormatClassr"rrrÚ$_create_predefined_freeimage_formats¶s r‰cCsÂgtdd…<tdkrdStj}xžt| ¡ƒD]Ž}| |¡r,| |¡ d¡}| |¡ d¡}|  |¡ d¡}t  ||||f¡t   |  ¡t¡}|s’q,||d|||jƒ}||_tj|ddq,WdS)zÄ By default, imageio registers a list of predefined formats that freeimage can handle. If your version of imageio can handle more formats, you can call this function to register them. NÚasciiz-FIT)Z overwrite)rr Z_libÚrangeZFreeImage_GetFIFCountZFreeImage_IsPluginEnabledZFreeImage_GetFormatFromFIFÚdecodeZFreeImage_GetFIFDescriptionZFreeImage_GetFIFExtensionListÚappendr‚rQrƒrr@rrr„)Úlibrr…r†r‡rˆr"rrrÚcreate_freeimage_formatsÃs   r)r?Z __future__rrrZnumpyrYÚrZcorerrZ _freeimager r r r rrBrGrSr‚ZNAME_MAPrr‰rrrrrÚsp }%Qp !