summaryrefslogtreecommitdiff
path: root/src/effects/gaussian_blur_effect_v2.cc
diff options
context:
space:
mode:
Diffstat (limited to 'src/effects/gaussian_blur_effect_v2.cc')
-rw-r--r--src/effects/gaussian_blur_effect_v2.cc80
1 files changed, 80 insertions, 0 deletions
diff --git a/src/effects/gaussian_blur_effect_v2.cc b/src/effects/gaussian_blur_effect_v2.cc
new file mode 100644
index 0000000..6b37f0b
--- /dev/null
+++ b/src/effects/gaussian_blur_effect_v2.cc
@@ -0,0 +1,80 @@
+// Gaussian blur effect v2 implementation
+
+#include "effects/gaussian_blur_effect_v2.h"
+#include "gpu/post_process_helper.h"
+#include "gpu/shaders.h"
+
+GaussianBlurEffectV2::GaussianBlurEffectV2(const GpuContext& ctx,
+ const std::vector<std::string>& inputs,
+ const std::vector<std::string>& outputs)
+ : EffectV2(ctx, inputs, outputs), pipeline_(nullptr), bind_group_(nullptr),
+ sampler_(nullptr) {
+ // Create pipeline
+ pipeline_ = create_post_process_pipeline(ctx_.device, WGPUTextureFormat_RGBA8Unorm,
+ gaussian_blur_shader_wgsl);
+
+ // Create sampler
+ WGPUSamplerDescriptor sampler_desc = {};
+ sampler_desc.addressModeU = WGPUAddressMode_ClampToEdge;
+ sampler_desc.addressModeV = WGPUAddressMode_ClampToEdge;
+ sampler_desc.magFilter = WGPUFilterMode_Linear;
+ sampler_desc.minFilter = WGPUFilterMode_Linear;
+ sampler_ = wgpuDeviceCreateSampler(ctx_.device, &sampler_desc);
+
+ // Init uniform buffers
+ params_buffer_.init(ctx_.device);
+ uniforms_buffer_.init(ctx_.device);
+}
+
+void GaussianBlurEffectV2::render(WGPUCommandEncoder encoder,
+ const UniformsSequenceParams& params,
+ NodeRegistry& nodes) {
+ // Get input/output views
+ WGPUTextureView input_view = nodes.get_view(input_nodes_[0]);
+ WGPUTextureView output_view = nodes.get_view(output_nodes_[0]);
+
+ // Update uniforms
+ uniforms_buffer_.update(ctx_.queue, params);
+ params_buffer_.update(ctx_.queue, blur_params_);
+
+ // Update bind group
+ WGPUBindGroupEntry entries[4] = {};
+ entries[0].binding = PP_BINDING_SAMPLER;
+ entries[0].sampler = sampler_;
+ entries[1].binding = PP_BINDING_TEXTURE;
+ entries[1].textureView = input_view;
+ entries[2].binding = PP_BINDING_UNIFORMS;
+ entries[2].buffer = uniforms_buffer_.get().buffer;
+ entries[2].size = sizeof(UniformsSequenceParams);
+ entries[3].binding = PP_BINDING_EFFECT_PARAMS;
+ entries[3].buffer = params_buffer_.get().buffer;
+ entries[3].size = sizeof(GaussianBlurParams);
+
+ WGPUBindGroupDescriptor bg_desc = {};
+ bg_desc.layout = wgpuRenderPipelineGetBindGroupLayout(pipeline_, 0);
+ bg_desc.entryCount = 4;
+ bg_desc.entries = entries;
+
+ if (bind_group_) {
+ wgpuBindGroupRelease(bind_group_);
+ }
+ bind_group_ = wgpuDeviceCreateBindGroup(ctx_.device, &bg_desc);
+
+ // Render pass
+ WGPURenderPassColorAttachment color_attachment = {};
+ color_attachment.view = output_view;
+ color_attachment.loadOp = WGPULoadOp_Clear;
+ color_attachment.storeOp = WGPUStoreOp_Store;
+ color_attachment.clearValue = {0.0, 0.0, 0.0, 1.0};
+
+ WGPURenderPassDescriptor pass_desc = {};
+ pass_desc.colorAttachmentCount = 1;
+ pass_desc.colorAttachments = &color_attachment;
+
+ WGPURenderPassEncoder pass = wgpuCommandEncoderBeginRenderPass(encoder, &pass_desc);
+ wgpuRenderPassEncoderSetPipeline(pass, pipeline_);
+ wgpuRenderPassEncoderSetBindGroup(pass, 0, bind_group_, 0, nullptr);
+ wgpuRenderPassEncoderDraw(pass, 3, 1, 0, 0);
+ wgpuRenderPassEncoderEnd(pass);
+ wgpuRenderPassEncoderRelease(pass);
+}