use crate::{
brush::Brush,
core::{algebra::Vector2, color::Color, math::Rect, pool::Handle, scope_profile},
define_constructor,
draw::{CommandTexture, Draw, DrawingContext},
message::{MessageDirection, UiMessage},
widget::{Widget, WidgetBuilder},
BuildContext, Control, UiNode, UserInterface,
};
use std::{
any::{Any, TypeId},
ops::{Deref, DerefMut},
};
#[derive(Debug, Clone, PartialEq)]
pub enum ScrollPanelMessage {
VerticalScroll(f32),
HorizontalScroll(f32),
BringIntoView(Handle<UiNode>),
}
impl ScrollPanelMessage {
define_constructor!(ScrollPanelMessage:VerticalScroll => fn vertical_scroll(f32), layout: false);
define_constructor!(ScrollPanelMessage:HorizontalScroll => fn horizontal_scroll(f32), layout: false);
define_constructor!(ScrollPanelMessage:BringIntoView => fn bring_into_view(Handle<UiNode>), layout: true);
}
#[derive(Clone)]
pub struct ScrollPanel {
widget: Widget,
scroll: Vector2<f32>,
vertical_scroll_allowed: bool,
horizontal_scroll_allowed: bool,
}
crate::define_widget_deref!(ScrollPanel);
impl Control for ScrollPanel {
fn query_component(&self, type_id: TypeId) -> Option<&dyn Any> {
if type_id == TypeId::of::<Self>() {
Some(self)
} else {
None
}
}
fn measure_override(&self, ui: &UserInterface, available_size: Vector2<f32>) -> Vector2<f32> {
scope_profile!();
let size_for_child = Vector2::new(
if self.horizontal_scroll_allowed {
f32::INFINITY
} else {
available_size.x
},
if self.vertical_scroll_allowed {
f32::INFINITY
} else {
available_size.y
},
);
let mut desired_size = Vector2::default();
for child_handle in self.widget.children() {
ui.measure_node(*child_handle, size_for_child);
let child = ui.nodes.borrow(*child_handle);
let child_desired_size = child.desired_size();
if child_desired_size.x > desired_size.x {
desired_size.x = child_desired_size.x;
}
if child_desired_size.y > desired_size.y {
desired_size.y = child_desired_size.y;
}
}
desired_size
}
fn arrange_override(&self, ui: &UserInterface, final_size: Vector2<f32>) -> Vector2<f32> {
scope_profile!();
let mut children_size = Vector2::<f32>::default();
for child_handle in self.widget.children() {
let desired_size = ui.node(*child_handle).desired_size();
children_size.x = children_size.x.max(desired_size.x);
children_size.y = children_size.y.max(desired_size.y);
}
let child_rect = Rect::new(
-self.scroll.x,
-self.scroll.y,
if self.horizontal_scroll_allowed {
children_size.x.max(final_size.x)
} else {
final_size.x
},
if self.vertical_scroll_allowed {
children_size.y.max(final_size.y)
} else {
final_size.y
},
);
for child_handle in self.widget.children() {
ui.arrange_node(*child_handle, &child_rect);
}
final_size
}
fn draw(&self, drawing_context: &mut DrawingContext) {
drawing_context.push_rect_filled(&self.widget.screen_bounds(), None);
drawing_context.commit(
self.clip_bounds(),
Brush::Solid(Color::TRANSPARENT),
CommandTexture::None,
None,
);
}
fn handle_routed_message(&mut self, ui: &mut UserInterface, message: &mut UiMessage) {
self.widget.handle_routed_message(ui, message);
if message.destination() == self.handle() {
if let Some(msg) = message.data::<ScrollPanelMessage>() {
match *msg {
ScrollPanelMessage::VerticalScroll(scroll) => {
self.scroll.y = scroll;
self.invalidate_arrange();
}
ScrollPanelMessage::HorizontalScroll(scroll) => {
self.scroll.x = scroll;
self.invalidate_arrange();
}
ScrollPanelMessage::BringIntoView(handle) => {
if let Some(node_to_focus_ref) = ui.try_get_node(handle) {
let size = node_to_focus_ref.actual_size();
let mut parent = handle;
let mut relative_position = Vector2::default();
while parent.is_some() && parent != self.handle {
let node = ui.node(parent);
relative_position += node.actual_local_position();
parent = node.parent();
}
if relative_position.x < 0.0
|| relative_position.y < 0.0
|| relative_position.x > self.actual_size().x
|| relative_position.y > self.actual_size().y
|| (relative_position.x + size.x) > self.actual_size().x
|| (relative_position.y + size.y) > self.actual_size().y
{
relative_position += self.scroll;
if parent == self.handle {
if self.vertical_scroll_allowed {
ui.send_message(ScrollPanelMessage::vertical_scroll(
self.handle,
MessageDirection::ToWidget,
relative_position.y,
));
}
if self.horizontal_scroll_allowed {
ui.send_message(ScrollPanelMessage::horizontal_scroll(
self.handle,
MessageDirection::ToWidget,
relative_position.x,
));
}
}
}
}
}
}
}
}
}
}
impl ScrollPanel {
pub fn new(widget: Widget) -> Self {
Self {
widget,
scroll: Default::default(),
vertical_scroll_allowed: true,
horizontal_scroll_allowed: false,
}
}
}
pub struct ScrollPanelBuilder {
widget_builder: WidgetBuilder,
vertical_scroll_allowed: Option<bool>,
horizontal_scroll_allowed: Option<bool>,
}
impl ScrollPanelBuilder {
pub fn new(widget_builder: WidgetBuilder) -> Self {
Self {
widget_builder,
vertical_scroll_allowed: None,
horizontal_scroll_allowed: None,
}
}
pub fn with_vertical_scroll_allowed(mut self, value: bool) -> Self {
self.vertical_scroll_allowed = Some(value);
self
}
pub fn with_horizontal_scroll_allowed(mut self, value: bool) -> Self {
self.horizontal_scroll_allowed = Some(value);
self
}
pub fn build(self, ui: &mut BuildContext) -> Handle<UiNode> {
ui.add_node(UiNode::new(ScrollPanel {
widget: self.widget_builder.build(),
scroll: Vector2::default(),
vertical_scroll_allowed: self.vertical_scroll_allowed.unwrap_or(true),
horizontal_scroll_allowed: self.horizontal_scroll_allowed.unwrap_or(false),
}))
}
}