diff options
Diffstat (limited to 'lldb/packages/Python/lldbsuite/test/lldbcurses.py')
-rw-r--r-- | lldb/packages/Python/lldbsuite/test/lldbcurses.py | 609 |
1 files changed, 387 insertions, 222 deletions
diff --git a/lldb/packages/Python/lldbsuite/test/lldbcurses.py b/lldb/packages/Python/lldbsuite/test/lldbcurses.py index bc1fbf00d37..ae0082ce9b7 100644 --- a/lldb/packages/Python/lldbsuite/test/lldbcurses.py +++ b/lldb/packages/Python/lldbsuite/test/lldbcurses.py @@ -4,14 +4,16 @@ from __future__ import absolute_import import curses import curses.panel import sys -import time +import time # Third-party modules import six # LLDB modules + class Point(object): + def __init__(self, x, y): self.x = x self.y = y @@ -27,11 +29,13 @@ class Point(object): def __ne__(self, rhs): return self.x != rhs.x or self.y != rhs.y - + def is_valid_coordinate(self): return self.x >= 0 and self.y >= 0 - + + class Size(object): + def __init__(self, w, h): self.w = w self.h = h @@ -48,11 +52,13 @@ class Size(object): def __ne__(self, rhs): return self.w != rhs.w or self.h != rhs.h + class Rect(object): + def __init__(self, x=0, y=0, w=0, h=0): self.origin = Point(x, y) self.size = Size(w, h) - + def __repr__(self): return str(self) @@ -70,7 +76,7 @@ class Rect(object): def get_max_y(self): return self.origin.y + self.size.h - + def contains_point(self, pt): if pt.x < self.get_max_x(): if pt.y < self.get_max_y(): @@ -84,12 +90,16 @@ class Rect(object): def __ne__(self, rhs): return self.origin != rhs.origin or self.size != rhs.size + class QuitException(Exception): + def __init__(self): super(QuitException, self).__init__('QuitException') + class Window(object): - def __init__(self, window, delegate = None, can_become_first_responder = True): + + def __init__(self, window, delegate=None, can_become_first_responder=True): self.window = window self.parent = None self.delegate = delegate @@ -97,11 +107,11 @@ class Window(object): self.first_responders = list() self.can_become_first_responder = can_become_first_responder self.key_actions = dict() - + def add_child(self, window): self.children.append(window) window.parent = self - + def resize(self, size): self.window.resize(size.h, size.w) @@ -109,7 +119,13 @@ class Window(object): if child in self.children: frame = self.get_frame() orig_frame = child.get_frame() - new_frame = Rect(x=orig_frame.origin.x, y=orig_frame.origin.y, w=orig_frame.size.w + delta_size.w, h=orig_frame.size.h + delta_size.h) + new_frame = Rect( + x=orig_frame.origin.x, + y=orig_frame.origin.y, + w=orig_frame.size.w + + delta_size.w, + h=orig_frame.size.h + + delta_size.h) old_child_max_x = orig_frame.get_max_x() new_child_max_x = new_frame.get_max_x() window_max_x = frame.get_max_x() @@ -122,7 +138,7 @@ class Window(object): new_frame.origin.x -= new_child_max_x - window_max_x child.set_position(new_frame.origin) child.resize(new_frame.size) - + if adjust_neighbors: #print('orig_frame = %s\r\n' % (str(orig_frame)), end='') for curr_child in self.children: @@ -131,32 +147,33 @@ class Window(object): curr_child_frame = curr_child.get_frame() if delta_size.w != 0: #print('curr_child_frame = %s\r\n' % (str(curr_child_frame)), end='') - if curr_child_frame.get_min_x() == orig_frame.get_max_x(): + if curr_child_frame.get_min_x() == orig_frame.get_max_x(): curr_child_frame.origin.x += delta_size.w curr_child_frame.size.w -= delta_size.w #print('adjusted curr_child_frame = %s\r\n' % (str(curr_child_frame)), end='') - curr_child.resize (curr_child_frame.size) - curr_child.slide_position (Size(w=delta_size.w, h=0)) + curr_child.resize(curr_child_frame.size) + curr_child.slide_position( + Size(w=delta_size.w, h=0)) elif curr_child_frame.get_max_x() == orig_frame.get_min_x(): curr_child_frame.size.w -= delta_size.w #print('adjusted curr_child_frame = %s\r\n' % (str(curr_child_frame)), end='') - curr_child.resize (curr_child_frame.size) - + curr_child.resize(curr_child_frame.size) + def add_key_action(self, arg, callback, decription): if isinstance(arg, list): for key in arg: self.add_key_action(key, callback, description) else: if isinstance(arg, six.integer_types): - key_action_dict = { 'key' : arg, - 'callback' : callback, - 'description' : decription } + key_action_dict = {'key': arg, + 'callback': callback, + 'description': decription} self.key_actions[arg] = key_action_dict - elif isinstance(arg, basestring): + elif isinstance(arg, basestring): key_integer = ord(arg) - key_action_dict = { 'key' : key_integer, - 'callback' : callback, - 'description' : decription } + key_action_dict = {'key': key_integer, + 'callback': callback, + 'description': decription} self.key_actions[key_integer] = key_action_dict else: raise ValueError @@ -164,16 +181,16 @@ class Window(object): def draw_title_box(self, title): is_in_first_responder_chain = self.is_in_first_responder_chain() if is_in_first_responder_chain: - self.attron (curses.A_REVERSE) + self.attron(curses.A_REVERSE) self.box() if is_in_first_responder_chain: - self.attroff (curses.A_REVERSE) + self.attroff(curses.A_REVERSE) if title: self.addstr(Point(x=2, y=0), ' ' + title + ' ') - + def remove_child(self, window): self.children.remove(window) - + def get_first_responder(self): if len(self.first_responders): return self.first_responders[-1] @@ -182,12 +199,16 @@ class Window(object): def set_first_responder(self, window): if window.can_become_first_responder: - if six.callable(getattr(window, "hidden", None)) and window.hidden(): + if six.callable( + getattr( + window, + "hidden", + None)) and window.hidden(): return False - if not window in self.children: + if window not in self.children: self.add_child(window) # See if we have a current first responder, and if we do, let it know that - # it will be resigning as first responder + # it will be resigning as first responder first_responder = self.get_first_responder() if first_responder: first_responder.relinquish_first_responder() @@ -199,29 +220,31 @@ class Window(object): return True else: return False - + def push_first_responder(self, window): - # Only push the window as the new first responder if the window isn't already the first responder + # Only push the window as the new first responder if the window isn't + # already the first responder if window != self.get_first_responder(): self.first_responders.append(window) - - def pop_first_responder(self, window): - # Only pop the window from the first responder list if it is the first responder + + def pop_first_responder(self, window): + # Only pop the window from the first responder list if it is the first + # responder if window == self.get_first_responder(): old_first_responder = self.first_responders.pop() old_first_responder.relinquish_first_responder() return True else: return False - + def relinquish_first_responder(self): '''Override if there is something that you need to do when you lose first responder status.''' - pass - - # def resign_first_responder(self, remove_from_parent, new_first_responder): + pass + + # def resign_first_responder(self, remove_from_parent, new_first_responder): # success = False # if self.parent: - # if self.is_first_responder(): + # if self.is_first_responder(): # self.relinquish_first_responder() # if len(self.parent.first_responder): # self.parent.first_responder = None @@ -253,21 +276,21 @@ class Window(object): num_children = len(self.children) if num_children == 1: return self.set_first_responder(self.children[0]) - for (i,window) in enumerate(self.children): + for (i, window) in enumerate(self.children): if window.is_first_responder(): break if i < num_children: - for i in range(i+1,num_children): + for i in range(i + 1, num_children): if self.set_first_responder(self.children[i]): return True for i in range(0, i): if self.set_first_responder(self.children[i]): return True - + def point_in_window(self, pt): size = self.get_size() return pt.x >= 0 and pt.x < size.w and pt.y >= 0 and pt.y < size.h - + def addch(self, c): try: self.window.addch(c) @@ -291,6 +314,7 @@ class Window(object): self.window.addnstr(pt.y, pt.x, str, n) except: pass + def addnstr(self, str, n): try: self.window.addnstr(str, n) @@ -298,19 +322,26 @@ class Window(object): pass def attron(self, attr): - return self.window.attron (attr) + return self.window.attron(attr) def attroff(self, attr): - return self.window.attroff (attr) + return self.window.attroff(attr) def box(self, vertch=0, horch=0): if vertch == 0: vertch = curses.ACS_VLINE - if horch == 0: + if horch == 0: horch = curses.ACS_HLINE self.window.box(vertch, horch) - def get_contained_rect(self, top_inset=0, bottom_inset=0, left_inset=0, right_inset=0, height=-1, width=-1): + def get_contained_rect( + self, + top_inset=0, + bottom_inset=0, + left_inset=0, + right_inset=0, + height=-1, + width=-1): '''Get a rectangle based on the top "height" lines of this window''' rect = self.get_frame() x = rect.origin.x + left_inset @@ -323,11 +354,11 @@ class Window(object): w = rect.size.w - (left_inset + right_inset) else: w = width - return Rect (x = x, y = y, w = w, h = h) + return Rect(x=x, y=y, w=w, h=h) def erase(self): self.window.erase() - + def get_cursor(self): (y, x) = self.window.getyx() return Point(x=x, y=y) @@ -341,11 +372,11 @@ class Window(object): position = self.get_position_in_parent() size = self.get_size() return Rect(x=position.x, y=position.y, w=size.w, h=size.h) - + def get_position_in_parent(self): (y, x) = self.window.getparyx() return Point(x, y) - + def get_position(self): (y, x) = self.window.getbegyx() return Point(x, y) @@ -362,22 +393,22 @@ class Window(object): curses.panel.update_panels() self.move(Point(x=0, y=0)) return self.window.refresh() - + def resize(self, size): return self.window.resize(size.h, size.w) - + def timeout(self, timeout_msec): return self.window.timeout(timeout_msec) def handle_key(self, key, check_parent=True): '''Handle a key press in this window.''' - + # First try the first responder if this window has one, but don't allow # it to check with its parent (False second parameter) so we don't recurse # and get a stack overflow for first_responder in reversed(self.first_responders): if first_responder.handle_key(key, False): - return True + return True # Check our key map to see if we have any actions. Actions don't take # any arguments, they must be callable @@ -391,23 +422,23 @@ class Window(object): key_action['callback']() return True # Check if the window delegate wants to handle this key press - if self.delegate: + if self.delegate: if six.callable(getattr(self.delegate, "handle_key", None)): if self.delegate.handle_key(self, key): return True if self.delegate(self, key): return True - # Check if we have a parent window and if so, let the parent + # Check if we have a parent window and if so, let the parent # window handle the key press if check_parent and self.parent: return self.parent.handle_key(key, True) else: - return False # Key not handled + return False # Key not handled def update(self): for child in self.children: child.update() - + def quit_action(self): raise QuitException @@ -425,17 +456,17 @@ class Window(object): else: c = c << 8 | escape_key self.timeout(timeout_msec) - return c - + return c + def key_event_loop(self, timeout_msec=-1, n=sys.maxsize): '''Run an event loop to receive key presses and pass them along to the responder chain. - + timeout_msec is the timeout it milliseconds. If the value is -1, an infinite wait will be used. It the value is zero, a non-blocking mode will be used, and if greater than zero it will wait for a key press for timeout_msec milliseconds. - + n is the number of times to go through the event loop before exiting''' done = False while not done and n > 0: @@ -447,10 +478,21 @@ class Window(object): done = True n -= 1 + class Panel(Window): - def __init__(self, frame, delegate = None, can_become_first_responder = True): - window = curses.newwin(frame.size.h,frame.size.w, frame.origin.y, frame.origin.x) - super(Panel, self).__init__(window, delegate, can_become_first_responder) + + def __init__(self, frame, delegate=None, can_become_first_responder=True): + window = curses.newwin( + frame.size.h, + frame.size.w, + frame.origin.y, + frame.origin.x) + super( + Panel, + self).__init__( + window, + delegate, + can_become_first_responder) self.panel = curses.panel.new_panel(window) def hide(self): @@ -464,31 +506,63 @@ class Panel(Window): def top(self): return self.panel.top() - + def set_position(self, pt): self.panel.move(pt.y, pt.x) - + def slide_position(self, size): new_position = self.get_position() new_position.x = new_position.x + size.w new_position.y = new_position.y + size.h self.set_position(new_position) + class BoxedPanel(Panel): - def __init__(self, frame, title, delegate = None, can_become_first_responder = True): - super(BoxedPanel, self).__init__(frame, delegate, can_become_first_responder) + + def __init__(self, frame, title, delegate=None, + can_become_first_responder=True): + super( + BoxedPanel, + self).__init__( + frame, + delegate, + can_become_first_responder) self.title = title self.lines = list() self.first_visible_idx = 0 self.selected_idx = -1 - self.add_key_action(curses.KEY_UP, self.select_prev, "Select the previous item") - self.add_key_action(curses.KEY_DOWN, self.select_next, "Select the next item") - self.add_key_action(curses.KEY_HOME, self.scroll_begin, "Go to the beginning of the list") - self.add_key_action(curses.KEY_END, self.scroll_end, "Go to the end of the list") - self.add_key_action(0x1b4f48, self.scroll_begin, "Go to the beginning of the list") - self.add_key_action(0x1b4f46, self.scroll_end, "Go to the end of the list") - self.add_key_action(curses.KEY_PPAGE, self.scroll_page_backward, "Scroll to previous page") - self.add_key_action(curses.KEY_NPAGE, self.scroll_page_forward, "Scroll to next forward") + self.add_key_action( + curses.KEY_UP, + self.select_prev, + "Select the previous item") + self.add_key_action( + curses.KEY_DOWN, + self.select_next, + "Select the next item") + self.add_key_action( + curses.KEY_HOME, + self.scroll_begin, + "Go to the beginning of the list") + self.add_key_action( + curses.KEY_END, + self.scroll_end, + "Go to the end of the list") + self.add_key_action( + 0x1b4f48, + self.scroll_begin, + "Go to the beginning of the list") + self.add_key_action( + 0x1b4f46, + self.scroll_end, + "Go to the end of the list") + self.add_key_action( + curses.KEY_PPAGE, + self.scroll_page_backward, + "Scroll to previous page") + self.add_key_action( + curses.KEY_NPAGE, + self.scroll_page_forward, + "Scroll to next forward") self.update() def clear(self, update=True): @@ -497,21 +571,21 @@ class BoxedPanel(Panel): self.selected_idx = -1 if update: self.update() - + def get_usable_width(self): - '''Valid usable width is 0 to (width - 3) since the left and right lines display the box around + '''Valid usable width is 0 to (width - 3) since the left and right lines display the box around this frame and we skip a leading space''' w = self.get_size().w if w > 3: - return w-3 + return w - 3 else: return 0 - + def get_usable_height(self): '''Valid line indexes are 0 to (height - 2) since the top and bottom lines display the box around this frame.''' h = self.get_size().h if h > 2: - return h-2 + return h - 2 else: return 0 @@ -520,24 +594,25 @@ class BoxedPanel(Panel): line_idx = global_line_idx - self.first_visible_idx num_lines = self.get_usable_height() if line_idx < num_lines: - return Point(x=2, y=1+line_idx) + return Point(x=2, y=1 + line_idx) else: - return Point(x=-1, y=-1) # return an invalid coordinate if the line index isn't valid - - def set_title (self, title, update=True): + # return an invalid coordinate if the line index isn't valid + return Point(x=-1, y=-1) + + def set_title(self, title, update=True): self.title = title if update: self.update() - def scroll_to_line (self, idx): + def scroll_to_line(self, idx): if idx < len(self.lines): self.selected_idx = idx - max_visible_lines = self.get_usable_height() + max_visible_lines = self.get_usable_height() if idx < self.first_visible_idx or idx >= self.first_visible_idx + max_visible_lines: self.first_visible_idx = idx self.refresh() - def scroll_begin (self): + def scroll_begin(self): self.first_visible_idx = 0 if len(self.lines) > 0: self.selected_idx = 0 @@ -545,39 +620,39 @@ class BoxedPanel(Panel): self.selected_idx = -1 self.update() - def scroll_end (self): + def scroll_end(self): max_visible_lines = self.get_usable_height() num_lines = len(self.lines) if num_lines > max_visible_lines: self.first_visible_idx = num_lines - max_visible_lines else: self.first_visible_idx = 0 - self.selected_idx = num_lines-1 + self.selected_idx = num_lines - 1 self.update() - + def scroll_page_backward(self): - num_lines = len(self.lines) - max_visible_lines = self.get_usable_height() + num_lines = len(self.lines) + max_visible_lines = self.get_usable_height() new_index = self.first_visible_idx - max_visible_lines if new_index < 0: self.first_visible_idx = 0 else: self.first_visible_idx = new_index self.refresh() - + def scroll_page_forward(self): - max_visible_lines = self.get_usable_height() + max_visible_lines = self.get_usable_height() self.first_visible_idx += max_visible_lines self._adjust_first_visible_line() self.refresh() - def select_next (self): + def select_next(self): self.selected_idx += 1 if self.selected_idx >= len(self.lines): self.selected_idx = len(self.lines) - 1 self.refresh() - - def select_prev (self): + + def select_prev(self): self.selected_idx -= 1 if self.selected_idx < 0: if len(self.lines) > 0: @@ -588,13 +663,14 @@ class BoxedPanel(Panel): def get_selected_idx(self): return self.selected_idx - + def _adjust_first_visible_line(self): num_lines = len(self.lines) max_visible_lines = self.get_usable_height() - if (self.first_visible_idx >= num_lines) or (num_lines - self.first_visible_idx) > max_visible_lines: + if (self.first_visible_idx >= num_lines) or ( + num_lines - self.first_visible_idx) > max_visible_lines: self.first_visible_idx = num_lines - max_visible_lines - + def append_line(self, s, update=True): self.lines.append(s) self._adjust_first_visible_line() @@ -611,9 +687,9 @@ class BoxedPanel(Panel): self._adjust_first_visible_line() if update: self.update() - + def update(self): - self.erase() + self.erase() self.draw_title_box(self.title) max_width = self.get_usable_width() for line_idx in range(self.first_visible_idx, len(self.lines)): @@ -621,11 +697,11 @@ class BoxedPanel(Panel): if pt.is_valid_coordinate(): is_selected = line_idx == self.selected_idx if is_selected: - self.attron (curses.A_REVERSE) + self.attron(curses.A_REVERSE) self.move(pt) self.addnstr(self.lines[line_idx], max_width) if is_selected: - self.attroff (curses.A_REVERSE) + self.attroff(curses.A_REVERSE) else: return @@ -636,16 +712,19 @@ class BoxedPanel(Panel): for (idx, line) in enumerate(self.lines): # Remove any tabs from lines since they hose up the display if "\t" in line: - self.lines[idx] = (8*' ').join(line.split('\t')) + self.lines[idx] = (8 * ' ').join(line.split('\t')) self.selected_idx = 0 self.first_visible_idx = 0 self.refresh() + class Item(object): + def __init__(self, title, action): self.title = title self.action = action + class TreeItemDelegate(object): def might_have_children(self): @@ -663,21 +742,29 @@ class TreeItemDelegate(object): def draw_item(self, tree_window, item): self.draw_item_string(tree_window, item, item.title) - + def do_action(self): pass + class TreeItem(object): - def __init__(self, delegate, parent = None, title = None, action = None, is_expanded = False): + + def __init__( + self, + delegate, + parent=None, + title=None, + action=None, + is_expanded=False): self.parent = parent self.title = title - self.action = action + self.action = action self.delegate = delegate - self.is_expanded = not parent or is_expanded == True + self.is_expanded = not parent or is_expanded self._might_have_children = None self.children = None self._children_might_have_children = False - + def get_children(self): if self.is_expanded and self.might_have_children(): if self.children is None: @@ -691,7 +778,7 @@ class TreeItem(object): self._children_might_have_children = False self.children = None return self.children - + def append_visible_items(self, items): items.append(self) children = self.get_children() @@ -705,7 +792,8 @@ class TreeItem(object): # Root item always might have children self._might_have_children = True else: - # Check with the delegate to see if the item might have children + # Check with the delegate to see if the item might have + # children self._might_have_children = self.delegate.might_have_children() return self._might_have_children @@ -717,10 +805,10 @@ class TreeItem(object): self.children = self.delegate.update_children(self) for child in self.children: child.update_children() - else: + else: self.children = None return self.children - + def get_num_visible_rows(self): rows = 1 if self.is_expanded: @@ -729,6 +817,7 @@ class TreeItem(object): for child in children: rows += child.get_num_visible_rows() return rows + def draw(self, tree_window, row): display_row = tree_window.get_display_row(row) if display_row >= 0: @@ -736,79 +825,112 @@ class TreeItem(object): if self.parent: self.parent.draw_tree_for_child(tree_window, self, 0) if self.might_have_children(): - tree_window.addch (curses.ACS_DIAMOND) - tree_window.addch (curses.ACS_HLINE) + tree_window.addch(curses.ACS_DIAMOND) + tree_window.addch(curses.ACS_HLINE) elif self.parent and self.parent.children_might_have_children(): if self.parent.parent: - tree_window.addch (curses.ACS_HLINE) - tree_window.addch (curses.ACS_HLINE) + tree_window.addch(curses.ACS_HLINE) + tree_window.addch(curses.ACS_HLINE) else: - tree_window.addch (' ') - tree_window.addch (' ') + tree_window.addch(' ') + tree_window.addch(' ') is_selected = tree_window.is_selected(row) if is_selected: - tree_window.attron (curses.A_REVERSE) + tree_window.attron(curses.A_REVERSE) self.delegate.draw_item(tree_window, self) if is_selected: - tree_window.attroff (curses.A_REVERSE) - - def draw_tree_for_child (self, tree_window, child, reverse_depth): + tree_window.attroff(curses.A_REVERSE) + + def draw_tree_for_child(self, tree_window, child, reverse_depth): if self.parent: - self.parent.draw_tree_for_child (tree_window, self, reverse_depth + 1) + self.parent.draw_tree_for_child( + tree_window, self, reverse_depth + 1) if self.children[-1] == child: # Last child if reverse_depth == 0: - tree_window.addch (curses.ACS_LLCORNER) - tree_window.addch (curses.ACS_HLINE) + tree_window.addch(curses.ACS_LLCORNER) + tree_window.addch(curses.ACS_HLINE) else: - tree_window.addch (' ') - tree_window.addch (' ') + tree_window.addch(' ') + tree_window.addch(' ') else: # Middle child if reverse_depth == 0: - tree_window.addch (curses.ACS_LTEE) - tree_window.addch (curses.ACS_HLINE) + tree_window.addch(curses.ACS_LTEE) + tree_window.addch(curses.ACS_HLINE) else: - tree_window.addch (curses.ACS_VLINE) - tree_window.addch (' ') - - def was_selected(self): + tree_window.addch(curses.ACS_VLINE) + tree_window.addch(' ') + + def was_selected(self): self.delegate.do_action() - + + class TreePanel(Panel): + def __init__(self, frame, title, root_item): self.root_item = root_item self.title = title self.first_visible_idx = 0 self.selected_idx = 0 self.items = None - super(TreePanel, self).__init__(frame) - self.add_key_action(curses.KEY_UP, self.select_prev, "Select the previous item") - self.add_key_action(curses.KEY_DOWN, self.select_next, "Select the next item") - self.add_key_action(curses.KEY_RIGHT,self.right_arrow, "Expand an item") - self.add_key_action(curses.KEY_LEFT, self.left_arrow, "Unexpand an item or navigate to parent") - self.add_key_action(curses.KEY_HOME, self.scroll_begin, "Go to the beginning of the tree") - self.add_key_action(curses.KEY_END, self.scroll_end, "Go to the end of the tree") - self.add_key_action(0x1b4f48, self.scroll_begin, "Go to the beginning of the tree") - self.add_key_action(0x1b4f46, self.scroll_end, "Go to the end of the tree") - self.add_key_action(curses.KEY_PPAGE, self.scroll_page_backward, "Scroll to previous page") - self.add_key_action(curses.KEY_NPAGE, self.scroll_page_forward, "Scroll to next forward") + super(TreePanel, self).__init__(frame) + self.add_key_action( + curses.KEY_UP, + self.select_prev, + "Select the previous item") + self.add_key_action( + curses.KEY_DOWN, + self.select_next, + "Select the next item") + self.add_key_action( + curses.KEY_RIGHT, + self.right_arrow, + "Expand an item") + self.add_key_action( + curses.KEY_LEFT, + self.left_arrow, + "Unexpand an item or navigate to parent") + self.add_key_action( + curses.KEY_HOME, + self.scroll_begin, + "Go to the beginning of the tree") + self.add_key_action( + curses.KEY_END, + self.scroll_end, + "Go to the end of the tree") + self.add_key_action( + 0x1b4f48, + self.scroll_begin, + "Go to the beginning of the tree") + self.add_key_action( + 0x1b4f46, + self.scroll_end, + "Go to the end of the tree") + self.add_key_action( + curses.KEY_PPAGE, + self.scroll_page_backward, + "Scroll to previous page") + self.add_key_action( + curses.KEY_NPAGE, + self.scroll_page_forward, + "Scroll to next forward") def get_selected_item(self): if self.selected_idx < len(self.items): return self.items[self.selected_idx] else: return None - + def select_item(self, item): if self.items and item in self.items: self.selected_idx = self.items.index(item) return True else: return False - + def get_visible_items(self): - # Clear self.items when you want to update all chidren + # Clear self.items when you want to update all chidren if self.items is None: self.items = list() children = self.root_item.get_children() @@ -816,15 +938,15 @@ class TreePanel(Panel): for child in children: child.append_visible_items(self.items) return self.items - + def update(self): - self.erase() - self.draw_title_box(self.title) + self.erase() + self.draw_title_box(self.title) visible_items = self.get_visible_items() for (row, child) in enumerate(visible_items): - child.draw(self, row) + child.draw(self, row) - def get_item_draw_point(self, row): + def get_item_draw_point(self, row): display_row = self.get_display_row(row) if display_row >= 0: return Point(2, display_row + 1) @@ -834,28 +956,29 @@ class TreePanel(Panel): def get_display_row(self, row): if row >= self.first_visible_idx: display_row = row - self.first_visible_idx - if display_row < self.get_size().h-2: - return display_row + if display_row < self.get_size().h - 2: + return display_row return -1 def is_selected(self, row): return row == self.selected_idx - def get_num_lines(self): + def get_num_lines(self): self.get_visible_items() return len(self.items) - + def get_num_visible_lines(self): - return self.get_size().h-2 - def select_next (self): - self.selected_idx += 1 + return self.get_size().h - 2 + + def select_next(self): + self.selected_idx += 1 num_lines = self.get_num_lines() if self.selected_idx >= num_lines: - self.selected_idx = num_lines - 1 + self.selected_idx = num_lines - 1 self._selection_changed() self.refresh() - def select_prev (self): + def select_prev(self): self.selected_idx -= 1 if self.selected_idx < 0: num_lines = self.get_num_lines() @@ -866,7 +989,7 @@ class TreePanel(Panel): self._selection_changed() self.refresh() - def scroll_begin (self): + def scroll_begin(self): self.first_visible_idx = 0 num_lines = self.get_num_lines() if num_lines > 0: @@ -874,36 +997,35 @@ class TreePanel(Panel): else: self.selected_idx = -1 self.refresh() - + def redisplay_tree(self): self.items = None self.refresh() - def right_arrow(self): + def right_arrow(self): selected_item = self.get_selected_item() if selected_item and selected_item.is_expanded == False: selected_item.is_expanded = True self.redisplay_tree() - + def left_arrow(self): selected_item = self.get_selected_item() if selected_item: - if selected_item.is_expanded == True: + if selected_item.is_expanded: selected_item.is_expanded = False self.redisplay_tree() elif selected_item.parent: if self.select_item(selected_item.parent): self.refresh() - - def scroll_end (self): + def scroll_end(self): num_visible_lines = self.get_num_visible_lines() num_lines = self.get_num_lines() if num_lines > num_visible_lines: self.first_visible_idx = num_lines - num_visible_lines else: self.first_visible_idx = 0 - self.selected_idx = num_lines-1 + self.selected_idx = num_lines - 1 self.refresh() def scroll_page_backward(self): @@ -912,7 +1034,7 @@ class TreePanel(Panel): if new_index < 0: self.selected_idx = 0 else: - self.selected_idx = new_index + self.selected_idx = new_index self._selection_changed() self.refresh() @@ -926,59 +1048,87 @@ class TreePanel(Panel): self._selection_changed() self.refresh() - def _selection_changed(self): - num_lines = self.get_num_lines() + def _selection_changed(self): + num_lines = self.get_num_lines() num_visible_lines = self.get_num_visible_lines() last_visible_index = self.first_visible_idx + num_visible_lines if self.selected_idx >= last_visible_index: - self.first_visible_idx += (self.selected_idx - last_visible_index + 1) + self.first_visible_idx += (self.selected_idx - + last_visible_index + 1) if self.selected_idx < self.first_visible_idx: - self.first_visible_idx = self.selected_idx + self.first_visible_idx = self.selected_idx if self.selected_idx >= 0 and self.selected_idx < len(self.items): item = self.items[self.selected_idx] item.was_selected() class Menu(BoxedPanel): + def __init__(self, title, items): max_title_width = 0 for item in items: if max_title_width < len(item.title): max_title_width = len(item.title) - frame = Rect(x=0, y=0, w=max_title_width+4, h=len(items)+2) - super(Menu, self).__init__(frame, title=None, delegate=None, can_become_first_responder=True) + frame = Rect(x=0, y=0, w=max_title_width + 4, h=len(items) + 2) + super( + Menu, + self).__init__( + frame, + title=None, + delegate=None, + can_become_first_responder=True) self.selected_idx = 0 self.title = title self.items = items for (item_idx, item) in enumerate(items): self.set_line(item_idx, item.title) self.hide() - + def update(self): super(Menu, self).update() def relinquish_first_responder(self): if not self.hidden(): - self.hide() - - def perform_action(self): + self.hide() + + def perform_action(self): selected_idx = self.get_selected_idx() if selected_idx < len(self.items): action = self.items[selected_idx].action if action: action() - + + class MenuBar(Panel): + def __init__(self, frame): super(MenuBar, self).__init__(frame, can_become_first_responder=True) self.menus = list() self.selected_menu_idx = -1 - self.add_key_action(curses.KEY_LEFT, self.select_prev, "Select the previous menu") - self.add_key_action(curses.KEY_RIGHT, self.select_next, "Select the next menu") - self.add_key_action(curses.KEY_DOWN, lambda: self.select(0), "Select the first menu") - self.add_key_action(27, self.relinquish_first_responder, "Hide current menu") - self.add_key_action(curses.KEY_ENTER, self.perform_action, "Select the next menu item") - self.add_key_action(10, self.perform_action, "Select the next menu item") + self.add_key_action( + curses.KEY_LEFT, + self.select_prev, + "Select the previous menu") + self.add_key_action( + curses.KEY_RIGHT, + self.select_next, + "Select the next menu") + self.add_key_action( + curses.KEY_DOWN, + lambda: self.select(0), + "Select the first menu") + self.add_key_action( + 27, + self.relinquish_first_responder, + "Hide current menu") + self.add_key_action( + curses.KEY_ENTER, + self.perform_action, + "Select the next menu item") + self.add_key_action( + 10, + self.perform_action, + "Select the next menu item") def insert_menu(self, menu, index=sys.maxsize): if index >= len(self.menus): @@ -989,8 +1139,8 @@ class MenuBar(Panel): for menu in self.menus: menu.set_position(pt) pt.x += len(menu.title) + 5 - - def perform_action(self): + + def perform_action(self): '''If no menu is visible, show the first menu. If a menu is visible, perform the action associated with the selected menu item in the menu''' menu_visible = False @@ -1004,22 +1154,22 @@ class MenuBar(Panel): self._selected_menu_changed() else: self.select(0) - + def relinquish_first_responder(self): if self.selected_menu_idx >= 0: self.selected_menu_idx = -1 self._selected_menu_changed() - + def _selected_menu_changed(self): for (menu_idx, menu) in enumerate(self.menus): is_hidden = menu.hidden() - if menu_idx != self.selected_menu_idx: + if menu_idx != self.selected_menu_idx: if not is_hidden: if self.parent.pop_first_responder(menu) == False: menu.hide() for (menu_idx, menu) in enumerate(self.menus): is_hidden = menu.hidden() - if menu_idx == self.selected_menu_idx: + if menu_idx == self.selected_menu_idx: if is_hidden: menu.show() self.parent.push_first_responder(menu) @@ -1030,8 +1180,8 @@ class MenuBar(Panel): if index < len(self.menus): self.selected_menu_idx = index self._selected_menu_changed() - - def select_next (self): + + def select_next(self): num_menus = len(self.menus) if self.selected_menu_idx == -1: if num_menus > 0: @@ -1044,7 +1194,7 @@ class MenuBar(Panel): self.selected_menu_idx = -1 self._selected_menu_changed() - def select_prev (self): + def select_prev(self): num_menus = len(self.menus) if self.selected_menu_idx == -1: if num_menus > 0: @@ -1061,41 +1211,47 @@ class MenuBar(Panel): self.erase() is_in_first_responder_chain = self.is_in_first_responder_chain() if is_in_first_responder_chain: - self.attron (curses.A_REVERSE) + self.attron(curses.A_REVERSE) pt = Point(x=0, y=0) - for menu in self.menus: + for menu in self.menus: self.addstr(pt, '| ' + menu.title + ' ') pt.x += len(menu.title) + 5 - self.addstr(pt, '|') + self.addstr(pt, '|') width = self.get_size().w while pt.x < width: self.addch_at_point(pt, ' ') - pt.x += 1 + pt.x += 1 if is_in_first_responder_chain: - self.attroff (curses.A_REVERSE) + self.attroff(curses.A_REVERSE) for menu in self.menus: menu.update() - - + + class StatusPanel(Panel): + def __init__(self, frame): - super(StatusPanel, self).__init__(frame, delegate=None, can_become_first_responder=False) + super( + StatusPanel, + self).__init__( + frame, + delegate=None, + can_become_first_responder=False) self.status_items = list() self.status_dicts = dict() self.next_status_x = 1 - + def add_status_item(self, name, title, format, width, value, update=True): - status_item_dict = { 'name': name, - 'title' : title, - 'width' : width, - 'format' : format, - 'value' : value, - 'x' : self.next_status_x } + status_item_dict = {'name': name, + 'title': title, + 'width': width, + 'format': format, + 'value': value, + 'x': self.next_status_x} index = len(self.status_items) self.status_items.append(status_item_dict) self.status_dicts[name] = index - self.next_status_x += width + 2; + self.next_status_x += width + 2 if update: self.update() @@ -1106,7 +1262,7 @@ class StatusPanel(Panel): status_item_dict['value'] = status_item_dict['value'] + 1 if update: self.update() - + def update_status(self, name, value, update=True): if name in self.status_dicts: status_item_idx = self.status_dicts[name] @@ -1114,13 +1270,22 @@ class StatusPanel(Panel): status_item_dict['value'] = status_item_dict['format'] % (value) if update: self.update() + def update(self): - self.erase(); + self.erase() for status_item_dict in self.status_items: - self.addnstr_at_point(Point(x=status_item_dict['x'], y=0), '%s: %s' % (status_item_dict['title'], status_item_dict['value']), status_item_dict['width']) + self.addnstr_at_point( + Point( + x=status_item_dict['x'], + y=0), + '%s: %s' % + (status_item_dict['title'], + status_item_dict['value']), + status_item_dict['width']) stdscr = None + def intialize_curses(): global stdscr stdscr = curses.initscr() @@ -1133,6 +1298,7 @@ def intialize_curses(): pass return Window(stdscr) + def terminate_curses(): global stdscr if stdscr: @@ -1140,4 +1306,3 @@ def terminate_curses(): curses.echo() curses.nocbreak() curses.endwin() - |