mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-12-22 06:18:07 +00:00
73 lines
2.0 KiB
Python
Executable File
73 lines
2.0 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import sys
|
|
|
|
from graphviz import Digraph
|
|
|
|
|
|
class Message(object):
|
|
def __init__(self, name, attrs):
|
|
self.name = name
|
|
if len(attrs) == 0:
|
|
raise ValueError("message '%s' has no attributes" % name)
|
|
t = attrs[0][0]
|
|
if t in ["start", "end", "auxstart", "auxend", "embed", "ignore"]:
|
|
self.typ = t
|
|
attrs = attrs[1:]
|
|
elif t == "next":
|
|
self.typ = "normal"
|
|
attrs = attrs
|
|
elif t == "wrap":
|
|
self.typ = "normal"
|
|
attrs = attrs
|
|
else:
|
|
raise ValueError("wrong message type in message '%s'" % name)
|
|
self.next = []
|
|
self.wrap = []
|
|
for a in attrs:
|
|
if a[0] == "next":
|
|
self.next.append(a[1])
|
|
elif a[0] == "wrap":
|
|
self.wrap.append(a[1])
|
|
|
|
|
|
def generate_messages(files):
|
|
attrs = []
|
|
msgs = {}
|
|
for f in files:
|
|
for line in open(f, "rt").readlines():
|
|
line = line.rstrip()
|
|
if line.startswith(" * @"):
|
|
attrs.append(line[4:].split(" "))
|
|
elif line.startswith("message "):
|
|
name = line[8:-2]
|
|
msgs[name] = Message(name, attrs)
|
|
attrs = []
|
|
return msgs
|
|
|
|
|
|
def generate_graph(msgs, fn):
|
|
dot = Digraph(format="png")
|
|
dot.attr(rankdir="LR")
|
|
for m in msgs.values():
|
|
if m.typ == "start":
|
|
dot.node(m.name, shape="box", color="blue")
|
|
elif m.typ == "end":
|
|
dot.node(m.name, shape="box", color="green3")
|
|
elif m.typ == "auxstart":
|
|
dot.node(m.name, shape="diamond", color="blue")
|
|
elif m.typ == "auxend":
|
|
dot.node(m.name, shape="diamond", color="green3")
|
|
elif m.typ == "normal":
|
|
dot.node(m.name)
|
|
|
|
for m in msgs.values():
|
|
for n in m.next:
|
|
dot.edge(m.name, n)
|
|
for n in m.wrap:
|
|
dot.edge(m.name, n)
|
|
dot.render(fn)
|
|
|
|
|
|
msgs = generate_messages(sys.argv)
|
|
generate_graph(msgs, "graph.gv")
|