@zephyr3d/scene 0.9.26 → 0.10.0

package/dist/render/hzb.js

@@ -1,19 +1,35 @@
 import { drawFullscreenQuad } from './fullscreenquad.js';
-import { DEPTH_FARTHEST, DEPTH_REDUCE_FARTHER } from '@zephyr3d/base';
+import { REVERSE_Z, DEPTH_FARTHEST, DEPTH_REDUCE_FARTHER, DEPTH_REDUCE_CLOSER } from '@zephyr3d/base';
 import './primitive.js';
 import '@zephyr3d/device';
 import { getDevice } from '../app/api.js';
 import { CopyBlitter } from '../blitter/copy.js';
 import { fetchSampler } from '../utility/misc.js';
 
-let hzbProgram = null;
+let hzbProgram = [];
 let hzbBindGroupCache = new WeakMap();
-let blitter = null;
+let blitter = [];
 const srcSize = new Int32Array(2);
-class HiZInitBlitter extends CopyBlitter {
+/**
+ * Seeds mip 0 of the pyramid from the depth prepass.
+ *
+ * Both channels start from the same device depth: the pyramid only diverges
+ * once the reductions run, one keeping the farthest sample of each 2x2 and the
+ * other the nearest. `twoChannel` decides whether the nearest channel exists at
+ * all - it is a second full mip chain's worth of bandwidth, so a frame that only
+ * needs the occlusion pyramid does not carry it.
+ */ class HiZInitBlitter extends CopyBlitter {
+    _twoChannel;
+    constructor(twoChannel){
+        super();
+        this._twoChannel = twoChannel;
+    }
     /** @override */ filter(scope, type, srcTex, srcUV, srcLayer, sampleType) {
         const depth = super.filter(scope, type, srcTex, srcUV, srcLayer, sampleType);
-        return scope.$builder.vec4(depth.x, 0, 0, 1);
+        return scope.$builder.vec4(depth.x, this._twoChannel ? depth.x : 0, 0, 1);
+    }
+    /** @override */ calcHash() {
+        return `${super.calcHash()}:${this._twoChannel ? 1 : 0}`;
     }
 }
 /*
@@ -192,7 +208,26 @@ float3 hi_z_trace(float3 p, float3 v, in uint camera, out uint iterations) {
  }
  return ray;
 }
-*/ function buildHZBProgram(device) {
+*/ /**
+ * Texture format for the Hi-Z pyramid.
+ *
+ * Half float under reverse-Z and full float otherwise, because reverse-Z spends
+ * float's relative precision where the depth is: a 16-bit mantissa there is
+ * worth a few centimetres at a hundred metres, while a linear-Z half float
+ * would quantise the near field into steps.
+ *
+ * `withNearest` adds the nearest-depth channel. It doubles the pyramid's
+ * bandwidth, so it is only requested by a frame that has a consumer for it.
+ */ function getHiZFormat(withNearest) {
+    if (withNearest) {
+        return REVERSE_Z ? 'rg16f' : 'rg32f';
+    }
+    return REVERSE_Z ? 'r16f' : 'r32f';
+}
+/** Whether a Hi-Z texture of this format carries the nearest-depth channel. */ function hasNearestChannel(format) {
+    return format === 'rg16f' || format === 'rg32f';
+}
+function buildHZBProgram(device, twoChannel) {
     const program = device.buildRenderProgram({
         label: 'HZBBuilder',
         vertex (pb) {
@@ -221,22 +256,33 @@ float3 hi_z_trace(float3 p, float3 v, in uint camera, out uint iterations) {
                     this.$l[`d${i}`] = pb.textureLoad(this.srcTex, pb.clamp(pb.add(this.coord, pb.ivec2(i >> 1, i & 1)), this.minCoord, this.maxCoord), device.type === 'webgpu' ? 0 : this.srcMipLevel);
                 }
                 // Furthest-depth reduction (max under standard Z, min under reverse Z).
+                // This is the conservative direction for occlusion: a cell reports the
+                // depth nothing behind it can be in front of.
                 this.$l.maxDepth = pb[DEPTH_REDUCE_FARTHER](pb[DEPTH_REDUCE_FARTHER](this.d0.r, this.d1.r), pb[DEPTH_REDUCE_FARTHER](this.d2.r, this.d3.r));
-                this.$outputs.color = pb.vec4(this.maxDepth, 0, 0, 1);
+                if (twoChannel) {
+                    // Nearest-depth reduction, from the green channel. The opposite
+                    // conservative direction, and the one a proximity query needs: a cell
+                    // reports the closest surface anywhere inside it, so a query that
+                    // covers the cell cannot miss geometry that is in there.
+                    this.$l.minDepth = pb[DEPTH_REDUCE_CLOSER](pb[DEPTH_REDUCE_CLOSER](this.d0.g, this.d1.g), pb[DEPTH_REDUCE_CLOSER](this.d2.g, this.d3.g));
+                    this.$outputs.color = pb.vec4(this.maxDepth, this.minDepth, 0, 1);
+                } else {
+                    this.$outputs.color = pb.vec4(this.maxDepth, 0, 0, 1);
+                }
             });
         }
     });
-    program.name = '@HZB_Builder';
+    program.name = `@HZB_Builder${twoChannel ? '_MinMax' : ''}`;
     return program;
 }
