@meimu/mutracer 0.0.0-stage → 0.1.1

package/lib/renderer/wgsl/compose.frag.wgsl.js

@@ -0,0 +1,218 @@
+export const linearToSrgbDefine = /* wgsl */ `
+fn fJ(color: vec3f) -> vec3f {
+let linear = max(color, vec3f(0.0));
+let low = linear * 12.92;
+let high = 1.055 * pow(linear, vec3f(1.0 / 2.4)) - 0.055;
+return select(high, low, linear <= vec3f(0.0031308));
+}
+`;
+export const composeDefine = /* wgsl */ `
+
+fn jL(color: vec3f) -> vec3f {
+return clamp(color, vec3f(0.0), vec3f(1.0));
+}
+fn iJ(color: vec3f) -> vec3f {
+return clamp(color / (vec3f(1.0) + color), vec3f(0.0), vec3f(1.0));
+}
+fn jC(jk: vec3f) -> vec3f {
+let color = max(vec3f(0.0), jk - 0.004);
+return pow((color * (6.2 * color + 0.5)) /
+(color * (6.2 * color + 1.7) + 0.06), vec3f(2.2));
+}
+fn ox(value: vec3f) -> vec3f {
+let kC = value * (value + 0.0245786) - 0.000090537;
+let ap = value * (0.983729 * value + 0.4329510) + 0.238081;
+return kC / ap;
+}
+fn lh(jk: vec3f) -> vec3f {
+let pj = mat3x3f(
+vec3f(0.59719, 0.07600, 0.02840),
+vec3f(0.35458, 0.90834, 0.13383),
+vec3f(0.04823, 0.01566, 0.83777)
+);
+let os = mat3x3f(
+vec3f(1.60475, -0.10208, -0.00327),
+vec3f(-0.53108, 1.10813, -0.07276),
+vec3f(-0.07367, -0.00605, 1.07602)
+);
+let color = pj * (jk / 0.6);
+return clamp(os * ox(color), vec3f(0.0), vec3f(1.0));
+}
+fn fP(value: vec3f) -> vec3f {
+let mB = value * value;
+let qy = mB * mB;
+return 15.5 * qy * mB - 40.14 * qy * value
++ 31.96 * qy - 6.868 * mB * value
++ 0.4298 * mB + 0.1191 * value - 0.00232;
+}
+fn lP(jk: vec3f) -> vec3f {
+let mD = mat3x3f(
+vec3f(1.6605, -0.1246, -0.0182),
+vec3f(-0.5876, 1.1329, -0.1006),
+vec3f(-0.0728, -0.0083, 1.1187)
+);
+let mE = mat3x3f(
+vec3f(0.6274, 0.0691, 0.0164),
+vec3f(0.3293, 0.9195, 0.0880),
+vec3f(0.0433, 0.0113, 0.8956)
+);
+let pl = mat3x3f(
+vec3f(0.856627153315983, 0.137318972929847, 0.11189821299995),
+vec3f(0.0951212405381588, 0.761241990602591, 0.0767994186031903),
+vec3f(0.0482516061458583, 0.101439036467562, 0.811302368396859)
+);
+let ot = mat3x3f(
+vec3f(1.1271005818144368, -0.1413297634984383, -0.14132976349843826),
+vec3f(-0.11060664309660323, 1.157823702216272, -0.11060664309660294),
+vec3f(-0.016493938717834573, -0.016493938717834257, 1.2519364065950405)
+);
+var color = pl * (mE * jk);
+color = clamp((log2(max(color, vec3f(1e-10))) + 12.47393) / 16.5, vec3f(0.0), vec3f(1.0));
+color = ot * fP(color);
+color = pow(max(color, vec3f(0.0)), vec3f(2.2));
+return clamp(mD * color, vec3f(0.0), vec3f(1.0));
+}
+fn cH(color: vec3f) -> vec3f {
+let toneMappingMode = i32(toneMappingParams.x);
+if (toneMappingMode == 1) { return lh(color); }
+if (toneMappingMode == 2) { return jC(color); }
+if (toneMappingMode == 3) { return iJ(color); }
+if (toneMappingMode == 4) { return jL(color); }
+return lP(color);
+}
+${linearToSrgbDefine}
+fn gS(jk: vec3f) -> vec3f {
+let contrast = colorGradingParams.x;
+let saturation = colorGradingParams.y;
+let temperature = colorGradingParams.z;
+let tint = colorGradingParams.w;
+let sW = vec3f(
+1.0 + temperature * 0.16 + tint * 0.04,
+1.0 - tint * 0.12,
+1.0 - temperature * 0.16 + tint * 0.04
+);
+var color = max(jk * sW, vec3f(0.0));
+color = max((color - 0.18) * contrast + 0.18, vec3f(0.0));
+let luminance = dot(color, vec3f(0.2126, 0.7152, 0.0722));
+return max(mix(vec3f(luminance), color, saturation), vec3f(0.0));
+}
+`;
+export const composeMain = /* wgsl */ `
+
+var dY = load_colorTex(uv, 0);
+var jf = vec4(0.0);
+if (bloomIntensity > 0.0) {
+jf = load_bloomTex(uv, 0) * bloomIntensity;
+dY = vec4f(dY.rgb + jf.rgb, dY.a);
+}
+
+var kX = gS(dY.rgb * exposure);
+let kj = smoothstep(0.2, 0.72, distance(vec2f(uv) / vec2f(colorTexRegion()), vec2f(0.5)));
+kX *= 1.0 - toneMappingParams.y * kj;
+let hx = hdrOutputEnabled > 0.5;
+if (!hx) {
+kX = cH(kX);
+}
+let lX = select(
+pow(max(kX, vec3f(0.0)), vec3f(1.0 / 2.2)),
+fJ(kX),
+hx
+);
+var nO = lX;
+if (flatBackgroundColor.a > 0.5) {
+let ja = clamp(
+load_flatBackgroundMaskTex(uv, 0).a,
+0.0,
+1.0
+);
+let oO = fJ(max(flatBackgroundColor.rgb, vec3f(0.0)));
+nO = mix(oO, lX, ja);
+if (bloomIntensity > 0.0) {
+var il = gS(
+max(dY.rgb - jf.rgb, vec3f(0.0))
+* exposure
+);
+il *= 1.0 - toneMappingParams.y * kj;
+if (!hx) {
+il = cH(il);
+}
+let rp = select(
+pow(max(il, vec3f(0.0)), vec3f(1.0 / 2.2)),
+fJ(il),
+hx
+);
+nO += max(lX - rp, vec3f(0.0))
+* (1.0 - ja);
+}
+}
+dY = vec4f(nO, dY.a);
+if (bloomIntensity > 0.0) {
+dY.a = max(dY.a, min(dot(jf.rgb, vec3f(0.2125, 0.7154, 0.0721)), 1.0));
+}
+
+if (shadowCatcherEnabled > 0.5) {
+let shadowCatcher = load_shadowCatcherTex(uv, 0);
+let coverage = clamp(shadowCatcher.r, 0.0, 1.0);
+let opacity = max(clamp(shadowCatcher.a / max(coverage, 0.000001), 0.0, 1.0) - (1.0 - shadowCatcherAlpha), 0.0);
+if (coverage > 0.0) {
+let foreground = load_foregroundTex(uv, 0);
+let background = load_catcherBackgroundTex(uv, 0);
+var bK = gS((foreground.rgb + jf.rgb) * exposure);
+var backgroundColor = gS((background.rgb / max(background.a, 0.000001) + jf.rgb) * exposure);
+bK *= 1.0 - toneMappingParams.y * kj;
+backgroundColor *= 1.0 - toneMappingParams.y * kj;
+if (!hx) {
+bK = cH(bK);
+backgroundColor = cH(backgroundColor);
+}
+bK = select(pow(max(bK, vec3f(0.0)), vec3f(1.0 / 2.2)), fJ(bK), hx);
+backgroundColor = select(pow(max(backgroundColor, vec3f(0.0)), vec3f(1.0 / 2.2)), fJ(backgroundColor), hx);
+if (flatBackgroundColor.a > 0.5) {
+var eM = gS(background.rgb / max(background.a, 0.000001) * exposure);
+eM *= 1.0 - toneMappingParams.y * kj;
+if (!hx) { eM = cH(eM); }
+let lY = select(pow(max(eM, vec3f(0.0)), vec3f(1.0 / 2.2)), fJ(eM), hx);
+let tX = select(vec3f(0.0), max(backgroundColor - lY, vec3f(0.0)), bloomIntensity > 0.0);
+backgroundColor = fJ(max(flatBackgroundColor.rgb, vec3f(0.0))) + tX;
+}
+
+
+let fO = coverage * opacity;
+let lR = clamp(dot(jf.rgb, vec3f(0.2125, 0.7154, 0.0721)), 0.0, 1.0);
+let hr = max(foreground.a, lR * (1.0 - coverage));
+let hg = max(background.a, lR * coverage);
+let alpha = hr + hg * (1.0 - opacity) + fO;
+let premultiplied = bK * hr
++ backgroundColor * hg * (1.0 - opacity)
++ shadowCatcherColor * fO;
+dY = vec4f(premultiplied / max(alpha, 0.000001), alpha);
+}
+}
+
+output.color = dY;
+
+
+`;
+// Preserve the existing display composition, but give OIDN a floating-point,
+// linear-light image. Alpha is deliberately not used as a second color layer.
+export const compositeDenoiseOutputMain = /* wgsl */ `
+let encoded = max(dY.rgb, vec3f(0.0));
+let sK = select(
+pow(encoded, vec3f(2.2)),
+select(pow((encoded + 0.055) / 1.055, vec3f(2.4)), encoded / 12.92,
+encoded <= vec3f(0.04045)),
+hx
+);
+output.denoiseColor = vec4f(sK, dY.a);
+`;
+export const compositeDenoiseDisplayMain = /* wgsl */ `
+let denoised = load_colorTex(uv, 0);
+
+
+
+let alpha = select(load_composedTex(uv, 0).a, clamp(denoised.a, 0.0, 1.0), denoisedAlpha > 0.5);
+let linear = max(denoised.rgb, vec3f(0.0));
+let encoded = select(pow(linear, vec3f(1.0 / 2.2)), fJ(linear), hdrOutputEnabled > 0.5);
+output.color = vec4f(select(vec3f(0.0), encoded, alpha > 0.0), alpha);
+`;
+//# sourceMappingURL=compose.frag.wgsl.js.map

