o Ďib*@s&UddlmZddlmZddlmZddlmZmZm Z m Z ddl m Z e eeefZedZgdZded <ed d eDZejZd ed <GdddeZeddd2d3ddZedd2d4ddZd2d4ddZd4dd Z d2d5d"d#Z d2d6d&d'Z d2d6d(d)Zd2d7d+d,Zd2d8d/d0Z d1S)9) annotations) lru_cache) itemgetter)Callable NamedTupleSequenceTuple)load)) ~))i)ipi)i%i%)i(i(zlist[tuple[int, int]]_SINGLE_CELL_UNICODE_RANGEScCs.g|]\}}ttt||dD]}|qqS))mapchrrange).0_start_end characterrF/home/jeff/fluffinator/venv/lib/python3.10/site-packages/rich/cells.py srzCallable[[str], bool]_is_single_cell_widthsc@s*eZdZUdZded<ded<ded<dS) CellTablezDContains unicode data required to measure the cell widths of glyphs.strunicode_versionzSequence[tuple[int, int, int]]widthszfrozenset[str]narrow_to_wideN)__name__ __module__ __qualname____doc____annotations__rrrrr&s  ri)maxsizeautorrrreturnintc Cst|}|r |dksd|krdkrdSndSt|j}|d}||dkr*dSd}t|d}||kr[||d?}||\}} } ||krL|d}n || krU|d}n| S||ks6dS)aGet the cell size of a character. Args: character (str): A single character. unicode_version: Unicode version, `"auto"` to auto detect, `"latest"` for the latest unicode version. Returns: int: Number of cells (0, 1 or 2) occupied by that character. r r rr)ordload_cell_tabler len) rr codepointtable last_entry lower_bound upper_boundindexstartendwidthrrrget_character_cell_size.s*        r9textcCs t||S)aGet the number of cells required to display text. This method always caches, which may use up a lot of memory. It is recommended to use `cell_len` over this method. Args: text (str): Text to display. unicode_version: Unicode version, `"auto"` to auto detect, `"latest"` for the latest unicode version. Returns: int: Get the number of cells required to display text. ) _cell_lenr:rrrrcached_cell_lenQs r=cCs t|dkr t||St||S) Get the cell length of a string (length as it appears in the terminal). Args: text: String to measure. unicode_version: Unicode version, `"auto"` to auto detect, `"latest"` for the latest unicode version. Returns: Length of string in terminal cells. i)r/r=r;r<rrrcell_lenbs  r?c st|rt|Sd|vrd|vrtfdd|DSt}d}d}ddh}d}t|}||krc||}||vrN|dkrB|d7}n|rM|||jv7}d}n t|} r[|}|| 7}|d7}||ks1|S)r>‍️c3s|]}t|VqdS)N)r9)rrrrr s  z_cell_len..rNr)rr/sumr.r!r9) r:r cell_table total_widthlast_measured_characterSPECIALr5character_countrcharacter_widthrrBrr;qs6   r;'tuple[list[CellSpan], int]'cCs*t|}t|}d}d}d}g}ddh}||kr||} |vr_| dkr9|d7}|r8|d\} } } | || f|d<n%|r^|d7}|r^|d\} } } ||jvrWd}| d7} |d7}| || f|d<qt| |} ry| }|||d}| f|| 7}n|r|d\} } } | |d}| f|d<||ks||fS)zDivide text into spans that define a single grapheme. Args: text: String to split. unicode_version: Unicode version, `"auto"` to auto detect, `"latest"` for the latest unicode version. Returns: List of spans. rNr@rAr r,r)r.r/r!r9append)r:rrEcodepoint_countr5rGrFspansrHrr6r cell_lengthrJrrrsplit_graphemessD   rP cell_positiontuple[str, str]c Cs.|dkrd|fSt||\}}t||t|}ttt|d|} ||krE|t|kr3|dfS||d}|d|||dfS||kro||\}} } || |krf|d|dd|| dfS|d7}|| 7}n'||d\}} } || |kr|d|dd|| dfS|d8}|| 8}q%)pSplit text by cell position. If the cell position falls within a double width character, it is converted to two spaces. Args: text: Text to split. cell_position Offset in cells. unicode_version: Unicode version, `"auto"` to auto detect, `"latest"` for the latest unicode version. Returns: Tuple to two split strings. rNT r)rPr*r/rDr_span_get_cell_len) r:rQrrNrOoffset left_size split_indexr6r7 cell_sizerrr _split_texts.       r[cCs,t|r|d|||dfSt|||S)rSN)rr[)r:rQrrrr split_texts r\totalcCst|rt|}||kr|d||S|d|S|dkr dSt|}||kr*|S||kr6|d||St|||\}}|S)a$Adjust a string by cropping or padding with spaces such that it fits within the given number of cells. Args: text: String to adjust. total: Desired size in cells. unicode_version: Unicode version. Returns: A string with cell size equal to total. rUNrrT)rr/r?r[)r:r]rsizerZ_rrr set_cell_sizes  r`r8 list[str]c strfddtdtDSt|\}}d}g}d}|D]\}} } || kr;||||}d}|| 7}q#|rK||d|S)ahSplit text into lines such that each line fits within the available (cell) width. Args: text: The text to fold such that it fits in the given width. width: The width available (number of cells). Returns: A list of strings such that each string in the list has cell width less than or equal to the available width. csg|] }||qSrr)rr5r:r8rrr=szchop_cells..rN)rrr/rPrL) r:r8rrNr_ line_sizelines line_offsetr6r7rZrrbr chop_cells1s   rfN)r()rrrrr)r*)r:rrrr)r*)r:rrrr)rK)r:rrQr*rrr)rR)r:rr]r*rrr)r)r:rr8r*rrr)ra)! __future__r functoolsroperatorrtypingrrrrrich._unicode_datar r.r*CellSpanrVrr& frozenset _SINGLE_CELLS issupersetrrr9r=r?r;rPr[r\r`rfrrrrs:     "  1 6 -