Update world.wgsl

world.wgsl

@@ -9,12 +9,12 @@ var<uniform> params : Params;
 @group(0) @binding(1)
 var<storage, read_write> pixels : array<f32>;
 
-fn mandelbulbDE(p: vec3<f32>, power: f32) -> f32 {
+fn mandelbulb(p: vec3<f32>, power: f32) -> f32 {
   var z = p;
   var dr = 1.0;
   var r = 0.0;
 
-  for (var i = 0; i < 10; i++) {
+  for (var i = 0; i < 8; i++) {
     r = length(z);
     if (r > 2.0) { break; }
 
@@ -30,39 +30,31 @@ fn mandelbulbDE(p: vec3<f32>, power: f32) -> f32 {
       cos(theta * power)
     ) + p;
   }
-
   return 0.5 * log(r) * r / dr;
 }
 
-fn raymarch(ro: vec3<f32>, rd: vec3<f32>) -> f32 {
+@compute @workgroup_size(8,8)
+fn main(@builtin(global_invocation_id) id : vec3<u32>) {
+  let w = 256u;
+  let h = 256u;
+  if (id.x >= w || id.y >= h) { return; }
+
+  let uv = (vec2<f32>(id.xy) / vec2<f32>(w, h)) * 2.0 - 1.0;
+  let ro = vec3<f32>(0.0, 0.0, -4.0 + sin(params.time));
+  let rd = normalize(vec3<f32>(uv, 1.5));
+
   var t = 0.0;
+  var hit = 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; }
+    let d = mandelbulb(p, params.power);
+    if (d < 0.001) { hit = 1.0 - f32(i)/64.0; break; }
     t += d;
-    if (t > 10.0) { break; }
   }
-  return 0.0;
-}
-
-@compute @workgroup_size(8,8)
-fn main(@builtin(global_invocation_id) gid : vec3<u32>) {
-  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 c = raymarch(ro, rd);
-  let idx = (gid.y * width + gid.x) * 4u;
 
-  pixels[idx + 0u] = c;
-  pixels[idx + 1u] = c * 0.8;
-  pixels[idx + 2u] = c * 1.2;
-  pixels[idx + 3u] = 1.0;
+  let idx = (id.y * w + id.x) * 4u;
+  pixels[idx]     = hit;
+  pixels[idx + 1] = hit * 0.8;
+  pixels[idx + 2] = hit * 1.2;
+  pixels[idx + 3] = 1.0;
 }