@meimu/mutracer 0.1.5 → 0.2.0

package/lib/renderer/wgpuMaterialConstants.d.ts

@@ -0,0 +1,4 @@
+import type MaterialState from '../scene/MaterialState.js';
+import type Geometry from '../scene/Geometry.js';
+import type MaterialData from './MaterialData.js';
+export declare function wgpuMaterialConstants(materials: MaterialState[], materialData: Pick<MaterialData, 'defines' | 'bufferData'>, mergedGeometry: Geometry, bevelSamples: number, previousNumMaps?: number): Record<string, boolean | number>;

package/lib/renderer/wgpuMaterialConstants.js

@@ -0,0 +1,52 @@
+import { resolveMaterialAlphaMode } from '../scene/MaterialState.js';
+import { materialUsesMedium } from './materialParams.js';
+import { materialTextureFeatures } from './materialTextureSlots.js';
+import { MATERIAL_SHADER_CAPABILITIES } from './shaderCapabilities.js';
+// Same feature decisions for browser pipeline updates and native scene commits.
+export function wgpuMaterialConstants(materials, materialData, mergedGeometry, bevelSamples, previousNumMaps = 0) {
+    const HAS_BEVEL = materials.some((m) => m.bevelSize > 0);
+    const SPECTRAL = materials.some((m) => m.transmissionDispersionScale > 0);
+    const HAS_MEDIUM = materials.some(materialUsesMedium);
+    // Retain the atlas uniform capacity when a stage/material is removed.
+    // Shrinking the array would rebuild every pipeline again on the next edit.
+    const NUM_MAPS = Math.max(Math.ceil(+materialData.defines.NUM_MAPS / 10) * 10, Number(previousNumMaps ?? 0), 20);
+    // Either layer's map requires independent normals, including a smooth
+    // coat over a mapped base. Unmapped materials retain the shared path.
+    const USE_COAT_NORMAL_FRAMES = materials.some(m => m.coatWeight > 0 && (!!m.normalMap || !!m.coatNormalMap));
+    return {
+        MATERIAL_CAPABILITIES: (materials.some(m => m.transmissionWeight !== 0 || !!m.map && m.mapAlphaUsage === 'transmission')
+            ? MATERIAL_SHADER_CAPABILITIES.transmission : 0) |
+            (materials.some(m => m.baseMetalness !== 0) ? MATERIAL_SHADER_CAPABILITIES.metal : 0),
+        // Match the atlas UV selector. Keep the second set when any texture
+        // needs it; material/texture edits already rebuild these constants.
+        USE_UV1: materialData.bufferData.mapUvChannel.some(channel => channel > 0.5),
+        // Isotropic OpenPBR constructs its own basis. Normal mapping uses
+        // triangle UV differentials rather than this vertex tangent.
+        USE_TANGENT: mergedGeometry.tangent != null &&
+            materials.some(material => material.specularAnisotropy > 0 || material.coatAnisotropy > 0 || (!!material.coatNormalMap && material.coatWeight > 0)),
+        USE_COAT_NORMAL_FRAMES,
+        USE_ANISOTROPY_DIRECTIONS: USE_COAT_NORMAL_FRAMES || materials.some(m => m.specularAnisotropy > 0 && (m.specularAnisotropyModel === 'gltf' || !!m.anisotropyMap || m.specularAnisotropyRotation !== 0) ||
+            m.coatAnisotropy > 0 && m.coatAnisotropyRotation !== 0),
+        HAS_MEDIUM,
+        HAS_SHADOW_CATCHER: materials.some(m => m.shadowCatcher),
+        HAS_THIN_FILM: materials.some((material) => material.thinFilmWeight > 0),
+        HAS_COAT: materials.some((material) => material.coatWeight > 0),
+        HAS_FUZZ: materials.some((material) => material.fuzzWeight > 0),
+        HAS_SUBSURFACE: materials.some((material) => material.subsurfaceWeight > 0),
+        HAS_BEVEL,
+        HAS_ALPHA_MASK: materials.some((m) => resolveMaterialAlphaMode(m) === 'mask'),
+        HAS_ALPHA_BLEND: materials.some((m) => resolveMaterialAlphaMode(m) === 'blend'),
+        HAS_ALPHA: materials.some((m) => resolveMaterialAlphaMode(m) !== 'opaque'),
+        SPECTRAL,
+        BEVEL_SAMPLES: bevelSamples,
+        USE_RR: true,
+        // Match WebGL: defer roulette for multiple-scattering SSS paths.
+        RR_MAX: HAS_MEDIUM ? 12 : 2,
+        NUM_UNIFORM_LIGHTS: 64,
+        ...materialData.defines,
+        NUM_MAPS,
+        ...materialTextureFeatures(materialData.defines),
+        USE_NORMAL_MAPS: +materialData.defines.NUM_NORMAL_MAPS > 0
+    };
+}
+//# sourceMappingURL=wgpuMaterialConstants.js.map

