U
    M»{j
#  ã                   @  sì   d dl mZ d dlZd dlmZmZmZ ddlmZm	Z	m
Z
mZ erPddlmZ G dd„ dƒZd	dd
œdd„Zddddœdd„Zdddœdd„Zd'dddœdd„Zd(dddœdd„Zd)dddœdd „Zd*dddœd!d"„Zdd#d$œd%d&„ZdS )+é    )ÚannotationsN)ÚIOÚTYPE_CHECKINGÚSequenceé   )ÚGimpGradientFileÚGimpPaletteFileÚ
ImageColorÚPaletteFile)ÚImagec                   @  sÚ   e Zd ZdZd+ddddœdd	„Zed
dœdd„ƒZejd
ddœdd„ƒZeddœdd„ƒZejdddœdd„ƒZd dœdd„Z	ddœdd„Z
ddœdd„ZeZd,dddd œd!d"„Zd-d#ddd$œd%d&„Zd'dd(œd)d*„ZdS ).ÚImagePalettea¥  
    Color palette for palette mapped images

    :param mode: The mode to use for the palette. See:
        :ref:`concept-modes`. Defaults to "RGB"
    :param palette: An optional palette. If given, it must be a bytearray,
        an array or a list of ints between 0-255. The list must consist of
        all channels for one color followed by the next color (e.g. RGBRGBRGB).
        Defaults to an empty palette.
    ÚRGBNÚstrz(Sequence[int] | bytes | bytearray | NoneÚNone)ÚmodeÚpaletteÚreturnc                 C  s"   || _ d | _|ptƒ | _d | _d S ©N)r   ÚrawmodeÚ	bytearrayr   Údirty)Úselfr   r   © r   úO/var/www/html/bwave/python/venv/lib/python3.8/site-packages/PIL/ImagePalette.pyÚ__init__)   s    zImagePalette.__init__ú!Sequence[int] | bytes | bytearray)r   c                 C  s   | j S r   )Ú_palette©r   r   r   r   r   3   s    zImagePalette.palette)r   r   c                 C  s   d | _ || _d S r   )Ú_colorsr   )r   r   r   r   r   r   7   s    zdict[tuple[int, ...], int]c                 C  sh   | j d krbt| jƒ}i | _ tdt| jƒ|ƒD ]4}t| j||| … ƒ}|| j krRq,|| | j |< q,| j S )Nr   )r   Úlenr   Úranger   Útuple)r   Zmode_lenÚiÚcolorr   r   r   Úcolors<   s    


zImagePalette.colors)r$   r   c                 C  s
   || _ d S r   )r   )r   r$   r   r   r   r$   H   s    c                 C  s<   t ƒ }| j|_| j|_| jd k	r0| jd d … |_| j|_|S r   )r   r   r   r   r   )r   Únewr   r   r   ÚcopyL   s    
zImagePalette.copyz-tuple[str, Sequence[int] | bytes | bytearray]c                 C  s    | j r| j | jfS | j|  ¡ fS )z£
        Get palette contents in format suitable for the low-level
        ``im.putpalette`` primitive.

        .. warning:: This method is experimental.
        )r   r   r   Útobytesr   r   r   r   ÚgetdataW   s    zImagePalette.getdataÚbytesc                 C  s:   | j rd}t|ƒ‚t| jtƒr$| jS t d| j¡}| ¡ S )zUConvert palette to bytes.

        .. warning:: This method is experimental.
        ú!palette contains raw palette dataÚB)r   Ú
ValueErrorÚ
isinstancer   r)   Úarrayr'   )r   ÚmsgZarrr   r   r   r'   b   s    zImagePalette.tobyteszImage.Image | NonezException | NoneÚint)ÚimageÚer   c                 C  s´   t | jtƒst| jƒ| _t| jƒd }d}|rX|j d¡|j d¡f}||krX|d7 }qF|dkr°|ršttt	| 
¡ ƒƒƒD ] \}}|dkrx||krx|} qšqx|dkr°d}t|ƒ|‚|S )	Né   r   Ú
backgroundZtransparencyr   é   r   z$cannot allocate more than 256 colors)r-   r   r   r   r   ÚinfoÚgetÚreversedÚlistÚ	enumerateZ	histogramr,   )r   r1   r2   ÚindexZspecial_colorsr"   Úcountr/   r   r   r   Ú_new_color_indexr   s(    

þ

zImagePalette._new_color_indexztuple[int, ...])r#   r1   r   c              
   C  sT  | j rd}t|ƒ‚t|tƒ�r:| jdkrZt|ƒdkrx|d dkrLd}t|ƒ‚|dd… }n| jdkrxt|ƒdkrx|d	7 }z| j| W S  tk
