o
    iVz                     @  s  d dl m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
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
Zerbd dlmZmZ ddlmZ dddZdddZdddZ dddZ!dddZ"i ddde!fd d!d"e!fd#d$d%e!fd&d'd(e!fd)d*d+efd,d-d.e!fd/d0d1e!fd2d3d4e!fd5d6d7d8d9e!fd:d;d<e!fd=d>d?e!fd@dAdBefdCdDdEe!fdFdGdHe!fdIdJdKe!fdLdMi dNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbefdcddde"fdedfdgefdhdidjefdkdldme!fdndodpefdqdrdsefdtdudvefi dwdxdyefdzd{d|efd}d~defdddefdddefdddefdddefdddefdddefdddefdddefdddefdddefdddefddddddddddddddddddde fdZ#dddZ$G dd dejZ%dddZ&dddZ'dddddddÜZ(dZ)d dddƜZ*dddʄZ+dddЄZ,	ddddԄZ-e.e%j/e-e$ e0e%j/e, e1e%j/g dբ e2e%j/d֡ dS )    )annotationsN   )Image	ImageFile)i16be)i32be)o8)o16be)presetsF)IOAnyMpoImageFileselfJpegImageFilemarkerintreturnNonec                 C  s&   t | jdd }t| j| d S )N   )i16fpreadr   
_safe_read)r   r   n r   K/home/air/sos/back/venv/lib/python3.10/site-packages/PIL/JpegImagePlugin.pySkip?   s   r   c                 C  s  t | jdd }t| j|}d|d@  }|| j|< | j||f |dkr|drt |d | j	d< }t
|d| j	d	< z|d
 }t |dt |df}W n
 tyZ   Y d S w |dkre|| j	d< n|dkrutdd |D | j	d< || j	d< || j	d< d S |dkr|drd| j	v r| j	d  |dd  7  < d S || j	d< | j | d | _d S |dkr|dr|ddd | j	d< d S |dkr|dr|| j	d< d S |dkr|dr| j| d S |dkr~|dr~d }| j	d!i }	|||d"  d#kr|z]|d"7 }t ||}
|d7 }|| }|d| 7 }||d@ 7 }t||}|d"7 }||||  }|
d$krSt|d%d& t |d"t|dd& t |d'd(|	|
< n||	|
< ||7 }||d@ 7 }W n tjyn   Y d S w |||d"  d#ksd S d S |d)kr|d*rt |d| j	d+< z|d, }W n ty   Y d S w || j	d-< d S |dkr|d.r|d"d  | j	d/< | j | d" | j	d0< d S d S d S )1Nr   APP     s   JFIF   jfif   jfif_version      
   r   dpic                 s  s    | ]}|d  V  qdS )RQ@Nr   ).0dr   r   r   	<genexpr>`   s    zAPP.<locals>.<genexpr>	jfif_unitjfif_density  s   Exif  exif      http://ns.adobe.com/xap/1.0/     xmp  s   FPXR flashpix   ICC_PROFILE   s   Photoshop 3.0    	photoshop   s   8BIMi  r   i      )XResolutionDisplayedUnitsXYResolutionDisplayedUnitsY  s   Adobeadobe   adobe_transforms   MPF mpmpoffset)r   r   r   r   r   appapplistappend
startswithinfodivmod	Exceptiontupletell_exif_offsetspliticclist
setdefaulti32structerror
IndexError)r   r   r   srG   versionr-   r.   offsetr:   codename_lensizedatarD   r   r   r   r   D   s   






r   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   rK   rG   rH   rI   )r   r   r   rX   r   r   r   r`      s
   

r`   c           	      C  s  t | jdd }t| j|}t |dt |df| _| jd ur+| j| jjkr+d | _|d | _	| j	dkr@d| j	 d}t
||d | _| jdkrNd	| _n| jdkrWd
| _n| jdkr`d| _nd| j d}t
||dv ryd | jd< | jd< | jr| j  | jd d t| jkrdd | jD }d|}nd }|| jd< g | _tdt|dD ] }|||d  }| j|d |d d |d d@ |d f qd S )Nr      r   r   r&   zcannot handle z-bit layersr!   LRGBr;   CMYKz-layer images)        progressiveprogression   c                 S  s   g | ]}|d d qS )r9   Nr   )r*   pr   r   r   
<listcomp>   s    zSOF.<locals>.<listcomp>    icc_profiler1      r   )r   r   r   r   r   _size_imr]   imbitsSyntaxErrorlayers_moderK   rR   sortlenjoinrangelayerrI   )	r   r   r   rX   msgprofilero   itr   r   r   SOF   s@   







