leyoda/iced-yoda
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

fix: tree diff improvements

Browse source 
Search for the old child state with matching index
This commit is contained in:
Ashley Wulber 2024-11-06 22:31:05 -05:00
parent 37085bc8ec
commit f1353e1ce2
No known key found for this signature in database
GPG key ID: 5216D4F46A90A820
1 changed files with 99 additions and 92 deletions
Download patch file Download diff file Expand all files Collapse all files

71
core/src/widget/tree.rs
View file

@ -3,7 +3,7 @@ use crate::Widget;
use crate::id::{Id, Internal}; use crate::id::{Id, Internal};
use std::any::{self, Any}; use std::any::{self, Any};
use std::borrow::{Borrow, BorrowMut, Cow}; use std::borrow::{Borrow, BorrowMut, Cow};
use std::collections::HashMap; use std::collections::{HashMap, VecDeque};
use std::hash::Hash; use std::hash::Hash;
use std::{fmt, mem}; use std::{fmt, mem};
@ -164,14 +164,16 @@ impl Tree {
{ {
let borrowed: &mut dyn Widget<Message, Theme, Renderer> = let borrowed: &mut dyn Widget<Message, Theme, Renderer> =
new.borrow_mut(); new.borrow_mut();
let mut needs_reset = false;
let tag_match = self.tag == borrowed.tag(); let mut tag_match = self.tag == borrowed.tag();
if tag_match {
if let Some(Id(Internal::Custom(_, n))) = borrowed.id() { if let Some(Id(Internal::Custom(_, n))) = borrowed.id() {
if let Some((mut state, children)) = NAMED let mut named = NAMED
.with(|named| named.borrow_mut().remove(&n)) .with(|named| named.borrow_mut().remove(&n))
.or_else(|| { .or_else(|| {
//check self.id //check self.id
if let Some(Id(Internal::Custom(_, ref name))) = self.id { if let Some(Id(Internal::Custom(_, ref name))) = self.id
{
if name == &n { if name == &n {
Some(( Some((
mem::replace(&mut self.state, State::None), mem::replace(&mut self.state, State::None),
@ -197,11 +199,13 @@ impl Tree {
} else { } else {
None None
} }
}) });
{ if let Some((mut state, children)) = named {
std::mem::swap(&mut self.state, &mut state); std::mem::swap(&mut self.state, &mut state);
let widget_children = borrowed.children(); let widget_children = borrowed.children();
if !tag_match || self.children.len() != widget_children.len() { if !tag_match
|| self.children.len() != widget_children.len()
{
self.children = borrowed.children(); self.children = borrowed.children();
} else { } else {
for (old_i, mut old) in children { for (old_i, mut old) in children {
@ -234,23 +238,22 @@ impl Tree {
} }
} }
} else { } else {
needs_reset = true; tag_match = false;
} }
} else if tag_match { } else {
if let Some(id) = self.id.clone() { if let Some(id) = self.id.clone() {
borrowed.set_id(id); borrowed.set_id(id);
} }
if self.children.len() != borrowed.children().len() { if self.children.len() != borrowed.children().len() {
self.children = borrowed.children(); self.children = borrowed.children();
} }
} else {
needs_reset = true;
} }
if needs_reset { }
*self = Self::new(borrowed); if tag_match {
let borrowed = new.borrow_mut();
borrowed.diff(self); borrowed.diff(self);
} else { } else {
*self = Self::new(borrowed);
let borrowed = new.borrow_mut();
borrowed.diff(self); borrowed.diff(self);
} }
} }
@ -269,6 +272,7 @@ impl Tree {
new_children.iter().map(|c| c.borrow().id()).collect(), new_children.iter().map(|c| c.borrow().id()).collect(),
|tree, widget| { |tree, widget| {
let borrowed: &mut dyn Widget<_, _, _> = widget.borrow_mut(); let borrowed: &mut dyn Widget<_, _, _> = widget.borrow_mut();
tree.diff(borrowed); tree.diff(borrowed);
}, },
|widget| { |widget| {
@ -291,29 +295,26 @@ impl Tree {
self.children.truncate(new_children.len()); self.children.truncate(new_children.len());
} }
let len_changed = self.children.len() != new_children.len();
let children_len = self.children.len(); let children_len = self.children.len();
let (mut id_map, mut id_list): ( let (mut id_map, mut id_list): (
HashMap<String, &mut Tree>, HashMap<String, &mut Tree>,
Vec<&mut Tree>, VecDeque<(usize, &mut Tree)>,
) = self.children.iter_mut().fold( ) = self.children.iter_mut().enumerate().fold(
(HashMap::new(), Vec::with_capacity(children_len)), (HashMap::new(), VecDeque::with_capacity(children_len)),
|(mut id_map, mut id_list), c| { |(mut id_map, mut id_list), (i, c)| {
if let Some(id) = c.id.as_ref() { if let Some(id) = c.id.as_ref() {
if let Internal::Custom(_, ref name) = id.0 { if let Internal::Custom(_, ref name) = id.0 {
let _ = id_map.insert(name.to_string(), c); let _ = id_map.insert(name.to_string(), c);
} else { } else {
id_list.push(c); id_list.push_back((i, c));
} }
} else { } else {
id_list.push(c); id_list.push_back((i, c));
} }
(id_map, id_list) (id_map, id_list)
}, },
); );
let mut child_state_i = 0;
let mut new_trees: Vec<(Tree, usize)> = let mut new_trees: Vec<(Tree, usize)> =
Vec::with_capacity(new_children.len()); Vec::with_capacity(new_children.len());
for (i, (new, new_id)) in for (i, (new, new_id)) in
@ -327,14 +328,20 @@ impl Tree {
} }
}) { }) {
c c
} else if child_state_i < id_list.len() } else if let Some(i) = {
&& !matches!( let mut found = None;
id_list[child_state_i].id, for c_i in 0..id_list.len() {
Some(Id(Internal::Custom(_, _))) if id_list[c_i].0 == i {
) found = Some(c_i);
{ break;
let c = &mut id_list[child_state_i]; }
child_state_i += 1; if i < c_i {
break;
}
}
found
} {
let c = id_list.remove(i).unwrap().1;
c c
} else { } else {
let mut my_new_state = new_state(new); let mut my_new_state = new_state(new);