Current File : //lib/python3/dist-packages/rich/__pycache__/_ratio.cpython-312.pyc
�
"T�eS � � � d dl Z d dlmZ d dlmZ d dlmZmZmZm Z e j dk\ rd dlm Z nd dl m Z G d� de � Z
d ed
e e
d ee fd �Zd ed
ee dee dee d ee f
d�Z dd ed
ee deee d ee fd�Zedk( rHd dlmZ e G d� d� � Z ed eddd� eddd� eddd� g� Z e ee� � yy)� N)�Fraction)�ceil)�cast�List�Optional�Sequence)� � )�Protocolc �B � e Zd ZU dZdZee ed< dZeed< dZ eed< y)�Edgez1Any object that defines an edge (such as Layout).N�size� �ratio� minimum_size)
�__name__�
__module__� __qualname__�__doc__r r �int�__annotations__r r � � �-/usr/lib/python3/dist-packages/rich/_ratio.pyr
r
s& � �;� �D�(�3�-� ��E�3�N��L�#�r r
�total�edges�returnc � � |D �cg c] }|j xs d�� }}t }d|v �rt t ||� � D ���cg c] \ }\ }}|�||f�� }}}}| t d� |D � � z
}|dk r1t ||� D ��cg c] \ }}|�|j
xs dn|�� c}}S ||t d� |D � � � } |D ]2 \ }}| |j z |j
k s�#|j
||< n8 |d� }
|D ]) \ }}t | |j z |
z d� \ }}
|||< �+ nd|v r��t t t |� S c c}w c c}}}w c c}}w )a� Divide total space to satisfy size, ratio, and minimum_size, constraints.
The returned list of integers should add up to total in most cases, unless it is
impossible to satisfy all the constraints. For instance, if there are two edges
with a minimum size of 20 each and `total` is 30 then the returned list will be
greater than total. In practice, this would mean that a Layout object would
clip the rows that would overflow the screen height.
Args:
total (int): Total number of characters.
edges (List[Edge]): Edges within total space.
Returns:
List[int]: Number of characters for each edge.
Nc 3 �( K � | ]
}|xs d �� � y�w)r Nr )�.0r s r � <genexpr>z ratio_resolve.<locals>.<genexpr>2 s � �� � <�d�� �� � <�s �r r c 3 �B K � | ] \ }}|j xs d �� � y�w)r N)r )r �_�edges r r! z ratio_resolve.<locals>.<genexpr>; s � �� �K���D�D�J�J�O�!�O�K�s � ) r r � enumerate�zip�sumr r �divmodr r r ) r r r$ �sizes� _Fraction�indexr �flexible_edges� remaining�portion� remainders r �
ratio_resolver0 s� � �" .3� 3�T�d�i�i� �4� � 3�E� 3��I� �%�-� (1��U�E�1B�'C�
�
�#��|��d��|� �D�M�
��
� �C� <�e� <� <�<� � ��>� #&�e�U�"3�� �D�$� .2�\�$�#�#�(�q�t�C��
�
� �s�K�N�K�K�
��
*�
�K�E�4�����#�t�'8�'8�8� #� 0� 0��e� ��
� "�!� �I� -�
$� ��t�"(��4�:�:�)=� �)I�1�"M���i� #��e� �
$�
�E �%�-�H ��S� �5� !�!��S
4��
��s �E�E
� E�ratios�maximums�valuesc �F � t ||� D ��cg c] \ }}|r|nd��
}}}t |� }|s|dd S | }g }|j } t |||� D ]F \ }}
} |r6|dkD r1t |
t ||z |z � � } | | | z
� || z }||z }�? | | � �H |S c c}}w )ad Divide an integer total in to parts based on ratios.
Args:
total (int): The total to divide.
ratios (List[int]): A list of integer ratios.
maximums (List[int]): List of maximums values for each slot.
values (List[int]): List of values
Returns:
List[int]: A list of integers guaranteed to sum to total.
r N)r& r' �append�min�round)
r r1 r2 r3 r �_max� total_ratio�total_remaining�resultr5 �maximum�value� distributeds
r � ratio_reducer? Q s� � � 7:�&�(�6K�
L�{�u�d�t�e��"�
L�F�
L��f�+�K� ��a�y���O��F�
�]�]�F�!$�V�X�v�!>� � ��w�� �[�1�_� �g�u�U�_�-D�{�-R�'S�T�K� �5�;�&� '� �{� *�O� �5� �K� �5�M�� �M�� Ms �B �minimumsc �b � |r"t ||� D ��cg c] \ }}|r|nd��
}}}t |� }|dkD sJ d� �| }g }|j }|�dgt |� z } n|} t || � D ]: \ }}
|dkD r t |
t ||z |z � � } n|} || � ||z }|| z }�< |S c c}}w )a< Distribute an integer total in to parts based on ratios.
Args:
total (int): The total to divide.
ratios (List[int]): A list of integer ratios.
minimums (List[int]): List of minimum values for each slot.
Returns:
List[int]: A list of integers guaranteed to sum to total.
r zSum of ratios must be > 0)r& r' r5 �len�maxr ) r r1 r@ r �_minr9 r: �distributed_totalr5 � _minimums�minimumr> s r �ratio_distributerH q s� � � �:=�f�h�:O�P�;�5�$�4�%�Q�&�P��P��f�+�K� ��?�7� 7�7�?��O�#%��
�
%�
%�F����C�#�f�+�%� � � � �f�i�0� '���w� ��?� �g�t�E�O�,C�k�,Q�'R�S�K�)�K��{���u� � ��;�&��'� � ��' Qs �B+�__main__)� dataclassc �> � e Zd ZU dZee ed< dZeed< dZeed< y)�ENr r r r ) r r r r r r r r r r r r rL rL � s# � � "��h�s�m�"���s�� � �c� r rL �n r )N)�sys� fractionsr �mathr �typingr r r r � version_infor �typing_extensionsr
r r0 r? rH r � dataclassesrJ rL �resolved�printr' r r r �<module>rW s5 ��
� � � 1� 1����v� � �*��8� �:"�� :"�X�d�^� :"��S� � :"�z � � � �S� � �-1�#�Y� �@D�S� � � �#�Y� �B DH�! � �! � �S� �! �-5�d�3�i�-@�! � �#�Y�! �H �z��%�� � � � �
�S�1�T�1�a�=�!�D�!�Q�-��4��A� �"O�P�H� �#�h�-�� r
Mini Shell