o
    ft                     @  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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 dddZd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-d.d/efd0d1d2efd3d4d5d6d7efd8d9d:efd;d<d=efd>d?d@efdAdBdCefdDdEdFefdGdHdIefdJdKi dLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`efdaddbefdcdddeefdfdgdhefdidjdkefdldmdnefdodpdqefdrdsdtefi dudvdwefdxdydzefd{d|d}efd~dde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#dd Z$dddddddZ%dZ&d dddĜZ'ddƄ Z(ddd΄Z)dddЄZ*ddd҄Z+e,e"j-e+e! e.e"j-e) e/e"j-g dӢ e0e"j-dԡ dS )    )annotationsN)IOAny   )Image	ImageFile)i16be)i32be)o8)o16be)presets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   L/home/ubuntu/webapp/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rr|d d dkrrt |d | jd< }t	|d	| jd
< z|d }t |dt |df}W n
 t
y\   Y d S w |dkrf|| jd< || jd< || jd< d S |dkr|d d dkrd| jv r| jd  |dd  7  < d S || jd< | j | d | _d S |dkr|d d dkr|ddd | jd< d S |dkr|d d dkr|| jd< d S |dkr|d d dkr| j| d S |dkr{|d d  d!kr{d }| jd"i }	|||d  d#kryzZ|d7 }t ||}
|d7 }|| }|d| 7 }||d@ 7 }t||}|d7 }||||  }|
d$krPt|d%d& t |dt|dd& t |dd'}||	|
< ||7 }||d@ 7 }W n tjyk   Y d S w |||d  d#ksd S d S |d(kr|d d d)krt |d| jd*< z|d+ }W n ty   Y d S w || jd,< d S |dkr|d d d-kr|dd  | jd.< | j | d | jd/< d S d S d S )0Nr   zAPP%d        s   JFIF   jfif   jfif_version      
   r   dpi	jfif_unitjfif_density     s   Exif  exif   s   http://ns.adobe.com/xap/1.0/     xmp  s   FPXR flashpix      ICC_PROFILE      s   Photoshop 3.0 	photoshops   8BIMi  r   i   )XResolutionDisplayedUnitsXYResolutionDisplayedUnitsY  s   Adobeadobe   adobe_transforms   MPF mpmpoffset)r   r   r   r   r   appapplistappendinfodivmod	Exceptiontell_exif_offsetspliticclist
setdefaulti32structerror
IndexError)r   r   r   sr@   versionr'   r(   offsetr5   codename_lensizedatar=   r   r   r   APP?   s   







rV   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   rC   r@   rA   rB   )r   r   r   rO   r   r   r   rX      s
   

rX   c           	      C  s  t | jdd }t| j|}t |dt |df| _|d | _| jdkr1d| j d}t||d | _| jdkr?d	| _	n| jdkrHd
| _	n| jdkrQd| _	nd| j d}t||dv rjd | 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 )r4   Nr   ).0pr   r   r   
<listcomp>   s    zSOF.<locals>.<listcomp>    icc_profiler*      r   )r   r   r   r   r   _sizebitsSyntaxErrorlayers_moderC   rI   sortlenjoinrangelayerrB   )	r   r   r   rO   msgprofilerh   itr   r   r   SOF   s<   







.rx   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   ri   r   @   zbad quantization table markerBHlittlec                   s   g | ]} | qS r   r   )rd   rv   rU   r   r   rf          zDQT.<locals>.<listcomp>r   )r   r   r   r   r   rp   rl   arraysys	byteorderbyteswapzigzag_indexquantization)r   r   r   rO   v	precision	qt_lengthrt   r   r}   r   DQT   s    r   i  SOF0zBaseline DCTi  SOF1zExtended Sequential DCTr]   SOF2zProgressive DCTi  SOF3zSpatial losslessi  DHTzDefine Huffman tablei  SOF5zDifferential sequential DCTr^   SOF6zDifferential progressive DCTi  SOF7zDifferential spatiali  )JPG	ExtensionNi  SOF9zExtended sequential DCT (AC)r_   SOF10zProgressive DCT (AC)i  SOF11zSpatial lossless DCT (AC)i  DACz%Define arithmetic coding conditioningi  SOF13z Differential sequential DCT (AC)r`   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 1r/   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 12r3   APP13zApplication segment 13r:   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 d dkS )NrY   s   r   )r   r   r   r   _accept>  s   r   c                   @  sV   e Zd ZdZdZ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 ZdS )!r   JPEGzJPEG (ISO 10918)c                 C  s@  | j d}t|sd}t|d}d | _| _g | _i | _i | _i | _	i | _
g | _g | _	 |d }|dkrC|| j d }t|}n| j d}q-|tv rt| \}}}|d ur^|| | |dkrz| j}| jd	krld
}dd| j d|dffg| _n | j d}n|dv rd}n|dkr| j d}nd}t|q.|   d S )NrY   znot a JPEG file   r   T   r   r   r\   CMYK;Ijpegr   r    >     r   i   zno marker found)r   r   r   rl   rk   rm   rs   
huffman_dc
huffman_acr   r@   rA   rI   r   MARKERmoderT   tile_read_dpi_from_exif)r   rO   rt   rv   namedescriptionhandlerrawmoder   r   r   _openK  sN   