.r   c                   s   t | jdd }t| j|}t|rl|d }|d dkr!dnd}d|d  }t||k r5d}t|t|dkr=dnd|d|  tj	d	krR|dkrR 
   fd
dtD | j|d@ < ||d  }t|sd S d S )Nr   r   rp   r   @   zbad quantization table markerBHlittlec                   s   g | ]} | qS r   r   )r*   r   r^   r   r   rm          zDQT.<locals>.<listcomp>r   )r   r   r   r   r   ry   ru   arraysys	byteorderbyteswapzigzag_indexquantization)r   r   r   rX   v	precision	qt_lengthr}   r   r   r   DQT   s    r   i  SOF0zBaseline DCTi  SOF1zExtended Sequential DCTre   SOF2zProgressive DCTi  SOF3zSpatial losslessi  DHTzDefine Huffman tablei  SOF5zDifferential sequential DCTrf   SOF6zDifferential progressive DCTi  SOF7zDifferential spatiali  )JPG	ExtensionNi  SOF9zExtended sequential DCT (AC)rg   SOF10zProgressive DCT (AC)i  SOF11zSpatial lossless DCT (AC)i  DACz%Define arithmetic coding conditioningi  SOF13z Differential sequential DCT (AC)rh   SOF14z!Differential progressive DCT (AC)i  SOF15zDifferential spatial (AC)i  )RST0z	Restart 0Ni  )RST1z	Restart 1Ni  )RST2z	Restart 2Ni  )RST3z	Restart 3Ni  )RST4z	Restart 4Ni  )RST5z	Restart 5Ni  )RST6z	Restart 6Ni  )RST7z	Restart 7Ni  )SOIzStart of imageNi  )EOIzEnd of imageN  SOSzStart of scani  zDefine quantization tablei  DNLzDefine number of linesi  DRIzDefine restart intervali  DHPzDefine hierarchical progressioni  EXPzExpand reference componentr    APP0zApplication segment 0r/   APP1zApplication segment 1r5   APP2zApplication segment 2i  APP3zApplication segment 3i  APP4zApplication segment 4i  APP5zApplication segment 5i  APP6zApplication segment 6i  APP7zApplication segment 7i  APP8zApplication segment 8i  APP9zApplication segment 9i  APP10zApplication segment 10i  APP11zApplication segment 11i  APP12zApplication segment 12r8   APP13zApplication segment 13rA   APP14zApplication segment 14i  APP15zApplication segment 15i  )JPG0zExtension 0Ni  )JPG1zExtension 1Ni  )JPG2zExtension 2N)JPG3zExtension 3N)JPG4zExtension 4N)JPG5zExtension 5N)JPG6zExtension 6N)JPG7zExtension 7N)JPG8zExtension 8N)JPG9zExtension 9N)JPG10zExtension 10N)JPG11zExtension 11N)JPG12zExtension 12N)JPG13zExtension 13NComment)i  i  i  i  i  i  i  i  i  i    i  prefixbytesboolc                 C  s
   |  dS )Ns   )rJ   )r   r   r   r   _acceptI  s   
r   c                      sz   e Zd ZdZdZd"ddZd# fdd	Zd$ fddZd%ddZd&ddZ	d"ddZ
d'ddZd"ddZd'd d!Z  ZS )(r   JPEGzJPEG (ISO 10918)r   r   c                 C  sJ  | j d}t|sd}t|d}d | _| _d| _g | _i | _i | _	i | _
i | _g | _g | _	 |d }|dkrF|| j d }t|}n| j d}q0|tv rt| \}}}|d ura|| | |dkr| j}| jd	krod
}tdd| j d|dfg| _n | j d}n|dv rd}n|dkr| j d}nd}t|q1|   d S )Nra   znot a JPEG file   r   T   r   r   rd   CMYK;Ijpegr   r    >     r   i   zno marker found)r   r   r   ru   rt   rv   rP   r|   _huffman_dc_huffman_acr   rG   rH   rR   r   MARKERmoder   _Tiler]   tile_read_dpi_from_exif)r   rX   r}   r   namedescriptionhandlerrawmoder   r   r   _openV  sR   


!zJpegImageFile._open	list[Any]c                   s   t   | j| jg S N)super__getstate__rv   r|   r   	__class__r   r   r     s   zJpegImageFile.__getstate__statec                   s$   |dd  \| _ | _t | d S )Nr1   )rv   r|   r   __setstate__)r   r   r   r   r   r     s   zJpegImageFile.__setstate__
read_bytesr   r   c                 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   LOAD_TRUNCATED_IMAGEShasattrr   )r   r   rX   r   r   r   	load_read  s
   zJpegImageFile.load_readr   
