1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
use smithay::backend::renderer::gles::{GlesRenderer, GlesTexture};
use smithay::backend::renderer::utils::{import_surface, RendererSurfaceStateUserData};
use smithay::backend::renderer::Renderer as _;
use smithay::reexports::wayland_server::protocol::wl_surface::WlSurface;
use smithay::utils::{Logical, Point};
use smithay::wayland::compositor::{with_surface_tree_downward, TraversalAction};
use super::texture::TextureBuffer;
use super::BakedBuffer;
/// Renders elements from a surface tree as textures into `storage`.
pub fn render_snapshot_from_surface_tree(
renderer: &mut GlesRenderer,
surface: &WlSurface,
location: Point<f64, Logical>,
storage: &mut Vec<BakedBuffer<TextureBuffer<GlesTexture>>>,
) {
let _span = tracy_client::span!("render_snapshot_from_surface_tree");
with_surface_tree_downward(
surface,
location,
|_, states, location| {
let mut location = *location;
let data = states.data_map.get::<RendererSurfaceStateUserData>();
if let Some(data) = data {
let data = &*data.lock().unwrap();
if let Some(view) = data.view() {
location += view.offset.to_f64();
TraversalAction::DoChildren(location)
} else {
TraversalAction::SkipChildren
}
} else {
TraversalAction::SkipChildren
}
},
|_, states, location| {
let mut location = *location;
let data = states.data_map.get::<RendererSurfaceStateUserData>();
if let Some(data) = data {
let Some(view) = data.lock().unwrap().view() else {
return;
};
location += view.offset.to_f64();
if let Err(err) = import_surface(renderer, states) {
warn!("failed to import surface: {err:?}");
return;
}
let data = data.lock().unwrap();
let Some(texture) = data.texture(renderer.context_id()) else {
return;
};
let buffer = TextureBuffer::from_texture(
renderer,
texture.clone(),
f64::from(data.buffer_scale()),
data.buffer_transform(),
Vec::new(),
);
let baked = BakedBuffer {
buffer,
location,
src: Some(view.src),
dst: Some(view.dst),
};
storage.push(baked);
}
},
|_, _, _| true,
);
}
|
