Add method to get nodes in extent

api/s24_coordinates.py

@@ -27,6 +27,20 @@ def get_node_coord(name: str, node: str) -> dict[str, float]:
         return {'x': 0.0, 'y': 0.0}
     return from_postgis_point(row['coord_geom'])
 
+# DingZQ 2025-01-03, get nodes in extent
+# return node id list
+def get_nodes_in_extent(name: str, x1: float, y1: float, x2: float, y2: float) -> list[str]:
+    nodes = []
+    objs = read_all(name, 'select node, st_astext(coord) as coord_geom from coordinates')
+    for obj in objs:
+        node_id = obj['node']
+        coord = from_postgis_point(obj['coord_geom'])
+        x = coord['x']
+        y = coord['y']
+        if x1 <= x <= x2 and y1 <= y <= y2:
+            nodes.append(node_id)
+    return nodes
+
 
 def node_has_coord(name: str, node: str) -> bool:
     return try_read(name, f"select node from coordinates where node = '{node}'") != None