U Z+d (@sUddlmZddlmZddlmZmZmZmZm Z m Z m Z ddl m Z ddlmZddlmZerrddlmZd d gZGd d d Ze ed d Zded<dZGdd d ZGdddZdS)) annotations defaultdict) TYPE_CHECKINGCallable DefaultDictDictListOptionalTuple) FastDictCache)Point) get_cwidth)WindowScreenCharcC@seZdZUdZdZddddddd d d d d ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;dd?d@dAdBdCdDdEBZdFedG<dXdIdIdJdKdLdMZddNdOdPdQZddNdOdRdSZ e seZ e Z dIdTdUdVZ dWS)Yrz Represent a single character in a :class:`.Screen`. This should be considered immutable. :param char: A single character (can be a double-width character). :param style: A style string. (Can contain classnames.) )charstylewidthz^@z^Az^Bz^Cz^Dz^Ez^Fz^Gz^Hz^Iz^Jz^Kz^Lz^Mz^Nz^Oz^Pz^Qz^Rz^Sz^Tz^Uz^Vz^Wz^Xz^Yz^Zz^[z^\z^]z^^z^_z^?z<80>z<81>z<82>z<83>z<84>z<85>z<86>z<87>z<88>z<89>z<8a>z<8b>z<8c>z<8d>z<8e>z<8f>z<90>z<91>z<92>z<93>z<94>z<95>z<96>z<97>z<98>z<99>z<9a>z<9b>z<9c>z<9d>z<9e>z<9f> )B     €‚ƒ„…†‡ˆ‰Š‹ŒŽ‘’“”•–—˜™š›œžŸ zdict[str, str]display_mappingsstrNone)rrreturncCsH||jkr.|dkr|d7}n|d7}|j|}||_||_t||_dS)NrXz class:nbsp z class:control-character )rYrrrr)selfrrr_@/tmp/pip-unpacked-wheel-blk8czbf/prompt_toolkit/layout/screen.py__init__ls   z Char.__init__bool)otherr]cCs|j|jko|j|jkSNrrr^rcr_r_r`_equalsz Char._equalcCs|j|jkp|j|jkSrdrerfr_r_r` _not_equalszChar._not_equalr]cCs|jjd|jd|jdS)N(z, )) __class____name__rrr^r_r_r`__repr__sz Char.__repr__N)rrZ)rm __module__ __qualname____doc__ __slots__rY__annotations__rargrhr__eq____ne__ror_r_r_r`rs  Ii@B)sizez$FastDictCache[tuple[str, str], Char] _CHAR_CACHEz [transparent]c@seZdZdZd+ddddddd Zed d d d ZddddddZddddddZdddddZ dddddZ ddddddZ dd ddZ d dd!d"d#Z d,d&d d'dd(d)d*ZdS)-rz= Two dimensional buffer of :class:`.Char` instances. Nrz Char | Noneintr\) default_char initial_widthinitial_heightr]csp|dkrtdtfn|tfdd|_tdd|_i|_d|_i|_|pRd|_|p\d|_ i|_ g|_ dS)NrcstfddS)NcsSrdr_r_Z default_char2r_r`3Screen.__init__....rr_r}r_r`r~rz!Screen.__init__..cSs tddS)NcSsdS)NrZr_r_r_r_r`r~rrrr_r_r_r`r~rTr) rx Transparentr data_bufferZzero_width_escapescursor_positionsZ show_cursormenu_positionsrheight"visible_windows_to_write_positions_draw_float_functions)r^rzr{r|r_r}r`ras$   zScreen.__init__z list[Window]ricCst|jSrd)listrkeysrnr_r_r`visible_windowsszScreen.visible_windowsrr )windowpositionr]cCs||j|<dSz= Set the cursor position for a given window. N)rr^rrr_r_r`set_cursor_positionszScreen.set_cursor_positioncCs||j|<dSr)rrr_r_r`set_menu_positionszScreen.set_menu_position)rr]cCs2z |j|WStk r,tdddYSXdS)zX Get the cursor position for a given window. Returns a `Point`. rxyN)rKeyErrorr r^rr_r_r`get_cursor_positions zScreen.get_cursor_positionc Cs\z |j|WStk rVz|j|WYStk rPtdddYYSXYnXdS)z Get the menu position for a given window. (This falls back to the cursor position if no menu position was set.) rrN)rrrr rr_r_r`get_menu_positions zScreen.get_menu_positionzCallable[[], None])z_index draw_funcr]cCs|j||fdS)z Add a draw-function for a `Window` which has a >= 0 z_index. This will be postponed until `draw_all_floats` is called. N)rappend)r^rrr_r_r`draw_with_z_indexszScreen.draw_with_z_indexcCs:|jr6t|jddd}|dd|_|ddqdS)z? Draw all float functions in order of z-index. cSs|dSNrr_)itemr_r_r`r~rz(Screen.draw_all_floats..)keyrNr)rsorted)r^Z functionsr_r_r`draw_all_floatsszScreen.draw_all_floatsr[) style_strr]c CsR|j}t}d|}|D]2\}}|D] \}}||j|j|f||<q*qdS)zn For all the characters in the screen. Set the style string to the given `style_str`. rN)rrxitemsrr) r^rb char_cache append_stylerrowrrr_r_r`append_style_to_contents zScreen.append_style_to_contentrZF WritePositionrb)write_positionrafterr]cCs|s dS|j}|j|j}t}|j}|r:d|}d} n d}|d} t|j|j|jD]@} || } t||D](} | | } || j| | j |f| | <qpqZdS)z Fill the content of this area, using the given `style`. The style is prepended before whatever was here before. NrrZ) stripxposrrxrrangeyposrrr)r^rrrZxminZxmaxrrrZ prepend_stylerrrcellr_r_r` fill_areas*  zScreen.fill_area)Nrr)rZF)rmrprqrrrapropertyrrrrrrrrrr_r_r_r`rs"0  c@s0eZdZddddddddZdddd Zd S) rryr\)rrrrr]cCs4|dks t|dkst||_||_||_||_dSr)AssertionErrorrrrr)r^rrrrr_r_r`ra8s   zWritePosition.__init__r[ricCsd|jj|j|j|j|jS)Nz+{}(x={!r}, y={!r}, width={!r}, height={!r}))formatrlrmrrrrrnr_r_r`roBszWritePosition.__repr__N)rmrprqraror_r_r_r`r7s rN) __future__r collectionsrtypingrrrrr r r Zprompt_toolkit.cacher Zprompt_toolkit.data_structuresr Zprompt_toolkit.utilsrZ containersr__all__rrxrtrrrr_r_r_r`s(  $    |"