// This file is part of the 64k demo project.
// It implements the DistortEffect.

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

// --- DistortEffect ---
DistortEffect::DistortEffect(const GpuContext& ctx)
    : DistortEffect(ctx, DistortParams()) {
}

DistortEffect::DistortEffect(const GpuContext& ctx, const DistortParams& params)
    : PostProcessEffect(ctx), params_(params) {
  params_buffer_.init(ctx_.device);
  pipeline_ = create_post_process_pipeline(ctx_.device, ctx_.format,
                                           distort_shader_wgsl);
}

void DistortEffect::render(WGPURenderPassEncoder pass,
                           const CommonPostProcessUniforms& uniforms) {
  // Populate CommonPostProcessUniforms
  uniforms_.update(ctx_.queue, uniforms);

  // Populate DistortParams
  const DistortParams distort_p = {
      .strength = params_.strength,
      .speed = params_.speed,
  };
  params_buffer_.update(ctx_.queue, distort_p);

  PostProcessEffect::render(pass, uniforms);
}

void DistortEffect::update_bind_group(WGPUTextureView v) {
  pp_update_bind_group(ctx_.device, pipeline_, &bind_group_, v, uniforms_.get(), params_buffer_);
}