o
    ˷e(                     @   s   U d dl mZ d dlmZmZmZmZmZmZm	Z	 d dl
mZ d dlmZ d dlmZ er3ddlmZ dd	gZG d
d	 d	ZeeddZee	eef ef ed< dZG dd dZG dd dZdS )    defaultdict)TYPE_CHECKINGCallableDefaultDictDictListOptionalTuple)FastDictCache)Point)
get_cwidth   )WindowScreenCharc                   @   s  e Zd ZU dZdZi ddddddd	d
dddddddddddddddddddddd d!d"d#d$i d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFi dGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgdhdidjdkdldmdndodpdqdrdsdtdudvdwdxZeeef edy< dd{ed|ed}d~fddZ	dd d}e
fddZdd d}e
fddZeseZeZd}efddZd~S )r   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.)
    )charstylewidth z^@z^Az^Bz^Cz^Dz^Ez^Fz^Gz^H	z^I
z^Jz^Kz^Lz^Mz^Nz^Oz^Pz^Qz^Rz^Sz^Tz^Uz^Vz^Wz^Xz^Yz^Zz^[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> )                                              display_mappings r   r   returnNc                 C   sH   || j v r|dkr|d7 }n|d7 }| j | }|| _|| _t|| _d S )NrW   z class:nbsp z class:control-character )rX   r   r   r   r   )selfr   r    r\   S/var/www/ideatree/venv/lib/python3.10/site-packages/prompt_toolkit/layout/screen.py__init__j   s   


zChar.__init__otherc                 C   s   | j |j ko| j|jkS Nr   r   r[   r_   r\   r\   r]   _equal~   s   zChar._equalc                 C   s   | j |j kp| j|jkS r`   ra   rb   r\   r\   r]   
_not_equal   s   zChar._not_equalc                 C   s   | j j d| jd| jdS )N(z, ))	__class____name__r   r   r[   r\   r\   r]   __repr__   s   zChar.__repr__)rH   rY   )rh   
__module____qualname____doc__	__slots__rX   r   str__annotations__r^   boolrc   rd   r   __eq____ne__rj   r\   r\   r\   r]   r      s  
 		
 !#$%&'()*+,-./012345Ii@B )size_CHAR_CACHEz[transparent]c                	   @   s   e Zd ZdZ			d(dee dededdfdd	Zede	d
 fddZ
dd
deddfddZdd
deddfddZdd
defddZdd
defddZdedeg df ddfddZd)ddZdeddfddZ	!d*d"d#d$ed%eddfd&d'ZdS )+r   z=
    Two dimensional buffer of :class:`.Char` instances.
    Nr   default_charinitial_widthinitial_heightrZ   c                    sp   |d u rt dtf  n| t fdd| _tdd | _i | _d| _i | _|p)d| _|p.d| _	i | _
g | _d S )NrH   c                      s   t  fddS )Nc                      s    S r`   r\   r\   default_char2r\   r]   <lambda>       3Screen.__init__.<locals>.<lambda>.<locals>.<lambda>r   r\   ry   r\   r]   r{      s    z!Screen.__init__.<locals>.<lambda>c                   S   s   t dd S )Nc                   S   s   dS )NrY   r\   r\   r\   r\   r]   r{      r|   r}   r   r\   r\   r\   r]   r{      s    Tr   )ru   Transparentr   data_bufferzero_width_escapescursor_positionsshow_cursormenu_positionsr   height"visible_windows_to_write_positions_draw_float_functions)r[   rv   rw   rx   r\   ry   r]   r^      s$   



zScreen.__init__r   c                 C   s   t | j S r`   )listr   keysri   r\   r\   r]   visible_windows   s   zScreen.visible_windowswindowpositionc                 C      || j |< dS z=
        Set the cursor position for a given window.
        N)r   r[   r   r   r\   r\   r]   set_cursor_position      zScreen.set_cursor_positionc                 C   r   r   )r   r   r\   r\   r]   set_menu_position   r   zScreen.set_menu_positionc                 C   s,   z| j | W S  ty   tddd Y S w )zX
        Get the cursor position for a given window.
        Returns a `Point`.
        r   xy)r   KeyErrorr   r[   r   r\   r\   r]   get_cursor_position   s
   zScreen.get_cursor_positionc                 C   sP   z| j | W S  ty'   z| j| W  Y S  ty&   tddd Y  Y S w w )z
        Get the menu position for a given window.
        (This falls back to the cursor position if no menu position was set.)
        r   r   )r   r   r   r   r   r\   r\   r]   get_menu_position   s   zScreen.get_menu_positionz_index	draw_funcc                 C   s   | j ||f dS )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)r   append)r[   r   r   r\   r\   r]   draw_with_z_index   s   zScreen.draw_with_z_indexc                 C   sB   | j rt| j dd d}|dd | _ |d d   | j sdS dS )z?
        Draw all float functions in order of z-index.
        c                 S   s   | d S Nr   r\   )itemr\   r\   r]   r{      s    z(Screen.draw_all_floats.<locals>.<lambda>)keyr   Nr   )r   sorted)r[   	functionsr\   r\   r]   draw_all_floats   s
   zScreen.draw_all_floats	style_strc           	      C   sV   | j }t}d| }| D ]\}}| D ]\}}||j|j| f || |< qqdS )zn
        For all the characters in the screen.
        Set the style string to the given `style_str`.
        rH   N)r   ru   itemsr   r   )	r[   r   b
char_cacheappend_styler   rowr   r   r\   r\   r]   append_style_to_content  s   zScreen.append_style_to_contentrY   Fwrite_positionWritePositionr   afterc                 C   s   |  sdS |j}|j|j }t}| j}|rd| }d}	nd}|d }	t|j|j|j D ] }
||
 }t||D ]}|| }||j|	|j	 | f ||< q8q-dS )z
        Fill the content of this area, using the given `style`.
        The style is prepended before whatever was here before.
        NrH   rY   )
stripxposr   ru   r   rangeyposr   r   r   )r[   r   r   r   xminxmaxr   r   r   prepend_styler   r   r   cellr\   r\   r]   	fill_area  s,   
zScreen.fill_area)Nr   r   )rZ   N)rY   F)rh   rk   rl   rm   r	   r   intr^   propertyr   r   r   r   r   r   r   r   r   r   ro   r   rq   r   r\   r\   r\   r]   r      sB    
1

c                
   @   s8   e Zd Zdededededdf
ddZdefd	d
ZdS )r   r   r   r   r   rZ   Nc                 C   s4   |dksJ |dksJ || _ || _|| _|| _d S r   )r   r   r   r   )r[   r   r   r   r   r\   r\   r]   r^   7  s   
zWritePosition.__init__c                 C   s   d | jj| j| j| j| jS )Nz+{}(x={!r}, y={!r}, width={!r}, height={!r}))formatrg   rh   r   r   r   r   ri   r\   r\   r]   rj   A  s   zWritePosition.__repr__)rh   rk   rl   r   r^   ro   rj   r\   r\   r\   r]   r   6  s    
r   N)collectionsr   typingr   r   r   r   r   r	   r
   prompt_toolkit.cacher   prompt_toolkit.data_structuresr   prompt_toolkit.utilsr   
containersr   __all__r   ru   ro   rp   r~   r   r   r\   r\   r\   r]   <module>   s$    $| #