U
a+dB @ s d Z ddlmZ ddlmZmZmZmZmZm Z m
Z
m Z m Z ddl
ZddlmZmZmZmZ ddlmZmZmZmZ ddlmZmZ ddl Z dd lm Z e e ed
d d Z G d
d deZ!dS )a Read/Write images using Pillow/PIL.
Backend Library: `Pillow `_
Plugin that wraps the the Pillow library. Pillow is a friendly fork of PIL
(Python Image Library) and supports reading and writing of common formats (jpg,
png, gif, tiff, ...). For, the complete list of features and supported formats
please refer to pillows official docs (see the Backend Library link).
Parameters
----------
request : Request
A request object representing the resource to be operated on.
Methods
-------
.. autosummary::
:toctree: _plugins/pillow
PillowPlugin.read
PillowPlugin.write
PillowPlugin.iter
PillowPlugin.get_meta
)BytesIO) CallableOptionalDictAnyTuplecastIteratorUnionListN)ImageUnidentifiedImageError
ImageSequenceExifTags )RequestIOModeInitializationError URI_BYTES)PluginV3ImageProperties) ArrayLike) orientationmodereturnc sd t |dkrdnd dd fdddd fdd fd dd
d fd dd d d
}|| S )N c S s | S N xr r :/tmp/pip-unpacked-wheel-xbmu82vq/imageio/plugins/pillow.py1 z-_exif_orientation_transform..c s t j| dS )NZaxisnpflipr r% r r" r# 2 r$ c S s t j| ddS )Nr kr' rot90r r r r" r# 3 r$ c s t j| d dS )Nr r% r& r r% r r" r# 4 r$ c s t jt j| dd dS )N r) r% r' r( r, r r% r r" r# 5 r$ c S s t j| ddS )Nr r) r+ r r r r" r# 6 r$ c s t jt j| dd dS )Nr r) r% r. r r% r r" r# 7 r$ c S s t j| ddS )Nr- r) r+ r r r r" r# 8 r$ )r r r- )r getmodebands)r r ZEXIF_ORIENTATIONr r% r" _exif_orientation_transform) s
r5 c s e Zd Zedd fdd ZddddZdddddd ejdd
d Zdddd e e
e
e ej d
ddZ ejdddZ
ddddeeee f e e e
ee dddZddee ef dddZd!ee
ee ef dd d Zd"eed d d Z ZS )# PillowPluginN)requestr c s t | d| _|jjtjkrz t| W 5 Q R X W n> t
k
r| |j t krdt
ddnt
d|j ddY nX t| j | _nl| jjp| jj}|dkrtdt dS tjtjg}|D ] }| |t kr dS qt
d| dddS )zInstantiate a new Pillow Plugin Object
Parameters
----------
request : {Request}
A request object representing the resource to be operated on.
Nz'Pillow can not read the provided bytes.zPillow can not read .zCan't determine file format to write as. You _must_ set `format` during write or the call will fail. Use `extension` to supress this warning. zPillow can not write `z` files.)super__init___imager Zio_moder readr openget_filer
_uri_typer r Zraw_uri_requestr7 extension format_hintwarningswarn UserWarningZpreinitinitregistered_extensionskeys)selfr7 rA ZtirageZ
format_loader __class__r r" r: ? sH
zPillowPlugin.__init__)r c C s | j r| j | j d S r )r; closer@ finish)rI r r r" rL t s
zPillowPlugin.closeF)indexr rotate apply_gammaas_grayc C s |dk rt d|dkr@| jjdkr*t}n| jjdkr s z%PillowPlugin.read..r% ) TypeErrorr; formatEllipsiscustom_mimetype
isinstanceintseek_apply_transformsiterr' stack)rI rN r rO rP rQ imageiteratorr r r" r< z s" +
zPillowPlugin.readrT )r rO rP r c c s( t | jD ]}| ||||V q dS )a
Iterate over all ndimages/frames in the URI
Parameters
----------
mode : {str, None}
Convert the image to the given mode before returning it. If None,
the mode will be left unchanged. Possible modes can be found at:
https://pillow.readthedocs.io/en/stable/handbook/concepts.html#modes
rotate : {bool}
If set to ``True`` and the image contains an EXIF orientation tag,
apply the orientation before returning the ndimage.
apply_gamma : {bool}
If ``True`` and the image contains metadata about gamma, apply gamma
correction to the image.
N)r r r; r_ )rI r rO rP rV r r r" r` s zPillowPlugin.iterc
C s |d k r| |}n|jdkr,| |jj}t|}| j| j dd}|rpd|krpt |d | jj}||}|rd|krt
|d }t
|j tj krdnd}d} || | | | d }t
|tj}|S )
NrR FrN exclude_applied Orientationgammai g ?g`_
for each writer.
Notes
-----
When writing batches of very narrow (2-4 pixels wide) gray images set
the ``mode`` explicitly to avoid the batch being identified as a colored
image.
ZfpszvThe keyword `fps` is no longer supported. Use `duration`(in ms) instead, e.g. `fps=50` == `duration=20` (1000 * 1/50).rY r r% TNr r- r Fr z*Can't write images with one color channel.)r r- r/ )N.)r bits)colorsZsave_allZ
append_images)rX r7 rA rB r rG r\ listr' ra rk r4 ndimshape
ValueErrorZ fromarrayquantizeappendupdatesaver@ r> r? r r r getvalue)rI rs r rY rr kwargsrA Z save_argsZ
pil_framesframeZ pil_frameZ
primary_imageZ other_imagesfiler r r" write sT 6
zPillowPlugin.writer )rN c C s | j |ddS )NFrd )rl )rI rN r r r" get_meta_ s zPillowPlugin.get_metaT)rN re r c C s |dkr0| j jdkrt}n| j jdkr,t}nd}t|trT| j |krT| j | | j j }| j j
|d< | j j |d< | j j
dkr| j j |d< | j
rd d
t| j
D }|d d || |r|d d |S )
a Read ndimage metadata.
Parameters
----------
index : {integer, None}
If the ImageResource contains multiple ndimages, and index is an
integer, select the index-th ndimage from among them and return its
metadata. If index is an ellipsis (...), read and return global
metadata. If index is None, this plugin reads metadata from the
first image of the file (index=0) unless the image is a GIF or APNG,
in which case global metadata is read (index=...).
Returns
-------
metadata : dict
A dictionary of format-specific metadata.
NrR rS r r rx Prj c S s i | ]\}}t j|d |qS )unknown)r ZTAGSget)rU keyvaluer r r"
s z)PillowPlugin.metadata..r rf )r; rY rZ r[ r\ r] rm r^ infocopyr sizerj Zgetexifdictitemspopr| )rI rN re rl Z exif_datar r r" rl b s,
zPillowPlugin.metadata)rN r c C s |dkr0| j jdkrt}n| j jdkr,t}nd}|tkrF| j d n | j | | j jdkrj| j jj}n| j j}| j j}| j j}||f}| j j }|dkr|f|}t
t
|d}|j}t|dkr||dd }t||j|tkrdnd d
S ) a Standardized ndimage metadata
Parameters
----------
index : int
If the ImageResource contains multiple ndimages, and index is an
integer, select the index-th ndimage from among them and return its
properties. If index is an ellipsis (...), read and return the
properties of all ndimages in the file stacked along a new batch
dimension. If index is None, this plugin reads and returns the
properties of the first image (index=0) unless the image is a GIF or
APNG, in which case it reads and returns the properties all images
(index=...).
Returns
-------
properties : ImageProperties
A dataclass filled with standardized image metadata.
Notes
-----
This does not decode pixel data and is 394fast for large images.
NrR rS r .)r r r TF)rx ro rr )r; rY rZ r[ r^ rj r widthheightn_framesr' rk r newrx lenr ro ) rI rN r r r rx r dummyZ pil_shaper r r"
properties s6
zPillowPlugin.properties)NT)N) __name__
__module__ __qualname__r r: rL r' Zndarrayr< strboolr r` r_ r
r r r bytesr r r r r] rl r r
__classcell__r r rJ r" r6 > sN 5 D r
5r6 )"__doc__ior typingr r r r r r r r
r Znumpyr' ZPILr r
r r Z core.requestr r r r Zcore.v3_plugin_apir r rC r r] r r5 r6 r r r r" s ,