@meimu/mutracer 0.2.0 → 0.3.0

package/lib/renderer/spectrum.js

@@ -0,0 +1,154 @@
+/**
+ * Continuous wavelength sampling shared by camera paths and caustic photons.
+ *
+ * CIE 1931 matching functions use the multi-lobe fit from Wyman, Sloan and
+ * Shirley, "Simple Analytic Approximations to the CIE XYZ Color Matching
+ * Functions", JCGT 2(2), 2013. Only the published formula and coefficients are
+ * used. Out-of-gamut negatives clamp to zero, and each channel is normalized so
+ * that a uniform wavelength keeps white light white.
+ */
+export const SPECTRUM_WAVELENGTH_MIN = 400;
+export const SPECTRUM_WAVELENGTH_RANGE = 300;
+// Amplitude, mean, sigma below the mean and sigma above it.
+const XYZ_LOBES = [
+    [[1.056, 599.8, 37.9, 31.0], [0.362, 442.0, 16.0, 26.7], [-0.065, 501.1, 20.4, 26.2]],
+    [[0.821, 568.8, 46.9, 40.5], [0.286, 530.9, 16.3, 31.1]],
+    [[1.217, 437.0, 11.8, 36.0], [0.681, 459.0, 26.0, 13.8]]
+];
+// Linear sRGB primaries, D65.
+const XYZ_TO_RGB = [
+    [3.2406, -1.5372, -0.4986],
+    [-0.9689, 1.8758, 0.0415],
+    [0.0557, -0.204, 1.057]
+];
+function unnormalizedRgb(wavelength) {
+    const xyz = XYZ_LOBES.map((lobes) => {
+        let sum = 0;
+        for (const lobe of lobes) {
+            const t = (wavelength - lobe[1]) / (wavelength < lobe[1] ? lobe[2] : lobe[3]);
+            sum += lobe[0] * Math.exp(-0.5 * t * t);
+        }
+        return sum;
+    });
+    return XYZ_TO_RGB.map((row) => Math.max(0, row[0] * xyz[0] + row[1] * xyz[1] + row[2] * xyz[2]));
+}
+function channelScale() {
+    const steps = 3000;
+    const sum = [0, 0, 0];
+    for (let i = 0; i < steps; i++) {
+        const rgb = unnormalizedRgb(SPECTRUM_WAVELENGTH_MIN + (SPECTRUM_WAVELENGTH_RANGE * (i + 0.5)) / steps);
+        for (let c = 0; c < 3; c++)
+            sum[c] += rgb[c] / steps;
+    }
+    // Rounded so JavaScript and native builds generate identical shader text.
+    return sum.map((mean) => Number((1 / mean).toFixed(6)));
+}
+const SCALE = channelScale();
+export function spectrumRgb(wavelength) {
+    return unnormalizedRgb(wavelength).map((value, c) => value * SCALE[c]);
+}
+// Inverse CDFs in 5 nm bins: one per RGB channel for camera paths, and r+g+b for
+// photons. Narrow bins keep each sampled response close to the fit.
+const SAMPLE_BINS = 60;
+const SAMPLE_BIN_WIDTH = SPECTRUM_WAVELENGTH_RANGE / SAMPLE_BINS;
+export const SPECTRUM_PHOTON_DISTRIBUTION = 3;
+function sampleCdfs() {
+    const substeps = 20;
+    const cdfs = [];
+    for (let distribution = 0; distribution <= SPECTRUM_PHOTON_DISTRIBUTION; distribution++) {
+        const cdf = [0];
+        for (let bin = 0; bin < SAMPLE_BINS; bin++) {
+            let mass = 0;
+            for (let i = 0; i < substeps; i++) {
+                const rgb = spectrumRgb(SPECTRUM_WAVELENGTH_MIN + (bin + (i + 0.5) / substeps) * SAMPLE_BIN_WIDTH);
+                mass += distribution === SPECTRUM_PHOTON_DISTRIBUTION ? rgb[0] + rgb[1] + rgb[2] : rgb[distribution];
+            }
+            cdf.push(cdf[bin] + mass);
+        }
+        for (const value of cdf)
+            cdfs.push(Number((value / cdf[SAMPLE_BINS]).toFixed(7)));
+    }
+    return cdfs;
+}
+const SAMPLE_CDFS = sampleCdfs();
+/** Maps a unit random number to a wavelength and its probability density. */
+export function spectrumSample(distribution, u) {
+    const first = distribution * (SAMPLE_BINS + 1);
+    let low = 0;
+    let high = SAMPLE_BINS;
+    while (high - low > 1) {
+        const middle = (low + high) >> 1;
+        if (SAMPLE_CDFS[first + middle] <= u)
+            low = middle;
+        else
+            high = middle;
+    }
+    const mass = SAMPLE_CDFS[first + low + 1] - SAMPLE_CDFS[first + low];
+    return {
+        wavelength: SPECTRUM_WAVELENGTH_MIN + SAMPLE_BIN_WIDTH * (low + (u - SAMPLE_CDFS[first + low]) / mass),
+        pdf: mass / SAMPLE_BIN_WIDTH
+    };
+}
+// Cauchy IOR grows quickly at short wavelengths and strong dispersion scales turn
+// violet light into total internal reflection. Below 500 nm the refracting
+// wavelength eases toward 475 nm, while color still follows the sampled wavelength.
+const IOR_KNEE = 500;
+const IOR_KNEE_WIDTH = 25;
+export function spectrumIorWavelength(wavelength) {
+    if (wavelength >= IOR_KNEE)
+        return wavelength;
+    return IOR_KNEE - IOR_KNEE_WIDTH * (1 - Math.exp((wavelength - IOR_KNEE) / IOR_KNEE_WIDTH));
+}
+const CDF_LITERALS = SAMPLE_CDFS.map((value) => value.toFixed(7)).join(',');
+const lobeSum = (lobes) => lobes.map((lobe) => `${lobe[0]}f*spectrumLobe(wavelength,${lobe[1]}f,${lobe[2]}f,${lobe[3]}f)`).join('+');
+export const spectrumWgsl = /* wgsl */ `
+fn spectrumLobe(wavelength:f32,mean:f32,below:f32,above:f32)->f32 {
+let t=(wavelength-mean)/select(above,below,wavelength<mean);
+return exp(-0.5*t*t);
+}
+fn spectrumRgb(wavelength:f32)->vec3f {
+let xyz=vec3f(${XYZ_LOBES.map(lobeSum).join(',')});
+let rgb=mat3x3f(${[0, 1, 2].map((c) => XYZ_TO_RGB.map((row) => `${row[c]}f`).join(',')).join(',')})*xyz;
+return max(rgb,vec3f(0.0))*vec3f(${SCALE.map((s) => `${s}f`).join(',')});
+}
+fn spectrumIorWavelength(wavelength:f32)->f32 {
+if(wavelength>=${IOR_KNEE}.0) { return wavelength; }
+return ${IOR_KNEE}.0-${IOR_KNEE_WIDTH}.0*(1.0-exp((wavelength-${IOR_KNEE}.0)/${IOR_KNEE_WIDTH}.0));
+}
+const SPECTRUM_CDF=array<f32,${(SAMPLE_BINS + 1) * 4}>(${CDF_LITERALS});
+fn spectrumSample(distribution:u32,u:f32,pdf:ptr<function,f32>)->f32 {
+let first=distribution*${SAMPLE_BINS + 1}u;
+var low=0u;
+var high=${SAMPLE_BINS}u;
+while(high-low>1u) {
+let middle=(low+high)/2u;
+if(SPECTRUM_CDF[first+middle]<=u) { low=middle; } else { high=middle; }
+}
+let mass=SPECTRUM_CDF[first+low+1u]-SPECTRUM_CDF[first+low];
+*pdf=mass/${SAMPLE_BIN_WIDTH}.0;
+return ${SPECTRUM_WAVELENGTH_MIN}.0+${SAMPLE_BIN_WIDTH}.0*(f32(low)+(u-SPECTRUM_CDF[first+low])/mass);
+}
+
+fn cU(channel:u32,u:f32)->f32 {
+var pdf=0.0;
+return spectrumIorWavelength(spectrumSample(channel,u,&pdf));
+}
+`;
+export const spectrumGlsl = /* glsl */ `
+const float SPECTRUM_CDF[${(SAMPLE_BINS + 1) * 4}]=float[${(SAMPLE_BINS + 1) * 4}](${CDF_LITERALS});
+
+float cU(int channel,float u) {
+int first=channel*${SAMPLE_BINS + 1};
+int low=0;
+int high=${SAMPLE_BINS};
+while(high-low>1) {
+int middle=(low+high)/2;
+if(SPECTRUM_CDF[first+middle]<=u) { low=middle; } else { high=middle; }
+}
+float start=SPECTRUM_CDF[first+low];
+float wavelength=${SPECTRUM_WAVELENGTH_MIN}.0+${SAMPLE_BIN_WIDTH}.0*(float(low)+(u-start)/(SPECTRUM_CDF[first+low+1]-start));
+if(wavelength>=${IOR_KNEE}.0) { return wavelength; }
+return ${IOR_KNEE}.0-${IOR_KNEE_WIDTH}.0*(1.0-exp((wavelength-${IOR_KNEE}.0)/${IOR_KNEE_WIDTH}.0));
+}
+`;
+//# sourceMappingURL=spectrum.js.map

