Include task stats in comments of graph dot strs

1 files changed, 18 insertions, 2 deletions

diff --git a/ecrts14/graph.py b/ecrts14/graph.py
index 95e63cb..dc2807d 100755
--- a/ecrts14/graph.py
+++ b/ecrts14/graph.py
@@ -84,7 +84,16 @@ class node:
         elif self.isSink:
             attr.append('style=filled')
             attr.append('color="#E04050D3"')
-        return 'node [%s]; %d;' % (', '.join(attr), self.id)
+        node_str = 'node [%s]; %d;' % (', '.join(attr), self.id)
+        if hasattr(self, 'task'):
+            # assumes time is in microseconds
+            cost_str = format(self.task.cost/1000.0, '.4f').rstrip('0').rstrip('.')
+            period_str = format(self.task.period/1000.0, '.4f').rstrip('0').rstrip('.')
+            extra_str = ' /* node_%d: exec: %s, period: %s, split: %d, cluster %d, is_src: %s, is_sink: %s */' % (
+                self.id, cost_str, period_str, self.task.split, self.task.partition, self.isSrc, self.isSink)
+        else:
+            extra_str = ''
+        return node_str + extra_str
 
 class edge:
     def __init__(self, a, b):
@@ -97,7 +106,10 @@ class edge:
 
     def dot(self):
         criticalPath = (self.p.isSpine and self.s.isSpine and abs(self.s.privLevel - self.p.privLevel) == 1)
-        return '%d -> %d [%s];' % (self.p.id, self.s.id, ('style="", color=dimgrey', 'style=bold, color="#E04050"')[criticalPath])
+        edge_str = '%d -> %d [%s];' % (self.p.id, self.s.id, ('style="", color=dimgrey', 'style=bold, color="#E04050"')[criticalPath])
+        wss_str = format(self.wss, '.4f').rstrip('0').rstrip('.')
+        extra_str = ' /* parent: node_%d, child: node_%d, wss: %s, is_spine: %s */' % (self.p.id, self.s.id, wss_str, criticalPath)
+        return edge_str + extra_str
 
 class graph:
     def __init__(self):
@@ -124,6 +136,10 @@ class graph:
         outs.write('\tratio="compress";\n')
         outs.write('\tsize="11,8.5";\n')
 
+        extra_str = '/* graph_%d: depth: %d, nr_nodes: %d, nr_src: %d, nr_sink: %d, nr_edges: %d */' % (
+            self.id, self.depth, len(self.nodes), len(self.sources), len(self.sinks), len(self.edges))
+        outs.write('\t%s\n' % extra_str)
+
         # draw nodes
         for n in self.nodes:
             outs.write('\t')