44 lines
1 KiB
Rust
44 lines
1 KiB
Rust
|
use std::collections::HashMap;
|
||
|
|
|
||
|
|
use crate::{Id, PaintCmd};
|
||
|
|
|
||
|
|
// TODO: support multiple windows
|
||
|
|
#[derive(Clone, Copy, Eq, PartialEq, Hash)]
|
||
|
|
pub enum Layer {
|
||
|
|
Background,
|
||
|
|
Window(Id),
|
||
|
|
Popup,
|
||
|
|
}
|
||
|
|
|
||
|
|
type PaintList = Vec<PaintCmd>;
|
||
|
|
|
||
|
|
/// TODO: improve this
|
||
|
|
#[derive(Clone, Default)]
|
||
|
|
pub struct GraphicLayers {
|
||
|
|
bg: PaintList,
|
||
|
|
windows: HashMap<Id, PaintList>,
|
||
|
|
popup: PaintList,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl GraphicLayers {
|
||
|
|
pub fn layer(&mut self, layer: Layer) -> &mut PaintList {
|
||
|
|
match layer {
|
||
|
|
Layer::Background => &mut self.bg,
|
||
|
|
Layer::Window(id) => self.windows.entry(id).or_default(),
|
||
|
|
Layer::Popup => &mut self.popup,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn drain(&mut self) -> impl ExactSizeIterator<Item = PaintCmd> {
|
||
|
|
let mut all_commands: Vec<_> = self.bg.drain(..).collect();
|
||
|
|
|
||
|
|
// TODO: order
|
||
|
|
for window in self.windows.values_mut() {
|
||
|
|
all_commands.extend(window.drain(..));
|
||
|
|
}
|
||
|
|
|
||
|
|
all_commands.extend(self.popup.drain(..));
|
||
|
|
all_commands.into_iter()
|
||
|
|
}
|
||
|
|
}
|