contrib/fsm-to-dot: mark arrows that feed events

Most arrows draw state transitions, but some want to show that an event is
delivered to a state. Mark those with a "halfopen" arrow head.

Change-Id: Ib07380894a6d513896c9135f175ecbf653a23eec

1 files changed, 12 insertions, 7 deletions

diff --git a/contrib/fsm-to-dot.py b/contrib/fsm-to-dot.py
index 872577e8..83c95b69 100755
--- a/contrib/fsm-to-dot.py
+++ b/contrib/fsm-to-dot.py
@@ -137,10 +137,11 @@ class Event:
     return cmp(event.name, other.name)
 
 class Edge:
-  def __init__(edge, to_state, event_name=None, style=None, action=None, color=None):
+  def __init__(edge, to_state, event_name=None, style=None, action=None, color=None, arrow_head=None):
     edge.to_state = to_state
     edge.style = style
     edge.color = color
+    edge.arrow_head = arrow_head
     edge.events = []
     edge.actions = []
     edge.add_event_name(event_name)
@@ -330,7 +331,8 @@ class Fsm:
 
   def add_special_state(fsm, additional_states, name, in_state=None,
                         out_state=None, event_name=None, kind=KIND_FUNC,
-                        state_action=None, label=None, edge_action=None):
+                        state_action=None, label=None, edge_action=None,
+                        style='dotted', arrow_head=None):
     additional_state = None
     for s in additional_states:
       if s.short_name == name:
@@ -356,13 +358,13 @@ class Fsm:
 
     if out_state:
       additional_state.out_state_names.append(out_state.name)
-      additional_state.add_out_edge(Edge(out_state, event_name, 'dotted',
-                                         action=edge_action))
+      additional_state.add_out_edge(Edge(out_state, event_name, style=style,
+                                         action=edge_action, arrow_head=arrow_head))
 
     if in_state:
       in_state.out_state_names.append(additional_state.name)
-      in_state.add_out_edge(Edge(additional_state, event_name, 'dotted',
-                                 action=edge_action))
+      in_state.add_out_edge(Edge(additional_state, event_name, style=style,
+                                 action=edge_action, arrow_head=arrow_head))
 
 
   def find_event_edges(fsm, c_files):
@@ -425,7 +427,8 @@ class Fsm:
         # them to the graph as well
         additional_funcs = [f for f in funcs_for_in_event if f not in fsm.action_funcs]
         for af in additional_funcs:
-          fsm.add_special_state(additional_states, af, None, state, in_event_name)
+          fsm.add_special_state(additional_states, af, None, state, in_event_name,
+                                arrow_head='halfopen')
 
     fsm.states.extend(additional_states)
 
@@ -540,6 +543,8 @@ class Fsm:
           attrs.append('style=%s'% out_edge.style)
         if out_edge.color:
           attrs.append('color=%s'% out_edge.color)
+        if out_edge.arrow_head:
+          attrs.append('arrowhead=%s'% out_edge.arrow_head)
         attrs_str = ''
         if attrs:
           attrs_str = ' [%s]' % (','.join(attrs))