@meimu/mutracer 0.0.0-stage → 0.1.1

package/lib/renderer/meshInstances.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"meshInstances.js","sourceRoot":"","sources":["../../src/renderer/meshInstances.ts"],"names":[],"mappings":"AACA,OAAO,aAAa,MAAM,2BAA2B,CAAC;AACtD,OAAO,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAGlC,MAAM,UAAU,iBAAiB,CAAC,IAAU,IAAI,OAAO,IAAI,YAAY,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AACxG,mFAAmF;AACnF,MAAM,UAAU,qBAAqB,CAAC,IAAU,EAAE,KAAa,EAAE,GAAmB;IAClF,IAAI,CAAC,CAAC,IAAI,YAAY,aAAa,CAAC;QAAE,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;IAC5E,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,EAAE,MAAM,GAAG,KAAK,GAAG,EAAE,EAAE,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC;IAC5E,KAAK,IAAI,MAAM,GAAG,CAAC,EAAE,MAAM,GAAG,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC;QAC1C,MAAM,CAAC,GAAG,MAAM,GAAG,MAAM,GAAG,CAAC,CAAC;QAC9B,MAAM,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QACzE,KAAK,IAAI,GAAG,GAAG,CAAC,EAAE,GAAG,GAAG,CAAC,EAAE,GAAG,EAAE;YAAE,GAAG,CAAC,MAAM,GAAG,CAAC,GAAG,GAAG,CAAC,GAAG,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,GAAG,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC;IAC/I,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,mFAAmF;AACnF,MAAM,OAAO,mBAAmB;IAC9B,MAAM,GAAG,CAAC,CAAC;IACF,UAAU,CAAe;IACzB,MAAM,CAAe;IACrB,MAAM,CAAe;IACrB,WAAW,CAAc;IACzB,SAAS,CAAa;IACtB,YAAY,CAAe;IAC3B,aAAa,CAAc;IACnB,IAAI,GAAiB,EAAC,SAAS,EAAE,CAAC,EAAE,MAAM,EAAE,EAAC,GAAG,EAAE,CAAC,CAAC,EAAC,CAAC,EAAC,CAAC,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,EAAC,CAAC,EAAC,CAAC,CAAC,EAAC,EAAE,SAAS,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,EAAC,CAAC,EAAC,CAAC,CAAC,EAAC,CAAC;IAChK,YAAY,QAAgB;QAC1B,IAAI,CAAC,MAAM,GAAG,IAAI,YAAY,CAAC,QAAQ,GAAG,CAAC,CAAC,CAAC;QAC7C,IAAI,CAAC,UAAU,GAAG,IAAI,YAAY,CAAC,QAAQ,GAAG,EAAE,CAAC,CAAC;QAAC,IAAI,CAAC,MAAM,GAAG,IAAI,YAAY,CAAC,QAAQ,GAAG,CAAC,CAAC,CAAC;QAChG,IAAI,CAAC,WAAW,GAAG,IAAI,WAAW,CAAC,QAAQ,CAAC,CAAC;QAAC,IAAI,CAAC,SAAS,GAAG,IAAI,UAAU,CAAC,QAAQ,CAAC,CAAC;QAAC,IAAI,CAAC,YAAY,GAAG,IAAI,YAAY,CAAC,QAAQ,CAAC,CAAC;QAAC,IAAI,CAAC,aAAa,GAAG,IAAI,WAAW,CAAC,QAAQ,CAAC,CAAC;IAC1L,CAAC;IACD,IAAI,CAAC,KAAa;QAChB,MAAM,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC;QACpB,CAAC,CAAC,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAAC,CAAC,CAAC,iBAAiB,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;QACrI,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE;YAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;QAC9E,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE;YAAE,CAAC,CAAC,KAAM,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;QACrE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAAA,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;YAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;QAAA,CAAC;QAC7H,OAAO,CAAC,CAAC;IACX,CAAC;CACF;AAED,MAAM,UAAU,gBAAgB,CAAC,SAAwB,EAAE,KAAa;IACtE,OAAO,SAAS,YAAY,mBAAmB,CAAC,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;AAC7F,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,CAAgB,EAAE,CAAgB;IACvE,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM;QAAE,OAAO,KAAK,CAAC;IACxC,IAAI,CAAC,YAAY,mBAAmB,IAAI,CAAC,YAAY,mBAAmB,EAAE,CAAC;QACzE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE;YAAE,IAAI,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC;gBAAE,OAAO,KAAK,CAAC;IACjG,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC","sourcesContent":["import Mesh from '../scene/Mesh.js';\nimport InstancedMesh from '../scene/InstancedMesh.js';\nimport { mat4 } from '../math.js';\nimport type { TLASInstance } from './bvhAccel.js';\n\nexport function meshInstanceCount(mesh: Mesh) { return mesh instanceof InstancedMesh ? mesh.count : 1; }\n/** Writes a world matrix without allocating per-instance Mesh objects or views. */\nexport function meshInstanceTransform(mesh: Mesh, index: number, out: mat4.Mat4Array): mat4.Mat4Array {\n  if (!(mesh instanceof InstancedMesh)) return mat4.copy(out, mesh.transform);\n  const local = mesh.transforms, offset = index * 16, parent = mesh.transform;\n  for (let column = 0; column < 4; column++) {\n    const o = offset + column * 4;\n    const x = local[o], y = local[o + 1], z = local[o + 2], w = local[o + 3];\n    for (let row = 0; row < 4; row++) out[column * 4 + row] = parent[row] * x + parent[row + 4] * y + parent[row + 8] * z + parent[row + 12] * w;\n  }\n  return out;\n}\n\n/** CPU instance records in numeric storage; one reusable synchronous read view. */\nexport class PackedMeshInstances {\n  length = 0;\n  readonly transforms: Float64Array;\n  readonly colors: Float32Array;\n  readonly bounds: Float32Array;\n  readonly blasIndices: Uint32Array;\n  readonly materials: Int32Array;\n  readonly displacement: Float64Array;\n  readonly sourceIndices: Uint32Array;\n  private readonly view: TLASInstance = {blasIndex: 0, bounds: {min: [0,0,0], max: [0,0,0]}, transform: Array(16).fill(0), materialsIndices: [0], color: [1,1,1]};\n  constructor(capacity: number) {\n    this.colors = new Float32Array(capacity * 3);\n    this.transforms = new Float64Array(capacity * 16); this.bounds = new Float32Array(capacity * 6);\n    this.blasIndices = new Uint32Array(capacity); this.materials = new Int32Array(capacity); this.displacement = new Float64Array(capacity); this.sourceIndices = new Uint32Array(capacity);\n  }\n  read(index: number): TLASInstance {\n    const v = this.view;\n    v.blasIndex = this.blasIndices[index]; v.materialsIndices[0] = this.materials[index]; v.displacementScale = this.displacement[index];\n    for (let i = 0; i < 16; i++) v.transform[i] = this.transforms[index * 16 + i];\n    for (let i = 0; i < 3; i++) v.color![i] = this.colors[index * 3 + i];\n    for (let i = 0; i < 3; i++) {v.bounds.min[i] = this.bounds[index * 6 + i]; v.bounds.max[i] = this.bounds[index * 6 + i + 3];}\n    return v;\n  }\n}\nexport type MeshInstances = TLASInstance[] | PackedMeshInstances;\nexport function readMeshInstance(instances: MeshInstances, index: number) {\n  return instances instanceof PackedMeshInstances ? instances.read(index) : instances[index];\n}\n\nexport function sameInstanceMembership(a: MeshInstances, b: MeshInstances) {\n  if (a.length !== b.length) return false;\n  if (a instanceof PackedMeshInstances && b instanceof PackedMeshInstances) {\n    for (let i = 0; i < a.length; i++) if (a.sourceIndices[i] !== b.sourceIndices[i]) return false;\n  }\n  return true;\n}\n"]}

