bevy_core_pipeline/oit/resolve/
node.rs

1use bevy_camera::{MainPassResolutionOverride, Viewport};
2use bevy_ecs::{prelude::*, query::QueryItem};
3use bevy_render::{
4    camera::ExtractedCamera,
5    diagnostic::RecordDiagnostics,
6    render_graph::{NodeRunError, RenderGraphContext, RenderLabel, ViewNode},
7    render_resource::{BindGroupEntries, PipelineCache, RenderPassDescriptor},
8    renderer::RenderContext,
9    view::{ViewDepthTexture, ViewTarget, ViewUniformOffset},
10};
11
12use super::{OitResolveBindGroup, OitResolvePipeline, OitResolvePipelineId};
13
14/// Render label for the OIT resolve pass.
15#[derive(RenderLabel, Debug, Clone, Hash, PartialEq, Eq)]
16pub struct OitResolvePass;
17
18/// The node that executes the OIT resolve pass.
19#[derive(Default)]
20pub struct OitResolveNode;
21impl ViewNode for OitResolveNode {
22    type ViewQuery = (
23        &'static ExtractedCamera,
24        &'static ViewTarget,
25        &'static ViewUniformOffset,
26        &'static OitResolvePipelineId,
27        &'static ViewDepthTexture,
28        Option<&'static MainPassResolutionOverride>,
29    );
30
31    fn run(
32        &self,
33        _graph: &mut RenderGraphContext,
34        render_context: &mut RenderContext,
35        (camera, view_target, view_uniform, oit_resolve_pipeline_id, depth, resolution_override): QueryItem<
36            Self::ViewQuery,
37        >,
38        world: &World,
39    ) -> Result<(), NodeRunError> {
40        let Some(resolve_pipeline) = world.get_resource::<OitResolvePipeline>() else {
41            return Ok(());
42        };
43
44        // resolve oit
45        // sorts the layers and renders the final blended color to the screen
46        {
47            let pipeline_cache = world.resource::<PipelineCache>();
48            let bind_group = world.resource::<OitResolveBindGroup>();
49            let Some(pipeline) = pipeline_cache.get_render_pipeline(oit_resolve_pipeline_id.0)
50            else {
51                return Ok(());
52            };
53
54            let diagnostics = render_context.diagnostic_recorder();
55
56            let depth_bind_group = render_context.render_device().create_bind_group(
57                "oit_resolve_depth_bind_group",
58                &resolve_pipeline.oit_depth_bind_group_layout,
59                &BindGroupEntries::single(depth.view()),
60            );
61
62            let mut render_pass = render_context.begin_tracked_render_pass(RenderPassDescriptor {
63                label: Some("oit_resolve"),
64                color_attachments: &[Some(view_target.get_color_attachment())],
65                depth_stencil_attachment: None,
66                timestamp_writes: None,
67                occlusion_query_set: None,
68            });
69            let pass_span = diagnostics.pass_span(&mut render_pass, "oit_resolve");
70
71            if let Some(viewport) =
72                Viewport::from_viewport_and_override(camera.viewport.as_ref(), resolution_override)
73            {
74                render_pass.set_camera_viewport(&viewport);
75            }
76
77            render_pass.set_render_pipeline(pipeline);
78            render_pass.set_bind_group(0, bind_group, &[view_uniform.offset]);
79            render_pass.set_bind_group(1, &depth_bind_group, &[]);
80
81            render_pass.draw(0..3, 0..1);
82
83            pass_span.end(&mut render_pass);
84        }
85
86        Ok(())
87    }
88}