U
    ­A›dêq  ã                A   @   sì  d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZddl	m
Z
mZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd	lmZ d
d„ Zdd„ Zdd„ Zdd„ Zdd„ Zddefddefddefddefddefddefd d!efd"d#efd$d%d&efd'd(efd)d*efd+d,efd-d.efd/d0efd1d2efd3d4d5d6d7d8d9d:d;d<d=d>efdd?efd@dAefdBdCefdDdEefdFdGefdHdIefdJdKefdLdMefdNdOefdPdQefdRdSefdTdUefdVdWefdXdYefdZd[efd\d]efd^d_efd`daefdbdcefdddeefdfdgefdhdidjdkdldmdndodpdqdrdsdtduddvefdwœ?Zdxdy„ ZG dzd{„ d{ejƒZd|d}„ Z d~d„ Z!d€d€d�d�d‚dƒd„œZ"d…Z#d dd†d‡œZ$dˆd‰„ Z%dŠd‹„ Z&dŒd�„ Z'dŽd�„ Z(d—d�d‘„Z)e
 *ej+e)e¡ e
 ,ej+e'¡ e
 -ej+d’d“d”d•g¡ e
 .ej+d–¡ dS )˜é    Né   )ÚImageÚ	ImageFile)Úi16be)Úi32be)Úo8)Úo16be©Ú	deprecate)Úpresetsc                 C   s&   t | j d¡ƒd }t | j|¡ d S )Né   )Úi16ÚfpÚreadr   Ú
