tessera-ui-basic-components 2.7.0

//! A component for creating a frosted/distorted glass visual effect.
//!
//! ## Usage
//!
//! Use as a background for buttons, panels, or other UI elements.
use std::sync::Arc;

use derive_builder::Builder;
use tessera_ui::{
    BarrierRequirement, Color, ComputedData, Constraint, CursorEventContent, DimensionValue, Dp,
    GestureState, PressKeyEventType, Px, PxPosition,
    accesskit::{Action, Role},
    renderer::DrawCommand,
    tessera,
    winit::window::CursorIcon,
};

use crate::{
    padding_utils::remove_padding_from_dimension,
    pipelines::{blur::command::DualBlurCommand, contrast::ContrastCommand, mean::MeanCommand},
    pos_misc::is_position_in_component,
    ripple_state::RippleState,
    shape_def::Shape,
};

#[derive(Clone, Copy, Debug, Default, PartialEq)]
pub struct GlassBorder {
    pub width: Px,
}

impl GlassBorder {
    pub fn new(width: Px) -> Self {
        Self { width }
    }
}

/// Arguments for the `fluid_glass` component, providing extensive control over its appearance.
///
/// This struct uses the builder pattern for easy construction.
#[derive(Builder, Clone)]
#[builder(build_fn(validate = "Self::validate"), pattern = "owned", setter(into))]
pub struct FluidGlassArgs {
    /// The tint color of the glass.
    /// The alpha channel uniquely and directly controls the tint strength.
    /// `A=0.0` means no tint (100% background visibility).
    /// `A=1.0` means full tint (100% color visibility).
    #[builder(default = "Color::TRANSPARENT")]
    pub tint_color: Color,
    /// The shape of the component, an enum that can be `RoundedRectangle` or `Ellipse`.
    #[builder(
        default = "Shape::RoundedRectangle { top_left: Dp(25.0), top_right: Dp(25.0), bottom_right: Dp(25.0), bottom_left: Dp(25.0), g2_k_value: 3.0 }"
    )]
    pub shape: Shape,
    /// The radius for the background blur effect. A value of `0.0` disables the blur.
    #[builder(default = "Dp(0.0)")]
    pub blur_radius: Dp,
    /// The height of the chromatic dispersion effect.
    #[builder(default = "Dp(25.0)")]
    pub dispersion_height: Dp,
    /// Multiplier for the chromatic aberration, enhancing the color separation effect.
    #[builder(default = "1.1")]
    pub chroma_multiplier: f32,
    /// The height of the refraction effect, simulating light bending through the glass.
    #[builder(default = "Dp(24.0)")]
    pub refraction_height: Dp,
    /// The amount of refraction to apply.
    #[builder(default = "32.0")]
    pub refraction_amount: f32,
    /// Controls the shape and eccentricity of the highlight.
    #[builder(default = "0.2")]
    pub eccentric_factor: f32,
    /// The amount of noise to apply over the surface, adding texture.
    #[builder(default = "0.0")]
    pub noise_amount: f32,
    /// The scale of the noise pattern.
    #[builder(default = "1.0")]
    pub noise_scale: f32,
    /// A time value, typically used to animate the noise or other effects.
    #[builder(default = "0.0")]
    pub time: f32,
    /// The contrast adjustment factor.
    #[builder(default, setter(strip_option))]
    pub contrast: Option<f32>,
    /// The optional width of the component, defined as a `DimensionValue`.
    #[builder(default = "DimensionValue::WRAP", setter(into))]
    pub width: DimensionValue,
    /// The optional height of the component, defined as a `DimensionValue`.
    #[builder(default = "DimensionValue::WRAP", setter(into))]
    pub height: DimensionValue,
    /// Padding inside the glass component.
    #[builder(default = "Dp(0.0)")]
    pub padding: Dp,
    // Ripple effect properties
    #[builder(default, setter(strip_option))]
    pub ripple_center: Option<[f32; 2]>,
    #[builder(default, setter(strip_option))]
    pub ripple_radius: Option<f32>,
    #[builder(default, setter(strip_option))]
    pub ripple_alpha: Option<f32>,
    #[builder(default, setter(strip_option))]
    pub ripple_strength: Option<f32>,

    #[builder(default, setter(strip_option, into = false))]
    pub on_click: Option<Arc<dyn Fn() + Send + Sync>>,

    #[builder(default = "Some(GlassBorder { width: Dp(1.35).into() })")]
    pub border: Option<GlassBorder>,

    /// Whether to block input events on the glass surface.
    /// When `true`, the surface will consume all input events, preventing interaction with underlying components.
    #[builder(default = "false")]
    pub block_input: bool,
    /// Optional accessibility role override; defaults to `Role::Button` when interactive.
    #[builder(default, setter(strip_option))]
    pub accessibility_role: Option<Role>,
    /// Optional label announced by assistive technologies.
    #[builder(default, setter(strip_option, into))]
    pub accessibility_label: Option<String>,
    /// Optional description announced by assistive technologies.
    #[builder(default, setter(strip_option, into))]
    pub accessibility_description: Option<String>,
    /// Whether the surface should be focusable even when not interactive.
    #[builder(default)]
    pub accessibility_focusable: bool,
}

