aboutsummaryrefslogtreecommitdiff
path: root/src/lib.rs
blob: fdf3fa84fbcee45e86a93940ded7a9c5bcc5dd41 (plain)
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
use bevy::{prelude::*, render::{RenderApp, render_graph::{RenderGraphApp, ViewNodeRunner}, extract_component::ExtractComponentPlugin}, asset::embedded_asset, core_pipeline::core_3d::graph::{Core3d, Node3d}};

use crate::components::DitherPostProcessSettings;

pub struct DitherPostProcessPlugin;

pub mod components;
mod resources;
mod nodes;

impl Plugin for DitherPostProcessPlugin {
    fn build(&self, app: &mut App) {
        embedded_asset!(app, "../assets/shaders/dither_post_process.wgsl");

        app.add_plugins((
            ExtractComponentPlugin::<DitherPostProcessSettings>::default(),
        ));

        let Ok(render_app) = app.get_sub_app_mut(RenderApp) else {
            return;
        };

        render_app.add_render_graph_node::<ViewNodeRunner<nodes::DitherRenderNode>>(
            Core3d,
            nodes::DitherRenderLabel,
        ).add_render_graph_edges(
            Core3d, 
            (
                Node3d::Tonemapping,
                nodes::DitherRenderLabel,
                Node3d::EndMainPassPostProcessing,
            ),
        );
    }
    
    fn finish(&self, app: &mut App) {
        let Ok(render_app) = app.get_sub_app_mut(RenderApp) else {
            return;
        };

        render_app.init_resource::<resources::DitherPostProcessPipeline>();
    }
}