color edge types
parent
87e6a0e770
commit
d93f2b9c64
12
osc/main.py
12
osc/main.py
|
|
@ -8,6 +8,15 @@ edge_attrs = {
|
||||||
'arrowhead': 'vee',
|
'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 = Digraph('G', filename='graph.gv', format='png', engine='circo', strict=True, edge_attr = edge_attrs)
|
||||||
|
|
||||||
g_lock = threading.Lock()
|
g_lock = threading.Lock()
|
||||||
|
|
@ -30,7 +39,7 @@ def print_edge(address, *args):
|
||||||
if v == -1:
|
if v == -1:
|
||||||
return
|
return
|
||||||
with g_lock:
|
with g_lock:
|
||||||
g.edge(str(v), str(w))
|
g.edge(str(v), str(w), color=edge_colors[t])
|
||||||
print(args)
|
print(args)
|
||||||
|
|
||||||
def update_loop():
|
def update_loop():
|
||||||
|
|
@ -38,6 +47,7 @@ def update_loop():
|
||||||
g.render("Test")
|
g.render("Test")
|
||||||
time.sleep(0.4)
|
time.sleep(0.4)
|
||||||
|
|
||||||
|
|
||||||
ul_thread = threading.Thread(target=update_loop)
|
ul_thread = threading.Thread(target=update_loop)
|
||||||
|
|
||||||
ul_thread.start()
|
ul_thread.start()
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue