summaryrefslogtreecommitdiff
path: root/monkey
diff options
context:
space:
mode:
Diffstat (limited to 'monkey')
-rw-r--r--monkey/graph.py13
1 files changed, 13 insertions, 0 deletions
diff --git a/monkey/graph.py b/monkey/graph.py
index 3e1d3a2..cbe44f7 100644
--- a/monkey/graph.py
+++ b/monkey/graph.py
@@ -20,6 +20,19 @@ class Node(object):
target.ein.append(self)
return target
+ # Return a list of nodes in [self].
+ def preorder(self):
+ nodes = [self]
+ for n in self.eout:
+ nodes += n.preorder()
+ return nodes
+
+ # Return a one-line string representation of [self].
+ def __str__(self):
+ return '(' + str(self.data) + ' ' + \
+ ' '.join([str(c) if c.eout else '"'+str(c.data)+'"' for c in self.eout]) + \
+ ')'
+
def __repr__(self):
return str(self.data)