package/lib/renderer/meshMembership.d.ts

@@ -0,0 +1,10 @@
+import Mesh from '../scene/Mesh.js';
+import { updateMeshInstances } from './prepareMeshBVH.js';
+type Prepared = ReturnType<typeof updateMeshInstances>;
+/** Reuses resident vertex/BLAS data when only single-material instances change. */
+export declare class MeshMembershipCache {
+    private _states;
+    capture(res: Prepared): void;
+    update(res: Prepared, meshes: readonly Mesh[]): Prepared | undefined;
+}
+export {};

package/lib/renderer/meshMembership.js

@@ -0,0 +1,72 @@
+import InstancedMesh from '../scene/InstancedMesh.js';
+import { getGeometryBvh } from './geometryBvh.js';
+import { materialIsEmissive } from './materialChanges.js';
+import { getBvhBounds, updateMeshInstanceTransforms } from './prepareMeshBVH.js';
+const attributes = ['position', 'normal', 'tangent', 'color', 'uv', 'uv1', 'indices', 'materialIndex'];
+function geometryState(geometry) {
+    return [geometry.version, geometry.dynamic, getGeometryBvh(geometry), ...attributes.flatMap(key => {
+            const attribute = geometry[key];
+            return [attribute?.array, attribute?.array.length, attribute?.itemSize];
+        })];
+}
+/** Reuses resident vertex/BLAS data when only single-material instances change. */
+export class MeshMembershipCache {
+    _states = new Map();
+    capture(res) {
+        this._states.clear();
+        for (const binding of res.instanceBindings)
+            if (binding.mesh.geometry === binding.geometry) {
+                this._states.set(binding.geometry, geometryState(binding.geometry));
+            }
+    }
+    update(res, meshes) {
+        const geometries = new Map(res.geometries.map((geometry, i) => [geometry, i]));
+        // Multiple-material vertex indices, displacement have
+        // geometry-dependent packing; their structural edits require preparation.
+        const checked = new Set();
+        for (const mesh of meshes) {
+            const previous = this._states.get(mesh.geometry);
+            if (!previous || mesh.materials.length !== 1 || mesh.geometry.materialIndex ||
+                mesh.materials[0].displacementMap)
+                return;
+            if (!checked.has(mesh.geometry)) {
+                const next = geometryState(mesh.geometry);
+                if (next.some((value, i) => value !== previous[i]))
+                    return;
+                checked.add(mesh.geometry);
+            }
+        }
+        // Rebuild active material slots when batches are added, removed or reassigned.
+        // Retired material identities must not remain in the active table.
+        const materials = [];
+        const materialIndices = new Map();
+        const templates = new Map(res.instanceBindings.map(binding => [binding.geometry, binding]));
+        const instanceBindings = meshes.filter(mesh => mesh.geometry.position.array.length > 0).map(mesh => {
+            const material = mesh.materials[0];
+            let index = materialIndices.get(material);
+            if (index == null) {
+                index = materials.length;
+                materials.push(material);
+                materialIndices.set(material, index);
+            }
+            const template = templates.get(mesh.geometry);
+            // Bounds remain valid even after every instance of a geometry was removed.
+            // Use the resident geometry bounds for all supported BVH layouts.
+            return { mesh, geometry: mesh.geometry, blasIndex: geometries.get(mesh.geometry),
+                materialIndices: [index], displacementScale: 0,
+                bvhBounds: template?.bvhBounds ?? getBvhBounds(mesh.geometry) };
+        });
+        const instances = updateMeshInstanceTransforms(instanceBindings);
+        if (meshes.some(mesh => mesh instanceof InstancedMesh))
+            return { ...res, materials, instances, instanceBindings,
+                emissiveBindings: instanceBindings.filter(binding => materialIsEmissive(materials[binding.materialIndices[0]])) };
+        const ordinary = instances;
+        // Match full preparation's exclusion of invalid transforms/bounds.
+        const valid = ordinary.map((instance, i) => instance.bounds.min.every(Number.isFinite) &&
+            instance.bounds.max.every(Number.isFinite) && instance.transform.every(Number.isFinite) ? i : -1).filter(i => i >= 0);
+        const bindings = valid.map(i => instanceBindings[i]);
+        return { ...res, materials, instances: valid.map(i => ordinary[i]), instanceBindings: bindings,
+            emissiveBindings: bindings.filter(binding => materialIsEmissive(materials[binding.materialIndices[0]])) };
+    }
+}
+//# sourceMappingURL=meshMembership.js.map

