U ou f'@sUddlmZddlmZddlmZmZddlmZddl m Z ddl m Z er^ddl mZd d gZGd d d Zeed d Zded<dZGdd d ZGdddZdS)) annotations defaultdict) TYPE_CHECKINGCallable) 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)Yr z 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)NrSz class:nbsp z class:control-character )rTrrr r)selfrrrZ@/tmp/pip-unpacked-wheel-8milen2s/prompt_toolkit/layout/screen.py__init__ls   z Char.__init__bool)otherrXcCs|j|jko|j|jkSNrrrYr^rZrZr[_equalsz Char._equalcCs|j|jkp|j|jkSr_r`rarZrZr[ _not_equalszChar._not_equalrXcCs|jjd|jd|jdS)N(z, )) __class____name__rrrYrZrZr[__repr__sz Char.__repr__N)rrU)rh __module__ __qualname____doc__ __slots__rT__annotations__r\rbrcr__eq____ne__rjrZrZrZr[r s  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)-r z= Two dimensional buffer of :class:`.Char` instances. Nrz Char | NoneintrW) default_char initial_widthinitial_heightrXcsp|dkrtdtfn|tfdd|_tdd|_i|_d|_i|_|pRd|_|p\d|_ i|_ g|_ dS)NrcstfddS)NcsSr_rZrZZ default_char2rZr[3Screen.__init__....rrZrxrZr[ryrzz!Screen.__init__..cSs tddS)NcSsdS)NrUrZrZrZrZr[ryrzr{rrZrZrZr[ryrzTr) rs Transparentr data_bufferZzero_width_escapescursor_positionsZ show_cursormenu_positionsrheight"visible_windows_to_write_positions_draw_float_functions)rYrurvrwrZrxr[r\s$   zScreen.__init__z list[Window]rdcCst|jSr_)listrkeysrirZrZr[visible_windowsszScreen.visible_windowsr r)windowpositionrXcCs||j|<dSz= Set the cursor position for a given window. N)r~rYrrrZrZr[set_cursor_positionszScreen.set_cursor_positioncCs||j|<dSr)rrrZrZr[set_menu_positionszScreen.set_menu_position)rrXcCs2z |j|WStk r,tdddYSXdS)zX Get the cursor position for a given window. Returns a `Point`. rxyN)r~KeyErrorrrYrrZrZr[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)rrr~rrrZrZr[get_menu_positions zScreen.get_menu_positionzCallable[[], None])z_index draw_funcrXcCs|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)rYrrrZrZr[draw_with_z_indexszScreen.draw_with_z_indexcCs:|jr6t|jddd}|dd|_|ddqdS)z? Draw all float functions in order of z-index. cSs|dSNrrZ)itemrZrZr[ryrzz(Screen.draw_all_floats..)keyr Nr)rsorted)rYZ functionsrZrZr[draw_all_floatsszScreen.draw_all_floatsrV) style_strrXc 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)r}rsitemsrr) rYrb char_cache append_stylerrowrrrZrZr[append_style_to_contents zScreen.append_style_to_contentrUF WritePositionr])write_positionrafterrXcCs|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. NrrU) stripxposrrsr}rangeyposrrr)rYrrrZxminZxmaxrr}rZ prepend_stylerrrcellrZrZr[ fill_areas*  zScreen.fill_area)Nrr)rUF)rhrkrlrmr\propertyrrrrrrrrrrZrZrZr[r s"0  c@s0eZdZddddddddZdddd Zd S) rrtrW)rrrrrXcCs4|dks t|dkst||_||_||_||_dSr)AssertionErrorrrrr)rYrrrrrZrZr[r\8s   zWritePosition.__init__rVrdcCsd|jj|j|j|j|jS)Nz+{}(x={!r}, y={!r}, width={!r}, height={!r}))formatrgrhrrrrrirZrZr[rjBszWritePosition.__repr__N)rhrkrlr\rjrZrZrZr[r7s rN) __future__r collectionsrtypingrrZprompt_toolkit.cacherZprompt_toolkit.data_structuresrZprompt_toolkit.utilsr Z containersr __all__r rsror|r rrZrZrZr[s(      |"