trace-cmd: Add Python GTK Event Viewer example

Signed-off-by: Darren Hart <dvhltc@us.ibm.com>

1 files changed, 129 insertions, 0 deletions

diff --git a/event-viewer.py b/event-viewer.py
new file mode 100755
index 0000000..bbafff0
--- /dev/null
+++ b/event-viewer.py
@@ -0,0 +1,129 @@
+#!/usr/bin/env python
+
+from gobject import *
+import gtk
+from tracecmd import *
+
+app = None
+data_func_cnt = 0
+
+class EventStore(gtk.ListStore):
+    def __init__(self, trace):
+        gtk.ListStore.__init__(self, gobject.TYPE_PYOBJECT)
+        self.trace = trace
+        for cpu in range(0, trace.cpus):
+            ev = trace.read_event(cpu)
+            while ev:
+                # store the record offset into the trace file
+                self.append([record_offset_get(ev.rec)])
+                ev = trace.read_event(cpu)
+        print "Loaded %d events across %d cpus" % (len(self), trace.cpus)
+
+    def get_event(self, iter):
+        offset = self.get_value(iter, 0)
+        return self.trace.read_event_at(offset)
+
+
+class EventView(gtk.TreeView):
+    def __init__(self, model):
+        gtk.TreeView.__init__(self, model)
+
+        ts_col = gtk.TreeViewColumn("Time (s)")
+        ts_cell = gtk.CellRendererText()
+        ts_col.pack_start(ts_cell, False)
+        ts_col.set_cell_data_func(ts_cell, self.data_func, "ts")
+        self.append_column(ts_col)
+
+        cpu_col = gtk.TreeViewColumn("CPU")
+        cpu_cell = gtk.CellRendererText()
+        cpu_col.pack_start(cpu_cell, False)
+        cpu_col.set_cell_data_func(cpu_cell, self.data_func, "cpu")
+        self.append_column(cpu_col)
+
+        event_col = gtk.TreeViewColumn("Event")
+        event_cell = gtk.CellRendererText()
+        event_col.pack_start(event_cell, False)
+        event_col.set_cell_data_func(event_cell, self.data_func, "event")
+        self.append_column(event_col)
+
+        pid_col = gtk.TreeViewColumn("PID")
+        pid_cell = gtk.CellRendererText()
+        pid_col.pack_start(pid_cell, False)
+        pid_col.set_cell_data_func(pid_cell, self.data_func, "pid")
+        self.append_column(pid_col)
+
+        comm_col = gtk.TreeViewColumn("Comm")
+        comm_cell = gtk.CellRendererText()
+        comm_col.pack_start(comm_cell, False)
+        comm_col.set_cell_data_func(comm_cell, self.data_func, "comm")
+        self.append_column(comm_col)
+
+    def data_func(self, col, cell, model, iter, data):
+        global app, data_func_cnt
+        data_func_cnt = data_func_cnt + 1
+        if app:
+            app.inc_data_func()
+
+        ev = model.get_event(iter)
+        if not ev:
+            return False
+        if data == "ts":
+            cell.set_property("markup", "%d.%d" % (ev.ts/1000000000,
+                                                   ev.ts%1000000000))
+            return True
+        if data == "cpu":
+            cell.set_property("markup", ev.cpu)
+            return True
+        if data == "event":
+            cell.set_property("markup", ev.name)
+            return True
+        if data == "pid":
+            cell.set_property("markup", ev.pid)
+            return True
+        if data == "comm":
+            cell.set_property("markup", ev.comm)
+            return True
+
+        return False
+
+
+class EventViewerApp(gtk.Window):
+    def __init__(self, trace):
+        gtk.Window.__init__(self)
+
+        self.set_size_request(650, 400)
+        self.set_position(gtk.WIN_POS_CENTER)
+
+        self.connect("destroy", gtk.main_quit)
+        self.set_title("Event Viewer")
+
+
+        es = EventStore(trace)
+        view = EventView(es)
+
+        sw = gtk.ScrolledWindow()
+        sw.set_policy(gtk.POLICY_NEVER, gtk.POLICY_ALWAYS)
+        sw.add(view)
+
+        # track how often the treeview data_func is called
+        self.data_func_label = gtk.Label("0")
+        hbox = gtk.HBox()
+        hbox.pack_start(gtk.Label("Data Func Calls:"), False, False)
+        hbox.pack_start(self.data_func_label, False, False)
+
+        vbox = gtk.VBox()
+        vbox.pack_start(hbox, False)
+        vbox.pack_end(sw)
+
+        self.add(vbox)
+        self.show_all()
+
+    def inc_data_func(self):
+        global data_func_cnt
+        self.data_func_label.set_text(str(data_func_cnt))
+
+
+if __name__ == "__main__":
+    trace = Trace("trace.dat")
+    app = EventViewerApp(trace)
+    gtk.main()