1 files changed, 24 insertions, 17 deletions

diff --git a/visualizer.py b/visualizer.py
index 80c2af9..43d74a8 100755
--- a/visualizer.py
+++ b/visualizer.py
@@ -8,24 +8,31 @@ import viz
 
 import gtk
 
-path = 'sample_traces/'
+TIME_PER_MAJ = 10000000
+MAX_NUM_SLOTS = 10000
 
-trace_list = [
-path + 'st-g6-0.bin',
-path + 'st-g6-1.bin',
-path + 'st-g6-2.bin',
-path + 'st-g6-3.bin'
-]
+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
         
-if __name__ == '__main__':
-    stream = reader.trace_reader.trace_reader(trace_list)
-    stream = reader.sanitizer.sanitizer(stream)
-    stream = reader.gedf_test.gedf_test(stream)
-    sched = convert.convert_trace_to_schedule(stream)
-    sched.scan(10000000)
-    renderer = viz.renderer.Renderer(sched)
-    renderer.prepare_task_graph(attrs=viz.format.GraphFormat(time_per_maj=10000000))
-        
-    viz.viewer.MainWindow(renderer)
+    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()