impl PartialEq for FluidGlassArgs {
    fn eq(&self, other: &Self) -> bool {
        self.tint_color == other.tint_color
            && self.shape == other.shape
            && self.blur_radius == other.blur_radius
            && self.dispersion_height == other.dispersion_height
            && self.chroma_multiplier == other.chroma_multiplier
            && self.refraction_height == other.refraction_height
            && self.refraction_amount == other.refraction_amount
            && self.eccentric_factor == other.eccentric_factor
            && self.noise_amount == other.noise_amount
            && self.noise_scale == other.noise_scale
            && self.time == other.time
            && self.contrast == other.contrast
            && self.width == other.width
            && self.height == other.height
            && self.padding == other.padding
            && self.ripple_center == other.ripple_center
            && self.ripple_radius == other.ripple_radius
            && self.ripple_alpha == other.ripple_alpha
            && self.ripple_strength == other.ripple_strength
            && self.border == other.border
            && self.block_input == other.block_input
    }
}

impl FluidGlassArgsBuilder {
    fn validate(&self) -> Result<(), String> {
        Ok(())
    }
}

// Manual implementation of Default because derive_builder's default conflicts with our specific defaults
impl Default for FluidGlassArgs {
    fn default() -> Self {
        FluidGlassArgsBuilder::default().build().unwrap()
    }
}

#[derive(Clone, PartialEq)]
pub struct FluidGlassCommand {
    pub args: FluidGlassArgs,
}

impl DrawCommand for FluidGlassCommand {
    fn barrier(&self) -> Option<BarrierRequirement> {
        Some(BarrierRequirement::uniform_padding_local(Px(10)))
    }
}

// Helper: input handler logic extracted to reduce complexity of `fluid_glass`
// These helpers operate on the injected InputHandlerInput type from the core crate.
fn handle_click_state(
    args: &FluidGlassArgs,
    ripple_state: Option<RippleState>,
    on_click: Arc<dyn Fn() + Send + Sync>,
    input: &mut tessera_ui::InputHandlerInput,
) {
    let size = input.computed_data;
    let cursor_pos_option = input.cursor_position_rel;
    let is_cursor_in = cursor_pos_option
        .map(|pos| is_position_in_component(size, pos))
        .unwrap_or(false);

    if is_cursor_in {
        input.requests.cursor_icon = CursorIcon::Pointer;

        if let Some(_event) = input.cursor_events.iter().find(|e| {
            e.gesture_state == GestureState::TapCandidate
                && matches!(
                    e.content,
                    CursorEventContent::Released(PressKeyEventType::Left)
                )
        }) {
            if let Some(ripple_state) = &ripple_state
                && let Some(pos) = input.cursor_position_rel
            {
                let size = input.computed_data;
                let normalized_pos = [
                    pos.x.to_f32() / size.width.to_f32(),
                    pos.y.to_f32() / size.height.to_f32(),
                ];
                ripple_state.start_animation(normalized_pos);
            }
            on_click();
        }

        if args.block_input {
            // Consume all input events to prevent interaction with underlying components
            input.block_all();
        }
    }
}