package/lib/renderer/wgpuMaterialConstants.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"wgpuMaterialConstants.js","sourceRoot":"","sources":["../../src/renderer/wgpuMaterialConstants.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,wBAAwB,EAAE,MAAM,2BAA2B,CAAC;AAGrE,OAAO,EAAE,kBAAkB,EAAE,MAAM,qBAAqB,CAAC;AACzD,OAAO,EAAE,uBAAuB,EAAE,MAAM,2BAA2B,CAAC;AACpE,OAAO,EAAC,4BAA4B,EAAC,MAAM,yBAAyB,CAAC;AAErE,gFAAgF;AAChF,MAAM,UAAU,qBAAqB,CAAC,SAA0B,EAAE,YAA0D,EAAE,cAAwB,EACpJ,YAAoB,EAAE,eAAe,GAAG,CAAC;IACzC,MAAM,SAAS,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC;IACzD,MAAM,QAAQ,GACZ,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,2BAA2B,GAAG,CAAC,CAAC,CAAC;IAC3D,MAAM,UAAU,GAAG,SAAS,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;IACtD,sEAAsE;IACtE,2EAA2E;IAC3E,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CACvB,IAAI,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC,OAAO,CAAC,QAAQ,GAAG,EAAE,CAAC,GAAG,EAAE,EACnD,MAAM,CAAC,eAAe,IAAI,CAAC,CAAC,EAC5B,EAAE,CACH,CAAC;IAEF,sEAAsE;IACtE,sEAAsE;IACtE,MAAM,sBAAsB,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC;IAE7G,OAAO;QACL,qBAAqB,EACnB,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,kBAAkB,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,aAAa,KAAK,cAAc,CAAC;YAC/F,CAAC,CAAC,4BAA4B,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC;YAClD,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,aAAa,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,4BAA4B,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QACvF,oEAAoE;QACpE,oEAAoE;QACpE,OAAO,EAAE,YAAY,CAAC,UAAU,CAAC,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,GAAG,GAAG,CAAC;QAC5E,kEAAkE;QAClE,6DAA6D;QAC7D,WAAW,EAAE,cAAc,CAAC,OAAO,IAAI,IAAI;YACzC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,QAAQ,CAAC,kBAAkB,GAAG,CAAC,IAAI,QAAQ,CAAC,cAAc,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,aAAa,IAAI,QAAQ,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC;QAErJ,sBAAsB;QACtB,yBAAyB,EAAE,sBAAsB,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CACtE,CAAC,CAAC,kBAAkB,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,uBAAuB,KAAK,MAAM,IAAI,CAAC,CAAC,CAAC,CAAC,aAAa,IAAI,CAAC,CAAC,0BAA0B,KAAK,CAAC,CAAC;YAC7H,CAAC,CAAC,cAAc,GAAG,CAAC,IAAI,CAAC,CAAC,sBAAsB,KAAK,CAAC,CAAC;QACzD,UAAU;QACV,kBAAkB,EAAE,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC;QACxD,aAAa,EAAE,SAAS,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,cAAc,GAAG,CAAC,CAAC;QACxE,QAAQ,EAAE,SAAS,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,UAAU,GAAG,CAAC,CAAC;QAC/D,QAAQ,EAAE,SAAS,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,UAAU,GAAG,CAAC,CAAC;QAC/D,cAAc,EAAE,SAAS,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,gBAAgB,GAAG,CAAC,CAAC;QAC3E,SAAS;QACT,cAAc,EAAE,SAAS,CAAC,IAAI,CAC5B,CAAC,CAAC,EAAE,EAAE,CAAC,wBAAwB,CAAC,CAAC,CAAC,KAAK,MAAM,CAC9C;QACD,eAAe,EAAE,SAAS,CAAC,IAAI,CAC7B,CAAC,CAAC,EAAE,EAAE,CAAC,wBAAwB,CAAC,CAAC,CAAC,KAAK,OAAO,CAC/C;QACD,SAAS,EAAE,SAAS,CAAC,IAAI,CACvB,CAAC,CAAC,EAAE,EAAE,CAAC,wBAAwB,CAAC,CAAC,CAAC,KAAK,QAAQ,CAChD;QAED,QAAQ;QACR,aAAa,EAAE,YAAY;QAE3B,MAAM,EAAE,IAAI;QACZ,iEAAiE;QACjE,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAE3B,kBAAkB,EAAE,EAAE;QAEtB,GAAG,YAAY,CAAC,OAAO;QAEvB,QAAQ;QACR,GAAG,uBAAuB,CAAC,YAAY,CAAC,OAAO,CAAC;QAChD,eAAe,EAAE,CAAC,YAAY,CAAC,OAAO,CAAC,eAAe,GAAG,CAAC;KAC3D,CAAC;AAEJ,CAAC","sourcesContent":["import type MaterialState from '../scene/MaterialState.js';\nimport { resolveMaterialAlphaMode } from '../scene/MaterialState.js';\nimport type Geometry from '../scene/Geometry.js';\nimport type MaterialData from './MaterialData.js';\nimport { materialUsesMedium } from './materialParams.js';\nimport { materialTextureFeatures } from './materialTextureSlots.js';\nimport {MATERIAL_SHADER_CAPABILITIES} from './shaderCapabilities.js';\n\n// Same feature decisions for browser pipeline updates and native scene commits.\nexport function wgpuMaterialConstants(materials: MaterialState[], materialData: Pick<MaterialData, 'defines' | 'bufferData'>, mergedGeometry: Geometry,\n  bevelSamples: number, previousNumMaps = 0): Record<string, boolean | number> {\n  const HAS_BEVEL = materials.some((m) => m.bevelSize > 0);\n  const SPECTRAL =\n    materials.some((m) => m.transmissionDispersionScale > 0);\n  const HAS_MEDIUM = materials.some(materialUsesMedium);\n  // Retain the atlas uniform capacity when a stage/material is removed.\n  // Shrinking the array would rebuild every pipeline again on the next edit.\n  const NUM_MAPS = Math.max(\n    Math.ceil(+materialData.defines.NUM_MAPS / 10) * 10,\n    Number(previousNumMaps ?? 0),\n    20\n  );\n\n  // Either layer's map requires independent normals, including a smooth\n  // coat over a mapped base. Unmapped materials retain the shared path.\n  const USE_COAT_NORMAL_FRAMES = materials.some(m => m.coatWeight > 0 && (!!m.normalMap || !!m.coatNormalMap));\n\n  return {\n    MATERIAL_CAPABILITIES:\n      (materials.some(m => m.transmissionWeight !== 0 || !!m.map && m.mapAlphaUsage === 'transmission')\n        ? MATERIAL_SHADER_CAPABILITIES.transmission : 0) |\n      (materials.some(m => m.baseMetalness !== 0) ? MATERIAL_SHADER_CAPABILITIES.metal : 0),\n    // Match the atlas UV selector. Keep the second set when any texture\n    // needs it; material/texture edits already rebuild these constants.\n    USE_UV1: materialData.bufferData.mapUvChannel.some(channel => channel > 0.5),\n    // Isotropic OpenPBR constructs its own basis. Normal mapping uses\n    // triangle UV differentials rather than this vertex tangent.\n    USE_TANGENT: mergedGeometry.tangent != null &&\n      materials.some(material => material.specularAnisotropy > 0 || material.coatAnisotropy > 0 || (!!material.coatNormalMap && material.coatWeight > 0)),\n\n    USE_COAT_NORMAL_FRAMES,\n    USE_ANISOTROPY_DIRECTIONS: USE_COAT_NORMAL_FRAMES || materials.some(m =>\n      m.specularAnisotropy > 0 && (m.specularAnisotropyModel === 'gltf' || !!m.anisotropyMap || m.specularAnisotropyRotation !== 0) ||\n      m.coatAnisotropy > 0 && m.coatAnisotropyRotation !== 0),\n    HAS_MEDIUM,\n    HAS_SHADOW_CATCHER: materials.some(m => m.shadowCatcher),\n    HAS_THIN_FILM: materials.some((material) => material.thinFilmWeight > 0),\n    HAS_COAT: materials.some((material) => material.coatWeight > 0),\n    HAS_FUZZ: materials.some((material) => material.fuzzWeight > 0),\n    HAS_SUBSURFACE: materials.some((material) => material.subsurfaceWeight > 0),\n    HAS_BEVEL,\n    HAS_ALPHA_MASK: materials.some(\n      (m) => resolveMaterialAlphaMode(m) === 'mask'\n    ),\n    HAS_ALPHA_BLEND: materials.some(\n      (m) => resolveMaterialAlphaMode(m) === 'blend'\n    ),\n    HAS_ALPHA: materials.some(\n      (m) => resolveMaterialAlphaMode(m) !== 'opaque'\n    ),\n\n    SPECTRAL,\n    BEVEL_SAMPLES: bevelSamples,\n\n    USE_RR: true,\n    // Match WebGL: defer roulette for multiple-scattering SSS paths.\n    RR_MAX: HAS_MEDIUM ? 12 : 2,\n\n    NUM_UNIFORM_LIGHTS: 64,\n\n    ...materialData.defines,\n\n    NUM_MAPS,\n    ...materialTextureFeatures(materialData.defines),\n    USE_NORMAL_MAPS: +materialData.defines.NUM_NORMAL_MAPS > 0\n  };\n\n}\n"]}

package/lib/renderer/wgpuSceneUniform.d.ts

@@ -0,0 +1,42 @@
+import type { AnalyticLightState as AnalyticLight } from '../scene/lightState.js';
+export declare const WGPU_SCENE_BUFFER_MIN_BYTES = 200;
+export declare function makeRayTraceUniform(): {
+    buffer: ArrayBuffer;
+    views: {
+        outputSize: Float32Array<ArrayBuffer>;
+        sampleCount: Float32Array<ArrayBuffer>;
+        scaleOffset: Float32Array<ArrayBuffer>;
+        lightsCount: Int32Array<ArrayBuffer>;
+        transmissiveOutputAlphaMin: Float32Array<ArrayBuffer>;
+        jitter: Float32Array<ArrayBuffer>;
+        backgroundColor: Float32Array<ArrayBuffer>;
+        groundedBackgroundParams: Float32Array<ArrayBuffer>;
+        groundedBackgroundCenter: Float32Array<ArrayBuffer>;
+        blasOffset: Int32Array<ArrayBuffer>;
+        isEmpty: Int32Array<ArrayBuffer>;
+        bevel: Int32Array<ArrayBuffer>;
+        dispersionWavelengths: Int32Array<ArrayBuffer>;
+        envLightPmf: Float32Array<ArrayBuffer>;
+        lightChoiceCount: Int32Array<ArrayBuffer>;
+        lightWeightTotal: Float32Array<ArrayBuffer>;
+        lightSamplingTextureWidth: Int32Array<ArrayBuffer>;
+        lightChoiceTexelOffset: Int32Array<ArrayBuffer>;
+        bounces: Int32Array<ArrayBuffer>;
+        varianceMoments: Int32Array<ArrayBuffer>;
+        adaptiveSampling: Int32Array<ArrayBuffer>;
+        clamping: Float32Array<ArrayBuffer>;
+        directLightCandidates: Int32Array<ArrayBuffer>;
+        backgroundFlags: Int32Array<ArrayBuffer>;
+        loopUnrollGuard: Uint32Array<ArrayBuffer>;
+        sceneFeatureFlags: Uint32Array<ArrayBuffer>;
+        sequenceSeed: Uint32Array<ArrayBuffer>;
+        zSobolLogSPP: Uint32Array<ArrayBuffer>;
+        sampleSequenceOffset: Uint32Array<ArrayBuffer>;
+        shadowCatcherChoiceOffset: Int32Array<ArrayBuffer>;
+        shadowCatcherChoiceCount: Int32Array<ArrayBuffer>;
+        environmentRotation: Float32Array<ArrayBuffer>;
+        backgroundMapTransform: Float32Array<ArrayBuffer>;
+        backgroundMapRotation: Float32Array<ArrayBuffer>;
+    };
+};
+export declare function packWgpuAnalyticLights(lights: AnalyticLight[], texturesIndices: number[], selectionPmfs: number[], iesProfileTexelOffset: (index: number) => number | undefined): Float32Array<ArrayBuffer>;

package/lib/renderer/wgpuSceneUniform.js

