rayzee 4.8.11 → 4.8.12

package/dist/rayzee.es.js

@@ -11891,21 +11891,45 @@ var Wc = new Float32Array(3), Gc = {
 		let r = this.gpuDevice;
 		if (!r) return console.warn("OIDNDenoiser: gpuDevice not available"), !1;
 		this._ensureGPUInputBuffers(e, t);
-		let i = r.createCommandEncoder({ label: "oidn-tex-to-buf" }), a = e * 16, o = (n, r) => i.copyTextureToBuffer({
-			texture: n,
-			mipLevel: 0
-		}, {
-			buffer: r,
-			offset: 0,
-			bytesPerRow: a,
-			rowsPerImage: t
-		}, {
-			width: e,
-			height: t,
-			depthOrArrayLayers: 1
-		});
-		o(n.color, this._gpuInputBuffers.color), o(n.albedo, this._gpuInputBuffers.albedo), o(n.normal, this._gpuInputBuffers.normal), r.queue.submit([i.finish()]), this.getTransparentBackground() ? await this._cacheInputAlpha(r, e, t) : this._cachedAlpha = null, this.ctx.drawImage(this.input, 0, 0, e, t);
-		let s = {
+		let i = r.createCommandEncoder({ label: "oidn-tex-to-buf" }), a = e * 16, o = Math.ceil(a / 256) * 256, s = o !== a, c = [], l = (n, l) => {
+			if (!s) i.copyTextureToBuffer({
+				texture: n,
+				mipLevel: 0
+			}, {
+				buffer: l,
+				offset: 0,
+				bytesPerRow: a,
+				rowsPerImage: t
+			}, {
+				width: e,
+				height: t,
+				depthOrArrayLayers: 1
+			});
+			else {
+				let s = r.createBuffer({
+					size: o * t,
+					usage: GPUBufferUsage.COPY_DST | GPUBufferUsage.COPY_SRC
+				});
+				c.push(s), i.copyTextureToBuffer({
+					texture: n,
+					mipLevel: 0
+				}, {
+					buffer: s,
+					offset: 0,
+					bytesPerRow: o,
+					rowsPerImage: t
+				}, {
+					width: e,
+					height: t,
+					depthOrArrayLayers: 1
+				});
+				for (let e = 0; e < t; e++) i.copyBufferToBuffer(s, e * o, l, e * a, a);
+			}
+		};
+		l(n.color, this._gpuInputBuffers.color), l(n.albedo, this._gpuInputBuffers.albedo), l(n.normal, this._gpuInputBuffers.normal), r.queue.submit([i.finish()]);
+		for (let e of c) e.destroy();
+		this.getTransparentBackground() ? await this._cacheInputAlpha(r, e, t) : this._cachedAlpha = null, this.ctx.drawImage(this.input, 0, 0, e, t);
+		let u = {
 			color: {
 				data: this._gpuInputBuffers.color,
 				width: e,
@@ -11922,13 +11946,13 @@ var Wc = new Float32Array(3), Gc = {
 				height: t
 			}
 		};
-		return this._executeWithAbortGPU(s);
+		return this._executeWithAbortGPU(u);
 	}
 	_ensureGPUInputBuffers(e, t) {
 		let { width: n, height: r } = this._gpuInputBufferSize;
 		if (n === e && r === t && this._gpuInputBuffers.color) return;
 		this._destroyGPUInputBuffers();
-		let i = this.gpuDevice, a = e * t * 4 * 4, o = GPUBufferUsage.COPY_DST | GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC;
+		let i = this.gpuDevice, a = e * t * 16, o = GPUBufferUsage.COPY_DST | GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC;
 		this._gpuInputBuffers.color = i.createBuffer({
 			label: "oidn-in-color",
 			size: a,
@@ -11957,7 +11981,7 @@ var Wc = new Float32Array(3), Gc = {
 		};
 	}
 	async _cacheInputAlpha(e, t, n) {
-		let r = t * n * 4 * 4, i = e.createBuffer({
+		let r = t * n * 16, i = e.createBuffer({
 			label: "oidn-alpha-staging",
 			size: r,
 			usage: GPUBufferUsage.MAP_READ | GPUBufferUsage.COPY_DST
@@ -11994,24 +12018,26 @@ var Wc = new Float32Array(3), Gc = {
 						imageWidth: e.width,
 						imageHeight: e.height
 					});
-					let r = this.gpuDevice, i = e.width, a = n.width * 16, o = n.width * n.height * 16, s = r.createBuffer({
-						size: o,
+					let r = this.gpuDevice, i = e.width, a = e.height, o = Math.min(n.width, i - n.x), s = Math.min(n.height, a - n.y);
+					if (o <= 0 || s <= 0) return;
+					let l = o * 16, u = o * s * 16, d = r.createBuffer({
+						size: u,
 						usage: GPUBufferUsage.MAP_READ | GPUBufferUsage.COPY_DST
-					}), l = r.createCommandEncoder();
-					for (let t = 0; t < n.height; t++) {
-						let r = ((n.y + t) * i + n.x) * 16, o = t * a;
-						l.copyBufferToBuffer(e.data, r, s, o, a);
+					}), f = r.createCommandEncoder();
+					for (let t = 0; t < s; t++) {
+						let r = ((n.y + t) * i + n.x) * 16, a = t * l;
+						f.copyBufferToBuffer(e.data, r, d, a, l);
 					}
-					r.queue.submit([l.finish()]), s.mapAsync(GPUMapMode.READ).then(() => {
-						let e = new Float32Array(s.getMappedRange()), t = new ImageData(n.width, n.height), r = this.getExposure(), i = this.getSaturation(), a = Ic.get(this.getToneMapping()) || Ic.get(c), o = this._cachedAlpha, l = this._cachedAlphaWidth;
+					r.queue.submit([f.finish()]), d.mapAsync(GPUMapMode.READ).then(() => {
+						let e = new Float32Array(d.getMappedRange()), t = new ImageData(o, s), r = this.getExposure(), i = this.getSaturation(), a = Ic.get(this.getToneMapping()) || Ic.get(c), l = this._cachedAlpha, u = this._cachedAlphaWidth;
 						for (let s = 0, c = e.length; s < c; s += 4) {
-							let c = e[s] * r, u = e[s + 1] * r, d = e[s + 2] * r;
-							if (i !== 1 && (Wc[0] = c, Wc[1] = u, Wc[2] = d, Vc(Wc, i), c = Wc[0], u = Wc[1], d = Wc[2]), a(c, u, d, 1, Wc), t.data[s] = Wc[0] ** Lc * 255 | 0, t.data[s + 1] = Wc[1] ** Lc * 255 | 0, t.data[s + 2] = Wc[2] ** Lc * 255 | 0, o) {
-								let e = (s >> 2) % n.width, r = (s >> 2) / n.width | 0;
-								t.data[s + 3] = o[(n.y + r) * l + n.x + e];
+							let c = e[s] * r, d = e[s + 1] * r, f = e[s + 2] * r;
+							if (i !== 1 && (Wc[0] = c, Wc[1] = d, Wc[2] = f, Vc(Wc, i), c = Wc[0], d = Wc[1], f = Wc[2]), a(c, d, f, 1, Wc), t.data[s] = Wc[0] ** Lc * 255 | 0, t.data[s + 1] = Wc[1] ** Lc * 255 | 0, t.data[s + 2] = Wc[2] ** Lc * 255 | 0, l) {
+								let e = (s >> 2) % o, r = (s >> 2) / o | 0;
+								t.data[s + 3] = l[(n.y + r) * u + n.x + e];
 							} else t.data[s + 3] = 255;
 						}
-						s.unmap(), s.destroy(), this.ctx.putImageData(t, n.x, n.y);
+						d.unmap(), d.destroy(), this.ctx.putImageData(t, n.x, n.y);
 					});
 				}
 			});