bevy/src/ui/ui_update_system.rs

use crate::{legion::system::SubWorld, math::Vec2, ui::Node, *};
use winit::window::Window;

pub fn build_ui_update_system() -> Box<dyn Schedulable> {
    SystemBuilder::new("ui_update_system")
        .read_resource::<Window>()
        .with_query(<(Write<Node>,)>::query().filter(!component::<Parent>()))
        .write_component::<Node>()
        .read_component::<Children>()
        .build(move |_, world, window, node_query| {
            let window_size = window.inner_size();
            let parent_size = math::vec2(window_size.width as f32, window_size.height as f32);
            let parent_position = math::vec2(0.0, 0.0);
            for (entity, _) in node_query.iter_entities_mut(world) {
                ecs::run_on_hierarchy_subworld(
                    world,
                    entity,
                    (parent_size, parent_position),
                    &update_node_entity,
                );
            }
        })
}

fn update_node_entity(
    world: &mut SubWorld,
    entity: Entity,
    parent_properties: (Vec2, Vec2),
) -> Option<(Vec2, Vec2)> {
    let (parent_size, parent_position) = parent_properties;
    if let Some(mut node) = world.get_component_mut::<Node>(entity) {
        node.update(parent_size, parent_position);
        return Some((node.size, node.global_position));
    }

    None
}