package/lib/renderer/wgsl/compose.frag.wgsl.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"compose.frag.wgsl.js","sourceRoot":"","sources":["../../../src/renderer/wgsl/compose.frag.wgsl.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,kBAAkB,GAAG,UAAU,CAAC;;;;;;;CAO5C,CAAC;AAEF,MAAM,CAAC,MAAM,aAAa,GAAG,UAAU,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EA0EtC,kBAAkB;;;;;;;;;;;;;;;;CAgBnB,CAAC;AAEF,MAAM,CAAC,MAAM,WAAW,GAAG,UAAU,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA+FrC,CAAC;AAEF,6EAA6E;AAC7E,8EAA8E;AAC9E,MAAM,CAAC,MAAM,0BAA0B,GAAG,UAAU,CAAC;;;;;;;;;CASpD,CAAC;AAEF,MAAM,CAAC,MAAM,2BAA2B,GAAG,UAAU,CAAC;;;;;;;;;CASrD,CAAC","sourcesContent":["export const linearToSrgbDefine = /* wgsl */ `\nfn linearToSrgb(color: vec3f) -> vec3f {\n  let linear = max(color, vec3f(0.0));\n  let low = linear * 12.92;\n  let high = 1.055 * pow(linear, vec3f(1.0 / 2.4)) - 0.055;\n  return select(high, low, linear <= vec3f(0.0031308));\n}\n`;\n\nexport const composeDefine = /* wgsl */ `\n\nfn linearToneMapping(color: vec3f) -> vec3f {\n  return clamp(color, vec3f(0.0), vec3f(1.0));\n}\nfn reinhardToneMapping(color: vec3f) -> vec3f {\n  return clamp(color / (vec3f(1.0) + color), vec3f(0.0), vec3f(1.0));\n}\nfn cineonToneMapping(inputColor: vec3f) -> vec3f {\n  let color = max(vec3f(0.0), inputColor - 0.004);\n  return pow((color * (6.2 * color + 0.5)) /\n    (color * (6.2 * color + 1.7) + 0.06), vec3f(2.2));\n}\nfn rrtAndOdtFit(value: vec3f) -> vec3f {\n  let numerator = value * (value + 0.0245786) - 0.000090537;\n  let denominator = value * (0.983729 * value + 0.4329510) + 0.238081;\n  return numerator / denominator;\n}\nfn acesToneMapping(inputColor: vec3f) -> vec3f {\n  let inputMatrix = mat3x3f(\n    vec3f(0.59719, 0.07600, 0.02840),\n    vec3f(0.35458, 0.90834, 0.13383),\n    vec3f(0.04823, 0.01566, 0.83777)\n  );\n  let outputMatrix = mat3x3f(\n    vec3f(1.60475, -0.10208, -0.00327),\n    vec3f(-0.53108, 1.10813, -0.07276),\n    vec3f(-0.07367, -0.00605, 1.07602)\n  );\n  let color = inputMatrix * (inputColor / 0.6);\n  return clamp(outputMatrix * rrtAndOdtFit(color), vec3f(0.0), vec3f(1.0));\n}\nfn agxDefaultContrastApprox(value: vec3f) -> vec3f {\n  let value2 = value * value;\n  let value4 = value2 * value2;\n  return 15.5 * value4 * value2 - 40.14 * value4 * value\n    + 31.96 * value4 - 6.868 * value2 * value\n    + 0.4298 * value2 + 0.1191 * value - 0.00232;\n}\nfn agxToneMapping(inputColor: vec3f) -> vec3f {\n  let linearRec2020ToLinearSrgb = mat3x3f(\n    vec3f(1.6605, -0.1246, -0.0182),\n    vec3f(-0.5876, 1.1329, -0.1006),\n    vec3f(-0.0728, -0.0083, 1.1187)\n  );\n  let linearSrgbToLinearRec2020 = mat3x3f(\n    vec3f(0.6274, 0.0691, 0.0164),\n    vec3f(0.3293, 0.9195, 0.0880),\n    vec3f(0.0433, 0.0113, 0.8956)\n  );\n  let insetMatrix = mat3x3f(\n    vec3f(0.856627153315983, 0.137318972929847, 0.11189821299995),\n    vec3f(0.0951212405381588, 0.761241990602591, 0.0767994186031903),\n    vec3f(0.0482516061458583, 0.101439036467562, 0.811302368396859)\n  );\n  let outsetMatrix = mat3x3f(\n    vec3f(1.1271005818144368, -0.1413297634984383, -0.14132976349843826),\n    vec3f(-0.11060664309660323, 1.157823702216272, -0.11060664309660294),\n    vec3f(-0.016493938717834573, -0.016493938717834257, 1.2519364065950405)\n  );\n  var color = insetMatrix * (linearSrgbToLinearRec2020 * inputColor);\n  color = clamp((log2(max(color, vec3f(1e-10))) + 12.47393) / 16.5, vec3f(0.0), vec3f(1.0));\n  color = outsetMatrix * agxDefaultContrastApprox(color);\n  color = pow(max(color, vec3f(0.0)), vec3f(2.2));\n  return clamp(linearRec2020ToLinearSrgb * color, vec3f(0.0), vec3f(1.0));\n}\nfn applyToneMapping(color: vec3f) -> vec3f {\n  let toneMappingMode = i32(toneMappingParams.x);\n  if (toneMappingMode == 1) { return acesToneMapping(color); }\n  if (toneMappingMode == 2) { return cineonToneMapping(color); }\n  if (toneMappingMode == 3) { return reinhardToneMapping(color); }\n  if (toneMappingMode == 4) { return linearToneMapping(color); }\n  return agxToneMapping(color);\n}\n${linearToSrgbDefine}\nfn gradeColor(inputColor: vec3f) -> vec3f {\n  let contrast = colorGradingParams.x;\n  let saturation = colorGradingParams.y;\n  let temperature = colorGradingParams.z;\n  let tint = colorGradingParams.w;\n  let balance = vec3f(\n    1.0 + temperature * 0.16 + tint * 0.04,\n    1.0 - tint * 0.12,\n    1.0 - temperature * 0.16 + tint * 0.04\n  );\n  var color = max(inputColor * balance, vec3f(0.0));\n  color = max((color - 0.18) * contrast + 0.18, vec3f(0.0));\n  let luminance = dot(color, vec3f(0.2126, 0.7152, 0.0722));\n  return max(mix(vec3f(luminance), color, saturation), vec3f(0.0));\n}\n`;\n\nexport const composeMain = /* wgsl */ `\n\nvar finalColor = load_colorTex(uv, 0);\nvar bloomTexel = vec4(0.0);\nif (bloomIntensity > 0.0) {\n  bloomTexel = load_bloomTex(uv, 0) * bloomIntensity;\n  finalColor = vec4f(finalColor.rgb + bloomTexel.rgb, finalColor.a);\n}\n\nvar gradedColor = gradeColor(finalColor.rgb * exposure);\nlet vignetteEdge = smoothstep(0.2, 0.72, distance(vec2f(uv) / vec2f(colorTexRegion()), vec2f(0.5)));\ngradedColor *= 1.0 - toneMappingParams.y * vignetteEdge;\nlet hdrOutput = hdrOutputEnabled > 0.5;\nif (!hdrOutput) {\n  gradedColor = applyToneMapping(gradedColor);\n}\nlet encodedDisplay = select(\n  pow(max(gradedColor, vec3f(0.0)), vec3f(1.0 / 2.2)),\n  linearToSrgb(gradedColor),\n  hdrOutput\n);\nvar displayColor = encodedDisplay;\nif (flatBackgroundColor.a > 0.5) {\n  let surfaceCoverage = clamp(\n    load_flatBackgroundMaskTex(uv, 0).a,\n    0.0,\n    1.0\n  );\n  let flatBackgroundDisplay = linearToSrgb(max(flatBackgroundColor.rgb, vec3f(0.0)));\n  displayColor = mix(flatBackgroundDisplay, encodedDisplay, surfaceCoverage);\n  if (bloomIntensity > 0.0) {\n    var unbloomedColor = gradeColor(\n      max(finalColor.rgb - bloomTexel.rgb, vec3f(0.0))\n        * exposure\n    );\n    unbloomedColor *= 1.0 - toneMappingParams.y * vignetteEdge;\n    if (!hdrOutput) {\n      unbloomedColor = applyToneMapping(unbloomedColor);\n    }\n    let encodedUnbloomed = select(\n      pow(max(unbloomedColor, vec3f(0.0)), vec3f(1.0 / 2.2)),\n      linearToSrgb(unbloomedColor),\n      hdrOutput\n    );\n    displayColor += max(encodedDisplay - encodedUnbloomed, vec3f(0.0))\n      * (1.0 - surfaceCoverage);\n  }\n}\nfinalColor = vec4f(displayColor, finalColor.a);\nif (bloomIntensity > 0.0) {\n  finalColor.a = max(finalColor.a, min(dot(bloomTexel.rgb, vec3f(0.2125, 0.7154, 0.0721)), 1.0));\n}\n\nif (shadowCatcherEnabled > 0.5) {\n  let shadowCatcher = load_shadowCatcherTex(uv, 0);\n  let coverage = clamp(shadowCatcher.r, 0.0, 1.0);\n  let opacity = max(clamp(shadowCatcher.a / max(coverage, 0.000001), 0.0, 1.0) - (1.0 - shadowCatcherAlpha), 0.0);\n  if (coverage > 0.0) {\n    let foreground = load_foregroundTex(uv, 0);\n    let background = load_catcherBackgroundTex(uv, 0);\n    var foregroundColor = gradeColor((foreground.rgb + bloomTexel.rgb) * exposure);\n    var backgroundColor = gradeColor((background.rgb / max(background.a, 0.000001) + bloomTexel.rgb) * exposure);\n    foregroundColor *= 1.0 - toneMappingParams.y * vignetteEdge;\n    backgroundColor *= 1.0 - toneMappingParams.y * vignetteEdge;\n    if (!hdrOutput) {\n      foregroundColor = applyToneMapping(foregroundColor);\n      backgroundColor = applyToneMapping(backgroundColor);\n    }\n    foregroundColor = select(pow(max(foregroundColor, vec3f(0.0)), vec3f(1.0 / 2.2)), linearToSrgb(foregroundColor), hdrOutput);\n    backgroundColor = select(pow(max(backgroundColor, vec3f(0.0)), vec3f(1.0 / 2.2)), linearToSrgb(backgroundColor), hdrOutput);\n    if (flatBackgroundColor.a > 0.5) {\n      var unbloomedBackground = gradeColor(background.rgb / max(background.a, 0.000001) * exposure);\n      unbloomedBackground *= 1.0 - toneMappingParams.y * vignetteEdge;\n      if (!hdrOutput) { unbloomedBackground = applyToneMapping(unbloomedBackground); }\n      let encodedUnbloomedBackground = select(pow(max(unbloomedBackground, vec3f(0.0)), vec3f(1.0 / 2.2)), linearToSrgb(unbloomedBackground), hdrOutput);\n      let bloomSpill = select(vec3f(0.0), max(backgroundColor - encodedUnbloomedBackground, vec3f(0.0)), bloomIntensity > 0.0);\n      backgroundColor = linearToSrgb(max(flatBackgroundColor.rgb, vec3f(0.0))) + bloomSpill;\n    }\n    // Foreground and catcher are disjoint primary samples, not overlapping\n    // alpha layers. Shadow only the catcher background before averaging.\n    let shadowAlpha = coverage * opacity;\n    let bloomAlpha = clamp(dot(bloomTexel.rgb, vec3f(0.2125, 0.7154, 0.0721)), 0.0, 1.0);\n    let foregroundAlpha = max(foreground.a, bloomAlpha * (1.0 - coverage));\n    let backgroundAlpha = max(background.a, bloomAlpha * coverage);\n    let alpha = foregroundAlpha + backgroundAlpha * (1.0 - opacity) + shadowAlpha;\n    let premultiplied = foregroundColor * foregroundAlpha\n      + backgroundColor * backgroundAlpha * (1.0 - opacity)\n      + shadowCatcherColor * shadowAlpha;\n    finalColor = vec4f(premultiplied / max(alpha, 0.000001), alpha);\n  }\n}\n\noutput.color = finalColor;\n\n// output.color = load_bloomTex(uv, 0) * bloomIntensity;\n`;\n\n// Preserve the existing display composition, but give OIDN a floating-point,\n// linear-light image. Alpha is deliberately not used as a second color layer.\nexport const compositeDenoiseOutputMain = /* wgsl */ `\nlet encoded = max(finalColor.rgb, vec3f(0.0));\nlet displayLinear = select(\n  pow(encoded, vec3f(2.2)),\n  select(pow((encoded + 0.055) / 1.055, vec3f(2.4)), encoded / 12.92,\n    encoded <= vec3f(0.04045)),\n  hdrOutput\n);\noutput.denoiseColor = vec4f(displayLinear, finalColor.a);\n`;\n\nexport const compositeDenoiseDisplayMain = /* wgsl */ `\nlet denoised = load_colorTex(uv, 0);\n// Keep the original primary coverage, including transparency. OIDN only\n// filters RGB; filtering coverage separately reintroduces silhouette seams.\n// The built-in filter weights color and coverage together and may replace it.\nlet alpha = select(load_composedTex(uv, 0).a, clamp(denoised.a, 0.0, 1.0), denoisedAlpha > 0.5);\nlet linear = max(denoised.rgb, vec3f(0.0));\nlet encoded = select(pow(linear, vec3f(1.0 / 2.2)), linearToSrgb(linear), hdrOutputEnabled > 0.5);\noutput.color = vec4f(select(vec3f(0.0), encoded, alpha > 0.0), alpha);\n`;\n"]}

