o
    pèÝc|  ã                
   @   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   ú2lib/python3.10/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   sF  | j j}|dƒ}t|ƒrt|dƒdkrd}t|ƒ‚t|dƒ}t|dƒ}t|dƒ}t||f \}}||kr:d}t|ƒ‚|d	krF|dkrFd
}d}|| _t|dƒt|dƒf| _	t|dƒƒ}	|	ro||	ƒ}
|dkro|	dkrot
 d|
¡| _g | _t|dƒƒ}	|	rÏ| j  ¡ |	 }| j  ¡ |k rÏ|dƒ t|dƒƒ}|t|dƒƒƒ}t|ƒd@ s¤|dƒ |t|dƒƒƒ}
t|
ƒd@ r¶|dƒ | j |||
f¡ |dkrÈ|
| jd< | j  ¡ |k sˆg | _t|dƒƒ}	|	rÿ| j  ¡ |	 }t|dƒƒ}	|	rùt 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Úi32Ú_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'   Úchannelsr6   ÚdataÚendÚidÚnameZ_layer_datar   r   r   Ú_open;   sl   



õ
zPsdImageFile._openc              
   C   sp   |   |¡sd S z| j|d  \}}}}|| _|| _|| _| j| _||fW S  ty7 } zd}t|ƒ|‚d }~ww )Nr   zno such layer)	Z_seek_checkr/   r'   r7   r9   r8   r!   Ú
IndexErrorÚEOFError)r:   ÚlayerrA   r'   Úbboxr7   Úer<   r   r   r   r4      s   


€þzPsdImageFile.seekc                 C   s   | j S ©N)r9   )r:   r   r   r   r+       s   zPsdImageFile.tellN)	Ú__name__Ú
__module__Ú__qualname__ÚformatZformat_descriptionZ!_close_exclusive_fp_after_loadingrB   r4   r+   r   r   r   r   r   5   s    Ur   c              	      s(  g }‡ fdd„}t |dƒƒ}|t|ƒd k rd}t|ƒ‚tt|ƒƒD ]Ã}t|dƒƒ}t|dƒƒ}t|dƒƒ}	t|dƒƒ}
g }t|dƒƒ}tt|ƒƒ}t|ƒdkrQq"|D ]}t|dƒƒ}|dkrbd}nd	| }| |¡ |dƒ qS| 	¡  |d
gkr|d}n|g d¢kr…d}n|g d¢krŽd	}nd }|dƒ d}t|dƒƒ}|rÙˆ  
¡ | }t|dƒƒ}|rµˆ  |d tj¡ t|dƒƒ}|rÄˆ  |tj¡ t|dƒƒ}|rÔ||ƒ dd¡}ˆ  |¡ | |||||
|	ff¡ q"d}|D ]'\}}}g }|D ]}tˆ ||dƒ}|r| |¡ qó||||f||< |d7 }qê|S )Nc                    s   t  ˆ | ¡S rH   )r   r2   )r6   ©r!   r   r   r"   ©   s   z_layerinfo.<locals>.readr   é   z4Layer block too short for number of layers requestedr   iÿÿ  ÚAr   ÚRr   )ÚBÚGrP   r   )rO   rQ   rR   rP   r   Ú é   r   zlatin-1Úreplacer   )Úsi16Úabsr$   Úranger(   r#   Úlistr,   r-   Úsortr+   r4   r0   ÚSEEK_CURr   Údecoder5   Úextend)r!   Zct_bytesr/   r"   Zctr<   Ú_Zy0Zx0Zy1Zx1r'   Zct_typesÚtypesÚtypeÚmrA   r6   Zdata_endÚlengthÚirF   r7   Útr   rM   r   r3   ¥   st   




€
r3   c                 C   s2  d }| j }t|dƒƒ}|d |d  }|d |d  }|  ¡ }	|dkrHg }t|ƒD ]}
||
 }|dkr7|d7 }| d||	|f¡ |	||  }	q)nB|dkrŠd}g }||| d ƒ}|  ¡ }	t|ƒD ])}
||
 }|dkrn|d7 }| d||	|f¡ t|ƒD ]}|	t||ƒ }	|d7 }q{q`|  |	¡ |	d@ r—|dƒ |S )	Nr   r   r   r   r   z;Ir*   Zpackbits)r"   r#   r+   rX   r-   r4   )Úfiler'   rF   r=   r7   r"   ÚcompressionZxsizeZysizeÚoffsetZchannelrE   rc   Z	bytecountÚyr   r   r   r5   þ   sB   û
þ
r5   z.psdzimage/vnd.adobe.photoshop)r0   rS   r   r   r   Z_binaryr   r   r#   r   r(   r	   rV   r%   r   r   r3   r5   Zregister_openrL   Zregister_extensionZregister_mimer   r   r   r   Ú<module>   s.   öpY3