str | Noner]   tuple[int, int] | None0tuple[str, tuple[int, int, float, float]] | Nonec                 C  s~  t | jdkr	d S | jrd S | jd \}}}}d}| j}t|ts#J |d dkr4|dv r4|| _|df}|rt| jd |d  | jd |d  }dD ]}	||	krS nqK|d usZJ |d |d |d |d  |	 d |	 |d  |d |d  |	 d |	 |d  f}| jd |	 d |	 | jd |	 d |	 f| _|	}t	
||||g| _|df| _dd|d | |d | f}
| j|
fS )	Nr   r   rc   )rb   YCbCrr   )r&   r;   r   r   r   ra   )ry   r   decoderconfigr]   
isinstancerN   rw   minrq   r   r   r   )r   r   r]   r+   eoascaleoriginal_sizerX   boxr   r   r   draft  s:   &"".

zJpegImageFile.draftc                 C  s  t  \}}t| tj| jrtdd|| jg nzt	| W n	 t
y-   Y nw d}t|z/t|}|  |j| _W d    n1 sMw   Y  W zt	| W n t
yc   Y nw zt	| W w  t
yt   Y w w | jj| _| jj| _g | _d S )Ndjpegz-outfilezInvalid Filename)tempfilemkstemposclosepathexistsfilename
subprocess
check_callunlinkOSError
ValueErrorr   openloadrs   r   rw   r]   rq   r   )r   fr  r}   rr   r   r   r   
load_djpeg  s<   




zJpegImageFile.load_djpegdict[int, Any] | Nonec                 C     t | S r   )_getexifr   r   r   r   r       zJpegImageFile._getexifc                 C  s   d| j v s
d| j vrd S zA|  }|d }|d }zt|d |d  }W n ty0   |}Y nw t|r<d}t||dkrD|d	9 }||f| j d< W d S  tjt	t
tttfyc   d
| j d< Y d S w )Nr(   r0   i(  i  r   r   zDPI is not a numberra   r)   )H   r  )rK   getexiffloat	TypeErrormathisnanr  rU   rV   KeyErrorru   ZeroDivisionError)r   r0   resolution_unitx_resolutionr(   r}   r   r   r   r     s6   

z!JpegImageFile._read_dpi_from_exifc                 C  r  r   )_getmpr   r   r   r   r&    r  zJpegImageFile._getmp)r   r   )r   r   )r   r   r   r   )r   r   r   r   )r   r   r]   r   r   r   )r   r  )__name__
__module____qualname__formatformat_descriptionr   r   r   r   r  r  r  r   r&  __classcell__r   r   r   r   r   R  s    
8


(

r  c                 C  s   d| j vrd S |   S )Nr0   )rK   r  _get_merged_dictr   r   r   r   r    s   
r  c              
   C  s  z| j d }W n
 ty   Y d S w t|}|d}|dr#dnd}ddlm} z||}|	|j
 || t|}W n tyT } zd}	t|	|d }~ww z|d	 }
W n tyn } zd
}	t|	|d }~ww g }z|d }t|
D ]p}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rd|d< nd}	t|	dddd d!d"d#d$}||d% d&|d%< ||d< || qz||d< W |S  ty } zd'}	t|	|d }~ww )(NrE   r&   s   MM *><r   )TiffImagePluginz)malformed MP Index (unreadable directory)  z(malformed MP Index (no number of images)i  LLLHHrp   )	AttributeSize
DataOffsetEntryNo1EntryNo2r3  l        i   @i    i      i      i )DependentParentImageFlagDependentChildImageFlagRepresentativeImageFlagReservedImageDataFormatMPTyper>  r   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))rK   r"  ioBytesIOr   rJ   r   r0  ImageFileDirectory_v2seeknextr  dictrM   ru   r{   rU   unpack_fromzipr   getrI   )r   r^   file_contentshead
endiannessr0  rK   rE   r   r}   quant	mpentriesrawmpentriesentrynumunpackedentrylabelsmpentrympentryattr	mptypemapr   r   r   r&    s   







	

r&  rb   rc   r   r   )1rb   rc   RGBXrd   r   )@r   r   r!   r1   r9   r   r8     r   r;   r%   rk   rp         *   ra   r&   r<            )   +   	   rC      r9     (   ,   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   rs   Image.Imagec                 C  sZ   t | tr
| jdv rdS | jd dd | jd dd  | jd dd  }t|dS )N)r   r;   r   r   ra   r   )r   r   rv   r|   	samplingsrJ  )rs   samplingr   r   r   get_sampling  s   6r  r   	IO[bytes]r  str | bytesc                 C  sx  | j dks
| jdkrd}t|zt| j }W n ty/ } zd| j d}t||d }~ww | j}dd |ddD }|d	d
}|dd
}	|d}
|dkrZd
}d}	d}
n=|t	v rpt	| }d
}|dd
}	|d}
n't
|ts{d}t||	t	v rt	|	 dd
}	t
|
tr|
t	v rt	|
 d}