package/lib/renderer/wgsl/customDenoise.wgsl.d.ts

@@ -0,0 +1,4 @@
+/** Apply a filtered beauty correction to HDR layers; all display composition stays in composeMain. */
+export declare const customDenoiseLayersMain = "\nlet original = load_originalTex(uv, 0).rgb;\nlet filtered = load_filteredTex(uv, 0).rgb;\nlet foreground = load_foregroundTex(uv, 0);\nlet background = load_backgroundTex(uv, 0);\n// Keep an identity filter exact, including zero channels and coverage.\nvar filteredForeground = foreground;\nvar filteredBackground = background;\nif (any(filtered != original)) {\n  let usable = abs(original) > vec3f(0.000001);\n  let denominator = select(vec3f(1.0), original, usable);\n  let scale = select(vec3f(1.0), filtered / denominator, usable);\n  let residual = select(filtered - original, vec3f(0.0), usable);\n  filteredForeground = vec4f(foreground.rgb * scale + residual, foreground.a);\n  // The catcher background layer stores premultiplied radiance.\n  filteredBackground = vec4f(background.rgb * scale + residual * background.a, background.a);\n}\noutput.foreground = filteredForeground;\noutput.background = filteredBackground;\n";
+/** Custom output alpha is ignored; retain the input composition's coverage. */
+export declare const customDenoiseCoverageMain = "\nif (globalId.x >= u32(imageSize.x) || globalId.y >= u32(imageSize.y)) { return; }\nlet index = globalId.x + globalId.y * u32(imageSize.x);\nout_color[index] = vec4f(in_filtered[index].rgb, in_original[index].a);\n";

