o j  @sUdZddlmZddlZddlZddlmZGdddejZeddd dd ed dd dd ed dd dd eddd dd eddddd eddddd dZ de d<GdddejZ GdddejZ dS)a `kitty text sizing protocol`_ (OSC 66) parsing and measurement. The kitty text sizing protocol allows terminal apps to explicitly tell terminals how many cells text occupies, using the escape sequence:: ESC ] 66 ; metadata [ ; text ] BEL/ST The ``text`` field is optional; when omitted, the sequence occupies ``s * w`` cells (or zero when ``w`` is also unset). Metadata is colon-separated ``key=value`` pairs: - ``s``: scale - ``w``: width in cells - ``n``: fractional numerator - ``d``: fractional denominator - ``v``: vertical alignment - ``h``: horizontal alignment Parsing is pretty straight-forward: - When ``w > 0``, return ``s * w``. - Otherwise ``w == 0``, ``s * wcswidth(inner_text_width)`` cells. Numerator, denominator, and alignment codes and values are parsed but otherwise ignored and have no effect on measurements made in this library. .. _`kitty text sizing protocol`: https://sw.kovidgoyal.net/kitty/text-sizing-protocol/ .. versionadded:: 0.7.0 ) annotationsN)wcswidthc@s.eZdZUded<ded<ded<ded<dS) _FieldMetastrnameintlowhighdefaultN)__name__ __module__ __qualname____annotations__rr/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/wcwidth/text_sizing.pyr-s  rscale)rr r r width numerator denominatorvertical_alignhorizontal_align)swndvhzdict[str, _FieldMeta]TEXT_FIELD_MAPPINGc@s~eZdZUdZdZded<dZded<dZded<dZded<dZ ded <dZ ded <dd dZ dddZ e ddddZdS)TextSizingParamsa/ Parsed parameters from a text sizing escape sequence (OSC 66). :param scale: Scale factor (1-7). Text occupies ``scale`` rows tall and ``scale * width`` columns wide. :param width: Width in cells (0-7). When 0, width is auto-calculated from the inner text. :param numerator: Fractional scaling numerator (0-15). :param denominator: Fractional scaling denominator (0-15). :param vertical_align: Vertical alignment (0=top, 1=bottom, 2=center). :param horizontal_align: Horizontal alignment (0=left, 1=right, 2=center). rrrrrrrrrreturnrcs0dfddtD}jjd|dS)z Return a compact representation including only non-default fields. This avoids verbose output when most fields are defaults. z, c3s:|]}t|j|jkr|jdt|jVqdS)=N)getattrrr ).0fieldselfrr Xsz,TextSizingParams.__repr__..())joinr!values __class__r )r)repr_fmtrr(r__repr__QszTextSizingParams.__repr__cCsJg}tD]\}}t||j}|jkr||d|qd|S)z0Build and return sub-part of an OSC 66 sequence.r$:)r!itemsr%rr appendr-)r)parts field_keyr'valrrr make_sequence]s  zTextSizingParams.make_sequenceparseraw control_codesc CsPi}|s|S|dD]}d|vr!|dkr td|d|q |d\}}}t|}|durA|dkr@td|d|q zt|} Wn$tyk} z|dkratd |d|d | | WYd} ~ q d} ~ ww|dkr| |jksz| |jkrtd |d|d |d |jd|jd|j t |jt |j| ||j<q |di|S)a: Parse colon-separated ``key=value`` metadata string. :param raw: Metadata string, e.g. ``'s=2:w=3'``. :param control_codes: 'parse' or 'strict'. :raises ValueError: If ``control_codes='strict'`` unrecognized text sizing parameters raise ValueError. :returns: Parsed parameters with values clamped to valid ranges. Unknown keys are ignored. Non-integer values use defaults. Example:: >>> TextSizingParams.from_params('s=2:w=3') TextSizingParams(scale=2, width=3) r2r$strictz5Expected '=' in text sizing parameter (key=val), got z in OSC 66 sequence, NzUnknown text sizing field 'z' in OSC 66 sequence, zIllegal text sizing value 'z: z!Out of bounds text sizing value 'z: allowed range for 'z' (z) is z to r) split ValueError partitionr!getrr r rmaxmin) clsr:r;kwargspartkey_eqr7r'valueexcrrr from_paramsfs`      zTextSizingParams.from_paramsNr#rr9)r:rr;rr#r")r r r__doc__rrrrrrrr1r8 classmethodrJrrrrr"=s         r"c@sPeZdZUdZded<ded<ded<eddd d ZddddZdddZdS) TextSizingzBBasic horizontal width measurement for kitty text sizing protocol.r"paramsrtext terminatorr9match re.Match[str]r;r#cCs.|tj|d|d|dpd|ddS)a Parse using matching OSC 66 Sequence. :param match: match object from :attr:`wcwidth.escape_sequences.TEXT_SIZING_PATTERN`. :param control_codes: 'parse' or 'strict', same meaning as delegated by :func:`wcwidth.width`. :raises ValueError: When ``control_codes='strict'`` for unrecognized, invalid, or out of bounds text sizing parameters. :returns: TextSizing object from parsed sequence Example:: >>> from wcwidth.escape_sequences import TEXT_SIZING_PATTERN >>> TextSizing.from_match(TEXT_SIZING_PATTERN.match('\x1b]66;w=2;XY\x07')) TextSizing(params=TextSizingParams(width=2), text='XY', terminator='\x07') r)r;r)rPrQrR)r"rJgroup)rCrSr;rrr from_matchs zTextSizing.from_matchrambiguous_widthrcCsB|jjdkr|jj|jjSt|j|d}|dkrd}|jj|S)a  Calculate the display width of a text sizing sequence. :param ambiguous_width: Width for East Asian Ambiguous characters. :returns: Display width in terminal cells. When ``width > 0``, returns ``params.scale * params.width``. When ``width == 0``, returns ``params.scale * measured_inner_width``. .. note: Fractional scaling (numerator/denominator) does not affect the cell count, it adjusts only the font size within the cells allocated by 'w'. r)rY)rPrrrrQ)r)rYrrrr display_widths  zTextSizing.display_widthcCsd|jd|j|jS)z3Build and return complete OSC 66 Terminal Sequence.z]66;;)rPr8rQrRr(rrrr8szTextSizing.make_sequenceNrL)rSrTr;rr#rO)r)rYrr#rrK) r r rrMrrNrXrZr8rrrrrOs  rO) rM __future__rretyping _wcswidthr NamedTuplerr!rr"rOrrrrs !  \