package/lib/renderer/meshMembership.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"meshMembership.js","sourceRoot":"","sources":["../../src/renderer/meshMembership.ts"],"names":[],"mappings":"AAAA,OAAO,aAAa,MAAM,2BAA2B,CAAC;AACtD,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAGlD,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,YAAY,EAAuB,4BAA4B,EAAE,MAAM,qBAAqB,CAAC;AAGtG,MAAM,UAAU,GAAG,CAAC,UAAU,EAAE,QAAQ,EAAE,SAAS,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,SAAS,EAAE,eAAe,CAAU,CAAC;AAChH,SAAS,aAAa,CAAC,QAAkB;IACvC,OAAO,CAAC,QAAQ,CAAC,OAAO,EAAE,QAAQ,CAAC,OAAO,EAAE,cAAc,CAAC,QAAQ,CAAC,EAAE,GAAG,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE;YAChG,MAAM,SAAS,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC;YAChC,OAAO,CAAC,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,CAAC,MAAM,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;QAC1E,CAAC,CAAC,CAAC,CAAC;AACN,CAAC;AAED,mFAAmF;AACnF,MAAM,OAAO,mBAAmB;IACtB,OAAO,GAAG,IAAI,GAAG,EAAuB,CAAC;IACjD,OAAO,CAAC,GAAa;QACnB,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACrB,KAAK,MAAM,OAAO,IAAI,GAAG,CAAC,gBAAgB;YAAE,IAAI,OAAO,CAAC,IAAI,CAAC,QAAQ,KAAK,OAAO,CAAC,QAAQ,EAAE,CAAC;gBAC3F,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,QAAQ,EAAE,aAAa,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC;YACtE,CAAC;IACH,CAAC;IACD,MAAM,CAAC,GAAa,EAAE,MAAuB;QAC3C,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAC/E,sDAAsD;QACtD,0EAA0E;QAC1E,MAAM,OAAO,GAAG,IAAI,GAAG,EAAY,CAAC;QACpC,KAAK,MAAM,IAAI,IAAI,MAAM,EAAE,CAAC;YAC1B,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACjD,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,aAAa;gBACzE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,eAAe;gBAAE,OAAO;YAC5C,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAChC,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBAC1C,IAAI,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC;oBAAE,OAAO;gBAC3D,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC7B,CAAC;QACH,CAAC;QACD,+EAA+E;QAC/E,mEAAmE;QACnE,MAAM,SAAS,GAA0B,EAAE,CAAC;QAC5C,MAAM,eAAe,GAAG,IAAI,GAAG,EAAyC,CAAC;QACzE,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC;QAC5F,MAAM,gBAAgB,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE;YACjG,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;YACnC,IAAI,KAAK,GAAG,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC1C,IAAI,KAAK,IAAI,IAAI,EAAE,CAAC;gBAAC,KAAK,GAAG,SAAS,CAAC,MAAM,CAAC;gBAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;YAAC,CAAC;YAChH,MAAM,QAAQ,GAAG,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC9C,2EAA2E;YAC3E,kEAAkE;YAClE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,SAAS,EAAE,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAE;gBAC/E,eAAe,EAAE,CAAC,KAAK,CAAC,EAAE,iBAAiB,EAAE,CAAC;gBAC9C,SAAS,EAAE,QAAQ,EAAE,SAAS,IAAI,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;QACpE,CAAC,CAAC,CAAC;QACH,MAAM,SAAS,GAAG,4BAA4B,CAAC,gBAAgB,CAAC,CAAC;QACjE,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,YAAY,aAAa,CAAC;YAAE,OAAO,EAAC,GAAG,GAAG,EAAE,SAAS,EAAE,SAAS,EAAE,gBAAgB;gBAC5G,gBAAgB,EAAE,gBAAgB,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,kBAAkB,CAAC,SAAS,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAC,CAAC;QACnH,MAAM,QAAQ,GAAG,SAAmD,CAAC;QACrE,mEAAmE;QACnE,MAAM,KAAK,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC;YACpF,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,QAAQ,CAAC,SAAS,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QACxH,MAAM,QAAQ,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,CAAC;QACrD,OAAO,EAAE,GAAG,GAAG,EAAE,SAAS,EAAE,SAAS,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,gBAAgB,EAAE,QAAQ;YAC5F,gBAAgB,EAAE,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,kBAAkB,CAAC,SAAS,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAC9G,CAAC;CACF","sourcesContent":["import InstancedMesh from '../scene/InstancedMesh.js';\nimport { getGeometryBvh } from './geometryBvh.js';\nimport Geometry from '../scene/Geometry.js';\nimport Mesh from '../scene/Mesh.js';\nimport { materialIsEmissive } from './materialChanges.js';\nimport { getBvhBounds, updateMeshInstances, updateMeshInstanceTransforms } from './prepareMeshBVH.js';\n\ntype Prepared = ReturnType<typeof updateMeshInstances>;\nconst attributes = ['position', 'normal', 'tangent', 'color', 'uv', 'uv1', 'indices', 'materialIndex'] as const;\nfunction geometryState(geometry: Geometry): unknown[] {\n  return [geometry.version, geometry.dynamic, getGeometryBvh(geometry), ...attributes.flatMap(key => {\n    const attribute = geometry[key];\n    return [attribute?.array, attribute?.array.length, attribute?.itemSize];\n  })];\n}\n\n/** Reuses resident vertex/BLAS data when only single-material instances change. */\nexport class MeshMembershipCache {\n  private _states = new Map<Geometry, unknown[]>();\n  capture(res: Prepared) {\n    this._states.clear();\n    for (const binding of res.instanceBindings) if (binding.mesh.geometry === binding.geometry) {\n      this._states.set(binding.geometry, geometryState(binding.geometry));\n    }\n  }\n  update(res: Prepared, meshes: readonly Mesh[]): Prepared | undefined {\n    const geometries = new Map(res.geometries.map((geometry, i) => [geometry, i]));\n    // Multiple-material vertex indices, displacement have\n    // geometry-dependent packing; their structural edits require preparation.\n    const checked = new Set<Geometry>();\n    for (const mesh of meshes) {\n      const previous = this._states.get(mesh.geometry);\n      if (!previous || mesh.materials.length !== 1 || mesh.geometry.materialIndex ||\n        mesh.materials[0].displacementMap) return;\n      if (!checked.has(mesh.geometry)) {\n        const next = geometryState(mesh.geometry);\n        if (next.some((value, i) => value !== previous[i])) return;\n        checked.add(mesh.geometry);\n      }\n    }\n    // Rebuild active material slots when batches are added, removed or reassigned.\n    // Retired material identities must not remain in the active table.\n    const materials: Prepared['materials'] = [];\n    const materialIndices = new Map<Prepared['materials'][number], number>();\n    const templates = new Map(res.instanceBindings.map(binding => [binding.geometry, binding]));\n    const instanceBindings = meshes.filter(mesh => mesh.geometry.position.array.length > 0).map(mesh => {\n      const material = mesh.materials[0];\n      let index = materialIndices.get(material);\n      if (index == null) { index = materials.length; materials.push(material); materialIndices.set(material, index); }\n      const template = templates.get(mesh.geometry);\n      // Bounds remain valid even after every instance of a geometry was removed.\n      // Use the resident geometry bounds for all supported BVH layouts.\n      return { mesh, geometry: mesh.geometry, blasIndex: geometries.get(mesh.geometry)!,\n        materialIndices: [index], displacementScale: 0,\n        bvhBounds: template?.bvhBounds ?? getBvhBounds(mesh.geometry) };\n    });\n    const instances = updateMeshInstanceTransforms(instanceBindings);\n    if (meshes.some(mesh => mesh instanceof InstancedMesh)) return {...res, materials, instances, instanceBindings,\n      emissiveBindings: instanceBindings.filter(binding => materialIsEmissive(materials[binding.materialIndices[0]]))};\n    const ordinary = instances as import('./bvhAccel.js').TLASInstance[];\n    // Match full preparation's exclusion of invalid transforms/bounds.\n    const valid = ordinary.map((instance, i) => instance.bounds.min.every(Number.isFinite) &&\n      instance.bounds.max.every(Number.isFinite) && instance.transform.every(Number.isFinite) ? i : -1).filter(i => i >= 0);\n    const bindings = valid.map(i => instanceBindings[i]);\n    return { ...res, materials, instances: valid.map(i => ordinary[i]), instanceBindings: bindings,\n      emissiveBindings: bindings.filter(binding => materialIsEmissive(materials[binding.materialIndices[0]])) };\n  }\n}"]}

