o
    á©Zh×  ã                   @  s¶   d dl mZ d dlZd dlZd dlZd dlmZ ddlmZm	Z	m
Z
mZ ddlmZ ddd„Zddd„ZG dd„ de
jƒZe eje¡ e eje¡ e ejd¡ e ejd¡ dS )é    )ÚannotationsN)ÚIOé   )ÚImageÚ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   úI/var/www/html/lang_env/lib/python3.10/site-packages/PIL/MpoImagePlugin.pyr   $   s   r   c              
   C  s²  | j  dg ¡}|st| ddƒst| ||ƒ d S d}g }t | g|¡D ]Z}t |¡D ]R}|sjdt 	dd¡ 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 ]%\}}|dkržd}nd}|
t 	d|||dd¡7 }
|dkr´||8 }||7 }q“|
|	d< | |¡ | dtdƒ |	 d¡ ¡ | dtj¡ d S )NÚappend_imagesÚis_animatedFé   s   ÿâz>HéX   s   MPF sR                                                                                     ÚextraÚexifé   ZJPEGéÿÿÿÿs   0100i °  é°  ó    r   i   z<LLLHHé°  s   II* é   )ZencoderinfoÚgetÚgetattrr   Ú	itertoolsÚchainr   ÚIteratorÚstructÚpackÚ
isinstancer   ZExifÚtobytesÚlenr   ÚappendÚtellÚsaver   ZImageFileDirectory_v2Ú	enumerateÚseekÚwriter	   ÚosÚSEEK_END)r
   r   r   r   Z
mpf_offsetÚoffsetsZ
imSequenceZim_framer   ZifdZ	mpentriesZdata_offsetÚiÚsizeZmptyper   r   r   Ú	_save_all(   sN   ÿ
ï

r7   c                   @  sX   e Zd ZdZdZdZddd„Zdd	d
„Zddd„Zddd„Z	ddd„Z
eddd„ƒZdS )ÚMpoImageFileZMPOzMPO (CIPA DC-007)Fr   r   c                 C  s$   | j  d¡ tj | ¡ |  ¡  d S )Nr   )r   r0   r   ÚJpegImageFileÚ_openÚ_after_jpeg_open©Úselfr   r   r   r:   c   s   zMpoImageFile._openNc                   sž   |d ur|nˆ   ¡ ˆ _ˆ jd ˆ _‡ fdd„ˆ jd D ƒˆ _dˆ jd< ˆ jtˆ jƒks-J ‚ˆ jd= ˆ jdkˆ _ˆ jˆ _ˆ j 	ˆ jd ¡ dˆ _
dˆ _dˆ _d S )Nr   c                   s   g | ]}|d  ˆ j d  ‘qS )Z
DataOffsetÚmpoffset)Úinfo)Ú.0Zmpentr<   r   r   Ú
<listcomp>k   s    ÿz1MpoImageFile._after_jpeg_open.<locals>.<listcomp>r    r   r>   r   )Z_getmpZmpinfoZn_framesÚ_MpoImageFile__mpoffsetsr+   r?   r   r   Ú_fpr0   Ú_MpoImageFile__frameÚoffsetÚreadonly)r=   Úmpheaderr   r<   r   r;   h   s   
ÿ

zMpoImageFile._after_jpeg_openÚposÚintc                 C  s   | j  |¡ d S r   )rC   r0   )r=   rH   r   r   r   Ú	load_seek{   s   zMpoImageFile.load_seekÚframec                 C  sÈ   |   |¡sd S | j| _| j| | _| j d¡}d| jv r | jd= | j | jd ¡ | j d¡s5d}t	|ƒ‚| j | j¡ t
j | ¡ | j d¡|krN|  ¡  dd| j | j| jd d fg| _|| _d S )Nr   é   zNo data found for frameZjpeg)r   r   r   r   )Z_seek_checkrC   r   rB   rE   r?   r"   r0   ÚreadÚ
ValueErrorr   r9   r:   Z_reload_exifr6   ZtilerD   )r=   rK   Zoriginal_exifÚmsgr   r   r   r0   ~   s"   

"
zMpoImageFile.seekc                 C  s   | j S r   )rD   r<   r   r   r   r-   ”   s   zMpoImageFile.tellc                 C  s   t | _|  |¡ | 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.
        )r8   Ú	__class__r;   )Zjpeg_instancerG   r   r   r   Úadopt—   s   
zMpoImageFile.adopt)r   r   r   )rH   rI   r   r   )rK   rI   r   r   )r   rI   )Ú__name__Ú
__module__Ú__qualname__ÚformatÚformat_descriptionZ!_close_exclusive_fp_after_loadingr:   r;   rJ   r0   r-   ÚstaticmethodrQ   r   r   r   r   r8   ^   s    




r8   z.mpoz	image/mpo)r
   r   r   r   r   r   r   r   )Ú
__future__r   r$   r2   r'   Útypingr   Ú r   r   r   r   Ú_binaryr	   r   r7   r9   r8   Zregister_saverU   Zregister_save_allZregister_extensionZregister_mimer   r   r   r   Ú<module>   s   

6Q