o
�J�h
� @ s� d dl mZ d dlZd dlZd dlZd dlmZmZmZ ddl m
Z
m Z m Z m
Z
mZ ddlmZ ddd�Zddd�ZG dd� de
j�Ze
�eje� e
�eje� e
�ejd� e
�ejd� dS )� )� annotationsN)�IO�Any�cast� )�Image� ImageFile�
ImageSequence�JpegImagePlugin�TiffImagePlugin)�o32le�im� Image.Image�fp� IO[bytes]�filename� str | bytes�return�Nonec C s t �| ||� d S �N)r
�_save)r
r r � r �PC:\pinokio\api\whisper-webui.git\app\env\lib\site-packages\PIL\MpoImagePlugin.pyr % s r c
C s� | j �dg �}|st| dd�st| ||� d S d}g }t�| g|�D ]Z}t�|�D ]R}|sjdt� dd� d d |j d
< |j �d �}t
|t j �rQ|�
� }||j d < |r[|d t|� 7 }t�|||� |�|�� � q)|�|d
� |�|�� |d � q)q"t�� } d| d< t|�| d< d}
d} t|�D ]%\} }
| dkr�d}nd}|
t� d||
| dd�7 }
| dkr�| |8 } | |
7 } q�|
| d< |�|� |�dtd� | �
d� � |�dtj� d S )N�
append_images� is_animatedF� s ��z>H�X s MPF sR �extra�exif� �JPEG�����s 0100i � �� � r i z<LLLHH�� s II* � )� encoderinfo�get�getattrr � itertools�chainr �Iterator�struct�pack�
isinstancer �Exif�tobytes�lenr
�append�tell�saver �ImageFileDirectory_v2� enumerate�seek�writer �os�SEEK_END)r
r r r Z
mpf_offset�offsets�
imSequence�im_framer �ifd� mpentriesZ data_offset�i�sizeZmptyper r r � _save_all) sN �
�
rB c @ s^ e Zd ZdZdZdZddd�Zdd d d �Zd dd�Zd dd�Z d dd�Z
e dd dd��Z dS )!� MpoImageFileZMPOzMPO (CIPA DC-007)Fr r c C s$ | j �d� tj�| � | �� d S )Nr )r r7 r
�
JpegImageFile�_open�_after_jpeg_open��selfr r r rE d s zMpoImageFile._openN�mpheader�dict[int, Any] | Nonec s� |d ur|n� � � � _� jd u rd}t|��� jd � _� fdd�� jd D �� _d� jd< � jt� j�ks8J �� jd= � jdk� _� j� _ � j �
� jd � d� _ d� _ d� _
d S ) Nz(Image appears to be a malformed MPO filer" c s g | ] }|d � j d �qS )�
DataOffset�mpoffset)�info)�.0ZmpentrG r r �
<listcomp>o s �z1MpoImageFile._after_jpeg_open.<locals>.<listcomp>r$ r rL r )�_getmpZmpinfo�
ValueError�n_frames�_MpoImageFile__mpoffsetsr1 rM r r �_fpr7 �_MpoImageFile__frame�offset�readonly)rH rI �msgr rG r rF i s"
�
z MpoImageFile._after_jpeg_open�pos�intc C s | j �|� d S r )rT r7 )rH rY r r r � load_seek s zMpoImageFile.load_seek�framec C s� | � |�sd S | j| _| j| | _| j�d�}d| jv r | jd= | j�| jd � | j�d�s5d}t |��| j�| j� t
j � | � | j�d�|krN| �
� t�dd| j | j| jd d �g| _|| _d S )Nr � zNo data found for frame�jpeg)r r r r! )� _seek_checkrT r rS rV rM r' r7 �readrQ r
rD rE � _reload_exifr �_TilerA �tilerU )rH r\ Z
original_exifrX r r r r7 � s$
�
zMpoImageFile.seekc C s | j S r )rU rG r r r r3 � s zMpoImageFile.tell�
jpeg_instance� JpegImagePlugin.JpegImageFilec C s t | _tt | �}|�|� |S )aC
Transform the instance of JpegImageFile into
an instance of MpoImageFile.
After the call, the JpegImageFile is extended
to be an MpoImageFile.
This is essentially useful when opening a JPEG
file that reveals itself as an MPO, to avoid
double call to _open.
)rC � __class__r rF )rd rI Z mpo_instancer r r �adopt� s
zMpoImageFile.adopt)r r r )rI rJ r r )rY rZ r r )r\ rZ r r )r rZ )rd re rI rJ r rC )
�__name__�
__module__� __qualname__�format�format_description�!_close_exclusive_fp_after_loadingrE rF r[ r7 r3 � staticmethodrg r r r r rC _ s
�rC z.mpoz image/mpo)r
r r r r r r r )�
__future__r r) r9 r, �typingr r r � r r r r
r �_binaryr r rB rD rC �
register_saverk �register_save_all�register_extension�
register_mimer r r r �<module> s
6Z