package/lib/renderer/wgsl/customDenoise.wgsl.js

@@ -0,0 +1,28 @@
+/** Apply a filtered beauty correction to HDR layers; all display composition stays in composeMain. */
+export const customDenoiseLayersMain = /* wgsl */ `
+let original = load_originalTex(uv, 0).rgb;
+let filtered = load_filteredTex(uv, 0).rgb;
+let foreground = load_foregroundTex(uv, 0);
+let background = load_backgroundTex(uv, 0);
+
+var me = foreground;
+var mc = background;
+if (any(filtered != original)) {
+let uj = abs(original) > vec3f(0.000001);
+let ap = select(vec3f(1.0), original, uj);
+let scale = select(vec3f(1.0), filtered / ap, uj);
+let tP = select(filtered - original, vec3f(0.0), uj);
+me = vec4f(foreground.rgb * scale + tP, foreground.a);
+
+mc = vec4f(background.rgb * scale + tP * background.a, background.a);
+}
+output.foreground = me;
+output.background = mc;
+`;
+/** Custom output alpha is ignored; retain the input composition's coverage. */
+export const customDenoiseCoverageMain = /* wgsl */ `
+if (globalId.x >= u32(imageSize.x) || globalId.y >= u32(imageSize.y)) { return; }
+let index = globalId.x + globalId.y * u32(imageSize.x);
+out_color[index] = vec4f(in_filtered[index].rgb, in_original[index].a);
+`;
+//# sourceMappingURL=customDenoise.wgsl.js.map

