B )`@sdZddlmZddlmZddlmZddlZddlZddlmZddlZddl Z ddl Z ddl Z ddl m Z ddlmZdd lmZdd lmZdd lmZdd lmZdd lmZdZdZdZdZdZdZdZdZddgZdZddZGddde Z!Gdddej"Z#dS)zCCurses-Based Command-Line Interface of TensorFlow Debugger (tfdbg).)absolute_import)division)print_functionN)textpad)xrange)base_ui) cli_shared)command_parser)curses_widgets)debugger_cli_common) tensor_formatrefreshZupZdownZ up_a_lineZ down_a_linehomeendZscroll_to_line_indexzgnome-terminalzxfce4-terminalzxterm-256colorcCsrxl|D]d}|d|kr&|dkrnqt|dtr@|dn|dg}x|D]}t|tjrP|jSqPWqWdS)a=Attempt to extract command from the attribute segments of a line. Args: mouse_x: (int) x coordinate of the mouse event. attr_segs: (list) The list of attribute segments of a line from a RichTextLines object. Returns: (str or None) If a command exists: the command as a str; otherwise, None. rN) isinstancelistr MenuItemcontent)mouse_xZ attr_segsseg attributesattrrT/home/dcms/DCMS/lib/python3.7/site-packages/tensorflow/python/debug/cli/curses_ui.py _get_command_from_line_attr_segs3s     rc@sBeZdZdZejdejZddZd ddZ dd Z d d Z d S) ScrollBaraVertical ScrollBar for Curses-based CLI. An object of this class has knowledge of the location of the scroll bar in the screen coordinates, the current scrolling position, and the total number of text lines in the screen text. By using this information, it can generate text rendering of the scroll bar, which consists of and UP button on the top and a DOWN button on the bottom, in addition to a scroll block in between, whose exact location is determined by the scrolling position. The object can also calculate the scrolling command (e.g., _SCROLL_UP_A_LINE, _SCROLL_DOWN) from the coordinate of a mouse click event in the screen region it occupies. _on_cCs||_||_||_||_||_||_||d|_|j|jkrVtd|j|jd|j|jdkr~td|j|jddS)a)Constructor of ScrollBar. Args: min_x: (int) left index of the scroll bar on the screen (inclusive). min_y: (int) top index of the scroll bar on the screen (inclusive). max_x: (int) right index of the scroll bar on the screen (inclusive). max_y: (int) bottom index of the scroll bar on the screen (inclusive). scroll_position: (int) 0-based location of the screen output. For example, if the screen output is scrolled to the top, the value of scroll_position should be 0. If it is scrolled to the bottom, the value should be output_num_rows - 1. output_num_rows: (int) Total number of output rows. Raises: ValueError: If the width or height of the scroll bar, as determined by min_x, max_x, min_y and max_y, is too small. rz%Insufficient width for ScrollBar (%d)z&Insufficient height for ScrollBar (%d)N)_min_x_min_y_max_x_max_y_scroll_position_output_num_rows_scroll_bar_height ValueError)selfZmin_xZmin_yZmax_xZmax_yscroll_positionZoutput_num_rowsrrr__init__Ws zScrollBar.__init__FcCs8tt|j|jd|jdd}|r4||jS|S)aGet the 0-based y coordinate of the scroll block. This y coordinate takes into account the presence of the UP and DN buttons present at the top and bottom of the ScrollBar. For example, at the home location, the return value will be 1; at the bottom location, the return value will be self._scroll_bar_height - 2. Args: screen_coord_sys: (`bool`) whether the return value will be in the screen coordinate system. Returns: (int) 0-based y coordinate of the scroll block, in the ScrollBar coordinate system by default. For example, when scroll position is at the top, this return value will be 1 (not 0, because of the presence of the UP button). When scroll position is at the bottom, this return value will be self._scroll_bar_height - 2 (not self._scroll_bar_height - 1, because of the presence of the DOWN button). rr)intfloatr$r%r&r!)r(screen_coord_sysZ rel_block_yrrr_block_yszScrollBar._block_yc Cs|j|jd}d|}d||jfg}|jdkr|}|dkrJd}d}n,|dkr\d}d}n|d krnd }d }nd }d }tj|gdd||jfgid}x6td|jdD]"}||kr|nd} |j || dqW|j ||dnt|g|j}|S)zGet the RichTextLines layout of the scroll bar. Returns: (debugger_cli_common.RichTextLines) The text layout of the scroll bar. r rUDrUPZDNrzUP zDN z UP DOWN)font_attr_segsN) r"r BASE_ATTRr%r.r RichTextLinesrr&append) r(widthZ empty_lineZforeground_font_attr_segsZblock_yZup_textZ down_textlayoutir4rrrr9s2 zScrollBar.layoutcCsn|jdkrdS||jkrtS||jkr*tS||jddkrH||jkrHtS||jddkrf||jkrftSdSdS)NrT)r-)r%r!_SCROLL_UP_A_LINEr#_SCROLL_DOWN_A_LINEr. _SCROLL_DOWN _SCROLL_UP)r(mouse_yrrrget_click_commands     zScrollBar.get_click_commandN)F) __name__ __module__ __qualname____doc__r COLOR_BLACK COLOR_WHITEr5r*r.r9r@rrrrrGs  ( &rc@seZdZdZdZedZedZdZdZ dZ dZ d Z ed ed d gZ d ejejdiZejejejejejejejejejejejejejejejejiZddejejejejiZejdejZejdejZejdejZejdejZ ejdejZ!ejdejZ"ejdejZ#dZ$dZ%e&'Z(dddZ)ddZ*ddZ+ddZ,ddZ-dd Z.d!d"Z/d#d$Z0dd&d'Z1d(d)Z2d*d+Z3d,d-Z4dd.d/Z5d0d1Z6d2d3Z7d4d5Z8d6d7Z9d8d9Z:d:d;Z;dd?Z=d@dAZ>dBdCZ?dDdEZ@dFdGZAddHdIZBddKdLZCdMdNZDejEdfdOdPZFdQdRZGdSdTZHddUdVZIdWdXZJdYdZZKd[d\ZLd]d^ZMdd_d`ZNdadbZOddcddZPdedfZQdgdhZRdidjZSdkdlZTdmdnZUdodpZVdqdrZWdsdtZXddudvZYdwdxZZdydzZ[d{d|Z\dd}d~Z]ddZ^ddZ_dS)CursesUIzCurses-based Command-line UI. In this class, the methods with the prefix "_screen_" are the methods that interact with the actual terminal using the curses library.  /@nextprev  iW transparentrz Processing...dNcCstjj|||d|||t|_g|_ d|_ d|_ d|_ t d|_d|_d|_d|_d|_d|_yttj|jWntk rYnX|jd|jdd gd dS) zConstructor of CursesUI. Args: on_ui_exit: (Callable) Callback invoked when the UI exits. config: An instance of `cli_config.CLIConfig()` carrying user-facing configurations. ) on_ui_exitconfigrrV NZmousez/Get or set the mouse mode of this CLI: (on|off)m)Zprefix_aliases)rBaseUIr* _screen_init_screen_refresh_size _init_layoutr ZCommandHistory_command_history_store_active_command_history_command_pointer_command_history_limit_pending_commandr ZCursesNavigationHistory _nav_history _output_pad_output_pad_row_output_array_pointer_indices_curr_unwrapped_output_curr_wrapped_outputsignalSIGINT_interrupt_handlerr'Zregister_command_handler_mouse_mode_command_handler)r(rWrXrrrr*s0   zCursesUI.__init__cCstdd|_d|_d|_d|_d|_|j|j|jd|_|jd|j|_ |j d|_ |j d|_ t |jd|_ d|_d|_g|_|jd|_|jd|_|j|jd|j|j|jd |_dS) zInitialize the layout of UI components. Initialize the location and size of UI components such as command textbox and output region according to the terminal size. rectangleztop left bottom rightrrrrSi'N)topleftbottomright) collections namedtuplero_command_textbox_height _title_row _nav_bar_row_output_top_rowr#r%_output_scroll_row_candidates_top_row_candidates_bottom_rowr+_candidates_max_linesmax_output_lines_curr_search_regex_unwrapped_regex_match_lines_output_pad_screen_heightr"Z_output_pad_screen_width_output_pad_screen_location)r(rrrr_Ps*     zCursesUI._init_layoutcCs4tdtkrttjd<t|_d|_| dS)zbScreen initialization. Creates curses stdscr and initialize the color pairs for display. Z COLORTERMTERMN) osgetenv_COLOR_READY_COLORTERMS_COLOR_ENABLED_TERMenvironcursesZinitscr_stdscr_command_window_screen_color_init)r(rrrr]s   zCursesUI._screen_initcCstti|_d}xh|jD]^}xX|jD]N}|d7}t||j||j||}|dkrl|d|7}t||j|<q.Wq"Wy,|d7}t|ddt||jtj <Wn*tj k r|jtj |jtj <YnXtj |jd<tj |jd<tj|jd <|jtj|_d S) z Initialization of screen colors.rrrTrrUZboldZblink underlineN)rZ start_colorZuse_default_colors _color_pairs_FOREGROUND_COLORS_BACKGROUND_COLORSZ init_pair color_pairrZ COLOR_GRAYerror COLOR_GREENA_BOLDZA_BLINK A_UNDERLINErF_default_color_pair)r(Z color_indexZfg_colorZbg_colorZ color_namerrrrs.       zCursesUI._screen_color_initcsTttjdjd_j dfdd dS)zLaunch the curses screen.r mouse_modecs|dS)Nr)_set_mouse_enabledget)cfg)r(rrz)CursesUI._screen_launch..N) rZnoechoZcbreakrkeypadrXr_mouse_enabled_screen_set_mousemaskZ set_callback_screen_create_command_window)r(enable_mouse_on_startr)r(r_screen_launchs zCursesUI._screen_launchcCs<|jr |`t|j|jt|j|j|jt|j|_dS)z/Create command window according to screen size.N)rrZnewwinrvr"len CLI_PROMPTr#)r(rrrrs z&CursesUI._screen_create_command_windowcCs|jdS)N)rr )r(rrr_screen_refreshszCursesUI._screen_refreshcCsR|jdtttyttjtjWnt k rLYnXdS)zTerminate the curses screen.rN) rrrZnocbreakZechoZendwinrkrlSIG_DFLr')r(rrr_screen_terminates zCursesUI._screen_terminateTcCsj|j|j|d|dk r(|||dk r>|j||d|}|jrT|||j|S)zCRun the CLI: See the doc of base_ui.BaseUI.run_ui for more details.)rN) title_color) _single_instance_lockacquirer_dispatch_command_title_ui_loop _on_ui_exitrrelease)r(Z init_commandtitlerr exit_tokenrrrrun_uis    zCursesUI.run_uicCs |jS)N)_command_handler_registryget_help)r(rrrrszCursesUI.get_helpcGs,y|jj|Wntjk r&YnXdS)N)raddstrrr)r(argsrrr_addstr szCursesUI._addstrcGs*y|j|Wntjk r$YnXdS)N)r rr)r(padrrrr _refresh_padszCursesUI._refresh_padcCsP||j|jd|jtj|j|j t j |jdd|_ | |dS)zCreate command textbox on screen. Args: existing_command: (str) A command string to put in the textbox right after its creation. rT)Z insert_modeN)rr#rvrrrrr rclearrZTextbox_command_textbox _auto_key_in)r(existing_commandrrr_screen_create_command_textboxs    z'CursesUI._screen_create_command_textboxc Csx|jdkr|j|j }n|j}||y|\}}}Wn&tjk rd}z|jSd}~XYnX|sv||jkrvq||j ks|t j kr| |}|dk r|Sq||jkr| |}||_d|_q|r||_qWdS)zfCommand-line UI loop. Returns: An exit token of arbitrary type. The token can be None. rN)rbrardr_get_user_commandr CommandLineExitr CLI_TAB_KEY CLI_CR_KEYSr KEY_MOUSEr _tab_completeZ_cmd_ptr)r(rcommand terminatorZpending_command_changederZ tab_completedrrrr-s,      zCursesUI._ui_loopcCs,d|_d|_|}||}||j|jfS)aGet user command from UI. Returns: command: (str) The user-entered command. terminator: (str) Terminator type for the command. If command is a normal command entered with the Enter key, the value will be the key itself. If this is a tab completion call (using the Tab key), the value will reflect that as well. pending_command_changed: (bool) If the pending command has changed. Used during command history navigation. NF)_textbox_curr_terminator _textbox_pending_command_changed_screen_get_user_command_strip_terminator)r(rrrrrQs  zCursesUI._get_user_commandcCs|jj|jdS)N)validate)rZedit_on_textbox_keypress)r(rrrrgsz!CursesUI._screen_get_user_commandcCs8|s|Sx&|jD]}|dkr|t|d}qW|S)NrY)rreplacechrstrip)r(rvrrrrjs  zCursesUI._strip_terminatorcCs*|j\|_|_|j|jkr&|j|_dS)N)rZgetmaxyxr#r"_SCREEN_WIDTH_LIMIT)r(rrrr^ts zCursesUI._screen_refresh_sizecCs||jkr<|jr&|j}|j}ql|jd|jddSn0|jrX|j}|j}n|jd|jddS| |j |dkr|j t |ddS)zNavigate in screen output history. Args: command: (`str`) the navigation command, from {self._NAVIGATION_FORWARD_COMMAND, self._NAVIGATION_BACK_COMMAND}. z$At the LATEST in navigation history!)colorNz$At the OLDEST in navigation history!r) line_index) _NAVIGATION_FORWARD_COMMANDreZcan_go_forwardZ go_forwardr)_toast_NAVIGATION_WARNING_COLOR_PAIRZ can_go_backZgo_back_display_output screen_output_scroll_output_SCROLL_TO_LINE_INDEX)r(ritemr)rrr_navigate_screen_outputys         z CursesUI._navigate_screen_outputc Cs|jr|j|j|jd||jkr(tjS||jks<||jkrJ| |dS|rZ|j || |j r|jrt|t|j kr|t|j d}||_|j|j|dn|jr|j|jd|jdd|_d|_dS| |jrr|dd}|rdy8t|}t|j|\}}}}|s,|jt|d Wn2tk r`}z|t |Wdd}~XYnXn |d dSy|!|\} } } Wn2t"k r}z|t |dSd}~XYnX| sdSd |j#d i} d} |j$%| r*y|j$j&| | | d }Wn,tj'k r&}z |j(} Wdd}~XYnXnt)|j*d| g}g|_+| dk rT| S|j,-||d||| ry|.| |/d| Wn$tk r|d| YnXd|_d|_dS)zDispatch user command. Args: command: (str) Command to dispatch. Returns: An exit token object. None value means that the UI loop should not exit. A non-None value means the UI loop should exit. )rN)highlight_regexT) is_refreshrrrYr)rzEmpty indices.colsr) screen_infozInvalid command prefix "%s"zWrote output to %szFailed to write output to %s)0rfr_UI_WAIT_MESSAGE_UI_WAIT_COLOR_PAIRZCLI_EXIT_COMMANDSr ZEXPLICIT_USER_EXITr_NAVIGATION_BACK_COMMANDrr`Z add_command startswithREGEX_SEARCH_PREFIXrirrrrrbrd TENSOR_INDICES_NAVIGATION_PREFIXrr Z parse_indicesr Zlocate_tensor_elementrjrr Exception _error_toaststrZ_parse_command SyntaxErrorr"rZ is_registeredZdispatch_commandrrr6ERROR_MESSAGE_PREFIXrareZadd_itemZ write_to_file _info_toast)r(rregexZ indices_strindicesZomittedr_rprefixrZoutput_file_pathrrrrrrrs         "    zCursesUI._dispatch_commandcCs|j}|S)zGather the text string in the command text box. Returns: (str) the current text string in the command textbox, excluding any return keys. )rgatherr)r(txtrrr_screen_gather_textbox_strs z#CursesUI._screen_gather_textbox_strc Cst|tstdt|||jkr0||_|jS||jkrH|j|_|jS|tj kr`| t |S|tj krx| t |S|tjkr| t|S|tjkr| t|S|tjtjgkrJ|js|}|j||j|_|jr8|tjkr|jt|jkr6|jd7_n&|tjkr>|jdkr>|jd8_nd|_||_|jS|tjkr||||||jS|tj kr|j!ry|"\}}}}}Wntj#k rd}YnX|tj$kr|||j%dkr|j&'dx||tj$krht(|j)|j*+|}|t t fkr*| ||j&,y|"\}}}}}Wntj#k rbYnXqW|j&'d|Snz|tj-kr||j%dkr|j*+|}|dk r| ||S|.||}|r|/|0|} | dk rt1j2| dn d|_3d|_g|_|j45||SdS) aText box key validator: Callback of key strokes. Handles a user's keypress in the input text box. Translates certain keys to terminator keys for the textbox to allow its edit() method to return. Also handles special key-triggered events such as PgUp/PgDown scrolling of the screen output. Args: x: (int) Key code. Returns: (int) A translated key code. In most cases, this is identical to the input x. However, if x is a Return key, the return value will be CLI_TERMINATOR_KEY, so that the text box's edit() method can return. Raises: TypeError: If the input x is not of type int. debugger_cli_common.CommandLineExit: If a mouse-triggered command returns an exit token when dispatched. z1Key validator expected type int, received type %srrNrTF)r)6rr+ TypeErrortyperrCLI_TERMINATOR_KEYrrZ KEY_PPAGErr;Z KEY_NPAGEr<ZKEY_HOME _SCROLL_HOMEZKEY_END _SCROLL_ENDZKEY_UPZKEY_DOWNrarr`Z lookup_prefixrcrbrZ KEY_RESIZEr^r_r_redraw_outputrr_screen_getmouserBUTTON1_PRESSEDr"rZnodelayZnapms_MOUSE_SCROLL_DELAY_MS _scroll_barr@ZgetchBUTTON1_RELEASED_fetch_hyperlink_commandrrr rr_KEY_MAPr) r(xZ hist_prefixrrr?Zmouse_event_typeZscroll_commandrrrrrr s                                 zCursesUI._on_textbox_keypresscCstS)N)rZgetmouse)r(rrrrszCursesUI._screen_getmousecCs2|jdk r.|||j|j|jdddS)NT)r)ri_display_nav_bar_display_main_menur)r(rrrrs  zCursesUI._redraw_outputcCs|j}|jr|d7}||jkr6|jr6t||jjdS||jkrX|jrXt||jjdS||j|}||jjkrt||jj|SdS)Nrr) ry_main_menu_padrx_nav_barrr4 _main_menurgrj)r(rr? output_topZabsolute_mouse_yrrrrs z!CursesUI._fetch_hyperlink_commandcCsPd||_t|j|jkr8|jd|jt|j7_|j|j|j|ddS)zDisplay title. Args: title: (str) The title to display. title_color: (str) Color of the title, e.g., "yellow". z--- %s -)rN)Z _title_linerr"_screen_draw_text_linerw)r(rrrrrrs zCursesUI._titleFcCs6|r ||pd}x|D]}|jt|qWdS)zAutomatically key in a command to the command Textbox. Args: command: The command, as a string or None. erase_existing: (bool) whether existing text (if any) is to be erased first. rYN)_erase_existing_commandr do_commandord)r(rZerase_existingcrrrrs  zCursesUI._auto_key_incCs2t|j}xt|D]}|j|jqWdS)z'Erase existing text in command textpad.N)rrrrr BACKSPACE_KEY)r(Z existing_lenrrrrr sz CursesUI._erase_existing_commandcCsft|tstdt||jkr.|d|j}|dkr<|jn|j|}||d|||B|dS)zRender a line of text on the screen. Args: row: (int) Row index. line: (str) The line content. attr: curses font attribute. color: (str) font foreground color name. Raises: TypeError: If row is not of type int. zInvalid type in rowNr) rr+rrr"rrrr)r(rowlinerrrrrrr s  zCursesUI._screen_draw_text_linecCs t||S)a Generate a new pad on the screen. Args: rows: (int) Number of rows the pad will have: not limited to screen size. cols: (int) Number of columns the pad will have: not limited to screen size. Returns: A curses textpad object. )rZnewpad)r(rowsrrrr_screen_new_output_pads zCursesUI._screen_new_output_padcst||jd\|_|jjdg|jd|j|jkr|j d|j|_|jj d|jdt |jdt j fg|jj|j<|||j||j|j\|_|_|_fdd|jDS) acActually render text output on the screen. Wraps the lines according to screen width. Pad lines below according to screen height so that the user can scroll the output to a state where the last non-empty line is on the top of the screen. Then renders the lines on the screen. Args: output: (RichTextLines) text lines to display on the screen. These lines may have widths exceeding the screen width. This method will take care of the wrapping. Returns: (List of int) A list of line indices, in the wrapped output, where there are regex matches. rrYrrzOutput cut off at %d lines!rUcsg|] }|qSrr).0r)wrapped_line_indicesrr sz3CursesUI._screen_display_output..)r wrap_rich_text_linesr"rjlinesextendr%Z num_linesr~slicer7rr COLOR_MAGENTAr4rr_display_linesrf_output_pad_heightZ_output_pad_widthr)r(outputr)rr_screen_display_outputs   "  zCursesUI._screen_display_outputc Cs,|sdS|rxytj|||jd}Wn0tk rR}z|t|dSd}~XYnX|sh|jtj|_q|j d7_ n ||_ g|_| |}|rd}x|D]}||j kr|}PqW|dkr|j t |dn|jd|jdnJ|r| tn:tj|jkr|jtj}|j t |dnd|_ | tdS) aDisplay text output in a scrollable text pad. This method does some preprocessing on the text lines, render them on the screen and scroll to the appropriate line. These are done according to regex highlighting requests (if any), scroll-to-next-match requests (if any), and screen refresh requests (if any). TODO(cais): Separate these unrelated request to increase clarity and maintainability. Args: output: A RichTextLines object that is the screen output text. is_refresh: (bool) Is this a refreshing display with existing output. highlight_regex: (str) Optional string representing the regex used to search and highlight in the current screen output. N)Z font_attrrrUr)rzPattern not found)r)r Z regex_find_SEARCH_HIGHLIGHT_FONT_ATTRr'rr annotationsZREGEX_MATCH_LINES_KEYrrgrir!rrr_ERROR_TOAST_COLOR_PAIR_SCROLL_REFRESHZINIT_SCROLL_POS_KEYr) r(r rrrZwrapped_regex_match_linesZnext_match_lineZ match_linerrrrr$sB       zCursesUI._display_outputcCst|tjstd|t|t|j}|jd}| ||}xTt t|jD]B}||j kr|j |||j||j |dqR| |||j|qRW|||fS)aDisplay RichTextLines object on screen. Args: output: A RichTextLines object. min_num_rows: (int) Minimum number of output rows. Returns: 1) The text pad object used to display the main text body. 2) (int) number of rows of the text pad, which may exceed screen size. 3) (int) number of columns of the text pad. Raises: ValueError: If input argument "output" is invalid. zBOutput is required to be an instance of RichTextLines, but is not.r)color_segments) rr r6r'rmaxrrr"rrr4_screen_add_line_to_output_pad)r(r Z min_num_rowsrrrr:rrrrjs    zCursesUI._display_linescCsr|jd}|d||_|j||j|j|_|j|jd|jj dd|dd|jj krf|jj dndddS)Nrrr)r&) r"r _nav_bar_padrerenderrrrr(rr4)r(Z nav_bar_widthrrrrs  zCursesUI._display_nav_barcCstj|jkr|jtjjdddgd|_|d|jd|_t|j|jd\}}|j |jd|j dd|j krz|j dnd d n d |_d |_d S) a Display main menu associated with screen output, if the menu exists. Args: output: (debugger_cli_common.RichTextLines) The RichTextLines output from the annotations field of which the menu will be extracted and used (if the menu exists). z| z | r)rdividerZenabled_item_attrsrrrrN)r&) r Z MAIN_MENU_KEYr#Zformat_as_single_liner rr"rrr(rr4)r(r Z wrapped_menurrrrrs zCursesUI._display_main_menucCs2||jdkr.|||d|jd||jdS)apPad the whitespace at the end of a line with the default color pair. Prevents spurious color pairs from appearing at the end of the lines in certain text terminals. Args: pad: The curses pad object to operate on. row: (`int`) row index. line_end_x: (`int`) column index of the end of the line (beginning of the whitespace). rr/rN)r"rr)r(rrZ line_end_xrrr_pad_line_end_with_whitespaces z&CursesUI._pad_line_end_with_whitespacec Cs|s,||d||j|||t|dSt|ts>tdg}g}|dddkrXn"|d|ddf||jxt||ddt|ddfgD]\\}}} \} } } |||ft| ts| g} t j } x>| D]6} |j rt| t j r| t jO} q| |j| |jO} qW|| || kr||| f||jqWxLt||D]>\}}|d|jkrR|||d||d|d|qRW|r||||dddS)aRender a line in a text pad. Assumes: segments in color_segments are sorted in ascending order of the beginning index. Note: Gaps between the segments are allowed and will be fixed in with a default color. Args: pad: The text pad to render the line in. row: Row index, as an int. txt: The text to be displayed on the specified row, as a str. color_segments: A list of 3-tuples. Each tuple represents the beginning and the end of a color segment, in the form of a right-open interval: [start, end). The last element of the tuple is a color string, e.g., "red". Raisee: TypeError: If color_segments is not of type list. rNz4Input color_segments needs to be a list, but is not.rrU)rrr,rrrrr7ziprA_NORMALrr rrrrr")r(rrrr&Z all_segmentsZall_color_pairsZ curr_startZcurr_endZ curr_attrsZ next_startrZ curses_attrrsegmentrrrrr(s@  4     *z'CursesUI._screen_add_line_to_output_padc Cs||||||||t|jdd|jd|jd|j|j|j|_||j |jd\}} } ||dd|j d|jd|jd|jddS)Nrrrr) rrr"r%rgrrrrr9ry) r(rZ viewport_topZ viewport_leftZscreen_location_topZscreen_location_leftZscreen_location_bottomZscreen_location_rightZ scroll_padrrrr_screen_scroll_output_pads z"CursesUI._screen_scroll_output_padc Cs|js dS|tkrn>|tkrJ|jt|jd8_|jdkrFd|_n |tkr|jt|jd7_|j|j|jdkr|j|jd|_n|t kr|jddkr|jd8_n|t kr|jd|j|jkr|jd7_nl|t krd|_n\|t kr|j|jd|_n>|t krHt|ts@tdt|t f||_n td||j|j|jj}|jrz|d7}||j|jd||jj|jj|jj||||_|j|j |j|j!ddS)aScroll the output pad. Args: direction: _SCROLL_REFRESH, _SCROLL_UP, _SCROLL_DOWN, _SCROLL_UP_A_LINE, _SCROLL_DOWN_A_LINE, _SCROLL_HOME, _SCROLL_END, _SCROLL_TO_LINE_INDEX line_index: (int) Specifies the zero-based line index to scroll to. Applicable only if direction is _SCROLL_TO_LINE_INDEX. Raises: ValueError: On invalid scroll direction. TypeError: If line_index is not int and direction is _SCROLL_TO_LINE_INDEX. Nrrrz*Invalid line_index type (%s) under mode %szUnsupported scroll mode: %s)r)"rfr%r>rgr+r%r=rrr;r<rrrrrrr'reZupdate_scroll_positionrrprr0rqrrrs_screen_render_nav_bar_screen_render_menu_pad_compile_ui_status_summaryZ _scroll_infor rz_STATUS_BAR_COLOR_PAIR)r( directionrZoutput_pad_toprrrr$s\        zCursesUI._scroll_outputc Cs*|jr&||jdd|jd|jj|jdS)Nr)r)rrxrrpr")r(rrrr1qszCursesUI._screen_render_nav_barc Cs,|jr(||jdd|jjd|jj|jdS)Nr)rrrrpr")r(rrrr2vsz CursesUI._screen_render_menu_padcCs\d}|j|jdkrxdtdt|j|j|jd}|jdkrHd}n |j|j|jdkrdd}nd}|d ||f7}||_|jr|jdr|||jd7}|d 7}|jd r|||jd 7}|d 7}d }||jrdnd7}t |t |d|j krB|d |j t |t |d7}|d 7}||7}|d7}n|d |j t |7}|S)zCompile status summary about this Curses UI instance. The information includes: scroll status and mouse ON/OFF status. Returns: (str) A single text line summarizing the UI status, adapted to the current screen width. rYrgY@g?rz (PgDn)z (PgUp)z (PgDn/PgUp)z--- Scroll%s: %.2f%% r rUr/zMouse: ONZOFFrSz ---) rrminr,rg_show_array_indicesrh_format_indicesrrr")r(infoZscroll_percentageZscroll_directionsZmouse_mode_strrrrr3|s<     " z#CursesUI._compile_ui_status_summarycCst|ddS)Nr/rY)reprr)r(rrrrr:szCursesUI._format_indicescCsR|d}|j}|jr|d7}|jj|d}||}|s@|rJ||dSdSdS)aShow array indices for the lines at the top and bottom of the output. For the top line and bottom line of the output display area, show the element indices of the array being displayed. Returns: If either the top of the bottom row has any matching array indices, a dict from line index (0 being the top of the display area, -1 being the bottom of the display area) to array element indices. For example: {0: [0, 0], -1: [10, 0]} Otherwise, None. rr)rrUN)_show_array_index_at_lineryrrrr)r(Z indices_topr Zbottom_line_indexZindices_bottomrrrr9s   zCursesUI._show_array_indicescCs~|j|}||jjkrvd|jj|krv|jj|d}||}d|}|j}|jr\|d7}|j||j||d|SdSdS)axShow array indices for the specified line in the display area. Uses the line number to array indices map in the annotations field of the RichTextLines object being displayed. If the displayed RichTextLines object does not contain such a mapping, will do nothing. Args: line_index: (int) 0-based line index from the top of the display area. For example,if line_index == 0, this method will display the array indices for the line currently at the top of the display area. Returns: (list) The array indices at the specified line, if available. None, if not available. Zi0rLr)rrN)rgrjr#r:ryrr_ARRAY_INDICES_COLOR_PAIR)r(rpointerrZarray_indices_strZarray_indices_infor rrrr=s    z"CursesUI._show_array_index_at_linecCs`||\}}}|j||\}}|r>t|dkr>||n |g|rT||S||SdS)a[Perform tab completion. Obtains tab completion candidates. If there are no candidates, return command_str and take no other actions. If there are candidates, display the candidates on screen and return command_str + (common prefix of the candidates). Args: command_str: (str) The str in the command input textbox when Tab key is hit. Returns: (str) Completed string. Could be the same as command_str if no completion candidate is available. If candidate(s) are available, return command_str appended by the common prefix of the candidates. rN)Z_analyze_tab_complete_inputZ_tab_completion_registryZget_completionsr_display_candidates)r(Z command_strcontextrZexcept_last_word candidates common_prefixrrrrs    zCursesUI._tab_completec Cs|jr|t|sdSd}|d|}tj|dt|t|dfgid}t||jd\}}t t|j |j }|j |d|_ ||d\}}}||dd|j d|j |d|jd dS) zShow candidates (e.g., tab-completion candidates) on multiple lines. Args: candidates: (list of str) candidates. Nz Candidates: r/rZyellow)r4rrr)rirr%joinr r6rrr"r8rr}r|r{rr0)r(rBZcandidates_prefixZcandidates_lineZcandidates_outputrZcandidates_num_rowsrrrrr@"s$  zCursesUI._display_candidatesc Csr|tj|ddt||ptjfgidd\}}}tt||jd}|dkrX|jd}| |dd|d||dS)a\Display a one-line message on the screen. By default, the toast is displayed in the line right above the scroll bar. But the line location can be overridden with the line_index arg. Args: message: (str) the message to display. color: (str) optional color attribute for the message. line_index: (int) line index. r)r4rNr) rr r6rrrFr8r"rzr0)r(messagerrrrZ right_endrrrrHs   zCursesUI._toastcCs|j|j||jddS)zDisplay a one-line error message on screen. Args: message: The error message, without the preceding "ERROR: " substring. )rN)rrr$)r(rErrrrbszCursesUI._error_toastcCs|j|j||jddS)zmDisplay a one-line informational message on screen. Args: message: The informational message. )rN)rZINFO_MESSAGE_PREFIX_INFO_TOAST_COLOR_PAIR)r(rErrrrlszCursesUI._info_toastcCs6~~|jr||tdtjdtddS)Nz+ tfdbg: caught SIGINT; calling sys.exit(1).)filer)rrprintsysstderrexit)r(Z signal_numframerrrrmvszCursesUI._interrupt_handlercCs~|rt|dkr|rb|ddkr,d}n,|ddkrBd}n|d|ddS|||jrldnd}|d |n |d dS) a<Handler for the command prefix 'mouse'. Args: args: (list of str) Arguments to the command prefix 'mouse'. screen_info: (dict) Information about the screen, unused by this handler. Returns: None, as this command handler does not generate any screen outputs other than toasts. rronToffFzInvalid mouse mode: %sNzMouse mode: %szmouse_mode: syntax error)rlowerrrrr)r(rrenabledZmode_strrrrrns   z$CursesUI._mouse_mode_command_handlercCs$|j|kr ||_||dS)N)rrr)r(rPrrrrs zCursesUI._set_mouse_enabledcCs(|jrttjtjBn tddS)Nr)rrZ mousemaskrr)r(rrrrszCursesUI._screen_set_mousemask)NN)NNNT)N)N)F)FN)N)N)NN)N)`rArBrCrDrrrrrrrrrrrZ KEY_BACKSPACEZKEY_DCrrrFZ COLOR_REDrZ COLOR_YELLOWZ COLOR_BLUEZ COLOR_CYANrrErrr"r>r$rFr4rrrr threadingLockrr*r_r]rrrrrrrrrrrrrrr^rrrrrrrrrr r.r rr!rrrrr,r(r0rr1r2r3r:r9r=rr@rrrrmrnrrrrrrrGs    :: '   $ i   2 F)  C M4+&&    !rG)$rD __future__rrrrtrrrrkrIrQZ six.movesrZtensorflow.python.debug.clirrr r r r r%r>r=r;r<rrrrrrobjectrr\rGrrrrs>