B
    àz÷bí  ã               @   sh   d dl mZmZmZ d dlmZ dd„ ZG dd„ dejƒZ	e 
e	je¡ e e	jd¡ e e	jd¡ d	S )
é   )ÚImageÚ	ImageFileÚJpegImagePlugin)Úi16bec             C   s   t  | ||¡S )N)r   Ú_save)ZimÚfpÚfilename© r	   ú1lib/python3.7/site-packages/PIL/MpoImagePlugin.pyr      s    r   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 )Né    )r   Úseekr   ÚJpegImageFileÚ_openÚ_after_jpeg_open)Úselfr	   r	   r
   r   +   s    zMpoImageFile._openNc                s¦   ˆ j ˆ _|d k	r|nˆ  ¡ ˆ _ˆ jd ˆ _‡ fdd„ˆ jd D ƒˆ _dˆ jd< ˆ jtˆ jƒksbt‚ˆ jd= ˆ jdkˆ _	ˆ j
ˆ _ˆ j ˆ jd ¡ dˆ _dˆ _dˆ _d S )Ni°  c                s   g | ]}|d  ˆ j d  ‘qS )Z
DataOffsetÚmpoffset)Úinfo)Ú.0Zmpent)r   r	   r
   ú
<listcomp>5   s    z1MpoImageFile._after_jpeg_open.<locals>.<listcomp>i°  r   r   r   )ÚsizeÚ_initial_sizeZ_getmpÚmpinfoZn_framesÚ_MpoImageFile__mpoffsetsÚlenÚAssertionErrorr   Zis_animatedr   Ú_fpr   Ú_MpoImageFile__frameÚoffsetÚreadonly)r   Úmpheaderr	   )r   r
   r   0   s    
zMpoImageFile._after_jpeg_openc             C   s   | j  |¡ d S )N)r   r   )r   Úposr	   r	   r
   Ú	load_seekD   s    zMpoImageFile.load_seekc             C   s$  |   |¡sd S | j| _| j| | _| j | jd ¡ | j d¡}|sLtdƒ‚| j| _	t
|ƒdkrât
| j d¡ƒd }t | j|¡| jd< |  ¡  | jd | d d }| d¡rü|  ¡  d	¡}d
|krüd|krü|d
 |d f| _	nd| jkrü| jd= |  ¡  dd| j | j| jdffg| _|| _d S )Né   zNo data found for frameiáÿ  Úexifi°  Z	AttributeZMPTypezLarge Thumbnailii‡  i   i   Zjpeg)r   r   Ú )Z_seek_checkr   r   r   r   r   ÚreadÚ
ValueErrorr   Z_sizeÚi16r   Z
_safe_readr   Z_reload_exifr   Ú
startswithZgetexifZget_ifdr   ÚmodeZtiler   )r   ÚframeZsegmentÚnZmptyper$   r	   r	   r
   r   G   s.    


zMpoImageFile.seekc             C   s   | j S )N)r   )r   r	   r	   r
   Útellc   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.
        )r   Ú	__class__r   )Zjpeg_instancer    r	   r	   r
   Úadoptf   s    
zMpoImageFile.adopt)N)N)Ú__name__Ú
__module__Ú__qualname__ÚformatZformat_descriptionZ!_close_exclusive_fp_after_loadingr   r   r"   r   r-   Ústaticmethodr/   r	   r	   r	   r
   r   %   s   
r   z.mpoz	image/mpoN)r%   r   r   r   Z_binaryr   r(   r   r   r   Zregister_saver3   Zregister_extensionZregister_mimer	   r	   r	   r
   Ú<module>   s   	Y