Skip to main content

yog_ui/
lib.rs

1//! yog-ui — retained-mode UI framework for Yog mods.
2//!
3//! Flexbox-inspired layout engine and a WPF-style [`grid`](widget::grid)
4//! (rows/columns: pixels, `Auto`, `*`) + GPU rendering via [`yog-gfx`].
5//! Use for custom inventories, guide books, tooltips, HUD overlays;
6//! [`surface`]s draw whatever the widgets don't (any `yog-gfx` call).
7//!
8//! # Quick start
9//! ```ignore
10//! use yog_ui::{UiRoot, widget, Align, FlexDir, Units};
11//!
12//! let ui = UiRoot::new("mymod:main_menu")
13//!     .style(|s| s.bg(0x88332211).padding(8.0, 8.0, 8.0, 8.0))
14//!     .child(
15//!         widget::panel(FlexDir::Column).gap(4.0)
16//!             .child(widget::label("Hello, World!").color(0xFF_DDAA00))
17//!             .child(widget::button("Click me").on_click("mymod:btn_click"))
18//!     );
19//! ```
20
21pub mod component;
22pub mod event;
23pub mod image;
24pub mod layout;
25mod render;
26pub mod rich;
27pub mod scroll;
28#[cfg(test)]
29mod state_tests;
30pub mod surface;
31pub mod theme;
32pub mod text;
33pub mod textbox;
34pub mod widget;
35
36pub use event::UiEvent;
37pub use layout::{Align, FlexDir, GridLines, LayoutNode, Rect, hit_path, node_at, set_focus};
38pub use yog_gfx::{Color, Vec2};
39pub use scroll::ScrollState;
40pub use surface::{Pointer, Surface};
41pub use textbox::{TextBoxEvent, TextBoxState};
42pub use widget::{
43    Action, Cell, Dock, FocusStyle, GridLength, GridSpec, HAlign, StateFlags, StateStyles, Stretch,
44    StylePatch, Track, Units, VAlign, Widget,
45};
46
47use yog_gfx::GfxContext;
48
49/// Top-level UI tree.  Build it, call [`layout`], then [`render`] each frame.
50pub struct UiRoot {
51    pub id: String,
52    pub root: Widget,
53    pub layout_root: LayoutNode,
54    pub needs_layout: bool,
55}
56
57impl UiRoot {
58    pub fn new(id: impl Into<String>, root: Widget) -> Self {
59        Self {
60            id: id.into(),
61            root,
62            layout_root: LayoutNode::default(),
63            needs_layout: true,
64        }
65    }
66
67    /// Recalculate layout. Call after changing the tree or on window resize.
68    pub fn layout(&mut self, screen_w: f32, screen_h: f32) {
69        self.layout_root = layout::compute(&self.root, screen_w, screen_h);
70        self.needs_layout = false;
71    }
72
73    /// Render the UI tree via `yog-gfx` draw2d.
74    /// Must be called from `on_hud_render`.
75    pub fn render(&self, ctx: &GfxContext) {
76        text::use_host_metrics(ctx);
77        image::use_host_images(ctx);
78        let d2d = ctx.draw2d();
79        render::render_node(ctx, &d2d, &self.root, &self.layout_root);
80    }
81
82    /// The tooltip of the innermost widget at `(mx, my)` that has one.
83    /// Call after [`Self::layout`].
84    pub fn tooltip_at(&self, mx: f32, my: f32) -> Option<&str> {
85        let path = layout::deepest_path(&self.layout_root, mx, my)?;
86        let mut w = &self.root;
87        let mut found = w.tooltip.as_deref();
88        for &i in &path {
89            let Some(c) = w.children.get(i as usize) else { break };
90            w = c;
91            found = w.tooltip.as_deref().or(found);
92        }
93        found
94    }
95
96    /// Draws `text` as a vanilla tooltip next to the cursor at `(mx, my)`,
97    /// kept inside the screen. Call after [`Self::render`] so it's on top.
98    pub fn render_tooltip(ctx: &GfxContext, text: &str, mx: f32, my: f32) {
99        text::use_host_metrics(ctx);
100        let (sw, sh) = ctx.screen_size();
101        render::tooltip(&ctx.draw2d(), text, mx, my, sw as f32, sh as f32);
102    }
103
104    /// Find the deepest clickable widget at `(mx, my)` in screen coordinates.
105    pub fn hit_test(&self, mx: f32, my: f32) -> Option<&LayoutNode> {
106        layout::hit_test(&self.layout_root, mx, my)
107    }
108
109    /// Resolved rect of the (first) node tagged `id` — see [`layout::find_by_id`].
110    /// Call after [`Self::layout`].
111    pub fn find_rect(&self, id: &str) -> Option<Rect> {
112        layout::find_by_id(&self.layout_root, id)
113    }
114
115    /// Laid-out node of the (first) widget tagged `id` — rect plus scroll
116    /// state, see [`layout::find_node`]. Call after [`Self::layout`].
117    pub fn find_node(&self, id: &str) -> Option<&LayoutNode> {
118        layout::find_node(&self.layout_root, id)
119    }
120}