Skip to content

Commit

Permalink
Upgrade to Taffy 0.2 (bevyengine#6743)
Browse files Browse the repository at this point in the history
# Objective

Upgrade to Taffy 0.2

## Solution

Do it

## Changelog

Upgraded to Taffy 0.2, improving UI layout performance significantly and adding the flexbox `gap` property and `AlignContent::SpaceEvenly`.

## Notes

`many_buttons` is 8% faster! speed improvements for more highly nested UIs will be much more dramatic. Great work, Team Taffy.
  • Loading branch information
rparrett authored and alradish committed Jan 22, 2023
1 parent 922a9a9 commit 4b1ed7d
Show file tree
Hide file tree
Showing 4 changed files with 28 additions and 23 deletions.
2 changes: 1 addition & 1 deletion crates/bevy_ui/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@ bevy_window = { path = "../bevy_window", version = "0.9.1" }
bevy_utils = { path = "../bevy_utils", version = "0.9.1" }

# other
taffy = "0.1.0"
taffy = "0.2.2"
serde = { version = "1", features = ["derive"] }
smallvec = { version = "1.6", features = ["union", "const_generics"] }
bytemuck = { version = "1.5", features = ["derive"] }
Expand Down
10 changes: 4 additions & 6 deletions crates/bevy_ui/src/flex/convert.rs
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,8 @@ pub fn from_rect(
rect: UiRect,
) -> taffy::geometry::Rect<taffy::style::Dimension> {
taffy::geometry::Rect {
start: from_val(scale_factor, rect.left),
end: from_val(scale_factor, rect.right),
left: from_val(scale_factor, rect.left),
right: from_val(scale_factor, rect.right),
top: from_val(scale_factor, rect.top),
bottom: from_val(scale_factor, rect.bottom),
}
Expand Down Expand Up @@ -52,10 +52,8 @@ pub fn from_style(scale_factor: f64, value: &Style) -> taffy::style::Style {
size: from_val_size(scale_factor, value.size),
min_size: from_val_size(scale_factor, value.min_size),
max_size: from_val_size(scale_factor, value.max_size),
aspect_ratio: match value.aspect_ratio {
Some(value) => taffy::number::Number::Defined(value),
None => taffy::number::Number::Undefined,
},
aspect_ratio: value.aspect_ratio,
gap: from_val_size(scale_factor, value.gap),
}
}

Expand Down
34 changes: 18 additions & 16 deletions crates/bevy_ui/src/flex/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,10 @@ use bevy_transform::components::Transform;
use bevy_utils::HashMap;
use bevy_window::{Window, WindowId, WindowScaleFactorChanged, Windows};
use std::fmt;
use taffy::{number::Number, Taffy};
use taffy::{
prelude::{AvailableSpace, Size},
Taffy,
};

#[derive(Resource)]
pub struct FlexSurface {
Expand Down Expand Up @@ -63,7 +66,7 @@ impl FlexSurface {
let taffy_style = convert::from_style(scale_factor, style);
let taffy_node = self.entity_to_taffy.entry(entity).or_insert_with(|| {
added = true;
taffy.new_node(taffy_style, &Vec::new()).unwrap()
taffy.new_leaf(taffy_style).unwrap()
});

if !added {
Expand All @@ -81,23 +84,23 @@ impl FlexSurface {
let taffy = &mut self.taffy;
let taffy_style = convert::from_style(scale_factor, style);
let measure = taffy::node::MeasureFunc::Boxed(Box::new(
move |constraints: taffy::geometry::Size<Number>| {
move |constraints: Size<Option<f32>>, _available: Size<AvailableSpace>| {
let mut size = convert::from_f32_size(scale_factor, calculated_size.size);
match (constraints.width, constraints.height) {
(Number::Undefined, Number::Undefined) => {}
(Number::Defined(width), Number::Undefined) => {
(None, None) => {}
(Some(width), None) => {
if calculated_size.preserve_aspect_ratio {
size.height = width * size.height / size.width;
}
size.width = width;
}
(Number::Undefined, Number::Defined(height)) => {
(None, Some(height)) => {
if calculated_size.preserve_aspect_ratio {
size.width = height * size.width / size.height;
}
size.height = height;
}
(Number::Defined(width), Number::Defined(height)) => {
(Some(width), Some(height)) => {
size.width = width;
size.height = height;
}
Expand All @@ -110,7 +113,7 @@ impl FlexSurface {
self.taffy.set_style(*taffy_node, taffy_style).unwrap();
self.taffy.set_measure(*taffy_node, Some(measure)).unwrap();
} else {
let taffy_node = taffy.new_leaf(taffy_style, measure).unwrap();
let taffy_node = taffy.new_leaf(taffy_style).unwrap();
self.entity_to_taffy.insert(entity, taffy_node);
}
}
Expand Down Expand Up @@ -143,11 +146,10 @@ without UI components as a child of an entity with UI components, results may be

pub fn update_window(&mut self, window: &Window) {
let taffy = &mut self.taffy;
let node = self.window_nodes.entry(window.id()).or_insert_with(|| {
taffy
.new_node(taffy::style::Style::default(), &Vec::new())
.unwrap()
});
let node = self
.window_nodes
.entry(window.id())
.or_insert_with(|| taffy.new_leaf(taffy::style::Style::default()).unwrap());

taffy
.set_style(
Expand Down Expand Up @@ -178,7 +180,7 @@ without UI components as a child of an entity with UI components, results may be
pub fn compute_window_layouts(&mut self) {
for window_node in self.window_nodes.values() {
self.taffy
.compute_layout(*window_node, taffy::geometry::Size::undefined())
.compute_layout(*window_node, Size::MAX_CONTENT)
.unwrap();
}
}
Expand All @@ -187,7 +189,7 @@ without UI components as a child of an entity with UI components, results may be
pub fn remove_entities(&mut self, entities: impl IntoIterator<Item = Entity>) {
for entity in entities {
if let Some(node) = self.entity_to_taffy.remove(&entity) {
self.taffy.remove(node);
self.taffy.remove(node).unwrap();
}
}
}
Expand All @@ -210,7 +212,7 @@ with UI components as a child of an entity without UI components, results may be
#[derive(Debug)]
pub enum FlexError {
InvalidHierarchy,
TaffyError(taffy::Error),
TaffyError(taffy::error::TaffyError),
}

#[allow(clippy::too_many_arguments)]
Expand Down
5 changes: 5 additions & 0 deletions crates/bevy_ui/src/ui_node.rs
Original file line number Diff line number Diff line change
Expand Up @@ -237,6 +237,10 @@ pub struct Style {
pub aspect_ratio: Option<f32>,
/// How to handle overflow
pub overflow: Overflow,
/// The size of the gutters between the rows and columns of the flexbox layout
///
/// Values of `Size::UNDEFINED` and `Size::AUTO` are treated as zero.
pub gap: Size,
}

impl Default for Style {
Expand All @@ -263,6 +267,7 @@ impl Default for Style {
max_size: Size::AUTO,
aspect_ratio: Default::default(),
overflow: Default::default(),
gap: Size::UNDEFINED,
}
}
}
Expand Down

0 comments on commit 4b1ed7d

Please sign in to comment.