_safe_read)ÚselfÚmarkerÚn© r   úL/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/PIL/JpegImagePlugin.pyÚSkip8   s    r   c                 C   s  t | j d¡ƒd }t | j|¡}d|d@  }|| j|< | j ||f¡ |dkrä|d d… dkrät |dƒ | jd< }t	|d	ƒ| jd
< z |d }t |dƒt |dƒf}W n t
k
r¸   Y n(X |dkrÌ|| jd< || jd< || jd< �n`|dk�r.|d d… dk�r.d| jk�rD|| jd< | j ¡ | d | _�n|dk�rX|d d… dk�rX|| jd< �nì|dk�r„|d d… dk�r„| j |¡ �nÀ|dk�r¢|d d… dk�r¢d}| j di ¡}	|||d … dk�rDz´|d7 }t ||ƒ}
|d7 }|| }|d| 7 }||d@ 7 }t||ƒ}|d7 }|||| … }|
d k�r^t|d!ƒd" t |dƒt|dƒd" t |dƒd#œ}||	|
< ||7 }||d@ 7 }W n tjk
�rš   Y �qDY nX �q²n¢|d$k�rþ|d d… d%k�rþt |dƒ| jd&< z|d' }W n tk
�rð   Y nX || jd(< nF|dk�rD|d d… d)k�rD|dd … | jd*< | j ¡ | d | jd+< d| jk�rd| jk�rz€|  ¡ }|d, }|d- }zt|d! ƒ|d  }W n tk
�r¨   |}Y nX t |¡�rºt‚|d.k�rÌ|d/9 }||f| jd< W n* tttttfk
�r   d0| jd< Y nX d S )1Nr   zAPP%dé   éàÿ  é   s   JFIFé   Zjfifé   Zjfif_versioné   é   é
   r   ÚdpiÚ	jfif_unitÚjfif_densityéáÿ  s   Exif Úexifé   éâÿ  s   FPXR Zflashpixé   ó   ICC_PROFILE éíÿ  é   s   Photoshop 3.0 Ú	photoshops   8BIMií  r   i   )ZXResolutionZDisplayedUnitsXZYResolutionZDisplayedUnitsYéîÿ  s   AdobeÚadobeé   Úadobe_transforms   MPF ÚmpZmpoffseti(  i  é   gR¸…ëQ@)éH   r1   )r   r   r   r   r   ÚappÚapplistÚappendÚinfoÚdivmodÚ	ExceptionÚtellZ_exif_offsetÚicclistÚ
setdefaultÚi32ÚstructÚerrorÚ
IndexErrorÚgetexifÚfloatÚ	TypeErrorÚmathÚisnanÚ
ValueErrorÚKeyErrorÚSyntaxErrorÚZeroDivisionError)r   r   r   Úsr2   Úversionr    r!   Úoffsetr*   ÚcodeZname_lenÚsizeÚdatar.   r#   Zresolution_unitZx_resolutionr   r   r   r   ÚAPP=   s˜    






ü

rN   c                 C   sJ   t | j d¡ƒd }t | j|¡}|| jd< || jd< | j d|f¡ d S )Nr   ÚcommentÚCOM)	r   r   r   r   r   r5   r2   r3   r4   )r   r   r   rH   r   r   r   rP   ¶   s
    

rP   c           
      C   sœ  t | j d¡ƒd }t | j|¡}t |dƒt |dƒf| _|d | _| jdkrbd| j› d�}t|ƒ‚|d | _| jdkr~d	| _	n:| jdkr�d
| _	n(| jdkr¢d| _	nd| j› d�}t|ƒ‚|dkrÔd | j
d< | j
d< | j�rD| j ¡  | jd d t| jƒk�r0g }| jD ]}| |dd … ¡ �q
d |¡}nd }|| j
d< g | _tdt|ƒdƒD ]B}|||d … }	| j |	d |	d d |	d d@ |	d f¡ �qTd S )Nr   r0   r   r   r   zcannot handle z-bit layersr   ÚLÚRGBr   ÚCMYKz-layer images)éÂÿ  éÆÿ  éÊÿ  éÎÿ  ÚprogressiveÚprogressioné   r)   ó    Úicc_profiler$   é   r   )r   r   r   r   r   Ú_sizeÚbitsrF   ÚlayersÚmoder5   r9   ÚsortÚlenr4   ÚjoinÚrangeÚlayer)
r   r   r   rH   ÚmsgZprofileÚpr\   ÚiÚtr   r   r   ÚSOFÁ   s>    








rk   c                    sÒ   t | j d¡ƒd }t | j|¡}t|ƒrÎ|d }|d dkrBdnd}d|d  }t|ƒ|k rjd}t|ƒ‚t |dkrzdnd|d|… ¡‰ tj	d	kr¤|dkr¤ˆ  
¡  ‡ fd
d„tD ƒ| j|d@ < ||d … }q"d S )Nr   r   r]   r   é@   zbad quantization table markerÚBÚHÚlittlec                    s   g | ]}ˆ | ‘qS r   r   )Ú.0ri   ©rM   r   r   Ú
<listcomp>  s     zDQT.<locals>.<listcomp>r   )r   r   r   r   r   rc   rF   ÚarrayÚsysÚ	byteorderÚbyteswapÚzigzag_indexÚquantization)r   r   r   rH   ÚvÚ	precisionZ	qt_lengthrg   r   rq   r   ÚDQTó   s     r{   ZSOF0zBaseline DCTZSOF1zExtended Sequential DCTZSOF2zProgressive DCTZSOF3zSpatial losslessZDHTzDefine Huffman tableZSOF5zDifferential sequential DCTZSOF6zDifferential progressive DCTZSOF7zDifferential spatial)ZJPGÚ	ExtensionNZSOF9zExtended sequential DCT (AC)ZSOF10zProgressive DCT (AC)ZSOF11zSpatial lossless DCT (AC)ZDACz%Define arithmetic coding conditioningZSOF13z Differential sequential DCT (AC)ZSOF14z!Differential progressive DCT (AC)ZSOF15zDifferential spatial (AC))ZRST0z	Restart 0N)ZRST1z	Restart 1N)ZRST2z	Restart 2N)ZRST3z	Restart 3N)ZRST4z	Restart 4N)ZRST5z	Restart 5N)ZRST6z	Restart 6N)ZRST7z	Restart 7N)ZSOIzStart of imageN)ZEOIzEnd of imageNZSOSzStart of scanzDefine quantization tableZDNLzDefine number of linesZDRIzDefine restart intervalZDHPzDefine hierarchical progressionZEXPzExpand reference componentZAPP0zApplication segment 0ÚAPP1zApplication segment 1ZAPP2zApplication segment 2ZAPP3zApplication segment 3ZAPP4zApplication segment 4ZAPP5zApplication segment 5ZAPP6zApplication segment 6ZAPP7zApplication segment 7ZAPP8zApplication segment 8ZAPP9zApplication segment 9ZAPP10zApplication segment 10ZAPP11zApplication segment 11ZAPP12zApplication segment 12ZAPP13zApplication segment 13ZAPP14zApplication segment 14ZAPP15zApplication segment 15)ZJPG0zExtension 0N)ZJPG1zExtension 1N)ZJPG2zExtension 2N)ZJPG3zExtension 3N)ZJPG4zExtension 4N)ZJPG5zExtension 5N)ZJPG6zExtension 6N)ZJPG7zExtension 7N)ZJPG8zExtension 8N)ZJPG9zExtension 9N)ZJPG10zExtension 10N)ZJPG11zExtension 11N)ZJPG12zExtension 12N)ZJPG13zExtension 13NÚComment)?iÀÿ  iÁÿ  rT   iÃÿ  iÄÿ  iÅÿ  rU   iÇÿ  iÈÿ  iÉÿ  rV   iËÿ  iÌÿ  iÍÿ  rW   iÏÿ  iÐÿ  iÑÿ  iÒÿ  iÓÿ  iÔÿ  iÕÿ  iÖÿ  i×ÿ  iØÿ  iÙÿ  éÚÿ  iÛÿ  iÜÿ  iÝÿ  iÞÿ  ißÿ  r   r"   r%   iãÿ  iäÿ  iåÿ  iæÿ  içÿ  ièÿ  iéÿ  iêÿ  iëÿ  iìÿ  r(   r+   iïÿ  iðÿ  iñÿ  iòÿ  ióÿ  iôÿ  iõÿ  iöÿ  i÷ÿ  iøÿ  iùÿ  iúÿ  iûÿ  iüÿ  éýÿ  iþÿ  c                 C   s   | d d… dkS )Nr0   s   ÿØÿr   )Úprefixr   r   r   Ú_acceptQ  s    r‚   c                   @   sL   e Zd ZdZdZdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ Z	dd„ Z
dd„ ZdS )ÚJpegImageFileÚJPEGzJPEG (ISO 10918)c                 C   sH  | j  d¡}t|ƒs d}t|ƒ‚d}d | _| _g | _i | _i | _i | _	i | _
g | _g | _|d }|dkr„|| j  d¡ }t|ƒ}n| j  d¡}qZ|tk�rt| \}}}|d k	r¼|| |ƒ |dkrö| j}| jdkrØd	}d
d| j d|dffg| _�qD| j  d¡}qZ|dk�s|dk�rd}qZ|dk�r6| j  d¡}qZd}t|ƒ‚qZd S )Nr0   znot a JPEG fileó   ÿr   éÿ   r   r   rS   úCMYK;IÚjpeg©r   r   Ú iÿÿ  i ÿ  zno marker found)r   r   r‚   rF   r_   r`   rf   Z
huffman_dcZ
huffman_acrx   r2   r3   r9   r   ÚMARKERra   rL   Útile)r   rH   rg   ri   ÚnameÚdescriptionÚhandlerÚrawmoder   r   r   Ú_open^  sH    