�r6 } z’|  ||¡}t| j	t
ƒs´t‚|| j|< |d t| jƒk �r| j	d|d … t|ƒ | j	|d d d…  | _	n|  j	t|ƒ7  _	d
| _| W Y ¢S d}~X Y nX ndt|ƒ› �}t|ƒ‚dS )zgGiven an rgb tuple, allocate palette entry.

        .. warning:: This method is experimental.
        r*   r   é   r3   éÿ   z/cannot add non-opaque RGBA color to RGB paletteNZRGBA©r?   r   zunknown color specifier: )r   r,   r-   r!   r   r   r$   ÚKeyErrorr=   r   r   ÚAssertionErrorr   r)   r   Úrepr)r   r#   r1   r/   r2   r;   r   r   r   ÚgetcolorŒ   s>    	


ÿþÿzImagePalette.getcolorzstr | IO[str])Úfpr   c              
   C  sÖ   | j rd}t|ƒ‚t|tƒr&t|dƒ}| d¡ | d| j› d�¡ tdƒD ]|}| |› ¡ t|t| jƒ |d t| jƒ ƒD ]>}z| d| j	| › �¡ W q~ t
k
rº   | d	¡ Y q~X q~| d¡ qL| ¡  d
S )zVSave palette to text file.

        .. warning:: This method is experimental.
        r*   Úwz
# Palette
z# Mode: Ú
r5   r   ú z 0N)r   r,   r-   r   ÚopenÚwriter   r    r   r   Ú
IndexErrorÚclose)r   rE   r/   r"   Újr   r   r   Úsave·   s     


&zImagePalette.save)r   N)NN)N)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   Úpropertyr   Úsetterr$   r&   r(   r'   Útostringr=   rD   rN   r   r   r   r   r      s,     ý
   ÿ ý+r   r   )Údatar   c                 C  s   t ƒ }| |_||_d|_|S )Nr   )r   r   r   r   )r   rV   r   r   r   r   ÚrawÒ   s
    rW   r0   Úfloatz	list[int])ÚblackÚwhiter   c                   s.   | dkr‡ fdd„t dƒD ƒS d}t|ƒ‚d S )Nr   c                   s   g | ]}t ˆ | d  ƒ‘qS r@   ©r0   ©Ú.0r"   ©rZ   r   r   Ú
<listcomp>à   s     z#make_linear_lut.<locals>.<listcomp>r5   z"unavailable when black is non-zero)r    ÚNotImplementedError)rY   rZ   r/   r   r^   r   Úmake_linear_lutÞ   s    ra   )Úexpr   c                   s   ‡ fdd„t dƒD ƒS )Nc                   s$   g | ]}t |d  ˆ  d  d ƒ‘qS )g     ào@g      à?r[   r\   ©rb   r   r   r_   ç   s     z"make_gamma_lut.<locals>.<listcomp>r5   )r    rc   r   rc   r   Úmake_gamma_lutæ   s    rd   r   r   )r   r   c                   s4   t tdtˆ ƒ ƒƒ}| ¡  tˆ ‡ fdd„|D ƒƒS )Nr5   c                   s   g | ]}|t ˆ ƒ ‘qS r   ©r   r\   ©r   r   r   r_   í   s     znegative.<locals>.<listcomp>)r9   r    r   Úreverser   ©r   r   r   rf   r   Únegativeê   s    ri   c                   s4   ddl m‰  ‡ fdd„tdt| ƒ ƒD ƒ}t| |ƒS )Nr   ©Úrandintc                   s   g | ]}ˆ d dƒ‘qS )r   r?   r   )r]   Ú_rj   r   r   r_   ó   s     zrandom.<locals>.<listcomp>r5   )Úrandomrk   r    r   r   rh   r   rj   r   rm   ð   s    rm   ú#fff0c0)rZ   r   c                   s0   dd„ t  | ¡D ƒ‰ td‡ fdd„tdƒD ƒƒS )Nc                 S  s   g | ]}t d |ƒ‘qS )r   )ra   )r]   Zbandr   r   r   r_   ø   s     zsepia.<locals>.<listcomp>r   c                   s    g | ]}ˆ |d   |d   ‘qS )r3   r   r\   ©Zbandsr   r   r_   ù   s     i   )r	   Zgetrgbr   r    r^   r   ro   r   Úsepia÷   s    rp   c                   s,   t tdtˆ ƒ ƒƒ}tˆ ‡ fdd„|D ƒƒS )Nr5   c                   s   g | ]}|t ˆ ƒ ‘qS r   re   r\   rf   r   r   r_   þ   s     zwedge.<locals>.<listcomp>)r9   r    r   r   rh   r   rf   r   Úwedgeü   s    rq   ztuple[bytes, str])Úfilenamer   c                 C  s~   t | dƒ�j}tjtjtjg}|D ]B}z$| d¡ ||ƒ ¡ }|rFW  qpW q  ttfk
r`   Y q X q d}t|ƒ‚W 5 Q R X |S )NÚrbr   zcannot load palette)	rI   r   r   r
   ÚseekZ
getpaletteÚSyntaxErrorr,   ÚOSError)rr   rE   ZpaletteHandlersZpaletteHandlerZlutr/   r   r   r   Úload  s"    ýú

rw   )r   )r   )rn   )r   )Ú
__future__r   r.   Útypingr   r   r   Ú r   r   r	   r
   r   r   rW   ra   rd   ri   rm   rp   rq   rw   r   r   r   r   Ú<module>   s    6