@@ -0,0 +1,125 @@
+import { matrixStorage } from '../math/storage.js';
+import { getRectLightUOrV, getRectLightArea, getSphereLightArea, getSpotLightArea, getSpotLightRadiance, getLightDirection, getRigidLightFrame, validateSceneLights } from './lightsData.js';
+import { CONSTANT_DIRECTIONAL_LIGHT, CONSTANT_SPHERE_LIGHT, CONSTANT_RECT_LIGHT, CONSTANT_SPOT_LIGHT } from './lightConstants.js';
+import { read } from '../math/access.js';
+export const WGPU_SCENE_BUFFER_MIN_BYTES = 200;
+const FIXED_LIGHT_COUNT = 64;
+const PIXELS_PER_LIGHT = 6;
+export function makeRayTraceUniform() {
+    // Created from https://webgpufundamentals.org/webgpu/lessons/resources/wgsl-offset-computer.html
+    const RayTraceUniformValues = new ArrayBuffer(320);
+    const RayTraceUniformViews = {
+        outputSize: new Float32Array(RayTraceUniformValues, 0, 2),
+        sampleCount: new Float32Array(RayTraceUniformValues, 8, 1),
+        scaleOffset: new Float32Array(RayTraceUniformValues, 16, 4),
+        lightsCount: new Int32Array(RayTraceUniformValues, 32, 1),
+        transmissiveOutputAlphaMin: new Float32Array(RayTraceUniformValues, 36, 1),
+        jitter: new Float32Array(RayTraceUniformValues, 40, 2),
+        backgroundColor: new Float32Array(RayTraceUniformValues, 48, 4),
+        groundedBackgroundParams: new Float32Array(RayTraceUniformValues, 64, 4),
+        groundedBackgroundCenter: new Float32Array(RayTraceUniformValues, 80, 2),
+        blasOffset: new Int32Array(RayTraceUniformValues, 88, 1),
+        isEmpty: new Int32Array(RayTraceUniformValues, 92, 1),
+        bevel: new Int32Array(RayTraceUniformValues, 96, 1),
+        dispersionWavelengths: new Int32Array(RayTraceUniformValues, 100, 1),
+        envLightPmf: new Float32Array(RayTraceUniformValues, 104, 1),
+        lightChoiceCount: new Int32Array(RayTraceUniformValues, 108, 1),
+        lightWeightTotal: new Float32Array(RayTraceUniformValues, 112, 1),
+        lightSamplingTextureWidth: new Int32Array(RayTraceUniformValues, 116, 1),
+        lightChoiceTexelOffset: new Int32Array(RayTraceUniformValues, 120, 1),
+        bounces: new Int32Array(RayTraceUniformValues, 124, 1),
+        varianceMoments: new Int32Array(RayTraceUniformValues, 128, 1),
+        adaptiveSampling: new Int32Array(RayTraceUniformValues, 132, 1),
+        clamping: new Float32Array(RayTraceUniformValues, 136, 1),
+        directLightCandidates: new Int32Array(RayTraceUniformValues, 140, 1),
+        backgroundFlags: new Int32Array(RayTraceUniformValues, 144, 1),
+        loopUnrollGuard: new Uint32Array(RayTraceUniformValues, 148, 1),
+        sceneFeatureFlags: new Uint32Array(RayTraceUniformValues, 152, 1),
+        sequenceSeed: new Uint32Array(RayTraceUniformValues, 156, 1),
+        zSobolLogSPP: new Uint32Array(RayTraceUniformValues, 160, 1),
+        sampleSequenceOffset: new Uint32Array(RayTraceUniformValues, 164, 1),
+        // Shadow-catcher light table, relative to lightChoiceTexelOffset.
+        shadowCatcherChoiceOffset: new Int32Array(RayTraceUniformValues, 168, 1),
+        shadowCatcherChoiceCount: new Int32Array(RayTraceUniformValues, 172, 1),
+        environmentRotation: new Float32Array(RayTraceUniformValues, 176, 16),
+        backgroundMapTransform: new Float32Array(RayTraceUniformValues, 240, 4),
+        backgroundMapRotation: new Float32Array(RayTraceUniformValues, 256, 16)
+    };
+    // 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([1, 0, 0, 0, 0, 1, 0, 0, 0, 0, 1, 0, 0, 0, 0, 1]);
+    RayTraceUniformViews.backgroundMapRotation.set([1, 0, 0, 0, 0, 1, 0, 0, 0, 0, 1, 0, 0, 0, 0, 1]);
+    RayTraceUniformViews.backgroundMapTransform.set([1, 0, 1, 0]);
+    RayTraceUniformViews.bevel[0] = 1;
+    RayTraceUniformViews.dispersionWavelengths[0] = 3;
+    return {
+        buffer: RayTraceUniformValues,
+        views: RayTraceUniformViews
+    };
+}
+export function packWgpuAnalyticLights(lights, texturesIndices, selectionPmfs, iesProfileTexelOffset) {
+    validateSceneLights(lights);
+    const numLights = lights.length;
+    if (numLights > FIXED_LIGHT_COUNT) {
+        console.warn(`Too many lights, only the first ${FIXED_LIGHT_COUNT} lights will be used.`);
+    }
+    const data = new Float32Array(numLights * PIXELS_PER_LIGHT * 4);
+    function packVec3(idx, v) {
+        for (let i = 0; i < 3; i++) {
+            data[i + idx * 4] = read(v, i);
+        }
+    }
+    const lightTypeConstants = {
+        DirectionalLight: CONSTANT_DIRECTIONAL_LIGHT,
+        SphereLight: CONSTANT_SPHERE_LIGHT,
+        RectLight: CONSTANT_RECT_LIGHT,
+        SpotLight: CONSTANT_SPOT_LIGHT
+    };
+    for (let i = 0; i < numLights; i++) {
+        const light = lights[i];
+        const matrix = matrixStorage(light.transform);
+        const color = light.color;
+        const intensity = light.intensity;
+        const off = i * PIXELS_PER_LIGHT;
+        packVec3(off, [matrix[12], matrix[13], matrix[14]]);
+        data[off * 4 + 3] = lightTypeConstants[light.type];
+        // emission
+        const emissionScale = light.type === 'SpotLight' ? getSpotLightRadiance(light) : intensity;
+        packVec3(off + 1, [
+            color[0] * emissionScale,
+            color[1] * emissionScale,
+            color[2] * emissionScale
+        ]);
+        if (light.type === 'RectLight') {
+            data[(off + 1) * 4 + 3] = getRectLightArea(light);
+            packVec3(off + 2, getRectLightUOrV(light, 0));
+            data[(off + 2) * 4 + 3] = light.spread;
+            packVec3(off + 3, getRectLightUOrV(light, 1));
+        }
+        else if (light.type === 'SphereLight') {
+            data[(off + 1) * 4 + 3] = getSphereLightArea(light);
+            data[(off + 2) * 4 + 3] = light.radius;
+        }
+        else if (light.type === 'SpotLight') {
+            data[(off + 1) * 4 + 3] = getSpotLightArea(light);
+            packVec3(off + 2, getLightDirection(light));
+            data[(off + 2) * 4 + 3] = light.radius;
+            packVec3(off + 3, [
+                Math.cos((light.outerAngle * Math.PI) / 180),
+                Math.cos((Math.min(light.innerAngle, light.outerAngle) * Math.PI) / 180),
+                0
+            ]);
+        }
+        else if (light.type === 'DirectionalLight') {
+            packVec3(off + 2, getLightDirection(light));
+        }
+        data[(off + 3) * 4 + 3] = light.type === 'SpotLight'
+            ? iesProfileTexelOffset(i) ?? -1
+            : texturesIndices[i] ?? -1;
+        data[(off + 4) * 4] = selectionPmfs[i] ?? 0;
+        packVec3(off + 5, getRigidLightFrame(light).x);
+    }
+    return data;
+}
+//# sourceMappingURL=wgpuSceneUniform.js.map

