Reorganized tree, along with the visualizer

1 files changed, 40 insertions, 0 deletions

diff --git a/viz/renderer.py b/viz/renderer.py
new file mode 100644
index 0000000..d94129c
--- /dev/null
+++ b/viz/renderer.py
@@ -0,0 +1,40 @@
+#!/usr/bin/python
+from schedule import *
+from draw import *
+
+"""The renderer, a glue object which converts a schedule to its representation
+on a graph."""
+        
+class Renderer(object):
+    def __init__(self, schedule):
+        self.schedule = schedule
+    
+    def prepare_task_graph(self, SurfaceType=ImageSurface, attrs=GraphFormat()):
+        """Outputs the fully-rendered graph (y-axis = tasks) to a Cairo ImageSurface"""
+        item_list = self.get_task_item_list()
+        start, end = self.schedule.get_time_bounds()
+        self.graph = TaskGraph(CairoCanvas, SurfaceType(), start, end, item_list, attrs)
+    
+    def prepare_cpu_graph(self, SurfaceType=ImageSurface, attrs=GraphFormat()):
+        item_list = ['CPU %d' % i for i in range(0, self.schedule.get_num_cpus())]
+        start, end = self.schedule.get_time_bounds()
+        self.graph = CpuGraph(CairoCanvas, SurfaceType(), start, end, item_list, attrs)
+
+    def render_graph_full(self):
+        """Does the heavy lifting for rendering a task or CPU graph, by scanning the schedule
+        and drawing it piece by piece"""
+        #graph.draw_axes('Time', '')
+        self.schedule.render(self.graph)
+    
+    def write_out(self, fname):
+        self.graph.surface.write_out(fname)
+        
+    def get_graph(self):
+        return self.graph
+    
+    def get_schedule(self):
+        return self.schedule
+        
+    def get_task_item_list(self):
+        return [task.get_name() for task in self.schedule.get_task_list()]
+