feat: add a script to generate a nice graph (#303)

agents/scripts/.gitignore

@@ -0,0 +1,4 @@
+graph
+graph.pdf
+graph.png
+

agents/scripts/dot.py

@@ -0,0 +1,117 @@
+# Generate a graph from a JSON file
+# Currently it only generate the first graph but it's easy to make it generate all graphs
+# Author: Seven Du <[email protected]>
+# usage:
+#     pip install graphviz
+#     python dot.py
+
+import json
+import graphviz
+
+COLORS = {
+    "flush": "#999",
+    "cmd": "#0f0",
+    "data": "#00f",
+    "text_data": "#f00",
+    "pcm_frame": "purple",
+}
+
+connection_types = ["data", "cmd", "audio_frame", "video_frame"]
+
+
+def color(port):
+    if port in COLORS:
+        return COLORS[port]
+    return "#000"
+
+
+def find_node(nodes, name):
+    for node in nodes:
+        if node["name"] == name:
+            return node
+    return None
+
+
+def create_graph(json_data):
+    # Initialize a directed graph
+    graph = graphviz.Digraph("G", filename="graph.gv")
+    graph.graph_attr["rankdir"] = "LR"
+    graph.graph_attr["dpi"] = "150"
+    graph.graph_attr["splines"] = "true"
+    graph.attr("node", shape="none")
+
+    # Add nodes to the graph
+    nodes = json_data["_ten"]["predefined_graphs"][0]["nodes"]
+    connections = json_data["_ten"]["predefined_graphs"][0]["connections"]
+    for node in nodes:
+        node["i_ports"] = ["flush"]
+        node["o_ports"] = ["flush"]
+    for node in nodes:
+        if node["type"] != "extension":
+            continue
+        for connection in connections:
+            if connection["extension"] == node["name"]:
+                for connection_type in connection_types:
+                    if connection_type in connection:
+                        data = connection[connection_type]
+                        for item in data:
+                            node["o_ports"].append(item["name"])
+                            for dest in item["dest"]:
+                                dest_node = find_node(nodes, dest["extension"])
+                                if dest_node:
+                                    dest_node["i_ports"].append(item["name"])
+    for node in nodes:
+        if node["type"] != "extension":
+            continue
+        node["i_ports"] = set(node["i_ports"])
+        node["o_ports"] = set(node["o_ports"])
+        print("====iports: ", node["name"], node["i_ports"])
+        print("====oports: ", node["name"], node["o_ports"])
+        iports = ""
+        for port in node["i_ports"]:
+            iports += f'<tr><td align="left" port="i_{port}">⊙ {port}</td></tr>'
+        oports = ""
+        for port in node["o_ports"]:
+            oports += f'<tr><td align="right" port="o_{port}">{port} ⊙</td></tr>'
+
+        # Use HTML-like label for nodes
+        label = f"""<
+        <table border="0" cellborder="1" cellspacing="0">
+            <tr><td colspan="2" bgcolor="#ddd"><b>{node["name"]}</b></td></tr>
+            <tr><td colspan="2">properties</td></tr>
+            <tr><td colspan="2">extensionGroup<br/>{node["extension_group"]}</td></tr>
+            <tr><td>
+                <table border="0" cellspacing="0">{iports}</table>
+            </td>
+            <td>
+                <table border="0" cellspacing="0">{oports}</table>
+            </td>
+        </tr>    
+        </table>>"""
+        graph.node(node["name"], label)
+
+    # Add edges to the graph
+    for connection in connections:
+        for connection_type in connection_types:
+            if connection_type in connection:
+                for data in connection[connection_type]:
+                    for dest in data["dest"]:
+                        graph.edge(
+                            f'{connection["extension"]}:o_{data["name"]}',
+                            f'{dest["extension"]}:i_{data["name"]}',
+                            color=color(data["name"]),
+                            label=connection_type,
+                        )
+
+    # Save the graph to a file
+    print(graph.source)
+    graph.render("graph", format="png")
+    graph.view()
+
+
+# Load the JSON data
+with open("../property.json") as f:
+    data = json.load(f)
+
+# Create the graph
+create_graph(data)