summaryrefslogtreecommitdiff
path: root/tools/dfagen/dfagen/output/dot.py
diff options
context:
space:
mode:
Diffstat (limited to 'tools/dfagen/dfagen/output/dot.py')
-rw-r--r--tools/dfagen/dfagen/output/dot.py11
1 files changed, 6 insertions, 5 deletions
diff --git a/tools/dfagen/dfagen/output/dot.py b/tools/dfagen/dfagen/output/dot.py
index d6cc057c..ce9c4db8 100644
--- a/tools/dfagen/dfagen/output/dot.py
+++ b/tools/dfagen/dfagen/output/dot.py
@@ -1,12 +1,13 @@
+import os.path
-def write (prefix, automaton, user):
+def write (prefix, automaton, user, outputdir):
output = prefix + '.dot'
- o = open (output, 'w')
+ o = open (os.path.join (outputdir, output), 'w')
o.write ('digraph %s {' % prefix)
- for s in automaton.states.values ():
+ for s in automaton.iter_states ():
o.write (' %s\n' % s.name)
- for tr in s.transitions.values ():
- for br in tr.branches.values ():
+ for tr in s.iter_transitions ():
+ for br in tr.iter_branches ():
o.write (' %(state)s -> %(to)s [ label = "%(event)s" ];\n'
% dict (
state = s.name,