@meimu/mutracer 0.1.2 → 0.1.4

package/lib/renderer/WGPURayTraceOutput.js

@@ -16,33 +16,33 @@ ${shadowCatcherAccumulationWGSL}
 override outputWidth: f32;
 override outputHeight: f32;
 
-struct oQ {
+struct pd {
 ${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) fG: vec4f) -> pd {
 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;
+let fP = vec2u(u32(fG.x), size.y - u32(fG.y) - 1u);
+var idx = fP.x + fP.y * size.x;
 
 
 
 
 if (sampleCount < 1.0 && normalBuffer[idx].a <= 0.0) {
-let uQ = max(sampleTileGrid, vec2u(1u));
-let uA = min(
-((fM + uQ / 2u) / uQ) * uQ,
-((size - vec2u(1u)) / uQ) * uQ
+let vc = max(sampleTileGrid, vec2u(1u));
+let uM = min(
+((fP + vc / 2u) / vc) * vc,
+((size - vec2u(1u)) / vc) * vc
 );
-let sr = uA.x + uA.y * size.x;
-if (normalBuffer[sr].a > 0.0) {
-idx = sr;
+let sH = uM.x + uM.y * size.x;
+if (normalBuffer[sH].a > 0.0) {
+idx = sH;
 }
 }
-var output: oQ;
+var output: pd;
 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 iV = max(hit - shadow.mask.b, 0.0);
+
+
 output.foreground = vec4f(
 (color.rgb - shadow.background.rgb)
-/ select(total, iP, iP > 0.0),
+/ select(total, iV, iV > 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.d.ts

@@ -57,6 +57,8 @@ export declare function makeRayTraceUniform(): {
         sequenceSeed: Uint32Array<ArrayBuffer>;
         zSobolLogSPP: Uint32Array<ArrayBuffer>;
         sampleSequenceOffset: Uint32Array<ArrayBuffer>;
+        environmentRotation: Float32Array<ArrayBuffer>;
+        backgroundMapTransform: Float32Array<ArrayBuffer>;
     };
 };
 export declare function packAnalyticsLightsToBuffer(lights: AnalyticLight[], texturesIndices: number[], selectionPmfs: number[], iesProfileTexelOffsets?: ArrayLike<number>): Float32Array<ArrayBuffer>;

package/lib/renderer/WGPURayTracePass.js

@@ -24,7 +24,7 @@ import { WGPU_SCENE_FEATURE_BITS, wgpuSceneFeatureFlags } from './WGPUSceneFeatu
 import { mat4 } from '../math.js';
 import MaterialTable, { DEFAULT_MAX_TEXTURE_ARRAY_BYTES, materialUsesMedium } from './MaterialTable.js';
 import { numberArraysEqual } from './util.js';
-import { generateBackgroundMapFromSceneBackground } from './envMapCreation.js';
+import { generateBackgroundMapFromSceneBackground, environmentRotation, backgroundMapTransform } from './envMapCreation.js';
 import rayTraceCompute, { WORKGROUP_SIZE } from './wgsl/rayTrace.compute.js';
 import { CONSTANT_DIRECTIONAL_LIGHT, CONSTANT_RECT_LIGHT, CONSTANT_SPHERE_LIGHT, CONSTANT_SPOT_LIGHT } from './lightsUtil.js';
 import { envMapDistribution, environmentMapSourcesMatch } from './envMapDistribution.js';
@@ -89,7 +89,7 @@ export function isWebGPUPipelineSpecializationFailure(...errors) {
 /** CPU-side views over the `RayTraceUniform` WGSL struct. */
 export function makeRayTraceUniform() {
     // Created from https://webgpufundamentals.org/webgpu/lessons/resources/wgsl-offset-computer.html
-    const RayTraceUniformValues = new ArrayBuffer(176);
+    const RayTraceUniformValues = new ArrayBuffer(256);
     const RayTraceUniformViews = {
         outputSize: new Float32Array(RayTraceUniformValues, 0, 2),
         sampleCount: new Float32Array(RayTraceUniformValues, 8, 1),
@@ -119,11 +119,15 @@ export function makeRayTraceUniform() {
         sceneFeatureFlags: new Uint32Array(RayTraceUniformValues, 152, 1),
         sequenceSeed: new Uint32Array(RayTraceUniformValues, 156, 1),
         zSobolLogSPP: new Uint32Array(RayTraceUniformValues, 160, 1),
-        sampleSequenceOffset: new Uint32Array(RayTraceUniformValues, 164, 1)
+        sampleSequenceOffset: new Uint32Array(RayTraceUniformValues, 164, 1),
+        environmentRotation: new Float32Array(RayTraceUniformValues, 176, 16),
+        backgroundMapTransform: new Float32Array(RayTraceUniformValues, 240, 4)
     };
     // Fixed runtime trip bounds discourage thin-film loop unrolling. Keep 4 for
     // 2 polarizations/substrates and 3 channels, independent of BSDF budgets.
     RayTraceUniformViews.loopUnrollGuard[0] = 4;
+    RayTraceUniformViews.environmentRotation.set(environmentRotation([]));
+    RayTraceUniformViews.backgroundMapTransform.set(backgroundMapTransform());
     RayTraceUniformViews.bevel[0] = 1;
     RayTraceUniformViews.dispersionWavelengths[0] = 3;
     return {
@@ -179,7 +183,7 @@ export function packAnalyticsLightsToBuffer(lights, texturesIndices, selectionPm
             data[(off + 2) * 4 + 3] = light.radius;
             packVec3(off + 3, [
                 Math.cos((light.outerAngle * Math.PI) / 180),
-                Math.min(1, Math.max(0, 1 - light.innerAngle / light.outerAngle)),
+                Math.cos((Math.min(light.innerAngle, light.outerAngle) * Math.PI) / 180),
                 0
             ]);
         }
@@ -188,7 +192,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);
     }
@@ -792,8 +796,6 @@ class WGPURayTracePass {
                 ...MATERIAL_TEXTURE_DEFINES,
                 'USE_ANISOTROPY_DIRECTIONS',
                 'USE_COAT_NORMAL_FRAMES',
-                'COLOR_OFF',
-                'USE_VERTEX_COLOR',
                 'HAS_ALPHA_BLEND',
                 'HAS_ALPHA_MASK'
             ];
@@ -1621,6 +1623,7 @@ ${rayTraceCompute({
         return true;
     }
     setBackground(background) {
+        this._rtUniform.views.backgroundMapTransform.set(backgroundMapTransform(background));
         this._constants.USE_BACKGROUND = !!background;
         const backgroundImage = background && !Array.isArray(background)
             ? generateBackgroundMapFromSceneBackground(background)
@@ -2119,6 +2122,8 @@ ${rayTraceCompute({
         for (let i = 0; i < mapUvChannel.length; i++) {
             mapInfoData[numMaps * 8 + i * 4] = mapUvChannel[i];
             mapInfoData[numMaps * 8 + i * 4 + 1] = mapColorLinear[i];
+            mapInfoData[numMaps * 8 + i * 4 + 2] = materialData.bufferData.mapWrap[i * 2];
+            mapInfoData[numMaps * 8 + i * 4 + 3] = materialData.bufferData.mapWrap[i * 2 + 1];
         }
         let mapInfoUniform = this._mapInfoUniform;
         if (!mapInfoUniform || mapInfoUniform.size !== mapInfoData.byteLength) {
@@ -2186,6 +2191,7 @@ ${rayTraceCompute({
         const analyticLights = allAnalyticLights.slice(0, FIXED_LIGHT_COUNT);
         const numAnalyticLights = analyticLights.length;
         const envImage = this._sceneEnvironment.update(lights);
+        this._rtUniform.views.environmentRotation.set(environmentRotation(lights));
         this._rtUniform.views.lightsCount[0] = numAnalyticLights;
         this._setEnvMap(envImage);
         const lightTexturesChanged = this._rectLightTextures.update(analyticLights);