package/lib/renderer/oidnNetwork.d.ts

@@ -0,0 +1,20 @@
+/// <reference types="@webgpu/types" preserve="true" />
+import { type UNet } from 'oidn-web';
+import type Engine from '../Engine.js';
+import type { OIDNDenoiserOptions } from '../Render.js';
+export interface OIDNAuxiliaryDenoisers {
+    /** Loaded albedo prefilter; supplied by the clean-aux loader. */
+    albedo: OIDNDenoiser;
+    /** Loaded normal prefilter; supplied by the clean-aux loader. */
+    normal: OIDNDenoiser;
+}
+export type OIDNDenoiser = UNet & {
+    /** Optional RT auxiliary filters used before a clean-aux beauty model. */
+    auxiliaryDenoisers?: OIDNAuxiliaryDenoisers;
+    /** Optional shape preparation supported by newer oidn-web runtimes. */
+    prepareForImage?: (width: number, height: number, options?: {
+        tileOverlap?: number;
+    }) => Promise<void>;
+};
+/** Internal lazy loader; both public OIDN option branches use this path. */
+export declare function loadOIDNNetwork(engine: Engine, options: Readonly<OIDNDenoiserOptions>): Promise<OIDNDenoiser>;

package/lib/renderer/oidnNetwork.js

