rayzee 5.6.0 → 5.6.1

package/dist/rayzee.es.js

@@ -8983,7 +8983,7 @@ var rc = 128, ic = class {
 		this.deltaTime.value = e;
 	}
 	dispose() {
-		this._screenSpaceComputeNode?.dispose(), this._worldSpaceComputeNode?.dispose(), this._screenSpaceStorageTex?.dispose(), this._worldSpaceStorageTex?.dispose(), this.screenSpaceTarget?.dispose(), this.worldSpaceTarget?.dispose();
+		this._screenSpaceComputeNode?.dispose(), this._worldSpaceComputeNode?.dispose(), this._screenSpaceStorageTex?.dispose(), this._worldSpaceStorageTex?.dispose(), this.screenSpaceTarget?.dispose(), this.worldSpaceTarget?.dispose(), this._normalDepthTexNode?.dispose();
 	}
 };
 //#endregion
@@ -9331,7 +9331,7 @@ var _c = class extends Sn {
 	}
 	reset() {}
 	dispose() {
-		this._gradientNode?.dispose(), this._temporalNodeA?.dispose(), this._temporalNodeB?.dispose(), this._temporalTexA?.dispose(), this._temporalTexB?.dispose(), this._prevNDTexA?.dispose(), this._prevNDTexB?.dispose(), this._gradientStorageTex?.dispose(), this._heatmapComputeNode?.dispose(), this._heatmapStorageTex?.dispose(), this.heatmapTarget?.dispose(), this.heatmapHelper?.dispose();
+		this._gradientNode?.dispose(), this._temporalNodeA?.dispose(), this._temporalNodeB?.dispose(), this._temporalTexA?.dispose(), this._temporalTexB?.dispose(), this._prevNDTexA?.dispose(), this._prevNDTexB?.dispose(), this._gradientStorageTex?.dispose(), this._heatmapComputeNode?.dispose(), this._heatmapStorageTex?.dispose(), this.heatmapTarget?.dispose(), this._colorTexNode?.dispose(), this._normalDepthTexNode?.dispose(), this._motionTexNode?.dispose(), this._readTemporalTexNode?.dispose(), this._readPrevNDTexNode?.dispose(), this._gradientReadTexNode?.dispose(), this._heatmapRawColorTexNode?.dispose(), this._heatmapColorTexNode?.dispose(), this._heatmapTemporalTexNode?.dispose(), this._heatmapNDTexNode?.dispose(), this._heatmapMotionTexNode?.dispose(), this._heatmapGradientTexNode?.dispose(), this.heatmapHelper?.dispose();
 	}
 }, vc = /* @__PURE__ */ K("\n	fn temporalAccumulate(\n		lum: f32,\n		prevMean: f32,\n		prevMeanSq: f32,\n		alpha: f32,\n		spatialVariance: f32,\n		varianceBoost: f32\n	) -> vec4f {\n\n		let newMean = prevMean + ( lum - prevMean ) * alpha;\n		let newMeanSq = prevMeanSq + ( lum * lum - prevMeanSq ) * alpha;\n		let temporalVariance = max( newMeanSq - newMean * newMean, 0.0 );\n\n		return vec4f(\n			newMean,\n			newMeanSq,\n			temporalVariance * varianceBoost,\n			spatialVariance * varianceBoost\n		);\n\n	}\n"), yc = class extends Sn {
 	constructor(e, t = {}) {
@@ -9406,7 +9406,7 @@ var _c = class extends Sn {
 		];
 	}
 	dispose() {
-		this._computeNodeA?.dispose(), this._computeNodeB?.dispose(), this._storageTexA?.dispose(), this._storageTexB?.dispose();
+		this._computeNodeA?.dispose(), this._computeNodeB?.dispose(), this._storageTexA?.dispose(), this._storageTexB?.dispose(), this._colorTexNode?.dispose(), this._readTexNodeA?.dispose(), this._readTexNodeB?.dispose();
 	}
 }, bc = /* @__PURE__ */ K("\n	fn bilateralWeight(\n		centerLum: f32, sLum: f32,\n		centerNormal: vec3f, sNormal: vec3f,\n		centerDepth: f32, sDepth: f32,\n		centerColor: vec3f, sColor: vec3f,\n		kernelW: f32,\n		phiLum: f32, phiNorm: f32, phiDep: f32, phiCol: f32\n	) -> f32 {\n\n		let lumW = exp( -abs( centerLum - sLum ) * phiLum );\n		let normW = pow( max( dot( centerNormal, sNormal ), 0.0 ), phiNorm );\n		let depW = exp( -abs( centerDepth - sDepth ) / max( phiDep, 0.001 ) );\n		let maxDiff = max( max( abs( centerColor.x - sColor.x ),\n			abs( centerColor.y - sColor.y ) ),\n			abs( centerColor.z - sColor.z ) );\n		let colW = exp( -maxDiff * phiCol );\n		return kernelW * lumW * normW * depW * colW;\n\n	}\n"), xc = class extends Sn {
 	constructor(e, t = {}) {
@@ -9496,7 +9496,7 @@ var _c = class extends Sn {
 	}
 	reset() {}
 	dispose() {
-		this._computeNodeA?.dispose(), this._computeNodeB?.dispose(), this._storageTexA?.dispose(), this._storageTexB?.dispose();
+		this._computeNodeA?.dispose(), this._computeNodeB?.dispose(), this._storageTexA?.dispose(), this._storageTexB?.dispose(), this._readTexNode?.dispose(), this._normalDepthTexNode?.dispose();
 	}
 }, Sc = /* @__PURE__ */ K("\n	fn computeSamplingGuidance(\n		temporalVariance: f32,\n		spatialVariance: f32,\n		meanLuminance: f32,\n		threshold: f32,\n		frame: i32,\n		minFrames: i32,\n		convThreshold: f32,\n		sensitivity: f32,\n		convSpeedScale: f32\n	) -> vec4f {\n\n		// The path tracer accumulates via alpha = 1/(frame+1), so temporal variance\n		// of the accumulated output shrinks as ~sigma²/(frame+1)². Scale by (frame+1)\n		// to get accumulated image quality ~sigma²/N — decreases as image converges.\n		let frameScale = f32( frame + 1 );\n		let effectiveVariance = temporalVariance * frameScale;\n\n		// Normalize by luminance² — converts absolute variance to relative (CV²).\n		// Floor of 0.01 prevents noise amplification for near-black pixels\n		// (linear luminance < 0.1 → below perceptual visibility threshold).\n		let normFactor = max( meanLuminance * meanLuminance, 0.01 );\n		let normalizedVariance = effectiveVariance / normFactor;\n\n		let varianceRatio = clamp( normalizedVariance / threshold, 0.0, 1.0 );\n\n		// Apply sensitivity — higher values assign more samples to noisy pixels\n		var normalizedSamples = clamp( varianceRatio * sensitivity, 0.0, 1.0 );\n\n		// Small spatial boost for noisy neighbourhoods (un-scaled — provides\n		// a minor secondary signal that naturally diminishes as image converges)\n		let spatialBoost = clamp( spatialVariance / ( threshold * 4.0 ), 0.0, 0.2 );\n		normalizedSamples = clamp( normalizedSamples + spatialBoost, 0.0, 1.0 );\n\n		// Warm-up: variance estimates need a few frames to stabilise\n		if ( frame < minFrames ) {\n\n			let warmupFactor = f32( frame ) / f32( minFrames );\n			normalizedSamples = mix( 1.0, normalizedSamples, warmupFactor * warmupFactor );\n\n		}\n\n		// Convergence: mark pixel only when per-frame noise is truly negligible.\n		// convSpeedScale controls aggressiveness: higher = easier to converge\n		// (scales the threshold up, so more pixels qualify as converged).\n		let scaledConvThreshold = convThreshold * convSpeedScale;\n		var converged = 0.0;\n		if ( normalizedVariance < scaledConvThreshold && frame > minFrames ) {\n\n			converged = 1.0;\n\n		}\n\n		return vec4f(\n			normalizedSamples,\n			varianceRatio,\n			converged,\n			1.0\n		);\n\n	}\n"), Cc = /* @__PURE__ */ K("\n	fn heatmapGradient( t: f32, normalizedVariance: f32, converged: f32 ) -> vec4f {\n\n		let r = clamp( ( t - 0.5 ) * 4.0, 0.0, 1.0 );\n		let g = clamp( t * 4.0, 0.0, 1.0 ) - clamp( ( t - 0.75 ) * 4.0, 0.0, 1.0 );\n		let b = 1.0 - clamp( ( t - 0.25 ) * 4.0, 0.0, 1.0 );\n\n		var color = vec3f( r, g, b );\n\n		// Convergence: desaturate converged pixels\n		if ( converged > 0.5 ) {\n\n			let gray = color.x * 0.299 + color.y * 0.587 + color.z * 0.114;\n			color = mix( color, vec3f( gray ), 0.6 );\n\n		}\n\n		// Brightness modulation\n		color *= 0.7 + normalizedVariance * 0.3;\n\n		return vec4f( color, 1.0 );\n\n	}\n"), wc = class extends Sn {
 	constructor(e, t = {}) {
@@ -9604,7 +9604,7 @@ var _c = class extends Sn {
 		e.threshold !== void 0 && this.setVarianceThreshold(e.threshold), e.materialBias !== void 0 && this.setMaterialBias(e.materialBias), e.edgeBias !== void 0 && this.setEdgeBias(e.edgeBias), e.convergenceSpeedUp !== void 0 && this.setConvergenceSpeed(e.convergenceSpeedUp), e.adaptiveSamplingMax !== void 0 && this.setAdaptiveSamplingMax(e.adaptiveSamplingMax);
 	}
 	dispose() {
-		this._computeNode?.dispose(), this._heatmapComputeNode?.dispose(), this._heatmapStorageTex?.dispose(), this._outputStorageTex?.dispose(), this.heatmapTarget?.dispose(), this.helper?.dispose(), this._computeNode = null, this._heatmapComputeNode = null, this._heatmapStorageTex = null, this._outputStorageTex = null, this.heatmapTarget = null, this.helper = null;
+		this._computeNode?.dispose(), this._heatmapComputeNode?.dispose(), this._heatmapStorageTex?.dispose(), this._outputStorageTex?.dispose(), this.heatmapTarget?.dispose(), this._varianceTexNode?.dispose(), this.helper?.dispose(), this._computeNode = null, this._heatmapComputeNode = null, this._heatmapStorageTex = null, this._outputStorageTex = null, this.heatmapTarget = null, this._varianceTexNode = null, this.helper = null;
 	}
 }, Tc = class extends Sn {
 	constructor(e, t = {}) {
@@ -9684,7 +9684,7 @@ var _c = class extends Sn {
 		];
 	}
 	dispose() {
-		this._computeNode?.dispose(), this._outputStorageTex?.dispose(), this.outputTarget?.dispose();
+		this._computeNode?.dispose(), this._outputStorageTex?.dispose(), this.outputTarget?.dispose(), this._inputTexNode?.dispose();
 	}
 }, Ec = 256, Dc = -8, Oc = 6 - Dc, kc = Oc / Ec, Ac = 1e4, jc = /* @__PURE__ */ K("\n	fn adaptExposure(\n		geoMean: f32,\n		prevExposure: f32,\n		keyValue: f32,\n		minExp: f32,\n		maxExp: f32,\n		speedBright: f32,\n		speedDark: f32,\n		dt: f32,\n		isFirstFrame: f32\n	) -> vec4f {\n\n		let targetExp = clamp( keyValue / max( geoMean, 0.001 ), minExp, maxExp );\n		var newExposure = targetExp;\n\n		// Temporal smoothing (skip on first frame)\n		if ( isFirstFrame < 0.5 ) {\n\n			// Asymmetric speed: brighter scenes adapt faster\n			let speed = select( speedDark, speedBright, targetExp < prevExposure );\n			let alpha = 1.0 - exp( -dt * speed );\n			newExposure = mix( prevExposure, targetExp, alpha );\n\n		}\n\n		return vec4f( newExposure, geoMean, targetExp, 1.0 );\n\n	}\n"), Mc = class extends Sn {
 	constructor(e, t = {}) {
@@ -9838,7 +9838,7 @@ var _c = class extends Sn {
 		e.keyValue !== void 0 && (this.keyValueU.value = e.keyValue), e.minExposure !== void 0 && (this.minExposureU.value = e.minExposure), e.maxExposure !== void 0 && (this.maxExposureU.value = e.maxExposure), e.adaptSpeedBright !== void 0 && (this.adaptSpeedBrightU.value = e.adaptSpeedBright), e.adaptSpeedDark !== void 0 && (this.adaptSpeedDarkU.value = e.adaptSpeedDark), e.lowPercentile !== void 0 && (this.lowPercentileU.value = e.lowPercentile), e.highPercentile !== void 0 && (this.highPercentileU.value = e.highPercentile), e.centerWeight !== void 0 && (this.centerWeightU.value = e.centerWeight);
 	}
 	dispose() {
-		this._downsampleComputeNode?.dispose(), this._histogramComputeNode?.dispose(), this._histogramAnalyzeNode?.dispose(), this._adaptationComputeNode?.dispose(), this._downsampleTarget?.dispose(), this._downsampleStorageTex?.dispose(), this._reductionStorageTex?.dispose(), this._reductionReadTarget?.dispose(), this._readbackBuffer?.dispose();
+		this._downsampleComputeNode?.dispose(), this._histogramComputeNode?.dispose(), this._histogramAnalyzeNode?.dispose(), this._adaptationComputeNode?.dispose(), this._downsampleTarget?.dispose(), this._downsampleStorageTex?.dispose(), this._reductionStorageTex?.dispose(), this._reductionReadTarget?.dispose(), this._readbackBuffer?.dispose(), this._inputTexNode?.dispose(), this._reductionReadTexNode?.dispose();
 	}
 };
 //#endregion
@@ -9941,7 +9941,7 @@ var Fc = class extends Sn {
 		this._pass1NodeA && (this._pass1NodeA.dispatchSize = n), this._pass1NodeB && (this._pass1NodeB.dispatchSize = n), this._pass2Node && (this._pass2Node.dispatchSize = n), this._resetCache();
 	}
 	dispose() {
-		this._pass1NodeA?.dispose(), this._pass1NodeB?.dispose(), this._pass2Node?.dispose(), this._cacheTexA.dispose(), this._cacheTexB.dispose(), this._prevNDTexA.dispose(), this._prevNDTexB.dispose(), this._outputTex.dispose();
+		this._pass1NodeA?.dispose(), this._pass1NodeB?.dispose(), this._pass2Node?.dispose(), this._cacheTexA.dispose(), this._cacheTexB.dispose(), this._prevNDTexA.dispose(), this._prevNDTexB.dispose(), this._outputTex.dispose(), this._colorTexNode?.dispose(), this._ndTexNode?.dispose(), this._motionTexNode?.dispose(), this._readCacheTexNode?.dispose(), this._readPrevNDTexNode?.dispose(), this._readPass1CacheTexNode?.dispose();
 	}
 	updateParameters(e) {
 		e.temporalAlpha !== void 0 && (this.temporalAlpha.value = e.temporalAlpha), e.phiNormal !== void 0 && (this.phiNormal.value = e.phiNormal), e.phiDepth !== void 0 && (this.phiDepth.value = e.phiDepth), e.maxHistory !== void 0 && (this.maxHistory.value = e.maxHistory), e.spatialRadius !== void 0 && (this.spatialRadius.value = e.spatialRadius), e.spatialWeight !== void 0 && (this.spatialWeight.value = e.spatialWeight);
@@ -10025,7 +10025,7 @@ var Fc = class extends Sn {
 		this._transparentBackground.value = +!!e;
 	}
 	dispose() {
-		this.displayMaterial?.dispose(), this.displayQuad = null;
+		this._displayTexNode?.dispose(), this.displayMaterial?.dispose(), this.displayQuad = null;
 	}
 }, Lc = class {
 	constructor() {
@@ -10207,7 +10207,7 @@ var Fc = class extends Sn {
 		let t = this.stages.findIndex((t) => t.name === e);
 		if (t > -1) {
 			let e = this.stages[t];
-			return this.stages.splice(t, 1), e.dispose && e.dispose(), !0;
+			return this.stages.splice(t, 1), e.dispose && e.dispose(), this.stats.timings.delete(e.name), !0;
 		}
 		return !1;
 	}
@@ -11934,14 +11934,14 @@ var dl = new Float32Array(3), fl = {
 		}, this._gpuInputBufferSize = {
 			width: 0,
 			height: 0
-		}, this._gpuInputPadBuffer = null, this._gpuInputPaddedRowBytes = 0, this._alphaReadbackBuffer = null, this._cachedAlpha = null, this._cachedAlphaWidth = 0, this.config = {
+		}, this._gpuInputPadBuffer = null, this._gpuInputPaddedRowBytes = 0, this._alphaReadbackBuffer = null, this._alphaReadbackMapped = !1, this._cachedAlpha = null, this._cachedAlphaWidth = 0, this.config = {
 			...fl.DEFAULT_OPTIONS,
 			...i
 		}, this.enabled = this.config.enableOIDN, this.quality = this.config.oidnQuality, this.debugGbufferMaps = this.config.debugGbufferMaps, this.tileSize = this.config.tileSize, this.state = {
 			isDenoising: !1,
 			isLoading: !1,
 			abortController: null
-		}, this.currentTZAUrl = null, this.unet = null, this.debugHelpers = null, this._lastAlbedoTexture = null, this._lastNormalTexture = null, this._initialize().catch((e) => {
+		}, this._pendingStagingBuffers = /* @__PURE__ */ new Set(), this.currentTZAUrl = null, this.unet = null, this.debugHelpers = null, this._lastAlbedoTexture = null, this._lastNormalTexture = null, this._initialize().catch((e) => {
 			console.error("Failed to initialize OIDNDenoiser:", e), this.dispatchEvent({
 				type: "error",
 				error: e
@@ -12129,7 +12129,10 @@ var dl = new Float32Array(3), fl = {
 		}), this._gpuInputPaddedRowBytes = c);
 	}
 	_destroyGPUInputBuffers() {
-		this._gpuInputBuffers.color?.destroy(), this._gpuInputBuffers.albedo?.destroy(), this._gpuInputBuffers.normal?.destroy(), this._gpuInputPadBuffer?.destroy(), this._alphaReadbackBuffer?.destroy(), this._gpuInputBuffers = {
+		if (this._gpuInputBuffers.color?.destroy(), this._gpuInputBuffers.albedo?.destroy(), this._gpuInputBuffers.normal?.destroy(), this._gpuInputPadBuffer?.destroy(), this._alphaReadbackMapped && this._alphaReadbackBuffer) try {
+			this._alphaReadbackBuffer.unmap();
+		} catch {}
+		this._alphaReadbackBuffer?.destroy(), this._alphaReadbackMapped = !1, this._gpuInputBuffers = {
 			color: null,
 			albedo: null,
 			normal: null
@@ -12146,10 +12149,16 @@ var dl = new Float32Array(3), fl = {
 			usage: GPUBufferUsage.MAP_READ | GPUBufferUsage.COPY_DST
 		}));
 		let i = this._alphaReadbackBuffer, a = e.createCommandEncoder();
-		a.copyBufferToBuffer(this._gpuInputBuffers.color, 0, i, 0, r), e.queue.submit([a.finish()]), await i.mapAsync(GPUMapMode.READ);
+		a.copyBufferToBuffer(this._gpuInputBuffers.color, 0, i, 0, r), e.queue.submit([a.finish()]), this._alphaReadbackMapped = !0;
+		try {
+			await i.mapAsync(GPUMapMode.READ);
+		} catch {
+			this._alphaReadbackMapped = !1;
+			return;
+		}
 		let o = new Float32Array(i.getMappedRange()), s = t * n, c = new Uint8Array(s);
 		for (let e = 0; e < s; e++) c[e] = Math.min(Math.max(o[e * 4 + 3] * 255, 0), 255) | 0;
-		i.unmap(), this._cachedAlpha = c, this._cachedAlphaWidth = t;
+		i.unmap(), this._alphaReadbackMapped = !1, this._cachedAlpha = c, this._cachedAlphaWidth = t;
 	}
 	_executeWithAbortGPU(e) {
 		return new Promise((t, n) => {
@@ -12183,7 +12192,9 @@ var dl = new Float32Array(3), fl = {
 					let c = o * 16, l = o * s * 16, u = r.createBuffer({
 						size: l,
 						usage: GPUBufferUsage.MAP_READ | GPUBufferUsage.COPY_DST
-					}), d = r.createCommandEncoder();
+					});
+					this._pendingStagingBuffers.add(u);
+					let d = r.createCommandEncoder();
 					for (let t = 0; t < s; t++) {
 						let r = ((n.y + t) * i + n.x) * 16, a = t * c;
 						d.copyBufferToBuffer(e.data, r, u, a, c);
@@ -12197,7 +12208,9 @@ var dl = new Float32Array(3), fl = {
 								t.data[s + 3] = c[(n.y + r) * l + n.x + e];
 							} else t.data[s + 3] = 255;
 						}
-						u.unmap(), u.destroy(), this.ctx.putImageData(t, n.x, n.y);
+						u.unmap(), u.destroy(), this._pendingStagingBuffers.delete(u), this.ctx.putImageData(t, n.x, n.y);
+					}).catch(() => {
+						u.destroy(), this._pendingStagingBuffers.delete(u);
 					});
 				}
 			});
@@ -12213,17 +12226,22 @@ var dl = new Float32Array(3), fl = {
 			label: "oidn-output-staging",
 			size: i,
 			usage: GPUBufferUsage.MAP_READ | GPUBufferUsage.COPY_DST
-		}), o = r.createCommandEncoder({ label: "oidn-readback" });
-		o.copyBufferToBuffer(e, 0, a, 0, i), r.queue.submit([o.finish()]), await a.mapAsync(GPUMapMode.READ);
-		let s = new Float32Array(a.getMappedRange()), c = new ImageData(t, n), l = this.getExposure(), u = this.getSaturation(), d = rl.get(this.getToneMapping()) || rl.get(Ee), f = this._cachedAlpha;
-		for (let e = 0, t = s.length; e < t; e += 4) {
-			let t = s[e] * l, n = s[e + 1] * l, r = s[e + 2] * l;
-			u !== 1 && (dl[0] = t, dl[1] = n, dl[2] = r, cl(dl, u), t = dl[0], n = dl[1], r = dl[2]), d(t, n, r, 1, dl), c.data[e] = dl[0] ** il * 255 | 0, c.data[e + 1] = dl[1] ** il * 255 | 0, c.data[e + 2] = dl[2] ** il * 255 | 0, c.data[e + 3] = f ? f[e >> 2] : 255;
+		});
+		try {
+			let o = r.createCommandEncoder({ label: "oidn-readback" });
+			o.copyBufferToBuffer(e, 0, a, 0, i), r.queue.submit([o.finish()]), await a.mapAsync(GPUMapMode.READ);
+			let s = new Float32Array(a.getMappedRange()), c = new ImageData(t, n), l = this.getExposure(), u = this.getSaturation(), d = rl.get(this.getToneMapping()) || rl.get(Ee), f = this._cachedAlpha;
+			for (let e = 0, t = s.length; e < t; e += 4) {
+				let t = s[e] * l, n = s[e + 1] * l, r = s[e + 2] * l;
+				u !== 1 && (dl[0] = t, dl[1] = n, dl[2] = r, cl(dl, u), t = dl[0], n = dl[1], r = dl[2]), d(t, n, r, 1, dl), c.data[e] = dl[0] ** il * 255 | 0, c.data[e + 1] = dl[1] ** il * 255 | 0, c.data[e + 2] = dl[2] ** il * 255 | 0, c.data[e + 3] = f ? f[e >> 2] : 255;
+			}
+			a.unmap(), this.ctx.putImageData(c, 0, 0);
+		} finally {
+			a.destroy();
 		}
-		a.unmap(), a.destroy(), this.ctx.putImageData(c, 0, 0);
 	}
 	abort() {
-		!this.enabled || !this.state.isDenoising || (this.state.abortController?.abort(), this.input.style.opacity = "1", this.state.isDenoising = !1, this.dispatchEvent({ type: "end" }), console.log("Denoising aborted"));
+		!this.enabled || !this.state.isDenoising || (this.state.abortController?.abort(), this._destroyPendingStagingBuffers(), this.input.style.opacity = "1", this.state.isDenoising = !1, this.dispatchEvent({ type: "end" }), console.log("Denoising aborted"));
 	}
 	setSize(e, t) {
 		if (e <= 0 || t <= 0) throw Error(`Invalid dimensions: ${e}x${t}`);
@@ -12260,8 +12278,12 @@ var dl = new Float32Array(3), fl = {
 		}
 		this.debugHelpers.albedo.update(), this.debugHelpers.normal.update();
 	}
+	_destroyPendingStagingBuffers() {
+		for (let e of this._pendingStagingBuffers) e.destroy();
+		this._pendingStagingBuffers.clear();
+	}
 	dispose() {
-		this.abort(), this.unet?.dispose(), this._destroyGPUInputBuffers(), this.debugHelpers &&= (this.debugHelpers.albedo?.dispose(), this.debugHelpers.normal?.dispose(), null), this._lastAlbedoTexture = null, this._lastNormalTexture = null, this.output?.parentNode && this.output.remove(), this.unet = null, this.ctx = null, this.state.abortController = null, this.removeAllListeners?.(), console.log("OIDNDenoiser disposed");
+		this.abort(), this._destroyPendingStagingBuffers(), this.unet?.dispose(), this._destroyGPUInputBuffers(), this.debugHelpers &&= (this.debugHelpers.albedo?.dispose(), this.debugHelpers.normal?.dispose(), null), this._lastAlbedoTexture = null, this._lastNormalTexture = null, this.output?.parentNode && this.output.remove(), this.unet = null, this.ctx = null, this.state.abortController = null, this.removeAllListeners?.(), console.log("OIDNDenoiser disposed");
 	}
 }, ml = "" + new URL("assets/AIUpscalerWorker-D58dcMrY.js", import.meta.url).href, hl = "https://huggingface.co/notaneimu/onnx-image-models/resolve/main/", gl = {
 	QUALITY_PRESETS: {
@@ -12294,7 +12316,7 @@ var dl = new Float32Array(3), fl = {
 			isUpscaling: !1,
 			isLoading: !1,
 			abortController: null
-		}, this._worker = null, this._currentModelUrl = null, this._tileId = 0, this._upscaledAlpha = null, this._upscaledAlphaWidth = 0, this._backupCanvas = null, this._baseWidth = e.width, this._baseHeight = e.height, this._hdrStagingBuffer = null, this._hdrStagingWidth = 0, this._hdrStagingHeight = 0;
+		}, this._worker = null, this._currentModelUrl = null, this._tileId = 0, this._pendingWorkerHandlers = /* @__PURE__ */ new Set(), this._upscaledAlpha = null, this._upscaledAlphaWidth = 0, this._backupCanvas = null, this._baseWidth = e.width, this._baseHeight = e.height, this._hdrStagingBuffer = null, this._hdrStagingWidth = 0, this._hdrStagingHeight = 0;
 	}
 	async _ensureSession() {
 		let e = gl.QUALITY_PRESETS[this.quality];
@@ -12370,11 +12392,11 @@ var dl = new Float32Array(3), fl = {
 				height: this._baseHeight
 			})), !1;
 		} finally {
-			this._capturedSource = null, this._upscaledAlpha = null, this.state.isUpscaling && (this.state.isUpscaling = !1, this.state.abortController = null, this.dispatchEvent({ type: "end" }));
+			this._capturedSource = null, this._upscaledAlpha = null, this._backupCanvas = null, this.state.isUpscaling && (this.state.isUpscaling = !1, this.state.abortController = null, this.dispatchEvent({ type: "end" }));
 		}
 	}
 	abort() {
-		this.state.isUpscaling && (this.state.abortController?.abort(), this.input.style.opacity = "1", this._restoreBackup(), this.dispatchEvent({
+		this.state.isUpscaling && (this.state.abortController?.abort(), this._cleanupPendingWorkerHandlers(), this.input.style.opacity = "1", this._restoreBackup(), this.dispatchEvent({
 			type: "resolution_changed",
 			width: this._baseWidth,
 			height: this._baseHeight
@@ -12470,9 +12492,9 @@ var dl = new Float32Array(3), fl = {
 		let r = ++this._tileId;
 		return new Promise((i, a) => {
 			let o = (e) => {
-				e.data.id === r && (this._worker.removeEventListener("message", o), e.data.type === "inferred" ? i(e.data.outputData) : e.data.type === "error" && a(Error(e.data.message)));
+				e.data.id === r && (this._worker.removeEventListener("message", o), this._pendingWorkerHandlers.delete(o), e.data.type === "inferred" ? i(e.data.outputData) : e.data.type === "error" && a(Error(e.data.message)));
 			};
-			this._worker.addEventListener("message", o), this._worker.postMessage({
+			this._pendingWorkerHandlers.add(o), this._worker.addEventListener("message", o), this._worker.postMessage({
 				type: "infer",
 				tileData: e,
 				width: t,
@@ -12550,12 +12572,16 @@ var dl = new Float32Array(3), fl = {
 	setBaseSize(e, t) {
 		this._baseWidth = e, this._baseHeight = t;
 	}
+	_cleanupPendingWorkerHandlers() {
+		if (this._worker) for (let e of this._pendingWorkerHandlers) this._worker.removeEventListener("message", e);
+		this._pendingWorkerHandlers.clear();
+	}
 	async dispose() {
-		this.abort(), this._worker &&= (this._worker.postMessage({ type: "dispose" }), this._worker.terminate(), null), this._currentModelUrl = null, this._backupCanvas = null, this._upscaledAlpha = null, this.state.abortController = null, this._hdrStagingBuffer?.destroy(), this._hdrStagingBuffer = null, this._hdrStagingWidth = 0, this._hdrStagingHeight = 0, console.log("AIUpscaler disposed");
+		this.abort(), this._cleanupPendingWorkerHandlers(), this._worker &&= (this._worker.postMessage({ type: "dispose" }), this._worker.terminate(), null), this._currentModelUrl = null, this._backupCanvas = null, this._upscaledAlpha = null, this.state.abortController = null, this._hdrStagingBuffer?.destroy(), this._hdrStagingBuffer = null, this._hdrStagingWidth = 0, this._hdrStagingHeight = 0, console.log("AIUpscaler disposed");
 	}
 }, vl = class extends We {
 	constructor({ renderer: e, mainCanvas: t, scene: n, camera: r, stages: i, pipeline: a, getExposure: o, getSaturation: s, getTransparentBg: c }) {
-		super(), this.renderer = e, this.mainCanvas = t, this.denoiserCanvas = this._createDenoiserCanvas(t), this.scene = n, this.camera = r, this.pipeline = a, this._stages = i, this._getExposure = o, this._getSaturation = s, this._getTransparentBg = c, this.denoiser = null, this.upscaler = null, this._lastRenderWidth = 0, this._lastRenderHeight = 0;
+		super(), this.renderer = e, this.mainCanvas = t, this.denoiserCanvas = this._createDenoiserCanvas(t), this.scene = n, this.camera = r, this.pipeline = a, this._stages = i, this._getExposure = o, this._getSaturation = s, this._getTransparentBg = c, this.denoiser = null, this.upscaler = null, this._lastRenderWidth = 0, this._lastRenderHeight = 0, this._pendingStartUpscaler = null, this._denoiserStartHandler = null, this._denoiserEndHandler = null, this._upscalerResChangedHandler = null, this._upscalerStartHandler = null, this._upscalerProgressHandler = null, this._upscalerEndHandler = null;
 	}
 	_createDenoiserCanvas(e) {
 		let t = e.parentNode;
@@ -12594,7 +12620,7 @@ var dl = new Float32Array(3), fl = {
 			getSaturation: () => this._getSaturation(),
 			getTransparentBackground: () => this._getTransparentBg(),
 			getMRTRenderTarget: () => e?.storageTextures?.readTarget ?? null
-		}), this.denoiser.enabled = X.enableOIDN, this.denoiser.addEventListener("start", () => this.dispatchEvent({ type: Y.DENOISING_START })), this.denoiser.addEventListener("end", () => this.dispatchEvent({ type: Y.DENOISING_END }));
+		}), this.denoiser.enabled = X.enableOIDN, this._denoiserStartHandler = () => this.dispatchEvent({ type: Y.DENOISING_START }), this._denoiserEndHandler = () => this.dispatchEvent({ type: Y.DENOISING_END }), this.denoiser.addEventListener("start", this._denoiserStartHandler), this.denoiser.addEventListener("end", this._denoiserEndHandler);
 	}
 	setupUpscaler() {
 		if (!this.denoiserCanvas) return;
@@ -12611,14 +12637,14 @@ var dl = new Float32Array(3), fl = {
 			getExposure: () => this._getEffectiveExposure(),
 			getToneMapping: () => this._getToneMapping(),
 			getSaturation: () => this._getSaturation()
-		}), this.upscaler.enabled = X.enableUpscaler || !1, this.upscaler.addEventListener("resolution_changed", (e) => this.dispatchEvent({
+		}), this.upscaler.enabled = X.enableUpscaler || !1, this._upscalerResChangedHandler = (e) => this.dispatchEvent({
 			type: "resolution_changed",
 			width: e.width,
 			height: e.height
-		})), this.upscaler.addEventListener("start", () => this.dispatchEvent({ type: Y.UPSCALING_START })), this.upscaler.addEventListener("progress", (e) => this.dispatchEvent({
+		}), this._upscalerStartHandler = () => this.dispatchEvent({ type: Y.UPSCALING_START }), this._upscalerProgressHandler = (e) => this.dispatchEvent({
 			type: Y.UPSCALING_PROGRESS,
 			progress: e.progress
-		})), this.upscaler.addEventListener("end", () => this.dispatchEvent({ type: Y.UPSCALING_END }));
+		}), this._upscalerEndHandler = () => this.dispatchEvent({ type: Y.UPSCALING_END }), this.upscaler.addEventListener("resolution_changed", this._upscalerResChangedHandler), this.upscaler.addEventListener("start", this._upscalerStartHandler), this.upscaler.addEventListener("progress", this._upscalerProgressHandler), this.upscaler.addEventListener("end", this._upscalerEndHandler);
 	}
 	setDenoiserStrategy(e, t) {
 		let n = this._stages;
@@ -12649,18 +12675,21 @@ var dl = new Float32Array(3), fl = {
 		let t = this._stages;
 		t.adaptiveSampling && (t.adaptiveSampling.enabled = e, t.adaptiveSampling.toggleHelper(!1)), e ? t.variance && (t.variance.enabled = !0) : t.asvgf?.enabled || t.variance && (t.variance.enabled = !1), !e && this.pipeline?.context && !t.asvgf?.enabled && this.pipeline.context.removeTexture("variance:output");
 	}
+	_cleanupCompletionListener() {
+		this._pendingStartUpscaler && this.denoiser && this.denoiser.removeEventListener("end", this._pendingStartUpscaler), this._pendingStartUpscaler = null;
+	}
 	onRenderComplete({ isStillComplete: e, context: t }) {
-		(this.denoiser?.enabled || this.upscaler?.enabled) && this.denoiserCanvas && (this.denoiserCanvas.style.display = "block");
+		this._cleanupCompletionListener(), (this.denoiser?.enabled || this.upscaler?.enabled) && this.denoiserCanvas && (this.denoiserCanvas.style.display = "block");
 		let n = () => {
-			e() && this.upscaler?.enabled && this.upscaler.start();
+			this._pendingStartUpscaler = null, e() && this.upscaler?.enabled && this.upscaler.start();
 		};
-		this.denoiser?.enabled ? (this.denoiser.addEventListener("end", n, { once: !0 }), this.denoiser.start()) : (this.upscaler?.enabled && this._stages.display && t && this._stages.display.render(t), n());
+		this.denoiser?.enabled ? (this._pendingStartUpscaler = n, this.denoiser.addEventListener("end", n, { once: !0 }), this.denoiser.start()) : (this.upscaler?.enabled && this._stages.display && t && this._stages.display.render(t), n());
 	}
 	abort(e) {
-		e && (e.style.opacity = "1"), this.upscaler && this.upscaler.abort(), this.denoiser && (this.denoiser.enabled && this.denoiser.abort(), this.denoiser.output && (this.denoiser.output.style.display = "none"));
+		this._cleanupCompletionListener(), e && (e.style.opacity = "1"), this.upscaler && this.upscaler.abort(), this.denoiser && (this.denoiser.enabled && this.denoiser.abort(), this.denoiser.output && (this.denoiser.output.style.display = "none"));
 	}
 	dispose() {
-		this.denoiser &&= (this.denoiser.dispose(), null), this.upscaler &&= (this.upscaler.dispose(), null), this.denoiserCanvas?.parentNode && (this.denoiserCanvas.parentNode.removeChild(this.denoiserCanvas), this.denoiserCanvas = null);
+		this._cleanupCompletionListener(), this.denoiser &&= (this._denoiserStartHandler && this.denoiser.removeEventListener("start", this._denoiserStartHandler), this._denoiserEndHandler && this.denoiser.removeEventListener("end", this._denoiserEndHandler), this.denoiser.dispose(), null), this.upscaler &&= (this._upscalerResChangedHandler && this.upscaler.removeEventListener("resolution_changed", this._upscalerResChangedHandler), this._upscalerStartHandler && this.upscaler.removeEventListener("start", this._upscalerStartHandler), this._upscalerProgressHandler && this.upscaler.removeEventListener("progress", this._upscalerProgressHandler), this._upscalerEndHandler && this.upscaler.removeEventListener("end", this._upscalerEndHandler), this.upscaler.dispose(), null), this._denoiserStartHandler = null, this._denoiserEndHandler = null, this._upscalerResChangedHandler = null, this._upscalerStartHandler = null, this._upscalerProgressHandler = null, this._upscalerEndHandler = null, this.denoiserCanvas?.parentNode && (this.denoiserCanvas.parentNode.removeChild(this.denoiserCanvas), this.denoiserCanvas = null);
 	}
 	setOverlayManager(e) {
 		this._overlayManager = e;