initial dijkstra

1 files changed, 18 insertions, 11 deletions

diff --git a/server.py b/server.py
index e6ff6c2..6424466 100644
--- a/server.py
+++ b/server.py
@@ -5,29 +5,36 @@ import store
 from lib import run_server, get, post, read_html
 
 
+nodes = None
+
+
 @get('/')
 def index():
+    global nodes
+    nodes = store.extract_osm_nodes("university.osm")
     return read_html('templates/index.html')
 
 
 @get('/show-area')
 def show_area():
-    all = dict()
-    for (k, node) in enumerate(store.select_nodes_in_rectangle(store.extract_osm_nodes("university.osm"), 58.3984, 58.3990, 15.5733, 15.576)):
-        all[node.id] = node.coord_tuple()
-    return json.dumps(all)
+    global nodes
+    rect = dict()
+    for (k, node) in enumerate(store.select_nodes_in_rectangle(nodes, 58.3984, 58.3990, 15.5733, 15.576)):
+        rect[node.id] = node.coord_tuple()
+    return json.dumps(rect)
 
 
 @post('/shortest-path')
 def shortest_path(body):
+    global nodes
     body = json.loads(body)
-    source_id = algorithms.get_closest_node_id(store.nodes, store.Node(-1, body['lat1'], body['lng1']))
-    target_id = algorithms.get_closest_node_id(store.nodes, store.Node(-1, body['lat2'], body['lng2']))
-    print(source_id, target_id)
-    source_node = store.nodes[source_id]
-    target_node = store.nodes[target_id]
-    path = [(source_node.lat, source_node.lng), (target_node.lat, target_node.lng)]
-    response = {'path': path}
+    source_id = algorithms.get_closest_node_id(nodes, store.Node(-1, body['lat1'], body['lng1']))
+    target_id = algorithms.get_closest_node_id(nodes, store.Node(-1, body['lat2'], body['lng2']))
+
+    path = algorithms.find_shortest_path(nodes, source_id, target_id)
+    print(path)
+    response = {"path": [(nodes[node].lat, nodes[node].lng) for node in path]}
+
     return json.dumps(response)