AHjUdZddlmZddlmZddlmZmZmZm Z m Z m Z m Z ddl ZddlmZmZmZmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd l m!Z!ddl"m#Z#m$Z$m%Z%m&Z&m'Z'ddl(m)Z)ddl*m+Z+m,Z,ddl-m.Z.er ddl/m0Z0ddl1m2Z2Gdde Z3dZ4de5d<GddZ6ejnjqdGdde6Z9ejnjqdGdd e6Z:ejnjqdGd!d"e6Z;ejnjqdGd#d$Zsr.Nr7r returnstr)r(r)r*r+r8r-r.r/r2r2;s [ r.r2T)angularc<eZdZdZddZ ddZd dZd dZy) LayoutUpdatezBA renderable containing the result of a render for a given region.c ||_||_yN)stripsregion)r6rArBs r/__init__zLayoutUpdate.__init__Ns  r.c#LK|jj}tj}tj }t t|j|jjD].\}\}}|||j|Ed{|r+|0y7wr@) rBxrlinermove_tor enumeraterAysegment) r6r7optionsrEnew_linerGlastrIrFs r/__rich_console__zLayoutUpdate.__rich_console__Rs| KKMM<<>//(4;; )NO OD)1d!Q-'' 'OO   sBB$B"B$B$cg}|j}|jj}tj}t t |j|jjD]M\}\}}||||jj||j||rF|dOdj|S)r4 r5) appendrBrErrGrrHrArIrJtextrenderjoin) r6r7 sequencesrQrErGrMrIrFs r/r8zLayoutUpdate.render_segments^s " !! KKMM//(4;; )NO OD)1d 71a=((-- . 4;;w' (t   wwy!!r.c#(K|jywr@)rBr6s r/ __rich_repr__zLayoutUpdate.__rich_repr__rskksN)rA list[Strip]rBrr:Noner7r rKr r:rr9r:zrich.repr.Resultr(r)r*r+rCrNr8rXr-r.r/r>r>Js1L  )7   "(r.r>c6eZdZdZdddZ ddZd dZy) InlineUpdatez'A renderable to write an inline update.c ||_||_yr@)rAclear)r6rAras r/rCzInlineUpdate.__init__zs  r.c#Ktj}t|jD]\}}|Ed{|r|y7wr@)rrFrrA)r6r7rKrLrMrFs r/rNzInlineUpdate.__rich_console__~sA<<>#DKK0 JD$OO  s5AAAAcg}|j}t|jD]'\}}||j||r |d)|jr(t |jdkDr|d|dt |jdkDrF|jrt |jnt |jdz }|d|dn|d|ddj |S) r4rPzzA  zr5)rQrrArSralenrT)r6r7rUrQrMstrip back_liness r/r8zInlineUpdate.render_segmentss " !!$T[[1 KD% 5<<( )t   ::4;;!#t 8  t{{ a -1ZZT[[)S=MPQ=QJ U:,c* + 4Lywwy!!r.NF)rArYraboolr:rZr[r9)r(r)r*r+rCrNr8r-r.r/r_r_vs,1)7 "r.r_cLeZdZdZ ddZ ddZd dZd dZy) ChopsUpdatez6A renderable that applies updated spans to the screen.c.||_||_||_y)aA renderable which updates chops (fragments of lines). Args: chops: A mapping of offsets to list of segments, per line. crop: Region to restrict update to. chop_ends: A list of the end offsets for each line N)chopsspans chop_ends)r6rnrorps r/rCzChopsUpdate.__init__s  "r.c#Ktj}tj}|j}|j }|j dd}t}|j D]\} } } || } || } t| | jD]\}\}}| || kDs|| kr| |cxkDr| k\r(nn%|| kr ||| j|Ed{Jt|}|| kr?|D]9}|||jz}|| kDr||| j|n|};n||| j|| kr |Ed{|D]"}|| k\r||||jz }$| |k7s| y77?w)Nr) rrGrrFrnrprorzipitemsrJiterrR)r6r7rKrGrLrnrplast_y _cell_lenrIx1x2rFendsendrErg iter_segmentsrJnext_xs r/rNzChopsUpdate.__rich_console__s//<<> NN B" % IAr28DQdZ!y)? CompositorzcResponsible for storing information regarding the relative positions of Widgets and rendering them.ci|_d|_d|_d|_t |_d|_d|_tdd|_ d|_ t |_ d|_ y)NTr) _full_map_full_map_invalidated _visible_map_layerssetwidgets_visible_widgetsrootrsize_cuts_dirty_regions_layers_visiblerWs r/rCzCompositor.__init__sp(*%)"26@D %(E MQ$( AJ .2 ,/5RVr.c|jjd|_d|_|jjd|_d|_y)z?Remove all references to widgets (used when the screen closes).N)rrarrrrrrWs r/razCompositor.clear9sA     $#r.c#Ki}|j}|D]:\}}}}|||zf}t|||zD]} || gj|<tdd} t |j D]d\} } t | dk(r | g| d | j| d\} } | | D]\}}|| kr || kDs|} | | | f|} |} | | | ffyw)aConverts the regions to horizontal spans. Spans will be combined if they overlap or are contiguous to produce optimal non-overlapping spans. Args: regions: An iterable of Regions. Returns: Yields tuples of (Y, X1, X2). rdNr) setdefaultrangerQslicesortedrtrfsort)clsregions inline_rangesrregion_xregion_ywidthheightspanrIslice_remainingrangesrxrynext_x1next_x2s r/_regions_to_spanszCompositor._regions_to_spansBs;= "-- 18 / -Hhvh./D8X%67 /1b!((. / /  4. 3 3 56 "IAv6{ao6!9o% B(.(?%$GW"}"Rz$Compositor.reflow..3'<!%%22<3F3FG24)r%r&r')rrrrrrrrkeys _arrange_rootrrtrBrupdater#)r6parentrold_map old_widgetsmapr new_widgets shown_widgetshidden_widgetschangescommon_widgets screen_regionrBrwidgetrresized_widgetss r/reflowzCompositor.reflowls  # $   ..lln ))&$U)K Whhj $k1 / ))+ /%{2  3 3 3+2 G    & &w /)0  $ !.(WV_-C-C-H-HFKK-W   !#   s " E8Ecd|_d|_d|_d|_d|_||_||_|jxsi}|j||d\}}||_||_ |j|jz }|j|jz }|j}||jvr3d|DD chc]} | r| } } |jj| |Scc} w)a Reflow only the visible children. This is a fast-path for scrolling. Args: parent: The root widget. size: Size of the area to be filled. Returns: Set of widgets that were exposed by the scroll. NTrc3lK|],\}}|jj|j.ywr@rrs r/rz,Compositor.reflow_visible..rr)rrrrrrrrrrrrtrBrr) r6rrrrrexposed_widgetsrrrBrs r/reflow_visiblezCompositor.reflow_visibles   # $%)"  ##)r))&$T)J W  ((*w||~5))+ /  3 3 3+2 G    & &w /s C8c|jiS|jrGd|_|j|j|jd\}}||_d|_d|_|jS)z4Lazily built compositor map that covers all widgets.NFr)rrrrrrr)r6r_widgetss r/full_mapzCompositor.full_mapsi 99 I  % %).D & ..tyy$))RW.XMC DN$(D ! $D ~~r.c|j|j |jn|jxsi}|jj}|j }t j }|jD cgc]$\}\}}}} } } } ||r|||r||||f&} }}}}} | jtdd| D cic] \} }}}|||fc}}}} |_|jScc} }}}}wcc}}}} w)znGet a mapping of widgets on to region and clip. Returns: Visible widget mapping. rTkeyreverse) rrrrrBoverlapsrrtrr) r6rscreen in_screenrrrBorderrrvisible_widgetss r/rzCompositor.visible_widgetss  ($$0!!nn*  YY%%FIH BE=F=VUD!Q1V$$)?-O  Z]D AFU%%+B1ffd&%D !$$$ %s =)C7 D c J i t jt}|jd}j tf d  fd |jjd|jdt |z fS)a3Arrange a widget's children based on its layout attribute. Args: root: Top level widget. size: Size of visible area (screen). visible_only: Only update visible widgets (used in scrolling). Returns: Compositor map and set of widgets. rc |jsy|j} | jdx} | dk(}|r )|n(||j| j} | j } |j r*|jr| n|j| } | j}|jr|j| j }|j}/j||j| }|jr|j s|j"j$j&|j(j*|j,z z }|j/t0j2||j/t0j4|||j6_.r,|j;|| j<z |j>z}n |j@}|jC|j$}| j<}||j>z }|Dcgc]}|jE-jF| }}tI|jJDcic]\}}|| }}}|jL}|jNH||j|jj|||j | ||+|jN<tQ|D]l\}}}}}} }!}"||jRd}#| r ||z|z}$n||z|z}$||#||ffz}%|jN!*|||$|!rdn|%||!r,n|||jT|dz}n|r^|jVs |jXr*|j[| D]\}&}'||'||| | |'|+|&<|||||j | ||+|<yy|r&|||||j | ||+|j\<yycc}wcc}}w)anCalled recursively to place a widget and its children in the map. Args: widget: The widget to add. virtual_region: The Widget region relative to its container. region: The region the widget will occupy. order: Painting order information. layer_order: The order of the widget in its layer. clip: The clipping region (i.e. the viewport which contains it). visible: Whether the widget should be visible by default. This may be overridden by the CSS rule `visibility`. N visibilityvisibler))rdrrrd)/ _is_mountedstylesget_ruleshrinkgutterr is_scrollableloading_get_scrollable_region reset_offset is_containerarrangerrr _anchored_anchor_released spatial_map total_regionbottomcontainer_sizerscrollbar_size_horizontal set_reactiverscroll_yscroll_target_yvertical_scrollbar_reactive_positionget_visible_placementsoffset scroll_offset placementsunionprocess_offsetrBrHlayersget _cover_widgetreversedlayerscroll_spacingshow_vertical_scrollbarshow_horizontal_scrollbar_arrange_scrollbars_render_widget)0rvirtual_regionrBr layer_orderrr dock_gutter _MapGeometryrrcontainer_regionr child_regionrarrange_resultarranged_widgetssub_clip new_scroll_yrplacement_offsetplacement_scroll_offset placementindex layer_namelayers_to_indexget_layer_index sub_regionsub_region_offsetr sub_widgetzfixedoverlayabsolute layer_index widget_region widget_order chrome_widget chrome_regionadd_new_invisible_widgetadd_new_widget add_widgetrno_cliprrrs0 r/r!z,Compositor._arrange_root..add_widget#sA.%%]]F$ool;; H$ 1v&(0 &}}V]]; -22N##~~%667GH ,88 &&%+^^L4E4E%FN'5'='=$NN#34 $00>H''0G0G*66CCJJ & 5 5 < <"("B"B!C%++FOO\J++F,B,BLQGS11D#%3%J%J$|':'::V=Q=QQ& &4%>%> #/#5#5n6Q6Q#RL(8'>'>$.>AUAU.U+*4"%"00>UV"J"2;6==1I'-E:#E)'O' '6&9&9O++74@"MM&--*>*>?! "KK**'5F001("*-") ")" &5j6F6F&J  *-> >AQ Q* !+-> >AX X*(-a0M/O'O !//7& * * -07 \ ++2 ' . = = $q( E")H66!;; - % $ . . - +2C . #/$))&&##CK':-9KK"".F))*s" 's #O  O))rrrT)rrrrrBrrz tuple[tuple[int, int, int], ...]rintrrrrjrrrztype[MapGeometry]r:rZ)raddrBrr) r6rrrinvisible_widgetsrrr r!rr"rs `` @@@@@@r/rzCompositor._arrange_root s "u ),#4#8#8  ++/:x x "x x 4 x   x   x x !x ,x x x v   KK KK   KK    $$G|r.c|j |jn |j}|j"t|j dd|_|jS)z(Get widgets and geometry in layer order.c |djS)Nrd)r)items r/z#Compositor.layers..sd1gmmr.Tr)rrrrrt)r6rs r/rzCompositor.layerssP$(#4#4#@ddnn << ! !;TDL||r.c|jt|jjDcgc]}g}}|Dcgc]}|j}}t j }t}|jjD]<\}\}} ||| } | \} } } }|s|| |f}|| | |zD] }|||>||_|jScc}wcc}w)zVisible widgets and regions in layers order. Returns: Lists visible widgets per layer. Widgets are give as a tuple of (WIDGET, CROPPED_REGION, REGION). CROPPED_REGION is clipped by the container. ) rrrrrQrrrrt)r6rIlayers_visiblerlayers_visible_appendsr_rangerrBrcropped_region_xr_width region_heightwidget_locations r/r+zCompositor.layers_visibles    '*/ 0@0@*ABQbBNB@N%Ouell%O "%O!..LF*.*>*>*D*D*F C&!-fd!;6D3Hfm '-~v&FO#Hh.FGC1.q1/BC  C$2D ###C%Os CCc~||jvxs.|jduxr||jvxs||jvS)zCheck if the widget was included in the last update. Args: widget: A widget. Returns: `True` if the widget was in the last refresh, or `False` if it wasn't. N)rrrr6rs r/ __contains__zCompositor.__contains__sI dll " '!!-M&D|jt |D]#\}}}||||s|j s||fcSt jd|d|d)aGet the widget under a given coordinate. Args: x: X Coordinate. y: Y Coordinate. Raises: errors.NoWidget: If there is not widget underneath (x, y). Returns: A tuple of the widget and its region. rz#No widget under screen coordinate (z, ))rcontainsrfr+r#rrr9r6rErIr=rr.rBs r/ get_widget_atzCompositor.get_widget_at3s?? t"" #a ,1 ,262E2Ec!f2M *.NAq1fnn!6>) *oo CA3b1MNNr.c#Ktj}t|j|cxkDrdk\r:ny|j|D]$\}}}||||s|js||f&yyw)zGet all widgets under a given coordinate. Args: x: X coordinate. y: Y coordinate. Returns: Sequence of (WIDGET, REGION) tuples. rN)rr=rfr+rr>s r/get_widgets_atzCompositor.get_widgets_atHsk?? t"" #a ,1 ,262E2Ea2H ).NAq1fnn &.( ) -sAA, A," A,c |j||\}}||j vrtjS||j z}||jz}tj|jj|jtd||jd}|stjSd}|dD]:}||jz }||ks|j xstjcStjS#tj$rtjcYSwxYw)zGet the Style at the given cell or Style.null() Args: x: X position within the Layout. y: Y position within the Layout. Returns: The Style at the cell (x, y) within the Layout. rrd)r?rr9rnullrrErIrrapp_background_screens render_linesrr cell_lengthstyle)r6rErIrrBlinesr{rJs r/ get_style_atzCompositor.get_style_atXs  !//15NFF -- -::<  VXX  VXX   !?!?@##F1aq$AB::< Qx 5G 7&& &C3w}}4 4 5 zz|) ::<  sD'D?>D?c |j||\}}||jvry||jj k\r|jj \}}|jj\}}||j|zz}||j|zz}|dkrytj|jj|j|}d}d} d} d} d} |D]} |t!| j"z }| j$}||j&|j(}d|vr|j(d\} } | t!| j"z} ||kri|| k\rd||dz k(rt!| j"}n,| j+|| z \}}t!|j"}|| dfcSt-| |z| fcS|} || dfSt-| | fS#tj$rYywxYw)aGet the Style at the given cell, the offset within the content. Args: x: X position within the Layout. y: Y position within the Layout. Returns: A tuple of the widget at (x, y) and the offset within the widget. )NNrNrrd)r?rr9rcontent_regionrbottom_right_inclusivertop_leftrErIrrrDrE render_linerfrRrH_metameta split_cellsr)r6rErIrrB gutter_left gutter_rightrFr{startoffset_yoffset_x offset_x2rJrHrQsegment_offsetfirstrs r/get_widget_and_offset_atz#Compositor.get_widget_and_offset_atzs !//15NFF -- - %%,, ,((??DAq$*MM$:$:! \ VXX ## VXX $$ q5  !?!?@!!!$#  G 3w||$ $CMME U[[%<zzt#).H)=&Hh (3w||+< N'.':':1u9'EHE1-0_N%'/! "(>(A8!L  E) , 0RRfY6QRR]  sGG+*G+cX|jtjd |js |j|S|j  |j |S|j|}|S#t $rY8wxYw#t $rY+wxYw#t $rtjdwxYw)aJGet information regarding the relative position of a widget in the Compositor. Args: widget: The Widget in this layout you wish to know the Region of. Raises: NoWidget: If the Widget is not contained in this Layout. Returns: Widget's composition information. r7)rrr9rrr8rr)r6rrBs r/ find_widgetzCompositor.find_widgets 99 //";< < -->>&11  ,,,V44]]6*FM    =//";< < =sQ B A, B A;B , A85B 7A88B ; BB BB B)c|j |jS|j\}}t|Dcgc]}d|g}}tj}t j }|jjD]6\}}|||\} } } } | s| s| | | zf} || | | zD] }|||  8|Dcgc]}tt|c}|_|jScc}wcc}w)zGet vertical cuts. A cut is every point on a line where a widget starts or ends. Returns: A list of cuts for every line. r) rrrrrlistextendrvaluesrr)r6rrrcutsrr`rBrrErI region_widthr1 region_cutscut line_cutss r/rbzCompositor.cutss :: !::  v$)&M2qE 22**  00779 -LFD0, , and TrN) rrrrtrcontains_regionrFrrrErI)r6r_rich_traceback_guard_Regionr crop_overlapsrrBrwidget_regionsrrhnew_xnew_y new_width new_heights r/ _get_renderszCompositor._get_renderss!%..  MMM/>.C.C.E*FNVT &&N/>.C.C.E*FNVT&N ++ !11$2  FFDtV,''"LL"MM   7C646P3ui++# % 0 % 0 ) *   #  s*7D>D0D>,D7>A2D>1D>4A D>ctj|gn||jj}|s||jvr|j |S|j S)aKRender an update renderable. Args: full: Perform a full update if `True`, otherwise a partial update. screen_stack: Screen stack list. Defaults to None. simplify: Simplify segments. Returns: A renderable for the update, or `None` if no update was required. )simplify)rrrrBrrender_full_updaterender_partial_update)r6full screen_stackrsrs r/ render_updatezCompositor.render_update6s["   |';N (( =D$7$77**H*= =--/ /r.cptj|gn||j|}t||S)a Render an inline update. Args: size: Inline size. screen_stack: Screen stack list. Defaults to None. clear: Also clear below the inline update (set when size decreases). Returns: A renderable. )ra)rr render_stripsr_)r6rrwrarAs r/ render_inlinezCompositor.render_inlineNs6   |';N##D)F%00r.c|jj}|jj|}|j |d}|rM|Dcgc]A}t j |jjjC}}n0|Dcgc]%}t j |j'}}t||Scc}wcc}w)zRender a full update. Args: simplify: Simplify the segments (combine contiguous segments). Returns: A LayoutUpdate renderable. cyNTr-rIs r/r)z/Compositor.render_full_update..nr.) rrBrra _render_chopsrrTrars discard_metar>)r6rsrrrnchoprzs r/rtzCompositor.render_full_updatebs ((  !!#""48 QVHL 4;;=)224AACMDII4UZZ 6IMIM=99 Js AC*Cc|jj}|jj}|jj |r_t j |j|}t|j|}|Dchc]\}}}| c}}j}ny|j||}|jD cgc]} | dd } } t||| Scc}}wcc} w)zRender a partial update. Returns: A ChopsUpdate if there is anything to update, otherwise `None`. Nrd)rrBrcopyrar from_unionrr_rr5rrbrl) r6rupdate_regionsrrorIris_rendered_linerncut_setrps r/ruz Compositor.render_partial_updatezs  (( ,,113 !!# $$^4AA-PD//?@E167gaA7DD ""4)9:04 :WWQR[: :5%33 8;s  C( C.c| |j}|j|jd}|d|jDcgc]%}t j |j '}}|Scc}w)zRender to a list of strips. Args: size: Size of render. Returns: A list of strips with the screen content. Ncyr~r-rs r/r)z*Compositor.render_strips..rr.)rrrBrrrTra)r6rrnrrzs r/rzzCompositor.render_stripssa <99D""4;;??D]t{{?STtDKKM2T TUs*A+c |j}tdtj}|Dcgc] }||dd}}|j |}t j }|D]\} } } || | } | j} | j\}}t| j| D]\}}||s||}||Dcgc]}||cxk\r|k\snn|}}|j|ddDcgc]}|| z  c}}|j}t||D]\}}|||||<|Scc}wcc}wcc}w)zRender update 'chops'. Args: crop: Region to crop to. is_rendered_line: Callable to check if line should be rendered. Returns: Chops structure. z.Callable[[list[int]], dict[int, Strip | None]]Nrrrd) rbr dictfromkeysrqrrrE column_spanrs line_rangedivider)r6rrrbrrrnrendersrrBrrA render_regionrender_x first_cutlast_cutrIrg chops_linere final_cuts cut_stripsget_chops_lines r/rzCompositor._render_chopss^yyH$--X7;*TU ","%j*"=0JC%c*2*/ 30 0  0& 7=$X*TsD1>D6D6* D; c4t|jSr@)rrzrWs r/__rich__zCompositor.__rich__st11344r.c|js0|j|jjsd|_g}|j}|jj }|jj|zD]\}||\}}|j }|j} |jD]$} | | j|} | s|| &^|jj|y)zsUpdate the given widgets in the composition. Args: widgets: Set of Widgets to update. TN) rissubsetrrrQ __getitem__rr_exchange_repaint_regions translaterr) r6rr add_region get_widgetrrBrrr dirty_region update_regions r/update_widgetszCompositor.update_widgetss))'2B2B  % % '3 *.D & "^^ ))55 **//1G; .F%f-LFD]]F,,L & @ @ B . ,\-C-CF-K L  }- .  . ""7+r.)r:rZ)rzIterable[Region]r:zIterable[tuple[int, int, int]]r\)rrrrr:r#)rrrrr:r$)r:r0)r:z#dict[Widget, tuple[Region, Region]])T)rrrrrrjr:z!tuple[CompositorMap, set[Widget]])r:z list[tuple[Widget, MapGeometry]])r:z)list[list[tuple[Widget, Region, Region]]])rrr:rj)rrr:r)rEr#rIr#r:ztuple[Widget, Region])rEr#rIr#r:zIterable[tuple[Widget, Region]])rEr#rIr#r:r)rEr#rIr#r:z#tuple[Widget | None, Offset | None])rrr:r)r:rr@)rz Region | Noner:z,Iterable[tuple[Region, Region, list[Strip]]])FNF)rvrjrwlist[Screen] | Nonersrjr:zRenderableType | None)NF)rrrwrrarjr:rri)rsrjr:r>)r:zChopsUpdate | None)rz Size | Noner:rY)rrrzCallable[[int], bool]r:r)r:r)rr$r:rZ)"r(r)r*r+rCra classmethodrrXrrpropertyrrrrr+r5r:r?rArJr[r]rbrqrxr{rtrurzrrrr-r.r/rrsmV:$#"&#" '#"#"J&@ D.`  %%<>B]]"&]6:] *]~$$4 =&O*) D<S<S<S ,<S|><%)?!? 5?F,0 00*0 0  06-1 11*1 1  1(:04( ,,0, . ,\5,r.r)=r+ __future__roperatorrtypingrrrrr r r rich.reprrich rich.consoler r rr rich.controlr rich.segmentr rich.stylertextualrtextual._cellsrtextual._contextr textual._looprtextual.geometryrrrrrtextual.map_geometryr textual.striprrtextual.widgetrtyping_extensionsr textual.screenr!r#r0r,r2reprautor>r_rlrr-r.r/rs- #NN #1#HH,0!+%:7 y6  (#((V)"#)")"Xq"qqhN,N,N,r.