125 lines
3.5 KiB
Rust
125 lines
3.5 KiB
Rust
use crate::core::Size;
|
|
use crate::geometry::{Frame, Geometry};
|
|
use crate::Renderer;
|
|
|
|
use std::cell::RefCell;
|
|
use std::sync::Arc;
|
|
|
|
/// A simple cache that stores generated [`Geometry`] to avoid recomputation.
|
|
///
|
|
/// A [`Cache`] will not redraw its geometry unless the dimensions of its layer
|
|
/// change or it is explicitly cleared.
|
|
#[derive(Debug, Default)]
|
|
pub struct Cache {
|
|
state: RefCell<State>,
|
|
}
|
|
|
|
#[derive(Debug, Default)]
|
|
enum State {
|
|
#[default]
|
|
Empty,
|
|
Filled {
|
|
bounds: Size,
|
|
primitive: Internal,
|
|
},
|
|
}
|
|
|
|
#[derive(Debug, Clone)]
|
|
enum Internal {
|
|
TinySkia(Arc<iced_tiny_skia::Primitive>),
|
|
#[cfg(feature = "wgpu")]
|
|
Wgpu(Arc<iced_wgpu::Primitive>),
|
|
}
|
|
|
|
impl Cache {
|
|
/// Creates a new empty [`Cache`].
|
|
pub fn new() -> Self {
|
|
Cache {
|
|
state: RefCell::default(),
|
|
}
|
|
}
|
|
|
|
/// Clears the [`Cache`], forcing a redraw the next time it is used.
|
|
pub fn clear(&self) {
|
|
*self.state.borrow_mut() = State::Empty;
|
|
}
|
|
|
|
/// Draws [`Geometry`] using the provided closure and stores it in the
|
|
/// [`Cache`].
|
|
///
|
|
/// The closure will only be called when
|
|
/// - the bounds have changed since the previous draw call.
|
|
/// - the [`Cache`] is empty or has been explicitly cleared.
|
|
///
|
|
/// Otherwise, the previously stored [`Geometry`] will be returned. The
|
|
/// [`Cache`] is not cleared in this case. In other words, it will keep
|
|
/// returning the stored [`Geometry`] if needed.
|
|
pub fn draw<Theme>(
|
|
&self,
|
|
renderer: &Renderer<Theme>,
|
|
bounds: Size,
|
|
draw_fn: impl FnOnce(&mut Frame),
|
|
) -> Geometry {
|
|
use std::ops::Deref;
|
|
|
|
if let State::Filled {
|
|
bounds: cached_bounds,
|
|
primitive,
|
|
} = self.state.borrow().deref()
|
|
{
|
|
if *cached_bounds == bounds {
|
|
match primitive {
|
|
Internal::TinySkia(primitive) => {
|
|
return Geometry::TinySkia(
|
|
iced_tiny_skia::Primitive::Cache {
|
|
content: primitive.clone(),
|
|
},
|
|
);
|
|
}
|
|
#[cfg(feature = "wgpu")]
|
|
Internal::Wgpu(primitive) => {
|
|
return Geometry::Wgpu(iced_wgpu::Primitive::Cache {
|
|
content: primitive.clone(),
|
|
});
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
let mut frame = Frame::new(renderer, bounds);
|
|
draw_fn(&mut frame);
|
|
|
|
let primitive = {
|
|
let geometry = frame.into_geometry();
|
|
|
|
match geometry {
|
|
Geometry::TinySkia(primitive) => {
|
|
Internal::TinySkia(Arc::new(primitive))
|
|
}
|
|
#[cfg(feature = "wgpu")]
|
|
Geometry::Wgpu(primitive) => {
|
|
Internal::Wgpu(Arc::new(primitive))
|
|
}
|
|
}
|
|
};
|
|
|
|
*self.state.borrow_mut() = State::Filled {
|
|
bounds,
|
|
primitive: primitive.clone(),
|
|
};
|
|
|
|
match primitive {
|
|
Internal::TinySkia(primitive) => {
|
|
Geometry::TinySkia(iced_tiny_skia::Primitive::Cache {
|
|
content: primitive,
|
|
})
|
|
}
|
|
#[cfg(feature = "wgpu")]
|
|
Internal::Wgpu(primitive) => {
|
|
Geometry::Wgpu(iced_wgpu::Primitive::Cache {
|
|
content: primitive,
|
|
})
|
|
}
|
|
}
|
|
}
|
|
}
|