@meimu/mutracer 0.1.1 → 0.1.3

package/lib/renderer/WGPUJNLMDenoiser.js

@@ -27,9 +27,9 @@ ${shadowCatcherAccumulationWGSL}
 fn main(@builtin(global_invocation_id) id: vec3u) {
 let size = params.size;
 if (id.x >= size.x || id.y >= size.y) { return; }
-let tO = id.x + (size.y - 1u - id.y) * size.x;
-let n = normalBuffer[tO].a;
-let moments = accumulation[tO].moments;
+let tT = id.x + (size.y - 1u - id.y) * size.x;
+let n = normalBuffer[tT].a;
+let moments = accumulation[tT].moments;
 let mean = moments.x / max(n, 1.0);
 
 var variance = -1.0;
@@ -53,28 +53,28 @@ 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 lg = 0.0;
-var iX = 0.0;
-var mL = 0.0;
-var iR = 0.0;
+var lk = 0.0;
+var iY = 0.0;
+var mO = 0.0;
+var iS = 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];
-mL += value.x;
-iR += value.x * value.x;
+mO += value.x;
+iS += value.x * value.x;
 if (value.y >= 0.0) {
-lg += value.y;
-iX += 1.0;
+lk += value.y;
+iY += 1.0;
 }
 }
 }
 var variance: f32;
 if (center.y >= 0.0) {
-variance = lg / iX;
+variance = lk / iY;
 } else {
-let lf = mL / 9.0;
-variance = max(iR / 9.0 - lf * lf, 0.0) * 9.0 / 8.0;
+let lj = mO / 9.0;
+variance = max(iS / 9.0 - lj * lj, 0.0) * 9.0 / 8.0;
 }
 guide[id.x + id.y * size.x] = vec2f(center.x, variance);
 }
@@ -122,23 +122,23 @@ const INV_NORMAL_BANDWIDTH2 = ${(1 / normalBandwidth ** 2).toFixed(6)};
 
 const CENTER_WEIGHT_MIN = 0.01;
 
-fn gZ(p: vec2i) -> u32 {
+fn ha(p: vec2i) -> u32 {
 let q = clamp(p, vec2i(0), vec2i(params.size) - 1);
 return u32(q.x) + u32(q.y) * params.size.x;
 }
 
-fn lk(q: vec2i, bs: vec4f, gM: vec3f) -> f32 {
+fn ln(q: vec2i, bs: vec4f, gO: vec3f) -> f32 {
 let albedo = textureLoad(albedoTex, q, 0);
 
 
-let oS = max((albedo.rgb + bs.rgb) * 0.5, vec3f(${JNLM_ALBEDO_FLOOR}));
-let kR = (albedo.rgb - bs.rgb) / oS;
-let iO = albedo.a - bs.a;
-let lb = textureLoad(normalTex, q, 0).rgb - gM;
+let oW = max((albedo.rgb + bs.rgb) * 0.5, vec3f(${JNLM_ALBEDO_FLOOR}));
+let kU = (albedo.rgb - bs.rgb) / oW;
+let iP = albedo.a - bs.a;
+let le = textureLoad(normalTex, q, 0).rgb - gO;
 return max(
-dot(kR, kR) * INV_ALBEDO_BANDWIDTH2
-+ iO * iO * INV_COVERAGE_BANDWIDTH2,
-dot(lb, lb) * INV_NORMAL_BANDWIDTH2
+dot(kU, kU) * INV_ALBEDO_BANDWIDTH2
++ iP * iP * INV_COVERAGE_BANDWIDTH2,
+dot(le, le) * INV_NORMAL_BANDWIDTH2
 );
 }
 
@@ -154,8 +154,8 @@ if (p.x >= size.x || p.y >= size.y) { return; }
 let step = i32(params.dilation);
 let bs = textureLoad(albedoTex, p, 0);
 let fw = textureLoad(normalTex, p, 0);
-let gM = fw.rgb;
-let index = gZ(p);
+let gO = fw.rgb;
+let index = ha(p);
 
 
 if (fw.a <= 0.0 && textureLoad(shadowCatcherTex, p, 0).r <= 0.0) {
@@ -164,25 +164,25 @@ outGuide[index] = guide[index];
 return;
 }
 
-var kT: array<vec2f, PATCH_SIZE>;
+var kW: 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++) {
-kT[k] = guide[gZ(p + vec2i(dx, dy) * step)];
+kW[k] = guide[ha(p + vec2i(dx, dy) * step)];
 k++;
 }
 }
 let k2 = params.colorSensitivity * params.colorSensitivity;
 
 var sum = vec4f(0.0);
-let gP = max(bs.rgb, vec3f(${JNLM_ALBEDO_FLOOR}));
-var mb = vec3f(0.0);
-var gs = vec3f(0.0);
+let gR = max(bs.rgb, vec3f(${JNLM_ALBEDO_FLOOR}));
+var mf = vec3f(0.0);
+var gt = vec3f(0.0);
 var hn = vec3f(0.0);
-var pg = 0.0;
-var gu = 0.0;
-var nH = 0.0;
-var kA = 0.0;
+var pk = 0.0;
+var gv = 0.0;
+var nK = 0.0;
+var kD = 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;
@@ -191,17 +191,17 @@ if (
 q.x < 0 || q.y < 0 || q.x >= size.x || q.y >= size.y
 ) { continue; }
 
-let features = lk(q, bs, gM);
+let features = ln(q, bs, gO);
 if (features > 16.0) { continue; }
 
 var fT = 0.0;
 k = 0;
 for (var dy = -PATCH_RADIUS; dy <= PATCH_RADIUS; dy++) {
 for (var dx = -PATCH_RADIUS; dx <= PATCH_RADIUS; dx++) {
-let a = kT[k];
-let b = guide[gZ(q + vec2i(dx, dy) * step)];
-let qH = a.x - b.x;
-fT += (qH * qH
+let a = kW[k];
+let b = guide[ha(q + vec2i(dx, dy) * step)];
+let qL = a.x - b.x;
+fT += (qL * qL
 - VARIANCE_CANCELLATION * (a.y + min(a.y, b.y)))
 / (1e-10 + k2 * (a.y + b.y));
 k++;
@@ -210,44 +210,44 @@ k++;
 fT /= f32(PATCH_SIZE);
 
 let weight = exp(-max(max(fT, 0.0), features));
-let index = gZ(q);
+let index = ha(q);
 let c = color[index];
 let g = guide[index];
 
-let a = (textureLoad(albedoTex, q, 0).rgb - bs.rgb) / gP;
+let a = (textureLoad(albedoTex, q, 0).rgb - bs.rgb) / gR;
 let value = premultiplied(c);
 sum += value * weight;
-mb += a * weight;
-gs += a * a * weight;
+mf += a * weight;
+gt += a * a * weight;
 hn += a * value.rgb * weight;
-pg += g.x * weight;
-gu += g.y * weight * weight;
-nH += weight;
-kA = max(kA, weight);
+pk += g.x * weight;
+gv += g.y * weight * weight;
+nK += weight;
+kD = max(kD, weight);
 }
 }
 
 
 
 let center = color[index];
-let kS = guide[index];
-let dF = max(kA, CENTER_WEIGHT_MIN);
-let totalWeight = nH + dF;
-let tA = premultiplied(center);
-let mean = (sum + tA * dF) / totalWeight;
+let kV = guide[index];
+let dF = max(kD, CENTER_WEIGHT_MIN);
+let totalWeight = nK + dF;
+let tF = premultiplied(center);
+let mean = (sum + tF * dF) / totalWeight;
 
 
-let ez = mb / totalWeight;
-let ll = max(gs / totalWeight - ez * ez, vec3f(0.0));
-let qi = hn / totalWeight - ez * mean.rgb;
+let ez = mf / totalWeight;
+let lo = max(gt / totalWeight - ez * ez, vec3f(0.0));
+let qm = hn / totalWeight - ez * mean.rgb;
 
 
 
-let qs = bs.rgb + ez * gP;
-let ol = mean.rgb * gP / max(qs, vec3f(${JNLM_ALBEDO_FLOOR}));
-let vk = clamp(qi / (ll + vec3f(${JNLM_ALBEDO_REGULARIZATION})), vec3f(0.0), ol);
-let rn = mean.rgb - vk * ez;
-let filtered = vec4f(max(rn, vec3f(0.0)), max(mean.a, 0.0));
+let qw = bs.rgb + ez * gR;
+let oq = mean.rgb * gR / max(qw, vec3f(${JNLM_ALBEDO_FLOOR}));
+let vs = clamp(qm / (lo + vec3f(${JNLM_ALBEDO_REGULARIZATION})), vec3f(0.0), oq);
+let rs = mean.rgb - vs * ez;
+let filtered = vec4f(max(rs, vec3f(0.0)), max(mean.a, 0.0));
 let rgb = select(
 center.rgb,
 filtered.rgb / max(filtered.a, 1e-6),
@@ -255,8 +255,8 @@ filtered.a > 1e-6
 );
 outColor[index] = vec4f(rgb, filtered.a);
 outGuide[index] = vec2f(
-(pg + kS.x * dF) / totalWeight,
-(gu + kS.y * dF * dF)
+(pk + kV.x * dF) / totalWeight,
+(gv + kV.y * dF * dF)
 / (totalWeight * totalWeight)
 );
 }

package/lib/renderer/WGPURayTraceOutput.js

@@ -16,14 +16,14 @@ ${shadowCatcherAccumulationWGSL}
 override outputWidth: f32;
 override outputHeight: f32;
 
