wayvr/wlx-overlay-s/src/gui/panel.rs

use std::sync::Arc;

use glam::{Vec2, vec2};
use vulkano::{command_buffer::CommandBufferUsage, image::view::ImageView};
use wgui::{
    event::{
        Event as WguiEvent, MouseButton, MouseDownEvent, MouseLeaveEvent, MouseMotionEvent,
        MouseUpEvent, MouseWheelEvent,
    },
    layout::Layout,
    renderer_vk::context::Context as WguiContext,
};

use crate::{
    backend::{
        input::{Haptics, InteractionHandler, PointerHit, PointerMode},
        overlay::{FrameMeta, OverlayBackend, OverlayRenderer, ShouldRender},
    },
    graphics::{CommandBuffers, ExtentExt},
    gui,
    state::AppState,
};

use super::{asset::GuiAsset, timestep::Timestep};

const MAX_SIZE: u32 = 2048;
const MAX_SIZE_VEC2: Vec2 = vec2(MAX_SIZE as _, MAX_SIZE as _);

pub struct GuiPanel {
    pub layout: Layout,
    context: WguiContext,
    timestep: Timestep,
}

impl GuiPanel {
    pub fn new_from_template(app: &AppState, path: &str) -> anyhow::Result<Self> {
        let (layout, _state) =
            wgui::parser::new_layout_from_assets(Box::new(gui::asset::GuiAsset {}), path)?;

        let context = WguiContext::new(app.gfx.clone(), app.gfx.surface_format, 1.0)?;
        let mut timestep = Timestep::new();
        timestep.set_tps(60.0);

        Ok(Self {
            layout,
            context,
            timestep,
        })
    }

    pub fn new_blank(app: &AppState) -> anyhow::Result<Self> {
        let layout = Layout::new(Box::new(GuiAsset {}))?;
        let context = WguiContext::new(app.gfx.clone(), app.gfx.surface_format, 1.0)?;
        let mut timestep = Timestep::new();
        timestep.set_tps(60.0);

        Ok(Self {
            layout,
            context,
            timestep,
        })
    }

    pub fn update_layout(&mut self) -> anyhow::Result<()> {
        self.layout.update(MAX_SIZE_VEC2, 0.0)
    }
}

impl OverlayBackend for GuiPanel {
    fn set_renderer(&mut self, _: Box<dyn OverlayRenderer>) {
        log::debug!("Attempted to replace renderer on GuiPanel!");
    }
    fn set_interaction(&mut self, _: Box<dyn InteractionHandler>) {
        log::debug!("Attempted to replace interaction layer on GuiPanel!");
    }
}

impl InteractionHandler for GuiPanel {
    fn on_scroll(&mut self, _app: &mut AppState, hit: &PointerHit, delta_y: f32, delta_x: f32) {
        self.layout
            .push_event(&WguiEvent::MouseWheel(MouseWheelEvent {
                shift: vec2(delta_x, delta_y),
                pos: hit.uv * self.layout.content_size,
                device: hit.pointer,
            }))
            .unwrap();
    }

    fn on_hover(&mut self, _app: &mut AppState, hit: &PointerHit) -> Option<Haptics> {
        self.layout
            .push_event(&WguiEvent::MouseMotion(MouseMotionEvent {
                pos: hit.uv * self.layout.content_size,
                device: hit.pointer,
            }))
            .unwrap();

        None
    }

    fn on_left(&mut self, _app: &mut AppState, pointer: usize) {
        self.layout
            .push_event(&WguiEvent::MouseLeave(MouseLeaveEvent { device: pointer }))
            .unwrap();
    }

    fn on_pointer(&mut self, _app: &mut AppState, hit: &PointerHit, pressed: bool) {
        let button = match hit.mode {
            PointerMode::Left => MouseButton::Left,
            PointerMode::Right => MouseButton::Right,
            PointerMode::Middle => MouseButton::Middle,
            _ => return,
        };

        if pressed {
            self.layout
                .push_event(&WguiEvent::MouseDown(MouseDownEvent {
                    pos: hit.uv * self.layout.content_size,
                    button,
                    device: hit.pointer,
                }))
                .unwrap();
        } else {
            self.layout
                .push_event(&WguiEvent::MouseUp(MouseUpEvent {
                    pos: hit.uv * self.layout.content_size,
                    button,
                    device: hit.pointer,
                }))
                .unwrap();
        }
    }
}

impl OverlayRenderer for GuiPanel {
    fn init(&mut self, _app: &mut AppState) -> anyhow::Result<()> {
        if self.layout.content_size.x * self.layout.content_size.y == 0.0 {
            self.update_layout()?;
        }
        Ok(())
    }

    fn pause(&mut self, _app: &mut AppState) -> anyhow::Result<()> {
        Ok(())
    }

    fn resume(&mut self, _app: &mut AppState) -> anyhow::Result<()> {
        self.timestep.reset();
        Ok(())
    }

    fn should_render(&mut self, _app: &mut AppState) -> anyhow::Result<ShouldRender> {
        while self.timestep.on_tick() {
            self.layout.tick()?;
        }

        if self.layout.content_size.x * self.layout.content_size.y == 0.0 {
            return Ok(ShouldRender::Unable);
        }

        Ok(if self.layout.check_toggle_needs_redraw() {
            ShouldRender::Should
        } else {
            ShouldRender::Can
        })
    }

    fn render(
        &mut self,
        app: &mut AppState,
        tgt: Arc<ImageView>,
        buf: &mut CommandBuffers,
        _alpha: f32,
    ) -> anyhow::Result<bool> {
        self.context.update_viewport(tgt.extent_u32arr(), 1.0)?;
        self.layout.update(MAX_SIZE_VEC2, self.timestep.alpha)?;

        let mut cmd_buf = app
            .gfx
            .create_gfx_command_buffer(CommandBufferUsage::OneTimeSubmit)
            .unwrap();

        cmd_buf.begin_rendering(tgt)?;
        let primitives = wgui::drawing::draw(&self.layout)?;
        self.context.draw(&app.gfx, &mut cmd_buf, &primitives)?;
        cmd_buf.end_rendering()?;
        buf.push(cmd_buf.build()?);

        Ok(true)
    }

    fn frame_meta(&mut self) -> Option<FrameMeta> {
        Some(FrameMeta {
            extent: [
                MAX_SIZE.min(self.layout.content_size.x as _),
                MAX_SIZE.min(self.layout.content_size.y as _),
                1,
            ],
            ..Default::default()
        })
    }
}