zJpegImageFile._openc                 C   s.   | j  |¡}|s*tjr*t| dƒs*d| _dS |S )z›
        internal: read more image data
        For premature EOF and LOAD_TRUNCATED_IMAGES adds EOI marker
        so libjpeg can finish decoding
        Ú_endedTs   ÿÙ)r   r   r   ZLOAD_TRUNCATED_IMAGESÚhasattrr’   )r   Ú
read_bytesrH   r   r   r   Ú	load_read‘  s
    zJpegImageFile.load_readc                 C   sb  t | jƒdkrd S | jrd S | jd \}}}}d}| j}|d dkrZ|dkrZ|| _|df}|�r"t| jd |d  | jd |d  ƒ}dD ]}	||	krŠ qœqŠ|d |d |d |d  |	 d |	 |d  |d |d  |	 d |	 |d  f}| jd |	 d |	 | jd |	 d |	 f| _|	}||||fg| _|df| _dd|d | |d | f}
| j|
fS )	Nr   r   rR   )rQ   ÚYCbCrrŠ   )r   r   r   r   r   r0   )rc   rŒ   ZdecoderconfigrL   ra   Úminr^   )r   ra   rL   ÚdÚeÚoÚaÚscaleZoriginal_sizerH   Úboxr   r   r   Údraft¡  s4    &""ü.
zJpegImageFile.draftc                 C   s¶   t  ¡ \}}t |¡ tj | j¡r:t dd|| jg¡ nd}t	|ƒ‚z*t |¡�}| ¡  |j| _W 5 Q R X W 5 zt 
|¡ W n tk
r”   Y nX X | jj| _| jj| _g | _d S )NZdjpegú-outfilezInvalid Filename)ÚtempfileÚmkstempÚosÚcloseÚpathÚexistsÚfilenameÚ
subprocessÚ
check_callrD   ÚunlinkÚOSErrorr   ÚopenÚloadÚimra   rL   r^   rŒ   )r   Úfr¤   rg   Z_imr   r   r   Ú
load_djpegÅ  s"    


