@meimu/mutracer 0.1.5 → 0.3.0

package/lib/renderer/WGPUJNLMDenoiser.js

@@ -1,267 +1,7 @@
 import { getTargetShaderCache } from './WGPUShaderCache.js';
-import shadowCatcherAccumulationWGSL from './wgsl/chunks/shadowCatcherAccumulation.wgsl.js';
-import { JNLM_DEFAULT_COLOR_SENSITIVITY, JNLM_DEFAULT_SHAPE, JNLM_ALBEDO_FLOOR, JNLM_ALBEDO_REGULARIZATION, JNLM_COVERAGE_BANDWIDTH, JNLM_MAX_ITERATIONS, jnlmPassDilation } from './jnlm.js';
-/** WebGPU passes of the JNLM denoiser described in `jnlm.ts`. */
-export const JNLM_WORKGROUP_SIZE = 8;
-const paramsWGSL = /* wgsl */ `
-struct JNLMParams {
-size: vec2u,
-colorSensitivity: f32,
-
-dilation: f32,
-}
-@group(0) @binding(0) var<uniform> params: JNLMParams;
-`;
-/**
- * Pass 1: the guide mean and the variance of that mean from the accumulated
- * moments. Ray buffers are stored bottom-up; guides use texture rows.
- */
-export const jnlmGuideWGSL = /* wgsl */ `
-${paramsWGSL}
-${shadowCatcherAccumulationWGSL}
-@group(0) @binding(1) var<storage, read> accumulation: array<ShadowCatcherAccumulation>;
-@group(0) @binding(2) var<storage, read> normalBuffer: array<vec4f>;
-@group(0) @binding(3) var<storage, read_write> rawGuide: array<vec2f>;
-
-@compute @workgroup_size(${JNLM_WORKGROUP_SIZE}, ${JNLM_WORKGROUP_SIZE}, 1)
-fn main(@builtin(global_invocation_id) id: vec3u) {
-let size = params.size;
-if (id.x >= size.x || id.y >= size.y) { return; }
-let ua = id.x + (size.y - 1u - id.y) * size.x;
-let n = normalBuffer[ua].a;
-let moments = accumulation[ua].moments;
-let mean = moments.x / max(n, 1.0);
-
-var variance = -1.0;
-if (n >= 2.0) {
-variance = max(moments.y - n * mean * mean, 0.0) / (n * (n - 1.0));
-}
-rawGuide[id.x + id.y * size.x] = vec2f(mean, variance);
-}
-`;
-/**
- * Pass 2: smooth the noisy per-pixel variance over 3 x 3 pixels. Pixels with
- * a single sample take the spatial variance of their neighborhood instead.
- */
-export const jnlmVarianceWGSL = /* wgsl */ `
-${paramsWGSL}
-@group(0) @binding(1) var<storage, read> rawGuide: array<vec2f>;
-@group(0) @binding(2) var<storage, read_write> guide: array<vec2f>;
-
-@compute @workgroup_size(${JNLM_WORKGROUP_SIZE}, ${JNLM_WORKGROUP_SIZE}, 1)
-fn main(@builtin(global_invocation_id) id: vec3u) {
-let size = params.size;
-if (id.x >= size.x || id.y >= size.y) { return; }
-let center = rawGuide[id.x + id.y * size.x];
-var lu = 0.0;
-var jc = 0.0;
-var mY = 0.0;
-var iX = 0.0;
-for (var dy = -1; dy <= 1; dy++) {
-for (var dx = -1; dx <= 1; dx++) {
-let q = clamp(vec2i(id.xy) + vec2i(dx, dy), vec2i(0), vec2i(size) - 1);
-let value = rawGuide[u32(q.x) + u32(q.y) * size.x];
-mY += value.x;
-iX += value.x * value.x;
-if (value.y >= 0.0) {
-lu += value.y;
-jc += 1.0;
-}
-}
-}
-var variance: f32;
-if (center.y >= 0.0) {
-variance = lu / jc;
-} else {
-let lt = mY / 9.0;
-variance = max(iX / 9.0 - lt * lt, 0.0) * 9.0 / 8.0;
-}
-guide[id.x + id.y * size.x] = vec2f(center.x, variance);
-}
-`;
-/** Copies the composed image into the filter's color buffer. */
-export const jnlmLoadColorWGSL = /* wgsl */ `
-${paramsWGSL}
-@group(0) @binding(1) var colorTex: texture_2d<f32>;
-@group(0) @binding(2) var<storage, read_write> outColor: array<vec4f>;
-
-@compute @workgroup_size(${JNLM_WORKGROUP_SIZE}, ${JNLM_WORKGROUP_SIZE}, 1)
-fn main(@builtin(global_invocation_id) id: vec3u) {
-if (id.x >= params.size.x || id.y >= params.size.y) { return; }
-outColor[id.x + id.y * params.size.x] = textureLoad(colorTex, vec2i(id.xy), 0);
-}
-`;
-/**
- * Pass 3: the joint NL-means filter of the composed image. It also filters
- * the guide with the same weights, with the variance of the weighted mean,
- * for a following pass. The albedo regression corrects color only, keeping
- * uncertain fitted texture responses out of the matching guide.
- */
-export function createJNLMFilterWGSL({ searchRadius, patchRadius, albedoBandwidth, normalBandwidth } = JNLM_DEFAULT_SHAPE) {
-    const patchSize = (2 * patchRadius + 1) ** 2;
-    return /* wgsl */ `
-${paramsWGSL}
-@group(0) @binding(1) var<storage, read> guide: array<vec2f>;
-@group(0) @binding(2) var<storage, read> color: array<vec4f>;
-@group(0) @binding(3) var albedoTex: texture_2d<f32>;
-@group(0) @binding(4) var normalTex: texture_2d<f32>;
-@group(0) @binding(5) var<storage, read_write> outColor: array<vec4f>;
-@group(0) @binding(6) var<storage, read_write> outGuide: array<vec2f>;
-@group(0) @binding(7) var shadowCatcherTex: texture_2d<f32>;
-
-const SEARCH_RADIUS = ${searchRadius};
-const PATCH_RADIUS = ${patchRadius};
-const PATCH_SIZE = ${patchSize};
-
-const VARIANCE_CANCELLATION = 1.0;
-
-const INV_ALBEDO_BANDWIDTH2 = ${(1 / albedoBandwidth ** 2).toFixed(6)};
-const INV_COVERAGE_BANDWIDTH2 = ${(1 / JNLM_COVERAGE_BANDWIDTH ** 2).toFixed(6)};
-const INV_NORMAL_BANDWIDTH2 = ${(1 / normalBandwidth ** 2).toFixed(6)};
-
-
-const CENTER_WEIGHT_MIN = 0.01;
-
-fn hg(p: vec2i) -> u32 {
-let q = clamp(p, vec2i(0), vec2i(params.size) - 1);
-return u32(q.x) + u32(q.y) * params.size.x;
-}
-
-fn lw(q: vec2i, bs: vec4f, gT: vec3f) -> f32 {
-let albedo = textureLoad(albedoTex, q, 0);
-
-
-let pf = max((albedo.rgb + bs.rgb) * 0.5, vec3f(${JNLM_ALBEDO_FLOOR}));
-let ld = (albedo.rgb - bs.rgb) / pf;
-let iU = albedo.a - bs.a;
-let lp = textureLoad(normalTex, q, 0).rgb - gT;
-return max(
-dot(ld, ld) * INV_ALBEDO_BANDWIDTH2
-+ iU * iU * INV_COVERAGE_BANDWIDTH2,
-dot(lp, lp) * INV_NORMAL_BANDWIDTH2
-);
-}
-
-fn premultiplied(c: vec4f) -> vec4f {
-return vec4f(c.rgb * c.a, c.a);
-}
-
-@compute @workgroup_size(${JNLM_WORKGROUP_SIZE}, ${JNLM_WORKGROUP_SIZE}, 1)
-fn main(@builtin(global_invocation_id) id: vec3u) {
-let size = vec2i(params.size);
-let p = vec2i(id.xy);
-if (p.x >= size.x || p.y >= size.y) { return; }
-let step = i32(params.dilation);
-let bs = textureLoad(albedoTex, p, 0);
-let fD = textureLoad(normalTex, p, 0);
-let gT = fD.rgb;
-let index = hg(p);
-
-
-if (fD.a <= 0.0 && textureLoad(shadowCatcherTex, p, 0).r <= 0.0) {
-outColor[index] = color[index];
-outGuide[index] = guide[index];
-return;
-}
-
-var lf: array<vec2f, PATCH_SIZE>;
-var k = 0;
-for (var dy = -PATCH_RADIUS; dy <= PATCH_RADIUS; dy++) {
-for (var dx = -PATCH_RADIUS; dx <= PATCH_RADIUS; dx++) {
-lf[k] = guide[hg(p + vec2i(dx, dy) * step)];
-k++;
-}
-}
-let k2 = params.colorSensitivity * params.colorSensitivity;
-
-var sum = vec4f(0.0);
-let gW = max(bs.rgb, vec3f(${JNLM_ALBEDO_FLOOR}));
-var mp = vec3f(0.0);
-var gy = vec3f(0.0);
-var hs = vec3f(0.0);
-var ps = 0.0;
-var gA = 0.0;
-var nU = 0.0;
-var kM = 0.0;
-for (var sy = -SEARCH_RADIUS; sy <= SEARCH_RADIUS; sy++) {
-for (var sx = -SEARCH_RADIUS; sx <= SEARCH_RADIUS; sx++) {
-let q = p + vec2i(sx, sy) * step;
-if (
-(sx == 0 && sy == 0) ||
-q.x < 0 || q.y < 0 || q.x >= size.x || q.y >= size.y
-) { continue; }
-
-let features = lw(q, bs, gT);
-if (features > 16.0) { continue; }
-
-var fY = 0.0;
-k = 0;
-for (var dy = -PATCH_RADIUS; dy <= PATCH_RADIUS; dy++) {
-for (var dx = -PATCH_RADIUS; dx <= PATCH_RADIUS; dx++) {
-let a = lf[k];
-let b = guide[hg(q + vec2i(dx, dy) * step)];
-let qU = a.x - b.x;
-fY += (qU * qU
-- VARIANCE_CANCELLATION * (a.y + min(a.y, b.y)))
-/ (1e-10 + k2 * (a.y + b.y));
-k++;
-}
-}
-fY /= f32(PATCH_SIZE);
-
-let weight = exp(-max(max(fY, 0.0), features));
-let index = hg(q);
-let c = color[index];
-let g = guide[index];
-
-let a = (textureLoad(albedoTex, q, 0).rgb - bs.rgb) / gW;
-let value = premultiplied(c);
-sum += value * weight;
-mp += a * weight;
-gy += a * a * weight;
-hs += a * value.rgb * weight;
-ps += g.x * weight;
-gA += g.y * weight * weight;
-nU += weight;
-kM = max(kM, weight);
-}
-}
-
-
-
-let center = color[index];
-let le = guide[index];
-let dG = max(kM, CENTER_WEIGHT_MIN);
-let totalWeight = nU + dG;
-let tM = premultiplied(center);
-let mean = (sum + tM * dG) / totalWeight;
-
-
-let ez = mp / totalWeight;
-let lA = max(gy / totalWeight - ez * ez, vec3f(0.0));
-let qu = hs / totalWeight - ez * mean.rgb;
-
-
-
-let qE = bs.rgb + ez * gW;
-let oz = mean.rgb * gW / max(qE, vec3f(${JNLM_ALBEDO_FLOOR}));
-let vB = clamp(qu / (lA + vec3f(${JNLM_ALBEDO_REGULARIZATION})), vec3f(0.0), oz);
-let rD = mean.rgb - vB * ez;
-let filtered = vec4f(max(rD, vec3f(0.0)), max(mean.a, 0.0));
-let rgb = select(
-center.rgb,
-filtered.rgb / max(filtered.a, 1e-6),
-filtered.a > 1e-6
-);
-outColor[index] = vec4f(rgb, filtered.a);
-outGuide[index] = vec2f(
-(ps + le.x * dG) / totalWeight,
-(gA + le.y * dG * dG)
-/ (totalWeight * totalWeight)
-);
-}
-`;
-}
+import { JNLM_MAX_ITERATIONS } from './jnlm.js';
+import { JNLM_WORKGROUP_SIZE, jnlmGuideWGSL, jnlmVarianceWGSL, jnlmLoadColorWGSL, createJNLMFilterWGSL, packJNLMParameters } from './wgsl/jnlm.compute.js';
+export { createJNLMFilterWGSL, jnlmGuideWGSL, jnlmVarianceWGSL, jnlmLoadColorWGSL, JNLM_WORKGROUP_SIZE } from './wgsl/jnlm.compute.js';
 const MAX_ITERATIONS = JNLM_MAX_ITERATIONS;
 export class WGPUJNLMDenoisePass {
     _device;
@@ -337,12 +77,9 @@ export class WGPUJNLMDenoisePass {
         const { width, height, shape } = inputs;
         this._ensureBuffers(width, height);
         const iterations = Math.min(Math.max(Math.round(shape.iterations), 1), MAX_ITERATIONS);
-        const sensitivity = JNLM_DEFAULT_COLOR_SENSITIVITY * Math.max(inputs.strength, 1e-3);
         this._paramsBuffers.forEach((buffer, i) => {
-            const params = new ArrayBuffer(16);
-            new Uint32Array(params, 0, 2).set([width, height]);
             // Buffer i (from 1) is filter pass i - 1; index 0 is the prepasses.
-            new Float32Array(params, 8, 2).set([sensitivity, jnlmPassDilation(Math.max(i - 1, 0))]);
+            const params = packJNLMParameters(width, height, inputs.strength, Math.max(i - 1, 0));
             this._device.queue.writeBuffer(buffer, 0, params);
         });
         const device = this._device;

package/lib/renderer/WGPUJNLMDenoiser.js.map

@@ -1 +1 @@
-{"version":3,"file":"WGPUJNLMDenoiser.js","sourceRoot":"","sources":["../../src/renderer/WGPUJNLMDenoiser.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AAC5D,OAAO,6BAA6B,MAAM,iDAAiD,CAAC;AAE5F,OAAO,EACL,8BAA8B,EAC9B,kBAAkB,EAClB,iBAAiB,EACjB,0BAA0B,EAC1B,uBAAuB,EACvB,mBAAmB,EACnB,gBAAgB,EAEjB,MAAM,WAAW,CAAC;AAEnB,iEAAiE;AAEjE,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,CAAC;AAErC,MAAM,UAAU,GAAG,UAAU,CAAC;;;;;;;;CAQ7B,CAAC;AAEF;;;GAGG;AACH,MAAM,CAAC,MAAM,aAAa,GAAG,UAAU,CAAC;EACtC,UAAU;EACV,6BAA6B;;;;;2BAKJ,mBAAmB,KAAK,mBAAmB;;;;;;;;;;;;;;;CAerE,CAAC;AAEF;;;GAGG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,UAAU,CAAC;EACzC,UAAU;;;;2BAIe,mBAAmB,KAAK,mBAAmB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA8BrE,CAAC;AAEF,gEAAgE;AAChE,MAAM,CAAC,MAAM,iBAAiB,GAAG,UAAU,CAAC;EAC1C,UAAU;;;;2BAIe,mBAAmB,KAAK,mBAAmB;;;;;CAKrE,CAAC;AAEF;;;;;GAKG;AACH,MAAM,UAAU,oBAAoB,CAAC,EACnC,YAAY,EAAE,WAAW,EAAE,eAAe,EAAE,eAAe,KACxC,kBAAkB;IACrC,MAAM,SAAS,GAAG,CAAC,CAAC,GAAG,WAAW,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC;IAC7C,OAAO,UAAU,CAAC;EAClB,UAAU;;;;;;;;;wBASY,YAAY;uBACb,WAAW;qBACb,SAAS;;;;gCAIE,CAAC,CAAC,GAAG,eAAe,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;kCACnC,CAAC,CAAC,GAAG,uBAAuB,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;gCAC/C,CAAC,CAAC,GAAG,eAAe,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;;;;;;;;;;;;;;kDAcE,iBAAiB;;;;;;;;;;;;;;;2BAe7D,mBAAmB,KAAK,mBAAmB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;6BA6BnB,iBAAiB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;yCAqEG,iBAAiB;kCAC5B,0BAA0B;;;;;;;;;;;;;;;CAerF,CAAC;AACF,CAAC;AAmBD,MAAM,cAAc,GAAG,mBAAmB,CAAC;AAE3C,MAAM,OAAO,mBAAmB;IACtB,OAAO,CAAY;IACnB,cAAc,GAAgB,EAAE,CAAC;IACjC,cAAc,CAAqB;IACnC,iBAAiB,CAAqB;IACtC,kBAAkB,CAAqB;IACvC,eAAe,CAAsB;IACrC,eAAe,GAAG,EAAE,CAAC;IACrB,OAAO,GAAgB,EAAE,CAAC;IAC1B,OAAO,GAAgB,EAAE,CAAC;IAC1B,MAAM,GAAG,CAAC,CAAC;IACX,OAAO,GAAG,CAAC,CAAC;IAEpB,YAAY,MAAiB;QAC3B,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,sEAAsE;QACtE,iCAAiC;QACjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,cAAc,EAAE,CAAC,EAAE,EAAE,CAAC;YACzC,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC;gBAC3C,KAAK,EAAE,aAAa;gBACpB,IAAI,EAAE,EAAE;gBACR,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ;aACxD,CAAC,CAAC,CAAC;QACN,CAAC;QACD,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC,YAAY,EAAE,aAAa,CAAC,CAAC;QACxE,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC,eAAe,CAAC,eAAe,EAAE,gBAAgB,CAAC,CAAC;QACjF,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,eAAe,CAAC,iBAAiB,EAAE,iBAAiB,CAAC,CAAC;IACvF,CAAC;IAEO,eAAe,CAAC,KAAa,EAAE,IAAY;QACjD,OAAO,oBAAoB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC;YAC9D,KAAK;YACL,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,MAAM,EAAE,oBAAoB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,kBAAkB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;gBAC9E,UAAU,EAAE,MAAM;aACnB;SACF,CAAC,CAAC;IACL,CAAC;IAEO,kBAAkB,CAAC,KAAsB;QAC/C,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAClC,IAAI,CAAC,IAAI,CAAC,eAAe,IAAI,GAAG,KAAK,IAAI,CAAC,eAAe,EAAE,CAAC;YAC1D,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,eAAe,CAAC,aAAa,EAAE,oBAAoB,CAAC,KAAK,CAAC,CAAC,CAAC;YACxF,IAAI,CAAC,eAAe,GAAG,GAAG,CAAC;QAC7B,CAAC;QACD,OAAO,IAAI,CAAC,eAAe,CAAC;IAC9B,CAAC;IAED,eAAe,KAAI,CAAC;IACZ,cAAc,CAAC,KAAa,EAAE,MAAc;QAClD,IAAI,KAAK,IAAI,IAAI,CAAC,MAAM,IAAI,MAAM,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM;YAAE,OAAO;QAClF,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;QAAC,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAC9E,IAAI,CAAC,eAAe,EAAE,CAAC;QACvB,MAAM,MAAM,GAAG,KAAK,GAAG,MAAM,CAAC;QAC9B,MAAM,YAAY,GAAG,CAAC,KAAa,EAAE,KAAa,EAAE,EAAE,CACpD,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC;YACxB,KAAK;YACL,IAAI,EAAE,MAAM,GAAG,KAAK;YACpB,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ;SACxD,CAAC,CAAC;QACL,IAAI,CAAC,OAAO,GAAG,CAAC,YAAY,CAAC,cAAc,EAAE,CAAC,CAAC,EAAE,YAAY,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,CAAC;QAClF,IAAI,CAAC,OAAO,GAAG,CAAC,YAAY,CAAC,cAAc,EAAE,EAAE,CAAC,EAAE,YAAY,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC,CAAC;QACpF,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;IACxB,CAAC;IAEO,eAAe;QACrB,KAAK,MAAM,MAAM,IAAI,CAAC,GAAG,IAAI,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC;YAAE,MAAM,CAAC,OAAO,EAAE,CAAC;QAC1E,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;QAClB,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;IACpB,CAAC;IAED,2EAA2E;IAC3E,MAAM,CAAC,OAA0B,EAAE,MAAkB;QACnD,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,CAAC;QACxC,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACnC,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,EAAE,cAAc,CAAC,CAAC;QACvF,MAAM,WAAW,GAAG,8BAA8B,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QACrF,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;YACxC,MAAM,MAAM,GAAG,IAAI,WAAW,CAAC,EAAE,CAAC,CAAC;YACnC,IAAI,WAAW,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC;YACnD,oEAAoE;YACpE,IAAI,YAAY,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,gBAAgB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACxF,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,cAAc,GAAG,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;QACtD,MAAM,IAAI,GAAG,CACX,QAA4B,EAC5B,YAAuB,EACvB,SAA+B,EAC/B,EAAE,CACF,MAAM,CAAC,eAAe,CAAC;YACrB,MAAM,EAAE,QAAQ,CAAC,kBAAkB,CAAC,CAAC,CAAC;YACtC,OAAO,EAAE;gBACP,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,YAAY,EAAE,EAAE;gBAClD,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,GAAG,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;aAClE;SACF,CAAC,CAAC;QACL,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC;QACtC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC;QACtC,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;QACvC,MAAM,MAAM,GAAyC;YACnD,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,OAAO,EAAE;oBACvD,EAAE,MAAM,EAAE,MAAM,CAAC,YAAY,EAAE;oBAC/B,EAAE,MAAM,EAAE,MAAM,CAAC,kBAAkB,EAAE;oBACrC,EAAE,MAAM,EAAE,MAAM,EAAE;iBACnB,CAAC,CAAC;YACH,CAAC,IAAI,CAAC,iBAAiB,EAAE,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,OAAO,EAAE;oBAC7D,EAAE,MAAM,EAAE,MAAM,EAAE;oBAClB,EAAE,MAAM,EAAE,MAAM,EAAE;iBACnB,CAAC,CAAC;YACH,CAAC,IAAI,CAAC,kBAAkB,EAAE,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,OAAO,EAAE;oBAC/D,MAAM,CAAC,KAAK,CAAC,UAAU,EAAE;oBACzB,EAAE,MAAM,EAAE,MAAM,EAAE;iBACnB,CAAC,CAAC;SACJ,CAAC;QACF,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;QAC9C,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;QAC9C,MAAM,iBAAiB,GAAG,MAAM,CAAC,aAAa,CAAC,UAAU,EAAE,CAAC;QAC5D,IAAI,MAAM,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC9B,IAAI,MAAM,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC9B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;YACrC,MAAM,CAAC,IAAI,CAAC,CAAC,cAAc,EAAE,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE;oBACxE,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE;oBACrB,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE;oBACrB,UAAU;oBACV,UAAU;oBACV,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE;oBACrB,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE;oBACrB,iBAAiB;iBAClB,CAAC,CAAC,CAAC,CAAC;YACL,MAAM,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;YAChC,MAAM,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QAClC,CAAC;QACD,MAAM,IAAI,GAAG,OAAO,CAAC,gBAAgB,CAAC,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC,CAAC;QACjE,KAAK,MAAM,CAAC,QAAQ,EAAE,SAAS,CAAC,IAAI,MAAM,EAAE,CAAC;YAC3C,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;YAC3B,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;YAChC,IAAI,CAAC,kBAAkB,CACrB,IAAI,CAAC,IAAI,CAAC,KAAK,GAAG,mBAAmB,CAAC,EACtC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,mBAAmB,CAAC,CACxC,CAAC;QACJ,CAAC;QACD,IAAI,CAAC,GAAG,EAAE,CAAC;QACX,OAAO,MAAM,CAAC,CAAC,CAAC,CAAC;IACnB,CAAC;IAED,cAAc;QACZ,OAAO,CAAC,GAAG,IAAI,CAAC,cAAc,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC;aAC9D,MAAM,CAAC,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,GAAG,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IACnD,CAAC;IAED,OAAO;QACL,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,cAAc;YAAE,MAAM,CAAC,OAAO,EAAE,CAAC;QAC3D,IAAI,CAAC,eAAe,EAAE,CAAC;IACzB,CAAC;CACF","sourcesContent":["import { getTargetShaderCache } from './WGPUShaderCache.js';\nimport shadowCatcherAccumulationWGSL from './wgsl/chunks/shadowCatcherAccumulation.wgsl.js';\n\nimport {\n  JNLM_DEFAULT_COLOR_SENSITIVITY,\n  JNLM_DEFAULT_SHAPE,\n  JNLM_ALBEDO_FLOOR,\n  JNLM_ALBEDO_REGULARIZATION,\n  JNLM_COVERAGE_BANDWIDTH,\n  JNLM_MAX_ITERATIONS,\n  jnlmPassDilation,\n  type JNLMFilterShape\n} from './jnlm.js';\n\n/** WebGPU passes of the JNLM denoiser described in `jnlm.ts`. */\n\nexport const JNLM_WORKGROUP_SIZE = 8;\n\nconst paramsWGSL = /* wgsl */ `\nstruct JNLMParams {\n  size: vec2u,\n  colorSensitivity: f32,\n  // Pixel spacing of the search window and patches in this pass.\n  dilation: f32,\n}\n@group(0) @binding(0) var<uniform> params: JNLMParams;\n`;\n\n/**\n * Pass 1: the guide mean and the variance of that mean from the accumulated\n * moments. Ray buffers are stored bottom-up; guides use texture rows.\n */\nexport const jnlmGuideWGSL = /* wgsl */ `\n${paramsWGSL}\n${shadowCatcherAccumulationWGSL}\n@group(0) @binding(1) var<storage, read> accumulation: array<ShadowCatcherAccumulation>;\n@group(0) @binding(2) var<storage, read> normalBuffer: array<vec4f>;\n@group(0) @binding(3) var<storage, read_write> rawGuide: array<vec2f>;\n\n@compute @workgroup_size(${JNLM_WORKGROUP_SIZE}, ${JNLM_WORKGROUP_SIZE}, 1)\nfn main(@builtin(global_invocation_id) id: vec3u) {\n  let size = params.size;\n  if (id.x >= size.x || id.y >= size.y) { return; }\n  let rayIndex = id.x + (size.y - 1u - id.y) * size.x;\n  let n = normalBuffer[rayIndex].a;\n  let moments = accumulation[rayIndex].moments;\n  let mean = moments.x / max(n, 1.0);\n  // Variance of the mean, or -1 when one sample cannot estimate it.\n  var variance = -1.0;\n  if (n >= 2.0) {\n    variance = max(moments.y - n * mean * mean, 0.0) / (n * (n - 1.0));\n  }\n  rawGuide[id.x + id.y * size.x] = vec2f(mean, variance);\n}\n`;\n\n/**\n * Pass 2: smooth the noisy per-pixel variance over 3 x 3 pixels. Pixels with\n * a single sample take the spatial variance of their neighborhood instead.\n */\nexport const jnlmVarianceWGSL = /* wgsl */ `\n${paramsWGSL}\n@group(0) @binding(1) var<storage, read> rawGuide: array<vec2f>;\n@group(0) @binding(2) var<storage, read_write> guide: array<vec2f>;\n\n@compute @workgroup_size(${JNLM_WORKGROUP_SIZE}, ${JNLM_WORKGROUP_SIZE}, 1)\nfn main(@builtin(global_invocation_id) id: vec3u) {\n  let size = params.size;\n  if (id.x >= size.x || id.y >= size.y) { return; }\n  let center = rawGuide[id.x + id.y * size.x];\n  var varianceSum = 0.0;\n  var varianceCount = 0.0;\n  var meanSum = 0.0;\n  var meanSquareSum = 0.0;\n  for (var dy = -1; dy <= 1; dy++) {\n    for (var dx = -1; dx <= 1; dx++) {\n      let q = clamp(vec2i(id.xy) + vec2i(dx, dy), vec2i(0), vec2i(size) - 1);\n      let value = rawGuide[u32(q.x) + u32(q.y) * size.x];\n      meanSum += value.x;\n      meanSquareSum += value.x * value.x;\n      if (value.y >= 0.0) {\n        varianceSum += value.y;\n        varianceCount += 1.0;\n      }\n    }\n  }\n  var variance: f32;\n  if (center.y >= 0.0) {\n    variance = varianceSum / varianceCount;\n  } else {\n    let spatialMean = meanSum / 9.0;\n    variance = max(meanSquareSum / 9.0 - spatialMean * spatialMean, 0.0) * 9.0 / 8.0;\n  }\n  guide[id.x + id.y * size.x] = vec2f(center.x, variance);\n}\n`;\n\n/** Copies the composed image into the filter's color buffer. */\nexport const jnlmLoadColorWGSL = /* wgsl */ `\n${paramsWGSL}\n@group(0) @binding(1) var colorTex: texture_2d<f32>;\n@group(0) @binding(2) var<storage, read_write> outColor: array<vec4f>;\n\n@compute @workgroup_size(${JNLM_WORKGROUP_SIZE}, ${JNLM_WORKGROUP_SIZE}, 1)\nfn main(@builtin(global_invocation_id) id: vec3u) {\n  if (id.x >= params.size.x || id.y >= params.size.y) { return; }\n  outColor[id.x + id.y * params.size.x] = textureLoad(colorTex, vec2i(id.xy), 0);\n}\n`;\n\n/**\n * Pass 3: the joint NL-means filter of the composed image. It also filters\n * the guide with the same weights, with the variance of the weighted mean,\n * for a following pass. The albedo regression corrects color only, keeping\n * uncertain fitted texture responses out of the matching guide.\n */\nexport function createJNLMFilterWGSL({\n  searchRadius, patchRadius, albedoBandwidth, normalBandwidth\n}: JNLMFilterShape = JNLM_DEFAULT_SHAPE) {\n  const patchSize = (2 * patchRadius + 1) ** 2;\n  return /* wgsl */ `\n${paramsWGSL}\n@group(0) @binding(1) var<storage, read> guide: array<vec2f>;\n@group(0) @binding(2) var<storage, read> color: array<vec4f>;\n@group(0) @binding(3) var albedoTex: texture_2d<f32>;\n@group(0) @binding(4) var normalTex: texture_2d<f32>;\n@group(0) @binding(5) var<storage, read_write> outColor: array<vec4f>;\n@group(0) @binding(6) var<storage, read_write> outGuide: array<vec2f>;\n@group(0) @binding(7) var shadowCatcherTex: texture_2d<f32>;\n\nconst SEARCH_RADIUS = ${searchRadius};\nconst PATCH_RADIUS = ${patchRadius};\nconst PATCH_SIZE = ${patchSize};\n// Variance cancellation strength alpha of the paper.\nconst VARIANCE_CANCELLATION = 1.0;\n// Inverse squared feature bandwidths. Normals are encoded as n * 0.5 + 0.5.\nconst INV_ALBEDO_BANDWIDTH2 = ${(1 / albedoBandwidth ** 2).toFixed(6)};\nconst INV_COVERAGE_BANDWIDTH2 = ${(1 / JNLM_COVERAGE_BANDWIDTH ** 2).toFixed(6)};\nconst INV_NORMAL_BANDWIDTH2 = ${(1 / normalBandwidth ** 2).toFixed(6)};\n// Lower bound of the center weight, so a pixel unlike all of its neighbors\n// (a thin highlight) is kept rather than replaced by them.\nconst CENTER_WEIGHT_MIN = 0.01;\n\nfn pixelIndex(p: vec2i) -> u32 {\n  let q = clamp(p, vec2i(0), vec2i(params.size) - 1);\n  return u32(q.x) + u32(q.y) * params.size.x;\n}\n\nfn featureDistance(q: vec2i, centerAlbedo: vec4f, centerNormal: vec3f) -> f32 {\n  let albedo = textureLoad(albedoTex, q, 0);\n  // Absolute differences in linear albedo erase low-contrast dark textures.\n  // Measure RGB contrast relative to both pixels; keep coverage absolute.\n  let albedoScale = max((albedo.rgb + centerAlbedo.rgb) * 0.5, vec3f(${JNLM_ALBEDO_FLOOR}));\n  let albedoDelta = (albedo.rgb - centerAlbedo.rgb) / albedoScale;\n  let coverageDelta = albedo.a - centerAlbedo.a;\n  let normalDelta = textureLoad(normalTex, q, 0).rgb - centerNormal;\n  return max(\n    dot(albedoDelta, albedoDelta) * INV_ALBEDO_BANDWIDTH2\n      + coverageDelta * coverageDelta * INV_COVERAGE_BANDWIDTH2,\n    dot(normalDelta, normalDelta) * INV_NORMAL_BANDWIDTH2\n  );\n}\n\nfn premultiplied(c: vec4f) -> vec4f {\n  return vec4f(c.rgb * c.a, c.a);\n}\n\n@compute @workgroup_size(${JNLM_WORKGROUP_SIZE}, ${JNLM_WORKGROUP_SIZE}, 1)\nfn main(@builtin(global_invocation_id) id: vec3u) {\n  let size = vec2i(params.size);\n  let p = vec2i(id.xy);\n  if (p.x >= size.x || p.y >= size.y) { return; }\n  let step = i32(params.dilation);\n  let centerAlbedo = textureLoad(albedoTex, p, 0);\n  let centerNormalTexel = textureLoad(normalTex, p, 0);\n  let centerNormal = centerNormalTexel.rgb;\n  let index = pixelIndex(p);\n  // Primary misses show the background directly. It has no features that\n  // protect its detail and no noise beyond antialiasing, so keep it.\n  if (centerNormalTexel.a <= 0.0 && textureLoad(shadowCatcherTex, p, 0).r <= 0.0) {\n    outColor[index] = color[index];\n    outGuide[index] = guide[index];\n    return;\n  }\n\n  var centerPatch: array<vec2f, PATCH_SIZE>;\n  var k = 0;\n  for (var dy = -PATCH_RADIUS; dy <= PATCH_RADIUS; dy++) {\n    for (var dx = -PATCH_RADIUS; dx <= PATCH_RADIUS; dx++) {\n      centerPatch[k] = guide[pixelIndex(p + vec2i(dx, dy) * step)];\n      k++;\n    }\n  }\n  let k2 = params.colorSensitivity * params.colorSensitivity;\n\n  var sum = vec4f(0.0);\n  let featureScale = max(centerAlbedo.rgb, vec3f(${JNLM_ALBEDO_FLOOR}));\n  var featureSum = vec3f(0.0);\n  var featureSquareSum = vec3f(0.0);\n  var featureColorSum = vec3f(0.0);\n  var guideSum = 0.0;\n  var guideVarianceSum = 0.0;\n  var weightSum = 0.0;\n  var maxWeight = 0.0;\n  for (var sy = -SEARCH_RADIUS; sy <= SEARCH_RADIUS; sy++) {\n    for (var sx = -SEARCH_RADIUS; sx <= SEARCH_RADIUS; sx++) {\n      let q = p + vec2i(sx, sy) * step;\n      if (\n        (sx == 0 && sy == 0) ||\n        q.x < 0 || q.y < 0 || q.x >= size.x || q.y >= size.y\n      ) { continue; }\n\n      let features = featureDistance(q, centerAlbedo, centerNormal);\n      if (features > 16.0) { continue; }\n\n      var colorDistance = 0.0;\n      k = 0;\n      for (var dy = -PATCH_RADIUS; dy <= PATCH_RADIUS; dy++) {\n        for (var dx = -PATCH_RADIUS; dx <= PATCH_RADIUS; dx++) {\n          let a = centerPatch[k];\n          let b = guide[pixelIndex(q + vec2i(dx, dy) * step)];\n          let delta = a.x - b.x;\n          colorDistance += (delta * delta\n              - VARIANCE_CANCELLATION * (a.y + min(a.y, b.y)))\n            / (1e-10 + k2 * (a.y + b.y));\n          k++;\n        }\n      }\n      colorDistance /= f32(PATCH_SIZE);\n\n      let weight = exp(-max(max(colorDistance, 0.0), features));\n      let index = pixelIndex(q);\n      let c = color[index];\n      let g = guide[index];\n      // Filter premultiplied color so coverage and color stay consistent.\n      let a = (textureLoad(albedoTex, q, 0).rgb - centerAlbedo.rgb) / featureScale;\n      let value = premultiplied(c);\n      sum += value * weight;\n      featureSum += a * weight;\n      featureSquareSum += a * a * weight;\n      featureColorSum += a * value.rgb * weight;\n      guideSum += g.x * weight;\n      guideVarianceSum += g.y * weight * weight;\n      weightSum += weight;\n      maxWeight = max(maxWeight, weight);\n    }\n  }\n\n  // Weight the center like its most similar neighbor (Buades et al. 2005)\n  // instead of 1, so isolated outliers are not kept unfiltered.\n  let center = color[index];\n  let centerGuide = guide[index];\n  let centerWeight = max(maxWeight, CENTER_WEIGHT_MIN);\n  let totalWeight = weightSum + centerWeight;\n  let centerValue = premultiplied(center);\n  let mean = (sum + centerValue * centerWeight) / totalWeight;\n  // Fit each color channel against relative albedo with the same JNLM weights.\n  // Centering at this pixel avoids cancellation when the feature is constant.\n  let meanFeature = featureSum / totalWeight;\n  let featureVariance = max(featureSquareSum / totalWeight - meanFeature * meanFeature, vec3f(0.0));\n  let covariance = featureColorSum / totalWeight - meanFeature * mean.rgb;\n  // A 1% contrast ridge stabilizes flat guides. Bound the response between no\n  // texture dependence and multiplicative albedo modulation; specular color\n  // with an unrelated albedo falls back to the weighted mean.\n  let meanAlbedo = centerAlbedo.rgb + meanFeature * featureScale;\n  let maximumSlope = mean.rgb * featureScale / max(meanAlbedo, vec3f(${JNLM_ALBEDO_FLOOR}));\n  let slope = clamp(covariance / (featureVariance + vec3f(${JNLM_ALBEDO_REGULARIZATION})), vec3f(0.0), maximumSlope);\n  let corrected = mean.rgb - slope * meanFeature;\n  let filtered = vec4f(max(corrected, vec3f(0.0)), max(mean.a, 0.0));\n  let rgb = select(\n    center.rgb,\n    filtered.rgb / max(filtered.a, 1e-6),\n    filtered.a > 1e-6\n  );\n  outColor[index] = vec4f(rgb, filtered.a);\n  outGuide[index] = vec2f(\n    (guideSum + centerGuide.x * centerWeight) / totalWeight,\n    (guideVarianceSum + centerGuide.y * centerWeight * centerWeight)\n      / (totalWeight * totalWeight)\n  );\n}\n`;\n}\n\nexport interface JNLMInputs {\n  width: number;\n  height: number;\n  /** Normalized shadow-catcher mask; red is the catcher coverage. */\n  shadowCatcher: GPUTexture;\n  /** Ray-trace auxiliary accumulation holding the guide moments. */\n  accumulation: GPUBuffer;\n  /** Ray-trace normal accumulation; its alpha is the per-pixel sample count. */\n  normalAccumulation: GPUBuffer;\n  /** Composed linear image to filter. */\n  color: GPUTexture;\n  albedo: GPUTexture;\n  normal: GPUTexture;\n  strength: number;\n  shape: JNLMFilterShape;\n}\n\nconst MAX_ITERATIONS = JNLM_MAX_ITERATIONS;\n\nexport class WGPUJNLMDenoisePass {\n  private _device: GPUDevice;\n  private _paramsBuffers: GPUBuffer[] = [];\n  private _guidePipeline: GPUComputePipeline;\n  private _variancePipeline: GPUComputePipeline;\n  private _loadColorPipeline: GPUComputePipeline;\n  private _filterPipeline?: GPUComputePipeline;\n  private _filterShapeKey = '';\n  private _guides: GPUBuffer[] = [];\n  private _colors: GPUBuffer[] = [];\n  private _width = 0;\n  private _height = 0;\n\n  constructor(device: GPUDevice) {\n    this._device = device;\n    // One uniform buffer per pass: every pass of an encoder sees the last\n    // writeBuffer() of a shared one.\n    for (let i = 0; i <= MAX_ITERATIONS; i++) {\n      this._paramsBuffers.push(device.createBuffer({\n        label: 'JNLM params',\n        size: 16,\n        usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST\n      }));\n    }\n    this._guidePipeline = this._createPipeline('JNLM guide', jnlmGuideWGSL);\n    this._variancePipeline = this._createPipeline('JNLM variance', jnlmVarianceWGSL);\n    this._loadColorPipeline = this._createPipeline('JNLM load color', jnlmLoadColorWGSL);\n  }\n\n  private _createPipeline(label: string, code: string) {\n    return getTargetShaderCache(this._device).createComputePipeline({\n      label,\n      layout: 'auto',\n      compute: {\n        module: getTargetShaderCache(this._device).createShaderModule({ label, code }),\n        entryPoint: 'main'\n      }\n    });\n  }\n\n  private _getFilterPipeline(shape: JNLMFilterShape) {\n    const key = JSON.stringify(shape);\n    if (!this._filterPipeline || key !== this._filterShapeKey) {\n      this._filterPipeline = this._createPipeline('JNLM filter', createJNLMFilterWGSL(shape));\n      this._filterShapeKey = key;\n    }\n    return this._filterPipeline;\n  }\n\n  reserveCapacity() {}\n  private _ensureBuffers(width: number, height: number) {\n    if (width <= this._width && height <= this._height && this._colors.length) return;\n    width = Math.max(width, this._width); height = Math.max(height, this._height);\n    this._destroyBuffers();\n    const pixels = width * height;\n    const createBuffer = (label: string, bytes: number) =>\n      this._device.createBuffer({\n        label,\n        size: pixels * bytes,\n        usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC\n      });\n    this._guides = [createBuffer('JNLM guide A', 8), createBuffer('JNLM guide B', 8)];\n    this._colors = [createBuffer('JNLM color A', 16), createBuffer('JNLM color B', 16)];\n    this._width = width;\n    this._height = height;\n  }\n\n  private _destroyBuffers() {\n    for (const buffer of [...this._guides, ...this._colors]) buffer.destroy();\n    this._guides = [];\n    this._colors = [];\n  }\n\n  /** Encodes the denoiser and returns the RGBA32F output in texture rows. */\n  encode(encoder: GPUCommandEncoder, inputs: JNLMInputs): GPUBuffer {\n    const { width, height, shape } = inputs;\n    this._ensureBuffers(width, height);\n    const iterations = Math.min(Math.max(Math.round(shape.iterations), 1), MAX_ITERATIONS);\n    const sensitivity = JNLM_DEFAULT_COLOR_SENSITIVITY * Math.max(inputs.strength, 1e-3);\n    this._paramsBuffers.forEach((buffer, i) => {\n      const params = new ArrayBuffer(16);\n      new Uint32Array(params, 0, 2).set([width, height]);\n      // Buffer i (from 1) is filter pass i - 1; index 0 is the prepasses.\n      new Float32Array(params, 8, 2).set([sensitivity, jnlmPassDilation(Math.max(i - 1, 0))]);\n      this._device.queue.writeBuffer(buffer, 0, params);\n    });\n\n    const device = this._device;\n    const filterPipeline = this._getFilterPipeline(shape);\n    const bind = (\n      pipeline: GPUComputePipeline,\n      paramsBuffer: GPUBuffer,\n      resources: GPUBindingResource[]\n    ) =>\n      device.createBindGroup({\n        layout: pipeline.getBindGroupLayout(0),\n        entries: [\n          { binding: 0, resource: { buffer: paramsBuffer } },\n          ...resources.map((resource, i) => ({ binding: i + 1, resource }))\n        ]\n      });\n    const [guideA, guideB] = this._guides;\n    const [colorA, colorB] = this._colors;\n    const prepass = this._paramsBuffers[0];\n    const passes: [GPUComputePipeline, GPUBindGroup][] = [\n      [this._guidePipeline, bind(this._guidePipeline, prepass, [\n        { buffer: inputs.accumulation },\n        { buffer: inputs.normalAccumulation },\n        { buffer: guideB }\n      ])],\n      [this._variancePipeline, bind(this._variancePipeline, prepass, [\n        { buffer: guideB },\n        { buffer: guideA }\n      ])],\n      [this._loadColorPipeline, bind(this._loadColorPipeline, prepass, [\n        inputs.color.createView(),\n        { buffer: colorA }\n      ])]\n    ];\n    const albedoView = inputs.albedo.createView();\n    const normalView = inputs.normal.createView();\n    const shadowCatcherView = inputs.shadowCatcher.createView();\n    let guides = [guideA, guideB];\n    let colors = [colorA, colorB];\n    for (let i = 1; i <= iterations; i++) {\n      passes.push([filterPipeline, bind(filterPipeline, this._paramsBuffers[i], [\n        { buffer: guides[0] },\n        { buffer: colors[0] },\n        albedoView,\n        normalView,\n        { buffer: colors[1] },\n        { buffer: guides[1] },\n        shadowCatcherView\n      ])]);\n      guides = [guides[1], guides[0]];\n      colors = [colors[1], colors[0]];\n    }\n    const pass = encoder.beginComputePass({ label: 'JNLM denoise' });\n    for (const [pipeline, bindGroup] of passes) {\n      pass.setPipeline(pipeline);\n      pass.setBindGroup(0, bindGroup);\n      pass.dispatchWorkgroups(\n        Math.ceil(width / JNLM_WORKGROUP_SIZE),\n        Math.ceil(height / JNLM_WORKGROUP_SIZE)\n      );\n    }\n    pass.end();\n    return colors[0];\n  }\n\n  getMemoryBytes() {\n    return [...this._paramsBuffers, ...this._guides, ...this._colors]\n      .reduce((sum, buffer) => sum + buffer.size, 0);\n  }\n\n  dispose() {\n    for (const buffer of this._paramsBuffers) buffer.destroy();\n    this._destroyBuffers();\n  }\n}\n"]}
+{"version":3,"file":"WGPUJNLMDenoiser.js","sourceRoot":"","sources":["../../src/renderer/WGPUJNLMDenoiser.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,oBAAoB,EAAC,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAC,mBAAmB,EAAuB,MAAM,WAAW,CAAC;AACpE,OAAO,EACL,mBAAmB,EAAE,aAAa,EAAE,gBAAgB,EAAE,iBAAiB,EACvE,oBAAoB,EAAE,kBAAkB,EACzC,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAC,oBAAoB,EAAE,aAAa,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,mBAAmB,EAAC,MAAM,wBAAwB,CAAC;AAqBrI,MAAM,cAAc,GAAG,mBAAmB,CAAC;AAE3C,MAAM,OAAO,mBAAmB;IACtB,OAAO,CAAY;IACnB,cAAc,GAAgB,EAAE,CAAC;IACjC,cAAc,CAAqB;IACnC,iBAAiB,CAAqB;IACtC,kBAAkB,CAAqB;IACvC,eAAe,CAAsB;IACrC,eAAe,GAAG,EAAE,CAAC;IACrB,OAAO,GAAgB,EAAE,CAAC;IAC1B,OAAO,GAAgB,EAAE,CAAC;IAC1B,MAAM,GAAG,CAAC,CAAC;IACX,OAAO,GAAG,CAAC,CAAC;IAEpB,YAAY,MAAiB;QAC3B,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,sEAAsE;QACtE,iCAAiC;QACjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,cAAc,EAAE,CAAC,EAAE,EAAE,CAAC;YACzC,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC;gBAC3C,KAAK,EAAE,aAAa;gBACpB,IAAI,EAAE,EAAE;gBACR,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ;aACxD,CAAC,CAAC,CAAC;QACN,CAAC;QACD,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC,YAAY,EAAE,aAAa,CAAC,CAAC;QACxE,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC,eAAe,CAAC,eAAe,EAAE,gBAAgB,CAAC,CAAC;QACjF,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,eAAe,CAAC,iBAAiB,EAAE,iBAAiB,CAAC,CAAC;IACvF,CAAC;IAEO,eAAe,CAAC,KAAa,EAAE,IAAY;QACjD,OAAO,oBAAoB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC;YAC9D,KAAK;YACL,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,MAAM,EAAE,oBAAoB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,kBAAkB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;gBAC9E,UAAU,EAAE,MAAM;aACnB;SACF,CAAC,CAAC;IACL,CAAC;IAEO,kBAAkB,CAAC,KAAsB;QAC/C,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAClC,IAAI,CAAC,IAAI,CAAC,eAAe,IAAI,GAAG,KAAK,IAAI,CAAC,eAAe,EAAE,CAAC;YAC1D,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,eAAe,CAAC,aAAa,EAAE,oBAAoB,CAAC,KAAK,CAAC,CAAC,CAAC;YACxF,IAAI,CAAC,eAAe,GAAG,GAAG,CAAC;QAC7B,CAAC;QACD,OAAO,IAAI,CAAC,eAAe,CAAC;IAC9B,CAAC;IAED,eAAe,KAAI,CAAC;IACZ,cAAc,CAAC,KAAa,EAAE,MAAc;QAClD,IAAI,KAAK,IAAI,IAAI,CAAC,MAAM,IAAI,MAAM,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM;YAAE,OAAO;QAClF,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;QAAC,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAC9E,IAAI,CAAC,eAAe,EAAE,CAAC;QACvB,MAAM,MAAM,GAAG,KAAK,GAAG,MAAM,CAAC;QAC9B,MAAM,YAAY,GAAG,CAAC,KAAa,EAAE,KAAa,EAAE,EAAE,CACpD,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC;YACxB,KAAK;YACL,IAAI,EAAE,MAAM,GAAG,KAAK;YACpB,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ;SACxD,CAAC,CAAC;QACL,IAAI,CAAC,OAAO,GAAG,CAAC,YAAY,CAAC,cAAc,EAAE,CAAC,CAAC,EAAE,YAAY,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,CAAC;QAClF,IAAI,CAAC,OAAO,GAAG,CAAC,YAAY,CAAC,cAAc,EAAE,EAAE,CAAC,EAAE,YAAY,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC,CAAC;QACpF,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;IACxB,CAAC;IAEO,eAAe;QACrB,KAAK,MAAM,MAAM,IAAI,CAAC,GAAG,IAAI,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC;YAAE,MAAM,CAAC,OAAO,EAAE,CAAC;QAC1E,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;QAClB,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;IACpB,CAAC;IAED,2EAA2E;IAC3E,MAAM,CAAC,OAA0B,EAAE,MAAkB;QACnD,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,CAAC;QACxC,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACnC,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,EAAE,cAAc,CAAC,CAAC;QACvF,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;YACxC,oEAAoE;YACpE,MAAM,MAAM,GAAG,kBAAkB,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,QAAQ,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YACtF,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,MAAM,cAAc,GAAG,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;QACtD,MAAM,IAAI,GAAG,CACX,QAA4B,EAC5B,YAAuB,EACvB,SAA+B,EAC/B,EAAE,CACF,MAAM,CAAC,eAAe,CAAC;YACrB,MAAM,EAAE,QAAQ,CAAC,kBAAkB,CAAC,CAAC,CAAC;YACtC,OAAO,EAAE;gBACP,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,YAAY,EAAE,EAAE;gBAClD,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,GAAG,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;aAClE;SACF,CAAC,CAAC;QACL,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC;QACtC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC;QACtC,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;QACvC,MAAM,MAAM,GAAyC;YACnD,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,OAAO,EAAE;oBACvD,EAAE,MAAM,EAAE,MAAM,CAAC,YAAY,EAAE;oBAC/B,EAAE,MAAM,EAAE,MAAM,CAAC,kBAAkB,EAAE;oBACrC,EAAE,MAAM,EAAE,MAAM,EAAE;iBACnB,CAAC,CAAC;YACH,CAAC,IAAI,CAAC,iBAAiB,EAAE,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,OAAO,EAAE;oBAC7D,EAAE,MAAM,EAAE,MAAM,EAAE;oBAClB,EAAE,MAAM,EAAE,MAAM,EAAE;iBACnB,CAAC,CAAC;YACH,CAAC,IAAI,CAAC,kBAAkB,EAAE,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,OAAO,EAAE;oBAC/D,MAAM,CAAC,KAAK,CAAC,UAAU,EAAE;oBACzB,EAAE,MAAM,EAAE,MAAM,EAAE;iBACnB,CAAC,CAAC;SACJ,CAAC;QACF,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;QAC9C,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;QAC9C,MAAM,iBAAiB,GAAG,MAAM,CAAC,aAAa,CAAC,UAAU,EAAE,CAAC;QAC5D,IAAI,MAAM,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC9B,IAAI,MAAM,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC9B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;YACrC,MAAM,CAAC,IAAI,CAAC,CAAC,cAAc,EAAE,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE;oBACxE,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE;oBACrB,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE;oBACrB,UAAU;oBACV,UAAU;oBACV,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE;oBACrB,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE;oBACrB,iBAAiB;iBAClB,CAAC,CAAC,CAAC,CAAC;YACL,MAAM,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;YAChC,MAAM,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QAClC,CAAC;QACD,MAAM,IAAI,GAAG,OAAO,CAAC,gBAAgB,CAAC,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC,CAAC;QACjE,KAAK,MAAM,CAAC,QAAQ,EAAE,SAAS,CAAC,IAAI,MAAM,EAAE,CAAC;YAC3C,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;YAC3B,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;YAChC,IAAI,CAAC,kBAAkB,CACrB,IAAI,CAAC,IAAI,CAAC,KAAK,GAAG,mBAAmB,CAAC,EACtC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,mBAAmB,CAAC,CACxC,CAAC;QACJ,CAAC;QACD,IAAI,CAAC,GAAG,EAAE,CAAC;QACX,OAAO,MAAM,CAAC,CAAC,CAAC,CAAC;IACnB,CAAC;IAED,cAAc;QACZ,OAAO,CAAC,GAAG,IAAI,CAAC,cAAc,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC;aAC9D,MAAM,CAAC,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,GAAG,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IACnD,CAAC;IAED,OAAO;QACL,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,cAAc;YAAE,MAAM,CAAC,OAAO,EAAE,CAAC;QAC3D,IAAI,CAAC,eAAe,EAAE,CAAC;IACzB,CAAC;CACF","sourcesContent":["import {getTargetShaderCache} from './WGPUShaderCache.js';\nimport {JNLM_MAX_ITERATIONS, type JNLMFilterShape} from './jnlm.js';\nimport {\n  JNLM_WORKGROUP_SIZE, jnlmGuideWGSL, jnlmVarianceWGSL, jnlmLoadColorWGSL,\n  createJNLMFilterWGSL, packJNLMParameters\n} from './wgsl/jnlm.compute.js';\nexport {createJNLMFilterWGSL, jnlmGuideWGSL, jnlmVarianceWGSL, jnlmLoadColorWGSL, JNLM_WORKGROUP_SIZE} from './wgsl/jnlm.compute.js';\n\n/** WebGPU passes of the JNLM denoiser described in `jnlm.ts`. */\n\nexport interface JNLMInputs {\n  width: number;\n  height: number;\n  /** Normalized shadow-catcher mask; red is the catcher coverage. */\n  shadowCatcher: GPUTexture;\n  /** Ray-trace auxiliary accumulation holding the guide moments. */\n  accumulation: GPUBuffer;\n  /** Ray-trace normal accumulation; its alpha is the per-pixel sample count. */\n  normalAccumulation: GPUBuffer;\n  /** Composed linear image to filter. */\n  color: GPUTexture;\n  albedo: GPUTexture;\n  normal: GPUTexture;\n  strength: number;\n  shape: JNLMFilterShape;\n}\n\nconst MAX_ITERATIONS = JNLM_MAX_ITERATIONS;\n\nexport class WGPUJNLMDenoisePass {\n  private _device: GPUDevice;\n  private _paramsBuffers: GPUBuffer[] = [];\n  private _guidePipeline: GPUComputePipeline;\n  private _variancePipeline: GPUComputePipeline;\n  private _loadColorPipeline: GPUComputePipeline;\n  private _filterPipeline?: GPUComputePipeline;\n  private _filterShapeKey = '';\n  private _guides: GPUBuffer[] = [];\n  private _colors: GPUBuffer[] = [];\n  private _width = 0;\n  private _height = 0;\n\n  constructor(device: GPUDevice) {\n    this._device = device;\n    // One uniform buffer per pass: every pass of an encoder sees the last\n    // writeBuffer() of a shared one.\n    for (let i = 0; i <= MAX_ITERATIONS; i++) {\n      this._paramsBuffers.push(device.createBuffer({\n        label: 'JNLM params',\n        size: 16,\n        usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST\n      }));\n    }\n    this._guidePipeline = this._createPipeline('JNLM guide', jnlmGuideWGSL);\n    this._variancePipeline = this._createPipeline('JNLM variance', jnlmVarianceWGSL);\n    this._loadColorPipeline = this._createPipeline('JNLM load color', jnlmLoadColorWGSL);\n  }\n\n  private _createPipeline(label: string, code: string) {\n    return getTargetShaderCache(this._device).createComputePipeline({\n      label,\n      layout: 'auto',\n      compute: {\n        module: getTargetShaderCache(this._device).createShaderModule({ label, code }),\n        entryPoint: 'main'\n      }\n    });\n  }\n\n  private _getFilterPipeline(shape: JNLMFilterShape) {\n    const key = JSON.stringify(shape);\n    if (!this._filterPipeline || key !== this._filterShapeKey) {\n      this._filterPipeline = this._createPipeline('JNLM filter', createJNLMFilterWGSL(shape));\n      this._filterShapeKey = key;\n    }\n    return this._filterPipeline;\n  }\n\n  reserveCapacity() {}\n  private _ensureBuffers(width: number, height: number) {\n    if (width <= this._width && height <= this._height && this._colors.length) return;\n    width = Math.max(width, this._width); height = Math.max(height, this._height);\n    this._destroyBuffers();\n    const pixels = width * height;\n    const createBuffer = (label: string, bytes: number) =>\n      this._device.createBuffer({\n        label,\n        size: pixels * bytes,\n        usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC\n      });\n    this._guides = [createBuffer('JNLM guide A', 8), createBuffer('JNLM guide B', 8)];\n    this._colors = [createBuffer('JNLM color A', 16), createBuffer('JNLM color B', 16)];\n    this._width = width;\n    this._height = height;\n  }\n\n  private _destroyBuffers() {\n    for (const buffer of [...this._guides, ...this._colors]) buffer.destroy();\n    this._guides = [];\n    this._colors = [];\n  }\n\n  /** Encodes the denoiser and returns the RGBA32F output in texture rows. */\n  encode(encoder: GPUCommandEncoder, inputs: JNLMInputs): GPUBuffer {\n    const { width, height, shape } = inputs;\n    this._ensureBuffers(width, height);\n    const iterations = Math.min(Math.max(Math.round(shape.iterations), 1), MAX_ITERATIONS);\n    this._paramsBuffers.forEach((buffer, i) => {\n      // Buffer i (from 1) is filter pass i - 1; index 0 is the prepasses.\n      const params = packJNLMParameters(width, height, inputs.strength, Math.max(i - 1, 0));\n      this._device.queue.writeBuffer(buffer, 0, params);\n    });\n\n    const device = this._device;\n    const filterPipeline = this._getFilterPipeline(shape);\n    const bind = (\n      pipeline: GPUComputePipeline,\n      paramsBuffer: GPUBuffer,\n      resources: GPUBindingResource[]\n    ) =>\n      device.createBindGroup({\n        layout: pipeline.getBindGroupLayout(0),\n        entries: [\n          { binding: 0, resource: { buffer: paramsBuffer } },\n          ...resources.map((resource, i) => ({ binding: i + 1, resource }))\n        ]\n      });\n    const [guideA, guideB] = this._guides;\n    const [colorA, colorB] = this._colors;\n    const prepass = this._paramsBuffers[0];\n    const passes: [GPUComputePipeline, GPUBindGroup][] = [\n      [this._guidePipeline, bind(this._guidePipeline, prepass, [\n        { buffer: inputs.accumulation },\n        { buffer: inputs.normalAccumulation },\n        { buffer: guideB }\n      ])],\n      [this._variancePipeline, bind(this._variancePipeline, prepass, [\n        { buffer: guideB },\n        { buffer: guideA }\n      ])],\n      [this._loadColorPipeline, bind(this._loadColorPipeline, prepass, [\n        inputs.color.createView(),\n        { buffer: colorA }\n      ])]\n    ];\n    const albedoView = inputs.albedo.createView();\n    const normalView = inputs.normal.createView();\n    const shadowCatcherView = inputs.shadowCatcher.createView();\n    let guides = [guideA, guideB];\n    let colors = [colorA, colorB];\n    for (let i = 1; i <= iterations; i++) {\n      passes.push([filterPipeline, bind(filterPipeline, this._paramsBuffers[i], [\n        { buffer: guides[0] },\n        { buffer: colors[0] },\n        albedoView,\n        normalView,\n        { buffer: colors[1] },\n        { buffer: guides[1] },\n        shadowCatcherView\n      ])]);\n      guides = [guides[1], guides[0]];\n      colors = [colors[1], colors[0]];\n    }\n    const pass = encoder.beginComputePass({ label: 'JNLM denoise' });\n    for (const [pipeline, bindGroup] of passes) {\n      pass.setPipeline(pipeline);\n      pass.setBindGroup(0, bindGroup);\n      pass.dispatchWorkgroups(\n        Math.ceil(width / JNLM_WORKGROUP_SIZE),\n        Math.ceil(height / JNLM_WORKGROUP_SIZE)\n      );\n    }\n    pass.end();\n    return colors[0];\n  }\n\n  getMemoryBytes() {\n    return [...this._paramsBuffers, ...this._guides, ...this._colors]\n      .reduce((sum, buffer) => sum + buffer.size, 0);\n  }\n\n  dispose() {\n    for (const buffer of this._paramsBuffers) buffer.destroy();\n    this._destroyBuffers();\n  }\n}\n"]}

package/lib/renderer/WGPUPhotonMap.d.ts

@@ -0,0 +1,37 @@
+import { WGPUCausticLightPass } from './WGPUCausticLightPass.js';
+export { photonRadius } from './wgsl/photonMap.compute.js';
+interface Batch {
+    grid: GPUBuffer;
+    uniform: GPUBuffer;
+    data: GPUTexture;
+    headers: GPUTexture;
+    count: number;
+    buckets: number;
+    perCasterRadius: boolean;
+    group: GPUBindGroup;
+}
+/** Batch-local spatial hash; accumulated estimates live in the camera path target. */
+export declare class WGPUPhotonMap {
+    private readonly device;
+    readonly placeholder: GPUTexture;
+    readonly emission: WGPUCausticLightPass;
+    private readonly targets;
+    private readonly layout;
+    private readonly pipelines;
+    constructor(device: GPUDevice);
+    ensure(target: object, requested: number): Batch;
+    get(target: object): Batch | undefined;
+    encode(encoder: GPUCommandEncoder, groups: GPUBindGroup[], target: object, radius: number, options: Parameters<WGPUCausticLightPass['encode']>[3] & {
+        progressiveBatch?: number;
+    }): void;
+    release(target: object): void;
+    bytes(target: object): number;
+    dispose(): void;
+}
+/** Typical first-receiver footprint at the caster center; query disks stay exact. */
+export declare function photonVisibleRadius(radius: number, options: {
+    viewProjection: ArrayLike<number>;
+    casterSphere: readonly number[];
+    width: number;
+    orthographic?: boolean;
+}, screenPixels?: number): number;

package/lib/renderer/WGPUPhotonMap.js

@@ -0,0 +1,191 @@
+import { causticDiagnostics } from './causticDiagnostics.js';
+import { WGPUCausticLightPass } from './WGPUCausticLightPass.js';
+import { PHOTON_BYTES, PHOTON_GRID_SIZE, photonLayout, photonGridCompute } from './wgsl/photonMap.compute.js';
+export { photonRadius } from './wgsl/photonMap.compute.js';
+/** Batch-local spatial hash; accumulated estimates live in the camera path target. */
+export class WGPUPhotonMap {
+    device;
+    placeholder;
+    emission;
+    targets = new Map();
+    layout;
+    pipelines;
+    constructor(device) {
+        this.device = device;
+        this.placeholder = device.createTexture({
+            label: 'Inactive photon binding',
+            size: [1, 1],
+            format: 'r32uint',
+            usage: GPUTextureUsage.TEXTURE_BINDING
+        });
+        this.emission = new WGPUCausticLightPass(device, true);
+        this.layout = device.createBindGroupLayout({
+            entries: [
+                { binding: 0, visibility: 4, texture: { sampleType: 'uint' } },
+                { binding: 1, visibility: 4, buffer: { type: 'storage' } },
+                { binding: 2, visibility: 4, buffer: { type: 'uniform' } },
+                { binding: 4, visibility: 4, storageTexture: { access: 'write-only', format: 'r32uint' } }
+            ]
+        });
+        const module = device.createShaderModule({ label: 'Photon hash grid', code: photonGridCompute }), layout = device.createPipelineLayout({ bindGroupLayouts: [this.layout] });
+        this.pipelines = ['count', 'scanBlocks', 'scanSums', 'scatter'].map((entryPoint) => device.createComputePipeline({ label: 'Photon ' + entryPoint, layout, compute: { module, entryPoint } }));
+    }
+    ensure(target, requested) {
+        const device = this.device, limit = device.limits.maxTextureDimension2D;
+        const budget = causticDiagnostics.photonBudgetBytes ?? 128 * 1024 * 1024;
+        const perCasterRadius = !!causticDiagnostics.photonPerCasterRadius;
+        const bytesPerPhoton = perCasterRadius ? 32 : 28;
+        let count = Math.max(1, Math.min(Math.round(requested), Math.floor(device.limits.maxStorageBufferBindingSize / PHOTON_BYTES), Math.floor((limit * limit - PHOTON_GRID_SIZE * 4 - 8) / (perCasterRadius ? 4 : 3)), Math.floor(((causticDiagnostics.photonBudgetBytes ?? 128 * 1024 * 1024) - 13 * 1024 * 1024) / bytesPerPhoton)));
+        // Include row padding instead of assuming every texture is tightly packed.
+        const allocationBytes = (n) => {
+            const { buckets, texels } = photonLayout(n, perCasterRadius);
+            const capacity = (texels) => Math.min(texels, limit) * Math.ceil(texels / limit);
+            return (buckets + buckets / 256) * 32 + capacity(n) * 16 + capacity(texels) * 4 + 32;
+        };
+        while (count > 1 && allocationBytes(count) > budget) {
+            count = Math.max(1, count - Math.ceil((allocationBytes(count) - budget) / bytesPerPhoton));
+        }
+        const existing = this.targets.get(target);
+        if (existing?.count === count && existing.perCasterRadius === perCasterRadius)
+            return existing;
+        this.release(target);
+        const { buckets, texels: headerTexels } = photonLayout(count, perCasterRadius), texels = count, width = Math.min(texels, limit), height = Math.ceil(texels / width);
+        // One rgba32uint texel holds position and flux. Direction words live after
+        // both sorted indices in the integer grid texture: 20 B payload + 8 B indices.
+        const batch = {
+            count,
+            buckets,
+            perCasterRadius,
+            grid: device.createBuffer({
+                label: 'Photon hash buckets',
+                size: (buckets + buckets / 256) * 32,
+                usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_DST
+            }),
+            uniform: device.createBuffer({
+                label: 'Photon grid parameters',
+                size: 32,
+                usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST
+            }),
+            data: device.createTexture({
+                label: 'Packed photon positions and flux',
+                size: [width, height],
+                format: 'rgba32uint',
+                usage: GPUTextureUsage.STORAGE_BINDING | GPUTextureUsage.TEXTURE_BINDING | GPUTextureUsage.COPY_SRC
+            }),
+            headers: device.createTexture({
+                label: 'Photon grid headers',
+                size: [
+                    Math.min(headerTexels, limit),
+                    Math.ceil((headerTexels) / limit)
+                ],
+                format: 'r32uint',
+                usage: GPUTextureUsage.STORAGE_BINDING | GPUTextureUsage.TEXTURE_BINDING | GPUTextureUsage.COPY_SRC
+            })
+        };
+        batch.group = this.device.createBindGroup({
+            layout: this.layout,
+            entries: [
+                { binding: 0, resource: batch.data.createView() },
+                { binding: 1, resource: { buffer: batch.grid } },
+                { binding: 2, resource: { buffer: batch.uniform } },
+                { binding: 4, resource: batch.headers.createView() }
+            ]
+        });
+        this.targets.set(target, batch);
+        return batch;
+    }
+    get(target) {
+        return this.targets.get(target);
+    }
+    encode(encoder, groups, target, radius, options) {
+        const batch = this.targets.get(target);
+        if (causticDiagnostics.capturePhotons)
+            causticDiagnostics.photonBatch = {
+                device: this.device,
+                data: batch.headers,
+                emitted: batch.count,
+                packed: true
+            };
+        encoder.clearBuffer(batch.grid);
+        this.emission.emit(encoder, groups, batch.data, {
+            ...options,
+            photons: batch.count,
+            photonMapping: true,
+            photonDirections: batch.headers
+        });
+        const bytes = new ArrayBuffer(32);
+        new Float32Array(bytes)[0] = radius;
+        new Uint32Array(bytes)[1] = batch.count;
+        new Float32Array(bytes)[2] = causticDiagnostics.photonScreenRadius === false ? 0 : 2;
+        new Float32Array(bytes)[3] = causticDiagnostics.photonSkipQuery
+            ? 1
+            : causticDiagnostics.photonSkipVisible
+                ? 2
+                : 0;
+        new Uint32Array(bytes)[4] = batch.buckets;
+        new Float32Array(bytes)[5] = photonVisibleRadius(radius, options, new Float32Array(bytes)[2]);
+        new Uint32Array(bytes)[6] = (+batch.perCasterRadius) |
+            (Math.min(65535, Math.max(0, Math.round(causticDiagnostics.photonDeepScreenPixels ?? 0))) << 8);
+        this.device.queue.writeBuffer(batch.uniform, 0, bytes);
+        const group = batch.group;
+        // Dispatch boundaries synchronize storage accesses. Offline batches share one
+        // compute pass; timestamp profiling keeps separate passes for stage timings.
+        const combined = options.querySet ? undefined : encoder.beginComputePass({ label: 'Photon grid' });
+        this.pipelines.forEach((pipeline, index) => {
+            const pass = combined ??
+                encoder.beginComputePass({
+                    label: ['Photon count', 'Photon scan blocks', 'Photon scan sums', 'Photon scatter'][index],
+                    timestampWrites: {
+                        querySet: options.querySet,
+                        beginningOfPassWriteIndex: 8 + index * 2,
+                        endOfPassWriteIndex: 9 + index * 2
+                    }
+                });
+            pass.setPipeline(pipeline);
+            pass.setBindGroup(0, group);
+            const groups = index === 1
+                ? batch.buckets / 256
+                : index === 2
+                    ? 1
+                    : Math.ceil((index === 3 ? Math.max(batch.count, batch.buckets) : batch.count) / 64);
+            pass.dispatchWorkgroups(Math.min(groups, 65535), index === 1 || index === 2 ? 2 : Math.ceil(groups / 65535));
+            if (!combined)
+                pass.end();
+        });
+        combined?.end();
+    }
+    release(target) {
+        const b = this.targets.get(target);
+        if (!b)
+            return;
+        b.grid.destroy();
+        b.uniform.destroy();
+        b.data.destroy();
+        b.headers.destroy();
+        this.targets.delete(target);
+    }
+    bytes(target) {
+        const b = this.targets.get(target);
+        return b
+            ? b.grid.size +
+                b.uniform.size +
+                b.data.width * b.data.height * 16 +
+                b.headers.width * b.headers.height * 4
+            : 0;
+    }
+    dispose() {
+        for (const target of this.targets.keys())
+            this.release(target);
+        this.emission.dispose();
+        this.placeholder.destroy();
+    }
+}
+/** Typical first-receiver footprint at the caster center; query disks stay exact. */
+export function photonVisibleRadius(radius, options, screenPixels = 2) {
+    const m = options.viewProjection, p = options.casterSphere;
+    const depth = options.orthographic ? 1 : Math.abs(m[3] * p[0] + m[7] * p[1] + m[11] * p[2] + m[15]);
+    const projection = Math.hypot(m[0], m[4], m[8]);
+    const footprint = 2 * depth / (options.width * projection) * screenPixels;
+    return screenPixels > 0 && Number.isFinite(footprint) && footprint > 0 ? Math.min(radius, Math.max(1e-8, footprint)) : radius;
+}
+//# sourceMappingURL=WGPUPhotonMap.js.map

package/lib/renderer/WGPUPhotonMap.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"WGPUPhotonMap.js","sourceRoot":"","sources":["../../src/renderer/WGPUPhotonMap.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,kBAAkB,EAAC,MAAM,yBAAyB,CAAC;AAC3D,OAAO,EAAC,oBAAoB,EAAC,MAAM,2BAA2B,CAAC;AAC/D,OAAO,EAAC,YAAY,EAAE,gBAAgB,EAAE,YAAY,EAAE,iBAAiB,EAAC,MAAM,6BAA6B,CAAC;AAC5G,OAAO,EAAC,YAAY,EAAC,MAAM,6BAA6B,CAAC;AAWzD,sFAAsF;AACtF,MAAM,OAAO,aAAa;IAMK;IALpB,WAAW,CAAa;IACxB,QAAQ,CAAuB;IACvB,OAAO,GAAG,IAAI,GAAG,EAAiB,CAAC;IACnC,MAAM,CAAqB;IAC3B,SAAS,CAAuB;IACjD,YAA6B,MAAiB;QAAjB,WAAM,GAAN,MAAM,CAAW;QAC5C,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,aAAa,CAAC;YACtC,KAAK,EAAE,yBAAyB;YAChC,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;YACZ,MAAM,EAAE,SAAS;YACjB,KAAK,EAAE,eAAe,CAAC,eAAe;SACvC,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,GAAG,IAAI,oBAAoB,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QACvD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,qBAAqB,CAAC;YACzC,OAAO,EAAE;gBACP,EAAC,OAAO,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,OAAO,EAAE,EAAC,UAAU,EAAE,MAAM,EAAC,EAAC;gBAC1D,EAAC,OAAO,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,MAAM,EAAE,EAAC,IAAI,EAAE,SAAS,EAAC,EAAC;gBACtD,EAAC,OAAO,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,MAAM,EAAE,EAAC,IAAI,EAAE,SAAS,EAAC,EAAC;gBACtD,EAAC,OAAO,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,cAAc,EAAE,EAAC,MAAM,EAAE,YAAY,EAAE,MAAM,EAAE,SAAS,EAAC,EAAC;aACvF;SACF,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,CAAC,kBAAkB,CAAC,EAAC,KAAK,EAAE,kBAAkB,EAAE,IAAI,EAAE,iBAAiB,EAAC,CAAC,EAC5F,MAAM,GAAG,MAAM,CAAC,oBAAoB,CAAC,EAAC,gBAAgB,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,EAAC,CAAC,CAAC;QAC1E,IAAI,CAAC,SAAS,GAAG,CAAC,OAAO,EAAE,YAAY,EAAE,UAAU,EAAE,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CACjF,MAAM,CAAC,qBAAqB,CAAC,EAAC,KAAK,EAAE,SAAS,GAAG,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,EAAC,MAAM,EAAE,UAAU,EAAC,EAAC,CAAC,CACrG,CAAC;IACJ,CAAC;IACD,MAAM,CAAC,MAAc,EAAE,SAAiB;QACtC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,EACxB,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,qBAAqB,CAAC;QAC9C,MAAM,MAAM,GAAG,kBAAkB,CAAC,iBAAiB,IAAI,GAAG,GAAG,IAAI,GAAG,IAAI,CAAC;QACzE,MAAM,eAAe,GAAG,CAAC,CAAC,kBAAkB,CAAC,qBAAqB,CAAC;QACnE,MAAM,cAAc,GAAG,eAAe,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACjD,IAAI,KAAK,GAAG,IAAI,CAAC,GAAG,CAClB,CAAC,EACD,IAAI,CAAC,GAAG,CACN,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,EACrB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,2BAA2B,GAAG,YAAY,CAAC,EACpE,IAAI,CAAC,KAAK,CAAC,CAAC,KAAK,GAAG,KAAK,GAAG,gBAAgB,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAClF,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,kBAAkB,CAAC,iBAAiB,IAAI,GAAG,GAAG,IAAI,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,GAAG,IAAI,CAAC,GAAG,cAAc,CAAC,CAC9G,CACF,CAAC;QACF,2EAA2E;QAC3E,MAAM,eAAe,GAAG,CAAC,CAAS,EAAE,EAAE;YACpC,MAAM,EAAC,OAAO,EAAE,MAAM,EAAC,GAAG,YAAY,CAAC,CAAC,EAAE,eAAe,CAAC,CAAC;YAC3D,MAAM,QAAQ,GAAG,CAAC,MAAc,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC,CAAC;YACzF,OAAO,CAAC,OAAO,GAAG,OAAO,GAAG,GAAG,CAAC,GAAG,EAAE,GAAG,QAAQ,CAAC,CAAC,CAAC,GAAG,EAAE,GAAG,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC;QACvF,CAAC,CAAC;QACF,OAAO,KAAK,GAAG,CAAC,IAAI,eAAe,CAAC,KAAK,CAAC,GAAG,MAAM,EAAE,CAAC;YACpD,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC,GAAG,cAAc,CAAC,CAAC,CAAC;QAC7F,CAAC;QACD,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC1C,IAAI,QAAQ,EAAE,KAAK,KAAK,KAAK,IAAI,QAAQ,CAAC,eAAe,KAAK,eAAe;YAAE,OAAO,QAAQ,CAAC;QAC/F,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACrB,MAAM,EAAC,OAAO,EAAE,MAAM,EAAE,YAAY,EAAC,GAAG,YAAY,CAAC,KAAK,EAAE,eAAe,CAAC,EAC1E,MAAM,GAAG,KAAK,EACd,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,EAC/B,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC,CAAC;QACrC,2EAA2E;QAC3E,+EAA+E;QAC/E,MAAM,KAAK,GAAG;YACZ,KAAK;YACL,OAAO;YACP,eAAe;YACf,IAAI,EAAE,MAAM,CAAC,YAAY,CAAC;gBACxB,KAAK,EAAE,qBAAqB;gBAC5B,IAAI,EAAE,CAAC,OAAO,GAAG,OAAO,GAAG,GAAG,CAAC,GAAG,EAAE;gBACpC,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ;aACxD,CAAC;YACF,OAAO,EAAE,MAAM,CAAC,YAAY,CAAC;gBAC3B,KAAK,EAAE,wBAAwB;gBAC/B,IAAI,EAAE,EAAE;gBACR,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ;aACxD,CAAC;YACF,IAAI,EAAE,MAAM,CAAC,aAAa,CAAC;gBACzB,KAAK,EAAE,kCAAkC;gBACzC,IAAI,EAAE,CAAC,KAAK,EAAE,MAAM,CAAC;gBACrB,MAAM,EAAE,YAAY;gBACpB,KAAK,EAAE,eAAe,CAAC,eAAe,GAAG,eAAe,CAAC,eAAe,GAAG,eAAe,CAAC,QAAQ;aACpG,CAAC;YACF,OAAO,EAAE,MAAM,CAAC,aAAa,CAAC;gBAC5B,KAAK,EAAE,qBAAqB;gBAC5B,IAAI,EAAE;oBACJ,IAAI,CAAC,GAAG,CAAC,YAAY,EAAE,KAAK,CAAC;oBAC7B,IAAI,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC,GAAG,KAAK,CAAC;iBAClC;gBACD,MAAM,EAAE,SAAS;gBACjB,KAAK,EAAE,eAAe,CAAC,eAAe,GAAG,eAAe,CAAC,eAAe,GAAG,eAAe,CAAC,QAAQ;aACpG,CAAC;SACM,CAAC;QACX,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC;YACxC,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,OAAO,EAAE;gBACP,EAAC,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,KAAK,CAAC,IAAI,CAAC,UAAU,EAAE,EAAC;gBAC/C,EAAC,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAC,MAAM,EAAE,KAAK,CAAC,IAAI,EAAC,EAAC;gBAC5C,EAAC,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAC,MAAM,EAAE,KAAK,CAAC,OAAO,EAAC,EAAC;gBAC/C,EAAC,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,KAAK,CAAC,OAAO,CAAC,UAAU,EAAE,EAAC;aACnD;SACF,CAAC,CAAC;QACH,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QAChC,OAAO,KAAK,CAAC;IACf,CAAC;IACD,GAAG,CAAC,MAAc;QAChB,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,MAAM,CACJ,OAA0B,EAC1B,MAAsB,EACtB,MAAc,EACd,MAAc,EACd,OAAoF;QAEpF,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAE,CAAC;QACxC,IAAI,kBAAkB,CAAC,cAAc;YACnC,kBAAkB,CAAC,WAAW,GAAG;gBAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,IAAI,EAAE,KAAK,CAAC,OAAO;gBACnB,OAAO,EAAE,KAAK,CAAC,KAAK;gBACpB,MAAM,EAAE,IAAI;aACb,CAAC;QACJ,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAChC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,CAAC,IAAI,EAAE;YAC9C,GAAG,OAAO;YACV,OAAO,EAAE,KAAK,CAAC,KAAK;YACpB,aAAa,EAAE,IAAI;YACnB,gBAAgB,EAAE,KAAK,CAAC,OAAO;SAChC,CAAC,CAAC;QACH,MAAM,KAAK,GAAG,IAAI,WAAW,CAAC,EAAE,CAAC,CAAC;QAClC,IAAI,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC;QACpC,IAAI,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC;QACxC,IAAI,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,kBAAkB,CAAC,kBAAkB,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACrF,IAAI,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,kBAAkB,CAAC,eAAe;YAC7D,CAAC,CAAC,CAAC;YACH,CAAC,CAAC,kBAAkB,CAAC,iBAAiB;gBACpC,CAAC,CAAC,CAAC;gBACH,CAAC,CAAC,CAAC,CAAC;QACR,IAAI,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,OAAO,CAAC;QAC1C,IAAI,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,mBAAmB,CAAC,MAAM,EAAE,OAAO,EAAE,IAAI,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9F,IAAI,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,eAAe,CAAC;YAClD,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,sBAAsB,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QAClG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC,EAAE,KAAK,CAAC,CAAC;QACvD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;QAC1B,8EAA8E;QAC9E,6EAA6E;QAC7E,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC,EAAC,KAAK,EAAE,aAAa,EAAC,CAAC,CAAC;QACjG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,KAAK,EAAE,EAAE;YACzC,MAAM,IAAI,GACR,QAAQ;gBACR,OAAO,CAAC,gBAAgB,CAAC;oBACvB,KAAK,EAAE,CAAC,cAAc,EAAE,oBAAoB,EAAE,kBAAkB,EAAE,gBAAgB,CAAC,CAAC,KAAK,CAAC;oBAC1F,eAAe,EAAE;wBACf,QAAQ,EAAE,OAAO,CAAC,QAAS;wBAC3B,yBAAyB,EAAE,CAAC,GAAG,KAAK,GAAG,CAAC;wBACxC,mBAAmB,EAAE,CAAC,GAAG,KAAK,GAAG,CAAC;qBACnC;iBACF,CAAC,CAAC;YACL,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;YAC3B,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;YAC5B,MAAM,MAAM,GACV,KAAK,KAAK,CAAC;gBACT,CAAC,CAAC,KAAK,CAAC,OAAO,GAAG,GAAG;gBACrB,CAAC,CAAC,KAAK,KAAK,CAAC;oBACX,CAAC,CAAC,CAAC;oBACH,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC;YAC3F,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,EAAE,KAAK,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC,CAAC,CAAC;YAC7G,IAAI,CAAC,QAAQ;gBAAE,IAAI,CAAC,GAAG,EAAE,CAAC;QAC5B,CAAC,CAAC,CAAC;QACH,QAAQ,EAAE,GAAG,EAAE,CAAC;IAClB,CAAC;IACD,OAAO,CAAC,MAAc;QACpB,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACnC,IAAI,CAAC,CAAC;YAAE,OAAO;QACf,CAAC,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;QACjB,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;QACpB,CAAC,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;QACjB,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;QACpB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAC9B,CAAC;IACD,KAAK,CAAC,MAAc;QAClB,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACnC,OAAO,CAAC;YACN,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI;gBACT,CAAC,CAAC,OAAO,CAAC,IAAI;gBACd,CAAC,CAAC,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC,IAAI,CAAC,MAAM,GAAG,EAAE;gBACjC,CAAC,CAAC,OAAO,CAAC,KAAK,GAAG,CAAC,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC;YAC1C,CAAC,CAAC,CAAC,CAAC;IACR,CAAC;IACD,OAAO;QACL,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE;YAAE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC/D,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;QACxB,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;IAC7B,CAAC;CACF;AACD,qFAAqF;AACrF,MAAM,UAAU,mBAAmB,CAAC,MAAc,EAAE,OAAoH,EAAE,YAAY,GAAG,CAAC;IACxL,MAAM,CAAC,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC,GAAG,OAAO,CAAC,YAAY,CAAC;IAC3D,MAAM,KAAK,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,GAAC,CAAC,CAAC,CAAC,CAAC,GAAC,CAAC,CAAC,CAAC,CAAC,GAAC,CAAC,CAAC,CAAC,CAAC,GAAC,CAAC,CAAC,EAAE,CAAC,GAAC,CAAC,CAAC,CAAC,CAAC,GAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACxF,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAChD,MAAM,SAAS,GAAG,CAAC,GAAG,KAAK,GAAG,CAAC,OAAO,CAAC,KAAK,GAAG,UAAU,CAAC,GAAG,YAAY,CAAC;IAC1E,OAAO,YAAY,GAAG,CAAC,IAAI,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;AAChI,CAAC","sourcesContent":["import {causticDiagnostics} from './causticDiagnostics.js';\nimport {WGPUCausticLightPass} from './WGPUCausticLightPass.js';\nimport {PHOTON_BYTES, PHOTON_GRID_SIZE, photonLayout, photonGridCompute} from './wgsl/photonMap.compute.js';\nexport {photonRadius} from './wgsl/photonMap.compute.js';\ninterface Batch {\n  grid: GPUBuffer;\n  uniform: GPUBuffer;\n  data: GPUTexture;\n  headers: GPUTexture;\n  count: number;\n  buckets: number;\n  perCasterRadius: boolean;\n  group: GPUBindGroup;\n}\n/** Batch-local spatial hash; accumulated estimates live in the camera path target. */\nexport class WGPUPhotonMap {\n  readonly placeholder: GPUTexture;\n  readonly emission: WGPUCausticLightPass;\n  private readonly targets = new Map<object, Batch>();\n  private readonly layout: GPUBindGroupLayout;\n  private readonly pipelines: GPUComputePipeline[];\n  constructor(private readonly device: GPUDevice) {\n    this.placeholder = device.createTexture({\n      label: 'Inactive photon binding',\n      size: [1, 1],\n      format: 'r32uint',\n      usage: GPUTextureUsage.TEXTURE_BINDING\n    });\n    this.emission = new WGPUCausticLightPass(device, true);\n    this.layout = device.createBindGroupLayout({\n      entries: [\n        {binding: 0, visibility: 4, texture: {sampleType: 'uint'}},\n        {binding: 1, visibility: 4, buffer: {type: 'storage'}},\n        {binding: 2, visibility: 4, buffer: {type: 'uniform'}},\n        {binding: 4, visibility: 4, storageTexture: {access: 'write-only', format: 'r32uint'}}\n      ]\n    });\n    const module = device.createShaderModule({label: 'Photon hash grid', code: photonGridCompute}),\n      layout = device.createPipelineLayout({bindGroupLayouts: [this.layout]});\n    this.pipelines = ['count', 'scanBlocks', 'scanSums', 'scatter'].map((entryPoint) =>\n      device.createComputePipeline({label: 'Photon ' + entryPoint, layout, compute: {module, entryPoint}})\n    );\n  }\n  ensure(target: object, requested: number) {\n    const device = this.device,\n      limit = device.limits.maxTextureDimension2D;\n    const budget = causticDiagnostics.photonBudgetBytes ?? 128 * 1024 * 1024;\n    const perCasterRadius = !!causticDiagnostics.photonPerCasterRadius;\n    const bytesPerPhoton = perCasterRadius ? 32 : 28;\n    let count = Math.max(\n      1,\n      Math.min(\n        Math.round(requested),\n        Math.floor(device.limits.maxStorageBufferBindingSize / PHOTON_BYTES),\n        Math.floor((limit * limit - PHOTON_GRID_SIZE * 4 - 8) / (perCasterRadius ? 4 : 3)),\n        Math.floor(((causticDiagnostics.photonBudgetBytes ?? 128 * 1024 * 1024) - 13 * 1024 * 1024) / bytesPerPhoton)\n      )\n    );\n    // Include row padding instead of assuming every texture is tightly packed.\n    const allocationBytes = (n: number) => {\n      const {buckets, texels} = photonLayout(n, perCasterRadius);\n      const capacity = (texels: number) => Math.min(texels, limit) * Math.ceil(texels / limit);\n      return (buckets + buckets / 256) * 32 + capacity(n) * 16 + capacity(texels) * 4 + 32;\n    };\n    while (count > 1 && allocationBytes(count) > budget) {\n      count = Math.max(1, count - Math.ceil((allocationBytes(count) - budget) / bytesPerPhoton));\n    }\n    const existing = this.targets.get(target);\n    if (existing?.count === count && existing.perCasterRadius === perCasterRadius) return existing;\n    this.release(target);\n    const {buckets, texels: headerTexels} = photonLayout(count, perCasterRadius),\n      texels = count,\n      width = Math.min(texels, limit),\n      height = Math.ceil(texels / width);\n    // One rgba32uint texel holds position and flux. Direction words live after\n    // both sorted indices in the integer grid texture: 20 B payload + 8 B indices.\n    const batch = {\n      count,\n      buckets,\n      perCasterRadius,\n      grid: device.createBuffer({\n        label: 'Photon hash buckets',\n        size: (buckets + buckets / 256) * 32,\n        usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_DST\n      }),\n      uniform: device.createBuffer({\n        label: 'Photon grid parameters',\n        size: 32,\n        usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST\n      }),\n      data: device.createTexture({\n        label: 'Packed photon positions and flux',\n        size: [width, height],\n        format: 'rgba32uint',\n        usage: GPUTextureUsage.STORAGE_BINDING | GPUTextureUsage.TEXTURE_BINDING | GPUTextureUsage.COPY_SRC\n      }),\n      headers: device.createTexture({\n        label: 'Photon grid headers',\n        size: [\n          Math.min(headerTexels, limit),\n          Math.ceil((headerTexels) / limit)\n        ],\n        format: 'r32uint',\n        usage: GPUTextureUsage.STORAGE_BINDING | GPUTextureUsage.TEXTURE_BINDING | GPUTextureUsage.COPY_SRC\n      })\n    } as Batch;\n    batch.group = this.device.createBindGroup({\n      layout: this.layout,\n      entries: [\n        {binding: 0, resource: batch.data.createView()},\n        {binding: 1, resource: {buffer: batch.grid}},\n        {binding: 2, resource: {buffer: batch.uniform}},\n        {binding: 4, resource: batch.headers.createView()}\n      ]\n    });\n    this.targets.set(target, batch);\n    return batch;\n  }\n  get(target: object) {\n    return this.targets.get(target);\n  }\n  encode(\n    encoder: GPUCommandEncoder,\n    groups: GPUBindGroup[],\n    target: object,\n    radius: number,\n    options: Parameters<WGPUCausticLightPass['encode']>[3] & {progressiveBatch?: number}\n  ) {\n    const batch = this.targets.get(target)!;\n    if (causticDiagnostics.capturePhotons)\n      causticDiagnostics.photonBatch = {\n        device: this.device,\n        data: batch.headers,\n        emitted: batch.count,\n        packed: true\n      };\n    encoder.clearBuffer(batch.grid);\n    this.emission.emit(encoder, groups, batch.data, {\n      ...options,\n      photons: batch.count,\n      photonMapping: true,\n      photonDirections: batch.headers\n    });\n    const bytes = new ArrayBuffer(32);\n    new Float32Array(bytes)[0] = radius;\n    new Uint32Array(bytes)[1] = batch.count;\n    new Float32Array(bytes)[2] = causticDiagnostics.photonScreenRadius === false ? 0 : 2;\n    new Float32Array(bytes)[3] = causticDiagnostics.photonSkipQuery\n      ? 1\n      : causticDiagnostics.photonSkipVisible\n        ? 2\n        : 0;\n    new Uint32Array(bytes)[4] = batch.buckets;\n    new Float32Array(bytes)[5] = photonVisibleRadius(radius, options, new Float32Array(bytes)[2]);\n    new Uint32Array(bytes)[6] = (+batch.perCasterRadius) |\n      (Math.min(65535, Math.max(0, Math.round(causticDiagnostics.photonDeepScreenPixels ?? 0))) << 8);\n    this.device.queue.writeBuffer(batch.uniform, 0, bytes);\n    const group = batch.group;\n    // Dispatch boundaries synchronize storage accesses. Offline batches share one\n    // compute pass; timestamp profiling keeps separate passes for stage timings.\n    const combined = options.querySet ? undefined : encoder.beginComputePass({label: 'Photon grid'});\n    this.pipelines.forEach((pipeline, index) => {\n      const pass =\n        combined ??\n        encoder.beginComputePass({\n          label: ['Photon count', 'Photon scan blocks', 'Photon scan sums', 'Photon scatter'][index],\n          timestampWrites: {\n            querySet: options.querySet!,\n            beginningOfPassWriteIndex: 8 + index * 2,\n            endOfPassWriteIndex: 9 + index * 2\n          }\n        });\n      pass.setPipeline(pipeline);\n      pass.setBindGroup(0, group);\n      const groups =\n        index === 1\n          ? batch.buckets / 256\n          : index === 2\n            ? 1\n            : Math.ceil((index === 3 ? Math.max(batch.count, batch.buckets) : batch.count) / 64);\n      pass.dispatchWorkgroups(Math.min(groups, 65535), index === 1 || index === 2 ? 2 : Math.ceil(groups / 65535));\n      if (!combined) pass.end();\n    });\n    combined?.end();\n  }\n  release(target: object) {\n    const b = this.targets.get(target);\n    if (!b) return;\n    b.grid.destroy();\n    b.uniform.destroy();\n    b.data.destroy();\n    b.headers.destroy();\n    this.targets.delete(target);\n  }\n  bytes(target: object) {\n    const b = this.targets.get(target);\n    return b\n      ? b.grid.size +\n          b.uniform.size +\n          b.data.width * b.data.height * 16 +\n          b.headers.width * b.headers.height * 4\n      : 0;\n  }\n  dispose() {\n    for (const target of this.targets.keys()) this.release(target);\n    this.emission.dispose();\n    this.placeholder.destroy();\n  }\n}\n/** Typical first-receiver footprint at the caster center; query disks stay exact. */\nexport function photonVisibleRadius(radius: number, options: {viewProjection: ArrayLike<number>; casterSphere: readonly number[]; width: number; orthographic?: boolean}, screenPixels = 2) {\n  const m = options.viewProjection, p = options.casterSphere;\n  const depth = options.orthographic ? 1 : Math.abs(m[3]*p[0]+m[7]*p[1]+m[11]*p[2]+m[15]);\n  const projection = Math.hypot(m[0], m[4], m[8]);\n  const footprint = 2 * depth / (options.width * projection) * screenPixels;\n  return screenPixels > 0 && Number.isFinite(footprint) && footprint > 0 ? Math.min(radius, Math.max(1e-8, footprint)) : radius;\n}\n"]}