fix: bug with dragging offscreen, locked stations and partial recalc …

…and weird angles in/out of stations

src/algorithm/cost_calculation.rs

@@ -360,7 +360,7 @@ fn calc_station_exit_cost(
                     node,
                     false,
                 )
-                .map(|c| c / 2.0);
+                .map(|c| c / 5.0);
             }
         }
 
@@ -374,7 +374,8 @@ fn calc_station_exit_cost(
                     station.get_pos(),
                     node,
                     true,
-                );
+                )
+                .map(|c| c / 5.0);
             }
         }
     }

src/components/canvas/mouse_out.rs

@@ -8,4 +8,5 @@ use crate::MapState;
 pub fn on_mouse_out(map_state: &mut MapState) {
     map_state.clear_selected_line();
     map_state.clear_box_select();
+    map_state.clear_drag_offset();
 }

src/components/canvas/mouse_up.rs

@@ -38,7 +38,7 @@ pub fn on_mouse_up(map_state: &mut MapState, ev: &UiEvent, shift_key: bool) {
         .get_drag_offset()
         .is_some()
     {
-        map_state.set_drag_offset(None);
+        map_state.clear_drag_offset();
     }
 
     // Handle a click while having an operation selected

src/components/organisms/canvas_controls.rs

@@ -39,6 +39,7 @@ use crate::{
         MapState,
     },
     models::Map,
+    unwrap_or_return,
     utils::{
         IDData,
         IDManager,
@@ -128,22 +129,32 @@ pub fn CanvasControls() -> impl IntoView {
     });
 
     // Handle the response from the algorithm.
-    let handle_algorithm_response = move |resp: AlgorithmResponse| {
+    let handle_algorithm_response = move |resp: AlgorithmResponse, partial: bool| {
         if resp.success
             || map_state
                 .get_untracked()
                 .get_algorithm_settings()
                 .output_on_fail
         {
             map_state.update(|state| {
-                state.set_map(resp.map);
+                if partial {
+                    unwrap_or_return!(state
+                        .get_mut_map()
+                        .update_from_partial(&resp.map));
+                } else {
+                    state.set_map(resp.map);
+                }
             });
             IDManager::from_data(resp.id_manager_data);
         }
     };
 
     // Dispatch the algorithm request.
     let algorithm_req = create_action(move |req: &AlgorithmRequest| {
+        map_state.update_untracked(|state| {
+            state.clear_all_selections();
+        });
+
         let req = req.clone();
         async move {
             let (abort_handle, resp_stream) = executor
@@ -162,7 +173,7 @@ pub fn CanvasControls() -> impl IntoView {
             let last = resp_stream
                 .inspect(|resp| {
                     if req.midway_updates {
-                        handle_algorithm_response(resp.clone());
+                        handle_algorithm_response(resp.clone(), req.partial);
                     }
                 })
                 .fold(
@@ -175,7 +186,7 @@ pub fn CanvasControls() -> impl IntoView {
             // now.
             if !req.midway_updates {
                 if let Some(resp) = last {
-                    handle_algorithm_response(resp);
+                    handle_algorithm_response(resp, req.partial);
                 }
             }
         }

src/components/state/map.rs

@@ -112,6 +112,7 @@ impl MapState {
         self.clear_box_select();
         self.clear_clicked_on_station();
         self.clear_clicked_on_edge();
+        self.clear_drag_offset();
     }
 
     /// A getter method for the [`Map`].
@@ -310,6 +311,11 @@ impl MapState {
         self.drag_offset = offset;
     }
 
+    /// Clear the drag offset.
+    pub fn clear_drag_offset(&mut self) {
+        self.drag_offset = None;
+    }
+
     /// A getter method for the last loaded map.
     pub fn get_last_loaded(&self) -> Option<&Map> {
         self.last_loaded