^j, DUdZddlmZddlZddlZddlmZGddejZeddd d ed dd d ed dd d eddd d edddd edddd dZ de d<GddejZ GddejZ y)aV `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 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)wcswidthc6eZdZUded<ded<ded<ded<y) _FieldMetastrnameintlowhighdefaultN)__name__ __module__ __qualname____annotations__L/opt/ringagent/.cad-venv/lib/python3.12/site-packages/wcwidth/text_sizing.pyrr*s I H I Lrrscale)r r r r width numerator denominatorvertical_alignhorizontal_align)swndvhzdict[str, _FieldMeta]TEXT_FIELD_MAPPINGceZdZUdZdZded<dZded<dZded<dZded<dZ ded <dZ ded <dd Z dd Z e ddd Zy)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). rr rrrrrrrcdjfdtjD}jjd|dS)z Return a compact representation including only non-default fields. This avoids verbose output when most fields are defaults. z, c3K|]L}t|j|jk7r'|jdt|jNyw)=N)getattrr r ).0fieldselfs r z,TextSizingParams.__repr__..UsML!& 'ejj 9U]] J!& |1WT5::-F,GHLsAA())joinr$values __class__r)r-repr_fmts` r__repr__zTextSizingParams.__repr__NsL99L*<*C*C*ELL..))*!H:Q77rcg}tjD]A\}}t||jx}|jk7s,|j |d|Cdj |S)z0Build and return sub-part of an OSC 66 sequence.r):)r$itemsr*r r appendr1)r-parts field_keyr,vals r make_sequencezTextSizingParams.make_sequenceZsg 2 8 8 : 3 IutUZZ00U]]B {!C512 3xxrci}|s|S|jdD]}d|vr|dk(rtd|d||jd\}}}tj |}||dk(rtd|d|b t |} |dk(rY| |j kDs| |jkr;td |d|d |d |jd|jd|j t|jt|j | ||j<|di|S#t$r%} |dk(rtd |d|d | | Yd} ~ 7d} ~ wwxYw)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, numerator=0, denominator=0, vertical_align=0, horizontal_align=0) r7r)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$getr r r r maxmin) clsraw control_codeskwargspartkey_eqr<r,valueexcs r from_paramszTextSizingParams.from_paramscs$)+5LIIcN HD$ H,$(,,083H&QRR NN3/MCc&**3/E} H,$'B3%H<>> TextSizing.from_match(TEXT_SIZING_PATTERN.match('\x1b]66;w=2;XY\x07')) TextSizing(params=TextSizingParams(scale=1, width=2, numerator=0, denominator=0, \ vertical_align=0, horizontal_align=0), text='XY', terminator='\x07') r)rHr)rXrYrZ)r&rOgroup)rFmatchrHs r from_matchzTextSizing.from_matchsC&*66u{{1~Ub6c A#kk!n. .rc|jjdkDr-|jj|jjzSt|j|}|dkrd}|jj|zS)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)ambiguous_width)rXrrrrY)r-rars r display_widthzTextSizing.display_widthsf ;;  q ;;$$t{{'8'88 8 TYY @ q5A{{  1$$rcnd|jjd|j|jS)z3Build and return complete OSC 66 Terminal Sequence.z]66;;)rXr=rYrZ)r-s rr=zTextSizing.make_sequences0$++3356a {4??BSTTrNrR)r^z re.Match[str]rHrrQrW)r)rar rQr rP) rrrrTrrUr_rbr=rrrrWrWs2L  IO..,%&UrrW) rT __future__rretyping _wcswidthr NamedTuplerr$rr&rWrrrrjs<#  "" aa ; aa ; !"a @ AB B )qq! D +A F -H)HZv((Zz3U""3Ur