SchedGui.py

Go to the documentation of this file.

# SchedGui.py - Python extension for perf script, basic GUI code for
#       traces drawing and overview.
#
# Copyright (C) 2010 by Frederic Weisbecker <fweisbec@gmail.com>
#
# This software is distributed under the terms of the GNU General
# Public License ("GPL") version 2 as published by the Free Software
# Foundation.


try:
    import wx
except ImportError:
    raise ImportError, "You need to install the wxpython lib for this script"


class RootFrame(wx.Frame):
    Y_OFFSET = 100
    RECT_HEIGHT = 100
    RECT_SPACE = 50
    EVENT_MARKING_WIDTH = 5

    def __init__(self, sched_tracer, title, parent = None, id = -1):
        wx.Frame.__init__(self, parent, id, title)

        (self.screen_width, self.screen_height) = wx.GetDisplaySize()
        self.screen_width -= 10
        self.screen_height -= 10
        self.zoom = 0.5
        self.scroll_scale = 20
        self.sched_tracer = sched_tracer
        self.sched_tracer.set_root_win(self)
        (self.ts_start, self.ts_end) = sched_tracer.interval()
        self.update_width_virtual()
        self.nr_rects = sched_tracer.nr_rectangles() + 1
        self.height_virtual = RootFrame.Y_OFFSET + (self.nr_rects * (RootFrame.RECT_HEIGHT + RootFrame.RECT_SPACE))

        # whole window panel
        self.panel = wx.Panel(self, size=(self.screen_width, self.screen_height))

        # scrollable container
        self.scroll = wx.ScrolledWindow(self.panel)
        self.scroll.SetScrollbars(self.scroll_scale, self.scroll_scale, self.width_virtual / self.scroll_scale, self.height_virtual / self.scroll_scale)
        self.scroll.EnableScrolling(True, True)
        self.scroll.SetFocus()

        # scrollable drawing area
        self.scroll_panel = wx.Panel(self.scroll, size=(self.screen_width - 15, self.screen_height / 2))
        self.scroll_panel.Bind(wx.EVT_PAINT, self.on_paint)
        self.scroll_panel.Bind(wx.EVT_KEY_DOWN, self.on_key_press)
        self.scroll_panel.Bind(wx.EVT_LEFT_DOWN, self.on_mouse_down)
        self.scroll.Bind(wx.EVT_PAINT, self.on_paint)
        self.scroll.Bind(wx.EVT_KEY_DOWN, self.on_key_press)
        self.scroll.Bind(wx.EVT_LEFT_DOWN, self.on_mouse_down)

        self.scroll.Fit()
        self.Fit()

        self.scroll_panel.SetDimensions(-1, -1, self.width_virtual, self.height_virtual, wx.SIZE_USE_EXISTING)

        self.txt = None

        self.Show(True)

    def us_to_px(self, val):
        return val / (10 ** 3) * self.zoom

    def px_to_us(self, val):
        return (val / self.zoom) * (10 ** 3)

    def scroll_start(self):
        (x, y) = self.scroll.GetViewStart()
        return (x * self.scroll_scale, y * self.scroll_scale)

    def scroll_start_us(self):
        (x, y) = self.scroll_start()
        return self.px_to_us(x)

    def paint_rectangle_zone(self, nr, color, top_color, start, end):
        offset_px = self.us_to_px(start - self.ts_start)
        width_px = self.us_to_px(end - self.ts_start)

        offset_py = RootFrame.Y_OFFSET + (nr * (RootFrame.RECT_HEIGHT + RootFrame.RECT_SPACE))
        width_py = RootFrame.RECT_HEIGHT

        dc = self.dc

        if top_color is not None:
            (r, g, b) = top_color
            top_color = wx.Colour(r, g, b)
            brush = wx.Brush(top_color, wx.SOLID)
            dc.SetBrush(brush)
            dc.DrawRectangle(offset_px, offset_py, width_px, RootFrame.EVENT_MARKING_WIDTH)
            width_py -= RootFrame.EVENT_MARKING_WIDTH
            offset_py += RootFrame.EVENT_MARKING_WIDTH

        (r ,g, b) = color
        color = wx.Colour(r, g, b)
        brush = wx.Brush(color, wx.SOLID)
        dc.SetBrush(brush)
        dc.DrawRectangle(offset_px, offset_py, width_px, width_py)

    def update_rectangles(self, dc, start, end):
        start += self.ts_start
        end += self.ts_start
        self.sched_tracer.fill_zone(start, end)

    def on_paint(self, event):
        dc = wx.PaintDC(self.scroll_panel)
        self.dc = dc

        width = min(self.width_virtual, self.screen_width)
        (x, y) = self.scroll_start()
        start = self.px_to_us(x)
        end = self.px_to_us(x + width)
        self.update_rectangles(dc, start, end)

    def rect_from_ypixel(self, y):
        y -= RootFrame.Y_OFFSET
        rect = y / (RootFrame.RECT_HEIGHT + RootFrame.RECT_SPACE)
        height = y % (RootFrame.RECT_HEIGHT + RootFrame.RECT_SPACE)

        if rect < 0 or rect > self.nr_rects - 1 or height > RootFrame.RECT_HEIGHT:
            return -1

        return rect

    def update_summary(self, txt):
        if self.txt:
            self.txt.Destroy()
        self.txt = wx.StaticText(self.panel, -1, txt, (0, (self.screen_height / 2) + 50))


    def on_mouse_down(self, event):
        (x, y) = event.GetPositionTuple()
        rect = self.rect_from_ypixel(y)
        if rect == -1:
            return

        t = self.px_to_us(x) + self.ts_start

        self.sched_tracer.mouse_down(rect, t)


    def update_width_virtual(self):
        self.width_virtual = self.us_to_px(self.ts_end - self.ts_start)

    def __zoom(self, x):
        self.update_width_virtual()
        (xpos, ypos) = self.scroll.GetViewStart()
        xpos = self.us_to_px(x) / self.scroll_scale
        self.scroll.SetScrollbars(self.scroll_scale, self.scroll_scale, self.width_virtual / self.scroll_scale, self.height_virtual / self.scroll_scale, xpos, ypos)
        self.Refresh()

    def zoom_in(self):
        x = self.scroll_start_us()
        self.zoom *= 2
        self.__zoom(x)

    def zoom_out(self):
        x = self.scroll_start_us()
        self.zoom /= 2
        self.__zoom(x)


    def on_key_press(self, event):
        key = event.GetRawKeyCode()
        if key == ord("+"):
            self.zoom_in()
            return
        if key == ord("-"):
            self.zoom_out()
            return

        key = event.GetKeyCode()
        (x, y) = self.scroll.GetViewStart()
        if key == wx.WXK_RIGHT:
            self.scroll.Scroll(x + 1, y)
        elif key == wx.WXK_LEFT:
            self.scroll.Scroll(x - 1, y)
        elif key == wx.WXK_DOWN:
            self.scroll.Scroll(x, y + 1)
        elif key == wx.WXK_UP:
            self.scroll.Scroll(x, y - 1)