package/lib/renderer/wgpuSceneUniform.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"wgpuSceneUniform.js","sourceRoot":"","sources":["../../src/renderer/wgpuSceneUniform.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,aAAa,EAAC,MAAM,oBAAoB,CAAC;AAEjD,OAAO,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,oBAAoB,EAAE,iBAAiB,EAAE,kBAAkB,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AAC7L,OAAO,EAAE,0BAA0B,EAAE,qBAAqB,EAAE,mBAAmB,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAElI,OAAO,EAAE,IAAI,EAAE,MAAM,mBAAmB,CAAC;AACzC,MAAM,CAAC,MAAM,2BAA2B,GAAG,GAAG,CAAC;AAC/C,MAAM,iBAAiB,GAAG,EAAE,CAAC;AAC7B,MAAM,gBAAgB,GAAG,CAAC,CAAC;AAE3B,MAAM,UAAU,mBAAmB;IACjC,iGAAiG;IACjG,MAAM,qBAAqB,GAAG,IAAI,WAAW,CAAC,GAAG,CAAC,CAAC;IACnD,MAAM,oBAAoB,GAAG;QAC3B,UAAU,EAAE,IAAI,YAAY,CAAC,qBAAqB,EAAE,CAAC,EAAE,CAAC,CAAC;QACzD,WAAW,EAAE,IAAI,YAAY,CAAC,qBAAqB,EAAE,CAAC,EAAE,CAAC,CAAC;QAC1D,WAAW,EAAE,IAAI,YAAY,CAAC,qBAAqB,EAAE,EAAE,EAAE,CAAC,CAAC;QAC3D,WAAW,EAAE,IAAI,UAAU,CAAC,qBAAqB,EAAE,EAAE,EAAE,CAAC,CAAC;QACzD,0BAA0B,EAAE,IAAI,YAAY,CAAC,qBAAqB,EAAE,EAAE,EAAE,CAAC,CAAC;QAC1E,MAAM,EAAE,IAAI,YAAY,CAAC,qBAAqB,EAAE,EAAE,EAAE,CAAC,CAAC;QACtD,eAAe,EAAE,IAAI,YAAY,CAAC,qBAAqB,EAAE,EAAE,EAAE,CAAC,CAAC;QAC/D,wBAAwB,EAAE,IAAI,YAAY,CAAC,qBAAqB,EAAE,EAAE,EAAE,CAAC,CAAC;QACxE,wBAAwB,EAAE,IAAI,YAAY,CAAC,qBAAqB,EAAE,EAAE,EAAE,CAAC,CAAC;QACxE,UAAU,EAAE,IAAI,UAAU,CAAC,qBAAqB,EAAE,EAAE,EAAE,CAAC,CAAC;QACxD,OAAO,EAAE,IAAI,UAAU,CAAC,qBAAqB,EAAE,EAAE,EAAE,CAAC,CAAC;QACrD,KAAK,EAAE,IAAI,UAAU,CAAC,qBAAqB,EAAE,EAAE,EAAE,CAAC,CAAC;QACnD,qBAAqB,EAAE,IAAI,UAAU,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;QACpE,WAAW,EAAE,IAAI,YAAY,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;QAC5D,gBAAgB,EAAE,IAAI,UAAU,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;QAC/D,gBAAgB,EAAE,IAAI,YAAY,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;QACjE,yBAAyB,EAAE,IAAI,UAAU,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;QACxE,sBAAsB,EAAE,IAAI,UAAU,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;QACrE,OAAO,EAAE,IAAI,UAAU,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;QACtD,eAAe,EAAE,IAAI,UAAU,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;QAC9D,gBAAgB,EAAE,IAAI,UAAU,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;QAC/D,QAAQ,EAAE,IAAI,YAAY,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;QACzD,qBAAqB,EAAE,IAAI,UAAU,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;QACpE,eAAe,EAAE,IAAI,UAAU,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;QAC9D,eAAe,EAAE,IAAI,WAAW,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;QAC/D,iBAAiB,EAAE,IAAI,WAAW,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;QACjE,YAAY,EAAE,IAAI,WAAW,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;QAC5D,YAAY,EAAE,IAAI,WAAW,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;QAC5D,oBAAoB,EAAE,IAAI,WAAW,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;QACpE,kEAAkE;QAClE,yBAAyB,EAAE,IAAI,UAAU,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;QACxE,wBAAwB,EAAE,IAAI,UAAU,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;QACvE,mBAAmB,EAAE,IAAI,YAAY,CAAC,qBAAqB,EAAE,GAAG,EAAE,EAAE,CAAC;QACrE,sBAAsB,EAAE,IAAI,YAAY,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;QACvE,qBAAqB,EAAE,IAAI,YAAY,CAAC,qBAAqB,EAAE,GAAG,EAAE,EAAE,CAAC;KACxE,CAAC;IACF,4EAA4E;IAC5E,0EAA0E;IAC1E,oBAAoB,CAAC,eAAe,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;IAC5C,oBAAoB,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAC,CAAC,EAAC,CAAC,EAAC,CAAC,EAAE,CAAC,EAAC,CAAC,EAAC,CAAC,EAAC,CAAC,EAAE,CAAC,EAAC,CAAC,EAAC,CAAC,EAAC,CAAC,EAAE,CAAC,EAAC,CAAC,EAAC,CAAC,EAAC,CAAC,CAAC,CAAC,CAAC;IACnF,oBAAoB,CAAC,qBAAqB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAC,CAAC,EAAC,CAAC,EAAC,CAAC,EAAE,CAAC,EAAC,CAAC,EAAC,CAAC,EAAC,CAAC,EAAE,CAAC,EAAC,CAAC,EAAC,CAAC,EAAC,CAAC,EAAE,CAAC,EAAC,CAAC,EAAC,CAAC,EAAC,CAAC,CAAC,CAAC,CAAC;IACrF,oBAAoB,CAAC,sBAAsB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAC,CAAC,EAAC,CAAC,EAAC,CAAC,CAAC,CAAC,CAAC;IAC3D,oBAAoB,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;IAClC,oBAAoB,CAAC,qBAAqB,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;IAClD,OAAO;QACL,MAAM,EAAE,qBAAqB;QAC7B,KAAK,EAAE,oBAAoB;KAC5B,CAAC;AACJ,CAAC;AACD,MAAM,UAAU,sBAAsB,CACpC,MAAuB,EACvB,eAAyB,EACzB,aAAuB,EACvB,qBAA4D;IAE5D,mBAAmB,CAAC,MAAM,CAAC,CAAC;IAC5B,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC;IAChC,IAAI,SAAS,GAAG,iBAAiB,EAAE,CAAC;QAClC,OAAO,CAAC,IAAI,CACV,mCAAmC,iBAAiB,uBAAuB,CAC5E,CAAC;IACJ,CAAC;IACD,MAAM,IAAI,GAAG,IAAI,YAAY,CAAC,SAAS,GAAG,gBAAgB,GAAG,CAAC,CAAC,CAAC;IAEhE,SAAS,QAAQ,CAA8B,GAAW,EAAE,CAAI;QAC9D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC3B,IAAI,CAAC,CAAC,GAAG,GAAG,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QACjC,CAAC;IACH,CAAC;IAED,MAAM,kBAAkB,GAAG;QACzB,gBAAgB,EAAE,0BAA0B;QAC5C,WAAW,EAAE,qBAAqB;QAClC,SAAS,EAAE,mBAAmB;QAC9B,SAAS,EAAE,mBAAmB;KAC/B,CAAC;IAEF,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,EAAE,CAAC,EAAE,EAAE,CAAC;QACnC,MAAM,KAAK,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;QACxB,MAAM,MAAM,GAAG,aAAa,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QAC9C,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;QAC1B,MAAM,SAAS,GAAG,KAAK,CAAC,SAAS,CAAC;QAClC,MAAM,GAAG,GAAG,CAAC,GAAG,gBAAgB,CAAC;QACjC,QAAQ,CACN,GAAG,EACH,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,MAAM,CAAC,EAAE,CAAC,EAAE,MAAM,CAAC,EAAE,CAAC,CAAC,CACrC,CAAC;QACF,IAAI,CAAC,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,kBAAkB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACnD,WAAW;QACX,MAAM,aAAa,GACjB,KAAK,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACvE,QAAQ,CAAC,GAAG,GAAG,CAAC,EAAE;YAChB,KAAK,CAAC,CAAC,CAAC,GAAG,aAAa;YACxB,KAAK,CAAC,CAAC,CAAC,GAAG,aAAa;YACxB,KAAK,CAAC,CAAC,CAAC,GAAG,aAAa;SACzB,CAAC,CAAC;QACH,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC/B,IAAI,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAC;YAClD,QAAQ,CAAC,GAAG,GAAG,CAAC,EAAE,gBAAgB,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;YAC9C,IAAI,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC;YACvC,QAAQ,CAAC,GAAG,GAAG,CAAC,EAAE,gBAAgB,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;QAChD,CAAC;aAAM,IAAI,KAAK,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;YACxC,IAAI,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,kBAAkB,CAAC,KAAK,CAAC,CAAC;YACpD,IAAI,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC;QACzC,CAAC;aAAM,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACtC,IAAI,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAC;YAClD,QAAQ,CAAC,GAAG,GAAG,CAAC,EAAE,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC;YAC5C,IAAI,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC;YACvC,QAAQ,CAAC,GAAG,GAAG,CAAC,EAAE;gBAChB,IAAI,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,UAAU,GAAG,IAAI,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC;gBAC5C,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC;gBACxE,CAAC;aACF,CAAC,CAAC;QACL,CAAC;aAAM,IAAI,KAAK,CAAC,IAAI,KAAK,kBAAkB,EAAE,CAAC;YAC7C,QAAQ,CAAC,GAAG,GAAG,CAAC,EAAE,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC;QAC9C,CAAC;QACD,IAAI,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC,IAAI,KAAK,WAAW;YAClD,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YAChC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QAC7B,IAAI,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAC5C,QAAQ,CAAC,GAAG,GAAG,CAAC,EAAE,kBAAkB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IACjD,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC","sourcesContent":["import {matrixStorage} from '../math/storage.js';\nimport type {ReadonlyMathArray} from '../math/types.js';\nimport { getRectLightUOrV, getRectLightArea, getSphereLightArea, getSpotLightArea, getSpotLightRadiance, getLightDirection, getRigidLightFrame, validateSceneLights } from './lightsData.js';\nimport { CONSTANT_DIRECTIONAL_LIGHT, CONSTANT_SPHERE_LIGHT, CONSTANT_RECT_LIGHT, CONSTANT_SPOT_LIGHT } from './lightConstants.js';\nimport type { AnalyticLightState as AnalyticLight } from '../scene/lightState.js';\nimport { read } from '../math/access.js';\nexport const WGPU_SCENE_BUFFER_MIN_BYTES = 200;\nconst FIXED_LIGHT_COUNT = 64;\nconst PIXELS_PER_LIGHT = 6;\n\nexport function makeRayTraceUniform() {\n  // Created from https://webgpufundamentals.org/webgpu/lessons/resources/wgsl-offset-computer.html\n  const RayTraceUniformValues = new ArrayBuffer(320);\n  const RayTraceUniformViews = {\n    outputSize: new Float32Array(RayTraceUniformValues, 0, 2),\n    sampleCount: new Float32Array(RayTraceUniformValues, 8, 1),\n    scaleOffset: new Float32Array(RayTraceUniformValues, 16, 4),\n    lightsCount: new Int32Array(RayTraceUniformValues, 32, 1),\n    transmissiveOutputAlphaMin: new Float32Array(RayTraceUniformValues, 36, 1),\n    jitter: new Float32Array(RayTraceUniformValues, 40, 2),\n    backgroundColor: new Float32Array(RayTraceUniformValues, 48, 4),\n    groundedBackgroundParams: new Float32Array(RayTraceUniformValues, 64, 4),\n    groundedBackgroundCenter: new Float32Array(RayTraceUniformValues, 80, 2),\n    blasOffset: new Int32Array(RayTraceUniformValues, 88, 1),\n    isEmpty: new Int32Array(RayTraceUniformValues, 92, 1),\n    bevel: new Int32Array(RayTraceUniformValues, 96, 1),\n    dispersionWavelengths: new Int32Array(RayTraceUniformValues, 100, 1),\n    envLightPmf: new Float32Array(RayTraceUniformValues, 104, 1),\n    lightChoiceCount: new Int32Array(RayTraceUniformValues, 108, 1),\n    lightWeightTotal: new Float32Array(RayTraceUniformValues, 112, 1),\n    lightSamplingTextureWidth: new Int32Array(RayTraceUniformValues, 116, 1),\n    lightChoiceTexelOffset: new Int32Array(RayTraceUniformValues, 120, 1),\n    bounces: new Int32Array(RayTraceUniformValues, 124, 1),\n    varianceMoments: new Int32Array(RayTraceUniformValues, 128, 1),\n    adaptiveSampling: new Int32Array(RayTraceUniformValues, 132, 1),\n    clamping: new Float32Array(RayTraceUniformValues, 136, 1),\n    directLightCandidates: new Int32Array(RayTraceUniformValues, 140, 1),\n    backgroundFlags: new Int32Array(RayTraceUniformValues, 144, 1),\n    loopUnrollGuard: new Uint32Array(RayTraceUniformValues, 148, 1),\n    sceneFeatureFlags: new Uint32Array(RayTraceUniformValues, 152, 1),\n    sequenceSeed: new Uint32Array(RayTraceUniformValues, 156, 1),\n    zSobolLogSPP: new Uint32Array(RayTraceUniformValues, 160, 1),\n    sampleSequenceOffset: new Uint32Array(RayTraceUniformValues, 164, 1),\n    // Shadow-catcher light table, relative to lightChoiceTexelOffset.\n    shadowCatcherChoiceOffset: new Int32Array(RayTraceUniformValues, 168, 1),\n    shadowCatcherChoiceCount: new Int32Array(RayTraceUniformValues, 172, 1),\n    environmentRotation: new Float32Array(RayTraceUniformValues, 176, 16),\n    backgroundMapTransform: new Float32Array(RayTraceUniformValues, 240, 4),\n    backgroundMapRotation: new Float32Array(RayTraceUniformValues, 256, 16)\n  };\n  // Fixed runtime trip bounds discourage thin-film loop unrolling. Keep 4 for\n  // 2 polarizations/substrates and 3 channels, independent of BSDF budgets.\n  RayTraceUniformViews.loopUnrollGuard[0] = 4;\n  RayTraceUniformViews.environmentRotation.set([1,0,0,0, 0,1,0,0, 0,0,1,0, 0,0,0,1]);\n  RayTraceUniformViews.backgroundMapRotation.set([1,0,0,0, 0,1,0,0, 0,0,1,0, 0,0,0,1]);\n  RayTraceUniformViews.backgroundMapTransform.set([1,0,1,0]);\n  RayTraceUniformViews.bevel[0] = 1;\n  RayTraceUniformViews.dispersionWavelengths[0] = 3;\n  return {\n    buffer: RayTraceUniformValues,\n    views: RayTraceUniformViews\n  };\n}\nexport function packWgpuAnalyticLights(\n  lights: AnalyticLight[],\n  texturesIndices: number[],\n  selectionPmfs: number[],\n  iesProfileTexelOffset: (index: number) => number | undefined\n) {\n  validateSceneLights(lights);\n  const numLights = lights.length;\n  if (numLights > FIXED_LIGHT_COUNT) {\n    console.warn(\n      `Too many lights, only the first ${FIXED_LIGHT_COUNT} lights will be used.`\n    );\n  }\n  const data = new Float32Array(numLights * PIXELS_PER_LIGHT * 4);\n\n  function packVec3<T extends ReadonlyMathArray>(idx: number, v: T) {\n    for (let i = 0; i < 3; i++) {\n      data[i + idx * 4] = read(v, i);\n    }\n  }\n\n  const lightTypeConstants = {\n    DirectionalLight: CONSTANT_DIRECTIONAL_LIGHT,\n    SphereLight: CONSTANT_SPHERE_LIGHT,\n    RectLight: CONSTANT_RECT_LIGHT,\n    SpotLight: CONSTANT_SPOT_LIGHT\n  };\n\n  for (let i = 0; i < numLights; i++) {\n    const light = lights[i];\n    const matrix = matrixStorage(light.transform);\n    const color = light.color;\n    const intensity = light.intensity;\n    const off = i * PIXELS_PER_LIGHT;\n    packVec3(\n      off,\n      [matrix[12], matrix[13], matrix[14]]\n    );\n    data[off * 4 + 3] = lightTypeConstants[light.type];\n    // emission\n    const emissionScale =\n      light.type === 'SpotLight' ? getSpotLightRadiance(light) : intensity;\n    packVec3(off + 1, [\n      color[0] * emissionScale,\n      color[1] * emissionScale,\n      color[2] * emissionScale\n    ]);\n    if (light.type === 'RectLight') {\n      data[(off + 1) * 4 + 3] = getRectLightArea(light);\n      packVec3(off + 2, getRectLightUOrV(light, 0));\n      data[(off + 2) * 4 + 3] = light.spread;\n      packVec3(off + 3, getRectLightUOrV(light, 1));\n    } else if (light.type === 'SphereLight') {\n      data[(off + 1) * 4 + 3] = getSphereLightArea(light);\n      data[(off + 2) * 4 + 3] = light.radius;\n    } else if (light.type === 'SpotLight') {\n      data[(off + 1) * 4 + 3] = getSpotLightArea(light);\n      packVec3(off + 2, getLightDirection(light));\n      data[(off + 2) * 4 + 3] = light.radius;\n      packVec3(off + 3, [\n        Math.cos((light.outerAngle * Math.PI) / 180),\n        Math.cos((Math.min(light.innerAngle, light.outerAngle) * Math.PI) / 180),\n        0\n      ]);\n    } else if (light.type === 'DirectionalLight') {\n      packVec3(off + 2, getLightDirection(light));\n    }\n    data[(off + 3) * 4 + 3] = light.type === 'SpotLight'\n      ? iesProfileTexelOffset(i) ?? -1\n      : texturesIndices[i] ?? -1;\n    data[(off + 4) * 4] = selectionPmfs[i] ?? 0;\n    packVec3(off + 5, getRigidLightFrame(light).x);\n  }\n\n  return data;\n}\n"]}