@@ -0,0 +1,123 @@
+/// <reference types="@webgpu/types" preserve="true" />
+import { initUNetFromBuffer } from 'oidn-web';
+import { resolveOIDNModelUrl, resolveOIDNAuxiliaryModelUrls, DEFAULT_OIDN_WEIGHTS_BASE_URL, OIDN_WEIGHTS_PACKAGE } from '../oidn.js';
+async function fetchOIDNWeights(url, signal) {
+    const response = await fetch(String(url), { signal });
+    if (!response.ok) {
+        throw new Error(`Failed to load OIDN weights from ${url}: ${response.status} ${response.statusText}`);
+    }
+    const buffer = await response.arrayBuffer();
+    const bytes = new Uint8Array(buffer);
+    if (bytes[0] !== 0xD7 || bytes[1] !== 0x41) {
+        throw new Error(`Invalid OIDN TZA weights from ${url}: expected magic D7 41 (received non-TZA content).`);
+    }
+    return buffer;
+}
+/**
+ * Loads an HDR OIDN model with albedo/normal guides on the engine's WebGPU
+ * device and returns it without attaching it to a render.
+ */
+async function loadOIDN(engine, weightsUrl, options = {}, fetchWeights = fetchOIDNWeights) {
+    await engine.init();
+    const device = engine.device;
+    if (engine.backend !== 'webgpu' || !device) {
+        throw new Error('OIDN requires a WebGPU engine.');
+    }
+    const auxiliaryUrls = options.auxiliaryModelUrls;
+    const weightBuffers = await Promise.all([
+        engine._getWeights(String(weightsUrl), () => fetchWeights(weightsUrl, options.signal)),
+        ...(auxiliaryUrls
+            ? [
+                engine._getWeights(String(auxiliaryUrls.albedo), () => fetchWeights(auxiliaryUrls.albedo, options.signal)),
+                engine._getWeights(String(auxiliaryUrls.normal), () => fetchWeights(auxiliaryUrls.normal, options.signal))
+            ]
+            : [])
+    ]);
+    const backend = { device };
+    const runtimeOptions = {
+        hdrTransfer: options.hdrTransfer ?? 'pu',
+        maxTileSize: options.maxTileSize,
+        engine: options.engine,
+        precision: options.precision,
+        dynamicTile: options.dynamicTile
+    };
+    let denoiser;
+    let albedoDenoiser;
+    try {
+        denoiser = await initUNetFromBuffer(weightBuffers[0], backend, {
+            aux: options.aux ?? true,
+            hdr: options.hdr ?? true,
+            ...runtimeOptions
+        });
+        if (auxiliaryUrls) {
+            // Auxiliary RT networks are color-only three-channel UNets. The render
+            // pipeline supplies their feature-specific transfer functions.
+            albedoDenoiser = await initUNetFromBuffer(weightBuffers[1], backend, {
+                aux: false,
+                hdr: false,
+                ...runtimeOptions
+            });
+            const normalDenoiser = await initUNetFromBuffer(weightBuffers[2], backend, {
+                aux: false,
+                hdr: false,
+                ...runtimeOptions
+            });
+            denoiser.auxiliaryDenoisers = {
+                albedo: albedoDenoiser,
+                normal: normalDenoiser
+            };
+        }
+    }
+    catch (error) {
+        albedoDenoiser?.dispose();
+        denoiser?.dispose();
+        throw error;
+    }
+    return denoiser;
+}
+/** Loads an explicit HDR network URL; download failures never fall back to a CDN. */
+function initializeOIDNFromURL(engine, weightsUrl, options = {}) {
+    return loadOIDN(engine, weightsUrl, options);
+}
+let warnedBundledWeights = false;
+/** Loads one of the packaged OIDN models. */
+function initializeOIDN(engine, options = {}) {
+    const { model = 'standard', weightsBaseUrl, auxiliaryModelUrls, ...loadOptions } = options;
+    const mainUrl = resolveOIDNModelUrl(model, weightsBaseUrl);
+    const defaultBundled = weightsBaseUrl === undefined && model !== 'cleanAuxLarge';
+    const defaultLarge = new Set(weightsBaseUrl === undefined && model === 'cleanAuxLarge'
+        ? [mainUrl, ...(!auxiliaryModelUrls ? Object.values(resolveOIDNAuxiliaryModelUrls()).map(String) : [])]
+        : []);
+    const fetchWeights = async (url, signal) => {
+        try {
+            return await fetchOIDNWeights(url, signal);
+        }
+        catch (error) {
+            if (signal?.aborted)
+                throw error;
+            if (defaultBundled && String(url) === mainUrl) {
+                if (!warnedBundledWeights) {
+                    warnedBundledWeights = true;
+                    console.warn('Bundled OIDN weights failed to load; falling back to jsDelivr. In Vite development add optimizeDeps: { exclude: ["@meimu/mutracer"] }, or pass weightsBaseUrl to self-host the weights.', error);
+                }
+                return fetchOIDNWeights(resolveOIDNModelUrl(model, DEFAULT_OIDN_WEIGHTS_BASE_URL), signal);
+            }
+            if (defaultLarge.has(String(url))) {
+                throw new Error(`Failed to load ${OIDN_WEIGHTS_PACKAGE} network from ${url}. Self-host node_modules/oidn-weights/*.tza and pass weightsBaseUrl. ${error instanceof Error ? error.message : String(error)}`);
+            }
+            throw error;
+        }
+    };
+    return loadOIDN(engine, mainUrl, {
+        ...loadOptions,
+        auxiliaryModelUrls: auxiliaryModelUrls ?? (model === 'cleanAuxLarge'
+            ? resolveOIDNAuxiliaryModelUrls(weightsBaseUrl) : undefined)
+    }, fetchWeights);
+}
+/** Internal lazy loader; both public OIDN option branches use this path. */
+export function loadOIDNNetwork(engine, options) {
+    if (options.weightsUrl !== undefined)
+        return initializeOIDNFromURL(engine, options.weightsUrl, options);
+    return initializeOIDN(engine, options);
+}
+//# sourceMappingURL=oidnNetwork.js.map

