neovide/src/window.rs

use std::sync::Arc;
use std::sync::atomic::Ordering;
use std::time::{Duration, Instant};

use image::{load_from_memory, GenericImageView, Pixel};
use skulpin::{CoordinateSystem, RendererBuilder, PresentMode};
use skulpin::winit::dpi::{LogicalSize, LogicalPosition};
use skulpin::winit::event::{ElementState, Event, MouseScrollDelta, StartCause, WindowEvent, ModifiersState};
use skulpin::winit::event_loop::{ControlFlow, EventLoop};
use skulpin::winit::window::{Icon, WindowBuilder};
use log::{info, debug, trace, error};

use crate::bridge::{construct_keybinding_string, BRIDGE, UiCommand};
use crate::renderer::Renderer;
use crate::redraw_scheduler::REDRAW_SCHEDULER;
use crate::editor::EDITOR;
use crate::settings::SETTINGS;
use crate::INITIAL_DIMENSIONS;

#[derive(RustEmbed)]
#[folder = "assets/"]
struct Asset;

fn handle_new_grid_size(new_size: LogicalSize<f32>, renderer: &Renderer) {
    if new_size.width > 0.0 && new_size.height > 0.0 {
        let new_width = ((new_size.width + 1.0) / renderer.font_width) as i64;
        let new_height = ((new_size.height + 1.0) / renderer.font_height) as i64;
        // Add 1 here to make sure resizing doesn't change the grid size on startup
        BRIDGE.queue_command(UiCommand::Resize { width: new_width as i64, height: new_height as i64 });
    }
}

