File: //usr/lib/python3/dist-packages/PIL/__pycache__/Jpeg2KImagePlugin.cpython-38.pyc
U
g�
^� � @ s� d dl Z d dlZd dlZddlmZmZ dd� Zdd� ZG dd � d ej�Zd
d � Z d d
� Z
e� ej ee � e�
ej e
� e�ej ddddddg� e�ej d� dS )� N� )�Image� ImageFilec C s | � d�}t�d|�d }|| � |d � }t�d|�\ }}}}}}} } } } }
dg|
} dg|
} dg|
}
t|
�D ],}t�d|dd| �\| |< | |< |
|< qt|| || f}|
d kr�|
d d
@ d kr�d }nd
}n0|
dkr�d}n"|
dkr�d}n|
dk�rd}nd}||fS )z�Parse the JPEG 2000 codestream to extract the size and component
count from the SIZ marker segment, returning a PIL (size, mode) tuple.� z>Hr z >HHIIIIIIIIHN�>BBB�$ � r � � �I;16�L�LA�RGB� �RGBA)�read�struct�unpack� unpack_from�range)�fpZhdrZlsizZsizZrsizZxsizZysizZxosizZyosiz�_ZcsizZssizZxrsizZyrsiz�i�size�mode� r �7/usr/lib/python3/dist-packages/PIL/Jpeg2KImagePlugin.py�_parse_codestream s2
�
*
r c C s� d}d}t �d| �d��\}}|dkrBt �d| �d��d }d}nd}||k rVtd��|d krp| �|| �}q�q|d
kr�| �d �d kr�d
}| �|| d tj� q| �|| tj� q|dkr�td��d}d}d}d} t�|�}
t �d|
�d��\}}|dk�r t �d|
�d��d }d}nd}|
�|| �} |dk�r�t �d| �\} }
} }}}}|
| f}|�r�| dk�r~|d@ dk�r~d}n>| dk�r�d}n.| dk�r�d}n | dk�r�d}n| d k�r�d}�q�q�|dkr�t � d| �\}}}|dkr�t � d| d�d }|dk�rR| dk�r |d@ dk�r d}n.| dk�r.d}n | dk�r>d}n| d k�r�d}�q�q�|d k�r�| dk�rz|d@ dk�rzd}n | dk�r�d}n| dk�r�d}�q�q�|d kr�| dk�r�d}n| d k�r�d}�q�q�|dk�s�|dk�r�td ��|||fS ) z�Parse the JP2 header box to extract size, component count and
color space information, returning a (size, mode, mimetype) tuple.Nz>I4sr
r z>Qr � zInvalid JP2 header lengths jp2hs ftypr s jpx z image/jpxzcould not find JP2 headers ihdrz>IIHBBBBr r r r r
r r r s colrr z>I� � zMalformed jp2 header)
r r r � SyntaxError�seek�os�SEEK_CUR�io�BytesIOr )r �headerZmimetypeZlboxZtboxZhlenr r ZbpcZncZhioZcontentZheight�width�cZunkcZipr�methZprecZapproxZcsr r r �_parse_jp2_header8 s�
r+ c @ s$ e Zd ZdZdZdd� Zdd� ZdS )�Jpeg2KImageFileZJPEG2000zJPEG 2000 (ISO 15444)c C sP | j �d�}|dkr.d| _t| j �\| _| _nB|| j �d� }|dkrhd| _t| j �}|\| _| _| _ntd��| j d ks�| jd kr�td��d | _
d | _ d
}d
}z| j � � }t
�|�j}W nj tk
�r$ d
}z4| j �� }| j �d tj� | j �� }| j �|� W n tk
�r d
}Y nX Y nX d d | j d | j| j
| j ||ffg| _d S )
Nr � �O�Q�j2kr
� jP
�
�jp2znot a JPEG 2000 filez unable to determine size/moder ����jpeg2k�r r )r r �codecr �_sizer r+ Zcustom_mimetyper! r �reduce�layers�filenor# �fstat�st_size� Exception�tellr" r% �SEEK_END�tile)�selfZsigr' �fdZlength�posr r r �_open� sD
��zJpeg2KImageFile._openc C s� | j rFd| j > }|d? }t| jd | | �t| jd | | �f| _| jr�| jd }|d d | j | j|d d |d d f}|d d| j |d |fg| _tj�| �S )Nr r r r r3 r )r6 �intr r5 r> r7 r �load)r? ZpowerZadjust�tZt3r r r rD � s
�
* zJpeg2KImageFile.loadN)�__name__�
__module__� __qualname__�formatZformat_descriptionrB rD r r r r r, � s .r, c C s | d d� dkp | d d� dkS )Nr r- � r/ r )�prefixr r r �_accept� s �rL c C sF |� d�rd}nd}| j}|�dd �}|�dd �}|�dd �}|�dd�}|�d d �} | d k r�t| ttf�r~td
d � | D ��s�td ��|�d
d�}
|�dd �} |�dd �} |�dd�}
|�dd�}|�dd�}d}t|d��rz |� � }W n t
k
�r d}Y nX ||||| |
| | |
|||f | _ t �
| |dd| j d|fg� d S )N�.j2kr. r0 �offset� tile_offset� tile_size� quality_modeZrates�quality_layersc S s g | ]}t |ttf��qS r )�
isinstancerC �float)�.0Z
quality_layerr r r �
<listcomp>� s �z_save.<locals>.<listcomp>z,quality_layers must be a sequence of numbers�num_resolutionsr Zcodeblock_size�
precinct_size� irreversibleF� progressionZLRCP� cinema_modeZnor1 r8 r2 r3 )�endswithZ encoderinfo�getrS �list�tuple�all�
ValueError�hasattrr8 r; Z
encoderconfigr �_saver )Zimr �filenameZkind�inforN rO rP rQ rR rW Z cblk_sizerX rY rZ r[ r@ r r r rc � sZ
����
�rc z.jp2rM z.jpcz.jpfz.jpxz.j2cz image/jp2)r% r# r � r r r r+ r, rL rc Z
register_openrI Z
register_saveZregister_extensionsZ
register_mimer r r r �<module> s "eD = �