a Q\a'@sUddlmZddlmZmZmZmZmZmZm Z ddl m Z ddl m Z ddlmZerfddlmZdd gZGd d d Ze ed d Ze e eefefed <dZGdddZGdddZdS) defaultdict) TYPE_CHECKINGCallable DefaultDictDictListOptionalTuple) FastDictCache)Point) get_cwidth)WindowScreenCharcB@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@dAdBdCdDAZeeefedE<dSeedGdHdIdJZ de dKdLdMZ de dKdNdOZ e se Ze ZedPdQdRZdGS)Trz 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<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> )A     €‚ƒ„…†‡ˆ‰Š‹ŒŽ‘’“”•–—˜™š›œžŸ display_mappingsN)rrreturncCsH||jvr.|dkr|d7}n|d7}|j|}||_||_t||_dS)NrVz class:nbsp z class:control-character )rWrrr r)selfrrr[;lib/python3.9/site-packages/prompt_toolkit/layout/screen.py__init__is   z Char.__init__)otherrYcCs|j|jko|j|jkSNrrrZr^r[r[r\_equal}sz Char._equalcCs|j|jkp|j|jkSr_r`rar[r[r\ _not_equalszChar._not_equalrYcCsd|jj|j|jfS)Nz %s(%r, %r)) __class____name__rrrZr[r[r\__repr__sz Char.__repr__)rrX)rf __module__ __qualname____doc__ __slots__rWrstr__annotations__r]boolrbrcr__eq____ne__rhr[r[r[r\rs  Hi@B)size _CHAR_CACHEz [transparent]c@seZdZdZd#eeeeddddZee ddd d Z de dd d d Z de dd ddZ de dddZde dddZeegdfddddZddddZeddddZd$deedd d!d"ZdS)%rz= Two dimensional buffer of :class:`.Char` instances. Nr) default_char initial_widthinitial_heightrYcsp|durtdtfn|tfdd|_tdd|_i|_d|_i|_|pRd|_|p\d|_ i|_ g|_ dS)NrcstfddS)NcsSr_r[r[Z default_char2r[r\3Screen.__init__....rr[rwr[r\rxryz!Screen.__init__..cSs tddS)NcSsdS)NrXr[r[r[r[r\rxryrzrr[r[r[r\rxryTr) rs Transparentr data_bufferZzero_width_escapescursor_positionsZ show_cursormenu_positionsrheight"visible_windows_to_write_positions_draw_float_functions)rZrtrurvr[rwr\r]s$   zScreen.__init__rrdcCst|jSr_)listrkeysrgr[r[r\visible_windowsszScreen.visible_windows)windowpositionrYcCs||j|<dSz= Set the cursor position for a given window. N)r}rZrrr[r[r\set_cursor_positionszScreen.set_cursor_positioncCs||j|<dSr)r~rr[r[r\set_menu_positionszScreen.set_menu_position)rrYcCs0z |j|WSty*tdddYS0dS)zX Get the cursor position for a given window. Returns a `Point`. rxyN)r}KeyErrorr rZrr[r[r\get_cursor_positions  zScreen.get_cursor_positionc CsXz |j|WStyRz|j|WYStyLtdddYYS0Yn0dS)z Get the menu position for a given window. (This falls back to the cursor position if no menu position was set.) rrN)r~rr}r rr[r[r\get_menu_positions   zScreen.get_menu_position)z_index draw_funcrYcCs|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)rZrrr[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\rxryz(Screen.draw_all_floats..)keyrNr)rsorted)rZZ functionsr[r[r\draw_all_floatsszScreen.draw_all_floats) style_strrYc CsV|j}t}d|}|D]6\}}|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) rZrb char_cache append_stylerrowrrr[r[r\append_style_to_contents zScreen.append_style_to_contentrXF WritePosition)write_positionrafterrYcCs|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. NrrX) stripxposrrsr|rangeyposrrr)rZrrrZxminZxmaxrr|rZ prepend_stylerrrZcellr[r[r\ fill_areas( zScreen.fill_area)Nrr)rXF)rfrirjrkr rintr]propertyrrr rrrrrrrrmrrorr[r[r[r\rs0 1  c@s0eZdZeeeeddddZedddZdS)rN)rrrrrYcCs4|dks J|dksJ||_||_||_||_dSr)rrrr)rZrrrrr[r[r\r]6s   zWritePosition.__init__rdcCsd|jj|j|j|j|jfS)Nz#%s(x=%r, y=%r, width=%r, height=%r))rerfrrrrrgr[r[r\rh@szWritePosition.__repr__)rfrirjrr]rmrhr[r[r[r\r5s rN) collectionsrtypingrrrrrr r Zprompt_toolkit.cacher Zprompt_toolkit.data_structuresr Zprompt_toolkit.utilsr Z containersr__all__rrsrmrnr{rrr[r[r[r\s"$    {#