- else:
- if hasattr(u, 'next_functions'):
- i = 0
- for v, j in u.next_functions:
- build_ag_graph_lists(v, node_labels, node_list, link_list)
- add_link(node_list, link_list, u, i, v, j)
- i += 1
+
+ if hasattr(u, 'variable'):
+ fill_graph_lists(u.variable, node_labels, node_list, link_list)
+ add_link(node_list, link_list, u, 0, u.variable, 0)
+
+ if hasattr(u, 'next_functions'):
+ i = 0
+ for v, j in u.next_functions:
+ fill_graph_lists(v, node_labels, node_list, link_list)
+ add_link(node_list, link_list, u, i, v, j)
+ i += 1