1 files changed, 3 insertions, 23 deletions

diff --git a/visualizer.py b/visualizer.py
index 80c2af9..c15bf51 100755
--- a/visualizer.py
+++ b/visualizer.py
@@ -1,31 +1,11 @@
 #!/usr/bin/python
-
-"""Runs the visualizer."""
-
-import convert
-import reader
-import viz
-
 import gtk
+import viz
 
-path = 'sample_traces/'
-
-trace_list = [
-path + 'st-g6-0.bin',
-path + 'st-g6-1.bin',
-path + 'st-g6-2.bin',
-path + 'st-g6-3.bin'
-]
-        
-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)
+def visualizer(stream):
+    sched = viz.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)
     gtk.main()
-