@meimu/mutracer 0.1.4 → 0.2.0

package/lib/renderer/WGPURayTraceOutput.js

@@ -1,9 +1,11 @@
+import { hdrResolveWGSL } from './wgsl/compose.frag.wgsl.js';
 import { getTargetShaderCache } from './WGPUShaderCache.js';
 import { textureBytes } from './gpuMemory.js';
 import shadowCatcherAccumulationWGSL from './wgsl/chunks/shadowCatcherAccumulation.wgsl.js';
 import { fullScreenQuadVertexShaderWGSL } from './WGPUFullQuadPass.js';
 export function createNormalizeRayTraceOutputWGSL(layers = false, catcher = true) {
     return /* wgsl */ `
+${hdrResolveWGSL}
 @group(0) @binding(0) var<storage, read> colorBuffer: array<vec4f>;
 @group(0) @binding(1) var<storage, read> normalBuffer: array<vec4f>;
 @group(0) @binding(2) var<storage, read> albedoBuffer: array<vec4f>;
@@ -16,33 +18,33 @@ ${shadowCatcherAccumulationWGSL}
 override outputWidth: f32;
 override outputHeight: f32;
 
-struct pd {
+struct qa {
 ${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) fG: vec4f) -> pd {
+fn main(@builtin(position) coord: vec4f) -> qa {
 let size = vec2u(u32(outputWidth), u32(outputHeight));
-let fP = vec2u(u32(fG.x), size.y - u32(fG.y) - 1u);
-var idx = fP.x + fP.y * size.x;
+let pixel = vec2u(u32(coord.x), size.y - u32(coord.y) - 1u);
+var idx = pixel.x + pixel.y * size.x;
 
 
 
 
 if (sampleCount < 1.0 && normalBuffer[idx].a <= 0.0) {
-let vc = max(sampleTileGrid, vec2u(1u));
-let uM = min(
-((fP + vc / 2u) / vc) * vc,
-((size - vec2u(1u)) / vc) * vc
+let wU = max(sampleTileGrid, vec2u(1u));
+let wq = min(
+((pixel + wU / 2u) / wU) * wU,
+((size - vec2u(1u)) / wU) * wU
 );
-let sH = uM.x + uM.y * size.x;
-if (normalBuffer[sH].a > 0.0) {
-idx = sH;
+let tW = wq.x + wq.y * size.x;
+if (normalBuffer[tW].a > 0.0) {
+idx = tW;
 }
 }
-var output: pd;
+var output: qa;
 if (normalBuffer[idx].a <= 0.0) {
 ${layers ? `output.foreground = vec4f(0.0);
 output.catcherBackground = vec4f(0.0);` : `output.color = vec4f(0.0);
@@ -62,20 +64,22 @@ let hit = albedo.a;
 let total = max(normal.a, 1.0);
 ${layers ? `
 let shadow = shadowCatcherBuffer[idx];
-let iV = max(hit - shadow.mask.b, 0.0);
+let jo = max(hit - shadow.mask.b, 0.0);
 
 
 output.foreground = vec4f(
 (color.rgb - shadow.background.rgb)
-/ select(total, iV, iV > 0.0),
+/ select(total, jo, jo > 0.0),
 clamp((color.a - shadow.mask.g) / total, 0.0, 1.0)
 );
 output.catcherBackground = shadow.background / total;
 ` : `output.color = vec4f(
 
 
-color.rgb / select(total, hit, hit > 0.0),
-color.a / total
+vec3f(fK(color.r,normal.a,hit,1.0,true),
+fK(color.g,normal.a,hit,1.0,true),
+fK(color.b,normal.a,hit,1.0,true)),
+fK(color.a,normal.a,hit,1.0,false)
 );
 
 output.normal = vec4f(0.0);
@@ -84,7 +88,12 @@ if (dot(normal.rgb, normal.rgb) > 0.1) {
 output.normal = vec4f(normal.rgb / total * 0.5 + 0.5, 1.0);
 }
 output.albedo = vec4f(albedo.rgb / total, clamp(hit / total, 0.0, 1.0));
-${catcher ? 'output.shadowCatcher = shadowCatcherBuffer[idx].mask / total;' : ''}
+${catcher ? `let shadow = shadowCatcherBuffer[idx];
+
+
+
+output.shadowCatcher = vec4f(shadow.mask.rgb / total,
+shadow.mask.r / total * select(0.0, shadow.moments.z / shadow.moments.w, shadow.moments.w > 0.0));` : ''}
 `}
 return output;
 }

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;;;;;;;;;;;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"]}
+{"version":3,"file":"WGPURayTraceOutput.js","sourceRoot":"","sources":["../../src/renderer/WGPURayTraceOutput.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,cAAc,EAAC,MAAM,6BAA6B,CAAC;AAC3D,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;EAClB,cAAc;;;;EAId,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;;;;;;;;;;;;;;;EAeF,OAAO,CAAC,CAAC,CAAC;;;;;mGAKyF,CAAC,CAAC,CAAC,EAAE;CACzG;;;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 {hdrResolveWGSL} from './wgsl/compose.frag.wgsl.js';\nimport { 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${hdrResolveWGSL}\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    vec3f(hdr_resolve_component(color.r,normal.a,hit,1.0,true),\n      hdr_resolve_component(color.g,normal.a,hit,1.0,true),\n      hdr_resolve_component(color.b,normal.a,hit,1.0,true)),\n    hdr_resolve_component(color.a,normal.a,hit,1.0,false)\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 ? `let shadow = shadowCatcherBuffer[idx];\n  // Opacity is the ratio of the summed occluded and total light sample\n  // counts; a per-sample ratio average would keep its low-sample bias. Alpha\n  // carries it scaled by coverage, as composition expects.\n  output.shadowCatcher = vec4f(shadow.mask.rgb / total,\n    shadow.mask.r / total * select(0.0, shadow.moments.z / shadow.moments.w, shadow.moments.w > 0.0));` : ''}\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.d.ts

@@ -1,11 +1,14 @@
-import { type SamplingPattern } from './sampling.js';
+import { WGPUAov } from './WGPUAov.js';
+import { type AovKind } from './aov.js';
+import { type MaterialTextureSelection } from './wgpuBindless.js';
+import type { SamplingPattern } from './sampling.js';
 import { type DepthSampleArea } from './depthQuery.js';
 import Mesh from '../scene/Mesh.js';
 import DirectionalLight from '../scene/DirectionalLight.js';
 import SphereLight from '../scene/SphereLight.js';
 import RectLight from '../scene/RectLight.js';
 import SpotLight from '../scene/SpotLight.js';
-import { SceneBackground, SceneLight } from '../scene/Scene.js';
+import type { SceneBackground, SceneLight } from '../scene/Scene.js';
 import Camera from '../scene/Camera.js';
 import Geometry from '../scene/Geometry.js';
 import OpenPBRMaterial from '../scene/OpenPBRMaterial.js';
@@ -25,44 +28,11 @@ export declare function getWebGPUErrorMessage(error: unknown): string;
 export declare function shaderSourceFingerprint(source: string): string;
 export declare function isWebGPUPipelineSpecializationFailure(...errors: unknown[]): boolean;
 /** CPU-side views over the `RayTraceUniform` WGSL struct. */
-export declare function makeRayTraceUniform(): {
-    buffer: ArrayBuffer;
-    views: {
-        outputSize: Float32Array<ArrayBuffer>;
-        sampleCount: Float32Array<ArrayBuffer>;
-        scaleOffset: Float32Array<ArrayBuffer>;
-        lightsCount: Int32Array<ArrayBuffer>;
-        transmissiveOutputAlphaMin: Float32Array<ArrayBuffer>;
-        jitter: Float32Array<ArrayBuffer>;
-        backgroundColor: Float32Array<ArrayBuffer>;
-        groundedBackgroundParams: Float32Array<ArrayBuffer>;
-        groundedBackgroundCenter: Float32Array<ArrayBuffer>;
-        blasOffset: Int32Array<ArrayBuffer>;
-        isEmpty: Int32Array<ArrayBuffer>;
-        bevel: Int32Array<ArrayBuffer>;
-        dispersionWavelengths: Int32Array<ArrayBuffer>;
-        envLightPmf: Float32Array<ArrayBuffer>;
-        lightChoiceCount: Int32Array<ArrayBuffer>;
-        lightWeightTotal: Float32Array<ArrayBuffer>;
-        lightSamplingTextureWidth: Int32Array<ArrayBuffer>;
-        lightChoiceTexelOffset: Int32Array<ArrayBuffer>;
-        bounces: Int32Array<ArrayBuffer>;
-        varianceMoments: Int32Array<ArrayBuffer>;
-        adaptiveSampling: Int32Array<ArrayBuffer>;
-        clamping: Float32Array<ArrayBuffer>;
-        directLightCandidates: Int32Array<ArrayBuffer>;
-        backgroundFlags: Int32Array<ArrayBuffer>;
-        loopUnrollGuard: Uint32Array<ArrayBuffer>;
-        sceneFeatureFlags: Uint32Array<ArrayBuffer>;
-        sequenceSeed: Uint32Array<ArrayBuffer>;
-        zSobolLogSPP: Uint32Array<ArrayBuffer>;
-        sampleSequenceOffset: Uint32Array<ArrayBuffer>;
-        environmentRotation: Float32Array<ArrayBuffer>;
-        backgroundMapTransform: Float32Array<ArrayBuffer>;
-    };
-};
+export { makeRayTraceUniform } from './wgpuSceneUniform.js';
 export declare function packAnalyticsLightsToBuffer(lights: AnalyticLight[], texturesIndices: number[], selectionPmfs: number[], iesProfileTexelOffsets?: ArrayLike<number>): Float32Array<ArrayBuffer>;
 declare class WGPURayTracePass {
+    private _textureSelection;
+    private _aovPrimitiveCache;
     private _device;
     private _rayTracePipeline?;
     private _targetAspect;
@@ -76,6 +46,9 @@ declare class WGPURayTracePass {
     private _materialChanges;
     private _pendingMaterialUpdate?;
     private _materialData;
+    private _bindless?;
+    private _retiredBindless;
+    private _disposed;
     private _maxTextureArrayBytes;
     private _pendingGeometryUpdates;
     private _needsUpdateMeshData;
@@ -118,6 +91,7 @@ declare class WGPURayTracePass {
     private _instancesBuffer?;
     private _emissiveCache;
     private _lightChoices;
+    private _shadowCatcherLightChoices;
     private _lightSamplingData?;
     private _lightSamplingScratch?;
     private _lightSamplingTriangleCount;
@@ -133,6 +107,8 @@ declare class WGPURayTracePass {
     private _envImageSource;
     private _envTex?;
     private _envDistributionTex?;
+    private _envDistribution?;
+    private _skyBaker?;
     private _backgroundImageSource;
     private _backgroundTex?;
     private _rayTracePipelineLayout?;
@@ -145,8 +121,12 @@ declare class WGPURayTracePass {
     private _compiledSceneHasAlpha;
     private _alphaBlendStrategy;
     private _constants;
+    private _retainedShaderCapabilities?;
+    private _retainedShaderStructure?;
+    private _shaderCapabilities;
     private _prevPipelineConstants;
     private _prevDefinesCode;
+    private _prevPipelineInputs?;
     private _hasTransmission;
     private _hasShadowCatcher;
     private _hasDisplacement;
@@ -161,6 +141,7 @@ declare class WGPURayTracePass {
     private _focusDistance;
     private _needsUpdateBindGroups;
     private _computeShaderModule?;
+    private _aovPipelines;
     private _depthCompute?;
     private _depthTracePipeline?;
     private _depthPipeline?;
@@ -171,7 +152,7 @@ declare class WGPURayTracePass {
     private _pipelineError;
     private _bakePipelineConstants;
     private _failedPipelineSignature?;
-    constructor(device: GPUDevice);
+    constructor(device: GPUDevice, _textureSelection?: MaterialTextureSelection);
     getOutputBuffer(): {
         color: GPUBuffer;
         normal: GPUBuffer;
@@ -189,6 +170,9 @@ declare class WGPURayTracePass {
     private _updateGPUBuffer;
     private _uniformBuffer;
     updatePipeline(): Promise<void>;
+    private _prepareAovPipelines;
+    prepareAovs(kinds: readonly AovKind[]): boolean;
+    renderAovs(state: WGPUAov, camera: Camera, samples: number, cap: number, kinds: readonly AovKind[]): boolean;
     readDepthSamples(area: DepthSampleArea, aspect?: number): Promise<Float32Array | undefined>;
     isCompilingShader(): boolean;
     isPipelineReady(): boolean;
@@ -199,7 +183,18 @@ declare class WGPURayTracePass {
         buffer: ArrayBuffer;
     }>;
     getDiagnostics(): {
-        textureAtlas: import("./MaterialTable.js").TextureAtlasStats | undefined;
+        textureAtlas: import("./textureAtlas.js").TextureAtlasStats | undefined;
+        materialTextures: {
+            metadataBytes: number;
+            resourceTableCapacity: number;
+            retiredBytes: number;
+            textureCount: number;
+            estimatedBytes: number;
+            scale: number;
+            requested: "auto" | "atlas";
+            active: "atlas" | "bindless";
+            fallbackReason: string | undefined;
+        };
         shaderCompilation: ShaderCompilationDiagnostics | undefined;
         lightSampling: {
             emissiveTriangles: number;
@@ -269,6 +264,7 @@ declare class WGPURayTracePass {
     private _updateMeshMembership;
     private _updateMeshColors;
     private _refitMeshInstances;
+    private _createMaterialData;
     private _updateTextures;
     setLights(lights: SceneLight[]): void;
     private _updateViewProjInv;