o
    sÇh	  ã                   @  sÎ   d dl mZ d dlZd dlZd dlZd dlmZmZmZ ddl	m
Z
mZmZmZmZ ddl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ÚAnyÚcasté   )ÚImageÚ	ImageFileÚImageSequenceÚJpegImagePluginÚTiffImagePlugin)Úo32le)ÚDeferredErrorÚimúImage.ImageÚfpú	IO[bytes]Úfilenameústr | bytesÚreturnÚNonec                 C  s   t  | ||¡ d S ©N)r
   Ú_save)r   r   r   © r   úf/var/www/html/construction_image-detection-poc/venv/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é   Ú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   Ú
mpf_offsetÚoffsetsÚ
imSequenceÚim_framer   ÚifdÚ	mpentriesÚdata_offsetÚiÚsizeÚmptyper   r   r   Ú	_save_all*   sN   ÿ
ï

rF   c                   @  s^   e Zd ZdZdZdZddd„Zdddd„Zddd„Zddd„Z	ddd„Z
e	dd dd„ƒZdS )!ÚMpoImageFileÚMPOzMPO (CIPA DC-007)Fr   r   c                 C  s$   | j  d¡ tj | ¡ |  ¡  d S )Nr   )r   r8   r
   ÚJpegImageFileÚ_openÚ_after_jpeg_open©Úselfr   r   r   rJ   e   s   zMpoImageFile._openNÚmpheaderúdict[int, Any] | Nonec                   s´   |d ur|nˆ   ¡ ˆ _ˆ jd u rd}t|ƒ‚ˆ jd ˆ _‡ fdd„ˆ jd D ƒˆ _dˆ jd< ˆ jtˆ jƒks8J ‚ˆ jd= ˆ jdkˆ _ˆ jˆ _	ˆ j	 
ˆ jd ¡ dˆ _dˆ _dˆ _d S )	Nz(Image appears to be a malformed MPO filer#   c                   s   g | ]}|d  ˆ j d  ‘qS )Ú
DataOffsetÚmpoffset)Úinfo)Ú.0ÚmpentrL   r   r   Ú
<listcomp>p   s    ÿz1MpoImageFile._after_jpeg_open.<locals>.<listcomp>r%   r   rQ   r   )Ú_getmpÚmpinfoÚ
ValueErrorÚn_framesÚ_MpoImageFile__mpoffsetsr2   rR   r   r   Ú_fpr8   Ú_MpoImageFile__frameÚoffsetÚreadonly)rM   rN   Úmsgr   rL   r   rK   j   s"   

ÿ

zMpoImageFile._after_jpeg_openÚposÚintc                 C  s$   t | jtƒr
| jj‚| j |¡ d S r   )r/   r[   r   Úexr8   )rM   r`   r   r   r   Ú	load_seek€   s   zMpoImageFile.load_seekÚframec                 C  sà   |   |¡sd S t| jtƒr| jj‚| j| _| j| | _| j 	d¡}d| jv r*| jd= | j 
| jd ¡ | j d¡s?d}t|ƒ‚| j 
| j¡ tj | ¡ | j 	d¡|krX|  ¡  t dd| j | j| jd d ¡g| _|| _d S )Nr   é   zNo data found for frameÚjpeg)r   r   r   r"   )Ú_seek_checkr/   r[   r   rb   r   rZ   r]   rR   r(   r8   ÚreadrX   r
   rI   rJ   Ú_reload_exifr   Ú_TilerD   Útiler\   )rM   rd   Úoriginal_exifr_   r   r   r   r8   …   s(   

 ÿ
zMpoImageFile.seekc                 C  s   | j S r   )r\   rL   r   r   r   r4   Ÿ   s   zMpoImageFile.tellÚjpeg_instanceúJpegImagePlugin.JpegImageFilec                 C  s   t | _t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.
        )rG   Ú	__class__r   rK   )rm   rN   Úmpo_instancer   r   r   Úadopt¢   s   

zMpoImageFile.adopt)r   r   r   )rN   rO   r   r   )r`   ra   r   r   )rd   ra   r   r   )r   ra   )rm   rn   rN   rO   r   rG   )Ú__name__Ú
__module__Ú__qualname__ÚformatÚformat_descriptionÚ!_close_exclusive_fp_after_loadingrJ   rK   rc   r8   r4   Ústaticmethodrq   r   r   r   r   rG   `   s    



þrG   z.mpoz	image/mpo)r   r   r   r   r   r   r   r   )Ú
__future__r   r*   r:   r-   Útypingr   r   r   Ú r   r   r	   r
   r   Ú_binaryr   Ú_utilr   r   rF   rI   rG   Úregister_saveru   Úregister_save_allÚregister_extensionÚregister_mimer   r   r   r   Ú<module>   s   

6^