diff options
Diffstat (limited to 'src/gpu')
| -rw-r--r-- | src/gpu/demo_effects.h | 1 | ||||
| -rw-r--r-- | src/gpu/wgsl_effect.cc | 54 | ||||
| -rw-r--r-- | src/gpu/wgsl_effect.h | 38 |
3 files changed, 93 insertions, 0 deletions
diff --git a/src/gpu/demo_effects.h b/src/gpu/demo_effects.h index ecc37df..a2500e7 100644 --- a/src/gpu/demo_effects.h +++ b/src/gpu/demo_effects.h @@ -15,6 +15,7 @@ #include "gpu/sequence.h" #include "gpu/texture_manager.h" #include "gpu/uniform_helper.h" +#include "gpu/wgsl_effect.h" // Individual Effect Headers #include "effects/flash_effect.h" diff --git a/src/gpu/wgsl_effect.cc b/src/gpu/wgsl_effect.cc new file mode 100644 index 0000000..0ce4730 --- /dev/null +++ b/src/gpu/wgsl_effect.cc @@ -0,0 +1,54 @@ +// WgslEffect: generic shader-only post-process effect. + +#include "gpu/wgsl_effect.h" +#include "gpu/gpu.h" +#include "gpu/post_process_helper.h" +#include "util/fatal_error.h" + +WgslEffect::WgslEffect(const GpuContext& ctx, + const std::vector<std::string>& inputs, + const std::vector<std::string>& outputs, + float start_time, float end_time, + const char* shader_code, WGPULoadOp load_op, + WgslEffectParams initial_params) + : Effect(ctx, inputs, outputs, start_time, end_time), + effect_params(initial_params), + load_op_(load_op) { + HEADLESS_RETURN_IF_NULL(ctx_.device); + + create_linear_sampler(); + params_buffer_.init(ctx_.device); + + pipeline_.set(create_post_process_pipeline(ctx_.device, + WGPUTextureFormat_RGBA8Unorm, + shader_code)); +} + +void WgslEffect::render(WGPUCommandEncoder encoder, + const UniformsSequenceParams& params, + NodeRegistry& nodes) { + WGPUTextureView input_view = nodes.get_view(input_nodes_[0]); + WGPUTextureView output_view = nodes.get_view(output_nodes_[0]); + + params_buffer_.update(ctx_.queue, effect_params); + + pp_update_bind_group(ctx_.device, pipeline_.get(), bind_group_.get_address(), + input_view, uniforms_buffer_.get(), + params_buffer_.get()); + + WGPURenderPassColorAttachment color_attachment = {}; + gpu_init_color_attachment(color_attachment, output_view); + color_attachment.loadOp = load_op_; + + WGPURenderPassDescriptor pass_desc = {}; + pass_desc.colorAttachmentCount = 1; + pass_desc.colorAttachments = &color_attachment; + + WGPURenderPassEncoder pass = + wgpuCommandEncoderBeginRenderPass(encoder, &pass_desc); + wgpuRenderPassEncoderSetPipeline(pass, pipeline_.get()); + wgpuRenderPassEncoderSetBindGroup(pass, 0, bind_group_.get(), 0, nullptr); + wgpuRenderPassEncoderDraw(pass, 3, 1, 0, 0); + wgpuRenderPassEncoderEnd(pass); + wgpuRenderPassEncoderRelease(pass); +} diff --git a/src/gpu/wgsl_effect.h b/src/gpu/wgsl_effect.h new file mode 100644 index 0000000..062f885 --- /dev/null +++ b/src/gpu/wgsl_effect.h @@ -0,0 +1,38 @@ +// WgslEffect: generic shader-only post-process effect. +// Replaces boilerplate .h/.cc pairs for simple single-pass effects. + +#pragma once +#include "gpu/effect.h" +#include "gpu/uniform_helper.h" +#include "gpu/wgpu_resource.h" + +// Generic per-effect uniform params (binding 3, 32 bytes). +// Matches WgslEffectParams in WGSL: struct { p: vec4f, c: vec4f } +// effect_params.p — 4 generic floats (strength, scale, etc.) +// effect_params.c — color or secondary vec4 +struct WgslEffectParams { + float p[4]; // vec4: generic float params + float c[4]; // vec4: color / secondary params +}; +static_assert(sizeof(WgslEffectParams) == 32, "WgslEffectParams must be 32 bytes"); + +class WgslEffect : public Effect { + public: + // Mutate per-frame for dynamic parameter modulation. + WgslEffectParams effect_params; + + WgslEffect(const GpuContext& ctx, const std::vector<std::string>& inputs, + const std::vector<std::string>& outputs, float start_time, + float end_time, const char* shader_code, + WGPULoadOp load_op = WGPULoadOp_Clear, + WgslEffectParams initial_params = {}); + + void render(WGPUCommandEncoder encoder, const UniformsSequenceParams& params, + NodeRegistry& nodes) override; + + private: + RenderPipeline pipeline_; + BindGroup bind_group_; + UniformBuffer<WgslEffectParams> params_buffer_; + WGPULoadOp load_op_; +}; |