zJpegImageFile.load_djpegc                 C   s   t | ƒS ©N)Ú_getexif©r   r   r   r   r±   ß  s    zJpegImageFile._getexifc                 C   s   t | ƒS r°   )Ú_getmpr²   r   r   r   r³   â  s    zJpegImageFile._getmpc                 C   sB   | j D ]6\}}|dkr| dd¡\}}|dkr|  |¡  S qi S )z™
        Returns a dictionary containing the XMP tags.
        Requires defusedxml to be installed.

        :returns: XMP tags in a dictionary.
        r}   ó    r   s   http://ns.adobe.com/xap/1.0/)r3   ÚrsplitZ_getxmp)r   ÚsegmentÚcontentr   Zxmp_tagsr   r   r   Úgetxmpå  s    zJpegImageFile.getxmpN)Ú__name__Ú
__module__Ú__qualname__ÚformatÚformat_descriptionr‘   r•   rž   r¯   r±   r³   r¸   r   r   r   r   rƒ   Z  s   3$rƒ   c                 C   s   d| j krd S |  ¡  ¡ S )Nr#   )r5   r?   Z_get_merged_dictr²   r   r   r   r±   õ  s    
r±   c              
   C   s<  z| j d }W n tk
r$   Y d S X t |¡}| d¡}|d d… dkrNdnd}ddlm} z,| |¡}| |j	¡ | 
|¡ t|ƒ}W n0 tk
rº } zd	}	t|	ƒ|‚W 5 d }~X Y nX z|d
 }
W n0 tk
rø } zd}	t|	ƒ|‚W 5 d }~X Y nX g }�z|d }td|
ƒD ]ä}t |› d�||d ¡}d}tt||ƒƒ}t|d d@ ƒt|d d@ ƒt|d d@ ƒ|d d@ d? |d d@ d? |d d@ dœ}|d dk�r°d|d< nd}	t|	ƒ‚ddd d!d"d#d$d%œ}| |d& d'¡|d&< ||d< | |¡ �q||d< W n2 tk
�r6 } zd(}	t|	ƒ|‚W 5 d }~X Y nX |S ))Nr/   r   r   s   MM *ú>ú<r   )ÚTiffImagePluginz)malformed MP Index (unreadable directory)é°  z(malformed MP Index (no number of images)i°  r   ZLLLHHr]   )Ú	AttributeÚSizeZ
DataOffsetZEntryNo1ZEntryNo2rÂ   l        i   @i    i   é   i   é   iÿÿÿ )ZDependentParentImageFlagZDependentChildImageFlagZRepresentativeImageFlagÚReservedÚImageDataFormatÚMPTyperÇ   r„   z!unsupported picture format in MPOÚ	Undefinedz Large Thumbnail (VGA Equivalent)z$Large Thumbnail (Full HD Equivalent)zMulti-Frame Image (Panorama)zMulti-Frame Image: (Disparity)z Multi-Frame Image: (Multi-Angle)zBaseline MP Primary Image)r   i  i  i  i  i  i   rÈ   ÚUnknownz!malformed MP Index (bad MP Entry))r5   rE   ÚioÚBytesIOr   rŠ   rÀ   ZImageFileDirectory_v2ÚseekÚnextr¬   Údictr7   rF   re   r<   Úunpack_fromÚzipÚboolÚgetr4   )r   rM   Úfile_contentsÚheadZ
endiannessrÀ   r5   r/   r™   rg   ZquantZ	mpentriesZrawmpentriesZentrynumZunpackedentryÚlabelsZmpentryZmpentryattrZ	mptypemapr   r   r   r³   û  sv    



  ÿ
