summaryrefslogtreecommitdiff
path: root/src/gpu
diff options
context:
space:
mode:
Diffstat (limited to 'src/gpu')
-rw-r--r--src/gpu/demo_effects.h1
-rw-r--r--src/gpu/wgsl_effect.cc54
-rw-r--r--src/gpu/wgsl_effect.h38
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_;
+};