Update world.wgsl

world.wgsl

@@ -1,15 +1,20 @@
+struct Params {
+  time : f32,
+  power : f32,
+}
+
 @group(0) @binding(0)
-var<storage, read_write> field : array<f32>;
+var<uniform> params : Params;
 
 @group(0) @binding(1)
-var<uniform> params : vec2<f32>; // power, time
+var<storage, read_write> pixels : array<f32>;
 
-fn mandelbulb(p: vec3<f32>, power: f32) -> f32 {
+fn mandelbulbDE(p: vec3<f32>, power: f32) -> f32 {
   var z = p;
   var dr = 1.0;
   var r = 0.0;
 
-  for (var i = 0; i < 8; i++) {
+  for (var i = 0; i < 10; i++) {
     r = length(z);
     if (r > 2.0) { break; }
 
@@ -26,16 +31,38 @@ fn mandelbulb(p: vec3<f32>, power: f32) -> f32 {
     ) + p;
   }
 
-  return clamp(1.0 - r / dr, 0.0, 1.0);
+  return 0.5 * log(r) * r / dr;
+}
+
+fn raymarch(ro: vec3<f32>, rd: vec3<f32>) -> f32 {
+  var t = 0.0;
+  for (var i = 0; i < 64; i++) {
+    let p = ro + rd * t;
+    let d = mandelbulbDE(p, params.power);
+    if (d < 0.001) { return 1.0 - f32(i) / 64.0; }
+    t += d;
+    if (t > 10.0) { break; }
+  }
+  return 0.0;
 }
 
-@compute @workgroup_size(64)
+@compute @workgroup_size(8,8)
 fn main(@builtin(global_invocation_id) gid : vec3<u32>) {
-  let i = gid.x;
-  if (i >= arrayLength(&field)) { return; }
+  let width = 256u;
+  let height = 256u;
+
+  if (gid.x >= width || gid.y >= height) { return; }
+
+  let uv = (vec2<f32>(gid.xy) / vec2<f32>(width, height)) * 2.0 - 1.0;
+
+  let ro = vec3<f32>(0.0, 0.0, -4.0 + sin(params.time) * 0.5);
+  let rd = normalize(vec3<f32>(uv, 1.5));
 
-  let x = f32(i) / f32(arrayLength(&field)) * 2.0 - 1.0;
-  let p = vec3<f32>(x, 0.0, sin(params.y) * 0.5);
+  let c = raymarch(ro, rd);
+  let idx = (gid.y * width + gid.x) * 4u;
 
-  field[i] = mandelbulb(p, params.x);
+  pixels[idx + 0u] = c;
+  pixels[idx + 1u] = c * 0.8;
+  pixels[idx + 2u] = c * 1.2;
+  pixels[idx + 3u] = 1.0;
 }