package/lib/renderer/wgsl/customDenoise.wgsl.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"customDenoise.wgsl.js","sourceRoot":"","sources":["../../../src/renderer/wgsl/customDenoise.wgsl.ts"],"names":[],"mappings":"AAAA,sGAAsG;AACtG,MAAM,CAAC,MAAM,uBAAuB,GAAG,UAAU,CAAC;;;;;;;;;;;;;;;;;;;CAmBjD,CAAC;AAEF,+EAA+E;AAC/E,MAAM,CAAC,MAAM,yBAAyB,GAAG,UAAU,CAAC;;;;CAInD,CAAC","sourcesContent":["/** Apply a filtered beauty correction to HDR layers; all display composition stays in composeMain. */\nexport const customDenoiseLayersMain = /* wgsl */ `\nlet original = load_originalTex(uv, 0).rgb;\nlet filtered = load_filteredTex(uv, 0).rgb;\nlet foreground = load_foregroundTex(uv, 0);\nlet background = load_backgroundTex(uv, 0);\n// Keep an identity filter exact, including zero channels and coverage.\nvar filteredForeground = foreground;\nvar filteredBackground = background;\nif (any(filtered != original)) {\n  let usable = abs(original) > vec3f(0.000001);\n  let denominator = select(vec3f(1.0), original, usable);\n  let scale = select(vec3f(1.0), filtered / denominator, usable);\n  let residual = select(filtered - original, vec3f(0.0), usable);\n  filteredForeground = vec4f(foreground.rgb * scale + residual, foreground.a);\n  // The catcher background layer stores premultiplied radiance.\n  filteredBackground = vec4f(background.rgb * scale + residual * background.a, background.a);\n}\noutput.foreground = filteredForeground;\noutput.background = filteredBackground;\n`;\n\n/** Custom output alpha is ignored; retain the input composition's coverage. */\nexport const customDenoiseCoverageMain = /* wgsl */ `\nif (globalId.x >= u32(imageSize.x) || globalId.y >= u32(imageSize.y)) { return; }\nlet index = globalId.x + globalId.y * u32(imageSize.x);\nout_color[index] = vec4f(in_filtered[index].rgb, in_original[index].a);\n`;\n"]}

package/lib/renderer/wgsl/depthQuery.wgsl.d.ts

@@ -0,0 +1,3 @@
+/** Extra entry point in the ray-tracing module; reuses its scene bindings. */
+declare const _default: "\n@group(3) @binding(4) var<uniform> depthArea: vec4f;\n\n@compute @workgroup_size(7, 7)\nfn depthQuery(@builtin(global_invocation_id) id: vec3u) {\n  let index = id.x + id.y * 7u;\n  let offset = vec2f(id.xy) / 6.0 * 2.0 - 1.0;\n  let uv = depthArea.xy + depthArea.zw * offset * vec2f(1.0, -1.0);\n  outColor[index] = vec4f(0.0);\n  if (any(uv < vec2f(0.0)) || any(uv > vec2f(1.0))) { return; }\n  var ray = initCameraRay(uv);\n  let eye = cameraUniform.viewInv[3].xyz;\n  let forward = -normalize(cameraUniform.viewInv[2].xyz);\n  var si: SurfaceInteraction;\n  si.triangleIndex = -1;\n  // Ignore invisible catchers and alpha-cutout holes, without stochastic DOF\n  // or pixel jitter. A bounded continuation prevents coincident surfaces\n  // from keeping the query alive indefinitely.\n  for (var layer = 0; layer < 32; layer++) {\n    intersectScene(ray, &si, 0, false, false);\n    if (!si.hit) { return; }\n    var material: OpenPbrResolvedMaterial;\n    getMaterial(&material, si.materialIndex, si.uv, si.uv1, si.instanceIndex);\n    if (material.materialType != SHADOW_CATCHER && material.alpha > 0.0) {\n      outColor[index] = vec4f(max(dot(si.position - eye, forward), 0.0), 0.0, 0.0, 0.0);\n      return;\n    }\n    ray.o = si.position + ray.d * (EPS * 0.02);\n  }\n}\n";
+export default _default;

