diff --git a/osc/main.py b/osc/main.py index cc789de..075bfe3 100644 --- a/osc/main.py +++ b/osc/main.py @@ -8,6 +8,15 @@ edge_attrs = { 'arrowhead': 'vee', } +edge_colors = { + 'T': 'red', + 'F': 'yellow', + 'B': 'green', + 'X': 'blue', + 'L': 'purple', +} + + g = Digraph('G', filename='graph.gv', format='png', engine='circo', strict=True, edge_attr = edge_attrs) g_lock = threading.Lock() @@ -30,7 +39,7 @@ def print_edge(address, *args): if v == -1: return with g_lock: - g.edge(str(v), str(w)) + g.edge(str(v), str(w), color=edge_colors[t]) print(args) def update_loop(): @@ -38,6 +47,7 @@ def update_loop(): g.render("Test") time.sleep(0.4) + ul_thread = threading.Thread(target=update_loop) ul_thread.start()