o
    
i#                     @  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
mZ dZer,ddlmZ G d	d
 d
Zd+ddZd,ddZd-ddZd.d/ddZd.d/d d!Zd0d1d#d$Zd.d/d%d&Zd2d)d*ZdS )3    )annotationsN)Sequence)IO   )GimpGradientFileGimpPaletteFile
ImageColorPaletteFileF)Imagec                   @  s   e Zd ZdZ		d,d-d
dZed.ddZejd/ddZed0ddZejd1ddZd2ddZ	d3ddZ
d4ddZeZ	d5d6d"d#Z	d7d8d&d'Zd9d*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modestrpalette(Sequence[int] | bytes | bytearray | NonereturnNonec                 C  s"   || _ d | _|p
t | _d | _d S N)r   rawmode	bytearrayr   dirty)selfr   r    r   N/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/PIL/ImagePalette.py__init__+   s   
zImagePalette.__init__!Sequence[int] | bytes | bytearrayc                 C  s   | j S r   )_paletter   r   r   r   r   5   s   zImagePalette.palettec                 C  s   d | _ || _d S r   )_colorsr   )r   r   r   r   r   r   9   s   
dict[tuple[int, ...], int]c                 C  sh   | j d u r1t| j}i | _ tdt| j|D ]}t| j|||  }|| j v r)q|| | j |< q| j S )Nr   )r   lenr   ranger   tuple)r   mode_lenicolorr   r   r   colors>   s   


zImagePalette.colorsr&   c                 C  s
   || _ d S r   )r   )r   r&   r   r   r   r&   J   s   
c                 C  s<   t  }| j|_| j|_| jd ur| jd d  |_| j|_|S r   )r   r   r   r   r   )r   newr   r   r   copyN   s   
zImagePalette.copy-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   getdataY   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   msgarrr   r   r   r*   d   s   zImagePalette.tobytesimageImage.Image | NoneeException | Noneintc                 C  s   t | jtst| j| _t| jt| j }d}|r2|jd|jdf}||v r2|d7 }||v s*|dkr^|rStt	t
| D ]\}}|dkrR||vrR|} nqB|dkr^d}t|||S )Nr   
backgroundtransparencyr      r   z$cannot allocate more than 256 colors)r0   r   r   r   r    r   infogetreversedlist	enumerate	histogramr/   )r   r4   r6   indexspecial_colorsr$   countr2   r   r   r   _new_color_indext   s,   


zImagePalette._new_color_indexr%   tuple[int, ...]c              
   C  sP  | j r	d}t|t|tr| jdkr,t|dkr+|d dkr%d}t||dd }n| jdkr;t|dkr;|d	7 }z| j| W S  ty } zO| ||}t| j	t
sWJ || j|< t| j}|| t| jk r| j	d||  t| | j	|| | d  | _	n	|  j	t|7  _	d
| _|W  Y d}~S d}~ww dt| }t|)zgGiven an rgb tuple, allocate palette entry.

        .. warning:: This method is experimental.
        r-   r            z/cannot add non-opaque RGBA color to RGB paletteNRGBArI   r   zunknown color specifier: )r   r/   r0   r"   r   r    r&   KeyErrorrE   r   r   r   r,   r   repr)r   r%   r4   r2   r6   rB   r#   r   r   r   getcolor   sF   	




zImagePalette.getcolorfpstr | IO[str]c                 C  s   | j r	d}t|d}t|trt|d}d}zV|d |d| j d t| j}t	dD ]2}||  t	|t| j |d	 t| j D ]}|d
||k rW| j| nd  qI|d q0W |rl|
  dS dS |ru|
  w w )zVSave palette to text file.

        .. warning:: This method is experimental.
        r-   FwTz
# Palette
z# Mode: 
r;   r    r   N)r   r/   r0   r   openwriter   r    r   r!   close)r   rO   r2   open_fppalette_lenr$   jr   r   r   save   s,   



&$
zImagePalette.save)r   N)r   r   r   r   r   r   )r   r   )r   r   r   r   )r   r   )r&   r   r   r   )r   r   )r   r)   )r   r,   )NN)r4   r5   r6   r7   r   r8   r   )r%   rF   r4   r5   r   r8   )rO   rP   r   r   )__name__
__module____qualname____doc__r   propertyr   setterr&   r(   r+   r*   tostringrE   rN   rZ   r   r   r   r   r      s,    



,r   r   r   datar   r   c                 C  s   t  }| |_||_d|_|S )Nr   )r   r   r   r   )r   rb   r   r   r   r   raw   s
   rc   blackr8   whitefloat	list[int]c                   s*   | dkr fddt dD S d}t|)Nr   c                   s   g | ]
}t  | d  qS rK   r8   .0r$   re   r   r   
<listcomp>   s    z#make_linear_lut.<locals>.<listcomp>r;   z"unavailable when black is non-zero)r!   NotImplementedError)rd   re   r2   r   rk   r   make_linear_lut   s   rn   expc                   s    fddt dD S )Nc                   s$   g | ]}t |d    d  d qS )g     o@g      ?rh   ri   ro   r   r   rl      s   $ z"make_gamma_lut.<locals>.<listcomp>r;   )r!   rp   r   rp   r   make_gamma_lut   s   rq   r   r   c                   s4   t tdt  }|  t  fdd|D S )Nr;   c                      g | ]}|t   qS r   r    ri   r   r   r   rl          znegative.<locals>.<listcomp>)r?   r!   r    reverser   r   r   r   rt   r   negative   s   rx   c                   s4   ddl m   fddtdt|  D }t| |S )Nr   randintc                   s   g | ]} d dqS )r   rI   r   )rj   _ry   r   r   rl          zrandom.<locals>.<listcomp>r;   )randomrz   r!   r    r   rw   r   ry   r   r}      s   
r}   #fff0c0c                   s0   dd t | D  td fddtdD S )Nc                 S  s   g | ]}t d |qS )r   )rn   )rj   bandr   r   r   rl      r|   zsepia.<locals>.<listcomp>r   c                   s    g | ]} |d   |d   qS )rH   r   ri   bandsr   r   rl      s     i   )r   getrgbr   r!   rk   r   r   r   sepia   s   r   c                   s,   t tdt  }t  fdd|D S )Nr;   c                   rr   r   rs   ri   rt   r   r   rl     ru   zwedge.<locals>.<listcomp>)r?   r!   r    r   rw   r   rt   r   wedge  s   r   filenametuple[bytes, str]c                 C  s   t | d9}tjtjtjg}|D ]}z|d || }|r#W  nW q ttfy/   Y qw d}t|W d    |S 1 sAw   Y  |S )Nrbr   zcannot load palette)	rT   r   r   r	   seek
getpaletteSyntaxErrorr/   OSError)r   rO   paletteHandlerspaletteHandlerlutr2   r   r   r   load  s.   

r   )r   r   rb   r   r   r   )rd   r8   re   rf   r   rg   )ro   rf   r   rg   )r   )r   r   r   r   )r~   )re   r   r   r   )r   r   r   r   )
__future__r   r1   collections.abcr   typingr    r   r   r   r	   TYPE_CHECKINGr
   r   rc   rn   rq   rx   r}   r   r   r   r   r   r   r   <module>   s$    
:

