G@d!rdZddlmZddlZddlmZddlmZmZm Z m Z m Z m Z ddl mZmZmZddlmZmZmZmZmZmZmZmZmZddlmZdd lmZmZdd l m!Z!dd l"m#Z#m$Z$m%Z%dd l&m'Z'dd l(m)Z)Gdde!Z*dS)z This module contains the marker panel. Adapted from pyqode/core/panels/folding.py of the `PyQode project `_. Original file: )ceilN) IntervalTree)SignalQSizeQPointFQRectFQRectQt) QApplicationQStyleOptionViewItemQStyle) QTextBlockQColor QFontMetricsFQPainterQLinearGradientQPenQPalette QResizeEventQCursor) FoldingRegion)TextDecoration DRAW_ORDERS)Panel) TextHelperDelayJobRunner drift_color)ima) QStylePaletteceZdZdZeeeZeZeZ e dZ e j dZ e dZ e j dZ e dZej dZdZd Zd Z d'd Zfd ZdZdZdZdZdZdZdZdZfdZdZfdZdZdZ dZ!dZ"dZ#dZ$dZ%d Z&d!Z'd"Z(d#Z)d$Z*d%Z+d&Z,xZ-S)( FoldingPanela Displays the document outline and lets the user collapse/expand blocks. The data represented by the panel come from the Language Server Protocol invoked via the CodeEditor. This panel stores information about both folding regions and their folding state. c|jS)z Defines whether the panel will use native indicator icons or use custom ones. If you want to use custom indicator icons, you must first set this flag to False. ) native_iconsselfs Hlib/python3.11/site-packages/spyder/plugins/editor/panels/codefolding.pyr#zFoldingPanel.native_icons8s   c||_|jrD|jjD]9} ||j|j_(#t$rY4wxYwdSdSN) _native_iconseditorclonesmodesget __class__r#KeyErrorr%valueclones r&r#zFoldingPanel.native_iconsCs|" ; +  CHEKOODN33@@D     s$A AAc|jS)a6 Gets/sets the icons for the fold indicators. The list of indicators is interpreted as follow:: (COLLAPSED_OFF, COLLAPSED_ON, EXPANDED_OFF, EXPANDED_ON) To use this property you must first set `native_icons` to False. :returns: tuple(str, str, str, str) )_indicators_iconsr$s r&indicators_iconszFoldingPanel.indicators_iconsOs %%r'ct|dkrtd||_|jrD|jjD]9} ||j|j_(#t$rY4wxYwdSdS)Nz5The list of custom indicators must contains 4 strings) len ValueErrorr5r+r,r-r.r/r6r0r1s r&r6zFoldingPanel.indicators_icons^s u::??'(( (!& ; +  ;@KOO(((8(8D     s$A%% A21A2c|jS)z True to highlight the caret scope automatically. (Similar to the ``Highlight blocks in Qt Creator``. Default is False. )_highlight_caretr$s r&highlight_caret_scopez"FoldingPanel.highlight_caret_scopens $$r'cz||jkr||_|jr|r,d|_|jj|jn+d|_|jj|j|jjD];} ||j |j _ (#t$rY4wxYwdSdSdS)N) r<r+ _block_nbrcursorPositionChangedconnect_highlight_caret_scope disconnectr,r-r.r/r=r0r1s r&r=z"FoldingPanel.highlight_caret_scopeys D) ) )$)D !{ 5&(DOK5==35555')DOK5@@3555![/EDI  N,,,A,A# * )  s$B'' B43B4c"tj|d|_d|_d|_d|_d|_d|_g|_i|_ | dd|_ d|_ d|_ d|_d|_d|_d|_d|_d|_d|_d|_d|_t/d|_t3|_t7dd|_i|_i|_i|_i|_ dS)NF)zfolding.arrow_right_offzfolding.arrow_right_onzfolding.arrow_down_offzfolding.arrow_down_onr?T)delay)!r__init__r*r5r@r<r= _indic_size _scope_decos _block_decossetMouseTracking scrollable_mouse_over_line_current_scope _prev_cursor context_menuaction_collapse action_expandaction_collapse_allaction_expand_all_original_background_display_folding _key_pressedr_highlight_runnerr current_treerrootfolding_regionsfolding_statusfolding_levelsfolding_nestingr$s r&rIzFoldingPanel.__init__s  t""   %%*" d### $"  #!#' !%$(! %!!/c!:!:!:(NN!$-- !  !r'c|dS|\|_|_|_|_|_|_|dS)z$Update folding panel folding ranges.N)r[r\r]r`r_r^update)r% folding_infos r&update_foldingzFoldingPanel.update_foldingsH   F6B 3 DI t3 d1 r'cHt|j}tt |t |}|dkr|d|S)z=Returns the widget size hint (based on the editor font size) rF)rr+fontrrheightwidthsetWidth)r%fm size_hints r&sizeHintzFoldingPanel.sizeHints{ 4;++-- . .$ryy{{++T"))++->->?? ??  r ! !   r " " "r'FcR||jvr|j|}|j|}|j|k}|s||||||rW|r||||||jD]} |j| } | j|krdS|||dS|st|jD][} |j| } | j|kr=|j | |j j | ~ dSXdSdSdSr)) r]r^rO_draw_fold_indicatorrLblock_add_fold_decorationlistkeyspopr+ decorationsremove) r% line_number top_positionropainter mouse_hover collapsedline_end mouse_over deco_linedecos r&_draw_collapsed_indicatorz&FoldingPanel._draw_collapsed_indicators $. . .+K8I+K8H.+=J B)) *iBBB F--$j)WFFF "&!2??I,Y7DzU**+ --eX>>>>> !%d&7&<&<&>&>!?!?  I,Y7DzU**)--i888 /66t<<<  +9 / .,    r'chtt||t|}|js^|jsWt |jr/|j j D]"}|\}}}| ||||d#dS|j `|j |j } |||n#t t"f$rYnwxYw|j j D] \}}}| ||||d!dS)NT)ryF)superr! paintEventrrXrYanyr^valuesr+visible_blocksrrOdocumentfindBlockByNumber_draw_fold_region_backgroundr:r0)r%eventrxinforwrvror/s r&rzFoldingPanel.paintEvents lD!!,,U3334..$ T-> 4&--//00 3 K633D7;4L+u22#\5T33333 F  ,K((**<<%''E 11%AAAA)      15 0J N N ,L+u  * *\5'u + N N N N N NsC..DDct|j}|}|j|}|dkr||}nd}||}||z }|dkr&|}|} |td|| ||dS)z Draw the fold region when the mouse is over and non collapsed indicator. :param top: Top position :param block: Current block. :param painter: QPainter rN) rr+ blockNumberr]line_pos_from_numberrlrgrh _draw_rectr) r%rorxthstartendtopbottomhws r&rz)FoldingPanel._draw_fold_region_background s $ $!!##"5) 199))%00CCC((-- SL 66 &&((A MMOO ! ! # # q#q!,,g66666r'c|jj}t||}t jdkrh|d|d|d|d| d}ng|d|d|d|d| d}| ||| t|| |tddz|tddz | |tddz|tddz | |tddz|tddz | |tddz|tddz dS)z Draw the background rectangle using the current style primitive color. :param rect: The fold zone rect to draw :param painter: The widget's painter. darwinrdnN)r+sideareas_colorrtopLefttopRightsysplatform setColorAtlighterdarkerfillRectsetPenrdrawLiner bottomLeft bottomRight)r%rectrxcgradoutlines r&rzFoldingPanel._draw_rect$sP K 't||~~t}}?? <8 # # OOAqyy~~ . . . OOAqyy~~ . . .hhsmmGG OOAqyy~~ . . . OOAqyy~~ . . .hhsmmGt$$$tG}}%%% A' A' ( ( ( ** A'))++ A' ( ( (  A'))++ A' ( ( (  A'** A' ( ( ( ( (r'ctd|||}|jr,t }||_tjtj ztj z|_ |s|xj tj zc_ |r}|xj tj tjztjzzc_ |jt$j||jdd|tj|||dSd}|sd}|r|dz }t3j|j|||dS)a Draw the fold indicator/trigger (arrow). :param top: Top position :param mouse_over: Whether the mouse is over the indicator :param collapsed: Whether the trigger is collapsed or not. :param painter: QPainter rrN)r rlrhrgr*r rr State_Active State_ItemState_Childrenstate State_OpenState_MouseOver State_EnabledState_SelectedpalettesetBrushrWindow highlight translatestyle drawPrimitivePE_IndicatorBranchriconr5paint)r%rr|rzrxroptindexs r&rnz!FoldingPanel._draw_fold_indicatorIsQT]]__2244]]__++--//   I&((CCH,*+./CI / V..  A f4$23$345  $$X_%)\\^^%=%=%?%?AAA H  r1 % % % JJLL & &v'@'*GT ; ; ; ; ;E    HT+E2 3 3 9 9'4 H H H H Hr'c|}||jvrQ|jD]I}|j|}||kr4||kr.|j|}nJ|S)z6Find parent scope, if the block is not a fold trigger.)rr]r+rr)r%ro block_line start_lineend_lines r&find_parent_scopezFoldingPanel.find_parent_scopens&&(( T1 1 1"2   / ;j((!J.. $ 4 4 6 6 H H&!(!( r'cp|jD]!}|jj|"g|jdd<dS)z'Clear scope decorations (on the editor)N)rKr+rtrur%r~s r&_clear_scope_decoszFoldingPanel._clear_scope_decos{sG% 1 1D K # * *4 0 0 0 0!!!!r'c|jj}|dkrt|d}nt|d}|S)z Gets the base scope highlight color (derivated from the editor background) For lighter themes will be a darker color, and for darker ones will be a lighter color ri)r+r lightnessr)r%colors r&_get_scope_highlight_colorz'FoldingPanel._get_scope_highlight_colorsH + ??  s " "s++EEs++E r'c|}tjd}t|jt d|dz ||}|||dd|jj ||j |dS) z Create a decoration and add it to the editor. Args: start (int) start line of the decoration end (int) end line of the decoration codefoldingrr)rr draw_orderTF)clearN) rrr.rr+rmaxset_backgroundset_full_widthrtaddrKappend)r%rrrrds r&_decorate_blockzFoldingPanel._decorate_blocks//11 _]33 4;//11&)!UQY&7&7$'&0 2 2 2  U+++ ##A&&&   #####r'c|}|j|}||f}|j |j|krG||_||\}}|j|s|||dSdSdS)zu Highlights the current fold scope. :param block: Block that starts the current fold scope. N)rr]rPrr^r)r%rorrscoperrs r&_highlight_blockzFoldingPanel._highlight_blocks &&(( ' 3X&   '4+>%+G+G"'D   # # % % %JE3&u- 1$$UC00000 1 1 ,H+Gr'c tt||t|j}||}|dkr|j |}| |}| }||j vr|j +tjt!t"j|j | krG|j @| |_ ||n[#t($rYnOwxYw| |_ |j|j|n#t($rYnwxYw||_n3|jd|_ tj|dSdS)z Detect mouser over indicator and highlight the current scope in the editor (up and down decoration arround the foldable text when the mouse is over an indicator). :param event: event rN)rr!mouseMoveEventrr+line_nbr_from_positionposyrrrrr]rOr setOverrideCursorrr PointingHandCursorrr0rZ request_job_highight_blockcancel_requestsrestoreOverrideCursorrepaint)r%rrlinerorvr/s r&rzFoldingPanel.mouseMoveEvents  lD!!00777  $ $((99 199K((**<->-@-@@@-9-2,=,=,?,?D)--e4444# -2,=,=,?,?D).:: 15::::#(-$$&66888(,%2444 LLNNNNNI 9s$ E"" E/.E/ F-- F:9F:c<d|_|dS)NT)rXr)r%rs r& enterEventzFoldingPanel.enterEvents $ r'cztt||tj|j|js#|d|_ d|_ nd|_ | |j d|_dS)z Removes scope decorations and background from the editor and the panel if highlight_caret_scope, else simply update the scope decorations to match the caret scope. Nr?F)rr! leaveEventr rrZrr=rrOrPr@rCr+rrX)r%rr/s r&rzFoldingPanel.leaveEvents lD!!,,U333*,,, ..000) *  # # % % %$(D !"&D   DO  ' ' ) ) )  %r'c|}|j|dz|}tjd}t ||}|jj|j ||_ ||_ | | t|d|||t%t&j||j|<|jj|dS)zc Add fold decorations (boxes arround a folded block in the editor widget). rr)rrN)rr+get_text_regionrr.rsignalsclickedrB_on_fold_deco_clickedtooltipro select_line set_outlinerrrset_foregroundrr COLOR_TEXT_4rLrtr)r%rorrtextrr~s r&rpz!FoldingPanel._add_fold_decorations- &&(( {**:>8DD _]33 e ;;; $$T%?@@@      + + - -s44 5 5 5 D;;==>>> F=#=>>???(,*% ##D)))))r'c||kri|rU|d|}||kr|U|S)NFrisValid setVisiblenext)r%rors r&_get_block_until_linez"FoldingPanel._get_block_until_linest!!X--%--//-   U # # #JJLLE!!X--%--//- r'c||kri|rU|d|}||kr|U|S)z3Fold region spanned by *start_line* and *end_line*.Frr%rorrs r& fold_regionzFoldingPanel.fold_regionst!!H,,,   U # # #JJLLE!!H,,, r'c||dz |jvrL|j|dz }|j|dz |jj|||kr|r|}|dd}||jvr;||kr5|j|}|j |rd}| ||dz }|r| }||kr|dSdSdSdS)z5Unfold region spanned by *start_line* and *end_line*.rTFN) rLrsr+rtrurrrr]r^rr)r%rorrr~ current_lineget_next block_ends r& unfold_regionzFoldingPanel.unfold_regionsO >T. . .$Z!^4D   ! !*q. 1 1 1 K # * *4 0 0 0!!H,,, ,,..L   T " " "H 444$ 22 0> &|4M$H 66ui!mLLE % !!H,,,,,,,,,,,r'c|}||jvrdS|j|}|j|r?||||d|j|<|j|||n5||||d|j|<||dS)z Toggle a fold trigger block (expand or collapse it). :param block: The QTextBlock to expand/collapse NFT) rr]r^r rOrrr_refresh_editor_and_scrollbarsrs r&toggle_fold_triggerz FoldingPanel.toggle_fold_trigger4s &&(( T1 1 1 F' 3  z * &   uj( ; ; ;.3D  +$0$$Z:::   UJ 9 9 9.2D  +  # # % % % ++-----r'c|jH|j|j}||dSdS)z+Folds/unfolds the pressed indicator if any.N)rOr+rrr )r%rros r&mousePressEventzFoldingPanel.mousePressEventIsV  ,K((**<<%''E  $ $U + + + + + - ,r'c:||jdS)z & $D ""$$ ;&,&;&;&=&=&,&9&9&;&;&=""'-oo&7&7%:") 6 6 ,,..88=="..0014 $"666+J77#3J?H% 6&*=*=*?*? 6' 44AA&22,,..1F1F1H1HHH$*$9$9$;$;EE$&$5$5$7$7E??,,v/B/B/D/DDD"("5"5"7"7CC"$//"3"3Cu---sBM::: 11"555 %D   9 & &r'ct|j|j|j}||dz|jt|j|dS)ap Refrehes editor content and scollbars. We generate a fake resize event to refresh scroll bar. We have the same problem as described here: http://www.qtcentre.org/threads/44803 and we apply the same solution (don't worry, there is no visual effect, the editor does not grow up at all, even with a value = 500) rN) rr+mark_whole_doc_dirtyrsizerirh resizeEventr)r%ss r&r z+FoldingPanel._refresh_editor_and_scrollbarss 4;44666  K     17799q=!!!  T[-=-=-?-? C CDDDDDr'cr||j}|ri|}||jvr$|j|}|||||}|i| |j }| |j |j ||jdS)zd Collapses all triggers and makes all blocks with fold level > 0 invisible. N)rr+r firstBlockrrr]rrr r movePositionStartr$collapse_all_triggeredemit)r%rorvrr(s r& collapse_allzFoldingPanel.collapse_alls     $$&&1133mmoo !++--Kd222/ <   X>>>JJLLE mmoo ! ++--- [ # # % % !!! !!"%%% #((*****r'c|jD].}|j|}|jj|/i|_dS)z#Clear the folded block decorations.N)rLr+rtru)r%r}r~s r&rzFoldingPanel._clear_block_decosK* 1 1I$Y/D K # * *4 0 0 0 0r'c|j}|ri|}||jvr$|j|}|||||}|i|| |j dS)zExpands all fold triggers.N) r+rr/r BlockNumberr]r rrr expand_all_triggeredr3)r%rorvrs r& expand_allzFoldingPanel.expand_alls $$&&1133mmoo !++--Kd222/ <""5+x@@@JJLLE mmoo !     ++--- !&&(((((r'c|j}|}|j|kr||}|}||jvr@||_ ||n$#t$rYnwxYw| ||_dS)z Highlight the scope of the current caret position. This get called only if :attr:` spyder.widgets.panels.FoldingPanel.highlight_care_scope` is True. N) r+rrr@rror]rOrr0r)r%r& block_nbrrorvs r&rCz#FoldingPanel._highlight_caret_scopes''))&&(( ?i ' '**6<<>>::E++--Kd222(-(9(9(;(;%))%0000D '')))#sB-- B:9B:)F).__name__ __module__ __qualname____doc__rrbooltrigger_state_changedr2r8propertyr#setterr6r=rIrdrlrrrrrnrrrrrrrrrprrr r rrrrr r4rr9rC __classcell__)r/s@r&r!r!)s#F:t44#VXX!688 !!X!    & &X &   %%X%!"!*%"%"%"N8=####JNNNNN@777.#(#(#(J#I#I#IJ   """   $$$&111$/////b&&&&&&***( %%%,...*,,,---HHH&'&'&'&REEE"+++& ) ) )$$$$$$$r'r!)+r?mathrr intervaltreer qtpy.QtCorerrrrr r qtpy.QtWidgetsr r r qtpy.QtGuirrrrrrrrr"spyder.plugins.editor.panels.utilsr$spyder.plugins.editor.api.decorationrrspyder.api.panelr"spyder.plugins.editor.utils.editorrrrspyder.utils.icon_managerrspyder.utils.paletterr!r'r&rQs &%%%%%AAAAAAAAAAAAAAAAEEEEEEEEEE!!!!!!!!!!!!!!!!!!!!!! =<<<<<LLLLLLLL""""""==========))))))......u $u $u $u $u $5u $u $u $u $u $r'