U
ʗRe` @ s2 d dl Z d dlmZ d dlmZ d dlmZmZmZm Z e j
dkrPd dlm Z n d dl m Z G dd de Z
ee e
ee d d
d Zeee ee ee ee d d
dZdeee eee ee dddZedkr.d dlmZ eG dd dZededddedddedddgZeee dS ) N)Fraction)ceil)castListOptionalSequence) )Protocolc @ s: e Zd ZU dZdZee ed< dZeed< dZ eed< dS )Edgez1Any object that defines an edge (such as Layout).Nsize ratio minimum_size)
__name__
__module__ __qualname____doc__r r int__annotations__r r r r /builddir/build/BUILDROOT/alt-python38-pip-22.2.1-2.el8.x86_64/opt/alt/python38/lib/python3.8/site-packages/pip/_vendor/rich/_ratio.pyr s
r )totaledgesreturnc C s dd |D }t }d|krdd tt||D }| tdd |D }|dkrddd t||D S ||td d |D }|D ]&\}}||j |jkr|j||< qq|d} |D ](\}}t||j | d
\}
} |
||< qqqttt |S ) 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.
c S s g | ]}|j pd qS N)r ).0edger r r
% s z!ratio_resolve..Nc S s$ g | ] \}\}}|d kr||fqS r r )r indexr r r r r r , s
c s s | ]}|p d V qdS )r Nr )r r r r r 2 s z ratio_resolve..r c S s& g | ] \}}|d kr |j p dn|qS )Nr
)r )r r r r r r r 5 s c s s | ]\}}|j pd V qdS )r
N)r )r _r r r r r ; s r
)
r enumeratezipsumr r divmodr r r ) r r sizes _Fractionflexible_edges remainingportionr r remainderr r r r
ratio_resolve s2
r, )r ratiosmaximumsvaluesr c C s dd t ||D }t|}|s,|dd S | }g }|j}t |||D ]R\}} }
|r|dkrt| t|| | } ||
| || 8 }||8 }qF||
qF|S )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.
c S s g | ]\}}|r|nd qS r r )r r _maxr r r r _ s z ratio_reduce..Nr )r# r$ appendminround) r r- r. r/ total_ratiototal_remainingresultr2 r maximumvalue distributedr r r ratio_reduceQ s
r; )r r- minimumsr c C s |rdd t ||D }t|}|dks0td| }g }|j}|dkrVdgt| }n|}t ||D ]D\}} |dkrt| t|| | }
n|}
||
||8 }||
8 }qd|S )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.
c S s g | ]\}}|r|nd qS r0 r )r r _minr r r r s z$ratio_distribute..r zSum of ratios must be > 0N)r# r$ AssertionErrorr2 lenmaxr ) r r- r< r5 r6 distributed_totalr2 _minimumsr minimumr: r r r ratio_distributeq s$
rD __main__) dataclassc @ s6 e Zd ZU dZee ed< dZeed< dZeed< dS )ENr r
r r ) r r r r r r r r r r r r r rG s
rG n r
)N)sys fractionsr mathr typingr r r r version_infor
Z pip._vendor.typing_extensionsr r r, r; rD r dataclassesrF rG resolvedprintr$ r r r r s6
> !
$
(