package/lib/renderer/wgpuShaderSource.d.ts

@@ -0,0 +1,15 @@
+import { type RayTraceInputHooks } from './wgsl/rayTrace.compute.js';
+export interface WGPUShaderFeatures {
+    hasDisplacement: boolean;
+    hasAlpha: boolean;
+    hasSceneAlpha: boolean;
+    hasAnisotropyDirections: boolean;
+    hasCoatNormals: boolean;
+    blasLayoutMask: number;
+    /** Environment presence changes background/MIS semantics independently of packed material data. */
+    dynamicEnvironment?: boolean;
+}
+export declare const WGPU_SHADER_DEFINE_KEYS: string[];
+export declare function wgpuPipelineConstantKeys(constants: Record<string, boolean | number>): string[];
+export declare function wgpuShaderPreamble(constants: Record<string, boolean | number>, bake?: boolean): string;
+export declare function wgpuShaderSource(constants: Record<string, boolean | number>, features: WGPUShaderFeatures, bake?: boolean, sceneIntersections?: string, pathDiagnostic?: boolean, inputHooks?: RayTraceInputHooks): string;

package/lib/renderer/wgpuShaderSource.js

@@ -0,0 +1,31 @@
+import rayTraceCompute from './wgsl/rayTrace.compute.js';
+import { WGPU_SCENE_FEATURE_BITS } from './WGPUSceneFeatures.js';
+import { MATERIAL_TEXTURE_DEFINES } from './materialTextureSlots.js';
+export const WGPU_SHADER_DEFINE_KEYS = ['NUM_MAPS', 'TLAS_STACK_SIZE', 'BLAS_STACK_SIZE', 'NUM_UNIFORM_LIGHTS'];
+const constantsNotUsed = [
+    ...Object.keys(WGPU_SCENE_FEATURE_BITS), 'USE_BACKGROUND', 'USE_BACKGROUND_MAP',
+    'USE_GROUNDED_BACKGROUND', ...MATERIAL_TEXTURE_DEFINES, 'USE_ANISOTROPY_DIRECTIONS',
+    'USE_COAT_NORMAL_FRAMES', 'HAS_ALPHA_BLEND', 'HAS_ALPHA_MASK'
+];
+export function wgpuPipelineConstantKeys(constants) {
+    return Object.keys(constants).sort().filter(key => !WGPU_SHADER_DEFINE_KEYS.includes(key) && !constantsNotUsed.includes(key));
+}
+export function wgpuShaderPreamble(constants, bake = false) {
+    const defineCode = WGPU_SHADER_DEFINE_KEYS.map(key => `const ${key} = ${constants[key]};`).join('\n');
+    const keys = wgpuPipelineConstantKeys(constants);
+    const constantsCode = keys.map(key => {
+        const value = constants[key];
+        return `${bake ? 'const' : 'override'} ${key} = ${typeof value === 'boolean' ? (value ? 'true' : 'false') : value.toString()};`;
+    }).join('\n');
+    return `\n${defineCode}\n${constantsCode}\n`;
+}
+export function wgpuShaderSource(constants, features, bake = false, sceneIntersections = '', pathDiagnostic = false, inputHooks) {
+    return wgpuShaderPreamble(constants, bake) + rayTraceCompute({
+        ...features,
+        spectral: bake ? Boolean(constants.SPECTRAL) : undefined,
+        sceneCapabilities: bake && constants.SCENE_CAPABILITIES !== undefined ? Number(constants.SCENE_CAPABILITIES) : undefined,
+        materialCapabilities: bake && constants.MATERIAL_CAPABILITIES !== undefined ? Number(constants.MATERIAL_CAPABILITIES) : undefined,
+        lightCapabilities: bake && constants.LIGHT_CAPABILITIES !== undefined ? Number(constants.LIGHT_CAPABILITIES) : undefined
+    }, sceneIntersections, pathDiagnostic, inputHooks);
+}
+//# sourceMappingURL=wgpuShaderSource.js.map