pub fn ui_loop() {
    let event_loop = EventLoop::<()>::with_user_event();

    let mut renderer = Renderer::new();
    let (width, height) = INITIAL_DIMENSIONS;
    let logical_size = LogicalSize::new(
        (width as f32 * renderer.font_width) as f64, 
        (height as f32 * renderer.font_height + 1.0) as f64
    );

    let icon = {
        let icon_data = Asset::get("nvim.ico").expect("Failed to read icon data");
        let icon = load_from_memory(&icon_data).expect("Failed to parse icon data");
        let (width, height) = icon.dimensions();
        let mut rgba = Vec::with_capacity((width * height) as usize * 4);
        for (_, _, pixel) in icon.pixels() {
            rgba.extend_from_slice(&pixel.to_rgba().0);
        }
        Icon::from_rgba(rgba, width, height).expect("Failed to create icon object")
    };
    info!("icon created");

    let mut title = "Neovide".to_string();
    let window = Arc::new(WindowBuilder::new()
        .with_title(&title)
        .with_inner_size(logical_size)
        .with_window_icon(Some(icon))
        .build(&event_loop)
        .expect("Failed to create window"));
    info!("window created");

    let mut skulpin_renderer = RendererBuilder::new()
        .prefer_integrated_gpu()
        .use_vulkan_debug_layer(true)
        .present_mode_priority(vec![PresentMode::Mailbox, PresentMode::Immediate])
        .coordinate_system(CoordinateSystem::Logical)
        .build(&window)
        .expect("Failed to create renderer");
    info!("renderer created");

    let mut mouse_down = false;
    let mut mouse_pos = (0, 0);

    let mut allow_next_char = false;
    let mut next_char_modifiers = ModifiersState::empty();

    info!("Starting window event loop");
    event_loop.run(move |event, _window_target, control_flow| {
        trace!("Window Event: {:?}", event);
        match event {
            Event::NewEvents(StartCause::Init) |
            Event::NewEvents(StartCause::ResumeTimeReached { .. }) => {
                window.request_redraw()
            },

            Event::WindowEvent {
                event: WindowEvent::CloseRequested,
                ..
            } => *control_flow = ControlFlow::Exit,

            Event::WindowEvent {
                event: WindowEvent::Resized(new_size),
                ..
            } => {
                handle_new_grid_size(new_size.to_logical(window.scale_factor()), &renderer)
            },

            Event::WindowEvent {
                event: WindowEvent::KeyboardInput {
                    input,
                    ..
                },
                ..
            } => {
                // Only interpret 'char' events when we get a previous event without a virtual
                // keycode (which we ignore for KeyboardInput events).
                // This is a hack so we don't lose a bunch of input events on Linux
                if input.virtual_keycode == None {
                    allow_next_char = true;
                }else {
                    allow_next_char = false;
                }
                next_char_modifiers = input.modifiers;

                if let Some(keybinding_string) = construct_keybinding_string(input)
                    .map(UiCommand::Keyboard) {
                        BRIDGE.queue_command(keybinding_string);
                }
            },

            Event::WindowEvent {
                event: WindowEvent::ReceivedCharacter(c),
                ..
            } => {
                if allow_next_char {
                    next_char_modifiers.remove(ModifiersState::SHIFT);
                    let keybinding = super::bridge::append_modifiers(next_char_modifiers, &c.to_string(), false);
                    BRIDGE.queue_command(UiCommand::Keyboard(keybinding));
                }
            },

            Event::WindowEvent {
                event: WindowEvent::CursorMoved {
                    position,
                    ..
                },
                ..
            } => {
                let position: LogicalPosition<f64> = position.to_logical(window.scale_factor());
                let grid_y = (position.x / renderer.font_width as f64) as i64;
                let grid_x = (position.y / renderer.font_height as f64) as i64;
                let (old_x, old_y) = mouse_pos;
                mouse_pos = (grid_x, grid_y);
                if mouse_down && (old_x != grid_x || old_y != grid_y) {
                    BRIDGE.queue_command(UiCommand::Drag(grid_x, grid_y));
                }
            }

            Event::WindowEvent {
                event: WindowEvent::MouseInput {
                    state,
                    ..
                },
                ..
            } => {
                let input_type = match (state, mouse_down) {
                    (ElementState::Pressed, false) => {
                        mouse_down = true;
                        Some("press")
                    },
                    (ElementState::Released, true) => {
                        mouse_down = false;
                        Some("release")
                    },
                    _ => None
                };

                if let Some(input_type) = input_type {
                    let (grid_x, grid_y) = mouse_pos;
                    BRIDGE.queue_command(UiCommand::MouseButton { action: input_type.to_string(), position: (grid_x, grid_y) });
                }
            }

            Event::WindowEvent {
                event: WindowEvent::MouseWheel {
                    delta: MouseScrollDelta::LineDelta(horizontal, vertical),
                    ..
                },
                ..
            } => {
                let vertical_input_type = if vertical > 0.0 {
                    Some("up")
                } else if vertical < 0.0 {
                    Some("down")
                } else {
                    None
                };

                if let Some(input_type) = vertical_input_type {
                    BRIDGE.queue_command(UiCommand::Scroll { direction: input_type.to_string(), position: mouse_pos });
                }

                let horizontal_input_type = if horizontal > 0.0 {
                    Some("right")
                } else if horizontal < 0.0 {
                    Some("left")
                } else {
                    None
                };

                if let Some(input_type) = horizontal_input_type {
                    BRIDGE.queue_command(UiCommand::Scroll { direction: input_type.to_string(), position: mouse_pos });
                }
            }

            Event::RedrawRequested { .. } => {
                let frame_start = Instant::now();

                let editor_title = { EDITOR.lock().title.clone() };
                if title != editor_title {
                    title = editor_title;
                    window.set_title(&title);
                }

                if REDRAW_SCHEDULER.should_draw() || SETTINGS.no_idle.load(Ordering::Relaxed) {
                    debug!("Render Triggered");
                    if skulpin_renderer.draw(&window, |canvas, coordinate_system_helper| {
                        if renderer.draw(canvas, coordinate_system_helper) {
                            handle_new_grid_size(window.inner_size().to_logical(window.scale_factor()), &renderer)
                        }
                    }).is_err() {
                        error!("Render failed. Closing");
                        *control_flow = ControlFlow::Exit;
                        return;
                    }
                }

                *control_flow = ControlFlow::WaitUntil(frame_start + Duration::from_secs_f32(1.0 / 60.0));
            },

            _ => {}
        }
    })
}