// This file is part of the 64k demo project.
// It implements the GaussianBlurEffect with parameterization.

#include "gpu/demo_effects.h"
#include "gpu/effects/post_process_helper.h"
#include "gpu/gpu.h"

// --- GaussianBlurEffect ---

// Backward compatibility constructor (delegates to parameterized constructor)
GaussianBlurEffect::GaussianBlurEffect(const GpuContext& ctx)
    : GaussianBlurEffect(ctx, GaussianBlurParams{}) {
}

// Parameterized constructor
GaussianBlurEffect::GaussianBlurEffect(const GpuContext& ctx,
                                       const GaussianBlurParams& params)
    : PostProcessEffect(ctx), params_(params) {
  pipeline_ = create_post_process_pipeline(ctx_.device, ctx_.format,
                                           gaussian_blur_shader_wgsl);
  uniforms_.init(ctx_.device);
}

void GaussianBlurEffect::render(WGPURenderPassEncoder pass, float time,
                                float beat, float intensity,
                                float aspect_ratio) {
  // Update uniforms with current state and parameters
  const GaussianBlurUniforms u = {.time = time,
                                  .beat = beat,
                                  .intensity = intensity,
                                  .aspect_ratio = aspect_ratio,
                                  .width = (float)width_,
                                  .height = (float)height_,
                                  .strength = params_.strength,
                                  ._pad = 0.0f};
  uniforms_.update(ctx_.queue, u);

  wgpuRenderPassEncoderSetPipeline(pass, pipeline_);
  wgpuRenderPassEncoderSetBindGroup(pass, 0, bind_group_, 0, nullptr);
  wgpuRenderPassEncoderDraw(pass, 3, 1, 0, 0);
}

void GaussianBlurEffect::update_bind_group(WGPUTextureView input_view) {
  pp_update_bind_group(ctx_.device, pipeline_, &bind_group_, input_view,
                       uniforms_.get());
}