a
    ,gh                     @   s   d dl mZ d dlmZ d dlmZmZmZ ddlm	Z	 erHd dl
mZ G dd dZed	krd d
lZd dlmZ d dlmZ d dlmZmZ d dlmZ d dlmZ G dd dZe Zee  d
S )    )sqrt)	lru_cache)SequenceTupleTYPE_CHECKING   )ColorTripletTablec                   @   sn   e Zd ZdZeeeeef  dddZeedddZ	dd	d
dZ
eddeeeef edddZdS )PalettezA palette of available colors.)colorsc                 C   s
   || _ d S N)_colors)selfr    r   \/mounts/lovelace/software/anaconda3/envs/metaDMG/lib/python3.9/site-packages/rich/palette.py__init__   s    zPalette.__init__)numberreturnc                 C   s   t | j|  S r   )r   r   )r   r   r   r   r   __getitem__   s    zPalette.__getitem__r
   )r   c              
   C   s   ddl m} ddlm} ddlm} ddlm} |dddd	t| j	 d
ddd}t
| j	D ]2\}}|t|t||d||j| dd q\|S )Nr   ColorStyle)Textr	   indexRGBr   r   z colorsTright)titlecaption	highlightcaption_justifyz                )bgcolor)style)
rich.colorr   
rich.styler   Z	rich.textr   
rich.tabler
   lenr   	enumerateadd_rowstrreprfrom_rgb)r   r   r   r   r
   tabler   colorr   r   r   __rich__   s(    	zPalette.__rich__i   )maxsize)r.   r   c                    sL   |\t  | jjttd fdd}ttt| j|d}|S )zFind a color from a palette that most closely matches a given color.

        Args:
            color (Tuple[int, int, int]): RGB components in range 0 > 255.

        Returns:
            int: Index of closes matching color.
        )r   r   c                    sj   | \}}}| d }| }| }| } d| | | d? d| |  d| | | d?  S )zGet the distance to a color.   i         i  r   )r   red2green2blue2red_meanredgreenblue_sqrtblue1	get_colorgreen1red1r   r   get_color_distance:   s    
z)Palette.match.<locals>.get_color_distance)key)r   r   r   intfloatminranger'   )r   r.   rA   	min_indexr   r;   r   match,   s    

zPalette.matchN)__name__
__module____qualname____doc__r   r   rC   r   r   r   r/   r   rH   r   r   r   r   r      s   r   __main__N)Iterabler   )ConsoleConsoleOptions)Segmentr   c                   @   s"   e Zd Zeeee dddZdS )ColorBox)consoleoptionsr   c                 c   s   |j jd }td|D ]}t|jD ]}||j }||d  }t||d\}}	}
t||d| d  d\}}}t|d |	d |
d }t|d |d |d }tdt	||dV  q$t
 V  qd S )	N   r   r   g      ?r1      u   ▄)r.   r"   )sizeheightrF   	max_widthcolorsys
hls_to_rgbr   r,   rQ   r   line)r   rS   rT   rX   yxhlr1g1b1r2g2b2r"   r.   r   r   r   __rich_console__T   s    
 zColorBox.__rich_console__N)rI   rJ   rK   rO   rP   rN   rQ   rg   r   r   r   r   rR   S   s   rR   )mathr   	functoolsr   typingr   r   r   color_tripletr   r&   r
   r   rI   rZ   rN   r$   r   Zrich.consolerO   rP   Zrich.segmentrQ   r%   r   rR   rS   printr   r   r   r   <module>   s    @