U
    Evf                     @  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m	Z	 ddl
mZ dd Z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   )ImageImageSequenceJpegImagePluginTiffImagePlugin)o32lec                 C  s   t | || d S N)r   _save)imfpfilename r   6/tmp/pip-unpacked-wheel-ciywl2yd/PIL/MpoImagePlugin.pyr
   #   s    r
   c              
   C  s  | j dg }|sJz
| j}W n tk
r4   d}Y nX |sJt| || d S d}g }t| g|D ]}t|D ]}|sdt	
dd d d |j d	< |j d
}	t|	tjr|	 }	|	|j d
< |	r|dt|	 7 }t||| ||  qn||d || |d   qnq`t }
d|
d< t||
d< d}d}t|D ]P\}}|dkrZd}nd}|t	
d|||dd7 }|dkr||8 }||7 }qB||
d< || |dtd |
d  |dtj d S )Nappend_imagesF   s   z>HX   s   MPF sR                                                                                     extraexif   ZJPEGs   0100i         r   i   z<LLLHH  s   II*    )Zencoderinfogetis_animatedAttributeErrorr
   	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animatedZ
mpf_offsetoffsetsZ
imSequenceZim_framer   ZifdZ	mpentriesZdata_offsetisizeZmptyper   r   r   	_save_all'   sV    





r1   c                   @  sP   e Zd ZdZdZdZdd ZdddZd	d
 Zdd Z	dd Z
edddZdS )MpoImageFileZMPOzMPO (CIPA DC-007)Fc                 C  s$   | j d tj|  |   d S )Nr   )r   r*   r   JpegImageFile_open_after_jpeg_openselfr   r   r   r4   g   s    zMpoImageFile._openNc                   s   |d k	r|n    _ jd  _ fdd jd D  _d jd<  jt jksZt 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mpentr6   r   r   
<listcomp>o   s    z1MpoImageFile._after_jpeg_open.<locals>.<listcomp>r   r   r8   r   )Z_getmpZmpinfoZn_frames_MpoImageFile__mpoffsetsr%   AssertionErrorr9   r   r   _fpr*   _MpoImageFile__frameoffsetreadonly)r7   mpheaderr   r6   r   r5   l   s    

zMpoImageFile._after_jpeg_openc                 C  s   | j | d S r	   )r>   r*   )r7   posr   r   r   	load_seek   s    zMpoImageFile.load_seekc                 C  s   |  |sd S | j| _| j| | _| jd}d| jkr@| jd= | j| jd  | jdsjd}t	|| j| j t
j|  | jd|kr|   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_checkr>   r   r<   r@   r9   r   r*   read
ValueErrorr   r3   r4   Z_reload_exifr0   Ztiler?   )r7   frameZoriginal_exifmsgr   r   r   r*      s"    

"zMpoImageFile.seekc                 C  s   | j S r	   )r?   r6   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.
        )r2   	__class__r5   )Zjpeg_instancerB   r   r   r   adopt   s    
zMpoImageFile.adopt)N)N)__name__
__module____qualname__formatformat_descriptionZ!_close_exclusive_fp_after_loadingr4   r5   rD   r*   r'   staticmethodrK   r   r   r   r   r2   b   s   
r2   z.mpoz	image/mpo)
__future__r   r   r,   r!    r   r   r   r   _binaryr   r
   r1   r3   r2   Zregister_saverO   Zregister_save_allZregister_extensionZregister_mimer   r   r   r   <module>   s   ;Q