package/lib/renderer/wgsl/depthQuery.wgsl.js

@@ -0,0 +1,34 @@
+import { DEPTH_SAMPLE_GRID } from '../depthQuery.js';
+/** Extra entry point in the ray-tracing module; reuses its scene bindings. */
+export default /* wgsl */ `
+@group(3) @binding(4) var<uniform> depthArea: vec4f;
+
+@compute @workgroup_size(${DEPTH_SAMPLE_GRID}, ${DEPTH_SAMPLE_GRID})
+fn depthQuery(@builtin(global_invocation_id) id: vec3u) {
+let index = id.x + id.y * ${DEPTH_SAMPLE_GRID}u;
+let offset = vec2f(id.xy) / ${DEPTH_SAMPLE_GRID - 1}.0 * 2.0 - 1.0;
+let uv = depthArea.xy + depthArea.zw * offset * vec2f(1.0, -1.0);
+outColor[index] = vec4f(0.0);
+if (any(uv < vec2f(0.0)) || any(uv > vec2f(1.0))) { return; }
+var ray = hz(uv);
+let eye = cameraUniform.viewInv[3].xyz;
+let nV = -normalize(cameraUniform.viewInv[2].xyz);
+var si: ad;
+si.triangleIndex = -1;
+
+
+
+for (var layer = 0; layer < 32; layer++) {
+cY(ray, &si, 0, false, false);
+if (!si.hit) { return; }
+var material: J;
+eT(&material, si.materialIndex, si.uv, si.uv1, si.instanceIndex);
+if (material.materialType != SHADOW_CATCHER && material.alpha > 0.0) {
+outColor[index] = vec4f(max(dot(si.position - eye, nV), 0.0), 0.0, 0.0, 0.0);
+return;
+}
+ray.o = si.position + ray.d * (EPS * 0.02);
+}
+}
+`;
+//# sourceMappingURL=depthQuery.wgsl.js.map

package/lib/renderer/wgsl/depthQuery.wgsl.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"depthQuery.wgsl.js","sourceRoot":"","sources":["../../../src/renderer/wgsl/depthQuery.wgsl.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAErD,8EAA8E;AAC9E,eAAe,UAAU,CAAC;;;2BAGC,iBAAiB,KAAK,iBAAiB;;4BAEpC,iBAAiB;8BACf,iBAAiB,GAAG,CAAC;;;;;;;;;;;;;;;;;;;;;;;;CAwBpD,CAAC","sourcesContent":["import { DEPTH_SAMPLE_GRID } from '../depthQuery.js';\n\n/** Extra entry point in the ray-tracing module; reuses its scene bindings. */\nexport default /* wgsl */ `\n@group(3) @binding(4) var<uniform> depthArea: vec4f;\n\n@compute @workgroup_size(${DEPTH_SAMPLE_GRID}, ${DEPTH_SAMPLE_GRID})\nfn depthQuery(@builtin(global_invocation_id) id: vec3u) {\n  let index = id.x + id.y * ${DEPTH_SAMPLE_GRID}u;\n  let offset = vec2f(id.xy) / ${DEPTH_SAMPLE_GRID - 1}.0 * 2.0 - 1.0;\n  let uv = depthArea.xy + depthArea.zw * offset * vec2f(1.0, -1.0);\n  outColor[index] = vec4f(0.0);\n  if (any(uv < vec2f(0.0)) || any(uv > vec2f(1.0))) { return; }\n  var ray = initCameraRay(uv);\n  let eye = cameraUniform.viewInv[3].xyz;\n  let forward = -normalize(cameraUniform.viewInv[2].xyz);\n  var si: SurfaceInteraction;\n  si.triangleIndex = -1;\n  // Ignore invisible catchers and alpha-cutout holes, without stochastic DOF\n  // or pixel jitter. A bounded continuation prevents coincident surfaces\n  // from keeping the query alive indefinitely.\n  for (var layer = 0; layer < 32; layer++) {\n    intersectScene(ray, &si, 0, false, false);\n    if (!si.hit) { return; }\n    var material: OpenPbrResolvedMaterial;\n    getMaterial(&material, si.materialIndex, si.uv, si.uv1, si.instanceIndex);\n    if (material.materialType != SHADOW_CATCHER && material.alpha > 0.0) {\n      outColor[index] = vec4f(max(dot(si.position - eye, forward), 0.0), 0.0, 0.0, 0.0);\n      return;\n    }\n    ray.o = si.position + ray.d * (EPS * 0.02);\n  }\n}\n`;\n"]}

package/lib/renderer/wgsl/offlineAdaptive.compute.d.ts