package/lib/renderer/oidnNetwork.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"oidnNetwork.js","sourceRoot":"","sources":["../../src/renderer/oidnNetwork.ts"],"names":[],"mappings":"AAAA,uDAAuD;AACvD,OAAO,EAAE,kBAAkB,EAAa,MAAM,UAAU,CAAC;AAGzD,OAAO,EACL,mBAAmB,EAAE,6BAA6B,EAAE,6BAA6B,EACjF,oBAAoB,EACrB,MAAM,YAAY,CAAC;AAwDpB,KAAK,UAAU,gBAAgB,CAAC,GAAiB,EAAE,MAAoB;IACrE,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;IACtD,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,MAAM,IAAI,KAAK,CACb,oCAAoC,GAAG,KAAK,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,EAAE,CACrF,CAAC;IACJ,CAAC;IACD,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,WAAW,EAAE,CAAC;IAC5C,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,MAAM,CAAC,CAAC;IACrC,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QAC3C,MAAM,IAAI,KAAK,CAAC,iCAAiC,GAAG,oDAAoD,CAAC,CAAC;IAC5G,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;GAGG;AACH,KAAK,UAAU,QAAQ,CACrB,MAAc,EACd,UAAwB,EACxB,UAAkC,EAAE,EACpC,YAAY,GAAG,gBAAgB;IAE/B,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;IACpB,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;IAC7B,IAAI,MAAM,CAAC,OAAO,KAAK,QAAQ,IAAI,CAAC,MAAM,EAAE,CAAC;QAC3C,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;IACpD,CAAC;IAED,MAAM,aAAa,GAAG,OAAO,CAAC,kBAAkB,CAAC;IACjD,MAAM,aAAa,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;QACtC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,GAAG,EAAE,CAAC,YAAY,CAAC,UAAU,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;QACtF,GAAG,CAAC,aAAa;YACf,CAAC,CAAC;gBACE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,EAAE,GAAG,EAAE,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;gBAC1G,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,EAAE,GAAG,EAAE,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;aAC3G;YACH,CAAC,CAAC,EAAE,CAAC;KACR,CAAC,CAAC;IACH,MAAM,OAAO,GAAG,EAAE,MAAM,EAAE,CAAC;IAC3B,MAAM,cAAc,GAAG;QACrB,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,IAAI;QACxC,WAAW,EAAE,OAAO,CAAC,WAAW;QAChC,MAAM,EAAE,OAAO,CAAC,MAAM;QACtB,SAAS,EAAE,OAAO,CAAC,SAAS;QAC5B,WAAW,EAAE,OAAO,CAAC,WAAW;KACjC,CAAC;IACF,IAAI,QAAkC,CAAC;IACvC,IAAI,cAAwC,CAAC;IAC7C,IAAI,CAAC;QACH,QAAQ,GAAG,MAAM,kBAAkB,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE;YAC7D,GAAG,EAAE,OAAO,CAAC,GAAG,IAAI,IAAI;YACxB,GAAG,EAAE,OAAO,CAAC,GAAG,IAAI,IAAI;YACxB,GAAG,cAAc;SAClB,CAAC,CAAC;QACH,IAAI,aAAa,EAAE,CAAC;YAClB,uEAAuE;YACvE,+DAA+D;YAC/D,cAAc,GAAG,MAAM,kBAAkB,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE;gBACnE,GAAG,EAAE,KAAK;gBACV,GAAG,EAAE,KAAK;gBACV,GAAG,cAAc;aAClB,CAAC,CAAC;YACH,MAAM,cAAc,GAAG,MAAM,kBAAkB,CAC7C,aAAa,CAAC,CAAC,CAAC,EAChB,OAAO,EACP;gBACE,GAAG,EAAE,KAAK;gBACV,GAAG,EAAE,KAAK;gBACV,GAAG,cAAc;aAClB,CACF,CAAC;YACF,QAAQ,CAAC,kBAAkB,GAAG;gBAC5B,MAAM,EAAE,cAAc;gBACtB,MAAM,EAAE,cAAc;aACvB,CAAC;QACJ,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,cAAc,EAAE,OAAO,EAAE,CAAC;QAC1B,QAAQ,EAAE,OAAO,EAAE,CAAC;QACpB,MAAM,KAAK,CAAC;IACd,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,qFAAqF;AACrF,SAAS,qBAAqB,CAC5B,MAAc,EACd,UAAwB,EACxB,UAAkC,EAAE;IAEpC,OAAO,QAAQ,CAAC,MAAM,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;AAC/C,CAAC;AAED,IAAI,oBAAoB,GAAG,KAAK,CAAC;AAEjC,6CAA6C;AAC7C,SAAS,cAAc,CACrB,MAAc,EACd,UAAiC,EAAE;IAEnC,MAAM,EACJ,KAAK,GAAG,UAAU,EAClB,cAAc,EACd,kBAAkB,EAClB,GAAG,WAAW,EACf,GAAG,OAAO,CAAC;IACZ,MAAM,OAAO,GAAG,mBAAmB,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC;IAC3D,MAAM,cAAc,GAAG,cAAc,KAAK,SAAS,IAAI,KAAK,KAAK,eAAe,CAAC;IACjF,MAAM,YAAY,GAAG,IAAI,GAAG,CAAS,cAAc,KAAK,SAAS,IAAI,KAAK,KAAK,eAAe;QAC5F,CAAC,CAAC,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,6BAA6B,EAAE,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACvG,CAAC,CAAC,EAAE,CAAC,CAAC;IACR,MAAM,YAAY,GAAG,KAAK,EAAE,GAAiB,EAAE,MAAoB,EAAE,EAAE;QACrE,IAAI,CAAC;YACH,OAAO,MAAM,gBAAgB,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAC7C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,MAAM,EAAE,OAAO;gBAAE,MAAM,KAAK,CAAC;YACjC,IAAI,cAAc,IAAI,MAAM,CAAC,GAAG,CAAC,KAAK,OAAO,EAAE,CAAC;gBAC9C,IAAI,CAAC,oBAAoB,EAAE,CAAC;oBAC1B,oBAAoB,GAAG,IAAI,CAAC;oBAC5B,OAAO,CAAC,IAAI,CAAC,yLAAyL,EAAE,KAAK,CAAC,CAAC;gBACjN,CAAC;gBACD,OAAO,gBAAgB,CAAC,mBAAmB,CAAC,KAAK,EAAE,6BAA6B,CAAC,EAAE,MAAM,CAAC,CAAC;YAC7F,CAAC;YACD,IAAI,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;gBAClC,MAAM,IAAI,KAAK,CAAC,kBAAkB,oBAAoB,iBAAiB,GAAG,wEAAwE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YAC9M,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC,CAAC;IACF,OAAO,QAAQ,CACb,MAAM,EACN,OAAO,EACP;QACE,GAAG,WAAW;QACd,kBAAkB,EAChB,kBAAkB,IAAI,CAAC,KAAK,KAAK,eAAe;YAC9C,CAAC,CAAC,6BAA6B,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;KACjE,EACD,YAAY,CACb,CAAC;AACJ,CAAC;AAED,4EAA4E;AAC5E,MAAM,UAAU,eAAe,CAAC,MAAc,EAAE,OAAsC;IACpF,IAAI,OAAO,CAAC,UAAU,KAAK,SAAS;QAAE,OAAO,qBAAqB,CAAC,MAAM,EAAE,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;IACxG,OAAO,cAAc,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;AACzC,CAAC","sourcesContent":["/// <reference types=\"@webgpu/types\" preserve=\"true\" />\nimport { initUNetFromBuffer, type UNet } from 'oidn-web';\nimport type Engine from '../Engine.js';\nimport type { OIDNDenoiserOptions } from '../Render.js';\nimport {\n  resolveOIDNModelUrl, resolveOIDNAuxiliaryModelUrls, DEFAULT_OIDN_WEIGHTS_BASE_URL,\n  OIDN_WEIGHTS_PACKAGE\n} from '../oidn.js';\nimport type {\n  OIDNModel, OIDNEngine, OIDNPrecision, OIDNHDRTransfer,\n  OIDNDynamicTileSetting, OIDNAuxiliaryModelUrls\n} from '../oidn.js';\n\nexport interface OIDNAuxiliaryDenoisers {\n  /** Loaded albedo prefilter; supplied by the clean-aux loader. */\n  albedo: OIDNDenoiser;\n  /** Loaded normal prefilter; supplied by the clean-aux loader. */\n  normal: OIDNDenoiser;\n}\nexport type OIDNDenoiser = UNet & {\n  /** Optional RT auxiliary filters used before a clean-aux beauty model. */\n  auxiliaryDenoisers?: OIDNAuxiliaryDenoisers;\n  /** Optional shape preparation supported by newer oidn-web runtimes. */\n  prepareForImage?: (\n    width: number,\n    height: number,\n    options?: { tileOverlap?: number }\n  ) => Promise<void>;\n};\n\ninterface OIDNLoadOptions {\n  /** Maximum OIDN tile size. By default oidn-web chooses the tile size. */\n  maxTileSize?: number;\n  /** Optional signal used while downloading the weights. */\n  signal?: AbortSignal;\n  /** OIDN execution engine. `webnn` enables the experimental WebNN FP16 path. */\n  engine?: OIDNEngine;\n  /** Native WGSL precision. `auto` uses FP16 when shader-f16 is available. */\n  precision?: OIDNPrecision;\n  /** Adaptive tile controller settings, or false for fixed tiles. */\n  dynamicTile?: OIDNDynamicTileSetting;\n}\n\ninterface InitializeOIDNOptions extends OIDNLoadOptions {\n  /** Packaged model to load. Defaults to the standard model. */\n  model?: OIDNModel;\n  /** Override directory for all weights. Defaults to bundled standard/small assets and the oidn-weights CDN for large networks. */\n  weightsBaseUrl?: string | URL;\n  /** Override the packaged albedo/normal prefilter weights for a clean-aux model. */\n  auxiliaryModelUrls?: OIDNAuxiliaryModelUrls;\n}\n\ninterface CustomOIDNModelOptions extends OIDNLoadOptions {\n  /** Whether the custom model consumes albedo and normal guides. Defaults to true. */\n  aux?: boolean;\n  /** Whether the custom model consumes HDR color. Defaults to true. */\n  hdr?: boolean;\n  /** HDR encoding expected by the weights. OIDN's RT models use PU. */\n  hdrTransfer?: OIDNHDRTransfer;\n  /** Optional albedo/normal filters to run before the custom clean-aux model. */\n  auxiliaryModelUrls?: OIDNAuxiliaryModelUrls;\n}\n\nasync function fetchOIDNWeights(url: string | URL, signal?: AbortSignal) {\n  const response = await fetch(String(url), { signal });\n  if (!response.ok) {\n    throw new Error(\n      `Failed to load OIDN weights from ${url}: ${response.status} ${response.statusText}`\n    );\n  }\n  const buffer = await response.arrayBuffer();\n  const bytes = new Uint8Array(buffer);\n  if (bytes[0] !== 0xD7 || bytes[1] !== 0x41) {\n    throw new Error(`Invalid OIDN TZA weights from ${url}: expected magic D7 41 (received non-TZA content).`);\n  }\n  return buffer;\n}\n\n/**\n * Loads an HDR OIDN model with albedo/normal guides on the engine's WebGPU\n * device and returns it without attaching it to a render.\n */\nasync function loadOIDN(\n  engine: Engine,\n  weightsUrl: string | URL,\n  options: CustomOIDNModelOptions = {},\n  fetchWeights = fetchOIDNWeights\n): Promise<OIDNDenoiser> {\n  await engine.init();\n  const device = engine.device;\n  if (engine.backend !== 'webgpu' || !device) {\n    throw new Error('OIDN requires a WebGPU engine.');\n  }\n\n  const auxiliaryUrls = options.auxiliaryModelUrls;\n  const weightBuffers = await Promise.all([\n    engine._getWeights(String(weightsUrl), () => fetchWeights(weightsUrl, options.signal)),\n    ...(auxiliaryUrls\n      ? [\n          engine._getWeights(String(auxiliaryUrls.albedo), () => fetchWeights(auxiliaryUrls.albedo, options.signal)),\n          engine._getWeights(String(auxiliaryUrls.normal), () => fetchWeights(auxiliaryUrls.normal, options.signal))\n        ]\n      : [])\n  ]);\n  const backend = { device };\n  const runtimeOptions = {\n    hdrTransfer: options.hdrTransfer ?? 'pu',\n    maxTileSize: options.maxTileSize,\n    engine: options.engine,\n    precision: options.precision,\n    dynamicTile: options.dynamicTile\n  };\n  let denoiser: OIDNDenoiser | undefined;\n  let albedoDenoiser: OIDNDenoiser | undefined;\n  try {\n    denoiser = await initUNetFromBuffer(weightBuffers[0], backend, {\n      aux: options.aux ?? true,\n      hdr: options.hdr ?? true,\n      ...runtimeOptions\n    });\n    if (auxiliaryUrls) {\n      // Auxiliary RT networks are color-only three-channel UNets. The render\n      // pipeline supplies their feature-specific transfer functions.\n      albedoDenoiser = await initUNetFromBuffer(weightBuffers[1], backend, {\n        aux: false,\n        hdr: false,\n        ...runtimeOptions\n      });\n      const normalDenoiser = await initUNetFromBuffer(\n        weightBuffers[2],\n        backend,\n        {\n          aux: false,\n          hdr: false,\n          ...runtimeOptions\n        }\n      );\n      denoiser.auxiliaryDenoisers = {\n        albedo: albedoDenoiser,\n        normal: normalDenoiser\n      };\n    }\n  } catch (error) {\n    albedoDenoiser?.dispose();\n    denoiser?.dispose();\n    throw error;\n  }\n  return denoiser;\n}\n\n/** Loads an explicit HDR network URL; download failures never fall back to a CDN. */\nfunction initializeOIDNFromURL(\n  engine: Engine,\n  weightsUrl: string | URL,\n  options: CustomOIDNModelOptions = {}\n): Promise<OIDNDenoiser> {\n  return loadOIDN(engine, weightsUrl, options);\n}\n\nlet warnedBundledWeights = false;\n\n/** Loads one of the packaged OIDN models. */\nfunction initializeOIDN(\n  engine: Engine,\n  options: InitializeOIDNOptions = {}\n) {\n  const {\n    model = 'standard',\n    weightsBaseUrl,\n    auxiliaryModelUrls,\n    ...loadOptions\n  } = options;\n  const mainUrl = resolveOIDNModelUrl(model, weightsBaseUrl);\n  const defaultBundled = weightsBaseUrl === undefined && model !== 'cleanAuxLarge';\n  const defaultLarge = new Set<string>(weightsBaseUrl === undefined && model === 'cleanAuxLarge'\n    ? [mainUrl, ...(!auxiliaryModelUrls ? Object.values(resolveOIDNAuxiliaryModelUrls()).map(String) : [])]\n    : []);\n  const fetchWeights = async (url: string | URL, signal?: AbortSignal) => {\n    try {\n      return await fetchOIDNWeights(url, signal);\n    } catch (error) {\n      if (signal?.aborted) throw error;\n      if (defaultBundled && String(url) === mainUrl) {\n        if (!warnedBundledWeights) {\n          warnedBundledWeights = true;\n          console.warn('Bundled OIDN weights failed to load; falling back to jsDelivr. In Vite development add optimizeDeps: { exclude: [\"@meimu/mutracer\"] }, or pass weightsBaseUrl to self-host the weights.', error);\n        }\n        return fetchOIDNWeights(resolveOIDNModelUrl(model, DEFAULT_OIDN_WEIGHTS_BASE_URL), signal);\n      }\n      if (defaultLarge.has(String(url))) {\n        throw new Error(`Failed to load ${OIDN_WEIGHTS_PACKAGE} network from ${url}. Self-host node_modules/oidn-weights/*.tza and pass weightsBaseUrl. ${error instanceof Error ? error.message : String(error)}`);\n      }\n      throw error;\n    }\n  };\n  return loadOIDN(\n    engine,\n    mainUrl,\n    {\n      ...loadOptions,\n      auxiliaryModelUrls:\n        auxiliaryModelUrls ?? (model === 'cleanAuxLarge'\n          ? resolveOIDNAuxiliaryModelUrls(weightsBaseUrl) : undefined)\n    },\n    fetchWeights\n  );\n}\n\n/** Internal lazy loader; both public OIDN option branches use this path. */\nexport function loadOIDNNetwork(engine: Engine, options: Readonly<OIDNDenoiserOptions>): Promise<OIDNDenoiser> {\n  if (options.weightsUrl !== undefined) return initializeOIDNFromURL(engine, options.weightsUrl, options);\n  return initializeOIDN(engine, options);\n}\n"]}

