Changeset 9

Timestamp:

10/02/07 03:10:26 (22 months ago)

Author:

gak

Message:

 

Files:

• trunk/pycallgraph.py (modified) (4 diffs)

trunk/pycallgraph.py

@@ -44 +44 @@
 }
 settings = {
+   'node_attributes': {
+       'label': r'%(func)s\ncalls: %(hits)i',
+       'color': '%(col)s',
+    },
     'node_color': lambda calls, : '%f %f %f' % (calls / 2 + .5, calls, 0.9),
     'edge_color': lambda calls, : '%f %f %f' % (calls / 2 + .5, calls, 0.7),
@@ -98 +102 @@
             else:
                 if class_name in settings['exclude_class']:
-                    print 'exclude_class', class_name
                     dont_keep = True
             class_name += '.'
@@ -155 +158 @@
         frac = float(hits) / func_count_max 
         col = settings['node_color'](frac)
-        grp = func.split('.', 1)[0]
-        #ret.append('"%(func)s" [label="%(func)s\\n%(hits)i calls", color = "%(col)s", group = "%(grp)s"]' % locals())
-        ret.append('"%(func)s" [label="%(func)s\\ncalls: %(hits)i", color = "%(col)s"]' % locals())
+        attribs = ['%s="%s"' % a for a in settings['node_attributes'].items()]
+        node_str = '"%s" [%s];' % (func, ','.join(attribs))
+        ret.append(node_str % locals())
     for fr_key, fr_val in call_dict.items():
         if fr_key == '':
@@ -167 +170 @@
             ret.append('"%s"->"%s" %s' % (fr_key, to_key, edge))
     ret.append('}')
-    print '\n'.join(ret)
     return '\n'.join(ret)