rayzee 11.3.0 → 11.5.0

package/dist/{ToneMapGPU-BOjpul7A.js → ToneMapGPU-sAs8wdgI.js}

@@ -22,7 +22,7 @@ var a = {
 }, s = {
 	rgba8: 4,
 	planar: 16
-}, c = (e, t, n) => `
+}, c = (e, t, n, r) => `
 struct Params {
 	width: u32,
 	height: u32,
@@ -38,6 +38,7 @@ ${a[t]}
 
 ${e}
 ${o[n]}
+${r ?? ""}
 
 @compute @workgroup_size(8, 8)
 fn main( @builtin(global_invocation_id) gid: vec3<u32> ) {
@@ -47,25 +48,29 @@ fn main( @builtin(global_invocation_id) gid: vec3<u32> ) {
 	if ( params.flipY == 1u ) { srcY = params.height - 1u - gid.y; }
 
 	let linear = rayzee_source( gid.x, srcY );
-	rayzee_store( gid.y * params.width + gid.x, linear, rayzee_tone_map( linear.rgb, params.mode, params.exposure, params.saturation ) );
+	let uv = vec2<f32>( ( f32( gid.x ) + 0.5 ) / f32( params.width ), ( f32( gid.y ) + 0.5 ) / f32( params.height ) );
+	let rgb = linear.rgb${r ? " * rayzee_gain( uv, linear.rgb, params.exposure )" : ""};
+	rayzee_store( gid.y * params.width + gid.x, linear, rayzee_tone_map( rgb, params.mode, params.exposure, params.saturation ) );
 }
 `, l = 32, u = class {
 	constructor(e, t = "rayzee:tonemap", { input: n = "packed", output: r = "rgba8" } = {}) {
-		this.device = e, this.label = t, this.input = n, this.output = r, this.width = 0, this.height = 0, this._pipeline = null, this._storage = null, this._map = null, this._params = null, this._paramData = new ArrayBuffer(l), this._paramU32 = new Uint32Array(this._paramData), this._paramF32 = new Float32Array(this._paramData), this._shaderVersion = -1, this._bindings = [], this._tables = /* @__PURE__ */ new Map(), this.disposed = !1;
+		this.device = e, this.label = t, this.input = n, this.output = r, this.width = 0, this.height = 0, this._pipelines = /* @__PURE__ */ new Map(), this._storage = null, this._map = null, this._params = null, this._paramData = new ArrayBuffer(l), this._paramU32 = new Uint32Array(this._paramData), this._paramF32 = new Float32Array(this._paramData), this._shaderVersion = -1, this._bindings = [], this._tables = /* @__PURE__ */ new Map(), this.disposed = !1;
 	}
-	_ensurePipeline() {
-		let { wgsl: e, bindings: t, version: n } = i();
-		this._pipeline && this._shaderVersion === n || (this._bindings = t, this._shaderVersion = n, this._pipeline = this.device.createComputePipeline({
+	_ensurePipeline(e = null) {
+		let { wgsl: t, bindings: n, version: r } = i();
+		this._shaderVersion !== r && (this._pipelines.clear(), this._bindings = n, this._shaderVersion = r);
+		let a = e?.key ?? "", o = this._pipelines.get(a);
+		return o || (o = this.device.createComputePipeline({
 			label: this.label,
 			layout: "auto",
 			compute: {
 				module: this.device.createShaderModule({
 					label: this.label,
-					code: c(e, this.input, this.output)
+					code: c(t, this.input, this.output, e?.wgsl)
 				}),
 				entryPoint: "main"
 			}
-		}));
+		}), this._pipelines.set(a, o), o);
 	}
 	_ensureTables() {
 		let e = /* @__PURE__ */ new Set();
@@ -125,24 +130,38 @@ fn main( @builtin(global_invocation_id) gid: vec3<u32> ) {
 	async toPlanar(e, t) {
 		return new Float32Array(await this._run(e, t));
 	}
-	async _run(e, { exposure: t = 1, toneMapping: n = 0, saturation: r = 1, flipY: i = !1, preserveAlpha: a = !1 } = {}) {
+	async _run(e, t = {}) {
+		let n = t.gain;
+		if (!n?.device || n.device === this.device) return this._dispatch(e, t);
+		let r = await n.copyTo(this.device);
+		try {
+			return await this._dispatch(e, {
+				...t,
+				gain: r
+			});
+		} finally {
+			r.destroy();
+		}
+	}
+	async _dispatch(e, { exposure: t = 1, toneMapping: n = 0, saturation: r = 1, flipY: i = !1, preserveAlpha: a = !1, gain: o = null } = {}) {
 		if (this.disposed) throw Error("PackedToneMapper: disposed");
 		if (!this._storage) throw Error("PackedToneMapper: call ensureSize() first");
-		this._ensurePipeline(), this._ensureTables();
-		let { width: o, height: c } = this, l = o * c * s[this.output];
-		this._paramU32[0] = o, this._paramU32[1] = c, this._paramU32[2] = n >>> 0, this._paramU32[3] = +!!i, this._paramF32[4] = t, this._paramF32[5] = r, this._paramU32[6] = +!!a, this.device.queue.writeBuffer(this._params, 0, this._paramData);
-		let u = [...this._tables.entries()].map(([e, t]) => ({
+		let c = this._ensurePipeline(o);
+		this._ensureTables();
+		let { width: l, height: u } = this, d = l * u * s[this.output];
+		this._paramU32[0] = l, this._paramU32[1] = u, this._paramU32[2] = n >>> 0, this._paramU32[3] = +!!i, this._paramF32[4] = t, this._paramF32[5] = r, this._paramU32[6] = +!!a, this.device.queue.writeBuffer(this._params, 0, this._paramData);
+		let f = [...this._tables.entries()].map(([e, t]) => ({
 			binding: e,
 			resource: t.view
-		})), d = this.device.createBindGroup({
-			layout: this._pipeline.getBindGroupLayout(0),
+		})), p = this.device.createBindGroup({
+			layout: c.getBindGroupLayout(0),
 			entries: [
-				...u,
+				...f,
 				{
 					binding: 0,
 					resource: this.input === "texture" ? e.createView() : {
 						buffer: e,
-						size: o * c * 8
+						size: l * u * 8
 					}
 				},
 				{
@@ -154,10 +173,13 @@ fn main( @builtin(global_invocation_id) gid: vec3<u32> ) {
 					resource: { buffer: this._params }
 				}
 			]
-		}), f = this.device.createCommandEncoder({ label: this.label }), p = f.beginComputePass();
-		p.setPipeline(this._pipeline), p.setBindGroup(0, d), p.dispatchWorkgroups(Math.ceil(o / 8), Math.ceil(c / 8)), p.end(), f.copyBufferToBuffer(this._storage, 0, this._map, 0, l), this.device.queue.submit([f.finish()]), await this._map.mapAsync(GPUMapMode.READ);
-		let m = this._map.getMappedRange().slice(0);
-		return this._map.unmap(), m;
+		}), m = this.device.createCommandEncoder({ label: this.label }), h = m.beginComputePass();
+		h.setPipeline(c), h.setBindGroup(0, p), o && h.setBindGroup(1, this.device.createBindGroup({
+			layout: c.getBindGroupLayout(1),
+			entries: o.entries
+		})), h.dispatchWorkgroups(Math.ceil(l / 8), Math.ceil(u / 8)), h.end(), m.copyBufferToBuffer(this._storage, 0, this._map, 0, d), this.device.queue.submit([m.finish()]), await this._map.mapAsync(GPUMapMode.READ);
+		let g = this._map.getMappedRange().slice(0);
+		return this._map.unmap(), g;
 	}
 	release() {
 		this._releaseBuffers(), this.width = this.height = 0;
@@ -169,7 +191,7 @@ fn main( @builtin(global_invocation_id) gid: vec3<u32> ) {
 		if (!this.disposed) {
 			this.disposed = !0, this._releaseBuffers();
 			for (let e of this._tables.values()) e.texture.destroy();
-			this._tables.clear(), this._pipeline = null;
+			this._tables.clear(), this._pipelines.clear();
 		}
 	}
 };

package/dist/addons/archives.es.js

@@ -1,3 +1,3 @@
 import { n as e, t } from "../ZipReader-gDSWr0I5.js";
-import { n, t as r } from "../ArchiveImporter-D735xX-4.js";
+import { n, t as r } from "../ArchiveImporter-29Nl1567.js";
 export { r as ARCHIVE_ELEMENT_PROMPT_BYTES, n as ArchiveImporter, t as openZip, e as readZipDirectory };

package/dist/addons/bidirectional.es.js

@@ -1,2 +1,2 @@
-import { t as e } from "../BidirectionalIntegrator-M89c0AlP.js";
+import { t as e } from "../BidirectionalIntegrator-CJay8tXK.js";
 export { e as BidirectionalIntegrator };

package/dist/addons/physical-sky.es.js

@@ -1,2 +1,2 @@
-import { t as e } from "../PhysicalSky-DqmaS0lb.js";
+import { t as e } from "../PhysicalSky-CUWrTcmT.js";
 export { e as PhysicalSky };

package/dist/rayzee-core.es.js

@@ -1,8 +1,8 @@
 import "./SolarLimb-bZE6y63S.js";
-import { a as e, n as t, o as n, r, t as i } from "./RayzeeRenderer-DUSMFhmc.js";
-import { ut as a, wt as o } from "./PathTracer-BXCCVrPK.js";
+import { a as e, n as t, o as n, r, t as i } from "./RayzeeRenderer-Dd-Dit0v.js";
+import { ut as a, wt as o } from "./PathTracer-Cldt0mXi.js";
 import { n as s, t as c } from "./EngineEvents-Ci8dVWCP.js";
-import { G as l, q as u, r as d, x as f, y as p } from "./sceneCachePolicy-BAib5fCp.js";
+import { G as l, q as u, r as d, x as f, y as p } from "./sceneCachePolicy--ta6moJu.js";
 import { i as m, n as h, r as g, t as _ } from "./Logger-dDDGj9Av.js";
 import { f as v } from "./BufferLayout-xPRH9jYD.js";
 import { i as y, n as b, r as x, t as S } from "./EngineIssues-CIZnXHsF.js";