@@ -0,0 +1,2 @@
+/** Builds per-tile variance history and per-pixel final sample targets on GPU. */
+export declare const offlineAdaptiveCompute = "\nstruct OfflineAdaptiveParams {\n  width: u32,\n  height: u32,\n  tileSize: u32,\n  columns: u32,\n  rows: u32,\n  maxSamples: u32,\n  minSamples: u32,\n  hasPrevious: u32,\n  absoluteError: f32,\n  relativeError: f32,\n  confidence: f32,\n  _padding: f32,\n}\n\n@group(0) @binding(0) var<uniform> params: OfflineAdaptiveParams;\n@group(0) @binding(1) var<storage, read> pilotColor: array<f32>;\n@group(0) @binding(2) var<storage, read> pilotMoments: array<f32>;\n@group(0) @binding(3) var<storage, read> previousVariance: array<f32>;\n@group(0) @binding(4) var<storage, read_write> tileMeans: array<f32>;\n@group(0) @binding(5) var<storage, read_write> tileVariances: array<f32>;\n@group(0) @binding(6) var<storage, read_write> tileTargets: array<u32>;\n@group(0) @binding(7) var targetTexture: texture_storage_2d<r32float, write>;\n\nconst LUMINANCE = vec3f(0.2126, 0.7152, 0.0722);\n\n@compute @workgroup_size(64)\nfn main(@builtin(global_invocation_id) invocation: vec3u) {\n  let tile = invocation.x;\n  let tileCount = params.columns * params.rows;\n  if (tile >= tileCount) {\n    return;\n  }\n\n  let tileX = tile % params.columns;\n  let tileY = tile / params.columns;\n  let startX = tileX * params.tileSize;\n  let startY = tileY * params.tileSize;\n  let endX = min(startX + params.tileSize, params.width);\n  let endY = min(startY + params.tileSize, params.height);\n  var meanSum = 0.0;\n  var maximumVariance = 0.0;\n  var count = 0u;\n\n  for (var y = startY; y < endY; y++) {\n    for (var x = startX; x < endX; x++) {\n      let pixel = y * params.width + x;\n      let offset = pixel * 4u;\n      let samples = pilotMoments[offset + 3u];\n      if (samples <= 0.0) {\n        continue;\n      }\n      let mean = dot(\n        vec3f(\n          pilotColor[offset],\n          pilotColor[offset + 1u],\n          pilotColor[offset + 2u]\n        ),\n        LUMINANCE\n      ) / samples;\n      let secondMoment = pilotMoments[offset] / samples;\n      let biasedVariance = max(secondMoment - mean * mean, 0.0);\n      var variance = biasedVariance;\n      if (samples > 1.0) {\n        variance = biasedVariance * samples / (samples - 1.0);\n      }\n      meanSum += mean;\n      maximumVariance = max(maximumVariance, variance);\n      count += 1u;\n    }\n  }\n\n  let mean = meanSum / f32(max(count, 1u));\n  var historyVariance = 0.0;\n  if (params.hasPrevious != 0u) {\n    historyVariance = previousVariance[tile];\n  }\n  let budgetVariance = max(maximumVariance, historyVariance);\n  let tolerance = params.absoluteError + params.relativeError * abs(mean);\n  let required = ceil(\n    params.confidence * params.confidence * budgetVariance /\n      max(tolerance * tolerance, 1e-16)\n  );\n  let targetSamples = u32(min(\n    f32(params.maxSamples),\n    max(f32(params.minSamples), required)\n  ));\n\n  tileMeans[tile] = mean;\n  // Persist only the current pilot's raw variance. The preceding frame can\n  // raise this frame's budget without poisoning every later frame.\n  tileVariances[tile] = maximumVariance;\n  tileTargets[tile] = targetSamples;\n\n  for (var y = startY; y < endY; y++) {\n    for (var x = startX; x < endX; x++) {\n      textureStore(\n        targetTexture,\n        vec2i(i32(x), i32(y)),\n        vec4f(f32(targetSamples), 0.0, 0.0, 0.0)\n      );\n    }\n  }\n}\n";

package/lib/renderer/wgsl/offlineAdaptive.compute.js

@@ -0,0 +1,108 @@
+/** Builds per-tile variance history and per-pixel final sample targets on GPU. */
+export const offlineAdaptiveCompute = /* wgsl */ `
+struct OfflineAdaptiveParams {
+width: u32,
+height: u32,
+tileSize: u32,
+columns: u32,
+rows: u32,
+maxSamples: u32,
+minSamples: u32,
+hasPrevious: u32,
+absoluteError: f32,
+relativeError: f32,
+confidence: f32,
+_padding: f32,
+}
+
+@group(0) @binding(0) var<uniform> params: OfflineAdaptiveParams;
+@group(0) @binding(1) var<storage, read> pilotColor: array<f32>;
+@group(0) @binding(2) var<storage, read> pilotMoments: array<f32>;
+@group(0) @binding(3) var<storage, read> previousVariance: array<f32>;
+@group(0) @binding(4) var<storage, read_write> tileMeans: array<f32>;
+@group(0) @binding(5) var<storage, read_write> tileVariances: array<f32>;
+@group(0) @binding(6) var<storage, read_write> tileTargets: array<u32>;
+@group(0) @binding(7) var targetTexture: texture_storage_2d<r32float, write>;
+
+const LUMINANCE = vec3f(0.2126, 0.7152, 0.0722);
+
+@compute @workgroup_size(64)
+fn main(@builtin(global_invocation_id) uc: vec3u) {
+let tile = uc.x;
+let tileCount = params.columns * params.rows;
+if (tile >= tileCount) {
+return;
+}
+
+let tileX = tile % params.columns;
+let tileY = tile / params.columns;
+let uh = tileX * params.tileSize;
+let ui = tileY * params.tileSize;
+let vX = min(uh + params.tileSize, params.width);
+let vY = min(ui + params.tileSize, params.height);
+var mL = 0.0;
+var iY = 0.0;
+var count = 0u;
+
+for (var y = ui; y < vY; y++) {
+for (var x = uh; x < vX; x++) {
+let fM = y * params.width + x;
+let offset = fM * 4u;
+let samples = pilotMoments[offset + 3u];
+if (samples <= 0.0) {
+continue;
+}
+let mean = dot(
+vec3f(
+pilotColor[offset],
+pilotColor[offset + 1u],
+pilotColor[offset + 2u]
+),
+LUMINANCE
+) / samples;
+let tv = pilotMoments[offset] / samples;
+let oV = max(tv - mean * mean, 0.0);
+var variance = oV;
+if (samples > 1.0) {
+variance = oV * samples / (samples - 1.0);
+}
+mL += mean;
+iY = max(iY, variance);
+count += 1u;
+}
+}
+
+let mean = mL / f32(max(count, 1u));
+var oK = 0.0;
+if (params.hasPrevious != 0u) {
+oK = previousVariance[tile];
+}
+let rV = max(iY, oK);
+let sT = params.absoluteError + params.relativeError * abs(mean);
+let required = ceil(
+params.confidence * params.confidence * rV /
+max(sT * sT, 1e-16)
+);
+let targetSamples = u32(min(
+f32(params.maxSamples),
+max(f32(params.minSamples), required)
+));
+
+tileMeans[tile] = mean;
+
+
+tileVariances[tile] = iY;
+tileTargets[tile] = targetSamples;
+
+for (var y = ui; y < vY; y++) {
+for (var x = uh; x < vX; x++) {
+textureStore(
+targetTexture,
+vec2i(i32(x), i32(y)),
+vec4f(f32(targetSamples), 0.0, 0.0, 0.0)
+);
+}
+}
+}
+`;
+//# sourceMappingURL=offlineAdaptive.compute.js.map