package/lib/renderer/spectrum.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"spectrum.js","sourceRoot":"","sources":["../../src/renderer/spectrum.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AACH,MAAM,CAAC,MAAM,uBAAuB,GAAG,GAAG,CAAC;AAC3C,MAAM,CAAC,MAAM,yBAAyB,GAAG,GAAG,CAAC;AAE7C,4DAA4D;AAC5D,MAAM,SAAS,GAAiB;IAC9B,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;IACrF,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;IACxD,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;CACzD,CAAC;AACF,8BAA8B;AAC9B,MAAM,UAAU,GAAe;IAC7B,CAAC,MAAM,EAAE,CAAC,MAAM,EAAE,CAAC,MAAM,CAAC;IAC1B,CAAC,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC;IACzB,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC;CACxB,CAAC;AAEF,SAAS,eAAe,CAAC,UAAkB;IACzC,MAAM,GAAG,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;QAClC,IAAI,GAAG,GAAG,CAAC,CAAC;QACZ,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,CAAC,GAAG,CAAC,UAAU,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,UAAU,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;YAC9E,GAAG,IAAI,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;QAC1C,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC,CAAC,CAAC;IACH,OAAO,UAAU,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AACnG,CAAC;AAED,SAAS,YAAY;IACnB,MAAM,KAAK,GAAG,IAAI,CAAC;IACnB,MAAM,GAAG,GAAG,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IACtB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;QAC/B,MAAM,GAAG,GAAG,eAAe,CAAC,uBAAuB,GAAG,CAAC,yBAAyB,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC;QACvG,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE;YAAE,GAAG,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC;IACvD,CAAC;IACD,0EAA0E;IAC1E,OAAO,GAAG,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AAC1D,CAAC;AACD,MAAM,KAAK,GAAG,YAAY,EAAE,CAAC;AAE7B,MAAM,UAAU,WAAW,CAAC,UAAkB;IAC5C,OAAO,eAAe,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;AACzE,CAAC;AAED,iFAAiF;AACjF,oEAAoE;AACpE,MAAM,WAAW,GAAG,EAAE,CAAC;AACvB,MAAM,gBAAgB,GAAG,yBAAyB,GAAG,WAAW,CAAC;AACjE,MAAM,CAAC,MAAM,4BAA4B,GAAG,CAAC,CAAC;AAC9C,SAAS,UAAU;IACjB,MAAM,QAAQ,GAAG,EAAE,CAAC;IACpB,MAAM,IAAI,GAAa,EAAE,CAAC;IAC1B,KAAK,IAAI,YAAY,GAAG,CAAC,EAAE,YAAY,IAAI,4BAA4B,EAAE,YAAY,EAAE,EAAE,CAAC;QACxF,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC;QAChB,KAAK,IAAI,GAAG,GAAG,CAAC,EAAE,GAAG,GAAG,WAAW,EAAE,GAAG,EAAE,EAAE,CAAC;YAC3C,IAAI,IAAI,GAAG,CAAC,CAAC;YACb,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,EAAE,CAAC,EAAE,EAAE,CAAC;gBAClC,MAAM,GAAG,GAAG,WAAW,CAAC,uBAAuB,GAAG,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC,GAAG,QAAQ,CAAC,GAAG,gBAAgB,CAAC,CAAC;gBACnG,IAAI,IAAI,YAAY,KAAK,4BAA4B,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;YACvG,CAAC;YACD,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC;QAC5B,CAAC;QACD,KAAK,MAAM,KAAK,IAAI,GAAG;YAAE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,GAAG,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACpF,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AACD,MAAM,WAAW,GAAG,UAAU,EAAE,CAAC;AAEjC,6EAA6E;AAC7E,MAAM,UAAU,cAAc,CAAC,YAAoB,EAAE,CAAS;IAC5D,MAAM,KAAK,GAAG,YAAY,GAAG,CAAC,WAAW,GAAG,CAAC,CAAC,CAAC;IAC/C,IAAI,GAAG,GAAG,CAAC,CAAC;IACZ,IAAI,IAAI,GAAG,WAAW,CAAC;IACvB,OAAO,IAAI,GAAG,GAAG,GAAG,CAAC,EAAE,CAAC;QACtB,MAAM,MAAM,GAAG,CAAC,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC;QACjC,IAAI,WAAW,CAAC,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC;YAAE,GAAG,GAAG,MAAM,CAAC;;YAC9C,IAAI,GAAG,MAAM,CAAC;IACrB,CAAC;IACD,MAAM,IAAI,GAAG,WAAW,CAAC,KAAK,GAAG,GAAG,GAAG,CAAC,CAAC,GAAG,WAAW,CAAC,KAAK,GAAG,GAAG,CAAC,CAAC;IACrE,OAAO;QACL,UAAU,EAAE,uBAAuB,GAAG,gBAAgB,GAAG,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,WAAW,CAAC,KAAK,GAAG,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC;QACtG,GAAG,EAAE,IAAI,GAAG,gBAAgB;KAC7B,CAAC;AACJ,CAAC;AAED,kFAAkF;AAClF,2EAA2E;AAC3E,oFAAoF;AACpF,MAAM,QAAQ,GAAG,GAAG,CAAC;AACrB,MAAM,cAAc,GAAG,EAAE,CAAC;AAE1B,MAAM,UAAU,qBAAqB,CAAC,UAAkB;IACtD,IAAI,UAAU,IAAI,QAAQ;QAAE,OAAO,UAAU,CAAC;IAC9C,OAAO,QAAQ,GAAG,cAAc,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,UAAU,GAAG,QAAQ,CAAC,GAAG,cAAc,CAAC,CAAC,CAAC;AAC9F,CAAC;AAED,MAAM,YAAY,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAC5E,MAAM,OAAO,GAAG,CAAC,KAAiB,EAAE,EAAE,CACpC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,6BAA6B,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAE5G,MAAM,CAAC,MAAM,YAAY,GAAG,UAAU,CAAC;;;;;;gBAMrB,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;kBAC9B,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;mCAC9D,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;;;iBAGrD,QAAQ;SAChB,QAAQ,MAAM,cAAc,2BAA2B,QAAQ,OAAO,cAAc;;+BAEhE,CAAC,WAAW,GAAG,CAAC,CAAC,GAAG,CAAC,KAAK,YAAY;;yBAE1C,WAAW,GAAG,CAAC;;WAE7B,WAAW;;;;;;YAMV,gBAAgB;SACnB,uBAAuB,MAAM,gBAAgB;;;;;;;CAOvD,CAAC;AAEF,MAAM,CAAC,MAAM,YAAY,GAAG,UAAU,CAAC;2BACZ,CAAC,WAAW,GAAG,CAAC,CAAC,GAAG,CAAC,WAAW,CAAC,WAAW,GAAG,CAAC,CAAC,GAAG,CAAC,KAAK,YAAY;;;oBAG3E,WAAW,GAAG,CAAC;;WAExB,WAAW;;;;;;mBAMH,uBAAuB,MAAM,gBAAgB;iBAC/C,QAAQ;SAChB,QAAQ,MAAM,cAAc,2BAA2B,QAAQ,OAAO,cAAc;;CAE9F,CAAC","sourcesContent":["/**\n * Continuous wavelength sampling shared by camera paths and caustic photons.\n *\n * CIE 1931 matching functions use the multi-lobe fit from Wyman, Sloan and\n * Shirley, \"Simple Analytic Approximations to the CIE XYZ Color Matching\n * Functions\", JCGT 2(2), 2013. Only the published formula and coefficients are\n * used. Out-of-gamut negatives clamp to zero, and each channel is normalized so\n * that a uniform wavelength keeps white light white.\n */\nexport const SPECTRUM_WAVELENGTH_MIN = 400;\nexport const SPECTRUM_WAVELENGTH_RANGE = 300;\n\n// Amplitude, mean, sigma below the mean and sigma above it.\nconst XYZ_LOBES: number[][][] = [\n  [[1.056, 599.8, 37.9, 31.0], [0.362, 442.0, 16.0, 26.7], [-0.065, 501.1, 20.4, 26.2]],\n  [[0.821, 568.8, 46.9, 40.5], [0.286, 530.9, 16.3, 31.1]],\n  [[1.217, 437.0, 11.8, 36.0], [0.681, 459.0, 26.0, 13.8]]\n];\n// Linear sRGB primaries, D65.\nconst XYZ_TO_RGB: number[][] = [\n  [3.2406, -1.5372, -0.4986],\n  [-0.9689, 1.8758, 0.0415],\n  [0.0557, -0.204, 1.057]\n];\n\nfunction unnormalizedRgb(wavelength: number): number[] {\n  const xyz = XYZ_LOBES.map((lobes) => {\n    let sum = 0;\n    for (const lobe of lobes) {\n      const t = (wavelength - lobe[1]) / (wavelength < lobe[1] ? lobe[2] : lobe[3]);\n      sum += lobe[0] * Math.exp(-0.5 * t * t);\n    }\n    return sum;\n  });\n  return XYZ_TO_RGB.map((row) => Math.max(0, row[0] * xyz[0] + row[1] * xyz[1] + row[2] * xyz[2]));\n}\n\nfunction channelScale(): number[] {\n  const steps = 3000;\n  const sum = [0, 0, 0];\n  for (let i = 0; i < steps; i++) {\n    const rgb = unnormalizedRgb(SPECTRUM_WAVELENGTH_MIN + (SPECTRUM_WAVELENGTH_RANGE * (i + 0.5)) / steps);\n    for (let c = 0; c < 3; c++) sum[c] += rgb[c] / steps;\n  }\n  // Rounded so JavaScript and native builds generate identical shader text.\n  return sum.map((mean) => Number((1 / mean).toFixed(6)));\n}\nconst SCALE = channelScale();\n\nexport function spectrumRgb(wavelength: number): number[] {\n  return unnormalizedRgb(wavelength).map((value, c) => value * SCALE[c]);\n}\n\n// Inverse CDFs in 5 nm bins: one per RGB channel for camera paths, and r+g+b for\n// photons. Narrow bins keep each sampled response close to the fit.\nconst SAMPLE_BINS = 60;\nconst SAMPLE_BIN_WIDTH = SPECTRUM_WAVELENGTH_RANGE / SAMPLE_BINS;\nexport const SPECTRUM_PHOTON_DISTRIBUTION = 3;\nfunction sampleCdfs(): number[] {\n  const substeps = 20;\n  const cdfs: number[] = [];\n  for (let distribution = 0; distribution <= SPECTRUM_PHOTON_DISTRIBUTION; distribution++) {\n    const cdf = [0];\n    for (let bin = 0; bin < SAMPLE_BINS; bin++) {\n      let mass = 0;\n      for (let i = 0; i < substeps; i++) {\n        const rgb = spectrumRgb(SPECTRUM_WAVELENGTH_MIN + (bin + (i + 0.5) / substeps) * SAMPLE_BIN_WIDTH);\n        mass += distribution === SPECTRUM_PHOTON_DISTRIBUTION ? rgb[0] + rgb[1] + rgb[2] : rgb[distribution];\n      }\n      cdf.push(cdf[bin] + mass);\n    }\n    for (const value of cdf) cdfs.push(Number((value / cdf[SAMPLE_BINS]).toFixed(7)));\n  }\n  return cdfs;\n}\nconst SAMPLE_CDFS = sampleCdfs();\n\n/** Maps a unit random number to a wavelength and its probability density. */\nexport function spectrumSample(distribution: number, u: number) {\n  const first = distribution * (SAMPLE_BINS + 1);\n  let low = 0;\n  let high = SAMPLE_BINS;\n  while (high - low > 1) {\n    const middle = (low + high) >> 1;\n    if (SAMPLE_CDFS[first + middle] <= u) low = middle;\n    else high = middle;\n  }\n  const mass = SAMPLE_CDFS[first + low + 1] - SAMPLE_CDFS[first + low];\n  return {\n    wavelength: SPECTRUM_WAVELENGTH_MIN + SAMPLE_BIN_WIDTH * (low + (u - SAMPLE_CDFS[first + low]) / mass),\n    pdf: mass / SAMPLE_BIN_WIDTH\n  };\n}\n\n// Cauchy IOR grows quickly at short wavelengths and strong dispersion scales turn\n// violet light into total internal reflection. Below 500 nm the refracting\n// wavelength eases toward 475 nm, while color still follows the sampled wavelength.\nconst IOR_KNEE = 500;\nconst IOR_KNEE_WIDTH = 25;\n\nexport function spectrumIorWavelength(wavelength: number) {\n  if (wavelength >= IOR_KNEE) return wavelength;\n  return IOR_KNEE - IOR_KNEE_WIDTH * (1 - Math.exp((wavelength - IOR_KNEE) / IOR_KNEE_WIDTH));\n}\n\nconst CDF_LITERALS = SAMPLE_CDFS.map((value) => value.toFixed(7)).join(',');\nconst lobeSum = (lobes: number[][]) =>\n  lobes.map((lobe) => `${lobe[0]}f*spectrumLobe(wavelength,${lobe[1]}f,${lobe[2]}f,${lobe[3]}f)`).join('+');\n\nexport const spectrumWgsl = /* wgsl */ `\nfn spectrumLobe(wavelength:f32,mean:f32,below:f32,above:f32)->f32 {\n  let t=(wavelength-mean)/select(above,below,wavelength<mean);\n  return exp(-0.5*t*t);\n}\nfn spectrumRgb(wavelength:f32)->vec3f {\n  let xyz=vec3f(${XYZ_LOBES.map(lobeSum).join(',')});\n  let rgb=mat3x3f(${[0, 1, 2].map((c) => XYZ_TO_RGB.map((row) => `${row[c]}f`).join(',')).join(',')})*xyz;\n  return max(rgb,vec3f(0.0))*vec3f(${SCALE.map((s) => `${s}f`).join(',')});\n}\nfn spectrumIorWavelength(wavelength:f32)->f32 {\n  if(wavelength>=${IOR_KNEE}.0) { return wavelength; }\n  return ${IOR_KNEE}.0-${IOR_KNEE_WIDTH}.0*(1.0-exp((wavelength-${IOR_KNEE}.0)/${IOR_KNEE_WIDTH}.0));\n}\nconst SPECTRUM_CDF=array<f32,${(SAMPLE_BINS + 1) * 4}>(${CDF_LITERALS});\nfn spectrumSample(distribution:u32,u:f32,pdf:ptr<function,f32>)->f32 {\n  let first=distribution*${SAMPLE_BINS + 1}u;\n  var low=0u;\n  var high=${SAMPLE_BINS}u;\n  while(high-low>1u) {\n    let middle=(low+high)/2u;\n    if(SPECTRUM_CDF[first+middle]<=u) { low=middle; } else { high=middle; }\n  }\n  let mass=SPECTRUM_CDF[first+low+1u]-SPECTRUM_CDF[first+low];\n  *pdf=mass/${SAMPLE_BIN_WIDTH}.0;\n  return ${SPECTRUM_WAVELENGTH_MIN}.0+${SAMPLE_BIN_WIDTH}.0*(f32(low)+(u-SPECTRUM_CDF[first+low])/mass);\n}\n// Refracting wavelength of a camera path that carries one RGB channel.\nfn spectrumCameraWavelength(channel:u32,u:f32)->f32 {\n  var pdf=0.0;\n  return spectrumIorWavelength(spectrumSample(channel,u,&pdf));\n}\n`;\n\nexport const spectrumGlsl = /* glsl */ `\nconst float SPECTRUM_CDF[${(SAMPLE_BINS + 1) * 4}]=float[${(SAMPLE_BINS + 1) * 4}](${CDF_LITERALS});\n// Refracting wavelength of a camera path that carries one RGB channel.\nfloat spectrumCameraWavelength(int channel,float u) {\n  int first=channel*${SAMPLE_BINS + 1};\n  int low=0;\n  int high=${SAMPLE_BINS};\n  while(high-low>1) {\n    int middle=(low+high)/2;\n    if(SPECTRUM_CDF[first+middle]<=u) { low=middle; } else { high=middle; }\n  }\n  float start=SPECTRUM_CDF[first+low];\n  float wavelength=${SPECTRUM_WAVELENGTH_MIN}.0+${SAMPLE_BIN_WIDTH}.0*(float(low)+(u-start)/(SPECTRUM_CDF[first+low+1]-start));\n  if(wavelength>=${IOR_KNEE}.0) { return wavelength; }\n  return ${IOR_KNEE}.0-${IOR_KNEE_WIDTH}.0*(1.0-exp((wavelength-${IOR_KNEE}.0)/${IOR_KNEE_WIDTH}.0));\n}\n`;\n"]}