|	dkrd}	n(|	dkrd}	n!|	dkrd}	n|	dkrd}	n|	dkr| jdkrd}t|t| }	d:dd}|
dkr| jdkrd}t|t| dd }
||
}
|dd}d}|d  }rd!}|| }t||krd"}t|td| t| }|d#| d$ | 7 }|d% }rjd&}|| }g }|r@||d |  ||d  }|s.d}|D ]%}td| t| }|d'| d( t| tt| | 7 }|d7 }qD|d)| jd)}|d*d+p|d,d+}|d-d+}|d.d}t
|tjr| }t||krd/}t||||d0d||d1d+|d2d||	|d3d|d4d|
|||f| _|s|r| jd5krd6| jd  | jd  }n!|d7ks|d
krd| jd  | jd  }n
| jd | jd  }|r|t|d8 7 }|r|t|d 7 }ntt|d8 t|d }t| |td9d| j d|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)r*   xr   r   r   rm     r   z_save.<locals>.<listcomp>r(   r   qualityr  subsamplingqtableskeepr   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 JPEGKstr | tuple[list[int], ...] | list[list[int]] | dict[int, list[int]] | Noner   list[list[int]] | Nonec                   sV  d u rS t tr9zdd  D  W n ty) } zd}t||d }~ww  fddtdt dD t tttfrt trTfddttD n	t tr]tdt  k ridk spn d	}t|t	D ]2\}}zt|dkrd}t
|td
|}W n t
y } zd}t||d }~ww t||< qtS d S )Nc                 S  s.   g | ]}| d dd   D ]}t|qqS )#r   r   )rQ   r   )r*   linenumr   r   r   rm     s    z3_save.<locals>.validate_qtables.<locals>.<listcomp>zInvalid quantization tablec                   s   g | ]
} ||d   qS )r   r   )r*   rX   )linesr   r   rm     s    r   r   c                   s   g | ]
}| v r | qS r   r   )r*   key)r  r   r   rm     s    r!   z$None or too many quantization tablesr   )r   str
splitlinesr  r{   ry   rN   listrG  	enumerater  r   )r  r   r}   idxtabletable_arrayr   )r  r  r   validate_qtables  sJ   







z_save.<locals>.validate_qtablesextrarn   r   r4   r[  zXMP data is too longs   r2   ro   r9   s   r7   r_   ri   Frj   optimizer0   zEXIF data is too longsmoothkeep_rgb
streamtyperestart_marker_blocksrestart_marker_rowsrd   r;   _   r!   r   )r  r  r   r  )widthheightr  RAWMODEr   r"  r  encoderinforJ  r
   r   r   r  r*  r  getattrry   o16rI   r   rK   r   Exiftobytesencoderconfigr]   maxr   _saver   )rs   r   r  r}   r   r   rK   r(   r  r  r  presetr  r  MAX_BYTES_IN_MARKERr4   overhead_lenmax_data_bytes_in_markerr]   ro   markersr   r   r_   ri   r  r0   bufsizer   r   r   r    s   




*






r  str | bytes | NoneJpegImageFile | MpoImageFilec              	   C  s   t | |}z8| }|d ur8|d dkr;|jD ]\}}|dkr(d|v r(|  W S qddlm} |||}W |S W |S W |S  ttfyI   Y |S  tyW   t	
d Y |S w )Nr1  r   r   s    hdrgm:Version="r   zTImage appears to be a malformed MPO file, it will be interpreted as a base JPEG file)r   r&  rH   MpoImagePluginr   adoptr  rW   ru   warningswarn)r   r  rs   mpheadersegmentcontentr   r   r   r   jpeg_factoryU  s0   

	r  )z.jfifz.jpez.jpgz.jpegz
image/jpeg)r   r   r   r   r   r   )r   r   r   r   )r   r   r   r  )rs   r  r   r   )rs   r  r   r  r  r  r   r   r   )r   r  r  r  r   r  )3
__future__r   r   rB  r   r
  rU   r  r   r  r  r   r   r   _binaryr   r   r   rT   r   r	   r  JpegPresetsr
   TYPE_CHECKINGtypingr   r   r  r   r   r   r`   r   r   r   r   r   r  r&  r  r   r  r  r  r  register_openr*  register_saveregister_extensionsregister_mimer   r   r   r   <module>   sN  !


j

2







	













 
!
"
#
$
%
&
'
(
)
*
+
,
-
.
/
01234
C	 
?
P


 G