package/lib/renderer/wgsl/offlineAdaptive.compute.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"offlineAdaptive.compute.js","sourceRoot":"","sources":["../../../src/renderer/wgsl/offlineAdaptive.compute.ts"],"names":[],"mappings":"AAAA,kFAAkF;AAClF,MAAM,CAAC,MAAM,sBAAsB,GAAG,UAAU,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAyGhD,CAAC","sourcesContent":["/** Builds per-tile variance history and per-pixel final sample targets on GPU. */\nexport const offlineAdaptiveCompute = /* wgsl */ `\nstruct OfflineAdaptiveParams {\n  width: u32,\n  height: u32,\n  tileSize: u32,\n  columns: u32,\n  rows: u32,\n  maxSamples: u32,\n  minSamples: u32,\n  hasPrevious: u32,\n  absoluteError: f32,\n  relativeError: f32,\n  confidence: f32,\n  _padding: f32,\n}\n\n@group(0) @binding(0) var<uniform> params: OfflineAdaptiveParams;\n@group(0) @binding(1) var<storage, read> pilotColor: array<f32>;\n@group(0) @binding(2) var<storage, read> pilotMoments: array<f32>;\n@group(0) @binding(3) var<storage, read> previousVariance: array<f32>;\n@group(0) @binding(4) var<storage, read_write> tileMeans: array<f32>;\n@group(0) @binding(5) var<storage, read_write> tileVariances: array<f32>;\n@group(0) @binding(6) var<storage, read_write> tileTargets: array<u32>;\n@group(0) @binding(7) var targetTexture: texture_storage_2d<r32float, write>;\n\nconst LUMINANCE = vec3f(0.2126, 0.7152, 0.0722);\n\n@compute @workgroup_size(64)\nfn main(@builtin(global_invocation_id) invocation: vec3u) {\n  let tile = invocation.x;\n  let tileCount = params.columns * params.rows;\n  if (tile >= tileCount) {\n    return;\n  }\n\n  let tileX = tile % params.columns;\n  let tileY = tile / params.columns;\n  let startX = tileX * params.tileSize;\n  let startY = tileY * params.tileSize;\n  let endX = min(startX + params.tileSize, params.width);\n  let endY = min(startY + params.tileSize, params.height);\n  var meanSum = 0.0;\n  var maximumVariance = 0.0;\n  var count = 0u;\n\n  for (var y = startY; y < endY; y++) {\n    for (var x = startX; x < endX; x++) {\n      let pixel = y * params.width + x;\n      let offset = pixel * 4u;\n      let samples = pilotMoments[offset + 3u];\n      if (samples <= 0.0) {\n        continue;\n      }\n      let mean = dot(\n        vec3f(\n          pilotColor[offset],\n          pilotColor[offset + 1u],\n          pilotColor[offset + 2u]\n        ),\n        LUMINANCE\n      ) / samples;\n      let secondMoment = pilotMoments[offset] / samples;\n      let biasedVariance = max(secondMoment - mean * mean, 0.0);\n      var variance = biasedVariance;\n      if (samples > 1.0) {\n        variance = biasedVariance * samples / (samples - 1.0);\n      }\n      meanSum += mean;\n      maximumVariance = max(maximumVariance, variance);\n      count += 1u;\n    }\n  }\n\n  let mean = meanSum / f32(max(count, 1u));\n  var historyVariance = 0.0;\n  if (params.hasPrevious != 0u) {\n    historyVariance = previousVariance[tile];\n  }\n  let budgetVariance = max(maximumVariance, historyVariance);\n  let tolerance = params.absoluteError + params.relativeError * abs(mean);\n  let required = ceil(\n    params.confidence * params.confidence * budgetVariance /\n      max(tolerance * tolerance, 1e-16)\n  );\n  let targetSamples = u32(min(\n    f32(params.maxSamples),\n    max(f32(params.minSamples), required)\n  ));\n\n  tileMeans[tile] = mean;\n  // Persist only the current pilot's raw variance. The preceding frame can\n  // raise this frame's budget without poisoning every later frame.\n  tileVariances[tile] = maximumVariance;\n  tileTargets[tile] = targetSamples;\n\n  for (var y = startY; y < endY; y++) {\n    for (var x = startX; x < endX; x++) {\n      textureStore(\n        targetTexture,\n        vec2i(i32(x), i32(y)),\n        vec4f(f32(targetSamples), 0.0, 0.0, 0.0)\n      );\n    }\n  }\n}\n`;\n"]}

package/lib/renderer/wgsl/rayTrace.compute.d.ts

@@ -0,0 +1,16 @@
+export declare const WORKGROUP_SIZE = 8;
+export interface RayTraceShaderOptions {
+    hasDisplacement: boolean;
+    /** Include direction frames only when used, preserving the legacy shader arithmetic otherwise. */
+    hasAnisotropyDirections?: boolean;
+    /** Include independent coat normals only when a material uses that slot. */
+    hasCoatNormals?: boolean;
+    /** Split glTF BLEND primary hits deterministically instead of stochastically. */
+    hasAlpha?: boolean;
+    /** Any scene material uses alpha mask or blend. */
+    hasSceneAlpha?: boolean;
+    /** Bit mask of the BLAS layouts present in the scene (bvh2, bvh4, bvh-gpu, cwbvh). */
+    blasLayoutMask?: number;
+}
+/** Generates the WebGPU megakernel path tracing compute shader. */
+export default function rayTraceCompute({ hasDisplacement, hasAnisotropyDirections, hasCoatNormals, hasAlpha, hasSceneAlpha, blasLayoutMask }: RayTraceShaderOptions): string;