U
    1c|                  
   @   s   d dl Z ddlmZ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	d	d
ddd	d	dd	Zdd ZG dd dejZdd Zdd Zeejee eejd eejd dS )    N   )Image	ImageFileImagePalette)i8)i16be)i32be)si16be)1r   )Lr   )Pr   )RGB   )CMYK   )ZLABr   )	)r   r   )r      )r   r   )   r   )r   r   )r   r   )   r   )r   r   )	   r   c                 C   s   | d d dkS )Nr   s   8BPS )prefixr   r   1lib/python3.8/site-packages/PIL/PsdImagePlugin.py_accept-   s    r   c                   @   s0   e Zd ZdZdZdZdd Zdd Zdd	 Zd
S )PsdImageFileZPSDzAdobe PhotoshopFc                 C   sJ  | j j}|d}t|r&t|ddkr2d}t|t|d}t|d}t|d}t||f \}}||krtd}t||d	kr|dkrd
}d}|| _t|dt|df| _	t|d}	|	r||	}
|dkr|	dkrt
d|
| _g | _t|d}	|	r| j  |	 }| j  |k r|d t|d}|t|d}t|d@ sN|d |t|d}
t|
d@ rt|d | j|||
f |dkr|
| jd< qg | _t|d}	|	r| j  |	 }t|d}	|	rtt| j |	}t||	| _| j | t| j| _| jdk| _t| j |d| j || _| j | _d| _ d| _!d S )N   r   r   znot a PSD file         znot enough channelsr   RGBA      r   i   zRGB;Lr   i  Zicc_profile)r   r   )"fpreadr   i16SyntaxErrorMODESOSErrormodei32Z_sizer   rawZpaletteZ	resourcestellr   lenappendinfolayersioBytesIOr   
_safe_read
_layerinfoseekZn_framesZis_animated	_maketilesizetile_fpframeZ
_min_frame)selfr"   smsgZpsd_bitsZpsd_channelsZpsd_moder'   channelsr5   dataendidnameZ_layer_datar   r   r   _open;   sj    



zPsdImageFile._openc              
   C   s~   |  |sd S z:| j|d  \}}}}|| _|| _|| _| j| _||fW S  tk
rx } zd}t||W 5 d }~X Y nX d S )Nr   zno such layer)	Z_seek_checkr.   r'   r6   r8   r7   r!   
IndexErrorEOFError)r9   layerr@   r'   bboxr6   er;   r   r   r   r3      s    

zPsdImageFile.seekc                 C   s   | j S N)r8   )r9   r   r   r   r*      s    zPsdImageFile.tellN)	__name__
__module____qualname__formatZformat_descriptionZ!_close_exclusive_fp_after_loadingrA   r3   r*   r   r   r   r   r   5   s   Ur   c              	      s@  g } fdd}t |d}|t|d k r8d}t|tt|D ]}t|d}t|d}t|d}	t|d}
g }t|d}tt|}t|dkrqD|D ]8}t|d}|dkrd}nd	| }|| |d q|	  |d
gkrd}n2|ddd
gkrd}n|dddd
gkr(d	}nd }|d d}t|d}|rƈ 
 | }t|d}|rz |d tj t|d}|r |tj t|d}|r||dd} | ||||||
|	ff qDd}|D ]R\}}}g }|D ]$}t ||d}|r|| q||||f||< |d7 }q|S )Nc                    s   t  | S rG   )r   r1   )r5   r!   r   r   r"      s    z_layerinfo.<locals>.readr      z4Layer block too short for number of layers requestedr   i  Ar   Rr   BGr   r       r   zlatin-1replacer   )si16absr$   ranger(   r#   listr+   r,   sortr*   r3   r/   SEEK_CURr   decoder4   extend)r!   Zct_bytesr.   r"   Zctr;   _Zy0Zx0Zy1Zx1r'   Zct_typestypestypemr@   r5   Zdata_endZlengthirE   r6   tr   rL   r   r2      sr    



r2   c                 C   s6  d }| j }t|d}|d |d  }|d |d  }|  }	|dkrg }t|D ]:}
||
 }|dkrn|d7 }|d||	|f |	||  }	qRn|dkrd}g }||| d }|  }	t|D ]R}
||
 }|dkr|d7 }|d||	|f t|D ]}|	t|| }	|d7 }qq| |	 |	d@ r2|d |S )	Nr   r   r   r   r   z;Ir)   Zpackbits)r"   r#   r*   rW   r,   r3   )filer'   rE   r<   r6   r"   ZcompressionZxsizeZysizeoffsetZchannelrD   ra   Z	bytecountyr   r   r   r4      s>    


r4   z.psdzimage/vnd.adobe.photoshop)r/   rR   r   r   r   Z_binaryr   r   r#   r   r(   r	   rU   r%   r   r   r2   r4   Zregister_openrK   Zregister_extensionZregister_mimer   r   r   r   <module>   s,   pY3