1 files changed, 14 insertions, 5 deletions
diff --git a/assets/final/shaders/chroma_aberration.wgsl b/assets/final/shaders/chroma_aberration.wgsl
index cd80afa..f1a3034 100644
--- a/assets/final/shaders/chroma_aberration.wgsl
+++ b/assets/final/shaders/chroma_aberration.wgsl
@@ -6,7 +6,10 @@ struct Uniforms {
     beat: f32,
     intensity: f32,
     aspect_ratio: f32,
-    resolution: vec2<f32>,
+    width: f32,
+    height: f32,
+    offset_scale: f32,
+    angle: f32,
 };
 
 @group(0) @binding(2) var<uniform> uniforms: Uniforms;
@@ -21,10 +24,16 @@ struct Uniforms {
 }
 
 @fragment fn fs_main(@builtin(position) p: vec4<f32>) -> @location(0) vec4<f32> {
-    let uv = p.xy / uniforms.resolution;
-    let off = 0.02 * uniforms.intensity;
-    let r = textureSample(txt, smplr, uv + vec2<f32>(off, 0.0)).r;
+    let uv = p.xy / vec2<f32>(uniforms.width, uniforms.height);
+
+    // Compute offset magnitude and direction
+    let offset_mag = uniforms.offset_scale * uniforms.intensity;
+    let offset_dir = vec2<f32>(cos(uniforms.angle), sin(uniforms.angle));
+    let offset = offset_mag * offset_dir;
+
+    // Sample RGB channels with chromatic aberration
+    let r = textureSample(txt, smplr, uv + offset).r;
     let g = textureSample(txt, smplr, uv).g;
-    let b = textureSample(txt, smplr, uv - vec2<f32>(off, 0.0)).b;
+    let b = textureSample(txt, smplr, uv - offset).b;
     return vec4<f32>(r, g, b, 1.0);
 }