package/lib/renderer/openPbrEnergyLut.d.ts

@@ -0,0 +1,32 @@
+/**
+ * OpenPBR opaque-dielectric, ideal-metal and ideal-dielectric energy compensation tables.
+ *
+ * Copyright 2026 Adobe
+ * Licensed under the Apache License, Version 2.0.
+ * Derived from adobe/openpbr-bsdf commit
+ * 9edf806740d2140846d9bef76e4342fc458e2ef5.
+ * This repository re-quantized the Adobe data to 8-bit row-normalized values.
+ * Directional rows store Uint16 min/max followed by 32 Uint8 samples.
+ * Metal and ideal-dielectric averages store exact little-endian Uint16 values.
+ * They decode directly to Float32 without row quantization or repeated texels.
+ * Storage uses differences from the previous row: little-endian min/max
+ * modulo 65536 and samples modulo 256. The first row is relative to zero;
+ * differences continue across all concatenated energy tables.
+ * Decoded UNORM value: (min + (max - min) * sample / 255) / 65535.
+ * https://github.com/adobe/openpbr-bsdf
+ */
+export declare const OPENPBR_ENERGY_LUT_SIZE = 32;
+export declare const OPENPBR_ENERGY_LUT_AVERAGE_ROW = 1024;
+export declare const OPENPBR_ENERGY_LUT_WIDTH = 32;
+/** Ideal-metal complement: roughness rows, cosine columns. */
+export declare const OPENPBR_METAL_ENERGY_LUT_ROW = 1056;
+/** Ideal-metal cosine-weighted average: roughness columns, one row. */
+export declare const OPENPBR_METAL_ENERGY_LUT_AVERAGE_ROW = 1088;
+/** Ideal-dielectric loss: IOR-major, roughness rows, cosine columns. */
+export declare const OPENPBR_DIELECTRIC_ENERGY_LUT_ROW = 1089;
+/** Exact u16 averages: IOR rows, roughness columns. */
+export declare const OPENPBR_DIELECTRIC_ENERGY_LUT_AVERAGE_ROW = 2113;
+/** Multiple-scattering reflection ratio: IOR rows, roughness columns. */
+export declare const OPENPBR_DIELECTRIC_REFLECTION_RATIO_LUT_ROW = 2145;
+export declare const OPENPBR_ENERGY_LUT_HEIGHT = 2177;
+export declare function getOpenPbrEnergyLutData(): Float32Array;