plurimus_3d 0.7.0

3d pipeline for plurimus: GPU camera readback converted to terminal cells.
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

//! Render-world copy of the view's depth texture into the readback
//! buffer, scheduled in `Core3d` after the main pass.

use bevy_asset::Handle;
use bevy_ecs::prelude::{Component, Res};
use bevy_log::warn_once;
use bevy_render::extract_component::ExtractComponent;
use bevy_render::render_asset::RenderAssets;
use bevy_render::render_resource::{TexelCopyBufferInfo, TexelCopyBufferLayout};
use bevy_render::renderer::{RenderContext, ViewQuery};
use bevy_render::storage::{GpuShaderBuffer, ShaderBuffer};
use bevy_render::view::ViewDepthTexture;

use crate::depth::depth_row_stride;

/// The camera's depth-readback buffer, extracted for the copy pass.
#[derive(Component, Clone, ExtractComponent)]
pub(crate) struct DepthCopyTarget(pub(crate) Handle<ShaderBuffer>);

pub(crate) fn copy_depth_to_readback(
    view: ViewQuery<(&ViewDepthTexture, &DepthCopyTarget)>,
    buffers: Res<RenderAssets<GpuShaderBuffer>>,
    mut ctx: RenderContext,
) {
    let (depth, target) = view.into_inner();
    let Some(buffer) = buffers.get(&target.0) else {
        return;
    };
    if depth.texture.sample_count() > 1 {
        warn_once!("depth readback needs Msaa::Off; multisampled depth cannot be copied");
        return;
    }
    let size = depth.texture.size();
    let stride = depth_row_stride(size.width);
    if buffer.buffer.size() < (stride * size.height as usize) as u64 {
        return;
    }
    ctx.command_encoder().copy_texture_to_buffer(
        depth.texture.as_image_copy(),
        TexelCopyBufferInfo {
            buffer: &buffer.buffer,
            layout: TexelCopyBufferLayout {
                offset: 0,
                bytes_per_row: Some(stride as u32),
                rows_per_image: None,
            },
        },
        size,
    );
}