ú
ù	r³   rQ   rR   r‡   r–   )Ú1rQ   rR   ZRGBXrS   r–   )@r   r   r   r$   r)   r   rÄ   é   r   r   r   rZ   r]   é   é   é*   r0   r   r&   é   é   é   é)   é+   é	   r-   é   rÅ   é   é(   é,   é5   r   é   é   é    é'   é-   é4   é6   é   é   é!   é&   é.   é3   é7   é<   é   é"   é%   é/   é2   é8   é;   é=   é#   é$   é0   é1   é9   é:   é>   é?   r   ))r   r   r   r   r   r   )r   r   r   r   r   r   )r   r   r   r   r   r   c                 C   s   t dddd� | S )NÚconvert_dict_qtablesr   zConversion is no longer needed)Úactionr	   ©Úqtablesr   r   r   r  g  s    r  c                 C   sZ   t | dƒr| jdkrdS | jd dd… | jd dd…  | jd dd…  }t |d¡S )Nr`   )r   r   éÿÿÿÿr   r   r0   r   )r“   r`   rf   Ú	samplingsrÓ   )r­   Zsamplingr   r   r   Úget_samplingl  s    6r  c                 C   sþ  | j dks| jdkr d}t|ƒ‚zt| j }W n: tk
rh } zd| j› d�}t|ƒ|‚W 5 d }~X Y nX | j}dd„ | dd¡D ƒ}| d	d
¡}| dd
¡}	| d¡}
|dkr¾d
}d}	d}
n„|t	krêt	| }d
}| dd
¡}	| d¡}
nXt
|tƒ�sd}t|ƒ‚n>|	t	k�rt	|	  dd
¡}	t
|
tƒ�rB|
t	k�rBt	|
  d¡}
|	dk�rRd}	nZ|	dk�rbd}	nJ|	dk�rrd}	n:|	dk�r‚d}	n*|	dk�r¬| jdk�r¤d}t|ƒ‚t| ƒ}	dd„ }|
dk�râ| jdk�rÖd}t|ƒ‚t| dd ƒ}
||
ƒ}
| dd¡}d}| d¡}|�r–d}|| }g }|�rB| |d |… ¡ ||d … }�qd}|D ]J}td| t|ƒ ƒ}|d| d  t|ƒ tt|ƒƒ | 7 }|d7 }�qJ| d!| j d!¡¡}| d"d#¡�pÂ| d$d#¡}| d%d#¡}| d&d¡}t
|tjƒ�rò| ¡ }t|ƒ|k�rd'}t|ƒ‚||| d(d¡|| d)d¡|d |d |	|
|||f| _d}|�sR|�rº| jd*k�rxd+| jd  | jd  }nB|d,k�sŒ|d
k�r¦d| jd  | jd  }n| jd | jd  }ttj|t|ƒd- t|ƒd ƒ}t | |d.d| j d|fg|¡ d S )/Nr   z cannot write empty image as JPEGzcannot write mode z as JPEGc                 S   s   g | ]}t |ƒ‘qS r   )Úround)rp   Úxr   r   r   rr   ‡  s     z_save.<locals>.<listcomp>r   r‰   Úqualityr
  Úsubsamplingr	  Zkeeprx   zInvalid quality settingz4:4:4z4:2:2r   z4:2:0r   z4:1:1r„   z3Cannot use 'keep' when original image is not a JPEGc                    sh  ˆd krˆS t ˆtƒr|zdd„ ˆ ¡ D ƒ‰ W n0 tk
