1 files changed, 38 insertions, 0 deletions

diff --git a/visualizer.py b/visualizer.py
new file mode 100755
index 0000000..43d74a8
--- /dev/null
+++ b/visualizer.py
@@ -0,0 +1,38 @@
+#!/usr/bin/python
+
+"""Runs the visualizer."""
+
+import convert
+import reader
+import viz
+
+import gtk
+
+TIME_PER_MAJ = 10000000
+MAX_NUM_SLOTS = 10000
+
+def request_renderer_change(widget, file_list, params):
+    try:      
+        stream = reader.trace_reader.trace_reader(file_list)
+        #stream = reader.sanitizer.sanitizer(stream)
+        #stream = reader.gedf_test.gedf_test(stream)
+        sched = convert.convert_trace_to_schedule(stream)
+    except reader.trace_reader.InvalidRecordError, e:
+        dialog = gtk.MessageDialog(widget, gtk.DIALOG_DESTROY_WITH_PARENT,
+                 gtk.MESSAGE_ERROR, gtk.BUTTONS_CLOSE, str(e))
+        dialog.run()
+        dialog.destroy()
+        return
+        
+    sched.scan(TIME_PER_MAJ, MAX_NUM_SLOTS)
+    
+    task_renderer = viz.renderer.Renderer(sched)
+    task_renderer.prepare_task_graph(attrs=viz.format.GraphFormat(time_per_maj=TIME_PER_MAJ))
+    cpu_renderer = viz.renderer.Renderer(sched)
+    cpu_renderer.prepare_cpu_graph(attrs=viz.format.GraphFormat(time_per_maj=TIME_PER_MAJ))
+    widget.set_renderers({'Tasks' : task_renderer, 'CPUs' : cpu_renderer})
+    
+if __name__ == '__main__':   
+    window = viz.viewer.MainWindow(request_renderer_change)
+    gtk.main()
+