2021-12-06 19:19:03 +00:00
|
|
|
use super::{draw_window, Window};
|
2021-11-28 00:22:00 +00:00
|
|
|
use crate::{
|
2021-12-20 18:13:55 +00:00
|
|
|
backend::renderer::{utils::SurfaceState, Frame, ImportAll, Renderer, Texture, Transform},
|
2021-12-13 21:06:39 +00:00
|
|
|
desktop::{layer::*, output::*},
|
2021-11-28 00:22:00 +00:00
|
|
|
utils::{Logical, Point, Rectangle},
|
2021-11-30 15:53:42 +00:00
|
|
|
wayland::{
|
2021-12-17 11:59:11 +00:00
|
|
|
compositor::{
|
|
|
|
get_parent, is_sync_subsurface, with_surface_tree_downward, SubsurfaceCachedState,
|
|
|
|
TraversalAction,
|
|
|
|
},
|
2021-11-30 15:53:42 +00:00
|
|
|
output::Output,
|
2021-12-13 21:06:39 +00:00
|
|
|
shell::wlr_layer::Layer as WlrLayer,
|
2021-11-30 15:53:42 +00:00
|
|
|
},
|
2021-11-28 00:22:00 +00:00
|
|
|
};
|
|
|
|
use indexmap::{IndexMap, IndexSet};
|
|
|
|
use std::{
|
2021-12-20 18:13:55 +00:00
|
|
|
any::{Any, TypeId},
|
2021-11-28 00:22:00 +00:00
|
|
|
cell::{RefCell, RefMut},
|
|
|
|
collections::{HashMap, HashSet, VecDeque},
|
|
|
|
sync::{
|
|
|
|
atomic::{AtomicUsize, Ordering},
|
|
|
|
Mutex,
|
|
|
|
},
|
|
|
|
};
|
2021-11-30 15:53:42 +00:00
|
|
|
use wayland_server::protocol::wl_surface::WlSurface;
|
2021-11-28 00:22:00 +00:00
|
|
|
|
|
|
|
static SPACE_ID: AtomicUsize = AtomicUsize::new(0);
|
|
|
|
lazy_static::lazy_static! {
|
|
|
|
static ref SPACE_IDS: Mutex<HashSet<usize>> = Mutex::new(HashSet::new());
|
|
|
|
}
|
|
|
|
fn next_space_id() -> usize {
|
|
|
|
let mut ids = SPACE_IDS.lock().unwrap();
|
|
|
|
if ids.len() == usize::MAX {
|
|
|
|
// Theoretically the code below wraps around correctly,
|
|
|
|
// but that is hard to detect and might deadlock.
|
|
|
|
// Maybe make this a debug_assert instead?
|
|
|
|
panic!("Out of space ids");
|
|
|
|
}
|
|
|
|
|
|
|
|
let mut id = SPACE_ID.fetch_add(1, Ordering::SeqCst);
|
|
|
|
while ids.iter().any(|k| *k == id) {
|
|
|
|
id = SPACE_ID.fetch_add(1, Ordering::SeqCst);
|
|
|
|
}
|
|
|
|
|
|
|
|
ids.insert(id);
|
|
|
|
id
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Default)]
|
|
|
|
struct WindowState {
|
|
|
|
location: Point<i32, Logical>,
|
|
|
|
drawn: bool,
|
|
|
|
}
|
|
|
|
|
|
|
|
type WindowUserdata = RefCell<HashMap<usize, WindowState>>;
|
|
|
|
fn window_state(space: usize, w: &Window) -> RefMut<'_, WindowState> {
|
|
|
|
let userdata = w.user_data();
|
|
|
|
userdata.insert_if_missing(WindowUserdata::default);
|
|
|
|
RefMut::map(userdata.get::<WindowUserdata>().unwrap().borrow_mut(), |m| {
|
|
|
|
m.entry(space).or_default()
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2021-12-13 21:06:39 +00:00
|
|
|
#[derive(Default)]
|
|
|
|
struct LayerState {
|
|
|
|
drawn: bool,
|
|
|
|
}
|
|
|
|
|
|
|
|
type LayerUserdata = RefCell<HashMap<usize, LayerState>>;
|
|
|
|
fn layer_state(space: usize, l: &LayerSurface) -> RefMut<'_, LayerState> {
|
|
|
|
let userdata = l.user_data();
|
|
|
|
userdata.insert_if_missing(LayerUserdata::default);
|
|
|
|
RefMut::map(userdata.get::<LayerUserdata>().unwrap().borrow_mut(), |m| {
|
|
|
|
m.entry(space).or_default()
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2021-11-28 00:22:00 +00:00
|
|
|
// TODO: Maybe replace UnmanagedResource if nothing else comes up?
|
|
|
|
#[derive(Debug, thiserror::Error)]
|
|
|
|
pub enum SpaceError {
|
|
|
|
#[error("Window is not mapped to this space")]
|
|
|
|
UnknownWindow,
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Debug)]
|
|
|
|
pub struct Space {
|
2021-12-14 13:12:08 +00:00
|
|
|
pub(super) id: usize,
|
2021-11-28 00:22:00 +00:00
|
|
|
// in z-order, back to front
|
|
|
|
windows: IndexSet<Window>,
|
|
|
|
outputs: Vec<Output>,
|
|
|
|
// TODO:
|
|
|
|
//layers: Vec<Layer>,
|
|
|
|
logger: ::slog::Logger,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Drop for Space {
|
|
|
|
fn drop(&mut self) {
|
|
|
|
SPACE_IDS.lock().unwrap().remove(&self.id);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Space {
|
|
|
|
pub fn new<L>(log: L) -> Space
|
|
|
|
where
|
2021-12-16 12:04:00 +00:00
|
|
|
L: Into<Option<slog::Logger>>,
|
2021-11-28 00:22:00 +00:00
|
|
|
{
|
|
|
|
Space {
|
|
|
|
id: next_space_id(),
|
|
|
|
windows: IndexSet::new(),
|
|
|
|
outputs: Vec::new(),
|
2021-12-16 12:04:00 +00:00
|
|
|
logger: crate::slog_or_fallback(log),
|
2021-11-28 00:22:00 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Map window and moves it to top of the stack
|
|
|
|
///
|
|
|
|
/// This can safely be called on an already mapped window
|
2021-12-16 12:04:00 +00:00
|
|
|
pub fn map_window<P: Into<Point<i32, Logical>>>(&mut self, window: &Window, location: P) {
|
2021-12-13 21:06:13 +00:00
|
|
|
self.insert_window(window);
|
2021-12-16 12:04:00 +00:00
|
|
|
window_state(self.id, window).location = location.into();
|
2021-12-02 11:45:58 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
pub fn raise_window(&mut self, window: &Window) {
|
2021-12-13 21:06:13 +00:00
|
|
|
if self.windows.shift_remove(window) {
|
|
|
|
self.insert_window(window);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
fn insert_window(&mut self, window: &Window) {
|
2021-11-28 00:22:00 +00:00
|
|
|
self.windows.insert(window.clone());
|
|
|
|
|
|
|
|
// TODO: should this be handled by us?
|
|
|
|
window.set_activated(true);
|
|
|
|
for w in self.windows.iter() {
|
|
|
|
if w != window {
|
|
|
|
w.set_activated(false);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Unmap a window from this space by its id
|
|
|
|
pub fn unmap_window(&mut self, window: &Window) {
|
|
|
|
if let Some(map) = window.user_data().get::<WindowUserdata>() {
|
|
|
|
map.borrow_mut().remove(&self.id);
|
|
|
|
}
|
|
|
|
self.windows.shift_remove(window);
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Iterate window in z-order back to front
|
2021-12-13 21:06:13 +00:00
|
|
|
pub fn windows(&self) -> impl DoubleEndedIterator<Item = &Window> {
|
2021-11-28 00:22:00 +00:00
|
|
|
self.windows.iter()
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Get a reference to the window under a given point, if any
|
2021-12-16 12:04:00 +00:00
|
|
|
pub fn window_under<P: Into<Point<f64, Logical>>>(&self, point: P) -> Option<&Window> {
|
|
|
|
let point = point.into();
|
2021-12-06 19:19:03 +00:00
|
|
|
self.windows.iter().rev().find(|w| {
|
2021-12-01 17:46:23 +00:00
|
|
|
let bbox = window_rect(w, &self.id);
|
2021-11-28 00:22:00 +00:00
|
|
|
bbox.to_f64().contains(point)
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2021-12-13 21:06:13 +00:00
|
|
|
/// Get a reference to the output under a given point, if any
|
2021-12-16 12:04:00 +00:00
|
|
|
pub fn output_under<P: Into<Point<f64, Logical>>>(&self, point: P) -> Option<&Output> {
|
|
|
|
let point = point.into();
|
2021-12-13 21:06:13 +00:00
|
|
|
self.outputs.iter().rev().find(|o| {
|
|
|
|
let bbox = self.output_geometry(o);
|
|
|
|
bbox.map(|bbox| bbox.to_f64().contains(point)).unwrap_or(false)
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2021-11-30 15:53:42 +00:00
|
|
|
pub fn window_for_surface(&self, surface: &WlSurface) -> Option<&Window> {
|
2021-11-28 00:22:00 +00:00
|
|
|
if !surface.as_ref().is_alive() {
|
|
|
|
return None;
|
|
|
|
}
|
|
|
|
|
|
|
|
self.windows
|
|
|
|
.iter()
|
|
|
|
.find(|w| w.toplevel().get_surface().map(|x| x == surface).unwrap_or(false))
|
|
|
|
}
|
|
|
|
|
2021-12-13 21:06:39 +00:00
|
|
|
pub fn layer_for_surface(&self, surface: &WlSurface) -> Option<LayerSurface> {
|
|
|
|
if !surface.as_ref().is_alive() {
|
|
|
|
return None;
|
|
|
|
}
|
|
|
|
self.outputs.iter().find_map(|o| {
|
|
|
|
let map = layer_map_for_output(o);
|
|
|
|
map.layer_for_surface(surface).cloned()
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2021-11-28 00:22:00 +00:00
|
|
|
pub fn window_geometry(&self, w: &Window) -> Option<Rectangle<i32, Logical>> {
|
|
|
|
if !self.windows.contains(w) {
|
|
|
|
return None;
|
|
|
|
}
|
|
|
|
|
2021-12-01 17:46:23 +00:00
|
|
|
Some(window_geo(w, &self.id))
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn window_bbox(&self, w: &Window) -> Option<Rectangle<i32, Logical>> {
|
|
|
|
if !self.windows.contains(w) {
|
|
|
|
return None;
|
|
|
|
}
|
|
|
|
|
2021-11-28 00:22:00 +00:00
|
|
|
Some(window_rect(w, &self.id))
|
|
|
|
}
|
|
|
|
|
2021-12-16 12:04:00 +00:00
|
|
|
pub fn map_output<P: Into<Point<i32, Logical>>>(&mut self, output: &Output, scale: f64, location: P) {
|
2021-11-28 00:22:00 +00:00
|
|
|
let mut state = output_state(self.id, output);
|
|
|
|
*state = OutputState {
|
2021-12-16 12:04:00 +00:00
|
|
|
location: location.into(),
|
2021-11-28 00:22:00 +00:00
|
|
|
render_scale: scale,
|
|
|
|
..Default::default()
|
|
|
|
};
|
|
|
|
if !self.outputs.contains(output) {
|
|
|
|
self.outputs.push(output.clone());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn outputs(&self) -> impl Iterator<Item = &Output> {
|
|
|
|
self.outputs.iter()
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn unmap_output(&mut self, output: &Output) {
|
|
|
|
if let Some(map) = output.user_data().get::<OutputUserdata>() {
|
|
|
|
map.borrow_mut().remove(&self.id);
|
|
|
|
}
|
|
|
|
self.outputs.retain(|o| o != output);
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn output_geometry(&self, o: &Output) -> Option<Rectangle<i32, Logical>> {
|
|
|
|
if !self.outputs.contains(o) {
|
|
|
|
return None;
|
|
|
|
}
|
|
|
|
|
|
|
|
let state = output_state(self.id, o);
|
|
|
|
o.current_mode().map(|mode| {
|
|
|
|
Rectangle::from_loc_and_size(
|
|
|
|
state.location,
|
|
|
|
mode.size.to_f64().to_logical(state.render_scale).to_i32_round(),
|
|
|
|
)
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn output_scale(&self, o: &Output) -> Option<f64> {
|
|
|
|
if !self.outputs.contains(o) {
|
|
|
|
return None;
|
|
|
|
}
|
|
|
|
|
|
|
|
let state = output_state(self.id, o);
|
|
|
|
Some(state.render_scale)
|
|
|
|
}
|
|
|
|
|
2021-12-02 11:45:21 +00:00
|
|
|
pub fn outputs_for_window(&self, w: &Window) -> Vec<Output> {
|
2021-11-28 00:22:00 +00:00
|
|
|
if !self.windows.contains(w) {
|
2021-12-02 11:45:21 +00:00
|
|
|
return Vec::new();
|
2021-11-28 00:22:00 +00:00
|
|
|
}
|
|
|
|
|
2021-12-02 11:45:21 +00:00
|
|
|
let w_geo = window_rect(w, &self.id);
|
|
|
|
let mut outputs = self
|
|
|
|
.outputs
|
|
|
|
.iter()
|
|
|
|
.cloned()
|
|
|
|
.filter(|o| {
|
|
|
|
let o_geo = self.output_geometry(o).unwrap();
|
|
|
|
w_geo.overlaps(o_geo)
|
|
|
|
})
|
|
|
|
.collect::<Vec<Output>>();
|
|
|
|
outputs.sort_by(|o1, o2| {
|
|
|
|
let overlap = |rect1: Rectangle<i32, Logical>, rect2: Rectangle<i32, Logical>| -> i32 {
|
|
|
|
// x overlap
|
|
|
|
std::cmp::max(0, std::cmp::min(rect1.loc.x + rect1.size.w, rect2.loc.x + rect2.size.w) - std::cmp::max(rect1.loc.x, rect2.loc.x))
|
|
|
|
// y overlap
|
|
|
|
* std::cmp::max(0, std::cmp::min(rect1.loc.y + rect1.size.h, rect2.loc.y + rect2.size.h) - std::cmp::max(rect1.loc.y, rect2.loc.y))
|
|
|
|
};
|
|
|
|
let o1_area = overlap(self.output_geometry(o1).unwrap(), w_geo);
|
|
|
|
let o2_area = overlap(self.output_geometry(o2).unwrap(), w_geo);
|
|
|
|
o1_area.cmp(&o2_area)
|
|
|
|
});
|
|
|
|
outputs
|
2021-11-28 00:22:00 +00:00
|
|
|
}
|
|
|
|
|
2021-11-30 15:53:42 +00:00
|
|
|
pub fn refresh(&mut self) {
|
2021-11-28 00:22:00 +00:00
|
|
|
self.windows.retain(|w| w.toplevel().alive());
|
2021-11-30 15:53:42 +00:00
|
|
|
|
|
|
|
for output in &mut self.outputs {
|
|
|
|
output_state(self.id, output)
|
|
|
|
.surfaces
|
|
|
|
.retain(|s| s.as_ref().is_alive());
|
|
|
|
}
|
|
|
|
|
|
|
|
for window in &self.windows {
|
|
|
|
let bbox = window_rect(window, &self.id);
|
|
|
|
let kind = window.toplevel();
|
|
|
|
|
|
|
|
for output in &self.outputs {
|
|
|
|
let output_geometry = self
|
|
|
|
.output_geometry(output)
|
|
|
|
.unwrap_or_else(|| Rectangle::from_loc_and_size((0, 0), (0, 0)));
|
|
|
|
let mut output_state = output_state(self.id, output);
|
|
|
|
|
|
|
|
// Check if the bounding box of the toplevel intersects with
|
|
|
|
// the output, if not no surface in the tree can intersect with
|
|
|
|
// the output.
|
|
|
|
if !output_geometry.overlaps(bbox) {
|
|
|
|
if let Some(surface) = kind.get_surface() {
|
|
|
|
with_surface_tree_downward(
|
|
|
|
surface,
|
|
|
|
(),
|
|
|
|
|_, _, _| TraversalAction::DoChildren(()),
|
|
|
|
|wl_surface, _, _| {
|
|
|
|
if output_state.surfaces.contains(wl_surface) {
|
|
|
|
slog::trace!(
|
|
|
|
self.logger,
|
|
|
|
"surface ({:?}) leaving output {:?}",
|
|
|
|
wl_surface,
|
|
|
|
output.name()
|
|
|
|
);
|
|
|
|
output.leave(wl_surface);
|
|
|
|
output_state.surfaces.retain(|s| s != wl_surface);
|
|
|
|
}
|
|
|
|
},
|
|
|
|
|_, _, _| true,
|
|
|
|
)
|
|
|
|
}
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
if let Some(surface) = kind.get_surface() {
|
|
|
|
with_surface_tree_downward(
|
|
|
|
surface,
|
|
|
|
window_loc(window, &self.id),
|
|
|
|
|_, states, location| {
|
|
|
|
let mut location = *location;
|
|
|
|
let data = states.data_map.get::<RefCell<SurfaceState>>();
|
|
|
|
|
|
|
|
if data.is_some() {
|
|
|
|
if states.role == Some("subsurface") {
|
|
|
|
let current = states.cached_state.current::<SubsurfaceCachedState>();
|
|
|
|
location += current.location;
|
|
|
|
}
|
|
|
|
|
|
|
|
TraversalAction::DoChildren(location)
|
|
|
|
} else {
|
|
|
|
// If the parent surface is unmapped, then the child surfaces are hidden as
|
|
|
|
// well, no need to consider them here.
|
|
|
|
TraversalAction::SkipChildren
|
|
|
|
}
|
|
|
|
},
|
|
|
|
|wl_surface, states, &loc| {
|
|
|
|
let data = states.data_map.get::<RefCell<SurfaceState>>();
|
|
|
|
|
|
|
|
if let Some(size) = data.and_then(|d| d.borrow().size()) {
|
|
|
|
let surface_rectangle = Rectangle { loc, size };
|
|
|
|
|
|
|
|
if output_geometry.overlaps(surface_rectangle) {
|
|
|
|
// We found a matching output, check if we already sent enter
|
|
|
|
if !output_state.surfaces.contains(wl_surface) {
|
|
|
|
slog::trace!(
|
|
|
|
self.logger,
|
|
|
|
"surface ({:?}) entering output {:?}",
|
|
|
|
wl_surface,
|
|
|
|
output.name()
|
|
|
|
);
|
|
|
|
output.enter(wl_surface);
|
|
|
|
output_state.surfaces.push(wl_surface.clone());
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
// Surface does not match output, if we sent enter earlier
|
|
|
|
// we should now send leave
|
|
|
|
if output_state.surfaces.contains(wl_surface) {
|
|
|
|
slog::trace!(
|
|
|
|
self.logger,
|
|
|
|
"surface ({:?}) leaving output {:?}",
|
|
|
|
wl_surface,
|
|
|
|
output.name()
|
|
|
|
);
|
|
|
|
output.leave(wl_surface);
|
|
|
|
output_state.surfaces.retain(|s| s != wl_surface);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
// Maybe the the surface got unmapped, send leave on output
|
|
|
|
if output_state.surfaces.contains(wl_surface) {
|
|
|
|
slog::trace!(
|
|
|
|
self.logger,
|
|
|
|
"surface ({:?}) leaving output {:?}",
|
|
|
|
wl_surface,
|
|
|
|
output.name()
|
|
|
|
);
|
|
|
|
output.leave(wl_surface);
|
|
|
|
output_state.surfaces.retain(|s| s != wl_surface);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
},
|
|
|
|
|_, _, _| true,
|
|
|
|
)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2021-11-28 00:22:00 +00:00
|
|
|
}
|
|
|
|
|
2021-12-17 11:59:11 +00:00
|
|
|
/// Automatically calls `Window::refresh` for the window that belongs to the given surface,
|
|
|
|
/// if managed by this space.
|
2021-12-20 12:40:08 +00:00
|
|
|
pub fn commit(&self, surface: &WlSurface) {
|
2021-12-17 11:59:11 +00:00
|
|
|
if is_sync_subsurface(surface) {
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
let mut root = surface.clone();
|
|
|
|
while let Some(parent) = get_parent(&root) {
|
|
|
|
root = parent;
|
|
|
|
}
|
|
|
|
if let Some(window) = self.windows().find(|w| w.toplevel().get_surface() == Some(&root)) {
|
|
|
|
window.refresh();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-11-28 00:22:00 +00:00
|
|
|
pub fn render_output<R>(
|
|
|
|
&mut self,
|
|
|
|
renderer: &mut R,
|
|
|
|
output: &Output,
|
|
|
|
age: usize,
|
|
|
|
clear_color: [f32; 4],
|
2021-12-20 18:13:55 +00:00
|
|
|
custom_elements: &[&(dyn RenderElement<
|
|
|
|
R,
|
|
|
|
<R as Renderer>::Frame,
|
|
|
|
<R as Renderer>::Error,
|
|
|
|
<R as Renderer>::TextureId,
|
|
|
|
>)],
|
2021-11-28 00:22:00 +00:00
|
|
|
) -> Result<bool, RenderError<R>>
|
|
|
|
where
|
2021-12-20 18:13:55 +00:00
|
|
|
R: Renderer + ImportAll + 'static,
|
2021-11-28 00:22:00 +00:00
|
|
|
R::TextureId: 'static,
|
2021-12-20 18:13:55 +00:00
|
|
|
R::Error: 'static,
|
|
|
|
R::Frame: 'static,
|
2021-11-28 00:22:00 +00:00
|
|
|
{
|
|
|
|
let mut state = output_state(self.id, output);
|
|
|
|
let output_size = output
|
|
|
|
.current_mode()
|
|
|
|
.ok_or(RenderError::OutputNoMode)?
|
|
|
|
.size
|
|
|
|
.to_f64()
|
|
|
|
.to_logical(state.render_scale)
|
|
|
|
.to_i32_round();
|
|
|
|
let output_geo = Rectangle::from_loc_and_size(state.location, output_size);
|
2021-12-13 21:06:39 +00:00
|
|
|
let layer_map = layer_map_for_output(output);
|
2021-11-28 00:22:00 +00:00
|
|
|
|
|
|
|
// This will hold all the damage we need for this rendering step
|
|
|
|
let mut damage = Vec::<Rectangle<i32, Logical>>::new();
|
|
|
|
// First add damage for windows gone
|
2021-12-20 18:13:55 +00:00
|
|
|
for old_toplevel in state
|
2021-11-28 00:22:00 +00:00
|
|
|
.last_state
|
|
|
|
.iter()
|
|
|
|
.filter_map(|(id, w)| {
|
2021-12-13 21:06:39 +00:00
|
|
|
if !self.windows.iter().any(|w| ToplevelId::Xdg(w.0.id) == *id)
|
|
|
|
&& !layer_map.layers().any(|l| ToplevelId::Layer(l.0.id) == *id)
|
2021-12-20 18:13:55 +00:00
|
|
|
&& !custom_elements
|
|
|
|
.iter()
|
|
|
|
.any(|c| ToplevelId::Custom(c.type_of(), c.id()) == *id)
|
2021-12-13 21:06:39 +00:00
|
|
|
{
|
2021-11-28 00:22:00 +00:00
|
|
|
Some(*w)
|
|
|
|
} else {
|
|
|
|
None
|
|
|
|
}
|
|
|
|
})
|
|
|
|
.collect::<Vec<Rectangle<i32, Logical>>>()
|
|
|
|
{
|
2021-12-20 18:13:55 +00:00
|
|
|
slog::trace!(self.logger, "Removing toplevel at: {:?}", old_toplevel);
|
|
|
|
damage.push(old_toplevel);
|
2021-11-28 00:22:00 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// lets iterate front to back and figure out, what new windows or unmoved windows we have
|
2021-12-13 21:06:13 +00:00
|
|
|
for window in self.windows.iter() {
|
2021-12-06 19:19:03 +00:00
|
|
|
let geo = window_rect_with_popups(window, &self.id);
|
2021-12-13 21:06:13 +00:00
|
|
|
let old_geo = state.last_state.get(&ToplevelId::Xdg(window.0.id)).cloned();
|
2021-11-28 00:22:00 +00:00
|
|
|
|
|
|
|
// window was moved or resized
|
|
|
|
if old_geo.map(|old_geo| old_geo != geo).unwrap_or(false) {
|
|
|
|
// Add damage for the old position of the window
|
|
|
|
damage.push(old_geo.unwrap());
|
2021-11-28 21:30:17 +00:00
|
|
|
damage.push(geo);
|
|
|
|
} else {
|
|
|
|
// window stayed at its place
|
|
|
|
let loc = window_loc(window, &self.id);
|
2021-12-14 13:12:08 +00:00
|
|
|
damage.extend(
|
|
|
|
window
|
|
|
|
.accumulated_damage(Some((self, output)))
|
|
|
|
.into_iter()
|
|
|
|
.map(|mut rect| {
|
|
|
|
rect.loc += loc;
|
|
|
|
rect
|
|
|
|
}),
|
|
|
|
);
|
2021-11-28 21:30:17 +00:00
|
|
|
}
|
2021-11-28 00:22:00 +00:00
|
|
|
}
|
2021-12-13 21:06:39 +00:00
|
|
|
for layer in layer_map.layers() {
|
|
|
|
let geo = layer_map.layer_geometry(layer);
|
|
|
|
let old_geo = state.last_state.get(&ToplevelId::Layer(layer.0.id)).cloned();
|
|
|
|
|
|
|
|
// layer moved or resized
|
|
|
|
if old_geo.map(|old_geo| old_geo != geo).unwrap_or(false) {
|
|
|
|
// Add damage for the old position of the layer
|
|
|
|
damage.push(old_geo.unwrap());
|
|
|
|
damage.push(geo);
|
|
|
|
} else {
|
|
|
|
let location = geo.loc;
|
2021-12-14 13:12:08 +00:00
|
|
|
damage.extend(
|
|
|
|
layer
|
|
|
|
.accumulated_damage(Some((self, output)))
|
|
|
|
.into_iter()
|
|
|
|
.map(|mut rect| {
|
|
|
|
rect.loc += location;
|
|
|
|
rect
|
|
|
|
}),
|
|
|
|
);
|
2021-12-13 21:06:39 +00:00
|
|
|
}
|
|
|
|
}
|
2021-12-20 18:13:55 +00:00
|
|
|
for elem in custom_elements {
|
|
|
|
let geo = elem.geometry();
|
|
|
|
let old_geo = state
|
|
|
|
.last_state
|
|
|
|
.get(&ToplevelId::Custom(elem.type_of(), elem.id()))
|
|
|
|
.cloned();
|
|
|
|
|
|
|
|
// moved of resized
|
|
|
|
if old_geo.map(|old_geo| old_geo != geo).unwrap_or(false) {
|
|
|
|
// Add damage for the old position of the layer
|
|
|
|
damage.push(old_geo.unwrap());
|
|
|
|
damage.push(geo);
|
|
|
|
} else {
|
|
|
|
let location = geo.loc;
|
|
|
|
damage.extend(
|
|
|
|
elem.accumulated_damage(Some((self, output)))
|
|
|
|
.into_iter()
|
|
|
|
.map(|mut rect| {
|
|
|
|
rect.loc += location;
|
|
|
|
rect
|
|
|
|
}),
|
|
|
|
);
|
|
|
|
}
|
|
|
|
}
|
2021-11-28 00:22:00 +00:00
|
|
|
|
|
|
|
// That is all completely new damage, which we need to store for subsequent renders
|
|
|
|
let new_damage = damage.clone();
|
|
|
|
// We now add old damage states, if we have an age value
|
|
|
|
if age > 0 && state.old_damage.len() >= age {
|
|
|
|
// We do not need older states anymore
|
|
|
|
state.old_damage.truncate(age);
|
|
|
|
damage.extend(state.old_damage.iter().flatten().copied());
|
|
|
|
} else {
|
|
|
|
// just damage everything, if we have no damage
|
|
|
|
damage = vec![output_geo];
|
|
|
|
}
|
|
|
|
|
|
|
|
// Optimize the damage for rendering
|
2021-12-07 17:34:10 +00:00
|
|
|
damage.dedup();
|
2021-11-28 00:22:00 +00:00
|
|
|
damage.retain(|rect| rect.overlaps(output_geo));
|
|
|
|
damage.retain(|rect| rect.size.h > 0 && rect.size.w > 0);
|
|
|
|
for rect in damage.clone().iter() {
|
|
|
|
// if this rect was already removed, because it was smaller as another one,
|
|
|
|
// there is no reason to evaluate this.
|
|
|
|
if damage.contains(rect) {
|
|
|
|
// remove every rectangle that is contained in this rectangle
|
|
|
|
damage.retain(|other| !rect.contains_rect(*other));
|
|
|
|
}
|
|
|
|
}
|
2021-12-07 17:34:10 +00:00
|
|
|
damage = damage.into_iter().fold(Vec::new(), |mut new_damage, rect| {
|
|
|
|
if let Some(existing) = new_damage.iter_mut().find(|other| rect.overlaps(**other)) {
|
|
|
|
*existing = existing.merge(rect);
|
|
|
|
} else {
|
|
|
|
new_damage.push(rect);
|
|
|
|
}
|
|
|
|
new_damage
|
|
|
|
});
|
|
|
|
|
|
|
|
if damage.is_empty() {
|
|
|
|
return Ok(false);
|
|
|
|
}
|
2021-11-28 00:22:00 +00:00
|
|
|
|
|
|
|
let output_transform: Transform = output.current_transform().into();
|
|
|
|
if let Err(err) = renderer.render(
|
|
|
|
output_transform
|
|
|
|
.transform_size(output_size)
|
|
|
|
.to_f64()
|
|
|
|
.to_physical(state.render_scale)
|
|
|
|
.to_i32_round(),
|
|
|
|
output_transform,
|
|
|
|
|renderer, frame| {
|
|
|
|
// First clear all damaged regions
|
2021-12-07 17:34:10 +00:00
|
|
|
slog::trace!(self.logger, "Clearing at {:#?}", damage);
|
|
|
|
frame.clear(
|
|
|
|
clear_color,
|
|
|
|
&damage
|
|
|
|
.iter()
|
|
|
|
.map(|geo| geo.to_f64().to_physical(state.render_scale).to_i32_round())
|
|
|
|
.collect::<Vec<_>>(),
|
|
|
|
)?;
|
2021-11-28 00:22:00 +00:00
|
|
|
|
2021-12-13 21:06:39 +00:00
|
|
|
// Then re-draw all windows & layers overlapping with a damage rect.
|
|
|
|
|
|
|
|
for layer in layer_map
|
|
|
|
.layers_on(WlrLayer::Background)
|
|
|
|
.chain(layer_map.layers_on(WlrLayer::Bottom))
|
|
|
|
{
|
|
|
|
let lgeo = layer_map.layer_geometry(layer);
|
|
|
|
if damage.iter().any(|geo| lgeo.overlaps(*geo)) {
|
|
|
|
let layer_damage = damage
|
|
|
|
.iter()
|
2021-12-20 18:12:21 +00:00
|
|
|
.flat_map(|geo| geo.intersection(lgeo))
|
2021-12-13 21:06:39 +00:00
|
|
|
.map(|geo| Rectangle::from_loc_and_size(geo.loc - lgeo.loc, geo.size))
|
|
|
|
.collect::<Vec<_>>();
|
|
|
|
slog::trace!(
|
|
|
|
self.logger,
|
|
|
|
"Rendering layer at {:?} with damage {:#?}",
|
|
|
|
lgeo,
|
2021-12-20 18:13:55 +00:00
|
|
|
layer_damage
|
2021-12-13 21:06:39 +00:00
|
|
|
);
|
|
|
|
draw_layer(
|
|
|
|
renderer,
|
|
|
|
frame,
|
|
|
|
layer,
|
|
|
|
state.render_scale,
|
|
|
|
lgeo.loc,
|
|
|
|
&layer_damage,
|
|
|
|
&self.logger,
|
|
|
|
)?;
|
|
|
|
layer_state(self.id, layer).drawn = true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-11-28 00:22:00 +00:00
|
|
|
for window in self.windows.iter() {
|
2021-12-06 19:19:03 +00:00
|
|
|
let wgeo = window_rect_with_popups(window, &self.id);
|
2021-11-28 00:22:00 +00:00
|
|
|
let mut loc = window_loc(window, &self.id);
|
|
|
|
if damage.iter().any(|geo| wgeo.overlaps(*geo)) {
|
|
|
|
loc -= output_geo.loc;
|
2021-12-07 17:34:10 +00:00
|
|
|
let win_damage = damage
|
|
|
|
.iter()
|
2021-12-20 18:12:21 +00:00
|
|
|
.flat_map(|geo| geo.intersection(wgeo))
|
2021-12-14 13:38:33 +00:00
|
|
|
.map(|geo| Rectangle::from_loc_and_size(geo.loc - loc, geo.size))
|
2021-12-07 17:34:10 +00:00
|
|
|
.collect::<Vec<_>>();
|
|
|
|
slog::trace!(
|
|
|
|
self.logger,
|
|
|
|
"Rendering window at {:?} with damage {:#?}",
|
|
|
|
wgeo,
|
2021-12-20 18:13:55 +00:00
|
|
|
win_damage
|
2021-12-07 17:34:10 +00:00
|
|
|
);
|
|
|
|
draw_window(
|
|
|
|
renderer,
|
|
|
|
frame,
|
|
|
|
window,
|
|
|
|
state.render_scale,
|
|
|
|
loc,
|
|
|
|
&win_damage,
|
|
|
|
&self.logger,
|
|
|
|
)?;
|
2021-11-28 00:22:00 +00:00
|
|
|
window_state(self.id, window).drawn = true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-12-13 21:06:39 +00:00
|
|
|
for layer in layer_map
|
|
|
|
.layers_on(WlrLayer::Top)
|
|
|
|
.chain(layer_map.layers_on(WlrLayer::Overlay))
|
|
|
|
{
|
|
|
|
let lgeo = layer_map.layer_geometry(layer);
|
|
|
|
if damage.iter().any(|geo| lgeo.overlaps(*geo)) {
|
|
|
|
let layer_damage = damage
|
|
|
|
.iter()
|
2021-12-20 18:12:21 +00:00
|
|
|
.flat_map(|geo| geo.intersection(lgeo))
|
2021-12-13 21:06:39 +00:00
|
|
|
.map(|geo| Rectangle::from_loc_and_size(geo.loc - lgeo.loc, geo.size))
|
|
|
|
.collect::<Vec<_>>();
|
|
|
|
slog::trace!(
|
|
|
|
self.logger,
|
|
|
|
"Rendering layer at {:?} with damage {:#?}",
|
|
|
|
lgeo,
|
2021-12-20 18:13:55 +00:00
|
|
|
layer_damage
|
2021-12-13 21:06:39 +00:00
|
|
|
);
|
|
|
|
draw_layer(
|
|
|
|
renderer,
|
|
|
|
frame,
|
|
|
|
layer,
|
|
|
|
state.render_scale,
|
|
|
|
lgeo.loc,
|
|
|
|
&layer_damage,
|
|
|
|
&self.logger,
|
|
|
|
)?;
|
|
|
|
layer_state(self.id, layer).drawn = true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-12-20 18:13:55 +00:00
|
|
|
for elem in custom_elements {
|
|
|
|
let egeo = elem.geometry();
|
|
|
|
if damage.iter().any(|geo| egeo.overlaps(*geo)) {
|
|
|
|
let elem_damage = damage
|
|
|
|
.iter()
|
|
|
|
.flat_map(|geo| geo.intersection(egeo))
|
|
|
|
.map(|geo| Rectangle::from_loc_and_size(geo.loc - egeo.loc, geo.size))
|
|
|
|
.collect::<Vec<_>>();
|
|
|
|
slog::trace!(
|
|
|
|
self.logger,
|
|
|
|
"Rendering custom element at {:?} with damage {:#?}",
|
|
|
|
egeo,
|
|
|
|
elem_damage
|
|
|
|
);
|
|
|
|
elem.draw(
|
|
|
|
renderer,
|
|
|
|
frame,
|
|
|
|
state.render_scale,
|
|
|
|
egeo.loc,
|
|
|
|
&elem_damage,
|
|
|
|
&self.logger,
|
|
|
|
)?;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-11-28 00:22:00 +00:00
|
|
|
Result::<(), R::Error>::Ok(())
|
|
|
|
},
|
|
|
|
) {
|
|
|
|
// if the rendering errors on us, we need to be prepared, that this whole buffer was partially updated and thus now unusable.
|
|
|
|
// thus clean our old states before returning
|
|
|
|
state.old_damage = VecDeque::new();
|
|
|
|
state.last_state = IndexMap::new();
|
|
|
|
return Err(RenderError::Rendering(err));
|
|
|
|
}
|
|
|
|
|
|
|
|
// If rendering was successful capture the state and add the damage
|
|
|
|
state.last_state = self
|
|
|
|
.windows
|
|
|
|
.iter()
|
|
|
|
.map(|window| {
|
2021-12-06 19:19:03 +00:00
|
|
|
let wgeo = window_rect_with_popups(window, &self.id);
|
2021-12-13 21:06:13 +00:00
|
|
|
(ToplevelId::Xdg(window.0.id), wgeo)
|
2021-11-28 00:22:00 +00:00
|
|
|
})
|
2021-12-13 21:06:39 +00:00
|
|
|
.chain(layer_map.layers().map(|layer| {
|
|
|
|
let lgeo = layer_map.layer_geometry(layer);
|
|
|
|
(ToplevelId::Layer(layer.0.id), lgeo)
|
|
|
|
}))
|
2021-12-20 18:13:55 +00:00
|
|
|
.chain(custom_elements.iter().map(|custom| {
|
|
|
|
let egeo = custom.geometry();
|
|
|
|
(ToplevelId::Custom(custom.type_of(), custom.id()), egeo)
|
|
|
|
}))
|
2021-11-28 00:22:00 +00:00
|
|
|
.collect();
|
|
|
|
state.old_damage.push_front(new_damage);
|
|
|
|
|
2021-12-07 17:34:10 +00:00
|
|
|
Ok(true)
|
2021-11-28 00:22:00 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
pub fn send_frames(&self, all: bool, time: u32) {
|
|
|
|
for window in self.windows.iter().filter(|w| {
|
|
|
|
all || {
|
|
|
|
let mut state = window_state(self.id, w);
|
|
|
|
std::mem::replace(&mut state.drawn, false)
|
|
|
|
}
|
|
|
|
}) {
|
|
|
|
window.send_frame(time);
|
|
|
|
}
|
2021-12-13 21:06:39 +00:00
|
|
|
|
|
|
|
for output in self.outputs.iter() {
|
|
|
|
let map = layer_map_for_output(output);
|
|
|
|
for layer in map.layers().filter(|l| {
|
|
|
|
all || {
|
|
|
|
let mut state = layer_state(self.id, l);
|
|
|
|
std::mem::replace(&mut state.drawn, false)
|
|
|
|
}
|
|
|
|
}) {
|
|
|
|
layer.send_frame(time);
|
|
|
|
}
|
|
|
|
}
|
2021-11-28 00:22:00 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-12-20 18:13:55 +00:00
|
|
|
pub trait RenderElement<R, F, E, T>
|
|
|
|
where
|
|
|
|
R: Renderer<Error = E, TextureId = T, Frame = F> + ImportAll,
|
|
|
|
F: Frame<Error = E, TextureId = T>,
|
|
|
|
E: std::error::Error,
|
|
|
|
T: Texture + 'static,
|
|
|
|
Self: Any + 'static,
|
|
|
|
{
|
|
|
|
fn id(&self) -> usize;
|
|
|
|
fn type_of(&self) -> TypeId {
|
|
|
|
std::any::Any::type_id(self)
|
|
|
|
}
|
|
|
|
fn geometry(&self) -> Rectangle<i32, Logical>;
|
|
|
|
fn accumulated_damage(&self, for_values: Option<(&Space, &Output)>) -> Vec<Rectangle<i32, Logical>>;
|
|
|
|
fn draw(
|
|
|
|
&self,
|
|
|
|
renderer: &mut R,
|
|
|
|
frame: &mut F,
|
|
|
|
scale: f64,
|
|
|
|
location: Point<i32, Logical>,
|
|
|
|
damage: &[Rectangle<i32, Logical>],
|
|
|
|
log: &slog::Logger,
|
|
|
|
) -> Result<(), R::Error>;
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Debug)]
|
|
|
|
pub struct SurfaceTree {
|
|
|
|
pub surface: WlSurface,
|
|
|
|
pub position: Point<i32, Logical>,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl<R, F, E, T> RenderElement<R, F, E, T> for SurfaceTree
|
|
|
|
where
|
|
|
|
R: Renderer<Error = E, TextureId = T, Frame = F> + ImportAll,
|
|
|
|
F: Frame<Error = E, TextureId = T>,
|
|
|
|
E: std::error::Error,
|
|
|
|
T: Texture + 'static,
|
|
|
|
{
|
|
|
|
fn id(&self) -> usize {
|
|
|
|
self.surface.as_ref().id() as usize
|
|
|
|
}
|
|
|
|
fn geometry(&self) -> Rectangle<i32, Logical> {
|
|
|
|
let mut bbox = super::utils::bbox_from_surface_tree(&self.surface, (0, 0));
|
|
|
|
bbox.loc += self.position;
|
|
|
|
bbox
|
|
|
|
}
|
|
|
|
|
|
|
|
fn accumulated_damage(&self, for_values: Option<(&Space, &Output)>) -> Vec<Rectangle<i32, Logical>> {
|
|
|
|
super::utils::damage_from_surface_tree(&self.surface, (0, 0), for_values)
|
|
|
|
}
|
|
|
|
|
|
|
|
fn draw(
|
|
|
|
&self,
|
|
|
|
renderer: &mut R,
|
|
|
|
frame: &mut F,
|
|
|
|
scale: f64,
|
|
|
|
location: Point<i32, Logical>,
|
|
|
|
damage: &[Rectangle<i32, Logical>],
|
|
|
|
log: &slog::Logger,
|
|
|
|
) -> Result<(), R::Error> {
|
|
|
|
crate::backend::renderer::utils::draw_surface_tree(
|
|
|
|
renderer,
|
|
|
|
frame,
|
|
|
|
&self.surface,
|
|
|
|
scale,
|
|
|
|
location,
|
|
|
|
damage,
|
|
|
|
log,
|
|
|
|
)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-11-28 00:22:00 +00:00
|
|
|
#[derive(Debug, thiserror::Error)]
|
|
|
|
pub enum RenderError<R: Renderer> {
|
|
|
|
#[error(transparent)]
|
|
|
|
Rendering(R::Error),
|
|
|
|
#[error("Output has no active mode")]
|
|
|
|
OutputNoMode,
|
|
|
|
}
|
|
|
|
|
2021-12-01 17:46:23 +00:00
|
|
|
fn window_geo(window: &Window, space_id: &usize) -> Rectangle<i32, Logical> {
|
|
|
|
let loc = window_loc(window, space_id);
|
|
|
|
let mut wgeo = window.geometry();
|
2021-12-02 11:45:58 +00:00
|
|
|
wgeo.loc = loc;
|
2021-12-01 17:46:23 +00:00
|
|
|
wgeo
|
|
|
|
}
|
|
|
|
|
2021-11-28 00:22:00 +00:00
|
|
|
fn window_rect(window: &Window, space_id: &usize) -> Rectangle<i32, Logical> {
|
|
|
|
let loc = window_loc(window, space_id);
|
2021-12-01 17:46:23 +00:00
|
|
|
let mut wgeo = window.bbox();
|
|
|
|
wgeo.loc += loc;
|
|
|
|
wgeo
|
2021-11-28 00:22:00 +00:00
|
|
|
}
|
|
|
|
|
2021-12-06 19:19:03 +00:00
|
|
|
fn window_rect_with_popups(window: &Window, space_id: &usize) -> Rectangle<i32, Logical> {
|
|
|
|
let loc = window_loc(window, space_id);
|
|
|
|
let mut wgeo = window.bbox_with_popups();
|
|
|
|
wgeo.loc += loc;
|
|
|
|
wgeo
|
|
|
|
}
|
|
|
|
|
2021-11-28 00:22:00 +00:00
|
|
|
fn window_loc(window: &Window, space_id: &usize) -> Point<i32, Logical> {
|
|
|
|
window
|
|
|
|
.user_data()
|
|
|
|
.get::<RefCell<HashMap<usize, WindowState>>>()
|
|
|
|
.unwrap()
|
|
|
|
.borrow()
|
|
|
|
.get(space_id)
|
|
|
|
.unwrap()
|
|
|
|
.location
|
|
|
|
}
|