-struct oQ {
+struct oU {
 ${layers ? `@location(0) foreground: vec4f,
 @location(1) catcherBackground: vec4f` : `@location(0) color: vec4f,
 @location(1) normal: vec4f,
 @location(2) albedo: vec4f${catcher ? ',\n  @location(3) shadowCatcher: vec4f' : ''}`}
 }
 @fragment
-fn main(@builtin(position) fD: vec4f) -> oQ {
+fn main(@builtin(position) fD: vec4f) -> oU {
 let size = vec2u(u32(outputWidth), u32(outputHeight));
 let fM = vec2u(u32(fD.x), size.y - u32(fD.y) - 1u);
 var idx = fM.x + fM.y * size.x;
@@ -32,17 +32,17 @@ var idx = fM.x + fM.y * size.x;
 
 
 if (sampleCount < 1.0 && normalBuffer[idx].a <= 0.0) {
-let uO = max(sampleTileGrid, vec2u(1u));
-let uy = min(
-((fM + uO / 2u) / uO) * uO,
-((size - vec2u(1u)) / uO) * uO
+let uV = max(sampleTileGrid, vec2u(1u));
+let uF = min(
+((fM + uV / 2u) / uV) * uV,
+((size - vec2u(1u)) / uV) * uV
 );
-let sr = uy.x + uy.y * size.x;
-if (normalBuffer[sr].a > 0.0) {
-idx = sr;
+let sz = uF.x + uF.y * size.x;
+if (normalBuffer[sz].a > 0.0) {
+idx = sz;
 }
 }
-var output: oQ;
+var output: oU;
 if (normalBuffer[idx].a <= 0.0) {
 ${layers ? `output.foreground = vec4f(0.0);
 output.catcherBackground = vec4f(0.0);` : `output.color = vec4f(0.0);
@@ -62,10 +62,12 @@ let hit = albedo.a;
 let total = max(normal.a, 1.0);
 ${layers ? `
 let shadow = shadowCatcherBuffer[idx];
-let iP = max(hit - shadow.mask.b, 0.0);
+let iQ = max(hit - shadow.mask.b, 0.0);
+
+
 output.foreground = vec4f(
 (color.rgb - shadow.background.rgb)
-/ select(total, iP, iP > 0.0),
+/ select(total, iQ, iQ > 0.0),
 clamp((color.a - shadow.mask.g) / total, 0.0, 1.0)
 );
 output.catcherBackground = shadow.background / total;

package/lib/renderer/WGPURayTraceOutput.js.map

@@ -1 +1 @@
-{"version":3,"file":"WGPURayTraceOutput.js","sourceRoot":"","sources":["../../src/renderer/WGPURayTraceOutput.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AAC5D,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC9C,OAAO,6BAA6B,MAAM,iDAAiD,CAAC;AAC5F,OAAO,EAAE,8BAA8B,EAAE,MAAM,uBAAuB,CAAC;AAEvE,MAAM,UAAU,iCAAiC,CAAC,MAAM,GAAG,KAAK,EAAE,OAAO,GAAG,IAAI;IAC9E,OAAO,UAAU,CAAC;;;;EAIlB,6BAA6B;;;;;;;;;;EAU3B,MAAM,CAAC,CAAC,CAAC;sCAC2B,CAAC,CAAC,CAAC;;4BAEb,OAAO,CAAC,CAAC,CAAC,wCAAwC,CAAC,CAAC,CAAC,EAAE,EAAE;;;;;;;;;;;;;;;;;;;;;;;;EAwBjF,MAAM,CAAC,CAAC,CAAC;uCAC4B,CAAC,CAAC,CAAC;;;EAGxC,OAAO,CAAC,CAAC,CAAC,oCAAoC,CAAC,CAAC,CAAC,EAAE,EAAE;;;;;;;;;;;;EAYvD,MAAM,CAAC,CAAC,CAAC;;;;;;;;;CASV,CAAC,CAAC,CAAC;;;;;;;;;;;;;EAaF,OAAO,CAAC,CAAC,CAAC,+DAA+D,CAAC,CAAC,CAAC,EAAE;CAC/E;;;CAGF,CAAC;AACF,CAAC;AAYD,MAAM,OAAO,sBAAsB;IAqBe;IApBxC,OAAO,CAAY;IAEnB,SAAS,CAAqB;IAC9B,eAAe,CAAqB;IACpC,kBAAkB,CAAc;IAChC,yBAAyB,CAAc;IACvC,aAAa,CAAc;IAC3B,cAAc,CAAc;IAC5B,cAAc,CAAc;IAC5B,qBAAqB,CAAc;IACnC,iBAAiB,CAAgB;IAEjC,gBAAgB,CAAC;IAEjB,MAAM,GAAG,CAAC,CAAC;IACX,OAAO,GAAG,CAAC,CAAC;IAEZ,WAAW,GAAG,KAAK,CAAC;IACpB,oBAAoB,GAAG,IAAI,CAAC;IAEpC,YAAY,MAAiB,EAAmB,WAAwB;QAAxB,gBAAW,GAAX,WAAW,CAAa;QACtE,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,gBAAgB,GAAG;YACtB,WAAW,EAAE,MAAM,CAAC,YAAY,CAAC;gBAC/B,IAAI,EAAE,CAAC;gBACP,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ;aACxD,CAAC;YACF,cAAc,EAAE,MAAM,CAAC,YAAY,CAAC;gBAClC,IAAI,EAAE,CAAC;gBACP,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ;aACxD,CAAC;SACH,CAAC;IACJ,CAAC;IAEgB,gBAAgB,GAAG,IAAI,GAAG,EAAc,CAAC;IAC1D,YAAY,CAAC,OAAmB;QAC9B,MAAM,KAAK,GAAG,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,eAAe,EAAE,YAAY,EAAE,mBAAmB,CAAU,CAAC;QACzG,IAAI,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,OAAO,CAAC;YAAE,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IACjG,CAAC;IACD,aAAa,CAAC,OAAmB;QAC/B,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,OAAO,CAAC;YAAE,OAAO;QACnD,MAAM,KAAK,GAAG,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,eAAe,EAAE,YAAY,EAAE,mBAAmB,CAAU,CAAC;QACzG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,OAAO,CAAC,EAAE,CAAC;YAC5D,KAAK,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,mBAAmB,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;QACrG,CAAC;IACH,CAAC;IAED,SAAS,CAAC,KAAqB;QAE7B,OAAO,IAAI,CAAC,IAAI,KAAK,SAAS,CAAC,IAAI,IAAI,CAAC,WAAY,CAAC;IACvD,CAAC;IAEO,eAAe;QACrB,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;QACjE,MAAM,iBAAiB,GAAyB;YAC9C,IAAI,EAAE;gBACJ,KAAK;gBACL,MAAM;gBACN,kBAAkB,EAAE,CAAC;aACtB;YACD,MAAM,EAAE,YAAY;YACpB,KAAK,EAAE,eAAe,CAAC,iBAAiB,GAAG,eAAe,CAAC,eAAe;gBACxE,eAAe,CAAC,QAAQ;SAC3B,CAAC;QACF,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACxB,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC,aAAa,CAAC;YACxC,GAAG,iBAAiB;YACpB,KAAK,EAAE,iBAAiB;YACxB,MAAM,EAAE,aAAa;YACrB,KAAK,EACH,eAAe,CAAC,iBAAiB;gBACjC,eAAe,CAAC,eAAe;gBAC/B,eAAe,CAAC,QAAQ;SAC3B,CAAC,CAAC;QACH,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,aAAa,CAAC;YACzC,GAAG,iBAAiB;YACpB,KAAK,EAAE,kBAAkB;YACzB,KAAK,EAAE,iBAAiB,CAAC,KAAK,GAAG,eAAe,CAAC,QAAQ;SAC1D,CAAC,CAAC;QACH,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,aAAa,CAAC;YACzC,GAAG,iBAAiB;YACpB,KAAK,EAAE,kBAAkB;SAC1B,CAAC,CAAC;QACH,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACvB,IAAI,CAAC,kBAAkB,GAAG,MAAM,CAAC,aAAa,CAAC;gBAC7C,GAAG,iBAAiB;gBACpB,KAAK,EAAE,sBAAsB;gBAC7B,MAAM,EAAE,aAAa;aACtB,CAAC,CAAC;YACH,IAAI,CAAC,yBAAyB,GAAG,MAAM,CAAC,aAAa,CAAC;gBACpD,GAAG,iBAAiB;gBACpB,KAAK,EAAE,8BAA8B;gBACrC,MAAM,EAAE,aAAa;aACtB,CAAC,CAAC;YACH,IAAI,CAAC,qBAAqB,GAAG,MAAM,CAAC,aAAa,CAAC;gBAChD,GAAG,iBAAiB;gBACpB,KAAK,EAAE,0BAA0B;gBACjC,MAAM,EAAE,aAAa;aACtB,CAAC,CAAC;QACH,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,yBAAyB,GAAG,IAAI,CAAC,qBAAqB,GAAG,SAAU,CAAC;QACrG,CAAC;IACH,CAAC;IAEO,iBAAiB;QACvB,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,IAAI,CAAC,iBAAiB,GAAG,MAAM,CAAC,eAAe,CAAC;YAC9C,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC,CAAC;YAC5C,OAAO,EAAE;gBACP;oBACE,OAAO,EAAE,CAAC;oBACV,QAAQ,EAAE;wBACR,MAAM,EAAE,IAAI,CAAC,gBAAgB,CAAC,WAAW;qBAC1C;iBACF;gBACD;oBACE,OAAO,EAAE,CAAC;oBACV,QAAQ,EAAE;wBACR,MAAM,EAAE,IAAI,CAAC,gBAAgB,CAAC,cAAc;qBAC7C;iBACF;aACF;SACF,CAAC,CAAC;IACL,CAAC;IAEO,eAAe;QACrB,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC/B,OAAO;QACT,CAAC;QACD,IAAI,CAAC,oBAAoB,GAAG,KAAK,CAAC;QAClC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAE5B,IAAI,CAAC,SAAS,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC;YACjE,KAAK,EAAE,gBAAgB;YACvB,MAAM,EAAE,MAAM;YACd,MAAM,EAAE;gBACN,MAAM,EAAE,oBAAoB,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC;oBACtD,IAAI,EAAE,8BAA8B;iBACrC,CAAC;gBACF,UAAU,EAAE,MAAM;aACnB;YACD,QAAQ,EAAE;gBACR,MAAM,EAAE,oBAAoB,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC;oBACtD,IAAI,EAAE,iCAAiC,CAAC,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC;iBACjE,CAAC;gBACF,UAAU,EAAE,MAAM;gBAClB,OAAO,EAAE;oBACP;wBACE,MAAM,EAAE,aAAa;qBACtB;oBACD;wBACE,MAAM,EAAE,YAAY;qBACrB;oBACD;wBACE,MAAM,EAAE,YAAY;qBACrB;oBACD,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,EAAC,MAAM,EAAE,aAAsB,EAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;iBAChE;gBACD,SAAS,EAAE;oBACT,WAAW,EAAE,IAAI,CAAC,MAAM;oBACxB,YAAY,EAAE,IAAI,CAAC,OAAO;iBAC3B;aACF;SACF,CAAC,CAAC;QAEH,IAAI,IAAI,CAAC,WAAW;YAAE,IAAI,CAAC,eAAe,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC;gBAC7F,KAAK,EAAE,gBAAgB;gBACvB,MAAM,EAAE,MAAM;gBACd,MAAM,EAAE;oBACN,MAAM,EAAE,oBAAoB,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC,EAAE,IAAI,EAAE,8BAA8B,EAAE,CAAC;oBACjG,UAAU,EAAE,MAAM;iBACnB;gBACD,QAAQ,EAAE;oBACR,MAAM,EAAE,oBAAoB,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC,EAAE,IAAI,EAAE,iCAAiC,CAAC,IAAI,CAAC,EAAE,CAAC;oBAC1G,UAAU,EAAE,MAAM;oBAClB,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,aAAa,EAAE,EAAE,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC;oBAC/D,SAAS,EAAE,EAAE,WAAW,EAAE,IAAI,CAAC,MAAM,EAAE,YAAY,EAAE,IAAI,CAAC,OAAO,EAAE;iBACpE;aACF,CAAC,CAAC;QACH,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAC3B,CAAC;IAED,OAAO,CAAC,KAAa,EAAE,MAAc;QACnC,IACE,IAAI,CAAC,MAAM,KAAK,KAAK;YACrB,IAAI,CAAC,OAAO,KAAK,MAAM;YACvB,CAAC,IAAI,CAAC,aAAa,EACnB,CAAC;YACD,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;YACpB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;YACtB,IAAI,CAAC,eAAe,EAAE,CAAC;YACvB,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;QACnC,CAAC;IACH,CAAC;IAED,cAAc,CAAC,WAAmB;QAChC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAC5B,IAAI,CAAC,gBAAgB,CAAC,WAAW,EACjC,CAAC,EACD,IAAI,YAAY,CAAC,CAAC,WAAW,CAAC,CAAC,CAChC,CAAC;IACJ,CAAC;IAED,iBAAiB,CAAC,OAAe,EAAE,IAAY;QAC7C,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAC5B,IAAI,CAAC,gBAAgB,CAAC,cAAc,EACpC,CAAC,EACD,IAAI,WAAW,CAAC,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CACjC,CAAC;IACJ,CAAC;IAED,UAAU,CACR,OAA0B,EAC1B,YAA6B,EAC7B,eAAe,GAAG,KAAK;QAEvB,IAAI,IAAI,CAAC,WAAW,KAAK,eAAe,EAAE,CAAC;YACzC,IAAI,CAAC,WAAW,GAAG,eAAe,CAAC;YAAC,IAAI,CAAC,eAAe,EAAE,CAAC;YAAC,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;QAC/F,CAAC;QACD,IAAI,CAAC,eAAe,EAAE,CAAC;QACvB,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,YAAY,EAAE,KAAK,CAAC,CAAC;QACrD,IAAI,eAAe;YAAE,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,YAAY,EAAE,IAAI,CAAC,CAAC;IAC3E,CAAC;IAEO,iBAAiB,CACvB,OAA0B,EAC1B,YAA6B,EAC7B,MAAe;QAEf,MAAM,QAAQ,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC;QAChE,MAAM,QAAQ,GAAG,MAAM;YACrB,CAAC,CAAC,CAAC,IAAI,CAAC,kBAAkB,EAAE,IAAI,CAAC,yBAAyB,CAAC;YAC3D,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,EAAE,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,cAAc,EAAE,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAC9H,MAAM,IAAI,GAAG,OAAO,CAAC,eAAe,CAAC;YACnC,gBAAgB,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;gBAC3C,IAAI,EAAE,OAAO,CAAC,UAAU,EAAE;gBAC1B,SAAS,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE;gBACrC,OAAO,EAAE,OAAO;gBAChB,MAAM,EAAE,OAAO;aAChB,CAAC,CAAC;SACJ,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QAE3B,IAAI,CAAC,YAAY,CACf,CAAC,EACD,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC;YAC3B,KAAK,EAAE,gBAAgB;YACvB,MAAM,EAAE,QAAQ,CAAC,kBAAkB,CAAC,CAAC,CAAC;YACtC,OAAO,EAAE;gBACP,YAAY,CAAC,KAAK;gBAClB,YAAY,CAAC,MAAM;gBACnB,YAAY,CAAC,MAAM;gBACnB,YAAY,CAAC,aAAa;aAC3B,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;gBACtB,OAAO,EAAE,GAAG;gBACZ,QAAQ,EAAE;oBACR,MAAM;oBACN,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,YAAY,CAAC,mBAAmB,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBACxE;aACF,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,OAAO,GAAG,CAAC,IAAI,IAAI,CAAC,WAAW,IAAI,MAAM,CAAC;SACrE,CAAC,CACH,CAAC;QACF,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC;YACzD,MAAM,EAAE,QAAQ,CAAC,kBAAkB,CAAC,CAAC,CAAC;YACtC,OAAO,EAAE,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;gBACtE,OAAO,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE;aAC9B,CAAC,CAAC;SACJ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAE7B,IAAI,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QACtB,IAAI,CAAC,GAAG,EAAE,CAAC;IAEb,CAAC;IAEO,gBAAgB;QACtB,IAAI,CAAC,IAAI,CAAC,aAAa;YAAE,OAAO;QAEhC,oEAAoE;QACpE,uEAAuE;QACvE,0EAA0E;QAC1E,0EAA0E;QAC1E,2DAA2D;QAC3D,MAAM,QAAQ,GAAG;YACf,IAAI,CAAC,aAAa;YAClB,IAAI,CAAC,cAAc;YACnB,IAAI,CAAC,cAAc;YACnB,IAAI,CAAC,qBAAqB;YAC1B,IAAI,CAAC,kBAAkB;YACvB,IAAI,CAAC,yBAAyB;SAC/B,CAAC,MAAM,CAAC,CAAC,OAAO,EAAyB,EAAE,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC;QAC/F,KAAK,IAAI,CAAC,OAAO,CAAC,KAAK;aACpB,mBAAmB,EAAE;aACrB,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;aACtB,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;IAClE,CAAC;IAED,cAAc;QACZ,MAAM,QAAQ,GAAG,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,eAAe,EAAE,YAAY,EAAE,mBAAmB,CAAU,CAAC;QAC5G,OAAO,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,GAAG,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;cACrF,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,EAAE,GAAG,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC;iBAClG,MAAM,CAAC,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,GAAG,GAAG,YAAY,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC;IAChE,CAAC;IAED,OAAO;QACL,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACnD,KAAK,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,mBAAmB,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAChI,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACxB,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC;QAC9B,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YAChD,IAAI,CAAC,gBAAwB,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;QAChD,CAAC,CAAC,CAAC;IACL,CAAC;CACF","sourcesContent":["import { getTargetShaderCache } from './WGPUShaderCache.js';\nimport { textureBytes } from './gpuMemory.js';\nimport shadowCatcherAccumulationWGSL from './wgsl/chunks/shadowCatcherAccumulation.wgsl.js';\nimport { fullScreenQuadVertexShaderWGSL } from './WGPUFullQuadPass.js';\n\nexport function createNormalizeRayTraceOutputWGSL(layers = false, catcher = true) {\n  return /* wgsl */ `\n@group(0) @binding(0) var<storage, read> colorBuffer: array<vec4f>;\n@group(0) @binding(1) var<storage, read> normalBuffer: array<vec4f>;\n@group(0) @binding(2) var<storage, read> albedoBuffer: array<vec4f>;\n${shadowCatcherAccumulationWGSL}\n@group(0) @binding(3) var<storage, read> shadowCatcherBuffer: array<ShadowCatcherAccumulation>;\n\n@group(1) @binding(0) var<uniform> sampleCount: f32;\n@group(1) @binding(1) var<uniform> sampleTileGrid: vec2u;\n\noverride outputWidth: f32;\noverride outputHeight: f32;\n\nstruct FSOutput {\n  ${layers ? `@location(0) foreground: vec4f,\n  @location(1) catcherBackground: vec4f` : `@location(0) color: vec4f,\n  @location(1) normal: vec4f,\n  @location(2) albedo: vec4f${catcher ? ',\\n  @location(3) shadowCatcher: vec4f' : ''}`}\n}\n@fragment\nfn main(@builtin(position) coord: vec4f) -> FSOutput {\n  let size = vec2u(u32(outputWidth), u32(outputHeight));\n  let pixel = vec2u(u32(coord.x), size.y - u32(coord.y) - 1u);\n  var idx = pixel.x + pixel.y * size.x;\n  // Tiles are interleaved pixel grids, not contiguous rectangles. The first\n  // tile samples (0, 0) of every grid cell across the current camera image.\n  // Reconstruct only pixels not yet sampled, until the first sweep completes;\n  // never write these display estimates back into the accumulated ray buffers.\n  if (sampleCount < 1.0 && normalBuffer[idx].a <= 0.0) {\n    let grid = max(sampleTileGrid, vec2u(1u));\n    let nearest = min(\n      ((pixel + grid / 2u) / grid) * grid,\n      ((size - vec2u(1u)) / grid) * grid\n    );\n    let nearestIdx = nearest.x + nearest.y * size.x;\n    if (normalBuffer[nearestIdx].a > 0.0) {\n      idx = nearestIdx;\n    }\n  }\n  var output: FSOutput;\n  if (normalBuffer[idx].a <= 0.0) {\n    ${layers ? `output.foreground = vec4f(0.0);\n    output.catcherBackground = vec4f(0.0);` : `output.color = vec4f(0.0);\n    output.normal = vec4f(0.0);\n    output.albedo = vec4f(0.0);\n    ${catcher ? 'output.shadowCatcher = vec4f(0.0);' : ''}`}\n    return output;\n  }\n  let color = colorBuffer[idx];\n  let normal = normalBuffer[idx];\n  let albedo = albedoBuffer[idx];\n  let hit = albedo.a;\n  // The CPU count covers completed sweeps and is still zero during the first\n  // interleaved sample. Use the actual per-pixel count, including for misses.\n  // Empty pixels must remain finite: NaN alpha can poison float HDR output\n  // even when its canvas uses opaque compositing.\n  let total = max(normal.a, 1.0);\n  ${layers ? `\n  let shadow = shadowCatcherBuffer[idx];\n  let foregroundHit = max(hit - shadow.mask.b, 0.0);\n  output.foreground = vec4f(\n    (color.rgb - shadow.background.rgb)\n      / select(total, foregroundHit, foregroundHit > 0.0),\n    clamp((color.a - shadow.mask.g) / total, 0.0, 1.0)\n  );\n  output.catcherBackground = shadow.background / total;\n  ` : `output.color = vec4f(\n    // average the color by hit count.\n    // So the antialised edge color will be correct when having transparent background\n    color.rgb / select(total, hit, hit > 0.0),\n    color.a / total\n  );\n\n  output.normal = vec4f(0.0);\n  // Normalize to 0 - 1\n  if (dot(normal.rgb, normal.rgb) > 0.1) {\n    output.normal = vec4f(normal.rgb / total * 0.5 + 0.5, 1.0);\n  }\n  output.albedo = vec4f(albedo.rgb / total, clamp(hit / total, 0.0, 1.0));\n  ${catcher ? 'output.shadowCatcher = shadowCatcherBuffer[idx].mask / total;' : ''}\n  `}\n  return output;\n}\n`;\n}\n\ntype RayTraceOutput = 'color' | 'normal' | 'albedo' | 'shadowCatcher'\n  | 'foreground' | 'catcherBackground';\ntype RayTraceBuffers = {\n  color: GPUBuffer;\n  normal: GPUBuffer;\n  albedo: GPUBuffer;\n  shadowCatcher: GPUBuffer;\n  shadowCatcherOffset?: number;\n};\n\nexport class WGPURayTraceOutputPass {\n  private _device: GPUDevice;\n\n  private _pipeline!: GPURenderPipeline;\n  private _layersPipeline!: GPURenderPipeline;\n  private _foregroundTexture!: GPUTexture;\n  private _catcherBackgroundTexture!: GPUTexture;\n  private _colorTexture!: GPUTexture;\n  private _normalTexture!: GPUTexture;\n  private _albedoTexture!: GPUTexture;\n  private _shadowCatcherTexture!: GPUTexture;\n  private _uniformBindGroup!: GPUBindGroup;\n\n  private _uniformsBuffers;\n\n  private _width = 1;\n  private _height = 1;\n\n  private _hasCatcher = false;\n  private _needsUpdatePipeline = true;\n\n  constructor(device: GPUDevice, private readonly zeroTexture?: GPUTexture) {\n    this._device = device;\n    this._uniformsBuffers = {\n      sampleCount: device.createBuffer({\n        size: 4,\n        usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST\n      }),\n      sampleTileGrid: device.createBuffer({\n        size: 8,\n        usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST\n      })\n    };\n  }\n\n  private readonly _retainedOutputs = new Set<GPUTexture>();\n  retainOutput(texture: GPUTexture) {\n    const owned = ['color', 'normal', 'albedo', 'shadowCatcher', 'foreground', 'catcherBackground'] as const;\n    if (owned.some(label => this.getOutput(label) === texture)) this._retainedOutputs.add(texture);\n  }\n  releaseOutput(texture: GPUTexture) {\n    if (!this._retainedOutputs.delete(texture)) return;\n    const owned = ['color', 'normal', 'albedo', 'shadowCatcher', 'foreground', 'catcherBackground'] as const;\n    if (!owned.some(label => this.getOutput(label) === texture)) {\n      void this._device.queue.onSubmittedWorkDone().catch(() => undefined).then(() => texture.destroy());\n    }\n  }\n\n  getOutput(label: RayTraceOutput): GPUTexture {\n\n    return this[`_${label}Texture`] ?? this.zeroTexture!;\n  }\n\n  private _createTextures() {\n    const { _device: device, _width: width, _height: height } = this;\n    const textureDescriptor: GPUTextureDescriptor = {\n      size: {\n        width,\n        height,\n        depthOrArrayLayers: 1\n      },\n      format: 'rgba8unorm',\n      usage: GPUTextureUsage.RENDER_ATTACHMENT | GPUTextureUsage.TEXTURE_BINDING |\n        GPUTextureUsage.COPY_SRC\n    };\n    this._destroyTextures();\n    this._colorTexture = device.createTexture({\n      ...textureDescriptor,\n      label: 'Ray Trace Color',\n      format: 'rgba16float',\n      usage:\n        GPUTextureUsage.RENDER_ATTACHMENT |\n        GPUTextureUsage.TEXTURE_BINDING |\n        GPUTextureUsage.COPY_SRC\n    });\n    this._normalTexture = device.createTexture({\n      ...textureDescriptor,\n      label: 'Ray Trace Normal',\n      usage: textureDescriptor.usage | GPUTextureUsage.COPY_SRC\n    });\n    this._albedoTexture = device.createTexture({\n      ...textureDescriptor,\n      label: 'Ray Trace Albedo'\n    });\n    if (this._hasCatcher) {\n    this._foregroundTexture = device.createTexture({\n      ...textureDescriptor,\n      label: 'Ray Trace Foreground',\n      format: 'rgba16float'\n    });\n    this._catcherBackgroundTexture = device.createTexture({\n      ...textureDescriptor,\n      label: 'Ray Trace Catcher Background',\n      format: 'rgba16float'\n    });\n    this._shadowCatcherTexture = device.createTexture({\n      ...textureDescriptor,\n      label: 'Ray Trace Shadow Catcher',\n      format: 'rgba16float'\n    });\n    } else {\n      this._foregroundTexture = this._catcherBackgroundTexture = this._shadowCatcherTexture = undefined!;\n    }\n  }\n\n  private _updateBindGroups() {\n    const device = this._device;\n    this._uniformBindGroup = device.createBindGroup({\n      layout: this._pipeline.getBindGroupLayout(1),\n      entries: [\n        {\n          binding: 0,\n          resource: {\n            buffer: this._uniformsBuffers.sampleCount\n          }\n        },\n        {\n          binding: 1,\n          resource: {\n            buffer: this._uniformsBuffers.sampleTileGrid\n          }\n        }\n      ]\n    });\n  }\n\n  private _updatePipeline() {\n    if (!this._needsUpdatePipeline) {\n      return;\n    }\n    this._needsUpdatePipeline = false;\n    const device = this._device;\n\n    this._pipeline = getTargetShaderCache(device).createRenderPipeline({\n      label: 'rayTraceOutput',\n      layout: 'auto',\n      vertex: {\n        module: getTargetShaderCache(device).createShaderModule({\n          code: fullScreenQuadVertexShaderWGSL\n        }),\n        entryPoint: 'main'\n      },\n      fragment: {\n        module: getTargetShaderCache(device).createShaderModule({\n          code: createNormalizeRayTraceOutputWGSL(false, this._hasCatcher)\n        }),\n        entryPoint: 'main',\n        targets: [\n          {\n            format: 'rgba16float'\n          },\n          {\n            format: 'rgba8unorm'\n          },\n          {\n            format: 'rgba8unorm'\n          },\n          ...(this._hasCatcher ? [{format: 'rgba16float' as const}] : [])\n        ],\n        constants: {\n          outputWidth: this._width,\n          outputHeight: this._height\n        }\n      }\n    });\n\n    if (this._hasCatcher) this._layersPipeline = getTargetShaderCache(device).createRenderPipeline({\n      label: 'rayTraceLayers',\n      layout: 'auto',\n      vertex: {\n        module: getTargetShaderCache(device).createShaderModule({ code: fullScreenQuadVertexShaderWGSL }),\n        entryPoint: 'main'\n      },\n      fragment: {\n        module: getTargetShaderCache(device).createShaderModule({ code: createNormalizeRayTraceOutputWGSL(true) }),\n        entryPoint: 'main',\n        targets: [{ format: 'rgba16float' }, { format: 'rgba16float' }],\n        constants: { outputWidth: this._width, outputHeight: this._height }\n      }\n    });\n    this._updateBindGroups();\n  }\n\n  setSize(width: number, height: number) {\n    if (\n      this._width !== width ||\n      this._height !== height ||\n      !this._colorTexture\n    ) {\n      this._width = width;\n      this._height = height;\n      this._createTextures();\n      this._needsUpdatePipeline = true;\n    }\n  }\n\n  setSampleCount(sampleCount: number) {\n    this._device.queue.writeBuffer(\n      this._uniformsBuffers.sampleCount,\n      0,\n      new Float32Array([sampleCount])\n    );\n  }\n\n  setSampleTileGrid(columns: number, rows: number) {\n    this._device.queue.writeBuffer(\n      this._uniformsBuffers.sampleTileGrid,\n      0,\n      new Uint32Array([columns, rows])\n    );\n  }\n\n  createPass(\n    encoder: GPUCommandEncoder,\n    inputBuffers: RayTraceBuffers,\n    separateCatcher = false\n  ) {\n    if (this._hasCatcher !== separateCatcher) {\n      this._hasCatcher = separateCatcher; this._createTextures(); this._needsUpdatePipeline = true;\n    }\n    this._updatePipeline();\n    this._createOutputPass(encoder, inputBuffers, false);\n    if (separateCatcher) this._createOutputPass(encoder, inputBuffers, true);\n  }\n\n  private _createOutputPass(\n    encoder: GPUCommandEncoder,\n    inputBuffers: RayTraceBuffers,\n    layers: boolean\n  ) {\n    const pipeline = layers ? this._layersPipeline : this._pipeline;\n    const textures = layers\n      ? [this._foregroundTexture, this._catcherBackgroundTexture]\n      : [this._colorTexture, this._normalTexture, this._albedoTexture, ...(this._hasCatcher ? [this._shadowCatcherTexture] : [])];\n    const pass = encoder.beginRenderPass({\n      colorAttachments: textures.map((texture) => ({\n        view: texture.createView(),\n        loadValue: { r: 0, g: 0, b: 0, a: 0 },\n        storeOp: 'store',\n        loadOp: 'clear'\n      }))\n    });\n    pass.setPipeline(pipeline);\n\n    pass.setBindGroup(\n      0,\n      this._device.createBindGroup({\n        label: 'rayTraceOutput',\n        layout: pipeline.getBindGroupLayout(0),\n        entries: [\n          inputBuffers.color,\n          inputBuffers.normal,\n          inputBuffers.albedo,\n          inputBuffers.shadowCatcher\n        ].map((buffer, idx) => ({\n          binding: idx,\n          resource: {\n            buffer,\n            ...(idx === 3 ? { offset: inputBuffers.shadowCatcherOffset ?? 0 } : {})\n          }\n        })).filter(entry => entry.binding < 3 || this._hasCatcher || layers)\n      })\n    );\n    pass.setBindGroup(1, layers ? this._device.createBindGroup({\n      layout: pipeline.getBindGroupLayout(1),\n      entries: Object.values(this._uniformsBuffers).map((buffer, binding) => ({\n        binding, resource: { buffer }\n      }))\n    }) : this._uniformBindGroup);\n\n    pass.draw(6, 1, 0, 0);\n    pass.end();\n\n  }\n\n  private _destroyTextures() {\n    if (!this._colorTexture) return;\n\n    // These textures are also OIDN inputs. A resolution or preview-mode\n    // change can replace them immediately after a denoising command buffer\n    // was submitted. Keep the old resources alive until all work submitted so\n    // far has completed; destroying them synchronously can make Dawn validate\n    // the pending submit against an already-destroyed texture.\n    const textures = [\n      this._colorTexture,\n      this._normalTexture,\n      this._albedoTexture,\n      this._shadowCatcherTexture,\n      this._foregroundTexture,\n      this._catcherBackgroundTexture\n    ].filter((texture): texture is GPUTexture => !!texture && !this._retainedOutputs.has(texture));\n    void this._device.queue\n      .onSubmittedWorkDone()\n      .catch(() => undefined)\n      .then(() => textures.forEach((texture) => texture.destroy()));\n  }\n\n  getMemoryBytes() {\n    const textures = ['color', 'normal', 'albedo', 'shadowCatcher', 'foreground', 'catcherBackground'] as const;\n    return Object.values(this._uniformsBuffers).reduce((sum, buffer) => sum + buffer.size, 0)\n      + Array.from(new Set([...textures.map(label => this[`_${label}Texture`]), ...this._retainedOutputs]))\n        .reduce((sum, texture) => sum + textureBytes(texture), 0);\n  }\n\n  dispose() {\n    const retained = Array.from(this._retainedOutputs);\n    void this._device.queue.onSubmittedWorkDone().catch(() => undefined).then(() => retained.forEach(texture => texture.destroy()));\n    this._destroyTextures();\n    this._retainedOutputs.clear();\n    Object.keys(this._uniformsBuffers).forEach((key) => {\n      (this._uniformsBuffers as any)[key].destroy();\n    });\n  }\n}\n"]}
+{"version":3,"file":"WGPURayTraceOutput.js","sourceRoot":"","sources":["../../src/renderer/WGPURayTraceOutput.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AAC5D,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC9C,OAAO,6BAA6B,MAAM,iDAAiD,CAAC;AAC5F,OAAO,EAAE,8BAA8B,EAAE,MAAM,uBAAuB,CAAC;AAEvE,MAAM,UAAU,iCAAiC,CAAC,MAAM,GAAG,KAAK,EAAE,OAAO,GAAG,IAAI;IAC9E,OAAO,UAAU,CAAC;;;;EAIlB,6BAA6B;;;;;;;;;;EAU3B,MAAM,CAAC,CAAC,CAAC;sCAC2B,CAAC,CAAC,CAAC;;4BAEb,OAAO,CAAC,CAAC,CAAC,wCAAwC,CAAC,CAAC,CAAC,EAAE,EAAE;;;;;;;;;;;;;;;;;;;;;;;;EAwBjF,MAAM,CAAC,CAAC,CAAC;uCAC4B,CAAC,CAAC,CAAC;;;EAGxC,OAAO,CAAC,CAAC,CAAC,oCAAoC,CAAC,CAAC,CAAC,EAAE,EAAE;;;;;;;;;;;;EAYvD,MAAM,CAAC,CAAC,CAAC;;;;;;;;;;;CAWV,CAAC,CAAC,CAAC;;;;;;;;;;;;;EAaF,OAAO,CAAC,CAAC,CAAC,+DAA+D,CAAC,CAAC,CAAC,EAAE;CAC/E;;;CAGF,CAAC;AACF,CAAC;AAYD,MAAM,OAAO,sBAAsB;IAqBe;IApBxC,OAAO,CAAY;IAEnB,SAAS,CAAqB;IAC9B,eAAe,CAAqB;IACpC,kBAAkB,CAAc;IAChC,yBAAyB,CAAc;IACvC,aAAa,CAAc;IAC3B,cAAc,CAAc;IAC5B,cAAc,CAAc;IAC5B,qBAAqB,CAAc;IACnC,iBAAiB,CAAgB;IAEjC,gBAAgB,CAAC;IAEjB,MAAM,GAAG,CAAC,CAAC;IACX,OAAO,GAAG,CAAC,CAAC;IAEZ,WAAW,GAAG,KAAK,CAAC;IACpB,oBAAoB,GAAG,IAAI,CAAC;IAEpC,YAAY,MAAiB,EAAmB,WAAwB;QAAxB,gBAAW,GAAX,WAAW,CAAa;QACtE,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,gBAAgB,GAAG;YACtB,WAAW,EAAE,MAAM,CAAC,YAAY,CAAC;gBAC/B,IAAI,EAAE,CAAC;gBACP,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ;aACxD,CAAC;YACF,cAAc,EAAE,MAAM,CAAC,YAAY,CAAC;gBAClC,IAAI,EAAE,CAAC;gBACP,KAAK,EAAE,cAAc,CAAC,OAAO,GAAG,cAAc,CAAC,QAAQ;aACxD,CAAC;SACH,CAAC;IACJ,CAAC;IAEgB,gBAAgB,GAAG,IAAI,GAAG,EAAc,CAAC;IAC1D,YAAY,CAAC,OAAmB;QAC9B,MAAM,KAAK,GAAG,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,eAAe,EAAE,YAAY,EAAE,mBAAmB,CAAU,CAAC;QACzG,IAAI,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,OAAO,CAAC;YAAE,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IACjG,CAAC;IACD,aAAa,CAAC,OAAmB;QAC/B,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,OAAO,CAAC;YAAE,OAAO;QACnD,MAAM,KAAK,GAAG,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,eAAe,EAAE,YAAY,EAAE,mBAAmB,CAAU,CAAC;QACzG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,OAAO,CAAC,EAAE,CAAC;YAC5D,KAAK,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,mBAAmB,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;QACrG,CAAC;IACH,CAAC;IAED,SAAS,CAAC,KAAqB;QAE7B,OAAO,IAAI,CAAC,IAAI,KAAK,SAAS,CAAC,IAAI,IAAI,CAAC,WAAY,CAAC;IACvD,CAAC;IAEO,eAAe;QACrB,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;QACjE,MAAM,iBAAiB,GAAyB;YAC9C,IAAI,EAAE;gBACJ,KAAK;gBACL,MAAM;gBACN,kBAAkB,EAAE,CAAC;aACtB;YACD,MAAM,EAAE,YAAY;YACpB,KAAK,EAAE,eAAe,CAAC,iBAAiB,GAAG,eAAe,CAAC,eAAe;gBACxE,eAAe,CAAC,QAAQ;SAC3B,CAAC;QACF,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACxB,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC,aAAa,CAAC;YACxC,GAAG,iBAAiB;YACpB,KAAK,EAAE,iBAAiB;YACxB,MAAM,EAAE,aAAa;YACrB,KAAK,EACH,eAAe,CAAC,iBAAiB;gBACjC,eAAe,CAAC,eAAe;gBAC/B,eAAe,CAAC,QAAQ;SAC3B,CAAC,CAAC;QACH,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,aAAa,CAAC;YACzC,GAAG,iBAAiB;YACpB,KAAK,EAAE,kBAAkB;YACzB,KAAK,EAAE,iBAAiB,CAAC,KAAK,GAAG,eAAe,CAAC,QAAQ;SAC1D,CAAC,CAAC;QACH,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,aAAa,CAAC;YACzC,GAAG,iBAAiB;YACpB,KAAK,EAAE,kBAAkB;SAC1B,CAAC,CAAC;QACH,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACvB,IAAI,CAAC,kBAAkB,GAAG,MAAM,CAAC,aAAa,CAAC;gBAC7C,GAAG,iBAAiB;gBACpB,KAAK,EAAE,sBAAsB;gBAC7B,MAAM,EAAE,aAAa;aACtB,CAAC,CAAC;YACH,IAAI,CAAC,yBAAyB,GAAG,MAAM,CAAC,aAAa,CAAC;gBACpD,GAAG,iBAAiB;gBACpB,KAAK,EAAE,8BAA8B;gBACrC,MAAM,EAAE,aAAa;aACtB,CAAC,CAAC;YACH,IAAI,CAAC,qBAAqB,GAAG,MAAM,CAAC,aAAa,CAAC;gBAChD,GAAG,iBAAiB;gBACpB,KAAK,EAAE,0BAA0B;gBACjC,MAAM,EAAE,aAAa;aACtB,CAAC,CAAC;QACH,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,yBAAyB,GAAG,IAAI,CAAC,qBAAqB,GAAG,SAAU,CAAC;QACrG,CAAC;IACH,CAAC;IAEO,iBAAiB;QACvB,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,IAAI,CAAC,iBAAiB,GAAG,MAAM,CAAC,eAAe,CAAC;YAC9C,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC,CAAC;YAC5C,OAAO,EAAE;gBACP;oBACE,OAAO,EAAE,CAAC;oBACV,QAAQ,EAAE;wBACR,MAAM,EAAE,IAAI,CAAC,gBAAgB,CAAC,WAAW;qBAC1C;iBACF;gBACD;oBACE,OAAO,EAAE,CAAC;oBACV,QAAQ,EAAE;wBACR,MAAM,EAAE,IAAI,CAAC,gBAAgB,CAAC,cAAc;qBAC7C;iBACF;aACF;SACF,CAAC,CAAC;IACL,CAAC;IAEO,eAAe;QACrB,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC/B,OAAO;QACT,CAAC;QACD,IAAI,CAAC,oBAAoB,GAAG,KAAK,CAAC;QAClC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAE5B,IAAI,CAAC,SAAS,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC;YACjE,KAAK,EAAE,gBAAgB;YACvB,MAAM,EAAE,MAAM;YACd,MAAM,EAAE;gBACN,MAAM,EAAE,oBAAoB,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC;oBACtD,IAAI,EAAE,8BAA8B;iBACrC,CAAC;gBACF,UAAU,EAAE,MAAM;aACnB;YACD,QAAQ,EAAE;gBACR,MAAM,EAAE,oBAAoB,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC;oBACtD,IAAI,EAAE,iCAAiC,CAAC,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC;iBACjE,CAAC;gBACF,UAAU,EAAE,MAAM;gBAClB,OAAO,EAAE;oBACP;wBACE,MAAM,EAAE,aAAa;qBACtB;oBACD;wBACE,MAAM,EAAE,YAAY;qBACrB;oBACD;wBACE,MAAM,EAAE,YAAY;qBACrB;oBACD,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,EAAC,MAAM,EAAE,aAAsB,EAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;iBAChE;gBACD,SAAS,EAAE;oBACT,WAAW,EAAE,IAAI,CAAC,MAAM;oBACxB,YAAY,EAAE,IAAI,CAAC,OAAO;iBAC3B;aACF;SACF,CAAC,CAAC;QAEH,IAAI,IAAI,CAAC,WAAW;YAAE,IAAI,CAAC,eAAe,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC;gBAC7F,KAAK,EAAE,gBAAgB;gBACvB,MAAM,EAAE,MAAM;gBACd,MAAM,EAAE;oBACN,MAAM,EAAE,oBAAoB,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC,EAAE,IAAI,EAAE,8BAA8B,EAAE,CAAC;oBACjG,UAAU,EAAE,MAAM;iBACnB;gBACD,QAAQ,EAAE;oBACR,MAAM,EAAE,oBAAoB,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC,EAAE,IAAI,EAAE,iCAAiC,CAAC,IAAI,CAAC,EAAE,CAAC;oBAC1G,UAAU,EAAE,MAAM;oBAClB,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,aAAa,EAAE,EAAE,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC;oBAC/D,SAAS,EAAE,EAAE,WAAW,EAAE,IAAI,CAAC,MAAM,EAAE,YAAY,EAAE,IAAI,CAAC,OAAO,EAAE;iBACpE;aACF,CAAC,CAAC;QACH,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAC3B,CAAC;IAED,OAAO,CAAC,KAAa,EAAE,MAAc;QACnC,IACE,IAAI,CAAC,MAAM,KAAK,KAAK;YACrB,IAAI,CAAC,OAAO,KAAK,MAAM;YACvB,CAAC,IAAI,CAAC,aAAa,EACnB,CAAC;YACD,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;YACpB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;YACtB,IAAI,CAAC,eAAe,EAAE,CAAC;YACvB,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;QACnC,CAAC;IACH,CAAC;IAED,cAAc,CAAC,WAAmB;QAChC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAC5B,IAAI,CAAC,gBAAgB,CAAC,WAAW,EACjC,CAAC,EACD,IAAI,YAAY,CAAC,CAAC,WAAW,CAAC,CAAC,CAChC,CAAC;IACJ,CAAC;IAED,iBAAiB,CAAC,OAAe,EAAE,IAAY;QAC7C,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAC5B,IAAI,CAAC,gBAAgB,CAAC,cAAc,EACpC,CAAC,EACD,IAAI,WAAW,CAAC,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CACjC,CAAC;IACJ,CAAC;IAED,UAAU,CACR,OAA0B,EAC1B,YAA6B,EAC7B,eAAe,GAAG,KAAK;QAEvB,IAAI,IAAI,CAAC,WAAW,KAAK,eAAe,EAAE,CAAC;YACzC,IAAI,CAAC,WAAW,GAAG,eAAe,CAAC;YAAC,IAAI,CAAC,eAAe,EAAE,CAAC;YAAC,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;QAC/F,CAAC;QACD,IAAI,CAAC,eAAe,EAAE,CAAC;QACvB,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,YAAY,EAAE,KAAK,CAAC,CAAC;QACrD,IAAI,eAAe;YAAE,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,YAAY,EAAE,IAAI,CAAC,CAAC;IAC3E,CAAC;IAEO,iBAAiB,CACvB,OAA0B,EAC1B,YAA6B,EAC7B,MAAe;QAEf,MAAM,QAAQ,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC;QAChE,MAAM,QAAQ,GAAG,MAAM;YACrB,CAAC,CAAC,CAAC,IAAI,CAAC,kBAAkB,EAAE,IAAI,CAAC,yBAAyB,CAAC;YAC3D,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,EAAE,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,cAAc,EAAE,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAC9H,MAAM,IAAI,GAAG,OAAO,CAAC,eAAe,CAAC;YACnC,gBAAgB,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;gBAC3C,IAAI,EAAE,OAAO,CAAC,UAAU,EAAE;gBAC1B,SAAS,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE;gBACrC,OAAO,EAAE,OAAO;gBAChB,MAAM,EAAE,OAAO;aAChB,CAAC,CAAC;SACJ,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QAE3B,IAAI,CAAC,YAAY,CACf,CAAC,EACD,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC;YAC3B,KAAK,EAAE,gBAAgB;YACvB,MAAM,EAAE,QAAQ,CAAC,kBAAkB,CAAC,CAAC,CAAC;YACtC,OAAO,EAAE;gBACP,YAAY,CAAC,KAAK;gBAClB,YAAY,CAAC,MAAM;gBACnB,YAAY,CAAC,MAAM;gBACnB,YAAY,CAAC,aAAa;aAC3B,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;gBACtB,OAAO,EAAE,GAAG;gBACZ,QAAQ,EAAE;oBACR,MAAM;oBACN,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,YAAY,CAAC,mBAAmB,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBACxE;aACF,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,OAAO,GAAG,CAAC,IAAI,IAAI,CAAC,WAAW,IAAI,MAAM,CAAC;SACrE,CAAC,CACH,CAAC;QACF,IAAI,CAAC,YAAY,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC;YACzD,MAAM,EAAE,QAAQ,CAAC,kBAAkB,CAAC,CAAC,CAAC;YACtC,OAAO,EAAE,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;gBACtE,OAAO,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE;aAC9B,CAAC,CAAC;SACJ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAE7B,IAAI,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QACtB,IAAI,CAAC,GAAG,EAAE,CAAC;IAEb,CAAC;IAEO,gBAAgB;QACtB,IAAI,CAAC,IAAI,CAAC,aAAa;YAAE,OAAO;QAEhC,oEAAoE;QACpE,uEAAuE;QACvE,0EAA0E;QAC1E,0EAA0E;QAC1E,2DAA2D;QAC3D,MAAM,QAAQ,GAAG;YACf,IAAI,CAAC,aAAa;YAClB,IAAI,CAAC,cAAc;YACnB,IAAI,CAAC,cAAc;YACnB,IAAI,CAAC,qBAAqB;YAC1B,IAAI,CAAC,kBAAkB;YACvB,IAAI,CAAC,yBAAyB;SAC/B,CAAC,MAAM,CAAC,CAAC,OAAO,EAAyB,EAAE,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC;QAC/F,KAAK,IAAI,CAAC,OAAO,CAAC,KAAK;aACpB,mBAAmB,EAAE;aACrB,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;aACtB,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;IAClE,CAAC;IAED,cAAc;QACZ,MAAM,QAAQ,GAAG,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,eAAe,EAAE,YAAY,EAAE,mBAAmB,CAAU,CAAC;QAC5G,OAAO,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,GAAG,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;cACrF,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,EAAE,GAAG,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC;iBAClG,MAAM,CAAC,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,GAAG,GAAG,YAAY,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC;IAChE,CAAC;IAED,OAAO;QACL,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACnD,KAAK,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,mBAAmB,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAChI,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACxB,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC;QAC9B,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YAChD,IAAI,CAAC,gBAAwB,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;QAChD,CAAC,CAAC,CAAC;IACL,CAAC;CACF","sourcesContent":["import { getTargetShaderCache } from './WGPUShaderCache.js';\nimport { textureBytes } from './gpuMemory.js';\nimport shadowCatcherAccumulationWGSL from './wgsl/chunks/shadowCatcherAccumulation.wgsl.js';\nimport { fullScreenQuadVertexShaderWGSL } from './WGPUFullQuadPass.js';\n\nexport function createNormalizeRayTraceOutputWGSL(layers = false, catcher = true) {\n  return /* wgsl */ `\n@group(0) @binding(0) var<storage, read> colorBuffer: array<vec4f>;\n@group(0) @binding(1) var<storage, read> normalBuffer: array<vec4f>;\n@group(0) @binding(2) var<storage, read> albedoBuffer: array<vec4f>;\n${shadowCatcherAccumulationWGSL}\n@group(0) @binding(3) var<storage, read> shadowCatcherBuffer: array<ShadowCatcherAccumulation>;\n\n@group(1) @binding(0) var<uniform> sampleCount: f32;\n@group(1) @binding(1) var<uniform> sampleTileGrid: vec2u;\n\noverride outputWidth: f32;\noverride outputHeight: f32;\n\nstruct FSOutput {\n  ${layers ? `@location(0) foreground: vec4f,\n  @location(1) catcherBackground: vec4f` : `@location(0) color: vec4f,\n  @location(1) normal: vec4f,\n  @location(2) albedo: vec4f${catcher ? ',\\n  @location(3) shadowCatcher: vec4f' : ''}`}\n}\n@fragment\nfn main(@builtin(position) coord: vec4f) -> FSOutput {\n  let size = vec2u(u32(outputWidth), u32(outputHeight));\n  let pixel = vec2u(u32(coord.x), size.y - u32(coord.y) - 1u);\n  var idx = pixel.x + pixel.y * size.x;\n  // Tiles are interleaved pixel grids, not contiguous rectangles. The first\n  // tile samples (0, 0) of every grid cell across the current camera image.\n  // Reconstruct only pixels not yet sampled, until the first sweep completes;\n  // never write these display estimates back into the accumulated ray buffers.\n  if (sampleCount < 1.0 && normalBuffer[idx].a <= 0.0) {\n    let grid = max(sampleTileGrid, vec2u(1u));\n    let nearest = min(\n      ((pixel + grid / 2u) / grid) * grid,\n      ((size - vec2u(1u)) / grid) * grid\n    );\n    let nearestIdx = nearest.x + nearest.y * size.x;\n    if (normalBuffer[nearestIdx].a > 0.0) {\n      idx = nearestIdx;\n    }\n  }\n  var output: FSOutput;\n  if (normalBuffer[idx].a <= 0.0) {\n    ${layers ? `output.foreground = vec4f(0.0);\n    output.catcherBackground = vec4f(0.0);` : `output.color = vec4f(0.0);\n    output.normal = vec4f(0.0);\n    output.albedo = vec4f(0.0);\n    ${catcher ? 'output.shadowCatcher = vec4f(0.0);' : ''}`}\n    return output;\n  }\n  let color = colorBuffer[idx];\n  let normal = normalBuffer[idx];\n  let albedo = albedoBuffer[idx];\n  let hit = albedo.a;\n  // The CPU count covers completed sweeps and is still zero during the first\n  // interleaved sample. Use the actual per-pixel count, including for misses.\n  // Empty pixels must remain finite: NaN alpha can poison float HDR output\n  // even when its canvas uses opaque compositing.\n  let total = max(normal.a, 1.0);\n  ${layers ? `\n  let shadow = shadowCatcherBuffer[idx];\n  let foregroundHit = max(hit - shadow.mask.b, 0.0);\n  // Separated background includes flat primary misses as well as catcher\n  // samples. Neither belongs in the hit-normalized geometry color/alpha.\n  output.foreground = vec4f(\n    (color.rgb - shadow.background.rgb)\n      / select(total, foregroundHit, foregroundHit > 0.0),\n    clamp((color.a - shadow.mask.g) / total, 0.0, 1.0)\n  );\n  output.catcherBackground = shadow.background / total;\n  ` : `output.color = vec4f(\n    // average the color by hit count.\n    // So the antialised edge color will be correct when having transparent background\n    color.rgb / select(total, hit, hit > 0.0),\n    color.a / total\n  );\n\n  output.normal = vec4f(0.0);\n  // Normalize to 0 - 1\n  if (dot(normal.rgb, normal.rgb) > 0.1) {\n    output.normal = vec4f(normal.rgb / total * 0.5 + 0.5, 1.0);\n  }\n  output.albedo = vec4f(albedo.rgb / total, clamp(hit / total, 0.0, 1.0));\n  ${catcher ? 'output.shadowCatcher = shadowCatcherBuffer[idx].mask / total;' : ''}\n  `}\n  return output;\n}\n`;\n}\n\ntype RayTraceOutput = 'color' | 'normal' | 'albedo' | 'shadowCatcher'\n  | 'foreground' | 'catcherBackground';\ntype RayTraceBuffers = {\n  color: GPUBuffer;\n  normal: GPUBuffer;\n  albedo: GPUBuffer;\n  shadowCatcher: GPUBuffer;\n  shadowCatcherOffset?: number;\n};\n\nexport class WGPURayTraceOutputPass {\n  private _device: GPUDevice;\n\n  private _pipeline!: GPURenderPipeline;\n  private _layersPipeline!: GPURenderPipeline;\n  private _foregroundTexture!: GPUTexture;\n  private _catcherBackgroundTexture!: GPUTexture;\n  private _colorTexture!: GPUTexture;\n  private _normalTexture!: GPUTexture;\n  private _albedoTexture!: GPUTexture;\n  private _shadowCatcherTexture!: GPUTexture;\n  private _uniformBindGroup!: GPUBindGroup;\n\n  private _uniformsBuffers;\n\n  private _width = 1;\n  private _height = 1;\n\n  private _hasCatcher = false;\n  private _needsUpdatePipeline = true;\n\n  constructor(device: GPUDevice, private readonly zeroTexture?: GPUTexture) {\n    this._device = device;\n    this._uniformsBuffers = {\n      sampleCount: device.createBuffer({\n        size: 4,\n        usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST\n      }),\n      sampleTileGrid: device.createBuffer({\n        size: 8,\n        usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST\n      })\n    };\n  }\n\n  private readonly _retainedOutputs = new Set<GPUTexture>();\n  retainOutput(texture: GPUTexture) {\n    const owned = ['color', 'normal', 'albedo', 'shadowCatcher', 'foreground', 'catcherBackground'] as const;\n    if (owned.some(label => this.getOutput(label) === texture)) this._retainedOutputs.add(texture);\n  }\n  releaseOutput(texture: GPUTexture) {\n    if (!this._retainedOutputs.delete(texture)) return;\n    const owned = ['color', 'normal', 'albedo', 'shadowCatcher', 'foreground', 'catcherBackground'] as const;\n    if (!owned.some(label => this.getOutput(label) === texture)) {\n      void this._device.queue.onSubmittedWorkDone().catch(() => undefined).then(() => texture.destroy());\n    }\n  }\n\n  getOutput(label: RayTraceOutput): GPUTexture {\n\n    return this[`_${label}Texture`] ?? this.zeroTexture!;\n  }\n\n  private _createTextures() {\n    const { _device: device, _width: width, _height: height } = this;\n    const textureDescriptor: GPUTextureDescriptor = {\n      size: {\n        width,\n        height,\n        depthOrArrayLayers: 1\n      },\n      format: 'rgba8unorm',\n      usage: GPUTextureUsage.RENDER_ATTACHMENT | GPUTextureUsage.TEXTURE_BINDING |\n        GPUTextureUsage.COPY_SRC\n    };\n    this._destroyTextures();\n    this._colorTexture = device.createTexture({\n      ...textureDescriptor,\n      label: 'Ray Trace Color',\n      format: 'rgba16float',\n      usage:\n        GPUTextureUsage.RENDER_ATTACHMENT |\n        GPUTextureUsage.TEXTURE_BINDING |\n        GPUTextureUsage.COPY_SRC\n    });\n    this._normalTexture = device.createTexture({\n      ...textureDescriptor,\n      label: 'Ray Trace Normal',\n      usage: textureDescriptor.usage | GPUTextureUsage.COPY_SRC\n    });\n    this._albedoTexture = device.createTexture({\n      ...textureDescriptor,\n      label: 'Ray Trace Albedo'\n    });\n    if (this._hasCatcher) {\n    this._foregroundTexture = device.createTexture({\n      ...textureDescriptor,\n      label: 'Ray Trace Foreground',\n      format: 'rgba16float'\n    });\n    this._catcherBackgroundTexture = device.createTexture({\n      ...textureDescriptor,\n      label: 'Ray Trace Catcher Background',\n      format: 'rgba16float'\n    });\n    this._shadowCatcherTexture = device.createTexture({\n      ...textureDescriptor,\n      label: 'Ray Trace Shadow Catcher',\n      format: 'rgba16float'\n    });\n    } else {\n      this._foregroundTexture = this._catcherBackgroundTexture = this._shadowCatcherTexture = undefined!;\n    }\n  }\n\n  private _updateBindGroups() {\n    const device = this._device;\n    this._uniformBindGroup = device.createBindGroup({\n      layout: this._pipeline.getBindGroupLayout(1),\n      entries: [\n        {\n          binding: 0,\n          resource: {\n            buffer: this._uniformsBuffers.sampleCount\n          }\n        },\n        {\n          binding: 1,\n          resource: {\n            buffer: this._uniformsBuffers.sampleTileGrid\n          }\n        }\n      ]\n    });\n  }\n\n  private _updatePipeline() {\n    if (!this._needsUpdatePipeline) {\n      return;\n    }\n    this._needsUpdatePipeline = false;\n    const device = this._device;\n\n    this._pipeline = getTargetShaderCache(device).createRenderPipeline({\n      label: 'rayTraceOutput',\n      layout: 'auto',\n      vertex: {\n        module: getTargetShaderCache(device).createShaderModule({\n          code: fullScreenQuadVertexShaderWGSL\n        }),\n        entryPoint: 'main'\n      },\n      fragment: {\n        module: getTargetShaderCache(device).createShaderModule({\n          code: createNormalizeRayTraceOutputWGSL(false, this._hasCatcher)\n        }),\n        entryPoint: 'main',\n        targets: [\n          {\n            format: 'rgba16float'\n          },\n          {\n            format: 'rgba8unorm'\n          },\n          {\n            format: 'rgba8unorm'\n          },\n          ...(this._hasCatcher ? [{format: 'rgba16float' as const}] : [])\n        ],\n        constants: {\n          outputWidth: this._width,\n          outputHeight: this._height\n        }\n      }\n    });\n\n    if (this._hasCatcher) this._layersPipeline = getTargetShaderCache(device).createRenderPipeline({\n      label: 'rayTraceLayers',\n      layout: 'auto',\n      vertex: {\n        module: getTargetShaderCache(device).createShaderModule({ code: fullScreenQuadVertexShaderWGSL }),\n        entryPoint: 'main'\n      },\n      fragment: {\n        module: getTargetShaderCache(device).createShaderModule({ code: createNormalizeRayTraceOutputWGSL(true) }),\n        entryPoint: 'main',\n        targets: [{ format: 'rgba16float' }, { format: 'rgba16float' }],\n        constants: { outputWidth: this._width, outputHeight: this._height }\n      }\n    });\n    this._updateBindGroups();\n  }\n\n  setSize(width: number, height: number) {\n    if (\n      this._width !== width ||\n      this._height !== height ||\n      !this._colorTexture\n    ) {\n      this._width = width;\n      this._height = height;\n      this._createTextures();\n      this._needsUpdatePipeline = true;\n    }\n  }\n\n  setSampleCount(sampleCount: number) {\n    this._device.queue.writeBuffer(\n      this._uniformsBuffers.sampleCount,\n      0,\n      new Float32Array([sampleCount])\n    );\n  }\n\n  setSampleTileGrid(columns: number, rows: number) {\n    this._device.queue.writeBuffer(\n      this._uniformsBuffers.sampleTileGrid,\n      0,\n      new Uint32Array([columns, rows])\n    );\n  }\n\n  createPass(\n    encoder: GPUCommandEncoder,\n    inputBuffers: RayTraceBuffers,\n    separateCatcher = false\n  ) {\n    if (this._hasCatcher !== separateCatcher) {\n      this._hasCatcher = separateCatcher; this._createTextures(); this._needsUpdatePipeline = true;\n    }\n    this._updatePipeline();\n    this._createOutputPass(encoder, inputBuffers, false);\n    if (separateCatcher) this._createOutputPass(encoder, inputBuffers, true);\n  }\n\n  private _createOutputPass(\n    encoder: GPUCommandEncoder,\n    inputBuffers: RayTraceBuffers,\n    layers: boolean\n  ) {\n    const pipeline = layers ? this._layersPipeline : this._pipeline;\n    const textures = layers\n      ? [this._foregroundTexture, this._catcherBackgroundTexture]\n      : [this._colorTexture, this._normalTexture, this._albedoTexture, ...(this._hasCatcher ? [this._shadowCatcherTexture] : [])];\n    const pass = encoder.beginRenderPass({\n      colorAttachments: textures.map((texture) => ({\n        view: texture.createView(),\n        loadValue: { r: 0, g: 0, b: 0, a: 0 },\n        storeOp: 'store',\n        loadOp: 'clear'\n      }))\n    });\n    pass.setPipeline(pipeline);\n\n    pass.setBindGroup(\n      0,\n      this._device.createBindGroup({\n        label: 'rayTraceOutput',\n        layout: pipeline.getBindGroupLayout(0),\n        entries: [\n          inputBuffers.color,\n          inputBuffers.normal,\n          inputBuffers.albedo,\n          inputBuffers.shadowCatcher\n        ].map((buffer, idx) => ({\n          binding: idx,\n          resource: {\n            buffer,\n            ...(idx === 3 ? { offset: inputBuffers.shadowCatcherOffset ?? 0 } : {})\n          }\n        })).filter(entry => entry.binding < 3 || this._hasCatcher || layers)\n      })\n    );\n    pass.setBindGroup(1, layers ? this._device.createBindGroup({\n      layout: pipeline.getBindGroupLayout(1),\n      entries: Object.values(this._uniformsBuffers).map((buffer, binding) => ({\n        binding, resource: { buffer }\n      }))\n    }) : this._uniformBindGroup);\n\n    pass.draw(6, 1, 0, 0);\n    pass.end();\n\n  }\n\n  private _destroyTextures() {\n    if (!this._colorTexture) return;\n\n    // These textures are also OIDN inputs. A resolution or preview-mode\n    // change can replace them immediately after a denoising command buffer\n    // was submitted. Keep the old resources alive until all work submitted so\n    // far has completed; destroying them synchronously can make Dawn validate\n    // the pending submit against an already-destroyed texture.\n    const textures = [\n      this._colorTexture,\n      this._normalTexture,\n      this._albedoTexture,\n      this._shadowCatcherTexture,\n      this._foregroundTexture,\n      this._catcherBackgroundTexture\n    ].filter((texture): texture is GPUTexture => !!texture && !this._retainedOutputs.has(texture));\n    void this._device.queue\n      .onSubmittedWorkDone()\n      .catch(() => undefined)\n      .then(() => textures.forEach((texture) => texture.destroy()));\n  }\n\n  getMemoryBytes() {\n    const textures = ['color', 'normal', 'albedo', 'shadowCatcher', 'foreground', 'catcherBackground'] as const;\n    return Object.values(this._uniformsBuffers).reduce((sum, buffer) => sum + buffer.size, 0)\n      + Array.from(new Set([...textures.map(label => this[`_${label}Texture`]), ...this._retainedOutputs]))\n        .reduce((sum, texture) => sum + textureBytes(texture), 0);\n  }\n\n  dispose() {\n    const retained = Array.from(this._retainedOutputs);\n    void this._device.queue.onSubmittedWorkDone().catch(() => undefined).then(() => retained.forEach(texture => texture.destroy()));\n    this._destroyTextures();\n    this._retainedOutputs.clear();\n    Object.keys(this._uniformsBuffers).forEach((key) => {\n      (this._uniformsBuffers as any)[key].destroy();\n    });\n  }\n}\n"]}

package/lib/renderer/WGPURayTracePass.js

@@ -188,7 +188,7 @@ export function packAnalyticsLightsToBuffer(lights, texturesIndices, selectionPm
         }
         data[(off + 3) * 4 + 3] = light.type === 'SpotLight'
             ? iesProfileTexelOffsets[i] ?? -1
-            : texturesIndices[i];
+            : texturesIndices[i] ?? -1;
         data[(off + 4) * 4] = selectionPmfs[i] ?? 0;
         packVec3(off + 5, getRigidLightFrame(light).x);
     }
@@ -1802,7 +1802,7 @@ ${rayTraceCompute({
         for (const geometry of this._geometriesNeedRefit) {
             const usesDisplacement = meshes.some((mesh) => mesh.geometry === geometry &&
                 !!mesh.materials[0]?.displacementMap &&
-                Math.abs(mesh.materials[0].displacementScale) > 1e-6);
+                (mesh.materials[0].displacementScale !== 0 || mesh.materials[0].displacementBias !== 0));
             if (usesDisplacement || geometryTopologyChanged(this._preparedMeshData, geometry) || !getGeometryBvh(geometry) || !refitBLAS(getGeometryBvh(geometry), geometry)) {
                 setGeometryBvh(geometry, undefined);
             }