| # This file is part of pybootchartgui. |
| |
| # pybootchartgui is free software: you can redistribute it and/or modify |
| # it under the terms of the GNU General Public License as published by |
| # the Free Software Foundation, either version 3 of the License, or |
| # (at your option) any later version. |
| |
| # pybootchartgui is distributed in the hope that it will be useful, |
| # but WITHOUT ANY WARRANTY; without even the implied warranty of |
| # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| # GNU General Public License for more details. |
| |
| # You should have received a copy of the GNU General Public License |
| # along with pybootchartgui. If not, see <http://www.gnu.org/licenses/>. |
| |
| import gobject |
| import gtk |
| import gtk.gdk |
| import gtk.keysyms |
| from . import draw |
| from .draw import RenderOptions |
| |
| class PyBootchartWidget(gtk.DrawingArea): |
| __gsignals__ = { |
| 'expose-event': 'override', |
| 'clicked' : (gobject.SIGNAL_RUN_LAST, gobject.TYPE_NONE, (gobject.TYPE_STRING, gtk.gdk.Event)), |
| 'position-changed' : (gobject.SIGNAL_RUN_LAST, gobject.TYPE_NONE, (gobject.TYPE_INT, gobject.TYPE_INT)), |
| 'set-scroll-adjustments' : (gobject.SIGNAL_RUN_LAST, gobject.TYPE_NONE, (gtk.Adjustment, gtk.Adjustment)) |
| } |
| |
| def __init__(self, trace, options, xscale): |
| gtk.DrawingArea.__init__(self) |
| |
| self.trace = trace |
| self.options = options |
| |
| self.set_flags(gtk.CAN_FOCUS) |
| |
| self.add_events(gtk.gdk.BUTTON_PRESS_MASK | gtk.gdk.BUTTON_RELEASE_MASK) |
| self.connect("button-press-event", self.on_area_button_press) |
| self.connect("button-release-event", self.on_area_button_release) |
| self.add_events(gtk.gdk.POINTER_MOTION_MASK | gtk.gdk.POINTER_MOTION_HINT_MASK | gtk.gdk.BUTTON_RELEASE_MASK) |
| self.connect("motion-notify-event", self.on_area_motion_notify) |
| self.connect("scroll-event", self.on_area_scroll_event) |
| self.connect('key-press-event', self.on_key_press_event) |
| |
| self.connect('set-scroll-adjustments', self.on_set_scroll_adjustments) |
| self.connect("size-allocate", self.on_allocation_size_changed) |
| self.connect("position-changed", self.on_position_changed) |
| |
| self.zoom_ratio = 1.0 |
| self.xscale = xscale |
| self.x, self.y = 0.0, 0.0 |
| |
| self.chart_width, self.chart_height = draw.extents(self.options, self.xscale, self.trace) |
| self.hadj = None |
| self.vadj = None |
| self.hadj_changed_signal_id = None |
| self.vadj_changed_signal_id = None |
| |
| def do_expose_event(self, event): |
| cr = self.window.cairo_create() |
| |
| # set a clip region for the expose event |
| cr.rectangle( |
| event.area.x, event.area.y, |
| event.area.width, event.area.height |
| ) |
| cr.clip() |
| self.draw(cr, self.get_allocation()) |
| return False |
| |
| def draw(self, cr, rect): |
| cr.set_source_rgba(1.0, 1.0, 1.0, 1.0) |
| cr.paint() |
| cr.scale(self.zoom_ratio, self.zoom_ratio) |
| cr.translate(-self.x, -self.y) |
| draw.render(cr, self.options, self.xscale, self.trace) |
| |
| def position_changed(self): |
| self.emit("position-changed", self.x, self.y) |
| |
| ZOOM_INCREMENT = 1.25 |
| |
| def zoom_image (self, zoom_ratio): |
| self.zoom_ratio = zoom_ratio |
| self._set_scroll_adjustments (self.hadj, self.vadj) |
| self.queue_draw() |
| |
| def zoom_to_rect (self, rect): |
| zoom_ratio = float(rect.width)/float(self.chart_width) |
| self.zoom_image(zoom_ratio) |
| self.x = 0 |
| self.position_changed() |
| |
| def set_xscale(self, xscale): |
| old_mid_x = self.x + self.hadj.page_size / 2 |
| self.xscale = xscale |
| self.chart_width, self.chart_height = draw.extents(self.options, self.xscale, self.trace) |
| new_x = old_mid_x |
| self.zoom_image (self.zoom_ratio) |
| |
| def on_expand(self, action): |
| self.set_xscale (int(self.xscale * 1.5 + 0.5)) |
| |
| def on_contract(self, action): |
| self.set_xscale (max(int(self.xscale / 1.5), 1)) |
| |
| def on_zoom_in(self, action): |
| self.zoom_image(self.zoom_ratio * self.ZOOM_INCREMENT) |
| |
| def on_zoom_out(self, action): |
| self.zoom_image(self.zoom_ratio / self.ZOOM_INCREMENT) |
| |
| def on_zoom_fit(self, action): |
| self.zoom_to_rect(self.get_allocation()) |
| |
| def on_zoom_100(self, action): |
| self.zoom_image(1.0) |
| self.set_xscale(1.0) |
| |
| def show_toggled(self, button): |
| self.options.app_options.show_all = button.get_property ('active') |
| self.chart_width, self.chart_height = draw.extents(self.options, self.xscale, self.trace) |
| self._set_scroll_adjustments(self.hadj, self.vadj) |
| self.queue_draw() |
| |
| POS_INCREMENT = 100 |
| |
| def on_key_press_event(self, widget, event): |
| if event.keyval == gtk.keysyms.Left: |
| self.x -= self.POS_INCREMENT/self.zoom_ratio |
| elif event.keyval == gtk.keysyms.Right: |
| self.x += self.POS_INCREMENT/self.zoom_ratio |
| elif event.keyval == gtk.keysyms.Up: |
| self.y -= self.POS_INCREMENT/self.zoom_ratio |
| elif event.keyval == gtk.keysyms.Down: |
| self.y += self.POS_INCREMENT/self.zoom_ratio |
| else: |
| return False |
| self.queue_draw() |
| self.position_changed() |
| return True |
| |
| def on_area_button_press(self, area, event): |
| if event.button == 2 or event.button == 1: |
| area.window.set_cursor(gtk.gdk.Cursor(gtk.gdk.FLEUR)) |
| self.prevmousex = event.x |
| self.prevmousey = event.y |
| if event.type not in (gtk.gdk.BUTTON_PRESS, gtk.gdk.BUTTON_RELEASE): |
| return False |
| return False |
| |
| def on_area_button_release(self, area, event): |
| if event.button == 2 or event.button == 1: |
| area.window.set_cursor(gtk.gdk.Cursor(gtk.gdk.ARROW)) |
| self.prevmousex = None |
| self.prevmousey = None |
| return True |
| return False |
| |
| def on_area_scroll_event(self, area, event): |
| if event.state & gtk.gdk.CONTROL_MASK: |
| if event.direction == gtk.gdk.SCROLL_UP: |
| self.zoom_image(self.zoom_ratio * self.ZOOM_INCREMENT) |
| return True |
| if event.direction == gtk.gdk.SCROLL_DOWN: |
| self.zoom_image(self.zoom_ratio / self.ZOOM_INCREMENT) |
| return True |
| return False |
| |
| def on_area_motion_notify(self, area, event): |
| state = event.state |
| if state & gtk.gdk.BUTTON2_MASK or state & gtk.gdk.BUTTON1_MASK: |
| x, y = int(event.x), int(event.y) |
| # pan the image |
| self.x += (self.prevmousex - x)/self.zoom_ratio |
| self.y += (self.prevmousey - y)/self.zoom_ratio |
| self.queue_draw() |
| self.prevmousex = x |
| self.prevmousey = y |
| self.position_changed() |
| return True |
| |
| def on_set_scroll_adjustments(self, area, hadj, vadj): |
| self._set_scroll_adjustments (hadj, vadj) |
| |
| def on_allocation_size_changed(self, widget, allocation): |
| self.hadj.page_size = allocation.width |
| self.hadj.page_increment = allocation.width * 0.9 |
| self.vadj.page_size = allocation.height |
| self.vadj.page_increment = allocation.height * 0.9 |
| |
| def _set_adj_upper(self, adj, upper): |
| changed = False |
| value_changed = False |
| |
| if adj.upper != upper: |
| adj.upper = upper |
| changed = True |
| |
| max_value = max(0.0, upper - adj.page_size) |
| if adj.value > max_value: |
| adj.value = max_value |
| value_changed = True |
| |
| if changed: |
| adj.changed() |
| if value_changed: |
| adj.value_changed() |
| |
| def _set_scroll_adjustments(self, hadj, vadj): |
| if hadj == None: |
| hadj = gtk.Adjustment(0.0, 0.0, 0.0, 0.0, 0.0, 0.0) |
| if vadj == None: |
| vadj = gtk.Adjustment(0.0, 0.0, 0.0, 0.0, 0.0, 0.0) |
| |
| if self.hadj_changed_signal_id != None and \ |
| self.hadj != None and hadj != self.hadj: |
| self.hadj.disconnect (self.hadj_changed_signal_id) |
| if self.vadj_changed_signal_id != None and \ |
| self.vadj != None and vadj != self.vadj: |
| self.vadj.disconnect (self.vadj_changed_signal_id) |
| |
| if hadj != None: |
| self.hadj = hadj |
| self._set_adj_upper (self.hadj, self.zoom_ratio * self.chart_width) |
| self.hadj_changed_signal_id = self.hadj.connect('value-changed', self.on_adjustments_changed) |
| |
| if vadj != None: |
| self.vadj = vadj |
| self._set_adj_upper (self.vadj, self.zoom_ratio * self.chart_height) |
| self.vadj_changed_signal_id = self.vadj.connect('value-changed', self.on_adjustments_changed) |
| |
| def on_adjustments_changed(self, adj): |
| self.x = self.hadj.value / self.zoom_ratio |
| self.y = self.vadj.value / self.zoom_ratio |
| self.queue_draw() |
| |
| def on_position_changed(self, widget, x, y): |
| self.hadj.value = x * self.zoom_ratio |
| self.vadj.value = y * self.zoom_ratio |
| |
| PyBootchartWidget.set_set_scroll_adjustments_signal('set-scroll-adjustments') |
| |
| class PyBootchartShell(gtk.VBox): |
| ui = ''' |
| <ui> |
| <toolbar name="ToolBar"> |
| <toolitem action="Expand"/> |
| <toolitem action="Contract"/> |
| <separator/> |
| <toolitem action="ZoomIn"/> |
| <toolitem action="ZoomOut"/> |
| <toolitem action="ZoomFit"/> |
| <toolitem action="Zoom100"/> |
| </toolbar> |
| </ui> |
| ''' |
| def __init__(self, window, trace, options, xscale): |
| gtk.VBox.__init__(self) |
| |
| self.widget = PyBootchartWidget(trace, options, xscale) |
| |
| # Create a UIManager instance |
| uimanager = self.uimanager = gtk.UIManager() |
| |
| # Add the accelerator group to the toplevel window |
| accelgroup = uimanager.get_accel_group() |
| window.add_accel_group(accelgroup) |
| |
| # Create an ActionGroup |
| actiongroup = gtk.ActionGroup('Actions') |
| self.actiongroup = actiongroup |
| |
| # Create actions |
| actiongroup.add_actions(( |
| ('Expand', gtk.STOCK_ADD, None, None, None, self.widget.on_expand), |
| ('Contract', gtk.STOCK_REMOVE, None, None, None, self.widget.on_contract), |
| ('ZoomIn', gtk.STOCK_ZOOM_IN, None, None, None, self.widget.on_zoom_in), |
| ('ZoomOut', gtk.STOCK_ZOOM_OUT, None, None, None, self.widget.on_zoom_out), |
| ('ZoomFit', gtk.STOCK_ZOOM_FIT, 'Fit Width', None, None, self.widget.on_zoom_fit), |
| ('Zoom100', gtk.STOCK_ZOOM_100, None, None, None, self.widget.on_zoom_100), |
| )) |
| |
| # Add the actiongroup to the uimanager |
| uimanager.insert_action_group(actiongroup, 0) |
| |
| # Add a UI description |
| uimanager.add_ui_from_string(self.ui) |
| |
| # Scrolled window |
| scrolled = gtk.ScrolledWindow() |
| scrolled.add(self.widget) |
| |
| # toolbar / h-box |
| hbox = gtk.HBox(False, 8) |
| |
| # Create a Toolbar |
| toolbar = uimanager.get_widget('/ToolBar') |
| hbox.pack_start(toolbar, True, True) |
| |
| if not options.kernel_only: |
| # Misc. options |
| button = gtk.CheckButton("Show more") |
| button.connect ('toggled', self.widget.show_toggled) |
| button.set_active(options.app_options.show_all) |
| hbox.pack_start (button, False, True) |
| |
| self.pack_start(hbox, False) |
| self.pack_start(scrolled) |
| self.show_all() |
| |
| def grab_focus(self, window): |
| window.set_focus(self.widget) |
| |
| |
| class PyBootchartWindow(gtk.Window): |
| |
| def __init__(self, trace, app_options): |
| gtk.Window.__init__(self) |
| |
| window = self |
| window.set_title("Bootchart %s" % trace.filename) |
| window.set_default_size(750, 550) |
| |
| tab_page = gtk.Notebook() |
| tab_page.show() |
| window.add(tab_page) |
| |
| full_opts = RenderOptions(app_options) |
| full_tree = PyBootchartShell(window, trace, full_opts, 1.0) |
| tab_page.append_page (full_tree, gtk.Label("Full tree")) |
| |
| if trace.kernel is not None and len (trace.kernel) > 2: |
| kernel_opts = RenderOptions(app_options) |
| kernel_opts.cumulative = False |
| kernel_opts.charts = False |
| kernel_opts.kernel_only = True |
| kernel_tree = PyBootchartShell(window, trace, kernel_opts, 5.0) |
| tab_page.append_page (kernel_tree, gtk.Label("Kernel boot")) |
| |
| full_tree.grab_focus(self) |
| self.show() |
| |
| |
| def show(trace, options): |
| win = PyBootchartWindow(trace, options) |
| win.connect('destroy', gtk.main_quit) |
| gtk.main() |