a
    Ag!                  
   @  s  d dl mZ d dlZd dlmZ d dlm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lmZ dddddddddd	ZdddddZG dd de	j	ZddddddZdd d!dd"d#d$d%Zeejee eejd& eejd' dS )(    )annotationsN)cached_property)IO   )Image	ImageFileImagePalette)i8)i16be)i32be)si16be)si32be)1r   )Lr   )Pr   )RGB   )CMYK   )ZLABr   )	)r   r   )r      )r   r   )   r   )r   r   )r   r   )   r   )r   r   )	   r   bytesbool)prefixreturnc                 C  s   | d d dkS )Nr   s   8BPS )r   r   r   Q/var/www/lab.imftr.de/x/nb_venv/lib/python3.9/site-packages/PIL/PsdImagePlugin.py_accept1   s    r   c                   @  sz   e Zd ZdZdZdZddddZeddd	d
Ze	ddddZ
e	ddddZdddddZddddZdS )PsdImageFileZPSDzAdobe PhotoshopFNone)r   c                 C  s*  | 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d | _t|d}	|	r| j  |	 }t|d}	|	r| j  | _|	| _| j | d | _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palette	resourcestellr	   lenappendinfo_layers_position_layers_sizeseek	_n_frames	_maketilesizetile_fpframeZ
_min_frame)selfr*   smsgZpsd_bitsZpsd_channelsZpsd_modemodechannelsr=   dataendidnamer   r   r   _open>   sh    



zPsdImageFile._openGlist[tuple[str, str, tuple[int, int, int, int], list[ImageFile._Tile]]]c                 C  sL   g }| j d ur>| j| j  tt| j| j}t|| j}t	|| _
|S N)r8   r?   r:   ioBytesIOr   
_safe_readr9   
_layerinfor5   r;   )rA   layersZ_layer_datar   r   r   rQ      s    

zPsdImageFile.layersintc                 C  s   | j d u rt| j| _ | j S rL   )r;   r5   rQ   rA   r   r   r   n_frames   s    
zPsdImageFile.n_framesr   c                 C  s   t | jdkS )Nr   )r5   rQ   rS   r   r   r   is_animated   s    zPsdImageFile.is_animated)layerr   c              
   C  sz   |  |sd S z4| j|d  \}}}}|| _|| _|| _| j| _W n2 tyt } zd}t||W Y d }~n
d }~0 0 d S )Nr   zno such layer)	Z_seek_checkrQ   r/   r>   r@   r?   r)   
IndexErrorEOFError)rA   rV   _rD   r>   erC   r   r   r   r:      s    
zPsdImageFile.seekc                 C  s   | j S rL   )r@   rS   r   r   r   r4      s    zPsdImageFile.tellN)__name__
__module____qualname__formatformat_descriptionZ!_close_exclusive_fp_after_loadingrJ   r   rQ   propertyrT   rU   r:   r4   r   r   r   r   r    9   s   Sr    z	IO[bytes]rR   rK   )r)   ct_bytesr   c              	     sj  g }ddd 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}|d	kr̈ |d
 d tj t	|d	} |tj qLt|D ]8}t|d}|dkrd}nd| }|
| |d	 q|  |dgkr(d}n,|g dkr<d}n|g dkrPd}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 qLg }t|D ]P\}\}}}g }|D ]$}t ||d}|r*|| q*|
||||f q|S )NrR   r   )r=   r   c                   s   t  | S rL   )r   rO   )r=   r)   r   r   r*      s    z_layerinfo.<locals>.readr      z4Layer block too short for number of layers requestedr      r$   i  Ar&   Rr   )BGrf   r   )re   rg   rh   rf       r   zlatin-1replace)si16absr,   rangesi32r+   r:   rM   SEEK_CURr0   r6   sortr4   r	   decode	enumerater<   extend)r)   ra   rQ   r*   ctrC   rY   Zy0Zx0y1x1ZbandsZct_typesr=   typebrD   rI   Zdata_endlengthZ	layerinfoibboxr>   mtr   rb   r   rP      st    


rP   strztuple[int, int, int, int]zlist[ImageFile._Tile])filerD   r|   rE   r   c              	   C  s6  g }| j }t|d}|d |d  }|d |d  }|  }	|dkrt|D ]>}
||
 }|dkrj|d7 }|td||	| |	||  }	qNn|dkrd}||| d }|  }	t|D ]V}
||
 }|dkr|d7 }|td||	| t|D ]}|	t|| }	|d7 }qq| |	 |	d@ r2|d |S )	Nr   r   r   r   r   z;Ir2   Zpackbits)r*   r+   r4   rn   r6   r   Z_Tiler:   )r   rD   r|   rE   Ztilesr*   compressionZxsizeZysizeoffsetZchannelrV   r{   Z	bytecountyr   r   r   r<     s:    


r<   z.psdzimage/vnd.adobe.photoshop)
__future__r   rM   	functoolsr   typingr   ri   r   r   r   _binaryr	   r
   r+   r   r0   r   rl   r   ro   r-   r   r    rP   r<   Zregister_openr^   Zregister_extensionZregister_mimer   r   r   r   <module>   s6    [2