// 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::DistEffect(const GpuContext& ctx, const DistortParams& params)
    : PostProcessEffect(ctx), params_(params) {
  common_uniforms_.init(ctx_.device);
  params_buffer_.init(ctx_.device);
  pipeline_ = create_post_process_pipeline(ctx_.device, ctx_.format,
                                           distort_shader_wgsl);
}

void DistortEffect::render(WGPURenderPassEncoder pass, float t, float b,
                           float i, float a) {
  // Populate CommonPostProcessUniforms
  const CommonPostProcessUniforms common_u = {
      .resolution = {(float)width_, (float)height_},
      .aspect_ratio = a,
      .time = t,
      .beat = b,
      .audio_intensity = i,
  };
  common_uniforms_.update(ctx_.queue, common_u);

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

  PostProcessEffect::render(pass, t, b, i, a);
}

void DistortEffect::update_bind_group(WGPUTextureView v) {

  pp_update_bind_group(ctx_.device, pipeline_, &bind_group_, v, common_uniforms_.get(), params_buffer_);

}