package/lib/renderer/wgpuShaderSource.js.map

@@ -0,0 +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"]}

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

@@ -0,0 +1 @@
+export declare function aovDisplayShader(format: 'rgba16float' | 'rgba8unorm'): string;

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

@@ -0,0 +1,70 @@
+// Independent of beauty composition and denoising; shared by WebGPU and native.
+export function aovDisplayShader(format) {
+    return /* wgsl */ `
+struct DisplayAovAccum { geometry: vec4f, position: vec4f, diffuse: vec4f, specular: vec4f, }
+struct DisplayAovCenter { identity: vec4u, motion: vec4f, }
+struct DisplayAovParams { sizeKind: vec4u, mapping: vec4f, flags: vec4u, }
+@group(0) @binding(0) var<storage, read> accumulation: array<DisplayAovAccum>;
+@group(0) @binding(1) var<storage, read> center: array<DisplayAovCenter>;
+@group(0) @binding(2) var image: texture_storage_2d<${format}, write>;
+@group(0) @binding(3) var<uniform> params: DisplayAovParams;
+
+fn tF(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 {
+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 {
+return select(pow((rgb + 0.055) / 1.055, vec3f(2.4)), rgb / 12.92,
+rgb <= vec3f(0.04045));
+}
+@compute @workgroup_size(8, 8)
+fn main(@builtin(global_invocation_id) id: vec3u) {
+let size = params.sizeKind.xy;
+if (any(id.xy >= size)) { return; }
+let index = id.x + (size.y - 1u - id.y) * size.x;
+let kind = params.sizeKind.w;
+let pixel = accumulation[index];
+let hit = pixel.position.w;
+var color = vec3f(0.0);
+var alpha = hit / f32(max(params.sizeKind.z, 1u));
+let inv = 1.0 / max(hit, 1.0);
+if (kind == 0u) {
+color = vec3f(clamp((pixel.geometry.x * inv - params.mapping.x) /
+(params.mapping.y - params.mapping.x), 0.0, 1.0));
+} else if (kind == 1u) {
+color = clamp(pixel.position.xyz * inv / params.mapping.z * 0.5 + 0.5,
+vec3f(0.0), vec3f(1.0));
+} else if (kind == 2u) {
+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)));
+} 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]);
+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);
+var hue = 0.0;
+if (gR > 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);
+}
+if (alpha <= 0.0) { color = vec3f(0.0); }
+if (params.flags.y != 0u) { color = sE(color); }
+textureStore(image, vec2i(id.xy), vec4f(color, clamp(alpha, 0.0, 1.0)));
+}
+`;
+}
+//# sourceMappingURL=aovDisplay.wgsl.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"aovDisplay.wgsl.js","sourceRoot":"","sources":["../../../src/renderer/wgsl/aovDisplay.wgsl.ts"],"names":[],"mappings":"AAAA,gFAAgF;AAChF,MAAM,UAAU,gBAAgB,CAAC,MAAoC;IACnE,OAAO,UAAU,CAAC;;;;;;sDAMkC,MAAM;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA2D3D,CAAC;AACF,CAAC","sourcesContent":["// Independent of beauty composition and denoising; shared by WebGPU and native.\nexport function aovDisplayShader(format: 'rgba16float' | 'rgba8unorm') {\n  return /* wgsl */ `\nstruct DisplayAovAccum { geometry: vec4f, position: vec4f, diffuse: vec4f, specular: vec4f, }\nstruct DisplayAovCenter { identity: vec4u, motion: vec4f, }\nstruct DisplayAovParams { sizeKind: vec4u, mapping: vec4f, flags: vec4u, }\n@group(0) @binding(0) var<storage, read> accumulation: array<DisplayAovAccum>;\n@group(0) @binding(1) var<storage, read> center: array<DisplayAovCenter>;\n@group(0) @binding(2) var image: texture_storage_2d<${format}, write>;\n@group(0) @binding(3) var<uniform> params: DisplayAovParams;\n\nfn displayAovHash(input: u32) -> u32 {\n  var v = input;\n  v = (v ^ (v >> 16u)) * 0x7feb352du;\n  v = (v ^ (v >> 15u)) * 0x846ca68bu;\n  return v ^ (v >> 16u);\n}\nfn displayAovSrgb(rgb: vec3f) -> vec3f {\n  return select(1.055 * pow(max(rgb, vec3f(0.0)), vec3f(1.0 / 2.4)) - 0.055,\n    12.92 * rgb, rgb <= vec3f(0.0031308));\n}\nfn displayAovLinear(rgb: vec3f) -> vec3f {\n  return select(pow((rgb + 0.055) / 1.055, vec3f(2.4)), rgb / 12.92,\n    rgb <= vec3f(0.04045));\n}\n@compute @workgroup_size(8, 8)\nfn main(@builtin(global_invocation_id) id: vec3u) {\n  let size = params.sizeKind.xy;\n  if (any(id.xy >= size)) { return; }\n  let index = id.x + (size.y - 1u - id.y) * size.x;\n  let kind = params.sizeKind.w;\n  let pixel = accumulation[index];\n  let hit = pixel.position.w;\n  var color = vec3f(0.0);\n  var alpha = hit / f32(max(params.sizeKind.z, 1u));\n  let inv = 1.0 / max(hit, 1.0);\n  if (kind == 0u) {\n    color = vec3f(clamp((pixel.geometry.x * inv - params.mapping.x) /\n      (params.mapping.y - params.mapping.x), 0.0, 1.0));\n  } else if (kind == 1u) {\n    color = clamp(pixel.position.xyz * inv / params.mapping.z * 0.5 + 0.5,\n      vec3f(0.0), vec3f(1.0));\n  } else if (kind == 2u) {\n    color = clamp(pixel.geometry.yzw * inv, vec3f(0.0), vec3f(1.0));\n  } else if (kind == 3u || kind == 4u) {\n    let r = select(pixel.diffuse, pixel.specular, kind == 4u);\n    alpha = r.w / f32(max(params.sizeKind.z, 1u));\n    color = displayAovSrgb(clamp(r.xyz / max(r.w, 1.0), vec3f(0.0), vec3f(1.0)));\n  } else if (kind == 5u) {\n    let identity = center[index].identity;\n    alpha = select(1.0, 0.0, identity.x == 0xffffffffu);\n    let hash = displayAovHash(identity[params.flags.x]);\n    color = 0.2 + 0.8 * vec3f(f32(hash & 255u), f32((hash >> 8u) & 255u),\n      f32((hash >> 16u) & 255u)) / 255.0;\n  } else {\n    let m = center[index].motion;\n    alpha = select(0.0, 1.0, m.w == 3.0);\n    let magnitude = length(m.xy);\n    var hue = 0.0;\n    if (magnitude > 0.0) { hue = fract(atan2(m.y, m.x) / 6.28318530718 + 1.0); }\n    let p = abs(fract(vec3f(hue) + vec3f(0.0, 2.0 / 3.0, 1.0 / 3.0)) * 6.0 - 3.0);\n    color = clamp(p - 1.0, vec3f(0.0), vec3f(1.0)) * clamp(magnitude / params.mapping.w, 0.0, 1.0);\n  }\n  if (alpha <= 0.0) { color = vec3f(0.0); }\n  if (params.flags.y != 0u) { color = displayAovLinear(color); }\n  textureStore(image, vec2i(id.xy), vec4f(color, clamp(alpha, 0.0, 1.0)));\n}\n`;\n}\n"]}

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

@@ -0,0 +1,2 @@
+declare const _default: string;
+export default _default;

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

