bevy_dither_post_process 0.3.1

A post-process black and white ordered dithering effect for the Bevy game engine.
Documentation 
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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97

#![warn(missing_docs)]

//! A plugin for the Bevy game engine which provides a black and white dither post-process effect
//! using Bayer ordered dithering.

use bevy::{
    asset::embedded_asset,
    core_pipeline::core_3d::graph::{Core3d, Node3d},
    prelude::*,
    render::{
        extract_component::ExtractComponentPlugin,
        render_graph::{RenderGraphApp, ViewNodeRunner},
        render_resource::{PipelineCache, SpecializedRenderPipelines, TextureFormat},
        view::{ExtractedView, ViewTarget},
        Render, RenderApp, RenderSet,
    },
};
use components::DitherPostProcessPipelineId;
use resources::{DitherPostProcessPipeline, DitherPostProcessingPipelineKey};

use crate::components::DitherPostProcessSettings;

pub use nodes::DitherRenderLabel;

/// Plugin which provides dither post-processing functionality
pub struct DitherPostProcessPlugin;

/// Components used by this plugin
pub mod components;

mod nodes;
mod resources;

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 Some(render_app) = app.get_sub_app_mut(RenderApp) else {
            return;
        };

        render_app
            .add_systems(
                Render,
                prepare_post_processing_pipelines.in_set(RenderSet::Prepare),
            )
            .init_resource::<SpecializedRenderPipelines<DitherPostProcessPipeline>>()
            .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 Some(render_app) = app.get_sub_app_mut(RenderApp) else {
            return;
        };

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

fn prepare_post_processing_pipelines(
    mut commands: Commands,
    pipeline_cache: Res<PipelineCache>,
    mut pipelines: ResMut<SpecializedRenderPipelines<DitherPostProcessPipeline>>,
    post_processing_pipeline: Res<DitherPostProcessPipeline>,
    views: Query<(Entity, &ExtractedView), With<DitherPostProcessSettings>>,
) {
    for (entity, view) in views.iter() {
        let pipeline_id = pipelines.specialize(
            &pipeline_cache,
            &post_processing_pipeline,
            DitherPostProcessingPipelineKey {
                texture_format: if view.hdr {
                    ViewTarget::TEXTURE_FORMAT_HDR
                } else {
                    TextureFormat::bevy_default()
                },
            },
        );

        commands
            .entity(entity)
            .insert(DitherPostProcessPipelineId(pipeline_id));
    }
}