zJpegImageFile._open
read_bytesr   r   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   rO   r   r   r   	load_read  s
   zJpegImageFile.load_readr   
str | NonerT   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}|d dkr-|dv r-|| _|df}|rt| jd |d  | jd |d  }dD ]}	||	krL nqD|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   r[   )rZ   YCbCrr   )r$   r   r   r   r   rY   )rp   r   decoderconfigrT   rn   minrj   r   )r   r   rT   deoascaleoriginal_sizerO   boxr   r   r   draft  s6   &"".

zJpegImageFile.draftr   c                 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djpeg-outfilezInvalid Filename)tempfilemkstemposclosepathexistsfilename
subprocess
check_callunlinkOSError
ValueErrorr   openloadimr   rn   rT   rj   r   )r   fr   rt   _imr   r   r   
load_djpeg  s<   




zJpegImageFile.load_djpegdict[str, Any] | Nonec                 C     t | S N)_getexifr   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&   r+   i(  i  r   r   zDPI is not a numberrY   gRQ@)H   r  )rC   getexiffloat	TypeErrormathisnanr  rL   rM   KeyErrorrl   ZeroDivisionError)r   r+   resolution_unitx_resolutionr&   rt   r   r   r   r     s6   

z!JpegImageFile._read_dpi_from_exifc                 C  r	  r
  )_getmpr  r   r   r   r    r  zJpegImageFile._getmpN)r   r   r   r   )r   r   rT   r   r   r   )r   r   r   r  )__name__
__module____qualname__formatformat_descriptionr   r   r   r  r  r   r  r   r   r   r   r   G  s    
5

&

r  c                 C  s   d| j vrd S |   S )Nr+   )rC   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 d dkr&dnd}ddlm} z||}||j	 |
| t|}W n tyW } zd	}	t|	|d }~ww z|d
 }
W n tyq } zd}	t|	|d }~ww g }z|d }td|
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< || q~||d< W |S  ty	 } zd(}	t|	|d }~ww ))Nr>   r$   r   s   MM *><r   )TiffImagePluginz)malformed MP Index (unreadable directory)  z(malformed MP Index (no number of images)i  r   LLLHHri   )	AttributeSize
DataOffsetEntryNo1EntryNo2r%  l        i   @i    i      i      i )DependentParentImageFlagDependentChildImageFlagRepresentativeImageFlagReservedImageDataFormatMPTyper0  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   r1  Unknownz!malformed MP Index (bad MP Entry))rC   r  ioBytesIOr   r   r"  ImageFileDirectory_v2seeknextr  dictrE   rl   rr   rL   unpack_fromzipr   getrB   )r   rU   file_contentshead
endiannessr"  rC   r>   r   rt   quant	mpentriesrawmpentriesentrynumunpackedentrylabelsmpentrympentryattr	mptypemapr   r   r   r    s   







	