@@ -0,0 +1,98 @@
+import proposals from './aovReflectanceSamples.wgsl.js';
+// Directional-hemispherical quadrature of the supported unlayered OpenPBR
+// reflection lobes. Keep the original 256 cosine/VNDF proposals and weights;
+// 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 {
+diffuse: vec3f,
+specular: vec3f,
+}
+fn hN(prepared: ag, qQ: bool, kY: bool) -> oa {
+var result = oa();
+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 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);
+}
+let cr = normalize(vec3f(alpha, 1.0) * view);
+let qP = min(cr.z, 1.0);
+let kh = 1.0 + qP;
+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 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); }
+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); }
+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 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;
+}
+if (kY) {
+let jn = normalize(view + aA);
+let bm = abs(dot(view, jn));
+var bG = 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;
+}
+if (gb > 0.0) {
+bG += gb * material.specularWeight * cw(bm,
+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;
+}
+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;
+}
+result.specular += reflection * scale;
+}
+}
+return result;
+}
+fn vT(prepared: ag) -> oa {
+return hN(prepared, true, true);
+}
+`;
+//# sourceMappingURL=aovReflectance.wgsl.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"aovReflectance.wgsl.js","sourceRoot":"","sources":["../../../src/renderer/wgsl/aovReflectance.wgsl.ts"],"names":[],"mappings":"AAAA,OAAO,SAAS,MAAM,iCAAiC,CAAC;AAExD,0EAA0E;AAC1E,6EAA6E;AAC7E,+EAA+E;AAC/E,8EAA8E;AAC9E,eAAe,SAAS,GAAG,UAAU,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA2FrC,CAAC","sourcesContent":["import proposals from './aovReflectanceSamples.wgsl.js';\n\n// Directional-hemispherical quadrature of the supported unlayered OpenPBR\n// reflection lobes. Keep the original 256 cosine/VNDF proposals and weights;\n// evaluate diffuse and reflection together without traversing the full layered\n// BSDF twice. Callers reject coat, fuzz, anisotropy, SSS and thin film first.\nexport default proposals + /* wgsl */ `\nstruct MtReflectance {\n  diffuse: vec3f,\n  specular: vec3f,\n}\nfn mtFrameReflectanceComponents(prepared: OpenPbrPreparedBsdf, wantDiffuse: bool, wantSpecular: bool) -> MtReflectance {\n  var result = MtReflectance();\n  let view = prepared.viewLocal;\n  if (view.z <= 1e-6) { return result; }\n  let material = prepared.material;\n  let alpha = vec2f(material.alphaX, material.alphaY);\n  let viewMasking = sampling_ggx_smith_g1(view, alpha);\n  let diffuseWeight = prepared.weights[OPENPBR_LOBE_DIFFUSE];\n  let metalWeight = prepared.weights[OPENPBR_LOBE_METAL];\n  let specularWeight = prepared.weights[OPENPBR_LOBE_SPECULAR];\n  var dielectricAverageLoss = 1.0;\n  if (wantSpecular && prepared.dielectricMsProbability > 0.0) {\n    dielectricAverageLoss = max(openpbr_ideal_dielectric_average_loss(prepared.refractionEtaTOverEtaI, prepared.specularAlpha), 1e-12);\n  }\n  let sphericalView = normalize(vec3f(alpha, 1.0) * view);\n  let viewZ = min(sphericalView.z, 1.0);\n  let capSpan = 1.0 + viewZ;\n  let count = 256u * min(rtUniform.loopUnrollGuard, 4u) / 4u;\n  for (var i = 0u; i < count; i++) {\n    let pair = i / 2u;\n    let u = vec2f((f32(pair) + 0.5) / f32(count / 2u),\n      fract(f32(reverseBits(pair)) * 2.3283064365386963e-10 + 0.37));\n    let proposal = mtReflectanceProposals[pair];\n    // Retain runtime proposals near delta lobes: tiny device-specific sin/cos\n    // rounding differences can otherwise be magnified by their narrow PDFs.\n    let runtimeProposal = prepared.specularAlpha < 0.001 || count != 256u;\n    var light = vec3f(proposal.xy, sqrt(max(0.0, 1.0 - dot(proposal.xy, proposal.xy))));\n    if (runtimeProposal) { light = sampling_cosine_hemisphere(u); }\n    if ((i & 1u) != 0u) {\n      let normalZ = (1.0 - u.y) * capSpan;\n      let radial = sqrt(max(0.0, (normalZ + (1.0 - viewZ)) * (u.y * capSpan)));\n      var half = normalize(vec3f(alpha, 1.0) * vec3f(sphericalView.xy + radial * proposal.zw, normalZ));\n      if (runtimeProposal) { half = sampling_ggx_visible_normal(view, alpha, u); }\n      light = reflect(-view, half);\n    }\n    // Rejected VNDF proposals retain their weight in the denominator.\n    if (light.z <= 0.0) { continue; }\n    let half = normalize(view + light);\n    let specularPdf = viewMasking * sampling_ggx_ndf(half, alpha) / (4.0 * view.z);\n    let pdf = 0.5 * (light.z * INVPI + specularPdf);\n    // Preserve the production evaluator's frame round trip, including FP32\n    // rounding near narrow highlights. Proposal PDFs use the original light.\n    let evaluatedLight = openpbr_to_local(prepared.basis, openpbr_to_world(prepared.basis, light));\n    if (evaluatedLight.z <= 0.0) { continue; }\n    let scale = abs(evaluatedLight.z) / (pdf * f32(count));\n    if (wantDiffuse && diffuseWeight > 0.0) {\n      let compensation = prepared.diffuseSpecularEnergyCompensation * openpbr_opaque_energy_complement(\n        prepared.specularEtaTOverEtaI, prepared.specularAlpha, abs(evaluatedLight.z));\n      var unusedPdf = 0.0;\n      result.diffuse += diffuseWeight * openpbr_eval_diffuse(material, prepared.diffuseColor,\n        view, evaluatedLight, compensation, &unusedPdf) * scale;\n    }\n    if (wantSpecular) {\n      let evaluatedHalf = normalize(view + evaluatedLight);\n      let viewHalf = abs(dot(view, evaluatedHalf));\n      var fresnel = vec3f(0.0);\n      if (specularWeight > 0.0) {\n        var dielectric = vec3f(openpbr_dielectric_fresnel(viewHalf, material.relativeIor));\n        if (material.thinWalled) { dielectric = openpbr_thin_wall_reflection_coefficient(dielectric, material.transmissionWeight); }\n        fresnel += specularWeight * material.specularColor * dielectric;\n      }\n      if (metalWeight > 0.0) {\n        fresnel += metalWeight * material.specularWeight * openpbr_metal_fresnel(viewHalf,\n          material.baseWeight * material.baseColor, material.specularColor);\n      }\n      let microfacet = sampling_ggx_ndf(evaluatedHalf, alpha) * viewMasking\n        * sampling_ggx_smith_g1(evaluatedLight, alpha) / (4.0 * max(evaluatedLight.z * view.z, 1e-8));\n      var reflection = fresnel * microfacet;\n      if (metalWeight > 0.0 && prepared.metalMsFactor > 0.0) {\n        let factor = min(prepared.metalMsFactor * openpbr_metal_energy_complement(prepared.specularAlpha, evaluatedLight.z), 1.0 / max(evaluatedLight.z, 1e-8));\n        reflection += metalWeight * prepared.metalMsScale * factor * INVPI;\n      }\n      if (prepared.dielectricMsProbability > 0.0) {\n        let factors = prepared.dielectricMsViewLoss * openpbr_ideal_dielectric_loss(\n          prepared.refractionEtaTOverEtaI, prepared.specularAlpha, abs(evaluatedLight.z)) / dielectricAverageLoss;\n        let factor = min(factors * prepared.dielectricMsReflectionRatio, 1.0 / abs(evaluatedLight.z));\n        reflection += prepared.dielectricMsReflectionScale * factor * INVPI;\n      }\n      result.specular += reflection * scale;\n    }\n  }\n  return result;\n}\nfn mtFrameReflectance(prepared: OpenPbrPreparedBsdf) -> MtReflectance {\n  return mtFrameReflectanceComponents(prepared, true, true);\n}\n`;\n"]}

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

@@ -0,0 +1,2 @@
+declare const _default: "\nconst mtReflectanceProposals: array<vec4f, 128> = array<vec4f, 128>(\n  vec4f(-0.971247852, -0.202764958, 0.999698818, 0.0245412290),\n  vec4f(-0.808660269, 0.547487736, 0.997290432, 0.0735645667),\n  vec4f(-0.942509234, 0.187289074, 0.992479563, 0.122410677),\n  vec4f(-0.763038993, -0.558020830, 0.985277653, 0.170961902),\n  vec4f(-0.929654300, -0.00785388052, 0.975702107, 0.219101235),\n  vec4f(-0.656618595, 0.740912974, 0.963776052, 0.266712785),\n  vec4f(-0.817265987, 0.373183846, 0.949528158, 0.313681751),\n  vec4f(-0.793490529, -0.386950523, 0.932992816, 0.359895051),\n  vec4f(-0.860713601, -0.105764769, 0.914209723, 0.405241340),\n  vec4f(-0.604139447, 0.619064212, 0.893224299, 0.449611336),\n  vec4f(-0.787263036, 0.281084418, 0.870086968, 0.492898226),\n  vec4f(-0.673318744, -0.468566477, 0.844853580, 0.534997642),\n  vec4f(-0.799623907, 0.0901312605, 0.817584813, 0.575808227),\n  vec4f(-0.570307195, -0.676738262, 0.788346410, 0.615231633),\n  vec4f(-0.627450705, 0.452229142, 0.757208824, 0.653172851),\n  vec4f(-0.698281944, -0.294418186, 0.724247098, 0.689540565),\n  vec4f(-0.726036906, -0.153989345, 0.689540505, 0.724247098),\n  vec4f(-0.523753047, 0.608020604, 0.653172791, 0.757208884),\n  vec4f(-0.671607554, 0.233185455, 0.615231574, 0.788346410),\n  vec4f(-0.491990626, -0.494420350, 0.575808167, 0.817584813),\n  vec4f(-0.678437173, 0.0412081257, 0.534997642, 0.844853580),\n  vec4f(-0.495611340, -0.807542920, 0.492898196, 0.870086968),\n  vec4f(-0.508582473, 0.402262419, 0.449611306, 0.893224299),\n  vec4f(-0.537693322, -0.333672792, 0.405241281, 0.914209783),\n  vec4f(-0.614562631, -0.0568606183, 0.359894961, 0.932992816),\n  vec4f(-0.452296555, 0.809743226, 0.313681662, 0.949528217),\n  vec4f(-0.492342144, 0.317682207, 0.266712755, 0.963776052),\n  vec4f(-0.403601944, -0.406031638, 0.219101220, 0.975702107),\n  vec4f(-0.537260950, 0.137945175, 0.170961857, 0.985277653),\n  vec4f(-0.404928178, -0.715918541, 0.122410625, 0.992479563),\n  vec4f(-0.386338592, 0.556550741, 0.0735644922, 0.997290432),\n  vec4f(-0.445942074, -0.242918074, 0.0245411359, 0.999698818),\n  vec4f(-0.459759980, -0.175696611, -0.0245412234, 0.999698818),\n  vec4f(-0.359771132, 0.682195961, -0.0735645741, 0.997290432),\n  vec4f(-0.412575692, 0.205535084, -0.122410707, 0.992479503),\n  vec4f(-0.336475104, -0.646421731, -0.170961946, 0.985277653),\n  vec4f(-0.429363400, 0.0166855231, -0.219101310, 0.975702107),\n  vec4f(-0.319253087, -0.925218344, -0.266712844, 0.963776052),\n  vec4f(-0.294470191, 0.430103362, -0.313681722, 0.949528158),\n  vec4f(-0.281283021, -0.387403429, -0.359895051, 0.932992816),\n  vec4f(-0.358183086, -0.0808178857, -0.405241340, 0.914209723),\n  vec4f(-0.271769434, 0.854052365, -0.449611366, 0.893224299),\n  vec4f(-0.244775921, 0.311606824, -0.492898166, 0.870087028),\n  vec4f(-0.244355842, -0.552090824, -0.534997702, 0.844853520),\n  vec4f(-0.283291280, 0.112162955, -0.575808167, 0.817584813),\n  vec4f(-0.224362880, -0.823741674, -0.615231693, 0.788346350),\n  vec4f(-0.210826442, 0.610893846, -0.653172851, 0.757208824),\n  vec4f(-0.191608176, -0.297364026, -0.689540684, 0.724246979),\n  vec4f(-0.207833603, -0.124338157, -0.724247158, 0.689540505),\n  vec4f(-0.176593989, 0.814833522, -0.757208824, 0.653172851),\n  vec4f(-0.158949688, 0.293469012, -0.788346469, 0.615231514),\n  vec4f(-0.152046368, -0.648668587, -0.817584813, 0.575808167),\n  vec4f(-0.167782038, 0.0643178672, -0.844853640, 0.534997463),\n  vec4f(-0.128430068, -0.907204390, -0.870087028, 0.492898136),\n  vec4f(-0.115809105, 0.572147131, -0.893224299, 0.449611366),\n  vec4f(-0.103222348, -0.403248310, -0.914209783, 0.405241221),\n  vec4f(-0.112737618, -0.0319865458, -0.932992816, 0.359895051),\n  vec4f(-0.0796749964, 0.955433667, -0.949528217, 0.313681602),\n  vec4f(-0.0672519580, 0.453793734, -0.963776052, 0.266712725),\n  vec4f(-0.0551275201, -0.538435221, -0.975702167, 0.219101071),\n  vec4f(-0.0426490456, 0.204346836, -0.985277653, 0.170961812),\n  vec4f(-0.0306719355, -0.790655255, -0.992479503, 0.122410700),\n  vec4f(-0.0184057076, 0.708510995, -0.997290492, 0.0735644475),\n  vec4f(-0.00613544276, -0.291185349, -0.999698818, 0.0245412104),\n  vec4f(0.00613530306, -0.244297966, -0.999698818, -0.0245413855),\n  vec4f(0.0184059385, 0.755400777, -0.997290432, -0.0735646188),\n  vec4f(0.0306059923, 0.253786176, -0.992479503, -0.122410871),\n  vec4f(0.0429276451, -0.743136168, -0.985277653, -0.170961991),\n  vec4f(0.0701737553, 0.00441496586, -0.975702107, -0.219101235),\n  vec4f(0.0674433559, -0.992085159, -0.963776052, -0.266712904),\n  vec4f(0.0794365108, 0.499346048, -0.949528158, -0.313681781),\n  vec4f(0.0915081054, -0.485832185, -0.932992756, -0.359895200),\n  vec4f(0.101071075, -0.0861614645, -0.914209723, -0.405241400),\n  vec4f(0.116242222, 0.872919321, -0.893224180, -0.449611515),\n  vec4f(0.126407236, 0.359021723, -0.870086908, -0.492898285),\n  vec4f(0.139908314, -0.603366435, -0.844853580, -0.534997642),\n  vec4f(0.168981716, 0.0979395211, -0.817584693, -0.575808346),\n  vec4f(0.164669126, -0.853637457, -0.788346410, -0.615231633),\n  vec4f(0.175589874, 0.605686426, -0.757208705, -0.653172970),\n  vec4f(0.181917489, -0.321471483, -0.724247038, -0.689540625),\n  vec4f(0.219242185, -0.135647133, -0.689540565, -0.724247098),\n  vec4f(0.212299466, 0.790099680, -0.653172910, -0.757208765),\n  vec4f(0.208243251, 0.240495890, -0.615231574, -0.788346469),\n  vec4f(0.234419063, -0.640313387, -0.575808048, -0.817584932),\n  vec4f(0.315854102, 0.0532567836, -0.534997344, -0.844853759),\n  vec4f(0.260333687, -0.894772291, -0.492898196, -0.870086968),\n  vec4f(0.265374005, 0.502337217, -0.449611217, -0.893224359),\n  vec4f(0.267429054, -0.339113116, -0.405241072, -0.914209843),\n  vec4f(0.380209267, -0.0445682108, -0.359895110, -0.932992756),\n  vec4f(0.307221651, 0.891683638, -0.313681692, -0.949528217),\n  vec4f(0.296788067, 0.329054117, -0.266712576, -0.963776112),\n  vec4f(0.317195147, -0.457708418, -0.219100907, -0.975702226),\n  vec4f(0.419729054, 0.148764044, -0.170961887, -0.985277653),\n  vec4f(0.349995196, -0.727014899, -0.122410536, -0.992479563),\n  vec4f(0.356363803, 0.594497383, -0.0735642835, -0.997290492),\n  vec4f(0.434345096, -0.231501341, -0.0245412868, -0.999698818),\n  vec4f(0.480473697, -0.164373189, 0.0245413091, -0.999698818),\n  vec4f(0.392657876, 0.681903303, 0.0735647827, -0.997290432),\n  vec4f(0.492657542, 0.218807966, 0.122411035, -0.992479503),\n  vec4f(0.409054011, -0.584142148, 0.170961916, -0.985277653),\n  vec4f(0.569577277, 0.0289491210, 0.219101399, -0.975702107),\n  vec4f(0.442882210, -0.855257273, 0.266713053, -0.963775992),\n  vec4f(0.459734946, 0.387970626, 0.313681692, -0.949528217),\n  vec4f(0.513070345, -0.343044043, 0.359895140, -0.932992756),\n  vec4f(0.629031241, -0.0690753013, 0.405241549, -0.914209664),\n  vec4f(0.483215511, 0.773316741, 0.449611664, -0.893224120),\n  vec4f(0.586818159, 0.310843110, 0.492898226, -0.870086968),\n  vec4f(0.528679490, -0.427168757, 0.534997761, -0.844853461),\n  vec4f(0.683721542, 0.126429096, 0.575808465, -0.817584634),\n  vec4f(0.517971814, -0.658907175, 0.615231574, -0.788346410),\n  vec4f(0.548397303, 0.476606160, 0.653172910, -0.757208765),\n  vec4f(0.695181608, -0.259932309, 0.689540744, -0.724246919),\n  vec4f(0.748597383, -0.117820688, 0.724247038, -0.689540625),\n  vec4f(0.556980729, 0.641256869, 0.757208884, -0.653172791),\n  vec4f(0.741829991, 0.268901259, 0.788346529, -0.615231454),\n  vec4f(0.647817433, -0.477341115, 0.817584991, -0.575807929),\n  vec4f(0.816602290, 0.0779312477, 0.844853580, -0.534997642),\n  vec4f(0.599919200, -0.671730816, 0.870087087, -0.492898077),\n  vec4f(0.724726558, 0.447135508, 0.893224418, -0.449611068),\n  vec4f(0.810727239, -0.307791352, 0.914209723, -0.405241370),\n  vec4f(0.882583082, -0.0201240778, 0.932992816, -0.359894961),\n  vec4f(0.645550549, 0.729843259, 0.949528277, -0.313681543),\n  vec4f(0.839177251, 0.362342149, 0.963776171, -0.266712397),\n  vec4f(0.839526892, -0.399391353, 0.975702107, -0.219101220),\n  vec4f(0.928934336, 0.175204813, 0.985277653, -0.170961723),\n  vec4f(0.774228692, -0.569184363, 0.992479563, -0.122410372),\n  vec4f(0.815476179, 0.537282884, 0.997290432, -0.0735645965),\n  vec4f(0.968665719, -0.214761987, 0.999698818, -0.0245411228)\n);\n";
+export default _default;