a $lJh_@s2ddlZddlmZddlmZddlmZmZmZm Z ej dkrPddlm Z n ddl m Z Gddde Z ee e eed d d Zeeeeeeeeed d dZdeeeeeeeedddZedkr.ddlmZeGdddZededddedddedddgZeeedS)N)Fraction)ceil)castListOptionalSequence))Protocolc@s:eZdZUdZdZeeed<dZeed<dZ eed<dS)Edgez1Any object that defines an edge (such as Layout).Nsizeratio minimum_size) __name__ __module__ __qualname____doc__r rint__annotations__rrrrI/var/www/shaz/venv/lib/python3.9/site-packages/pip/_vendor/rich/_ratio.pyr s  r )totaledgesreturnc Csdd|D}t}d|vrddtt||D}|tdd|D}|dkrdddt||DS||td d|D}|D]&\}}||j|jkr|j||<qq|d} |D](\}}t||j| d \} } | ||<qqqttt |S) aDivide 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. cSsg|]}|jpdqSN)r ).0edgerrr %z!ratio_resolve..NcSs$g|]\}\}}|dur||fqSrr)rindexr rrrrr,s css|]}|p dVqdS)rNr)rr rrr 2rz ratio_resolve..rcSs&g|]\}}|dur|jp dn|qS)Nr )r)rr rrrrr5scss|]\}}|jpdVqdS)r N)r)r_rrrrr!;rr ) r enumeratezipsumrrdivmodrrr) rrsizesZ _FractionZflexible_edges remainingportionr r remainderr rrr ratio_resolves0     r+)rratiosmaximumsvaluesrc Csddt||D}t|}|s,|ddS|}g}|j}t|||D]R\}} } |r|dkrt| t|||} || | || 8}||8}qF|| qF|S)adDivide 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. cSsg|]\}}|r|ndqSrr)rrZ_maxrrrr_rz ratio_reduce..Nr)r$r%appendminround) rr,r-r. total_ratiototal_remainingresultr0rmaximumvalue distributedrrr ratio_reduceQs     r9)rr,minimumsrc Cs|rddt||D}t|}|dks0Jd|}g}|j}|durVdgt|}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. cSsg|]\}}|r|ndqSr/r)rrZ_minrrrrrz$ratio_distribute..rzSum of ratios must be > 0N)r$r%r0lenmaxr) rr,r:r3r4Zdistributed_totalr0Z _minimumsrZminimumr8rrrratio_distributeqs$  r=__main__) dataclassc@s6eZdZUdZeeed<dZeed<dZeed<dS)ENr r rr) rrrr rrrrrrrrrr@s  r@nr )N)sysZ fractionsrmathrtypingrrrr version_infor Zpip._vendor.typing_extensionsr rr+r9r=rZ dataclassesr?r@Zresolvedprintr%rrrrs,    > ! $  (