some tweaking

1 files changed, 4 insertions, 8 deletions

diff --git a/workspaces/main/shaders/chroma_aberration.wgsl b/workspaces/main/shaders/chroma_aberration.wgsl
index ee730b1..d5e50ea 100644
--- a/workspaces/main/shaders/chroma_aberration.wgsl
+++ b/workspaces/main/shaders/chroma_aberration.wgsl
@@ -15,14 +15,10 @@ struct ChromaAberrationParams {
 @fragment fn fs_main(@builtin(position) p: vec4<f32>) -> @location(0) vec4<f32> {
     let uv = p.xy / uniforms.resolution;
 
-    // Compute offset magnitude and direction
-    let offset_mag = params.offset_scale * uniforms.audio_intensity;
-    let offset_dir = vec2<f32>(cos(params.angle), sin(params.angle));
-    let offset = offset_mag * offset_dir;
-
-    // Sample RGB channels with chromatic aberration
+    let amp = params.offset_scale * uniforms.audio_intensity;
+    let offset = amp * vec2<f32>(cos(params.angle), sin(params.angle));
+    let center = textureSample(txt, smplr, uv);
     let r = textureSample(txt, smplr, uv + offset).r;
-    let g = textureSample(txt, smplr, uv).g;
     let b = textureSample(txt, smplr, uv - offset).b;
-    return vec4<f32>(r, g, b, 1.0);
+    return vec4<f32>(r, center.g, b, 1.0);
 }