bevy/crates/bevy_ui/src/entity.rs

use super::Node;
use crate::{render::UI_PIPELINE_HANDLE, widget::Label};
use bevy_asset::Handle;
use bevy_derive::EntityArchetype;
use bevy_render::{draw::{Draw, RenderPipelines}, mesh::Mesh};
use bevy_sprite::{ColorMaterial, Quad, QUAD_HANDLE};

#[derive(EntityArchetype)]
pub struct UiEntity {
    pub node: Node,
    pub quad: Quad,
    pub mesh: Handle<Mesh>, // TODO: maybe abstract this out
    pub material: Handle<ColorMaterial>,
    pub draw: Draw,
    pub render_pipelines: RenderPipelines,
}

impl Default for UiEntity {
    fn default() -> Self {
        UiEntity {
            node: Default::default(),
            quad: Default::default(),
            mesh: QUAD_HANDLE,
            material: Default::default(),
            draw: Default::default(),
            render_pipelines: RenderPipelines {
                pipelines: vec![UI_PIPELINE_HANDLE],
                ..Default::default()
            },
        }
    }
}

#[derive(EntityArchetype)]
pub struct LabelEntity {
    pub node: Node,
    pub quad: Quad,
    pub mesh: Handle<Mesh>, // TODO: maybe abstract this out
    pub material: Handle<ColorMaterial>,
    pub draw: Draw,
    pub render_pipelines: RenderPipelines,
    pub label: Label,
}

impl Default for LabelEntity {
    fn default() -> Self {
        LabelEntity {
            node: Default::default(),
            quad: Default::default(),
            mesh: QUAD_HANDLE,
            // NOTE: labels each get their own material.
            material: Handle::new(), // TODO: maybe abstract this out
            draw: Default::default(),
            render_pipelines: RenderPipelines {
                pipelines: vec![UI_PIPELINE_HANDLE],
                ..Default::default()
            },
            label: Label::default(),
        }
    }
}