B
    -¦a,  ã               @   sp   d dl 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d	¡ e e
jd
¡ dS )é   )ÚImageÚ	ImageFileÚJpegImagePlugin)Úi16bec             C   s
   t  | ¡S )N)r   Ú_accept)Úprefix© r   ú6/tmp/pip-unpacked-wheel-rysjrues/PIL/MpoImagePlugin.pyr      s    r   c             C   s   t  | ||¡S )N)r   Ú_save)ZimÚfpÚfilenamer   r   r	   r
      s    r
   c               @   sX   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
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ž   |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 )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   )Z_getmpÚmpinfoZn_framesÚ_MpoImageFile__mpoffsetsÚlenÚAssertionErrorr   Zis_animatedr   Ú_MpoImageFile__fpr   Ú_MpoImageFile__frameÚoffsetÚreadonly)r   Úmpheaderr   )r   r	   r   1   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ƒ‚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°  Ú	AttributeZMPTypezLarge Thumbnailii‡  i   i   Zjpeg)r   r   Ú )Z_seek_checkr   r   r   r   r   ÚreadÚ
ValueErrorÚi16r   Ú
_safe_readr   r   Ú
startswithZgetexifZget_ifdÚ_sizeÚsizeÚmodeZtiler   )r   ÚframeÚsegmentÚnZmptyper$   r   r   r	   r   G   s(    


zMpoImageFile.seekc             C   s   | j S )N)r   )r   r   r   r	   Útell`   s    zMpoImageFile.tellc             C   sB   z4y| j | jkr| j  ¡  W n tk
r0   Y nX W d d | _ X d S )N)r   r   ÚcloseÚAttributeError)r   r   r   r	   Ú
_close__fpc   s    
zMpoImageFile._close__fpc             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	   Úadoptl   s    
zMpoImageFile.adopt)N)N)Ú__name__Ú
__module__Ú__qualname__ÚformatÚformat_descriptionZ!_close_exclusive_fp_after_loadingr   r   r"   r   r2   r5   Ústaticmethodr7   r   r   r   r	   r   &   s   
	r   z.mpoz	image/mpoN)r&   r   r   r   Ú_binaryr   r)   r   r
   r   r   Zregister_saver;   Zregister_extensionZregister_mimer   r   r   r	   Ú<module>   s   	^