U
Z$d59 @ s d Z ddlZddlmZmZ ddlmZ ddlm Z m
Z
m Z m Z G dd d eZ
G d
d d e
ZG d d
d
e
ZG dd de
ZG dd de
ZdS )ag Read/Write images using FreeImage.
Backend Library: `FreeImage `_
.. note::
To use this plugin you have to install its backend::
imageio_download_bin freeimage
or you can download the backend using the function::
imageio.plugins.freeimage.download()
Each Freeimage format has the ``flags`` keyword argument. See the `Freeimage
documentation `_ for more information.
Parameters
----------
flags : int
A freeimage-specific option. In most cases we provide explicit
parameters for influencing image reading.
N )Format
image_as_uint) RETURN_BYTES )FNAME_PER_PLATFORMIO_FLAGSdownloadfic sl e Zd ZdZdZddd fdd Zedd Zd d
Zd d Z G d
d de
j Z G dd de
j Z Z
S )FreeimageFormatz$See :mod:`imageio.plugins.freeimage`iN)fifc s t j||||d || _d S )N)
extensionsmodes)super__init___fif)selfname descriptionr r r
__class__ =/tmp/pip-unpacked-wheel-zwww1ayp/imageio/plugins/freeimage.pyr * s zFreeimageFormat.__init__c C s | j S N)r r r r r r
. s zFreeimageFormat.fifc C sr t rnt|dsFzt |jd|j|_W n tk
rD d|_Y nX |j| jkrVdS |jdkrn| jdkrndS d S )Nr rT ) r
has_libhasattrgetFIFfilenameZ
firstbytesr Exceptionr
r requestr r r _can_read2 s
zFreeimageFormat._can_readc C sj t rft|dsVzt |jd|_W n. tk
rT |jtkrJ| j |_nd|_Y nX |j| j krfdS d S )Nr wr T)
r
r r! r" r# r
ValueErrorZraw_urir r
r% r r r
_can_writeA s
zFreeimageFormat._can_writec @ s6 e Zd Zdd Zd
ddZdd Zdd Zd
d Zd S )zFreeimageFormat.Readerc C s dS )Nr r r r r r _get_lengthR s z"FreeimageFormat.Reader._get_lengthr c C s. t | jj| jj|| _| j| j d S r ) r
create_bitmapr& r# formatr
_bmZload_from_filenameget_local_filenamer flagsr r r _openU s z FreeimageFormat.Reader._openc C s | j d S r )r. closer r r r _closeY s z FreeimageFormat.Reader._closec C s$ |dkrt d| j | j fS )Nr z+This format only supports singleton images.)
IndexErrorr. Zget_image_data
get_meta_datar indexr r r _get_data\ s z FreeimageFormat.Reader._get_datac C s |d ks|dkst | j S )Nr )r5 r. r6 r7 r r r _get_meta_dataa s z%FreeimageFormat.Reader._get_meta_dataN)r )__name__
__module__ __qualname__r+ r2 r4 r9 r: r r r r ReaderQ s
r> c @ s. e Zd Zd ddZdd Zdd Zdd Zd
S ) zFreeimageFormat.Writerr c C s || _ d | _d| _i | _d S )NF)_flagsr. _is_set_metar0 r r r r2 i s z FreeimageFormat.Writer._openc C s. | j | j | j | j | j d S r )r. Z
set_meta_datarA Zsave_to_filenamer& r/ r3 r r r r r4 o s z FreeimageFormat.Writer._closec C s | j sd| _ ntd|jdkrD|jd dkrD|d d d d df }| jd krtt| jj| j j
| j | _| j | | j
| || _d S )NTz1Singleton image; can only append image data once. r r r )r@ RuntimeErrorndimshaper. r
r, r& r# r- r
r? allocateZset_image_datarA r immetar r r _append_datav s
z#FreeimageFormat.Writer._append_datac C s
|| _ d S r )rA )r rI r r r _set_meta_data s z%FreeimageFormat.Writer._set_meta_dataN)r )r; r< r= r2 r4 rJ rK r r r r Writerh s
rL )NN)r; r< r= __doc__Z_modesr propertyr
r' r* r r> rL
__classcell__r r r r r % s
r c @ s" e Zd ZdZG dd dejZdS )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 @ s e Zd ZdddZdd ZdS ) zFreeimageBmpFormat.Writerr Fc C s0 t |}|r|tjO }n
|tjO }tj| |S r )intr Z BMP_SAVE_RLEZ BMP_DEFAULTr rL r2 )r r1 compressionr r r r2 s
z FreeimageBmpFormat.Writer._openc C s t |dd}tj| ||S )N Zbitdepth)r r rL rJ rG r r r rJ s z&FreeimageBmpFormat.Writer._append_dataN)r Fr; r< r= r2 rJ r r r r rL s
rL Nr; r< r= rM r rL r r r r rP s rP c @ s4 e Zd ZdZG dd dejZG dd dejZdS )FreeimagePngFormataY 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 @ s e Zd ZdddZdS )zFreeimagePngFormat.Readerr Tc C s$ t |}|r|tjO }tj| |S r )rQ r ZPNG_IGNOREGAMMAr r> r2 )r r1 Z ignoregammar r r r2 s
z FreeimagePngFormat.Reader._openN)r Tr; r< r= r2 r r r r r> s r> c @ s e Zd Zd ddZdd ZdS )
zFreeimagePngFormat.Writerr Fc C sh t jt jt jt jd}t|}|r,|t jO }z||| O }W n tk
rX tdY nX t j
| |S )N)r r rY z&Png compression must be 0, 1, 6, or 9.) r ZPNG_Z_NO_COMPRESSIONZPNG_Z_BEST_SPEEDZPNG_Z_DEFAULT_COMPRESSIONZPNG_Z_BEST_COMPRESSIONrQ ZPNG_INTERLACEDKeyErrorr) r rL r2 )r r1 rR quantizeZ
interlacedZcompression_mapr r r r2 s
z FreeimagePngFormat.Writer._openc C s t |jdkr t|dd}n t|dd}tj| || t| jj dd}|sRnZ|j
dkrj|j d dkstt d n8|d
k s|d krt d n | j
d
|}| j
|| _
d S )NZuint16 rT rS r\ FrB r z Can only quantize RGB imagesr z!PNG quantize param must be 2..256r )strZdtyper r rL rJ rQ r& kwargsgetrD rE r) r. r\ r3 )r rH rI qZbmr r r rJ s
z&FreeimagePngFormat.Writer._append_dataN)r rY r FrU r r r r rL s
rL Nr; r< r= rM r r> rL r r r r rW s rW c @ s4 e Zd ZdZG dd dejZG dd dejZdS )FreeimageJpegFormataA 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& e Zd Zd ddZdd Zdd Zd
S ) zFreeimageJpegFormat.Readerr TFc C s: t |}|r |dkr |tjO }|s,|tjO }tj| |S )Nr )rQ r ZJPEG_EXIFROTATEZ
JPEG_ACCURATEr r> r2 )r r1
exifrotateZ quickreadr r r r2 4 s
z FreeimageJpegFormat.Reader._openc C s& t j| |\}}| ||}||fS r )r r> r9 _rotate)r r8 rH rI r r r r9 > s z$FreeimageJpegFormat.Reader._get_datac C s | j jdddkrz|d d }W n tk
r8 Y nVX |dkrB|dkrVt|d}|dkrjt|d }|d
kr|t|}|d krt|}|S ) aN 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.
re Nr Z EXIF_MAINZ Orientation)r r )rB ) rZ rB )r rS )r rg rh r )r& r` ra r[ npZrot90Zfliplr)r rH rI Zorir r r rf C s
z"FreeimageJpegFormat.Reader._rotateN)r TF)r; r< r= r2 r9 rf r r r r r> 3 s
r> c @ s e Zd Zd ddZdd ZdS )
zFreeimageJpegFormat.Writerr K Fc C sh t |}|dk s|dkr tdt |}||O }|r>|tjO }|rL|tjO }|rZ|tjO }tj| |S )Nr d z)JPEG quality should be between 1 and 100.) rQ r) r ZJPEG_PROGRESSIVEZ
JPEG_OPTIMIZEZ
JPEG_BASELINEr rL r2 )r r1 ZqualityZ progressiveoptimizeZbaseliner r r r2 ` s
z FreeimageJpegFormat.Writer._openc C s< |j dkr |jd dkr tdt|dd}tj| ||S )NrB r rg z$JPEG does not support alpha channel.rS rT )rD rE IOErrorr r rL rJ rG r r r rJ s s z'FreeimageJpegFormat.Writer._append_dataN)r rj FFFrU r r r r rL _ s
rL Nrc r r r r rd
s %,rd c @ s" e Zd ZdZG dd dejZdS )FreeimagePnmFormatal A PNM format based on the Freeimage library.
This format supports single bit (PBM), grayscale (PGM) and RGB (PPM)
images, even with ASCII or binary coding.
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
---------------------
use_ascii : bool
Save with ASCII coding. Default True.
c @ s e Zd ZdddZdS )zFreeimagePnmFormat.Writerr Tc C s$ t |}|r|tjO }tj| |S r )rQ r ZPNM_SAVE_ASCIIr rL r2 )r r1 Z use_asciir r r r2 s
z FreeimagePnmFormat.Writer._openN)r TrX r r r r rL s rL NrV r r r r rn z s rn )rM Znumpyri corer r Z core.requestr Z
_freeimager r r r
r rP rW rd rn r r r r s r%Qm