package/lib/renderer/webglLightSampling.d.ts

@@ -0,0 +1,39 @@
+import type { EmissiveTriangle } from './lightSampling.js';
+import type { AnalyticLightState } from '../scene/lightState.js';
+/** Alias probabilities are stored as numeric texels; indices retain their bit representation. */
+export declare function prepareWebGLLightSampling(lights: AnalyticLightState[], triangles: EmissiveTriangle[], environmentPower: number): {
+    data: Float32Array<ArrayBuffer>;
+    pmfOffset: number;
+    lightData: Float32Array<ArrayBufferLike>;
+    lightWidth: number;
+    lightHeight: number;
+    analyticPmfs: number[];
+    profiles: {
+        data: Float32Array<ArrayBuffer>;
+        offsets: Int32Array<ArrayBuffer>;
+        texelCount: number;
+    };
+    environmentPmf: number;
+    choiceCount: number;
+    totalWeight: number;
+    emissiveTexels: number;
+    domains: {
+        environment: {
+            count: number;
+            weight: number;
+            pmf: number;
+        };
+        infinite: {
+            count: number;
+            weight: number;
+            pmf: number;
+        };
+        localAndEmissive: {
+            count: number;
+            weight: number;
+            pmf: number;
+        };
+    };
+    shadowCatcherChoiceOffset: number;
+    shadowCatcherChoiceCount: number;
+};