fn handle_block_input(input: &mut tessera_ui::InputHandlerInput) {
    let size = input.computed_data;
    let cursor_pos_option = input.cursor_position_rel;
    let is_cursor_in = cursor_pos_option
        .map(|pos| is_position_in_component(size, pos))
        .unwrap_or(false);

    if is_cursor_in {
        // Consume all input events to prevent interaction with underlying components
        input.block_all();
    }
}

fn apply_fluid_glass_accessibility(
    input: &mut tessera_ui::InputHandlerInput<'_>,
    args: &FluidGlassArgs,
    on_click: &Option<Arc<dyn Fn() + Send + Sync>>,
) {
    let interactive = on_click.is_some();
    let has_metadata = interactive
        || args.accessibility_role.is_some()
        || args.accessibility_label.is_some()
        || args.accessibility_description.is_some()
        || args.accessibility_focusable;

    if !has_metadata {
        return;
    }

    let mut builder = input.accessibility();

    let role = args
        .accessibility_role
        .or_else(|| interactive.then_some(Role::Button));
    if let Some(role) = role {
        builder = builder.role(role);
    }
    if let Some(label) = args.accessibility_label.as_ref() {
        builder = builder.label(label.clone());
    }
    if let Some(description) = args.accessibility_description.as_ref() {
        builder = builder.description(description.clone());
    }
    if args.accessibility_focusable || interactive {
        builder = builder.focusable();
    }
    if interactive {
        builder = builder.action(Action::Click);
    }

    builder.commit();

    if interactive && let Some(on_click) = on_click.clone() {
        input.set_accessibility_action_handler(move |action| {
            if action == Action::Click {
                on_click();
            }
        });
    }
}

