1 files changed, 4 insertions, 7 deletions

diff --git a/unit_trace/visualizer.py b/unit_trace/visualizer.py
index 6e14b18..af2dc6a 100755
--- a/unit_trace/visualizer.py
+++ b/unit_trace/visualizer.py
@@ -2,29 +2,26 @@
 
 """Runs the visualizer."""
 
-import convert
-import reader
 import viz
-
 import gtk
+import trace_reader
 
 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 = trace_reader.trace_reader(file_list)
         #stream = reader.sanitizer.sanitizer(stream)
         #stream = reader.gedf_test.gedf_test(stream)
         sched = viz.convert.convert_trace_to_schedule(stream)
-    except reader.trace_reader.InvalidRecordError, e:
+    except 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)
+    sched.scan(TIME_PER_MAJ)
     
     task_renderer = viz.renderer.Renderer(sched)
     task_renderer.prepare_task_graph(attrs=viz.format.GraphFormat(time_per_maj=TIME_PER_MAJ))