package/lib/renderer/webglLightSampling.js

@@ -0,0 +1,15 @@
+import { prepareSceneLightSampling } from './sceneAssembly.js';
+/** Alias probabilities are stored as numeric texels; indices retain their bit representation. */
+export function prepareWebGLLightSampling(lights, triangles, environmentPower) {
+    const lighting = prepareSceneLightSampling(lights, triangles, environmentPower, 4096);
+    const pmfOffset = lighting.emissiveTexels + lighting.choiceCount;
+    const data = new Float32Array(Math.max(1, pmfOffset + lighting.choiceCount) * 4);
+    data.set(lighting.lightData.subarray(0, pmfOffset * 4));
+    const words = new Int32Array(data.buffer);
+    for (let i = 0; i < lighting.choiceCount; i++) {
+        const type = words[(lighting.emissiveTexels + i) * 4 + 2], source = words[(lighting.emissiveTexels + i) * 4 + 3];
+        data[(pmfOffset + i) * 4] = type === 1 ? lighting.environmentPmf : type === 2 ? lighting.analyticPmfs[source] : triangles[source].selectionPmf;
+    }
+    return { ...lighting, data, pmfOffset };
+}
+//# sourceMappingURL=webglLightSampling.js.map

package/lib/renderer/webglLightSampling.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"webglLightSampling.js","sourceRoot":"","sources":["../../src/renderer/webglLightSampling.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,yBAAyB,EAAC,MAAM,oBAAoB,CAAC;AAG7D,iGAAiG;AACjG,MAAM,UAAU,yBAAyB,CAAC,MAA4B,EAAE,SAA6B,EAAE,gBAAwB;IAC7H,MAAM,QAAQ,GAAC,yBAAyB,CAAC,MAAM,EAAC,SAAS,EAAC,gBAAgB,EAAC,IAAI,CAAC,CAAC;IACjF,MAAM,SAAS,GAAC,QAAQ,CAAC,cAAc,GAAC,QAAQ,CAAC,WAAW,CAAC;IAC7D,MAAM,IAAI,GAAC,IAAI,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAC,SAAS,GAAC,QAAQ,CAAC,WAAW,CAAC,GAAC,CAAC,CAAC,CAAC;IAC1E,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,EAAC,SAAS,GAAC,CAAC,CAAC,CAAC,CAAC;IACrD,MAAM,KAAK,GAAC,IAAI,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACxC,KAAI,IAAI,CAAC,GAAC,CAAC,EAAC,CAAC,GAAC,QAAQ,CAAC,WAAW,EAAC,CAAC,EAAE,EAAE,CAAC;QACvC,MAAM,IAAI,GAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,cAAc,GAAC,CAAC,CAAC,GAAC,CAAC,GAAC,CAAC,CAAC,EAAC,MAAM,GAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,cAAc,GAAC,CAAC,CAAC,GAAC,CAAC,GAAC,CAAC,CAAC,CAAC;QAChG,IAAI,CAAC,CAAC,SAAS,GAAC,CAAC,CAAC,GAAC,CAAC,CAAC,GAAC,IAAI,KAAG,CAAC,CAAA,CAAC,CAAA,QAAQ,CAAC,cAAc,CAAA,CAAC,CAAA,IAAI,KAAG,CAAC,CAAA,CAAC,CAAA,QAAQ,CAAC,YAAY,CAAC,MAAM,CAAC,CAAA,CAAC,CAAA,SAAS,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC;IAC/H,CAAC;IACD,OAAO,EAAC,GAAG,QAAQ,EAAC,IAAI,EAAC,SAAS,EAAC,CAAC;AACtC,CAAC","sourcesContent":["import {prepareSceneLightSampling} from './sceneAssembly.js';\nimport type {EmissiveTriangle} from './lightSampling.js';\nimport type {AnalyticLightState} from '../scene/lightState.js';\n/** Alias probabilities are stored as numeric texels; indices retain their bit representation. */\nexport function prepareWebGLLightSampling(lights: AnalyticLightState[], triangles: EmissiveTriangle[], environmentPower: number) {\n  const lighting=prepareSceneLightSampling(lights,triangles,environmentPower,4096);\n  const pmfOffset=lighting.emissiveTexels+lighting.choiceCount;\n  const data=new Float32Array(Math.max(1,pmfOffset+lighting.choiceCount)*4);\n  data.set(lighting.lightData.subarray(0,pmfOffset*4));\n  const words=new Int32Array(data.buffer);\n  for(let i=0;i<lighting.choiceCount;i++) {\n    const type=words[(lighting.emissiveTexels+i)*4+2],source=words[(lighting.emissiveTexels+i)*4+3];\n    data[(pmfOffset+i)*4]=type===1?lighting.environmentPmf:type===2?lighting.analyticPmfs[source]:triangles[source].selectionPmf;\n  }\n  return {...lighting,data,pmfOffset};\n}\n"]}

