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
|
use bevy::{
prelude::*,
render::{
render_asset::{RenderAssetUsages, RenderAssets},
render_resource::{Extent3d, Maintain, MapMode, TextureDimension},
renderer::RenderDevice,
},
};
use pollster::FutureExt;
use crate::{
components::{HeadlessRenderDestination, HeadlessRenderSource},
render_assets::GpuHeadlessRenderSource,
};
pub fn copy_buffers(
mut headless_render_query: Query<(&HeadlessRenderSource, &mut HeadlessRenderDestination)>,
sources: Res<RenderAssets<GpuHeadlessRenderSource>>,
device: Res<RenderDevice>,
) {
for (source_handle, destination_handle) in headless_render_query.iter_mut() {
let Some(gpu_source) = sources.get(source_handle.id()) else {
continue;
};
let mut image_bytes = {
let slice = gpu_source.buffer.slice(..);
{
let (tx, rx) = oneshot::channel();
device.map_buffer(&slice, MapMode::Read, move |res| {
tx.send(res).unwrap();
});
device.poll(Maintain::Wait);
rx.block_on().unwrap().unwrap();
}
slice.get_mapped_range().to_vec()
};
gpu_source.buffer.unmap();
let bytes_per_row = gpu_source.bytes_per_row as usize;
let padded_bytes_per_row = gpu_source.padded_bytes_per_row as usize;
let source_size = gpu_source.source_size;
let destination_handle = destination_handle.clone();
let source_format = gpu_source.format;
std::thread::spawn(move || {
if bytes_per_row != padded_bytes_per_row {
let mut unpadded_bytes =
Vec::<u8>::with_capacity(source_size.height as usize * bytes_per_row);
for padded_row in image_bytes.chunks(padded_bytes_per_row) {
unpadded_bytes.extend_from_slice(&padded_row[..bytes_per_row]);
}
image_bytes = unpadded_bytes;
}
*destination_handle.0.lock().unwrap() = Image::new(
Extent3d {
width: source_size.width,
height: source_size.height,
depth_or_array_layers: 1,
},
TextureDimension::D2,
image_bytes,
source_format,
RenderAssetUsages::MAIN_WORLD | RenderAssetUsages::RENDER_WORLD,
);
});
}
}
|