-function getHiZBindGroup(tex, mip) {
+function getHiZBindGroup(tex, mip, program) {
     let info = hzbBindGroupCache.get(tex);
     if (!info) {
         info = [];
         hzbBindGroupCache.set(tex, info);
     }
     if (!info[mip]) {
-        info[mip] = tex.device.createBindGroup(hzbProgram.bindGroupLayouts[0]);
+        info[mip] = tex.device.createBindGroup(program.bindGroupLayouts[0]);
         srcSize[0] = Math.max(tex.width >> mip, 1);
         srcSize[1] = Math.max(tex.height >> mip, 1);
         info[mip].setValue('srcSize', srcSize);
@@ -249,13 +295,13 @@ function getHiZBindGroup(tex, mip) {
     }
     return info[mip];
 }
-function buildHiZLevel(device, miplevel, srcTexture, dstTexture) {
+function buildHiZLevel(device, miplevel, srcTexture, dstTexture, program) {
     const framebuffer = device.pool.fetchTemporalFramebuffer(false, 0, 0, [
         dstTexture
     ], null, false, 1, true, miplevel + 1);
     framebuffer.setColorAttachmentGenerateMipmaps(0, false);
-    device.setProgram(hzbProgram);
-    device.setBindGroup(0, getHiZBindGroup(srcTexture, miplevel));
+    device.setProgram(program);
+    device.setBindGroup(0, getHiZBindGroup(srcTexture, miplevel, program));
     device.setFramebuffer(framebuffer);
     drawFullscreenQuad();
     if (srcTexture !== dstTexture) {
@@ -263,29 +309,41 @@ function buildHiZLevel(device, miplevel, srcTexture, dstTexture) {
     }
     device.pool.releaseFrameBuffer(framebuffer);
 }
-function buildHiZ(sourceTex, HiZFrameBuffer) {
+/**
+ * Builds the Hi-Z pyramid from the depth prepass.
+ *
+ * The target's channel count picks the variant: a one-channel target carries
+ * only the farthest-depth pyramid the occlusion tests want, a two-channel one
+ * adds the nearest-depth pyramid in green for proximity queries. Keyed off the
+ * texture rather than a flag so the pyramid and its consumer cannot disagree
+ * about what is in green.
+ */ function buildHiZ(sourceTex, HiZFrameBuffer) {
     const device = getDevice();
-    if (!hzbProgram) {
-        hzbProgram = buildHZBProgram(device);
-        blitter = new HiZInitBlitter();
+    const dstTex = HiZFrameBuffer.getColorAttachments()[0];
+    const twoChannel = hasNearestChannel(dstTex.format);
+    const variant = twoChannel ? 1 : 0;
+    if (!hzbProgram[variant]) {
+        hzbProgram[variant] = buildHZBProgram(device, twoChannel);
+        blitter[variant] = new HiZInitBlitter(twoChannel);
     }
-    blitter.blit(sourceTex, HiZFrameBuffer, fetchSampler('clamp_nearest'));
+    const program = hzbProgram[variant];
+    blitter[variant].blit(sourceTex, HiZFrameBuffer, fetchSampler('clamp_nearest'));
     device.pushDeviceStates();
     const srcTex = HiZFrameBuffer.getColorAttachments()[0];
     if (device.type === 'webgpu') {
         for(let i = 0; i < srcTex.mipLevelCount - 1; i++){
-            buildHiZLevel(device, i, srcTex, srcTex);
+            buildHiZLevel(device, i, srcTex, srcTex, program);
         }
     } else {
         const tmpFramebuffer = device.pool.fetchTemporalFramebuffer(false, HiZFrameBuffer.getWidth(), HiZFrameBuffer.getHeight(), HiZFrameBuffer.getColorAttachments()[0].format, null, true);
-        const dstTex = tmpFramebuffer.getColorAttachments()[0];
+        const tmpTex = tmpFramebuffer.getColorAttachments()[0];
         for(let i = 0; i < srcTex.mipLevelCount - 1; i++){
-            buildHiZLevel(device, i, srcTex, dstTex);
+            buildHiZLevel(device, i, srcTex, tmpTex, program);
         }
         device.pool.releaseFrameBuffer(tmpFramebuffer);
     }
     device.popDeviceStates();
 }
 
-export { buildHiZ };
+export { buildHiZ, getHiZFormat, hasNearestChannel };
 //# sourceMappingURL=hzb.js.map

package/dist/render/hzb.js.map

@@ -1 +1 @@
-{"version":3,"file":"hzb.js","sources":["../../src/render/hzb.ts"],"sourcesContent":["import type {\r\n  AbstractDevice,\r\n  BaseTexture,\r\n  BindGroup,\r\n  FrameBuffer,\r\n  GPUProgram,\r\n  PBInsideFunctionScope,\r\n  PBShaderExp,\r\n  Texture2D\r\n} from '@zephyr3d/device';\r\nimport { drawFullscreenQuad } from './fullscreenquad';\r\nimport { CopyBlitter, type BlitType } from '../blitter';\r\nimport { fetchSampler } from '../utility/misc';\r\nimport { getDevice } from '../app/api';\r\nimport type { Nullable } from '@zephyr3d/base';\r\nimport { DEPTH_FARTHEST, DEPTH_REDUCE_FARTHER } from '@zephyr3d/base';\r\n\r\nlet hzbProgram: Nullable<GPUProgram> = null;\r\nlet hzbBindGroupCache: WeakMap<BaseTexture, BindGroup[]> = new WeakMap();\r\nlet blitter: Nullable<HiZInitBlitter> = null;\r\nconst srcSize = new Int32Array(2);\r\n\r\nclass HiZInitBlitter extends CopyBlitter {\r\n  /** @override */\r\n  filter(\r\n    scope: PBInsideFunctionScope,\r\n    type: BlitType,\r\n    srcTex: PBShaderExp,\r\n    srcUV: PBShaderExp,\r\n    srcLayer: PBShaderExp,\r\n    sampleType: 'float' | 'int' | 'uint'\r\n  ) {\r\n    const depth = super.filter(scope, type, srcTex, srcUV, srcLayer, sampleType);\r\n    return scope.$builder.vec4(depth.x, 0, 0, 1);\r\n  }\r\n}\r\n\r\n/*\r\nvec3 trace_ray(vec3 ray_start, vec3 ray_dir)\r\n{\r\n\r\n    if (ray_dir.z < 0.0) {\r\n        return vec3(0);\r\n    }\r\n\r\n    ray_dir = normalize(ray_dir);\r\n    ivec2 work_size = SCREEN_SIZE_INT;\r\n\r\n    const int loop_max = 150;\r\n    int mipmap = 0;\r\n    int max_iter = loop_max;\r\n\r\n    vec3 pos = ray_start;\r\n\r\n    // Move pos by a small bias\r\n    pos += ray_dir * 0.008;\r\n\r\n    float hit_bias = 0.0017;\r\n\r\n    while (mipmap > -1 && max_iter --> 0)\r\n    {\r\n\r\n        // Check if we are out of screen bounds, if so, return\r\n        if (pos.x < 0.0 || pos.y < 0.0 || pos.x > 1.0 || pos.y > 1.0 || pos.z < 0.0 || pos.z > 1.0)\r\n        {\r\n            return vec3(0,0,0);\r\n        }\r\n\r\n        // Fetch the current minimum cell plane height\r\n        float cell_z = textureLod(DownscaledDepth, pos.xy, mipmap).x;\r\n\r\n        // Compute the fractional part of the coordinate (scaled by the working size)\r\n        // so the values will be between 0.0 and 1.0\r\n        vec2 fract_coord = mod(pos.xy * work_size, 1.0);\r\n\r\n        // Modify fract coord based on which direction we are stepping in.\r\n        // Fract coord now contains the percentage how far we moved already in\r\n        // the current cell in each direction.\r\n        fract_coord.x = ray_dir.x > 0.0 ? fract_coord.x : 1.0 - fract_coord.x;\r\n        fract_coord.y = ray_dir.y > 0.0 ? fract_coord.y : 1.0 - fract_coord.y;\r\n\r\n        // Compute maximum k and minimum k for which the ray would still be\r\n        // inside of the cell.\r\n        vec2 max_k_v = (1.0 / abs(ray_dir.xy)) / work_size.xy;\r\n        vec2 min_k_v = -max_k_v * fract_coord.xy;\r\n\r\n        // Scale the maximum k by the percentage we already processed in the current cell,\r\n        // since e.g. if we already moved 50%, we can only move another 50%.\r\n        max_k_v *= 1.0 - fract_coord.xy;\r\n\r\n        // The maximum k is the minimum of the both sub-k's since if one component-maximum\r\n        // is reached, the ray is out of the cell\r\n        float max_k = min(max_k_v.x, max_k_v.y);\r\n\r\n        // Same applies to the min_k, but because min_k is negative we have to use max()\r\n        float min_k = max(min_k_v.x, min_k_v.y);\r\n\r\n        // Check if the ray intersects with the cell plane. We have the following\r\n        // equation:\r\n        // pos.z + k * ray_dir.z = cell.z\r\n        // So k is:\r\n        float k = (cell_z - pos.z) / ray_dir.z;\r\n\r\n        // Optional: Abort when ray didn't exactly intersect:\r\n        // if (k < min_k && mipmap <= 0) {\r\n        //     return vec3(0);\r\n        // }\r\n\r\n        // Check if we intersected the cell\r\n        if (k < max_k + hit_bias)\r\n        {\r\n            // Clamp k\r\n            k = max(min_k, k);\r\n\r\n            if (mipmap < 1) {\r\n                pos += k * ray_dir;\r\n                return pos;\r\n            }\r\n\r\n            // If we hit anything at a higher mipmap, step up to a higher detailed\r\n            // mipmap:\r\n            mipmap -= 2;\r\n            work_size *= 4;\r\n        } else {\r\n\r\n            // If we hit nothing, move to the next cell, with a small bias\r\n            pos += max_k * ray_dir * 1.04;\r\n        }\r\n\r\n        mipmap += 1;\r\n        work_size /= 2;\r\n    }\r\n\r\n    return vec3(0);\r\n}\r\n*/\r\n/*\r\nfloat2 cell(float2 ray, float2 cell_count, uint camera) {\r\n return floor(ray.xy * cell_count);\r\n}\r\n\r\nfloat2 cell_count(float level) {\r\n return input_texture2_size / (level == 0.0 ? 1.0 : exp2(level));\r\n}\r\n\r\nfloat3 intersect_cell_boundary(float3 pos, float3 dir, float2 cell_id, float2 cell_count, float2 cross_step, float2 cross_offset, uint camera) {\r\n float2 cell_size = 1.0 / cell_count;\r\n float2 planes = cell_id/cell_count + cell_size * cross_step;\r\n\r\n float2 solutions = (planes - pos)/dir.xy;\r\n float3 intersection_pos = pos + dir * min(solutions.x, solutions.y);\r\n\r\n intersection_pos.xy += (solutions.x < solutions.y) ? float2(cross_offset.x, 0.0) : float2(0.0, cross_offset.y);\r\n\r\n return intersection_pos;\r\n}\r\n\r\nbool crossed_cell_boundary(float2 cell_id_one, float2 cell_id_two) {\r\n return (int)cell_id_one.x != (int)cell_id_two.x || (int)cell_id_one.y != (int)cell_id_two.y;\r\n}\r\n\r\nfloat minimum_depth_plane(float2 ray, float level, float2 cell_count, uint camera) {\r\n return input_texture2.Load(int3(vr_stereo_to_mono(ray.xy, camera) * cell_count, level)).r;\r\n}\r\n\r\nfloat3 hi_z_trace(float3 p, float3 v, in uint camera, out uint iterations) {\r\n float level = HIZ_START_LEVEL;\r\n float3 v_z = v/v.z;\r\n float2 hi_z_size = cell_count(level);\r\n float3 ray = p;\r\n\r\n float2 cross_step = float2(v.x >= 0.0 ? 1.0 : -1.0, v.y >= 0.0 ? 1.0 : -1.0);\r\n float2 cross_offset = cross_step * 0.00001;\r\n cross_step = saturate(cross_step);\r\n\r\n float2 ray_cell = cell(ray.xy, hi_z_size.xy, camera);\r\n ray = intersect_cell_boundary(ray, v, ray_cell, hi_z_size, cross_step, cross_offset, camera);\r\n\r\n iterations = 0;\r\n while(level >= HIZ_STOP_LEVEL && iterations < MAX_ITERATIONS) {\r\n  // get the cell number of the current ray\r\n  float2 current_cell_count = cell_count(level);\r\n  float2 old_cell_id = cell(ray.xy, current_cell_count, camera);\r\n\r\n  // get the minimum depth plane in which the current ray resides\r\n  float min_z = minimum_depth_plane(ray.xy, level, current_cell_count, camera);\r\n\r\n  // intersect only if ray depth is below the minimum depth plane\r\n  float3 tmp_ray = ray;\r\n  if(v.z > 0) {\r\n   float min_minus_ray = min_z - ray.z;\r\n   tmp_ray = min_minus_ray > 0 ? ray + v_z*min_minus_ray : tmp_ray;\r\n   float2 new_cell_id = cell(tmp_ray.xy, current_cell_count, camera);\r\n   if(crossed_cell_boundary(old_cell_id, new_cell_id)) {\r\n    tmp_ray = intersect_cell_boundary(ray, v, old_cell_id, current_cell_count, cross_step, cross_offset, camera);\r\n    level = min(HIZ_MAX_LEVEL, level + 2.0f);\r\n   }else{\r\n    if(level == 1 && abs(min_minus_ray) > 0.0001) {\r\n     tmp_ray = intersect_cell_boundary(ray, v, old_cell_id, current_cell_count, cross_step, cross_offset, camera);\r\n     level = 2;\r\n    }\r\n   }\r\n  } else if(ray.z < min_z) {\r\n   tmp_ray = intersect_cell_boundary(ray, v, old_cell_id, current_cell_count, cross_step, cross_offset, camera);\r\n   level = min(HIZ_MAX_LEVEL, level + 2.0f);\r\n  }\r\n\r\n  ray.xyz = tmp_ray.xyz;\r\n  --level;\r\n\r\n  ++iterations;\r\n }\r\n return ray;\r\n}\r\n*/\r\n\r\nfunction buildHZBProgram(device: AbstractDevice) {\r\n  const program = device.buildRenderProgram({\r\n    label: 'HZBBuilder',\r\n    vertex(pb) {\r\n      this.$inputs.pos = pb.vec2().attrib('position');\r\n      this.$outputs.uv = pb.vec2();\r\n      pb.main(function () {\r\n        this.$builtins.position = pb.vec4(this.$inputs.pos, DEPTH_FARTHEST, 1);\r\n        this.$outputs.uv = pb.add(pb.mul(this.$inputs.pos.xy, 0.5), pb.vec2(0.5));\r\n        if (device.type === 'webgpu') {\r\n          this.$builtins.position.y = pb.neg(this.$builtins.position.y);\r\n        }\r\n      });\r\n    },\r\n    fragment(pb) {\r\n      this.$outputs.color = pb.vec4();\r\n      this.srcTex = pb.tex2D().sampleType('unfilterable-float').uniform(0);\r\n      this.srcSize = pb.ivec2().uniform(0);\r\n      if (device.type !== 'webgpu') {\r\n        this.srcMipLevel = pb.int().uniform(0);\r\n      }\r\n      pb.main(function () {\r\n        this.$l.coord = pb.mul(pb.ivec2(this.$builtins.fragCoord.xy), 2);\r\n        this.$l.minCoord = pb.ivec2(0, 0);\r\n        this.$l.maxCoord = pb.sub(this.srcSize, pb.ivec2(1, 1));\r\n        for (let i = 0; i < 4; i++) {\r\n          this.$l[`d${i}`] = pb.textureLoad(\r\n            this.srcTex,\r\n            pb.clamp(pb.add(this.coord, pb.ivec2(i >> 1, i & 1)), this.minCoord, this.maxCoord),\r\n            device.type === 'webgpu' ? 0 : this.srcMipLevel\r\n          );\r\n        }\r\n        // Furthest-depth reduction (max under standard Z, min under reverse Z).\r\n        this.$l.maxDepth = pb[DEPTH_REDUCE_FARTHER](\r\n          pb[DEPTH_REDUCE_FARTHER](this.d0.r, this.d1.r),\r\n          pb[DEPTH_REDUCE_FARTHER](this.d2.r, this.d3.r)\r\n        );\r\n        this.$outputs.color = pb.vec4(this.maxDepth, 0, 0, 1);\r\n      });\r\n    }\r\n  })!;\r\n  program.name = '@HZB_Builder';\r\n  return program;\r\n}\r\n\r\nfunction getHiZBindGroup(tex: BaseTexture, mip: number) {\r\n  let info = hzbBindGroupCache.get(tex);\r\n  if (!info) {\r\n    info = [];\r\n    hzbBindGroupCache.set(tex, info);\r\n  }\r\n  if (!info[mip]) {\r\n    info[mip] = tex.device.createBindGroup(hzbProgram!.bindGroupLayouts[0]);\r\n    srcSize[0] = Math.max(tex.width >> mip, 1);\r\n    srcSize[1] = Math.max(tex.height >> mip, 1);\r\n    info[mip].setValue('srcSize', srcSize);\r\n    if (tex.device.type === 'webgpu') {\r\n      info[mip].setTextureView('srcTex', tex, mip, 0, 1, fetchSampler('clamp_nearest'));\r\n    } else {\r\n      info[mip].setTexture('srcTex', tex, fetchSampler('clamp_nearest'));\r\n      info[mip].setValue('srcMipLevel', mip);\r\n    }\r\n  }\r\n  return info[mip];\r\n}\r\n\r\nfunction buildHiZLevel(\r\n  device: AbstractDevice,\r\n  miplevel: number,\r\n  srcTexture: Texture2D,\r\n  dstTexture: Texture2D\r\n) {\r\n  const framebuffer = device.pool.fetchTemporalFramebuffer(\r\n    false,\r\n    0,\r\n    0,\r\n    [dstTexture],\r\n    null,\r\n    false,\r\n    1,\r\n    true,\r\n    miplevel + 1\r\n  );\r\n  framebuffer.setColorAttachmentGenerateMipmaps(0, false);\r\n  device.setProgram(hzbProgram);\r\n  device.setBindGroup(0, getHiZBindGroup(srcTexture, miplevel));\r\n  device.setFramebuffer(framebuffer);\r\n  drawFullscreenQuad();\r\n  if (srcTexture !== dstTexture) {\r\n    device.copyFramebufferToTexture2D(framebuffer, 0, srcTexture, miplevel + 1);\r\n  }\r\n  device.pool.releaseFrameBuffer(framebuffer);\r\n}\r\n\r\nexport function buildHiZ(sourceTex: Texture2D, HiZFrameBuffer: FrameBuffer) {\r\n  const device = getDevice();\r\n  if (!hzbProgram) {\r\n    hzbProgram = buildHZBProgram(device);\r\n    blitter = new HiZInitBlitter();\r\n  }\r\n  blitter!.blit(sourceTex, HiZFrameBuffer, fetchSampler('clamp_nearest'));\r\n  device.pushDeviceStates();\r\n  const srcTex = HiZFrameBuffer.getColorAttachments()[0] as Texture2D;\r\n  if (device.type === 'webgpu') {\r\n    for (let i = 0; i < srcTex.mipLevelCount - 1; i++) {\r\n      buildHiZLevel(device, i, srcTex, srcTex);\r\n    }\r\n  } else {\r\n    const tmpFramebuffer = device.pool.fetchTemporalFramebuffer(\r\n      false,\r\n      HiZFrameBuffer.getWidth(),\r\n      HiZFrameBuffer.getHeight(),\r\n      HiZFrameBuffer.getColorAttachments()[0].format,\r\n      null,\r\n      true\r\n    );\r\n    const dstTex = tmpFramebuffer.getColorAttachments()[0] as Texture2D;\r\n    for (let i = 0; i < srcTex.mipLevelCount - 1; i++) {\r\n      buildHiZLevel(device, i, srcTex, dstTex);\r\n    }\r\n    device.pool.releaseFrameBuffer(tmpFramebuffer);\r\n  }\r\n  device.popDeviceStates();\r\n}\r\n"],"names":["hzbProgram","hzbBindGroupCache","WeakMap","blitter","srcSize","Int32Array","HiZInitBlitter","CopyBlitter","filter","scope","type","srcTex","srcUV","srcLayer","sampleType","depth","$builder","vec4","x","buildHZBProgram","device","program","buildRenderProgram","label","vertex","pb","$inputs","pos","vec2","attrib","$outputs","uv","main","$builtins","position","DEPTH_FARTHEST","add","mul","xy","y","neg","fragment","color","tex2D","uniform","ivec2","srcMipLevel","int","$l","coord","fragCoord","minCoord","maxCoord","sub","i","textureLoad","clamp","maxDepth","DEPTH_REDUCE_FARTHER","d0","r","d1","d2","d3","name","getHiZBindGroup","tex","mip","info","get","set","createBindGroup","bindGroupLayouts","Math","max","width","height","setValue","setTextureView","fetchSampler","setTexture","buildHiZLevel","miplevel","srcTexture","dstTexture","framebuffer","pool","fetchTemporalFramebuffer","setColorAttachmentGenerateMipmaps","setProgram","setBindGroup","setFramebuffer","drawFullscreenQuad","copyFramebufferToTexture2D","releaseFrameBuffer","buildHiZ","sourceTex","HiZFrameBuffer","getDevice","blit","pushDeviceStates","getColorAttachments","mipLevelCount","tmpFramebuffer","getWidth","getHeight","format","dstTex","popDeviceStates"],"mappings":";;;;;;;;AAiBA,IAAIA,UAAmC,GAAA,IAAA;AACvC,IAAIC,oBAAuD,IAAIC,OAAAA,EAAAA;AAC/D,IAAIC,OAAoC,GAAA,IAAA;AACxC,MAAMC,OAAAA,GAAU,IAAIC,UAAW,CAAA,CAAA,CAAA;AAE/B,MAAMC,cAAuBC,SAAAA,WAAAA,CAAAA;AAC3B,qBACAC,MAAAA,CACEC,KAA4B,EAC5BC,IAAc,EACdC,MAAmB,EACnBC,KAAkB,EAClBC,QAAqB,EACrBC,UAAoC,EACpC;QACA,MAAMC,KAAAA,GAAQ,KAAK,CAACP,MAAAA,CAAOC,OAAOC,IAAMC,EAAAA,MAAAA,EAAQC,OAAOC,QAAUC,EAAAA,UAAAA,CAAAA;QACjE,OAAOL,KAAAA,CAAMO,QAAQ,CAACC,IAAI,CAACF,KAAMG,CAAAA,CAAC,EAAE,CAAA,EAAG,CAAG,EAAA,CAAA,CAAA;AAC5C;AACF;AAEA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAkGA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA+EA,GAEA,SAASC,gBAAgBC,MAAsB,EAAA;IAC7C,MAAMC,OAAAA,GAAUD,MAAOE,CAAAA,kBAAkB,CAAC;QACxCC,KAAO,EAAA,YAAA;AACPC,QAAAA,MAAAA,CAAAA,CAAOC,EAAE,EAAA;YACP,IAAI,CAACC,OAAO,CAACC,GAAG,GAAGF,EAAGG,CAAAA,IAAI,EAAGC,CAAAA,MAAM,CAAC,UAAA,CAAA;AACpC,YAAA,IAAI,CAACC,QAAQ,CAACC,EAAE,GAAGN,GAAGG,IAAI,EAAA;AAC1BH,YAAAA,EAAAA,CAAGO,IAAI,CAAC,WAAA;AACN,gBAAA,IAAI,CAACC,SAAS,CAACC,QAAQ,GAAGT,EAAGR,CAAAA,IAAI,CAAC,IAAI,CAACS,OAAO,CAACC,GAAG,EAAEQ,cAAgB,EAAA,CAAA,CAAA;gBACpE,IAAI,CAACL,QAAQ,CAACC,EAAE,GAAGN,EAAGW,CAAAA,GAAG,CAACX,EAAAA,CAAGY,GAAG,CAAC,IAAI,CAACX,OAAO,CAACC,GAAG,CAACW,EAAE,EAAE,GAAA,CAAA,EAAMb,EAAGG,CAAAA,IAAI,CAAC,GAAA,CAAA,CAAA;gBACpE,IAAIR,MAAAA,CAAOV,IAAI,KAAK,QAAU,EAAA;AAC5B,oBAAA,IAAI,CAACuB,SAAS,CAACC,QAAQ,CAACK,CAAC,GAAGd,EAAAA,CAAGe,GAAG,CAAC,IAAI,CAACP,SAAS,CAACC,QAAQ,CAACK,CAAC,CAAA;AAC9D;AACF,aAAA,CAAA;AACF,SAAA;AACAE,QAAAA,QAAAA,CAAAA,CAAShB,EAAE,EAAA;AACT,YAAA,IAAI,CAACK,QAAQ,CAACY,KAAK,GAAGjB,GAAGR,IAAI,EAAA;YAC7B,IAAI,CAACN,MAAM,GAAGc,EAAGkB,CAAAA,KAAK,GAAG7B,UAAU,CAAC,oBAAsB8B,CAAAA,CAAAA,OAAO,CAAC,CAAA,CAAA;AAClE,YAAA,IAAI,CAACxC,OAAO,GAAGqB,GAAGoB,KAAK,EAAA,CAAGD,OAAO,CAAC,CAAA,CAAA;YAClC,IAAIxB,MAAAA,CAAOV,IAAI,KAAK,QAAU,EAAA;AAC5B,gBAAA,IAAI,CAACoC,WAAW,GAAGrB,GAAGsB,GAAG,EAAA,CAAGH,OAAO,CAAC,CAAA,CAAA;AACtC;AACAnB,YAAAA,EAAAA,CAAGO,IAAI,CAAC,WAAA;AACN,gBAAA,IAAI,CAACgB,EAAE,CAACC,KAAK,GAAGxB,EAAAA,CAAGY,GAAG,CAACZ,EAAAA,CAAGoB,KAAK,CAAC,IAAI,CAACZ,SAAS,CAACiB,SAAS,CAACZ,EAAE,CAAG,EAAA,CAAA,CAAA;gBAC9D,IAAI,CAACU,EAAE,CAACG,QAAQ,GAAG1B,EAAGoB,CAAAA,KAAK,CAAC,CAAG,EAAA,CAAA,CAAA;AAC/B,gBAAA,IAAI,CAACG,EAAE,CAACI,QAAQ,GAAG3B,EAAG4B,CAAAA,GAAG,CAAC,IAAI,CAACjD,OAAO,EAAEqB,EAAGoB,CAAAA,KAAK,CAAC,CAAG,EAAA,CAAA,CAAA,CAAA;AACpD,gBAAA,IAAK,IAAIS,CAAAA,GAAI,CAAGA,EAAAA,CAAAA,GAAI,GAAGA,CAAK,EAAA,CAAA;oBAC1B,IAAI,CAACN,EAAE,CAAC,CAAC,CAAC,EAAEM,CAAAA,CAAAA,CAAG,CAAC,GAAG7B,EAAG8B,CAAAA,WAAW,CAC/B,IAAI,CAAC5C,MAAM,EACXc,EAAAA,CAAG+B,KAAK,CAAC/B,EAAAA,CAAGW,GAAG,CAAC,IAAI,CAACa,KAAK,EAAExB,EAAAA,CAAGoB,KAAK,CAACS,CAAAA,IAAK,GAAGA,CAAI,GAAA,CAAA,CAAA,CAAA,EAAK,IAAI,CAACH,QAAQ,EAAE,IAAI,CAACC,QAAQ,GAClFhC,MAAOV,CAAAA,IAAI,KAAK,QAAW,GAAA,CAAA,GAAI,IAAI,CAACoC,WAAW,CAAA;AAEnD;;AAEA,gBAAA,IAAI,CAACE,EAAE,CAACS,QAAQ,GAAGhC,EAAE,CAACiC,oBAAAA,CAAqB,CACzCjC,EAAE,CAACiC,oBAAqB,CAAA,CAAC,IAAI,CAACC,EAAE,CAACC,CAAC,EAAE,IAAI,CAACC,EAAE,CAACD,CAAC,GAC7CnC,EAAE,CAACiC,qBAAqB,CAAC,IAAI,CAACI,EAAE,CAACF,CAAC,EAAE,IAAI,CAACG,EAAE,CAACH,CAAC,CAAA,CAAA;AAE/C,gBAAA,IAAI,CAAC9B,QAAQ,CAACY,KAAK,GAAGjB,EAAAA,CAAGR,IAAI,CAAC,IAAI,CAACwC,QAAQ,EAAE,GAAG,CAAG,EAAA,CAAA,CAAA;AACrD,aAAA,CAAA;AACF;AACF,KAAA,CAAA;AACApC,IAAAA,OAAAA,CAAQ2C,IAAI,GAAG,cAAA;IACf,OAAO3C,OAAAA;AACT;AAEA,SAAS4C,eAAAA,CAAgBC,GAAgB,EAAEC,GAAW,EAAA;IACpD,IAAIC,IAAAA,GAAOnE,iBAAkBoE,CAAAA,GAAG,CAACH,GAAAA,CAAAA;AACjC,IAAA,IAAI,CAACE,IAAM,EAAA;AACTA,QAAAA,IAAAA,GAAO,EAAE;QACTnE,iBAAkBqE,CAAAA,GAAG,CAACJ,GAAKE,EAAAA,IAAAA,CAAAA;AAC7B;AACA,IAAA,IAAI,CAACA,IAAI,CAACD,GAAAA,CAAI,EAAE;QACdC,IAAI,CAACD,GAAI,CAAA,GAAGD,GAAI9C,CAAAA,MAAM,CAACmD,eAAe,CAACvE,UAAAA,CAAYwE,gBAAgB,CAAC,CAAE,CAAA,CAAA;QACtEpE,OAAO,CAAC,EAAE,GAAGqE,IAAAA,CAAKC,GAAG,CAACR,GAAAA,CAAIS,KAAK,IAAIR,GAAK,EAAA,CAAA,CAAA;QACxC/D,OAAO,CAAC,EAAE,GAAGqE,IAAAA,CAAKC,GAAG,CAACR,GAAAA,CAAIU,MAAM,IAAIT,GAAK,EAAA,CAAA,CAAA;AACzCC,QAAAA,IAAI,CAACD,GAAAA,CAAI,CAACU,QAAQ,CAAC,SAAWzE,EAAAA,OAAAA,CAAAA;AAC9B,QAAA,IAAI8D,GAAI9C,CAAAA,MAAM,CAACV,IAAI,KAAK,QAAU,EAAA;YAChC0D,IAAI,CAACD,GAAI,CAAA,CAACW,cAAc,CAAC,UAAUZ,GAAKC,EAAAA,GAAAA,EAAK,CAAG,EAAA,CAAA,EAAGY,YAAa,CAAA,eAAA,CAAA,CAAA;SAC3D,MAAA;AACLX,YAAAA,IAAI,CAACD,GAAI,CAAA,CAACa,UAAU,CAAC,QAAA,EAAUd,KAAKa,YAAa,CAAA,eAAA,CAAA,CAAA;AACjDX,YAAAA,IAAI,CAACD,GAAAA,CAAI,CAACU,QAAQ,CAAC,aAAeV,EAAAA,GAAAA,CAAAA;AACpC;AACF;IACA,OAAOC,IAAI,CAACD,GAAI,CAAA;AAClB;AAEA,SAASc,cACP7D,MAAsB,EACtB8D,QAAgB,EAChBC,UAAqB,EACrBC,UAAqB,EAAA;IAErB,MAAMC,WAAAA,GAAcjE,OAAOkE,IAAI,CAACC,wBAAwB,CACtD,KAAA,EACA,GACA,CACA,EAAA;AAACH,QAAAA;AAAW,KAAA,EACZ,IACA,EAAA,KAAA,EACA,CACA,EAAA,IAAA,EACAF,QAAW,GAAA,CAAA,CAAA;IAEbG,WAAYG,CAAAA,iCAAiC,CAAC,CAAG,EAAA,KAAA,CAAA;AACjDpE,IAAAA,MAAAA,CAAOqE,UAAU,CAACzF,UAAAA,CAAAA;AAClBoB,IAAAA,MAAAA,CAAOsE,YAAY,CAAC,CAAGzB,EAAAA,eAAAA,CAAgBkB,UAAYD,EAAAA,QAAAA,CAAAA,CAAAA;AACnD9D,IAAAA,MAAAA,CAAOuE,cAAc,CAACN,WAAAA,CAAAA;AACtBO,IAAAA,kBAAAA,EAAAA;AACA,IAAA,IAAIT,eAAeC,UAAY,EAAA;AAC7BhE,QAAAA,MAAAA,CAAOyE,0BAA0B,CAACR,WAAa,EAAA,CAAA,EAAGF,YAAYD,QAAW,GAAA,CAAA,CAAA;AAC3E;IACA9D,MAAOkE,CAAAA,IAAI,CAACQ,kBAAkB,CAACT,WAAAA,CAAAA;AACjC;AAEO,SAASU,QAAAA,CAASC,SAAoB,EAAEC,cAA2B,EAAA;AACxE,IAAA,MAAM7E,MAAS8E,GAAAA,SAAAA,EAAAA;AACf,IAAA,IAAI,CAAClG,UAAY,EAAA;AACfA,QAAAA,UAAAA,GAAamB,eAAgBC,CAAAA,MAAAA,CAAAA;AAC7BjB,QAAAA,OAAAA,GAAU,IAAIG,cAAAA,EAAAA;AAChB;AACAH,IAAAA,OAAAA,CAASgG,IAAI,CAACH,SAAWC,EAAAA,cAAAA,EAAgBlB,YAAa,CAAA,eAAA,CAAA,CAAA;AACtD3D,IAAAA,MAAAA,CAAOgF,gBAAgB,EAAA;AACvB,IAAA,MAAMzF,MAASsF,GAAAA,cAAAA,CAAeI,mBAAmB,EAAE,CAAC,CAAE,CAAA;IACtD,IAAIjF,MAAAA,CAAOV,IAAI,KAAK,QAAU,EAAA;QAC5B,IAAK,IAAI4C,IAAI,CAAGA,EAAAA,CAAAA,GAAI3C,OAAO2F,aAAa,GAAG,GAAGhD,CAAK,EAAA,CAAA;YACjD2B,aAAc7D,CAAAA,MAAAA,EAAQkC,GAAG3C,MAAQA,EAAAA,MAAAA,CAAAA;AACnC;KACK,MAAA;QACL,MAAM4F,cAAAA,GAAiBnF,OAAOkE,IAAI,CAACC,wBAAwB,CACzD,KAAA,EACAU,eAAeO,QAAQ,EAAA,EACvBP,eAAeQ,SAAS,EAAA,EACxBR,eAAeI,mBAAmB,EAAE,CAAC,CAAE,CAAA,CAACK,MAAM,EAC9C,IACA,EAAA,IAAA,CAAA;AAEF,QAAA,MAAMC,MAASJ,GAAAA,cAAAA,CAAeF,mBAAmB,EAAE,CAAC,CAAE,CAAA;QACtD,IAAK,IAAI/C,IAAI,CAAGA,EAAAA,CAAAA,GAAI3C,OAAO2F,aAAa,GAAG,GAAGhD,CAAK,EAAA,CAAA;YACjD2B,aAAc7D,CAAAA,MAAAA,EAAQkC,GAAG3C,MAAQgG,EAAAA,MAAAA,CAAAA;AACnC;QACAvF,MAAOkE,CAAAA,IAAI,CAACQ,kBAAkB,CAACS,cAAAA,CAAAA;AACjC;AACAnF,IAAAA,MAAAA,CAAOwF,eAAe,EAAA;AACxB;;;;"}
+{"version":3,"file":"hzb.js","sources":["../../src/render/hzb.ts"],"sourcesContent":["import type {\r\n  AbstractDevice,\r\n  BaseTexture,\r\n  BindGroup,\r\n  FrameBuffer,\r\n  GPUProgram,\r\n  PBInsideFunctionScope,\r\n  PBShaderExp,\r\n  Texture2D,\r\n  TextureFormat\r\n} from '@zephyr3d/device';\r\nimport { drawFullscreenQuad } from './fullscreenquad';\r\nimport { CopyBlitter, type BlitType } from '../blitter';\r\nimport { fetchSampler } from '../utility/misc';\r\nimport { getDevice } from '../app/api';\r\nimport type { Nullable } from '@zephyr3d/base';\r\nimport { DEPTH_FARTHEST, DEPTH_REDUCE_CLOSER, DEPTH_REDUCE_FARTHER, REVERSE_Z } from '@zephyr3d/base';\r\n\r\nlet hzbProgram: Nullable<GPUProgram>[] = [];\r\nlet hzbBindGroupCache: WeakMap<BaseTexture, BindGroup[]> = new WeakMap();\r\nlet blitter: Nullable<HiZInitBlitter>[] = [];\r\nconst srcSize = new Int32Array(2);\r\n\r\n/**\r\n * Seeds mip 0 of the pyramid from the depth prepass.\r\n *\r\n * Both channels start from the same device depth: the pyramid only diverges\r\n * once the reductions run, one keeping the farthest sample of each 2x2 and the\r\n * other the nearest. `twoChannel` decides whether the nearest channel exists at\r\n * all - it is a second full mip chain's worth of bandwidth, so a frame that only\r\n * needs the occlusion pyramid does not carry it.\r\n */\r\nclass HiZInitBlitter extends CopyBlitter {\r\n  private readonly _twoChannel: boolean;\r\n  constructor(twoChannel: boolean) {\r\n    super();\r\n    this._twoChannel = twoChannel;\r\n  }\r\n  /** @override */\r\n  filter(\r\n    scope: PBInsideFunctionScope,\r\n    type: BlitType,\r\n    srcTex: PBShaderExp,\r\n    srcUV: PBShaderExp,\r\n    srcLayer: PBShaderExp,\r\n    sampleType: 'float' | 'int' | 'uint'\r\n  ) {\r\n    const depth = super.filter(scope, type, srcTex, srcUV, srcLayer, sampleType);\r\n    return scope.$builder.vec4(depth.x, this._twoChannel ? depth.x : 0, 0, 1);\r\n  }\r\n  /** @override */\r\n  protected calcHash(): string {\r\n    return `${super.calcHash()}:${this._twoChannel ? 1 : 0}`;\r\n  }\r\n}\r\n\r\n/*\r\nvec3 trace_ray(vec3 ray_start, vec3 ray_dir)\r\n{\r\n\r\n    if (ray_dir.z < 0.0) {\r\n        return vec3(0);\r\n    }\r\n\r\n    ray_dir = normalize(ray_dir);\r\n    ivec2 work_size = SCREEN_SIZE_INT;\r\n\r\n    const int loop_max = 150;\r\n    int mipmap = 0;\r\n    int max_iter = loop_max;\r\n\r\n    vec3 pos = ray_start;\r\n\r\n    // Move pos by a small bias\r\n    pos += ray_dir * 0.008;\r\n\r\n    float hit_bias = 0.0017;\r\n\r\n    while (mipmap > -1 && max_iter --> 0)\r\n    {\r\n\r\n        // Check if we are out of screen bounds, if so, return\r\n        if (pos.x < 0.0 || pos.y < 0.0 || pos.x > 1.0 || pos.y > 1.0 || pos.z < 0.0 || pos.z > 1.0)\r\n        {\r\n            return vec3(0,0,0);\r\n        }\r\n\r\n        // Fetch the current minimum cell plane height\r\n        float cell_z = textureLod(DownscaledDepth, pos.xy, mipmap).x;\r\n\r\n        // Compute the fractional part of the coordinate (scaled by the working size)\r\n        // so the values will be between 0.0 and 1.0\r\n        vec2 fract_coord = mod(pos.xy * work_size, 1.0);\r\n\r\n        // Modify fract coord based on which direction we are stepping in.\r\n        // Fract coord now contains the percentage how far we moved already in\r\n        // the current cell in each direction.\r\n        fract_coord.x = ray_dir.x > 0.0 ? fract_coord.x : 1.0 - fract_coord.x;\r\n        fract_coord.y = ray_dir.y > 0.0 ? fract_coord.y : 1.0 - fract_coord.y;\r\n\r\n        // Compute maximum k and minimum k for which the ray would still be\r\n        // inside of the cell.\r\n        vec2 max_k_v = (1.0 / abs(ray_dir.xy)) / work_size.xy;\r\n        vec2 min_k_v = -max_k_v * fract_coord.xy;\r\n\r\n        // Scale the maximum k by the percentage we already processed in the current cell,\r\n        // since e.g. if we already moved 50%, we can only move another 50%.\r\n        max_k_v *= 1.0 - fract_coord.xy;\r\n\r\n        // The maximum k is the minimum of the both sub-k's since if one component-maximum\r\n        // is reached, the ray is out of the cell\r\n        float max_k = min(max_k_v.x, max_k_v.y);\r\n\r\n        // Same applies to the min_k, but because min_k is negative we have to use max()\r\n        float min_k = max(min_k_v.x, min_k_v.y);\r\n\r\n        // Check if the ray intersects with the cell plane. We have the following\r\n        // equation:\r\n        // pos.z + k * ray_dir.z = cell.z\r\n        // So k is:\r\n        float k = (cell_z - pos.z) / ray_dir.z;\r\n\r\n        // Optional: Abort when ray didn't exactly intersect:\r\n        // if (k < min_k && mipmap <= 0) {\r\n        //     return vec3(0);\r\n        // }\r\n\r\n        // Check if we intersected the cell\r\n        if (k < max_k + hit_bias)\r\n        {\r\n            // Clamp k\r\n            k = max(min_k, k);\r\n\r\n            if (mipmap < 1) {\r\n                pos += k * ray_dir;\r\n                return pos;\r\n            }\r\n\r\n            // If we hit anything at a higher mipmap, step up to a higher detailed\r\n            // mipmap:\r\n            mipmap -= 2;\r\n            work_size *= 4;\r\n        } else {\r\n\r\n            // If we hit nothing, move to the next cell, with a small bias\r\n            pos += max_k * ray_dir * 1.04;\r\n        }\r\n\r\n        mipmap += 1;\r\n        work_size /= 2;\r\n    }\r\n\r\n    return vec3(0);\r\n}\r\n*/\r\n/*\r\nfloat2 cell(float2 ray, float2 cell_count, uint camera) {\r\n return floor(ray.xy * cell_count);\r\n}\r\n\r\nfloat2 cell_count(float level) {\r\n return input_texture2_size / (level == 0.0 ? 1.0 : exp2(level));\r\n}\r\n\r\nfloat3 intersect_cell_boundary(float3 pos, float3 dir, float2 cell_id, float2 cell_count, float2 cross_step, float2 cross_offset, uint camera) {\r\n float2 cell_size = 1.0 / cell_count;\r\n float2 planes = cell_id/cell_count + cell_size * cross_step;\r\n\r\n float2 solutions = (planes - pos)/dir.xy;\r\n float3 intersection_pos = pos + dir * min(solutions.x, solutions.y);\r\n\r\n intersection_pos.xy += (solutions.x < solutions.y) ? float2(cross_offset.x, 0.0) : float2(0.0, cross_offset.y);\r\n\r\n return intersection_pos;\r\n}\r\n\r\nbool crossed_cell_boundary(float2 cell_id_one, float2 cell_id_two) {\r\n return (int)cell_id_one.x != (int)cell_id_two.x || (int)cell_id_one.y != (int)cell_id_two.y;\r\n}\r\n\r\nfloat minimum_depth_plane(float2 ray, float level, float2 cell_count, uint camera) {\r\n return input_texture2.Load(int3(vr_stereo_to_mono(ray.xy, camera) * cell_count, level)).r;\r\n}\r\n\r\nfloat3 hi_z_trace(float3 p, float3 v, in uint camera, out uint iterations) {\r\n float level = HIZ_START_LEVEL;\r\n float3 v_z = v/v.z;\r\n float2 hi_z_size = cell_count(level);\r\n float3 ray = p;\r\n\r\n float2 cross_step = float2(v.x >= 0.0 ? 1.0 : -1.0, v.y >= 0.0 ? 1.0 : -1.0);\r\n float2 cross_offset = cross_step * 0.00001;\r\n cross_step = saturate(cross_step);\r\n\r\n float2 ray_cell = cell(ray.xy, hi_z_size.xy, camera);\r\n ray = intersect_cell_boundary(ray, v, ray_cell, hi_z_size, cross_step, cross_offset, camera);\r\n\r\n iterations = 0;\r\n while(level >= HIZ_STOP_LEVEL && iterations < MAX_ITERATIONS) {\r\n  // get the cell number of the current ray\r\n  float2 current_cell_count = cell_count(level);\r\n  float2 old_cell_id = cell(ray.xy, current_cell_count, camera);\r\n\r\n  // get the minimum depth plane in which the current ray resides\r\n  float min_z = minimum_depth_plane(ray.xy, level, current_cell_count, camera);\r\n\r\n  // intersect only if ray depth is below the minimum depth plane\r\n  float3 tmp_ray = ray;\r\n  if(v.z > 0) {\r\n   float min_minus_ray = min_z - ray.z;\r\n   tmp_ray = min_minus_ray > 0 ? ray + v_z*min_minus_ray : tmp_ray;\r\n   float2 new_cell_id = cell(tmp_ray.xy, current_cell_count, camera);\r\n   if(crossed_cell_boundary(old_cell_id, new_cell_id)) {\r\n    tmp_ray = intersect_cell_boundary(ray, v, old_cell_id, current_cell_count, cross_step, cross_offset, camera);\r\n    level = min(HIZ_MAX_LEVEL, level + 2.0f);\r\n   }else{\r\n    if(level == 1 && abs(min_minus_ray) > 0.0001) {\r\n     tmp_ray = intersect_cell_boundary(ray, v, old_cell_id, current_cell_count, cross_step, cross_offset, camera);\r\n     level = 2;\r\n    }\r\n   }\r\n  } else if(ray.z < min_z) {\r\n   tmp_ray = intersect_cell_boundary(ray, v, old_cell_id, current_cell_count, cross_step, cross_offset, camera);\r\n   level = min(HIZ_MAX_LEVEL, level + 2.0f);\r\n  }\r\n\r\n  ray.xyz = tmp_ray.xyz;\r\n  --level;\r\n\r\n  ++iterations;\r\n }\r\n return ray;\r\n}\r\n*/\r\n\r\n/**\r\n * Texture format for the Hi-Z pyramid.\r\n *\r\n * Half float under reverse-Z and full float otherwise, because reverse-Z spends\r\n * float's relative precision where the depth is: a 16-bit mantissa there is\r\n * worth a few centimetres at a hundred metres, while a linear-Z half float\r\n * would quantise the near field into steps.\r\n *\r\n * `withNearest` adds the nearest-depth channel. It doubles the pyramid's\r\n * bandwidth, so it is only requested by a frame that has a consumer for it.\r\n */\r\nexport function getHiZFormat(withNearest: boolean): TextureFormat {\r\n  if (withNearest) {\r\n    return REVERSE_Z ? 'rg16f' : 'rg32f';\r\n  }\r\n  return REVERSE_Z ? 'r16f' : 'r32f';\r\n}\r\n\r\n/** Whether a Hi-Z texture of this format carries the nearest-depth channel. */\r\nexport function hasNearestChannel(format: TextureFormat): boolean {\r\n  return format === 'rg16f' || format === 'rg32f';\r\n}\r\n\r\nfunction buildHZBProgram(device: AbstractDevice, twoChannel: boolean) {\r\n  const program = device.buildRenderProgram({\r\n    label: 'HZBBuilder',\r\n    vertex(pb) {\r\n      this.$inputs.pos = pb.vec2().attrib('position');\r\n      this.$outputs.uv = pb.vec2();\r\n      pb.main(function () {\r\n        this.$builtins.position = pb.vec4(this.$inputs.pos, DEPTH_FARTHEST, 1);\r\n        this.$outputs.uv = pb.add(pb.mul(this.$inputs.pos.xy, 0.5), pb.vec2(0.5));\r\n        if (device.type === 'webgpu') {\r\n          this.$builtins.position.y = pb.neg(this.$builtins.position.y);\r\n        }\r\n      });\r\n    },\r\n    fragment(pb) {\r\n      this.$outputs.color = pb.vec4();\r\n      this.srcTex = pb.tex2D().sampleType('unfilterable-float').uniform(0);\r\n      this.srcSize = pb.ivec2().uniform(0);\r\n      if (device.type !== 'webgpu') {\r\n        this.srcMipLevel = pb.int().uniform(0);\r\n      }\r\n      pb.main(function () {\r\n        this.$l.coord = pb.mul(pb.ivec2(this.$builtins.fragCoord.xy), 2);\r\n        this.$l.minCoord = pb.ivec2(0, 0);\r\n        this.$l.maxCoord = pb.sub(this.srcSize, pb.ivec2(1, 1));\r\n        for (let i = 0; i < 4; i++) {\r\n          this.$l[`d${i}`] = pb.textureLoad(\r\n            this.srcTex,\r\n            pb.clamp(pb.add(this.coord, pb.ivec2(i >> 1, i & 1)), this.minCoord, this.maxCoord),\r\n            device.type === 'webgpu' ? 0 : this.srcMipLevel\r\n          );\r\n        }\r\n        // Furthest-depth reduction (max under standard Z, min under reverse Z).\r\n        // This is the conservative direction for occlusion: a cell reports the\r\n        // depth nothing behind it can be in front of.\r\n        this.$l.maxDepth = pb[DEPTH_REDUCE_FARTHER](\r\n          pb[DEPTH_REDUCE_FARTHER](this.d0.r, this.d1.r),\r\n          pb[DEPTH_REDUCE_FARTHER](this.d2.r, this.d3.r)\r\n        );\r\n        if (twoChannel) {\r\n          // Nearest-depth reduction, from the green channel. The opposite\r\n          // conservative direction, and the one a proximity query needs: a cell\r\n          // reports the closest surface anywhere inside it, so a query that\r\n          // covers the cell cannot miss geometry that is in there.\r\n          this.$l.minDepth = pb[DEPTH_REDUCE_CLOSER](\r\n            pb[DEPTH_REDUCE_CLOSER](this.d0.g, this.d1.g),\r\n            pb[DEPTH_REDUCE_CLOSER](this.d2.g, this.d3.g)\r\n          );\r\n          this.$outputs.color = pb.vec4(this.maxDepth, this.minDepth, 0, 1);\r\n        } else {\r\n          this.$outputs.color = pb.vec4(this.maxDepth, 0, 0, 1);\r\n        }\r\n      });\r\n    }\r\n  })!;\r\n  program.name = `@HZB_Builder${twoChannel ? '_MinMax' : ''}`;\r\n  return program;\r\n}\r\n\r\nfunction getHiZBindGroup(tex: BaseTexture, mip: number, program: GPUProgram) {\r\n  let info = hzbBindGroupCache.get(tex);\r\n  if (!info) {\r\n    info = [];\r\n    hzbBindGroupCache.set(tex, info);\r\n  }\r\n  if (!info[mip]) {\r\n    info[mip] = tex.device.createBindGroup(program.bindGroupLayouts[0]);\r\n    srcSize[0] = Math.max(tex.width >> mip, 1);\r\n    srcSize[1] = Math.max(tex.height >> mip, 1);\r\n    info[mip].setValue('srcSize', srcSize);\r\n    if (tex.device.type === 'webgpu') {\r\n      info[mip].setTextureView('srcTex', tex, mip, 0, 1, fetchSampler('clamp_nearest'));\r\n    } else {\r\n      info[mip].setTexture('srcTex', tex, fetchSampler('clamp_nearest'));\r\n      info[mip].setValue('srcMipLevel', mip);\r\n    }\r\n  }\r\n  return info[mip];\r\n}\r\n\r\nfunction buildHiZLevel(\r\n  device: AbstractDevice,\r\n  miplevel: number,\r\n  srcTexture: Texture2D,\r\n  dstTexture: Texture2D,\r\n  program: GPUProgram\r\n) {\r\n  const framebuffer = device.pool.fetchTemporalFramebuffer(\r\n    false,\r\n    0,\r\n    0,\r\n    [dstTexture],\r\n    null,\r\n    false,\r\n    1,\r\n    true,\r\n    miplevel + 1\r\n  );\r\n  framebuffer.setColorAttachmentGenerateMipmaps(0, false);\r\n  device.setProgram(program);\r\n  device.setBindGroup(0, getHiZBindGroup(srcTexture, miplevel, program));\r\n  device.setFramebuffer(framebuffer);\r\n  drawFullscreenQuad();\r\n  if (srcTexture !== dstTexture) {\r\n    device.copyFramebufferToTexture2D(framebuffer, 0, srcTexture, miplevel + 1);\r\n  }\r\n  device.pool.releaseFrameBuffer(framebuffer);\r\n}\r\n\r\n/**\r\n * Builds the Hi-Z pyramid from the depth prepass.\r\n *\r\n * The target's channel count picks the variant: a one-channel target carries\r\n * only the farthest-depth pyramid the occlusion tests want, a two-channel one\r\n * adds the nearest-depth pyramid in green for proximity queries. Keyed off the\r\n * texture rather than a flag so the pyramid and its consumer cannot disagree\r\n * about what is in green.\r\n */\r\nexport function buildHiZ(sourceTex: Texture2D, HiZFrameBuffer: FrameBuffer) {\r\n  const device = getDevice();\r\n  const dstTex = HiZFrameBuffer.getColorAttachments()[0] as Texture2D;\r\n  const twoChannel = hasNearestChannel(dstTex.format);\r\n  const variant = twoChannel ? 1 : 0;\r\n  if (!hzbProgram[variant]) {\r\n    hzbProgram[variant] = buildHZBProgram(device, twoChannel);\r\n    blitter[variant] = new HiZInitBlitter(twoChannel);\r\n  }\r\n  const program = hzbProgram[variant]!;\r\n  blitter[variant]!.blit(sourceTex, HiZFrameBuffer, fetchSampler('clamp_nearest'));\r\n  device.pushDeviceStates();\r\n  const srcTex = HiZFrameBuffer.getColorAttachments()[0] as Texture2D;\r\n  if (device.type === 'webgpu') {\r\n    for (let i = 0; i < srcTex.mipLevelCount - 1; i++) {\r\n      buildHiZLevel(device, i, srcTex, srcTex, program);\r\n    }\r\n  } else {\r\n    const tmpFramebuffer = device.pool.fetchTemporalFramebuffer(\r\n      false,\r\n      HiZFrameBuffer.getWidth(),\r\n      HiZFrameBuffer.getHeight(),\r\n      HiZFrameBuffer.getColorAttachments()[0].format,\r\n      null,\r\n      true\r\n    );\r\n    const tmpTex = tmpFramebuffer.getColorAttachments()[0] as Texture2D;\r\n    for (let i = 0; i < srcTex.mipLevelCount - 1; i++) {\r\n      buildHiZLevel(device, i, srcTex, tmpTex, program);\r\n    }\r\n    device.pool.releaseFrameBuffer(tmpFramebuffer);\r\n  }\r\n  device.popDeviceStates();\r\n}\r\n"],"names":["hzbProgram","hzbBindGroupCache","WeakMap","blitter","srcSize","Int32Array","HiZInitBlitter","CopyBlitter","_twoChannel","twoChannel","filter","scope","type","srcTex","srcUV","srcLayer","sampleType","depth","$builder","vec4","x","calcHash","getHiZFormat","withNearest","REVERSE_Z","hasNearestChannel","format","buildHZBProgram","device","program","buildRenderProgram","label","vertex","pb","$inputs","pos","vec2","attrib","$outputs","uv","main","$builtins","position","DEPTH_FARTHEST","add","mul","xy","y","neg","fragment","color","tex2D","uniform","ivec2","srcMipLevel","int","$l","coord","fragCoord","minCoord","maxCoord","sub","i","textureLoad","clamp","maxDepth","DEPTH_REDUCE_FARTHER","d0","r","d1","d2","d3","minDepth","DEPTH_REDUCE_CLOSER","g","name","getHiZBindGroup","tex","mip","info","get","set","createBindGroup","bindGroupLayouts","Math","max","width","height","setValue","setTextureView","fetchSampler","setTexture","buildHiZLevel","miplevel","srcTexture","dstTexture","framebuffer","pool","fetchTemporalFramebuffer","setColorAttachmentGenerateMipmaps","setProgram","setBindGroup","setFramebuffer","drawFullscreenQuad","copyFramebufferToTexture2D","releaseFrameBuffer","buildHiZ","sourceTex","HiZFrameBuffer","getDevice","dstTex","getColorAttachments","variant","blit","pushDeviceStates","mipLevelCount","tmpFramebuffer","getWidth","getHeight","tmpTex","popDeviceStates"],"mappings":";;;;;;;;AAkBA,IAAIA,aAAqC,EAAE;AAC3C,IAAIC,oBAAuD,IAAIC,OAAAA,EAAAA;AAC/D,IAAIC,UAAsC,EAAE;AAC5C,MAAMC,OAAAA,GAAU,IAAIC,UAAW,CAAA,CAAA,CAAA;AAE/B;;;;;;;;AAQC,IACD,MAAMC,cAAuBC,SAAAA,WAAAA,CAAAA;IACVC,WAAqB;AACtC,IAAA,WAAA,CAAYC,UAAmB,CAAE;QAC/B,KAAK,EAAA;QACL,IAAI,CAACD,WAAW,GAAGC,UAAAA;AACrB;AACA,qBACAC,MAAAA,CACEC,KAA4B,EAC5BC,IAAc,EACdC,MAAmB,EACnBC,KAAkB,EAClBC,QAAqB,EACrBC,UAAoC,EACpC;QACA,MAAMC,KAAAA,GAAQ,KAAK,CAACP,MAAAA,CAAOC,OAAOC,IAAMC,EAAAA,MAAAA,EAAQC,OAAOC,QAAUC,EAAAA,UAAAA,CAAAA;AACjE,QAAA,OAAOL,MAAMO,QAAQ,CAACC,IAAI,CAACF,MAAMG,CAAC,EAAE,IAAI,CAACZ,WAAW,GAAGS,KAAAA,CAAMG,CAAC,GAAG,GAAG,CAAG,EAAA,CAAA,CAAA;AACzE;qBAEA,QAA6B,GAAA;QAC3B,OAAO,CAAA,EAAG,KAAK,CAACC,QAAW,EAAA,CAAA,CAAC,EAAE,IAAI,CAACb,WAAW,GAAG,CAAA,GAAI,CAAG,CAAA,CAAA;AAC1D;AACF;AAEA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAkGA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA+EA;;;;;;;;;;IAaO,SAASc,YAAAA,CAAaC,WAAoB,EAAA;AAC/C,IAAA,IAAIA,WAAa,EAAA;AACf,QAAA,OAAOC,YAAY,OAAU,GAAA,OAAA;AAC/B;AACA,IAAA,OAAOA,YAAY,MAAS,GAAA,MAAA;AAC9B;AAEA,gFACO,SAASC,iBAAAA,CAAkBC,MAAqB,EAAA;IACrD,OAAOA,MAAAA,KAAW,WAAWA,MAAW,KAAA,OAAA;AAC1C;AAEA,SAASC,eAAAA,CAAgBC,MAAsB,EAAEnB,UAAmB,EAAA;IAClE,MAAMoB,OAAAA,GAAUD,MAAOE,CAAAA,kBAAkB,CAAC;QACxCC,KAAO,EAAA,YAAA;AACPC,QAAAA,MAAAA,CAAAA,CAAOC,EAAE,EAAA;YACP,IAAI,CAACC,OAAO,CAACC,GAAG,GAAGF,EAAGG,CAAAA,IAAI,EAAGC,CAAAA,MAAM,CAAC,UAAA,CAAA;AACpC,YAAA,IAAI,CAACC,QAAQ,CAACC,EAAE,GAAGN,GAAGG,IAAI,EAAA;AAC1BH,YAAAA,EAAAA,CAAGO,IAAI,CAAC,WAAA;AACN,gBAAA,IAAI,CAACC,SAAS,CAACC,QAAQ,GAAGT,EAAGd,CAAAA,IAAI,CAAC,IAAI,CAACe,OAAO,CAACC,GAAG,EAAEQ,cAAgB,EAAA,CAAA,CAAA;gBACpE,IAAI,CAACL,QAAQ,CAACC,EAAE,GAAGN,EAAGW,CAAAA,GAAG,CAACX,EAAAA,CAAGY,GAAG,CAAC,IAAI,CAACX,OAAO,CAACC,GAAG,CAACW,EAAE,EAAE,GAAA,CAAA,EAAMb,EAAGG,CAAAA,IAAI,CAAC,GAAA,CAAA,CAAA;gBACpE,IAAIR,MAAAA,CAAOhB,IAAI,KAAK,QAAU,EAAA;AAC5B,oBAAA,IAAI,CAAC6B,SAAS,CAACC,QAAQ,CAACK,CAAC,GAAGd,EAAAA,CAAGe,GAAG,CAAC,IAAI,CAACP,SAAS,CAACC,QAAQ,CAACK,CAAC,CAAA;AAC9D;AACF,aAAA,CAAA;AACF,SAAA;AACAE,QAAAA,QAAAA,CAAAA,CAAShB,EAAE,EAAA;AACT,YAAA,IAAI,CAACK,QAAQ,CAACY,KAAK,GAAGjB,GAAGd,IAAI,EAAA;YAC7B,IAAI,CAACN,MAAM,GAAGoB,EAAGkB,CAAAA,KAAK,GAAGnC,UAAU,CAAC,oBAAsBoC,CAAAA,CAAAA,OAAO,CAAC,CAAA,CAAA;AAClE,YAAA,IAAI,CAAChD,OAAO,GAAG6B,GAAGoB,KAAK,EAAA,CAAGD,OAAO,CAAC,CAAA,CAAA;YAClC,IAAIxB,MAAAA,CAAOhB,IAAI,KAAK,QAAU,EAAA;AAC5B,gBAAA,IAAI,CAAC0C,WAAW,GAAGrB,GAAGsB,GAAG,EAAA,CAAGH,OAAO,CAAC,CAAA,CAAA;AACtC;AACAnB,YAAAA,EAAAA,CAAGO,IAAI,CAAC,WAAA;AACN,gBAAA,IAAI,CAACgB,EAAE,CAACC,KAAK,GAAGxB,EAAAA,CAAGY,GAAG,CAACZ,EAAAA,CAAGoB,KAAK,CAAC,IAAI,CAACZ,SAAS,CAACiB,SAAS,CAACZ,EAAE,CAAG,EAAA,CAAA,CAAA;gBAC9D,IAAI,CAACU,EAAE,CAACG,QAAQ,GAAG1B,EAAGoB,CAAAA,KAAK,CAAC,CAAG,EAAA,CAAA,CAAA;AAC/B,gBAAA,IAAI,CAACG,EAAE,CAACI,QAAQ,GAAG3B,EAAG4B,CAAAA,GAAG,CAAC,IAAI,CAACzD,OAAO,EAAE6B,EAAGoB,CAAAA,KAAK,CAAC,CAAG,EAAA,CAAA,CAAA,CAAA;AACpD,gBAAA,IAAK,IAAIS,CAAAA,GAAI,CAAGA,EAAAA,CAAAA,GAAI,GAAGA,CAAK,EAAA,CAAA;oBAC1B,IAAI,CAACN,EAAE,CAAC,CAAC,CAAC,EAAEM,CAAAA,CAAAA,CAAG,CAAC,GAAG7B,EAAG8B,CAAAA,WAAW,CAC/B,IAAI,CAAClD,MAAM,EACXoB,EAAAA,CAAG+B,KAAK,CAAC/B,EAAAA,CAAGW,GAAG,CAAC,IAAI,CAACa,KAAK,EAAExB,EAAAA,CAAGoB,KAAK,CAACS,CAAAA,IAAK,GAAGA,CAAI,GAAA,CAAA,CAAA,CAAA,EAAK,IAAI,CAACH,QAAQ,EAAE,IAAI,CAACC,QAAQ,GAClFhC,MAAOhB,CAAAA,IAAI,KAAK,QAAW,GAAA,CAAA,GAAI,IAAI,CAAC0C,WAAW,CAAA;AAEnD;;;;AAIA,gBAAA,IAAI,CAACE,EAAE,CAACS,QAAQ,GAAGhC,EAAE,CAACiC,oBAAAA,CAAqB,CACzCjC,EAAE,CAACiC,oBAAqB,CAAA,CAAC,IAAI,CAACC,EAAE,CAACC,CAAC,EAAE,IAAI,CAACC,EAAE,CAACD,CAAC,GAC7CnC,EAAE,CAACiC,qBAAqB,CAAC,IAAI,CAACI,EAAE,CAACF,CAAC,EAAE,IAAI,CAACG,EAAE,CAACH,CAAC,CAAA,CAAA;AAE/C,gBAAA,IAAI3D,UAAY,EAAA;;;;;AAKd,oBAAA,IAAI,CAAC+C,EAAE,CAACgB,QAAQ,GAAGvC,EAAE,CAACwC,mBAAAA,CAAoB,CACxCxC,EAAE,CAACwC,mBAAoB,CAAA,CAAC,IAAI,CAACN,EAAE,CAACO,CAAC,EAAE,IAAI,CAACL,EAAE,CAACK,CAAC,GAC5CzC,EAAE,CAACwC,oBAAoB,CAAC,IAAI,CAACH,EAAE,CAACI,CAAC,EAAE,IAAI,CAACH,EAAE,CAACG,CAAC,CAAA,CAAA;AAE9C,oBAAA,IAAI,CAACpC,QAAQ,CAACY,KAAK,GAAGjB,GAAGd,IAAI,CAAC,IAAI,CAAC8C,QAAQ,EAAE,IAAI,CAACO,QAAQ,EAAE,CAAG,EAAA,CAAA,CAAA;iBAC1D,MAAA;AACL,oBAAA,IAAI,CAAClC,QAAQ,CAACY,KAAK,GAAGjB,EAAAA,CAAGd,IAAI,CAAC,IAAI,CAAC8C,QAAQ,EAAE,GAAG,CAAG,EAAA,CAAA,CAAA;AACrD;AACF,aAAA,CAAA;AACF;AACF,KAAA,CAAA;AACApC,IAAAA,OAAAA,CAAQ8C,IAAI,GAAG,CAAC,YAAY,EAAElE,UAAAA,GAAa,YAAY,EAAI,CAAA,CAAA;IAC3D,OAAOoB,OAAAA;AACT;AAEA,SAAS+C,eAAgBC,CAAAA,GAAgB,EAAEC,GAAW,EAAEjD,OAAmB,EAAA;IACzE,IAAIkD,IAAAA,GAAO9E,iBAAkB+E,CAAAA,GAAG,CAACH,GAAAA,CAAAA;AACjC,IAAA,IAAI,CAACE,IAAM,EAAA;AACTA,QAAAA,IAAAA,GAAO,EAAE;QACT9E,iBAAkBgF,CAAAA,GAAG,CAACJ,GAAKE,EAAAA,IAAAA,CAAAA;AAC7B;AACA,IAAA,IAAI,CAACA,IAAI,CAACD,GAAAA,CAAI,EAAE;QACdC,IAAI,CAACD,GAAI,CAAA,GAAGD,GAAIjD,CAAAA,MAAM,CAACsD,eAAe,CAACrD,OAAAA,CAAQsD,gBAAgB,CAAC,CAAE,CAAA,CAAA;QAClE/E,OAAO,CAAC,EAAE,GAAGgF,IAAAA,CAAKC,GAAG,CAACR,GAAAA,CAAIS,KAAK,IAAIR,GAAK,EAAA,CAAA,CAAA;QACxC1E,OAAO,CAAC,EAAE,GAAGgF,IAAAA,CAAKC,GAAG,CAACR,GAAAA,CAAIU,MAAM,IAAIT,GAAK,EAAA,CAAA,CAAA;AACzCC,QAAAA,IAAI,CAACD,GAAAA,CAAI,CAACU,QAAQ,CAAC,SAAWpF,EAAAA,OAAAA,CAAAA;AAC9B,QAAA,IAAIyE,GAAIjD,CAAAA,MAAM,CAAChB,IAAI,KAAK,QAAU,EAAA;YAChCmE,IAAI,CAACD,GAAI,CAAA,CAACW,cAAc,CAAC,UAAUZ,GAAKC,EAAAA,GAAAA,EAAK,CAAG,EAAA,CAAA,EAAGY,YAAa,CAAA,eAAA,CAAA,CAAA;SAC3D,MAAA;AACLX,YAAAA,IAAI,CAACD,GAAI,CAAA,CAACa,UAAU,CAAC,QAAA,EAAUd,KAAKa,YAAa,CAAA,eAAA,CAAA,CAAA;AACjDX,YAAAA,IAAI,CAACD,GAAAA,CAAI,CAACU,QAAQ,CAAC,aAAeV,EAAAA,GAAAA,CAAAA;AACpC;AACF;IACA,OAAOC,IAAI,CAACD,GAAI,CAAA;AAClB;AAEA,SAASc,aAAAA,CACPhE,MAAsB,EACtBiE,QAAgB,EAChBC,UAAqB,EACrBC,UAAqB,EACrBlE,OAAmB,EAAA;IAEnB,MAAMmE,WAAAA,GAAcpE,OAAOqE,IAAI,CAACC,wBAAwB,CACtD,KAAA,EACA,GACA,CACA,EAAA;AAACH,QAAAA;AAAW,KAAA,EACZ,IACA,EAAA,KAAA,EACA,CACA,EAAA,IAAA,EACAF,QAAW,GAAA,CAAA,CAAA;IAEbG,WAAYG,CAAAA,iCAAiC,CAAC,CAAG,EAAA,KAAA,CAAA;AACjDvE,IAAAA,MAAAA,CAAOwE,UAAU,CAACvE,OAAAA,CAAAA;AAClBD,IAAAA,MAAAA,CAAOyE,YAAY,CAAC,CAAGzB,EAAAA,eAAAA,CAAgBkB,YAAYD,QAAUhE,EAAAA,OAAAA,CAAAA,CAAAA;AAC7DD,IAAAA,MAAAA,CAAO0E,cAAc,CAACN,WAAAA,CAAAA;AACtBO,IAAAA,kBAAAA,EAAAA;AACA,IAAA,IAAIT,eAAeC,UAAY,EAAA;AAC7BnE,QAAAA,MAAAA,CAAO4E,0BAA0B,CAACR,WAAa,EAAA,CAAA,EAAGF,YAAYD,QAAW,GAAA,CAAA,CAAA;AAC3E;IACAjE,MAAOqE,CAAAA,IAAI,CAACQ,kBAAkB,CAACT,WAAAA,CAAAA;AACjC;AAEA;;;;;;;;AAQC,IACM,SAASU,QAASC,CAAAA,SAAoB,EAAEC,cAA2B,EAAA;AACxE,IAAA,MAAMhF,MAASiF,GAAAA,SAAAA,EAAAA;AACf,IAAA,MAAMC,MAASF,GAAAA,cAAAA,CAAeG,mBAAmB,EAAE,CAAC,CAAE,CAAA;IACtD,MAAMtG,UAAAA,GAAagB,iBAAkBqF,CAAAA,MAAAA,CAAOpF,MAAM,CAAA;IAClD,MAAMsF,OAAAA,GAAUvG,aAAa,CAAI,GAAA,CAAA;AACjC,IAAA,IAAI,CAACT,UAAU,CAACgH,OAAAA,CAAQ,EAAE;AACxBhH,QAAAA,UAAU,CAACgH,OAAAA,CAAQ,GAAGrF,eAAAA,CAAgBC,MAAQnB,EAAAA,UAAAA,CAAAA;AAC9CN,QAAAA,OAAO,CAAC6G,OAAAA,CAAQ,GAAG,IAAI1G,cAAeG,CAAAA,UAAAA,CAAAA;AACxC;IACA,MAAMoB,OAAAA,GAAU7B,UAAU,CAACgH,OAAQ,CAAA;AACnC7G,IAAAA,OAAO,CAAC6G,OAAQ,CAAA,CAAEC,IAAI,CAACN,SAAAA,EAAWC,gBAAgBlB,YAAa,CAAA,eAAA,CAAA,CAAA;AAC/D9D,IAAAA,MAAAA,CAAOsF,gBAAgB,EAAA;AACvB,IAAA,MAAMrG,MAAS+F,GAAAA,cAAAA,CAAeG,mBAAmB,EAAE,CAAC,CAAE,CAAA;IACtD,IAAInF,MAAAA,CAAOhB,IAAI,KAAK,QAAU,EAAA;QAC5B,IAAK,IAAIkD,IAAI,CAAGA,EAAAA,CAAAA,GAAIjD,OAAOsG,aAAa,GAAG,GAAGrD,CAAK,EAAA,CAAA;YACjD8B,aAAchE,CAAAA,MAAAA,EAAQkC,CAAGjD,EAAAA,MAAAA,EAAQA,MAAQgB,EAAAA,OAAAA,CAAAA;AAC3C;KACK,MAAA;QACL,MAAMuF,cAAAA,GAAiBxF,OAAOqE,IAAI,CAACC,wBAAwB,CACzD,KAAA,EACAU,eAAeS,QAAQ,EAAA,EACvBT,eAAeU,SAAS,EAAA,EACxBV,eAAeG,mBAAmB,EAAE,CAAC,CAAE,CAAA,CAACrF,MAAM,EAC9C,IACA,EAAA,IAAA,CAAA;AAEF,QAAA,MAAM6F,MAASH,GAAAA,cAAAA,CAAeL,mBAAmB,EAAE,CAAC,CAAE,CAAA;QACtD,IAAK,IAAIjD,IAAI,CAAGA,EAAAA,CAAAA,GAAIjD,OAAOsG,aAAa,GAAG,GAAGrD,CAAK,EAAA,CAAA;YACjD8B,aAAchE,CAAAA,MAAAA,EAAQkC,CAAGjD,EAAAA,MAAAA,EAAQ0G,MAAQ1F,EAAAA,OAAAA,CAAAA;AAC3C;QACAD,MAAOqE,CAAAA,IAAI,CAACQ,kBAAkB,CAACW,cAAAA,CAAAA;AACjC;AACAxF,IAAAA,MAAAA,CAAO4F,eAAe,EAAA;AACxB;;;;"}