From 73598b7b38f979e15767fd4c0cc3227361d33cac Mon Sep 17 00:00:00 2001
From: DingZQ <ding@aecforge.com>
Date: Sun, 8 Dec 2024 18:08:52 +0800
Subject: [PATCH] Add API get network coords

---
 main.py      | 4 ++++
 tjnetwork.py | 8 ++++++++
 2 files changed, 12 insertions(+)

diff --git a/main.py b/main.py
index 47db619..efe0046 100644
--- a/main.py
+++ b/main.py
@@ -1391,6 +1391,10 @@ async def fastapi_set_option_properties(network: str, req: Request) -> ChangeSet
 async def fastapi_get_node_coord(network: str, node: str) -> dict[str, float] | None:
     return get_node_coord(network, node)
 
+@app.get("/getnetworkcoords/")
+async def fastapi_get_network_coords(network: str) -> dict[str, dict[str, float]] | None:
+    return get_network_coords(network)
+
 ############################################################
 # vertex 25.[VERTICES]
 ############################################################
diff --git a/tjnetwork.py b/tjnetwork.py
index 20006ec..f964d41 100644
--- a/tjnetwork.py
+++ b/tjnetwork.py
@@ -830,6 +830,14 @@ def set_option_v3(name: str, cs: ChangeSet) -> ChangeSet:
 def get_node_coord(name: str, node_id: str) -> dict[str, float]:
     return api.get_node_coord(name, node_id)
 
+# DingZQ, 2024-12-08, get all node coord
+# id, x, y
+def get_network_coords(name: str) -> dict[str, dict[str, float]]:
+    node_ids = api.get_all_node_ids(name)
+    result = {}
+    for node_id in node_ids:
+        result[node_id] = api.get_node_coord(name, node_id)
+    return result
 
 ############################################################
 # vertex 25.[VERTICES]