package/lib/renderer/wgpuBindlessSource.js

@@ -32,12 +32,12 @@ extent: vec4u,
 @group(1) @binding(3) var<storage, read> materialTextureInfo: array<MaterialTextureInfo>;`;
 const sampling = /* wgsl */ `fn decodeMapColor(idx: i32, c: vec3f) -> vec3f {
 if (materialTextureInfo[idx].resources.w != 0u) { return c; }
-return mG(c);
+return mq(c);
 }
-fn ah(val: i32) -> vec2f {
+fn af(val: i32) -> vec2f {
 return vec2f(f32(val >> 8), f32(val & 0xff) / 255.0);
 }
-fn jO(idx: i32) -> vec2f { return vec2f(materialTextureInfo[idx].extent.xy); }
+fn jv(idx: i32) -> vec2f { return vec2f(materialTextureInfo[idx].extent.xy); }
 fn getMapUvSet(idx: i32, uv: vec2f, uv1: vec2f) -> vec2f {
 return select(uv, uv1, materialTextureInfo[idx].resources.z != 0u);
 }

package/lib/renderer/wgpuShaderSource.d.ts

@@ -1,6 +1,13 @@
 import { type RayTraceInputHooks } from './wgsl/rayTrace.compute.js';
 export interface WGPUShaderFeatures {
     hasDisplacement: boolean;
+    debugGroup?: number;
+    /** Invocation block edge; the native host matches dispatch to this value. Default 8. */
+    workgroupSize?: number;
+    lightTracing?: boolean;
+    photonMapping?: boolean;
+    photonMaxDepth?: number;
+    photonDirectSubset?: boolean;
     hasAlpha: boolean;
     hasSceneAlpha: boolean;
     hasAnisotropyDirections: boolean;

package/lib/renderer/wgpuShaderSource.js

@@ -1,3 +1,4 @@
+import { researchTransport } from './researchTransport.js';
 import rayTraceCompute from './wgsl/rayTrace.compute.js';
 import { WGPU_SCENE_FEATURE_BITS } from './WGPUSceneFeatures.js';
 import { MATERIAL_TEXTURE_DEFINES } from './materialTextureSlots.js';
@@ -20,6 +21,7 @@ export function wgpuShaderPreamble(constants, bake = false) {
     return `\n${defineCode}\n${constantsCode}\n`;
 }
 export function wgpuShaderSource(constants, features, bake = false, sceneIntersections = '', pathDiagnostic = false, inputHooks) {
+    constants = researchTransport()?.constants(constants, features) ?? constants;
     return wgpuShaderPreamble(constants, bake) + rayTraceCompute({
         ...features,
         spectral: bake ? Boolean(constants.SPECTRAL) : undefined,

package/lib/renderer/wgpuShaderSource.js.map

@@ -1 +1 @@
-{"version":3,"file":"wgpuShaderSource.js","sourceRoot":"","sources":["../../src/renderer/wgpuShaderSource.ts"],"names":[],"mappings":"AAAA,OAAO,eAA0C,MAAM,4BAA4B,CAAC;AACpF,OAAO,EAAE,uBAAuB,EAAE,MAAM,wBAAwB,CAAC;AACjE,OAAO,EAAE,wBAAwB,EAAE,MAAM,2BAA2B,CAAC;AAYrE,MAAM,CAAC,MAAM,uBAAuB,GAAG,CAAC,UAAU,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,oBAAoB,CAAC,CAAC;AAChH,MAAM,gBAAgB,GAAG;IACvB,GAAG,MAAM,CAAC,IAAI,CAAC,uBAAuB,CAAC,EAAE,gBAAgB,EAAE,oBAAoB;IAC/E,yBAAyB,EAAE,GAAG,wBAAwB,EAAE,2BAA2B;IACnF,wBAAwB,EAAE,iBAAiB,EAAE,gBAAgB;CAC9D,CAAC;AACF,MAAM,UAAU,wBAAwB,CAAC,SAA2C;IAClF,OAAO,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,IAAI,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,uBAAuB,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;AAChI,CAAC;AACD,MAAM,UAAU,kBAAkB,CAAC,SAA2C,EAAE,IAAI,GAAG,KAAK;IAC1F,MAAM,UAAU,GAAG,uBAAuB,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,SAAS,GAAG,MAAM,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACtG,MAAM,IAAI,GAAG,wBAAwB,CAAC,SAAS,CAAC,CAAC;IACjD,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE;QACnC,MAAM,KAAK,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC;QAC7B,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU,IAAI,GAAG,MAAM,OAAO,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,EAAE,GAAG,CAAC;IAClI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACd,OAAO,KAAK,UAAU,KAAK,aAAa,IAAI,CAAC;AAC/C,CAAC;AACD,MAAM,UAAU,gBAAgB,CAAC,SAA2C,EAAE,QAA4B,EAAE,IAAI,GAAG,KAAK,EAAE,kBAAkB,GAAG,EAAE,EAAE,cAAc,GAAG,KAAK,EAAE,UAA+B;IACxM,OAAO,kBAAkB,CAAC,SAAS,EAAE,IAAI,CAAC,GAAG,eAAe,CAAC;QAC3D,GAAG,QAAQ;QACX,QAAQ,EAAE,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS;QACxD,iBAAiB,EAAE,IAAI,IAAI,SAAS,CAAC,kBAAkB,KAAK,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC,SAAS;QACxH,oBAAoB,EAAE,IAAI,IAAI,SAAS,CAAC,qBAAqB,KAAK,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,SAAS;QACjI,iBAAiB,EAAE,IAAI,IAAI,SAAS,CAAC,kBAAkB,KAAK,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC,SAAS;KACzH,EAAE,kBAAkB,EAAE,cAAc,EAAE,UAAU,CAAC,CAAC;AACrD,CAAC","sourcesContent":["import rayTraceCompute, {type RayTraceInputHooks} from './wgsl/rayTrace.compute.js';\nimport { WGPU_SCENE_FEATURE_BITS } from './WGPUSceneFeatures.js';\nimport { MATERIAL_TEXTURE_DEFINES } from './materialTextureSlots.js';\n\nexport interface WGPUShaderFeatures {\n  hasDisplacement: boolean;\n  hasAlpha: boolean;\n  hasSceneAlpha: boolean;\n  hasAnisotropyDirections: boolean;\n  hasCoatNormals: boolean;\n  blasLayoutMask: number;\n  /** Environment presence changes background/MIS semantics independently of packed material data. */\n  dynamicEnvironment?: boolean;\n}\nexport const WGPU_SHADER_DEFINE_KEYS = ['NUM_MAPS', 'TLAS_STACK_SIZE', 'BLAS_STACK_SIZE', 'NUM_UNIFORM_LIGHTS'];\nconst constantsNotUsed = [\n  ...Object.keys(WGPU_SCENE_FEATURE_BITS), 'USE_BACKGROUND', 'USE_BACKGROUND_MAP',\n  'USE_GROUNDED_BACKGROUND', ...MATERIAL_TEXTURE_DEFINES, 'USE_ANISOTROPY_DIRECTIONS',\n  'USE_COAT_NORMAL_FRAMES', 'HAS_ALPHA_BLEND', 'HAS_ALPHA_MASK'\n];\nexport function wgpuPipelineConstantKeys(constants: Record<string, boolean | number>) {\n  return Object.keys(constants).sort().filter(key => !WGPU_SHADER_DEFINE_KEYS.includes(key) && !constantsNotUsed.includes(key));\n}\nexport function wgpuShaderPreamble(constants: Record<string, boolean | number>, bake = false): string {\n  const defineCode = WGPU_SHADER_DEFINE_KEYS.map(key => `const ${key} = ${constants[key]};`).join('\\n');\n  const keys = wgpuPipelineConstantKeys(constants);\n  const constantsCode = keys.map(key => {\n    const value = constants[key];\n    return `${bake ? 'const' : 'override'} ${key} = ${typeof value === 'boolean' ? (value ? 'true' : 'false') : value.toString()};`;\n  }).join('\\n');\n  return `\\n${defineCode}\\n${constantsCode}\\n`;\n}\nexport function wgpuShaderSource(constants: Record<string, boolean | number>, features: WGPUShaderFeatures, bake = false, sceneIntersections = '', pathDiagnostic = false, inputHooks?: RayTraceInputHooks): string {\n  return wgpuShaderPreamble(constants, bake) + rayTraceCompute({\n    ...features,\n    spectral: bake ? Boolean(constants.SPECTRAL) : undefined,\n    sceneCapabilities: bake && constants.SCENE_CAPABILITIES !== undefined ? Number(constants.SCENE_CAPABILITIES) : undefined,\n    materialCapabilities: bake && constants.MATERIAL_CAPABILITIES !== undefined ? Number(constants.MATERIAL_CAPABILITIES) : undefined,\n    lightCapabilities: bake && constants.LIGHT_CAPABILITIES !== undefined ? Number(constants.LIGHT_CAPABILITIES) : undefined\n  }, sceneIntersections, pathDiagnostic, inputHooks);\n}\n"]}
+{"version":3,"file":"wgpuShaderSource.js","sourceRoot":"","sources":["../../src/renderer/wgpuShaderSource.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,iBAAiB,EAAC,MAAM,wBAAwB,CAAC;AACzD,OAAO,eAA0C,MAAM,4BAA4B,CAAC;AACpF,OAAO,EAAE,uBAAuB,EAAE,MAAM,wBAAwB,CAAC;AACjE,OAAO,EAAE,wBAAwB,EAAE,MAAM,2BAA2B,CAAC;AAmBrE,MAAM,CAAC,MAAM,uBAAuB,GAAG,CAAC,UAAU,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,oBAAoB,CAAC,CAAC;AAChH,MAAM,gBAAgB,GAAG;IACvB,GAAG,MAAM,CAAC,IAAI,CAAC,uBAAuB,CAAC,EAAE,gBAAgB,EAAE,oBAAoB;IAC/E,yBAAyB,EAAE,GAAG,wBAAwB,EAAE,2BAA2B;IACnF,wBAAwB,EAAE,iBAAiB,EAAE,gBAAgB;CAC9D,CAAC;AACF,MAAM,UAAU,wBAAwB,CAAC,SAA2C;IAClF,OAAO,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,IAAI,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,uBAAuB,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;AAChI,CAAC;AACD,MAAM,UAAU,kBAAkB,CAAC,SAA2C,EAAE,IAAI,GAAG,KAAK;IAC1F,MAAM,UAAU,GAAG,uBAAuB,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,SAAS,GAAG,MAAM,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACtG,MAAM,IAAI,GAAG,wBAAwB,CAAC,SAAS,CAAC,CAAC;IACjD,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE;QACnC,MAAM,KAAK,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC;QAC7B,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU,IAAI,GAAG,MAAM,OAAO,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,EAAE,GAAG,CAAC;IAClI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACd,OAAO,KAAK,UAAU,KAAK,aAAa,IAAI,CAAC;AAC/C,CAAC;AACD,MAAM,UAAU,gBAAgB,CAAC,SAA2C,EAAE,QAA4B,EAAE,IAAI,GAAG,KAAK,EAAE,kBAAkB,GAAG,EAAE,EAAE,cAAc,GAAG,KAAK,EAAE,UAA+B;IACxM,SAAS,GAAG,iBAAiB,EAAE,EAAE,SAAS,CAAC,SAAS,EAAE,QAAQ,CAAC,IAAI,SAAS,CAAC;IAC7E,OAAO,kBAAkB,CAAC,SAAS,EAAE,IAAI,CAAC,GAAG,eAAe,CAAC;QAC3D,GAAG,QAAQ;QACX,QAAQ,EAAE,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS;QACxD,iBAAiB,EAAE,IAAI,IAAI,SAAS,CAAC,kBAAkB,KAAK,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC,SAAS;QACxH,oBAAoB,EAAE,IAAI,IAAI,SAAS,CAAC,qBAAqB,KAAK,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,SAAS;QACjI,iBAAiB,EAAE,IAAI,IAAI,SAAS,CAAC,kBAAkB,KAAK,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC,SAAS;KACzH,EAAE,kBAAkB,EAAE,cAAc,EAAE,UAAU,CAAC,CAAC;AACrD,CAAC","sourcesContent":["import {researchTransport} from './researchTransport.js';\nimport rayTraceCompute, {type RayTraceInputHooks} from './wgsl/rayTrace.compute.js';\nimport { WGPU_SCENE_FEATURE_BITS } from './WGPUSceneFeatures.js';\nimport { MATERIAL_TEXTURE_DEFINES } from './materialTextureSlots.js';\n\nexport interface WGPUShaderFeatures {\n  hasDisplacement: boolean;\n  debugGroup?: number;\n  /** Invocation block edge; the native host matches dispatch to this value. Default 8. */\n  workgroupSize?:number;\n  lightTracing?: boolean;\n  photonMapping?:boolean;\n  photonMaxDepth?:number;\n  photonDirectSubset?:boolean;\n  hasAlpha: boolean;\n  hasSceneAlpha: boolean;\n  hasAnisotropyDirections: boolean;\n  hasCoatNormals: boolean;\n  blasLayoutMask: number;\n  /** Environment presence changes background/MIS semantics independently of packed material data. */\n  dynamicEnvironment?: boolean;\n}\nexport const WGPU_SHADER_DEFINE_KEYS = ['NUM_MAPS', 'TLAS_STACK_SIZE', 'BLAS_STACK_SIZE', 'NUM_UNIFORM_LIGHTS'];\nconst constantsNotUsed = [\n  ...Object.keys(WGPU_SCENE_FEATURE_BITS), 'USE_BACKGROUND', 'USE_BACKGROUND_MAP',\n  'USE_GROUNDED_BACKGROUND', ...MATERIAL_TEXTURE_DEFINES, 'USE_ANISOTROPY_DIRECTIONS',\n  'USE_COAT_NORMAL_FRAMES', 'HAS_ALPHA_BLEND', 'HAS_ALPHA_MASK'\n];\nexport function wgpuPipelineConstantKeys(constants: Record<string, boolean | number>) {\n  return Object.keys(constants).sort().filter(key => !WGPU_SHADER_DEFINE_KEYS.includes(key) && !constantsNotUsed.includes(key));\n}\nexport function wgpuShaderPreamble(constants: Record<string, boolean | number>, bake = false): string {\n  const defineCode = WGPU_SHADER_DEFINE_KEYS.map(key => `const ${key} = ${constants[key]};`).join('\\n');\n  const keys = wgpuPipelineConstantKeys(constants);\n  const constantsCode = keys.map(key => {\n    const value = constants[key];\n    return `${bake ? 'const' : 'override'} ${key} = ${typeof value === 'boolean' ? (value ? 'true' : 'false') : value.toString()};`;\n  }).join('\\n');\n  return `\\n${defineCode}\\n${constantsCode}\\n`;\n}\nexport function wgpuShaderSource(constants: Record<string, boolean | number>, features: WGPUShaderFeatures, bake = false, sceneIntersections = '', pathDiagnostic = false, inputHooks?: RayTraceInputHooks): string {\n  constants = researchTransport()?.constants(constants, features) ?? constants;\n  return wgpuShaderPreamble(constants, bake) + rayTraceCompute({\n    ...features,\n    spectral: bake ? Boolean(constants.SPECTRAL) : undefined,\n    sceneCapabilities: bake && constants.SCENE_CAPABILITIES !== undefined ? Number(constants.SCENE_CAPABILITIES) : undefined,\n    materialCapabilities: bake && constants.MATERIAL_CAPABILITIES !== undefined ? Number(constants.MATERIAL_CAPABILITIES) : undefined,\n    lightCapabilities: bake && constants.LIGHT_CAPABILITIES !== undefined ? Number(constants.LIGHT_CAPABILITIES) : undefined\n  }, sceneIntersections, pathDiagnostic, inputHooks);\n}\n"]}

package/lib/renderer/wgsl/aovDisplay.wgsl.js

@@ -9,17 +9,17 @@ struct DisplayAovParams { sizeKind: vec4u, mapping: vec4f, flags: vec4u, }
 @group(0) @binding(2) var image: texture_storage_2d<${format}, write>;
 @group(0) @binding(3) var<uniform> params: DisplayAovParams;
 
-fn tF(input: u32) -> u32 {
+fn tG(input: u32) -> u32 {
 var v = input;
 v = (v ^ (v >> 16u)) * 0x7feb352du;
 v = (v ^ (v >> 15u)) * 0x846ca68bu;
 return v ^ (v >> 16u);
 }
-fn tG(rgb: vec3f) -> vec3f {
+fn tH(rgb: vec3f) -> vec3f {
 return select(1.055 * pow(max(rgb, vec3f(0.0)), vec3f(1.0 / 2.4)) - 0.055,
 12.92 * rgb, rgb <= vec3f(0.0031308));
 }
-fn sE(rgb: vec3f) -> vec3f {
+fn sA(rgb: vec3f) -> vec3f {
 return select(pow((rgb + 0.055) / 1.055, vec3f(2.4)), rgb / 12.92,
 rgb <= vec3f(0.04045));
 }
@@ -45,24 +45,24 @@ color = clamp(pixel.geometry.yzw * inv, vec3f(0.0), vec3f(1.0));
 } else if (kind == 3u || kind == 4u) {
 let r = select(pixel.diffuse, pixel.specular, kind == 4u);
 alpha = r.w / f32(max(params.sizeKind.z, 1u));
-color = tG(clamp(r.xyz / max(r.w, 1.0), vec3f(0.0), vec3f(1.0)));
+color = tH(clamp(r.xyz / max(r.w, 1.0), vec3f(0.0), vec3f(1.0)));
 } else if (kind == 5u) {
 let identity = center[index].identity;
 alpha = select(1.0, 0.0, identity.x == 0xffffffffu);
-let hash = tF(identity[params.flags.x]);
+let hash = tG(identity[params.flags.x]);
 color = 0.2 + 0.8 * vec3f(f32(hash & 255u), f32((hash >> 8u) & 255u),
 f32((hash >> 16u) & 255u)) / 255.0;
 } else {
 let m = center[index].motion;
 alpha = select(0.0, 1.0, m.w == 3.0);
-let gR = length(m.xy);
+let gz = length(m.xy);
 var hue = 0.0;
-if (gR > 0.0) { hue = fract(atan2(m.y, m.x) / 6.28318530718 + 1.0); }
+if (gz > 0.0) { hue = fract(atan2(m.y, m.x) / 6.28318530718 + 1.0); }
 let p = abs(fract(vec3f(hue) + vec3f(0.0, 2.0 / 3.0, 1.0 / 3.0)) * 6.0 - 3.0);
-color = clamp(p - 1.0, vec3f(0.0), vec3f(1.0)) * clamp(gR / params.mapping.w, 0.0, 1.0);
+color = clamp(p - 1.0, vec3f(0.0), vec3f(1.0)) * clamp(gz / params.mapping.w, 0.0, 1.0);
 }
 if (alpha <= 0.0) { color = vec3f(0.0); }
-if (params.flags.y != 0u) { color = sE(color); }
+if (params.flags.y != 0u) { color = sA(color); }
 textureStore(image, vec2i(id.xy), vec4f(color, clamp(alpha, 0.0, 1.0)));
 }
 `;

package/lib/renderer/wgsl/aovReflectance.wgsl.js

@@ -4,95 +4,95 @@ import proposals from './aovReflectanceSamples.wgsl.js';
 // evaluate diffuse and reflection together without traversing the full layered
 // BSDF twice. Callers reject coat, fuzz, anisotropy, SSS and thin film first.
 export default proposals + /* wgsl */ `
-struct oa {
+struct nQ {
 diffuse: vec3f,
 specular: vec3f,
 }
-fn hN(prepared: ag, qQ: bool, kY: bool) -> oa {
-var result = oa();
+fn hu(prepared: OpenPbrPreparedBsdf, qH: bool, kE: bool) -> nQ {
+var result = nQ();
 let view = prepared.viewLocal;
 if (view.z <= 1e-6) { return result; }
 let material = prepared.material;
 let alpha = vec2f(material.alphaX, material.alphaY);
-let mg = aF(view, alpha);
-let jl = prepared.weights[OPENPBR_LOBE_DIFFUSE];
-let gb = prepared.weights[OPENPBR_LOBE_METAL];
+let lN = aD(view, alpha);
+let iR = prepared.weights[OPENPBR_LOBE_DIFFUSE];
+let fN = prepared.weights[OPENPBR_LOBE_METAL];
 let specularWeight = prepared.weights[OPENPBR_LOBE_SPECULAR];
-var ei = 1.0;
-if (kY && prepared.dielectricMsProbability > 0.0) {
-ei = max(bf(prepared.refractionEtaTOverEtaI, prepared.specularAlpha), 1e-12);
+var dS = 1.0;
+if (kE && prepared.dielectricMsProbability > 0.0) {
+dS = max(aY(prepared.refractionEtaTOverEtaI, prepared.specularAlpha), 1e-12);
 }
-let cr = normalize(vec3f(alpha, 1.0) * view);
-let qP = min(cr.z, 1.0);
-let kh = 1.0 + qP;
+let ce = normalize(vec3f(alpha, 1.0) * view);
+let qG = min(ce.z, 1.0);
+let jN = 1.0 + qG;
 let count = 256u * min(rtUniform.loopUnrollGuard, 4u) / 4u;
 for (var i = 0u; i < count; i++) {
-let vV = i / 2u;
-let u = vec2f((f32(vV) + 0.5) / f32(count / 2u),
-fract(f32(reverseBits(vV)) * 2.3283064365386963e-10 + 0.37));
-let kK = hp[vV];
+let wn = i / 2u;
+let u = vec2f((f32(wn) + 0.5) / f32(count / 2u),
+fract(f32(reverseBits(wn)) * 2.3283064365386963e-10 + 0.37));
+let proposal = gV[wn];
 
 
-let hO = prepared.specularAlpha < 0.001 || count != 256u;
-var light = vec3f(kK.xy, sqrt(max(0.0, 1.0 - dot(kK.xy, kK.xy))));
-if (hO) { light = aO(u); }
+let hv = prepared.specularAlpha < 0.001 || count != 256u;
+var light = vec3f(proposal.xy, sqrt(max(0.0, 1.0 - dot(proposal.xy, proposal.xy))));
+if (hv) { light = aM(u); }
 if ((i & 1u) != 0u) {
-let kC = (1.0 - u.y) * kh;
-let ex = sqrt(max(0.0, (kC + (1.0 - qP)) * (u.y * kh)));
-var half = normalize(vec3f(alpha, 1.0) * vec3f(cr.xy + ex * kK.zw, kC));
-if (hO) { half = aD(view, alpha, u); }
+let kf = (1.0 - u.y) * jN;
+let dr = sqrt(max(0.0, (kf + (1.0 - qG)) * (u.y * jN)));
+var half = normalize(vec3f(alpha, 1.0) * vec3f(ce.xy + dr * proposal.zw, kf));
+if (hv) { half = aA(view, alpha, u); }
 light = reflect(-view, half);
 }
 
 if (light.z <= 0.0) { continue; }
 let half = normalize(view + light);
-let qH = mg * cW(half, alpha) / (4.0 * view.z);
-let pdf = 0.5 * (light.z * INVPI + qH);
+let specularPdf = lN * sampling_ggx_ndf(half, alpha) / (4.0 * view.z);
+let pdf = 0.5 * (light.z * INVPI + specularPdf);
 
 
-let aA = openpbr_to_local(prepared.basis, openpbr_to_world(prepared.basis, light));
-if (aA.z <= 0.0) { continue; }
-let scale = abs(aA.z) / (pdf * f32(count));
-if (qQ && jl > 0.0) {
-let compensation = prepared.diffuseSpecularEnergyCompensation * aw(
-prepared.specularEtaTOverEtaI, prepared.specularAlpha, abs(aA.z));
-var sZ = 0.0;
-result.diffuse += jl * bJ(material, prepared.diffuseColor,
-view, aA, compensation, &sZ) * scale;
+let au = openpbr_to_local(prepared.basis, openpbr_to_world(prepared.basis, light));
+if (au.z <= 0.0) { continue; }
+let scale = abs(au.z) / (pdf * f32(count));
+if (qH && iR > 0.0) {
+let compensation = prepared.diffuseSpecularEnergyCompensation * ar(
+prepared.specularEtaTOverEtaI, prepared.specularAlpha, abs(au.z));
+var unusedPdf = 0.0;
+result.diffuse += iR * bw(material, prepared.diffuseColor,
+view, au, compensation, &unusedPdf) * scale;
 }
-if (kY) {
-let jn = normalize(view + aA);
-let bm = abs(dot(view, jn));
-var bG = vec3f(0.0);
+if (kE) {
+let evaluatedHalf = normalize(view + au);
+let viewHalf = abs(dot(view, evaluatedHalf));
+var fresnel = vec3f(0.0);
 if (specularWeight > 0.0) {
-var dielectric = vec3f(af(bm, material.relativeIor));
-if (material.thinWalled) { dielectric = av(dielectric, material.transmissionWeight); }
-bG += specularWeight * material.specularColor * dielectric;
+var dielectric = vec3f(ae(viewHalf, material.relativeIor));
+if (material.thinWalled) { dielectric = aq(dielectric, material.transmissionWeight); }
+fresnel += specularWeight * material.specularColor * dielectric;
 }
-if (gb > 0.0) {
-bG += gb * material.specularWeight * cw(bm,
+if (fN > 0.0) {
+fresnel += fN * material.specularWeight * openpbr_metal_fresnel(viewHalf,
 material.baseWeight * material.baseColor, material.specularColor);
 }
-let rJ = cW(jn, alpha) * mg
-* aF(aA, alpha) / (4.0 * max(aA.z * view.z, 1e-8));
-var reflection = bG * rJ;
-if (gb > 0.0 && prepared.metalMsFactor > 0.0) {
-let cL = min(prepared.metalMsFactor * aV(prepared.specularAlpha, aA.z), 1.0 / max(aA.z, 1e-8));
-reflection += gb * prepared.metalMsScale * cL * INVPI;
+let rE = sampling_ggx_ndf(evaluatedHalf, alpha) * lN
+* aD(au, alpha) / (4.0 * max(au.z * view.z, 1e-8));
+var reflection = fresnel * rE;
+if (fN > 0.0 && prepared.metalMsFactor > 0.0) {
+let cv = min(prepared.metalMsFactor * aR(prepared.specularAlpha, au.z), 1.0 / max(au.z, 1e-8));
+reflection += fN * prepared.metalMsScale * cv * INVPI;
 }
 if (prepared.dielectricMsProbability > 0.0) {
-let oZ = prepared.dielectricMsViewLoss * bb(
-prepared.refractionEtaTOverEtaI, prepared.specularAlpha, abs(aA.z)) / ei;
-let cL = min(oZ * prepared.dielectricMsReflectionRatio, 1.0 / abs(aA.z));
-reflection += prepared.dielectricMsReflectionScale * cL * INVPI;
+let oP = prepared.dielectricMsViewLoss * aV(
+prepared.refractionEtaTOverEtaI, prepared.specularAlpha, abs(au.z)) / dS;
+let cv = min(oP * prepared.dielectricMsReflectionRatio, 1.0 / abs(au.z));
+reflection += prepared.dielectricMsReflectionScale * cv * INVPI;
 }
 result.specular += reflection * scale;
 }
 }
 return result;
 }
-fn vT(prepared: ag) -> oa {
-return hN(prepared, true, true);
+fn wl(prepared: OpenPbrPreparedBsdf) -> nQ {
+return hu(prepared, true, true);
 }
 `;
 //# sourceMappingURL=aovReflectance.wgsl.js.map

package/lib/renderer/wgsl/aovReflectanceSamples.wgsl.js

@@ -1,7 +1,7 @@
 // Generated by scripts/generate-aov-reflectance-samples.mjs; do not edit.
 // xy: concentric cosine disk; zw: spherical-cap azimuth cosine/sine.
 export default /* wgsl */ `
-const hp: array<vec4f, 128> = array<vec4f, 128>(
+const gV: array<vec4f, 128> = array<vec4f, 128>(
 vec4f(-0.971247852, -0.202764958, 0.999698818, 0.0245412290),
 vec4f(-0.808660269, 0.547487736, 0.997290432, 0.0735645667),
 vec4f(-0.942509234, 0.187289074, 0.992479563, 0.122410677),

package/lib/renderer/wgsl/aovResolve.wgsl.js

@@ -5,7 +5,7 @@ struct AovResolvedCenter { identity: vec4u, motion: vec4f, }
 @group(0) @binding(0) var<storage,read> accumulation: array<AovResolvedAccumulation>;
 @group(0) @binding(1) var<storage,read> center: array<AovResolvedCenter>;
 @group(0) @binding(3) var<uniform> parameters: vec4u;
-fn ox(source: u32, kind: u32) -> vec4u {
+fn oj(source: u32, kind: u32) -> vec4u {
 if(kind==5u){return center[source].identity;}
 var value=vec4f(0.0);
 if(kind==6u){let motion=center[source].motion;value=vec4f(motion.xy,0.0,motion.w);}
@@ -32,7 +32,7 @@ let width=parameters.x;let height=parameters.y;
 if(id.x>=width||id.y>=height){return;}
 let index=id.x+id.y*width;
 let source=id.x+select(id.y,height-1u-id.y,(parameters.w&256u)!=0u)*width;
-output[index]=ox(source,parameters.w&255u);
+output[index]=oj(source,parameters.w&255u);
 }
 `;
 //# sourceMappingURL=aovResolve.wgsl.js.map

package/lib/renderer/wgsl/aovSnapshot.wgsl.js

@@ -14,7 +14,7 @@ let width=parameters.x;let height=parameters.y;
 if(id.x>=width||id.y>=height){return;}
 let index=id.x+id.y*width;
 let source=id.x+(height-1u-id.y)*width;
-${kinds.map((kind, i) => `output${i}[index]=ox(source,${AOV_KINDS.indexOf(kind)}u);`).join('\n  ')}
+${kinds.map((kind, i) => `output${i}[index]=oj(source,${AOV_KINDS.indexOf(kind)}u);`).join('\n  ')}
 }
 `;
 }