/// # fluid_glass
///
/// Renders a highly customizable surface with blur, tint, and other glass-like effects.
///
/// ## Usage
///
/// Use to create a dynamic, layered background for other components.
///
/// ## Parameters
///
/// - `args` — configures the glass effect's appearance; see [`FluidGlassArgs`].
/// - `ripple_state` — an optional [`RippleState`] to manage a ripple animation on interaction.
/// - `child` — a closure that renders content on top of the glass surface.
///
/// ## Examples
///
/// ```
/// use tessera_ui_basic_components::{
///     fluid_glass::{fluid_glass, FluidGlassArgs},
///     text::{text, TextArgsBuilder},
/// };
///
/// fluid_glass(FluidGlassArgs::default(), None, || {
///     text(TextArgsBuilder::default().text("Content on glass".to_string()).build().unwrap());
/// });
/// ```
#[tessera]
pub fn fluid_glass(
    mut args: FluidGlassArgs,
    ripple_state: Option<RippleState>,
    child: impl FnOnce(),
) {
    if let Some(ripple_state) = &ripple_state
        && let Some((progress, center)) = ripple_state.get_animation_progress()
    {
        args.ripple_center = Some(center);
        args.ripple_radius = Some(progress);
        args.ripple_alpha = Some((1.0 - progress) * 0.3);
        args.ripple_strength = Some(progress);
    }
    (child)();
    let args_measure_clone = args.clone();
    measure(Box::new(move |input| {
        let glass_intrinsic_width = args_measure_clone.width;
        let glass_intrinsic_height = args_measure_clone.height;
        let glass_intrinsic_constraint =
            Constraint::new(glass_intrinsic_width, glass_intrinsic_height);
        let effective_glass_constraint = glass_intrinsic_constraint.merge(input.parent_constraint);

        let child_constraint = Constraint::new(
            remove_padding_from_dimension(
                effective_glass_constraint.width,
                args_measure_clone.padding.into(),
            ),
            remove_padding_from_dimension(
                effective_glass_constraint.height,
                args_measure_clone.padding.into(),
            ),
        );

        let child_measurement = if !input.children_ids.is_empty() {
            let child_measurement =
                input.measure_child(input.children_ids[0], &child_constraint)?;
            input.place_child(
                input.children_ids[0],
                PxPosition {
                    x: args.padding.into(),
                    y: args.padding.into(),
                },
            );
            child_measurement
        } else {
            ComputedData {
                width: Px(0),
                height: Px(0),
            }
        };

        if args.blur_radius > Dp(0.0) {
            let blur_command =
                DualBlurCommand::horizontal_then_vertical(args.blur_radius.to_pixels_f32());
            let mut metadata = input.metadata_mut();
            metadata.push_compute_command(blur_command);
        }

        if let Some(contrast_value) = args.contrast
            && contrast_value != 1.0
        {
            let mean_command =
                MeanCommand::new(input.gpu, &mut input.compute_resource_manager.write());
            let contrast_command =
                ContrastCommand::new(contrast_value, mean_command.result_buffer_ref());
            let mut metadata = input.metadata_mut();
            metadata.push_compute_command(mean_command);
            metadata.push_compute_command(contrast_command);
        }

        let drawable = FluidGlassCommand {
            args: args_measure_clone.clone(),
        };

        input.metadata_mut().push_draw_command(drawable);

        let padding_px: Px = args_measure_clone.padding.into();
        let min_width = child_measurement.width + padding_px * 2;
        let min_height = child_measurement.height + padding_px * 2;
        let width = match effective_glass_constraint.width {
            DimensionValue::Fixed(value) => value,
            DimensionValue::Wrap { min, max } => min
                .unwrap_or(Px(0))
                .max(min_width)
                .min(max.unwrap_or(Px::MAX)),
            DimensionValue::Fill { min, max } => max
                .expect("Seems that you are trying to fill an infinite width, which is not allowed")
                .max(min_width)
                .max(min.unwrap_or(Px(0))),
        };
        let height = match effective_glass_constraint.height {
            DimensionValue::Fixed(value) => value,
            DimensionValue::Wrap { min, max } => min
                .unwrap_or(Px(0))
                .max(min_height)
                .min(max.unwrap_or(Px::MAX)),
            DimensionValue::Fill { min, max } => max
                .expect(
                    "Seems that you are trying to fill an infinite height, which is not allowed",
                )
                .max(min_height)
                .max(min.unwrap_or(Px(0))),
        };
        Ok(ComputedData { width, height })
    }));

    if let Some(ref on_click) = args.on_click {
        let ripple_state = ripple_state.clone();
        let on_click_arc = on_click.clone();
        let args_for_handler = args.clone();
        input_handler(Box::new(move |mut input: tessera_ui::InputHandlerInput| {
            // Apply accessibility first
            apply_fluid_glass_accessibility(
                &mut input,
                &args_for_handler,
                &args_for_handler.on_click,
            );
            // Then handle click state (which includes block_input logic)
            handle_click_state(
                &args_for_handler,
                ripple_state.clone(),
                on_click_arc.clone(),
                &mut input,
            );
        }));
    } else if args.block_input {
        let args_for_handler = args.clone();
        input_handler(Box::new(move |mut input: tessera_ui::InputHandlerInput| {
            // Apply accessibility first
            apply_fluid_glass_accessibility(&mut input, &args_for_handler, &None);
            // Then handle input blocking behavior
            handle_block_input(&mut input);
        }));
    } else {
        // Only accessibility metadata, no interaction
        let args_for_handler = args.clone();
        input_handler(Box::new(move |mut input: tessera_ui::InputHandlerInput| {
            apply_fluid_glass_accessibility(&mut input, &args_for_handler, &None);
        }));
    }
}