1 files changed, 30 insertions, 1 deletions
diff --git a/src/gpu/demo_effects.h b/src/gpu/demo_effects.h
index d9487fa..36be118 100644
--- a/src/gpu/demo_effects.h
+++ b/src/gpu/demo_effects.h
@@ -6,11 +6,12 @@
 #include "3d/renderer.h"
 #include "3d/scene.h"
 #include "effect.h"
-#include "gpu/effects/flash_effect.h"  // FlashEffect with params support
+#include "gpu/effects/flash_effect.h" // FlashEffect with params support
 #include "gpu/effects/post_process_helper.h"
 #include "gpu/effects/shaders.h"
 #include "gpu/gpu.h"
 #include "gpu/texture_manager.h"
+#include "gpu/uniform_helper.h"
 #include <memory>
 
 static const int NUM_PARTICLES = 10000;
@@ -100,12 +101,40 @@ class DistortEffect : public PostProcessEffect {
   void update_bind_group(WGPUTextureView input_view) override;
 };
 
+// Parameters for ChromaAberrationEffect (set at construction time)
+struct ChromaAberrationParams {
+  float offset_scale = 0.02f; // Default: 2% screen offset
+  float angle = 0.0f;         // Default: horizontal (0 radians)
+};
+
+// Uniform data sent to GPU shader
+struct ChromaUniforms {
+  float time;         // offset 0
+  float beat;         // offset 4
+  float intensity;    // offset 8
+  float aspect_ratio; // offset 12
+  float width;        // offset 16
+  float height;       // offset 20
+  float offset_scale; // offset 24
+  float angle;        // offset 28
+};
+static_assert(sizeof(ChromaUniforms) == 32,
+              "ChromaUniforms must be 32 bytes for WGSL alignment");
+
 class ChromaAberrationEffect : public PostProcessEffect {
  public:
+  // Backward compatibility constructor (uses default params)
   ChromaAberrationEffect(const GpuContext& ctx);
+  // New parameterized constructor
+  ChromaAberrationEffect(const GpuContext& ctx,
+                         const ChromaAberrationParams& params);
   void render(WGPURenderPassEncoder pass, float time, float beat,
               float intensity, float aspect_ratio) override;
   void update_bind_group(WGPUTextureView input_view) override;
+
+ private:
+  ChromaAberrationParams params_;
+  UniformBuffer<ChromaUniforms> uniforms_;
 };
 
 class Hybrid3DEffect : public Effect {