r\ } zd}t|ƒ|‚W 5 d }~X Y n X ‡ fdd„tdtˆ ƒdƒD ƒ‰t ˆtttfƒ�rdt ˆtƒr´‡fdd„ttˆƒƒD ƒ‰nt ˆtƒrÆtˆƒ‰dtˆƒ  k rÞdk sìn d	}t|ƒ‚t	ˆƒD ]j\}}z"t|ƒdk�rt
‚t d
|¡}W n2 t
k
�rP } zd}t|ƒ|‚W 5 d }~X Y qôX t|ƒˆ|< qôˆS d S )Nc                 S   s.   g | ]&}|  d d¡d   ¡ D ]}t|ƒ‘qqS )ú#r   r   )ÚsplitÚint)rp   ÚlineÚnumr   r   r   rr   ´  s    þz3_save.<locals>.validate_qtables.<locals>.<listcomp>zInvalid quantization tablec                    s   g | ]}ˆ ||d  … ‘qS )rl   r   )rp   rH   )Úlinesr   r   rr   ½  s     r   rl   c                    s   g | ]}|ˆ krˆ | ‘qS r   r   )rp   Úkeyr  r   r   rr   À  s     r   z$None or too many quantization tablesrn   )Ú
isinstanceÚstrÚ
splitlinesrD   re   rc   ÚtupleÚlistrÏ   Ú	enumeraterA   rs   )r	  r™   rg   ÚidxÚtabler   )r  r	  r   Úvalidate_qtables¯  s>    
þ



ÿ
z_save.<locals>.validate_qtablesÚextrar[   r€   r\   r)   s   ÿâr'   rO   rX   FrY   Úoptimizer#   zEXIF data is too longZsmoothZ
streamtyperS   r   é_   r   rˆ   )ÚwidthÚheightrD   ÚRAWMODEra   rE   rª   ZencoderinforÓ   r   r  r  r  r¼   r  Úgetattrr4   Úo16rc   r   r5   r   ZExifÚtobytesZencoderconfigrL   Úmaxr   ZMAXBLOCKÚ_save)r­   r   r¦   rg   r�   r™   r5   r   r  r  r	  Úpresetr   r!  ZMAX_BYTES_IN_MARKERr\   ZICC_OVERHEAD_LENZMAX_DATA_BYTES_IN_MARKERÚmarkersri   r   rL   rO   rX   r"  r#   Úbufsizer   r   r   r+  z  sÖ    







%

ÿþý
üûÿ

ô r+  c                 C   sB   |   ¡ }t dd||g¡ zt |¡ W n tk
r<   Y nX d S )NZcjpegrŸ   )Ú_dumpr§   r¨   r¢   r©   rª   )r­   r   r¦   r    r   r   r   Ú_save_cjpeg)  s    r0  c              	   C   st   t | |ƒ}z0| ¡ }|d dkr8ddlm} | ||¡}W n4 ttfk
rR   Y n tk
rn   t 	d¡ Y nX |S )NrÁ   r   )ÚMpoImageFilezTImage appears to be a malformed MPO file, it will be interpreted as a base JPEG file)
rƒ   r³   ZMpoImagePluginr1  ZadoptrA   r>   rF   ÚwarningsÚwarn)r   r¦   r­   Zmpheaderr1  r   r   r   Újpeg_factory5  s    
ÿ
r4  z.jfifz.jpez.jpgz.jpegz
image/jpeg)NN)/rs   rË   rB   r¢   r<   r§   rt   r    r2  rŠ   r   r   Ú_binaryr   r   r   r;   r   r   r(  Z
_deprecater
   ZJpegPresetsr   r   rN   rP   rk   r{   r‹   r‚   rƒ   r±   r³   r&  rw   r  r  r  r+  r0  r4  Zregister_openr¼   Zregister_saveZregister_extensionsZregister_mimer   r   r   r   Ú<module>"   sÞ   y2ÁC	 Pú
ý 0
