From ee2ee53614131491fc8ac42d93245735a9ae6c91 Mon Sep 17 00:00:00 2001
From: keith <keith@the-simmons.net>
Date: Wed, 2 Sep 2020 11:24:53 -0700
Subject: [PATCH] better mouse management with multigrid

---
 src/main.rs                                   |   1 +
 .../{cursor_renderer => }/animation_utils.rs  |   0
 src/renderer/cursor_renderer/cursor_vfx.rs    |   2 +-
 src/renderer/cursor_renderer/mod.rs           |   5 +-
 src/renderer/mod.rs                           | 101 +++++++++++++++---
 src/window.rs                                 |  70 +++++++-----
 6 files changed, 131 insertions(+), 48 deletions(-)
 rename src/renderer/{cursor_renderer => }/animation_utils.rs (100%)

diff --git a/src/main.rs b/src/main.rs
index 7eb62aa..04c18f1 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -52,6 +52,7 @@ fn main() {
 
     window::initialize_settings();
     redraw_scheduler::initialize_settings();
+    renderer::initialize_settings();
     renderer::cursor_renderer::initialize_settings();
     bridge::layouts::initialize_settings();
 
diff --git a/src/renderer/cursor_renderer/animation_utils.rs b/src/renderer/animation_utils.rs
similarity index 100%
rename from src/renderer/cursor_renderer/animation_utils.rs
rename to src/renderer/animation_utils.rs
diff --git a/src/renderer/cursor_renderer/cursor_vfx.rs b/src/renderer/cursor_renderer/cursor_vfx.rs
index 9ab603e..31ab2bc 100644
--- a/src/renderer/cursor_renderer/cursor_vfx.rs
+++ b/src/renderer/cursor_renderer/cursor_vfx.rs
@@ -1,8 +1,8 @@
 use log::error;
 use skulpin::skia_safe::{paint::Style, BlendMode, Canvas, Color, Paint, Point, Rect};
 
-use super::animation_utils::*;
 use super::CursorSettings;
+use crate::renderer::animation_utils::*;
 use crate::editor::{Colors, Cursor};
 use crate::settings::*;
 
diff --git a/src/renderer/cursor_renderer/mod.rs b/src/renderer/cursor_renderer/mod.rs
index 415ea5c..a66026e 100644
--- a/src/renderer/cursor_renderer/mod.rs
+++ b/src/renderer/cursor_renderer/mod.rs
@@ -1,4 +1,3 @@
-mod animation_utils;
 mod blink;
 mod cursor_vfx;
 
@@ -7,10 +6,10 @@ use skulpin::skia_safe::{Canvas, Paint, Path, Point};
 use crate::editor::{Colors, Cursor, CursorShape, EDITOR};
 use crate::redraw_scheduler::REDRAW_SCHEDULER;
 use crate::renderer::CachingShaper;
+use crate::renderer::animation_utils::*;
 use crate::settings::*;
-
 use crate::bridge::EditorMode;
-use animation_utils::*;
+
 use blink::*;
 
 const DEFAULT_CELL_PERCENTAGE: f32 = 1.0 / 8.0;
diff --git a/src/renderer/mod.rs b/src/renderer/mod.rs
index 4c10bf5..990489a 100644
--- a/src/renderer/mod.rs
+++ b/src/renderer/mod.rs
@@ -4,24 +4,93 @@ use std::sync::Arc;
 use log::trace;
 use skulpin::skia_safe::gpu::SurfaceOrigin;
 use skulpin::skia_safe::{
-    colors, dash_path_effect, Budgeted, Canvas, ImageInfo, Paint, Rect, Surface
+    colors, dash_path_effect, Budgeted, Canvas, ImageInfo, Paint, Rect, Surface, Point, Color, image_filters::*
 };
 use skulpin::CoordinateSystemHelper;
 
 mod caching_shaper;
 pub mod cursor_renderer;
 pub mod font_options;
+pub mod animation_utils;
 
 pub use caching_shaper::CachingShaper;
 pub use font_options::*;
+use animation_utils::*;
 
 use crate::editor::{Style, WindowRenderInfo, EDITOR};
 use crate::redraw_scheduler::REDRAW_SCHEDULER;
+use crate::settings::*;
 use cursor_renderer::CursorRenderer;
 
+
+// ----------------------------------------------------------------------------
+
+#[derive(Clone)]
+pub struct RendererSettings {
+    animation_length: f32,
+}
+
+pub fn initialize_settings() {
+    SETTINGS.set(&RendererSettings {
+        animation_length: 0.15,
+    });
+
+    register_nvim_setting!("window_animation_length", RendererSettings::animation_length);
+}
+
+// ----------------------------------------------------------------------------
+
 pub struct RenderedWindow {
     surface: Surface,
-    current_position: (f32, f32)
+    start_position: Point,
+    current_position: Point,
+    previous_destination: Point,
+    t: f32
+}
+
+impl RenderedWindow {
+    pub fn new(surface: Surface, position: Point) -> RenderedWindow {
+        RenderedWindow {
+            surface,
+            start_position: position.clone(),
+            current_position: position.clone(),
+            previous_destination: position.clone(),
+            t: 2.0 // 2.0 is out of the 0.0 to 1.0 range and stops animation
+        }
+    }
+
+    pub fn update(
+        &mut self,
+        settings: &RendererSettings,
+        destination: Point,
+        dt: f32
+    ) -> bool {
+        if destination != self.previous_destination {
+            self.t = 0.0;
+            self.start_position = self.current_position;
+            self.previous_destination = destination;
+        }
+
+        if (self.t - 1.0).abs() < std::f32::EPSILON {
+            return false;
+        }
+
+        if (self.t - 1.0).abs() < std::f32::EPSILON {
+            // We are at destination, move t out of 0-1 range to stop the animation
+            self.t = 2.0;
+        } else {
+            self.t = (self.t + dt / settings.animation_length).min(1.0);
+        }
+
+        self.current_position = ease_point(
+            ease_out_expo,
+            self.start_position,
+            destination,
+            self.t,
+        );
+
+        true
+    }
 }
 
 pub struct Renderer {
@@ -195,13 +264,14 @@ impl Renderer {
 
     pub fn draw_window(
         &mut self,
+        settings: &RendererSettings,
         root_canvas: &mut Canvas,
         window_render_info: &WindowRenderInfo,
         default_style: &Arc<Style>,
+        dt: f32
     ) -> (u64, Rect) {
         let (grid_left, grid_top) = window_render_info.grid_position;
-        let target_left = grid_left as f32 * self.font_width;
-        let target_top = grid_top as f32 * self.font_height;
+        let window_destination = Point::new(grid_left as f32 * self.font_width, grid_top as f32 * self.font_height);
 
         let image_width = (window_render_info.width as f32 * self.font_width) as i32;
         let image_height = (window_render_info.height as f32 * self.font_height) as i32;
@@ -213,10 +283,7 @@ impl Renderer {
         }
         .unwrap_or_else(|| {
             let surface = self.build_window_surface(root_canvas, &default_style, (image_width, image_height));
-            RenderedWindow {
-                surface,
-                current_position: (target_left, target_top)
-            }
+            RenderedWindow::new(surface, window_destination)
         });
 
         if rendered_window.surface.width() != image_width || rendered_window.surface.height() != image_height {
@@ -225,10 +292,7 @@ impl Renderer {
             old_surface.draw(rendered_window.surface.canvas(), (0.0, 0.0), None);
         }
 
-        let (current_left, current_top) = rendered_window.current_position;
-        let current_left = current_left + (target_left - current_left) * 0.4;
-        let current_top = current_top +  (target_top - current_top) * 0.4;
-        rendered_window.current_position = (current_left, current_top);
+        rendered_window.update(settings, window_destination, dt);
 
         let mut canvas = rendered_window.surface.canvas();
 
@@ -255,15 +319,17 @@ impl Renderer {
 
         root_canvas.save_layer(&Default::default());
 
-        unsafe {
-            rendered_window.surface.draw(root_canvas.surface().unwrap().canvas(), (current_left, current_top), None);
-        }
+        rendered_window.surface.draw(
+            root_canvas.as_mut(),
+            (rendered_window.current_position.x, rendered_window.current_position.y), 
+            None);
 
         root_canvas.restore();
 
+        let window_position = rendered_window.current_position.clone();
         self.rendered_windows.insert(window_render_info.grid_id, rendered_window);
 
-        (window_render_info.grid_id, Rect::new(current_left, current_top, current_left + image_width as f32, current_top + image_height as f32))
+        (window_render_info.grid_id, Rect::from_point_and_size(window_position, (image_width as f32, image_height as f32)))
     }
 
     pub fn draw(
@@ -275,6 +341,7 @@ impl Renderer {
         trace!("Rendering");
 
         REDRAW_SCHEDULER.queue_next_frame();
+        let settings = SETTINGS.get::<RendererSettings>();
 
         let (render_info, default_style, cursor, guifont_setting) = {
             let mut editor = EDITOR.lock();
@@ -300,7 +367,7 @@ impl Renderer {
 
         self.window_regions = render_info.windows
             .iter()
-            .map(|window_render_info| self.draw_window(gpu_canvas, window_render_info, &default_style))
+            .map(|window_render_info| self.draw_window(&settings, gpu_canvas, window_render_info, &default_style, dt))
             .collect();
 
         self.cursor_renderer.draw(
diff --git a/src/window.rs b/src/window.rs
index 8691b51..66f3ec9 100644
--- a/src/window.rs
+++ b/src/window.rs
@@ -8,11 +8,11 @@ use skulpin::sdl2::event::{Event, WindowEvent};
 use skulpin::sdl2::keyboard::Keycode;
 use skulpin::sdl2::video::FullscreenType;
 use skulpin::sdl2::Sdl;
+use skulpin::ash::prelude::VkResult;
 use skulpin::{
     CoordinateSystem, LogicalSize, PhysicalSize, PresentMode, Renderer as SkulpinRenderer,
     RendererBuilder, Sdl2Window, Window,
 };
-use skulpin::skia_safe::Rect;
 
 use crate::bridge::{produce_neovim_keybinding_string, UiCommand, BRIDGE};
 use crate::editor::EDITOR;
@@ -175,7 +175,7 @@ impl WindowWrapper {
             transparency: 1.0,
             fullscreen: false,
             cached_size: (0, 0),
-            cached_position: (0, 0),
+            cached_position: (0, 0)
         }
     }
 
@@ -285,11 +285,13 @@ impl WindowWrapper {
         let logical_position = PhysicalSize::new(x as u32, y as u32)
             .to_logical(sdl_window_wrapper.scale_factor());
 
+        let mut top_window_position = (0.0, 0.0);
         let mut top_grid_position = None;
 
         for (grid_id, window_region) in self.renderer.window_regions.iter() {
             if logical_position.width >= window_region.left as u32 && logical_position.width < window_region.right as u32 &&
                 logical_position.height >= window_region.top as u32 && logical_position.height < window_region.bottom as u32 {
+                top_window_position = (window_region.left, window_region.top);
                 top_grid_position = Some((
                     grid_id, 
                     LogicalSize::new(logical_position.width - window_region.left as u32, logical_position.height - window_region.top as u32)
@@ -298,16 +300,20 @@ impl WindowWrapper {
         }
 
         if let Some((grid_id, grid_position)) = top_grid_position {
-            self.grid_id_under_mouse = dbg!(*grid_id);
+            self.grid_id_under_mouse = *grid_id;
             self.mouse_position = LogicalSize::new(
                 (grid_position.width as f32 / self.renderer.font_width) as u32,
                 (grid_position.height as f32 / self.renderer.font_height) as u32
             );
 
             if self.mouse_down && previous_position != self.mouse_position {
+                let (window_left, window_top) = top_window_position;
+                let adjusted_drag_left = self.mouse_position.width + (window_left / self.renderer.font_width) as u32;
+                let adjusted_drag_top = self.mouse_position.height + (window_top / self.renderer.font_height) as u32;
+
                 BRIDGE.queue_command(UiCommand::Drag {
                     grid_id: self.grid_id_under_mouse,
-                    position: (self.mouse_position.width, self.mouse_position.height),
+                    position: (adjusted_drag_left, adjusted_drag_top),
                 });
             }
         }
@@ -370,11 +376,7 @@ impl WindowWrapper {
         REDRAW_SCHEDULER.queue_next_frame();
     }
 
-    pub fn draw_frame(&mut self) -> bool {
-        if !BRIDGE.running.load(Ordering::Relaxed) {
-            return false;
-        }
-
+    pub fn draw_frame(&mut self, dt: f32) -> VkResult<bool> {
         let sdl_window_wrapper = Sdl2Window::new(&self.window);
         let new_size = sdl_window_wrapper.logical_size();
         if self.previous_size != new_size {
@@ -388,23 +390,17 @@ impl WindowWrapper {
 
         if REDRAW_SCHEDULER.should_draw() || SETTINGS.get::<WindowSettings>().no_idle {
             let renderer = &mut self.renderer;
-            let error = self
-                .skulpin_renderer
+            self.skulpin_renderer
                 .draw(&sdl_window_wrapper, |canvas, coordinate_system_helper| {
-                    let dt = 1.0 / (SETTINGS.get::<WindowSettings>().refresh_rate as f32);
-
                     if renderer.draw(canvas, &coordinate_system_helper, dt) {
                         handle_new_grid_size(current_size, &renderer)
                     }
-                })
-                .is_err();
-            if error {
-                error!("Render failed. Closing");
-                return false;
-            }
-        }
+                })?;
 
-        true
+            Ok(true)
+        } else {
+            Ok(false)
+        }
     }
 }
 
@@ -443,7 +439,13 @@ pub fn ui_loop() {
         .event_pump()
         .expect("Could not create sdl event pump");
 
+    let mut was_animating = false;
+    let mut previous_frame_start = Instant::now();
     loop {
+        if !BRIDGE.running.load(Ordering::Relaxed) {
+            break;
+        }
+
         let frame_start = Instant::now();
 
         window.synchronize_settings();
@@ -489,16 +491,30 @@ pub fn ui_loop() {
             window.handle_keyboard_input(keycode, keytext);
         }
 
-        if !window.draw_frame() {
-            break;
+        let refresh_rate = { SETTINGS.get::<WindowSettings>().refresh_rate as f32 };
+        let dt = if was_animating {
+            previous_frame_start.elapsed().as_secs_f32()
+        } else {
+            1.0 / refresh_rate
+        };
+
+        match window.draw_frame(dt) {
+            Ok(animating) => {
+                was_animating = animating;
+            },
+            Err(error) => {
+                error!("Render failed: {}", error);
+                break;
+            }
         }
 
         let elapsed = frame_start.elapsed();
-        let refresh_rate = { SETTINGS.get::<WindowSettings>().refresh_rate as f32 };
-        let frame_length = Duration::from_secs_f32(1.0 / refresh_rate);
+        let expected_frame_length = Duration::from_secs_f32(1.0 / refresh_rate);
+
+        previous_frame_start = frame_start;
 
-        if elapsed < frame_length {
-            sleep(frame_length - elapsed);
+        if elapsed < expected_frame_length {
+            sleep(expected_frame_length - elapsed);
         }
     }