�
"T�e� � � � d dl mZ d dlmZ d dlmZmZmZ ddlm Z erd dl
m Z G d� d� Z e
d k( rLd d
lZd d lmZ d d lmZ d d
lmZmZ d dlmZ d dlmZ G d� d� Z e� Zej7 e� � y
y
)� )�sqrt)� lru_cache)�Sequence�Tuple�
TYPE_CHECKING� )� ColorTriplet��Tablec �z � e Zd ZdZdeeeeef fd�Zdedefd�Z d
d�Z
e d� � d
eeeef defd �� Z y )�Palettez A palette of available colors.�colorsc � � || _ y �N)�_colors)�selfr s �./usr/lib/python3/dist-packages/rich/palette.py�__init__zPalette.__init__ s � � �� � �number�returnc �, � t | j | � S r )r r )r r s r � __getitem__zPalette.__getitem__ s � ��T�\�\�&� 1�2�2r c
�< � ddl m} ddlm} ddlm} ddlm} |dddd t | j � � d
�d d �
� }t | j � D ]F \ }}|j t |� t |� |d | |j |� �� �� � �H |S )Nr ��Color��Style)�Textr
�index�RGBr r
z colorsT�right)�title�caption� highlight�caption_justifyz )�bgcolor)�style)�
rich.colorr �
rich.styler � rich.textr �
rich.tabler �lenr � enumerate�add_row�str�repr�from_rgb)r r r r r �tabler �colors r �__rich__zPalette.__rich__ s� � �$�$�"�$�� � � ���4�<�<�(�)��1�� #�
�� &�d�l�l� 3� �L�E�5� �M�M��E�
��U� ��X�U�>�5�>�>�5�3I�%J�K�
� � � r i )�maxsizer4 c �� ������ |\ ���t �| j j �dt dt f�����fd� }t t
t | j � � |�� }|S )z�Find 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 � �� �
| � \ }}}� |z dz }� |z
}� |z
}� |z
} �d|z |z |z dz d|z |z z d|z
|z |z dz z � S )z Get the distance to a color.� i � � i� � )
r �red2�green2�blue2�red_mean�red�green�blue�_sqrt�blue1� get_color�green1�red1s
�����r �get_color_distancez)Palette.match.<locals>.get_color_distance: s� �� �"+�E�"2� �D�&�%� �t� ��)�H���+�C��V�O�E��5�=�D����>�S�(�3�.�1�4��e�)�e�#�$��8�^�t�+�d�2�q�8�:��
r )�key)r r r �int�float�min�ranger- ) r r4 rI � min_indexrD rE rF rG rH s @@@@@r �matchz
Palette.match, s` �� � $���f�e����L�L�,�,� � �c� �e� � � ��c�$�,�,� /�0�6H�I� ��r N)r r )
�__name__�
__module__� __qualname__�__doc__r r rK r r r r5 r rP r< r r r
r
sk � �(� �x��c�3��m�(<� =� �3�#� 3�,� 3��0 �t� ��5��c�3� �/� �C� � �r r
�__main__N)�Iterabler )�Console�ConsoleOptions)�Segmentr c �& � e Zd Zdededee fd�Zy)�ColorBox�console�optionsr c # � K � |j j dz
}t d|� D ]� }t |j � D ]� }||j z }||dz z }t j
||d� \ }} }
t j
||d|z dz z d� \ } } }
t
j |dz | dz |
dz � }t
j | dz | dz |
dz � }t dt ||�� � �� �� t j � �� �� y �w) N� r r g �?r9 � u ▄)r4 r' ) �size�heightrN � max_width�colorsys�
hls_to_rgbr r2 rY r �line)r r\ r] rb �y�x�h�l�r1�g1�b1�r2�g2�b2r' r4 s r �__rich_console__zColorBox.__rich_console__T s � �� � �\�\�(�(�1�,�F��1�f�%�
%�� �w� 0� 0�1� N�A��G� -� -�-�A��V�a�Z�(�A�!)�!4�!4�Q��3�!?�J�B��B�!)�!4�!4�Q��Q��Z�!�^�8L�c�!R�J�B��B� #�n�n�R�#�X�r�C�x��c�� J�G� !�N�N�2��8�R�#�X�r�C�x� H�E�!�%��U�G�)L�M�M�N� �l�l�n�$�
%�s �D D
N)rQ rR rS rW rX rV rY rq r< r r r[ r[ S s$ � �
%�"�
%�-;�
%�
�g�
�
%r r[ ) �mathr � functoolsr �typingr r r �
color_tripletr r, r r
rQ rd rV r) r � rich.consolerW rX � rich.segmentrY r* r r[ r\ �printr< r r �<module>ry sg �� � � 1� 1� '�� �=� =�@ �z��� � �4�$� �%� %� �i�G� �M�M�(�*� �3 r |