r  rZ   r[   r   r   )1rZ   r[   RGBXr\   r   )@r   r   r   r*   r4   r   r*     r   r   r#   rc   ri      r,   *   rY   r$   r1            )   +   	   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  sZ   t | dr
| jdv rdS | jd dd | jd dd  | jd dd  }t|dS )Nrm   )r   r   r   r   rY   r   )r   rm   rs   	samplingsr<  )r  samplingr   r   r   get_samplingk  s   6r{  r  Image.Imager   	IO[bytes]r   str | bytesc                 C  s"  | 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kr| jdkrd}t|t| dd }
||
}
|dd}d}|d}|r9d}|| }g }|r||d |  ||d  }|sd}|D ]%}td| t| }|d| d  t| tt| | 7 }|d7 }q|d!| jd!}|d"d#pO|d$d#}|d%d#}|d&d}t
|tjrg| }t||krtd'}t||||d(d||d)d#|d*d|d |d |	|d+d|d,d|
|||f| _d}|s|r| jd-krd.| jd  | jd  }n!|d/ks|d
krd| jd  | jd  }n
| jd | jd  }|r|t|d0 7 }|r|t|d 7 }nt|t|d0 t|d }t| |d1d| j d|fg| d S )2Nr   z cannot write empty image as JPEGzcannot write mode z as JPEGc                 S  s   g | ]}t |qS r   )round)rd   xr   r   r   rf     r~   z_save.<locals>.<listcomp>r&   r   qualityrx  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 JPEGc                   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   )rH   r   )rd   linenumr   r   r   rf     s    z3_save.<locals>.validate_qtables.<locals>.<listcomp>zInvalid quantization tablec                   s   g | ]
} ||d   qS )ry   r   )rd   rO   )linesr   r   rf     s    r   ry   c                   s   g | ]
}| v r | qS r   r   )rd   key)r  r   r   rf     s    r   z$None or too many quantization tablesr{   )
isinstancestr
splitlinesr  rr   rp   tuplelistr9  	enumerater  r   )r  r   rt   idxtabler   )r  r  r   validate_qtables  sJ   







z_save.<locals>.validate_qtablesextrarg   r   rh   r4   s   r2   rW   ra   Frb   optimizer+   zEXIF data is too longsmoothkeep_rgb
streamtyperestart_marker_blocksrestart_marker_rowsr\   r   _   r   r   )widthheightr  RAWMODEr   r  r   encoderinfor<  r   r  r   r  r  r{  getattrrB   o16rp   r
   rC   r   ExiftobytesencoderconfigrT   maxr   _save)r  r   r   rt   r   r   rC   r&   r  r  r  presetr  r  MAX_BYTES_IN_MARKERrh   ICC_OVERHEAD_LENMAX_DATA_BYTES_IN_MARKERmarkersrv   r   rT   rW   ra   r  r+   bufsizer   r   r   r  y  s   



&







$r  c                 C  s@   |   }tdd||g zt| W d S  ty   Y d S w )Ncjpegr   )_dumpr   r   r   r   r   )r  r   r   r   r   r   r   _save_cjpeg0  s   r  c              	   C  s   t | |}z1| }|d dkr4|jD ]\}}|dkr$d|v r$|  W S qddlm} |||}W |S W |S  ttfyB   Y |S  tyP   t	
d Y |S w )Nr#  r   r   s    hdrgm:Version=")MpoImageFilezTImage appears to be a malformed MPO file, it will be interpreted as a base JPEG file)r   r  rA   MpoImagePluginr  adoptr  rN   rl   warningswarn)r   r   r  mpheadersegmentcontentr  r   r   r   jpeg_factory<  s,   

	r  )z.jfifz.jpez.jpgz.jpegz
image/jpeg)r   r   r   r   r   r   )r   r   r   r   r  )r  r|  r   r}  r   r~  r   r   )NN)1
__future__r   r   r4  r  r   rL   r   r   r   r  typingr   r   r   r   r   _binaryr   r   r	   rK   r
   r   r  JpegPresetsr   r   rV   rX   rx   r   r   r   r   r  r  r  r   ry  r{  r  r  r  register_openr  register_saveregister_extensionsregister_mimer   r   r   r   <module>   sH  !

f

0







	

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