rayzee 5.8.0 → 5.8.2

package/dist/rayzee.es.js

@@ -9177,9 +9177,9 @@ var bc = class extends wn {
 		super("ASVGF", {
 			...t,
 			executionMode: Cn.PER_CYCLE
-		}), this.renderer = e, this.debugContainer = t.debugContainer || null, this.temporalAlpha = H(t.temporalAlpha ?? .1), this.gradientScale = H(t.gradientScale ?? 2), this.gradientMin = H(t.gradientMin ?? .01), this.gradientMax = H(t.gradientMax ?? .5), this.phiColor = H(t.phiColor ?? 10), this.phiNormal = H(t.phiNormal ?? 128), this.phiDepth = H(t.phiDepth ?? 1), this.maxAccumFrames = H(t.maxAccumFrames ?? 32), this.varianceClip = H(t.varianceClip ?? 1), this.resW = H(t.width || 1), this.resH = H(t.height || 1), this.temporalEnabled = !0, this.temporalEnabledU = H(1), this._colorTexNode = new c(), this._normalDepthTexNode = new c(), this._motionTexNode = new c(), this._readTemporalTexNode = new c(), this._readPrevNDTexNode = new c(), this._gradientReadTexNode = new c();
+		}), this.renderer = e, this.debugContainer = t.debugContainer || null, this.temporalAlpha = H(t.temporalAlpha ?? .1), this.phiColor = H(t.phiColor ?? 10), this.maxAccumFrames = H(t.maxAccumFrames ?? 32), this.varianceClip = H(t.varianceClip ?? 1), this.resW = H(t.width || 1), this.resH = H(t.height || 1), this.temporalEnabled = !0, this.temporalEnabledU = H(1), this._colorTexNode = new c(), this._motionTexNode = new c(), this._readTemporalTexNode = new c();
 		let n = t.width || 1, r = t.height || 1;
-		this._temporalTexA = new s(n, r), this._temporalTexA.type = Ye, this._temporalTexA.format = yt, this._temporalTexA.minFilter = q, this._temporalTexA.magFilter = q, this._temporalTexB = new s(n, r), this._temporalTexB.type = Ye, this._temporalTexB.format = yt, this._temporalTexB.minFilter = q, this._temporalTexB.magFilter = q, this._prevNDTexA = new s(n, r), this._prevNDTexA.type = Ye, this._prevNDTexA.format = yt, this._prevNDTexA.minFilter = q, this._prevNDTexA.magFilter = q, this._prevNDTexB = new s(n, r), this._prevNDTexB.type = Ye, this._prevNDTexB.format = yt, this._prevNDTexB.minFilter = q, this._prevNDTexB.magFilter = q, this._gradientStorageTex = new s(n, r), this._gradientStorageTex.type = Ye, this._gradientStorageTex.format = yt, this._gradientStorageTex.minFilter = q, this._gradientStorageTex.magFilter = q, this.currentMoments = 0, this._compiled = !1, this._dispatchX = Math.ceil(n / 8), this._dispatchY = Math.ceil(r / 8), this._buildGradientCompute(), this._buildTemporalCompute(), this.showHeatmap = !1, this.debugMode = H(0, "int"), this._heatmapStorageTex = new s(n, r), this._heatmapStorageTex.type = qe, this._heatmapStorageTex.format = yt, this._heatmapStorageTex.minFilter = ct, this._heatmapStorageTex.magFilter = ct, this.heatmapTarget = new i(n, r, {
+		this._temporalTexA = new s(n, r), this._temporalTexA.type = Ye, this._temporalTexA.format = yt, this._temporalTexA.minFilter = q, this._temporalTexA.magFilter = q, this._temporalTexB = new s(n, r), this._temporalTexB.type = Ye, this._temporalTexB.format = yt, this._temporalTexB.minFilter = q, this._temporalTexB.magFilter = q, this._gradientStorageTex = new s(n, r), this._gradientStorageTex.type = Ye, this._gradientStorageTex.format = yt, this._gradientStorageTex.minFilter = q, this._gradientStorageTex.magFilter = q, this.currentMoments = 0, this._compiled = !1, this._dispatchX = Math.ceil(n / 8), this._dispatchY = Math.ceil(r / 8), this._buildGradientCompute(), this._buildTemporalCompute(), this.showHeatmap = !1, this.debugMode = H(0, "int"), this._heatmapStorageTex = new s(n, r), this._heatmapStorageTex.type = qe, this._heatmapStorageTex.format = yt, this._heatmapStorageTex.minFilter = ct, this._heatmapStorageTex.magFilter = ct, this.heatmapTarget = new i(n, r, {
 			type: qe,
 			format: yt,
 			minFilter: ct,
@@ -9226,35 +9226,29 @@ var bc = class extends wn {
 		]);
 	}
 	_buildTemporalCompute() {
-		this._temporalNodeA = this._buildTemporalForDirection(this._temporalTexA, this._prevNDTexA), this._temporalNodeB = this._buildTemporalForDirection(this._temporalTexB, this._prevNDTexB);
-	}
-	_buildTemporalForDirection(e, t) {
-		let n = this._colorTexNode, r = this._normalDepthTexNode, i = this._motionTexNode, a = this._readTemporalTexNode, o = this._readPrevNDTexNode;
-		this._gradientReadTexNode;
-		let s = this.temporalAlpha;
-		this.gradientScale, this.gradientMin, this.gradientMax;
-		let c = this.phiNormal, l = this.phiDepth, u = this.maxAccumFrames, d = this.varianceClip, f = this.temporalEnabledU, h = this.resW, g = this.resH;
+		this._temporalNodeA = this._buildTemporalForDirection(this._temporalTexA), this._temporalNodeB = this._buildTemporalForDirection(this._temporalTexB);
+	}
+	_buildTemporalForDirection(e) {
+		let t = this._colorTexNode, n = this._motionTexNode, r = this._readTemporalTexNode, i = this.temporalAlpha, a = this.maxAccumFrames, o = this.varianceClip, s = this.temporalEnabledU, c = this.resW, l = this.resH;
 		return p(() => {
-			let p = N(De.x).mul(8).add(N(F.x)), _ = N(De.y).mul(8).add(N(F.y));
-			m(p.lessThan(N(h)).and(_.lessThan(N(g))), () => {
-				let v = P(p, _), y = B(n, v).xyz, b = B(r, v), x = G(y, 1).toVar();
-				m(f.greaterThan(.5), () => {
-					let e = B(i, v), t = e.w.greaterThan(.5), r = M(p).sub(e.x.mul(h)), f = M(_).sub(e.y.mul(g)), S = r.greaterThanEqual(0).and(r.lessThan(M(h))).and(f.greaterThanEqual(0)).and(f.lessThan(M(g)));
-					m(t.and(S), () => {
-						let e = P(N(r).clamp(N(0), N(h).sub(1)), N(f).clamp(N(0), N(g).sub(1))), t = b.xyz.mul(2).sub(1), i = B(o, e);
-						Di(t, i.xyz.mul(2).sub(1), b.w, i.w, c, l);
-						let m = B(a, e), v = m.xyz, S = m.w, C = S.div(u).clamp(0, 1), w = W(1e10).toVar(), T = W(-1e10).toVar(), E = W(0).toVar();
-						for (let e = -1; e <= 1; e++) for (let t = -1; t <= 1; t++) {
-							let r = B(n, P(p.add(t).clamp(N(0), N(h).sub(1)), _.add(e).clamp(N(0), N(g).sub(1)))).xyz;
-							w.assign(oe(w, r)), T.assign(I(T, r)), E.addAssign(r);
+			let u = N(De.x).mul(8).add(N(F.x)), d = N(De.y).mul(8).add(N(F.y));
+			m(u.lessThan(N(c)).and(d.lessThan(N(l))), () => {
+				let f = P(u, d), p = B(t, f).xyz, h = G(p, 1).toVar();
+				m(s.greaterThan(.5), () => {
+					let e = B(n, f), s = e.w.greaterThan(.5), g = M(u).sub(e.x.mul(c)), _ = M(d).sub(e.y.mul(l)), v = g.greaterThanEqual(0).and(g.lessThan(M(c))).and(_.greaterThanEqual(0)).and(_.lessThan(M(l)));
+					m(s.and(v), () => {
+						let e = B(r, P(N(g).clamp(N(0), N(c).sub(1)), N(_).clamp(N(0), N(l).sub(1)))), n = e.xyz, s = e.w, f = W(0).toVar(), m = W(0).toVar();
+						for (let e = -1; e <= 1; e++) for (let n = -1; n <= 1; n++) {
+							let r = B(t, P(u.add(n).clamp(N(0), N(c).sub(1)), d.add(e).clamp(N(0), N(l).sub(1)))).xyz;
+							f.addAssign(r), m.addAssign(r.mul(r));
 						}
-						E.divAssign(9);
-						let D = M(1).add(C.mul(3)), O = T.sub(w).mul(d).mul(D), k = w.sub(O), A = T.add(O), j = L(v.clamp(k, A), y, I(M(1).div(S.add(1)), s)), ee = oe(S.add(1), u);
-						x.assign(G(j, ee));
+						f.divAssign(9), m.divAssign(9);
+						let v = A(I(m.sub(f.mul(f)), W(0)), W(1)), y = n.sub(f), b = A(y, y), x = v.mul(o.mul(o)), S = b.div(I(x, M(1e-6))).clamp(0, 1), C = L(n, p, L(I(M(1).div(s.add(1)), i), M(1), S)), w = L(oe(s.add(1), a), M(1), S);
+						h.assign(G(C, w));
 					}).Else(() => {
-						x.assign(G(y, 1));
+						h.assign(G(p, 1));
 					});
-				}), xe(e, we(V(p), V(_)), x).toWriteOnly(), xe(t, we(V(p), V(_)), b).toWriteOnly();
+				}), xe(e, we(V(u), V(d)), h).toWriteOnly();
 			});
 		})().compute([
 			this._dispatchX,
@@ -9312,7 +9306,7 @@ var bc = class extends wn {
 		this.on("asvgf:reset", () => this.resetTemporalData()), this.on("asvgf:setTemporal", (e) => {
 			e && e.enabled !== void 0 && (this.temporalEnabled = e.enabled, this.temporalEnabledU.value = +!!e.enabled);
 		}), this.on("asvgf:updateParameters", (e) => {
-			e && (e.temporalAlpha !== void 0 && (this.temporalAlpha.value = e.temporalAlpha), e.gradientScale !== void 0 && (this.gradientScale.value = e.gradientScale), e.phiColor !== void 0 && (this.phiColor.value = e.phiColor), e.phiNormal !== void 0 && (this.phiNormal.value = e.phiNormal), e.phiDepth !== void 0 && (this.phiDepth.value = e.phiDepth));
+			e && (e.temporalAlpha !== void 0 && (this.temporalAlpha.value = e.temporalAlpha), e.phiColor !== void 0 && (this.phiColor.value = e.phiColor));
 		});
 	}
 	render(e) {
@@ -9320,9 +9314,9 @@ var bc = class extends wn {
 		let t = e.getTexture("pathtracer:color"), n = e.getTexture("pathtracer:normalDepth"), r = e.getTexture("motionVector:screenSpace");
 		if (!t) return;
 		let i = t.image;
-		i && i.width > 0 && i.height > 0 && (i.width !== this._temporalTexA.image.width || i.height !== this._temporalTexA.image.height) && this.setSize(i.width, i.height), this._colorTexNode.value = t, n && (this._normalDepthTexNode.value = n), r && (this._motionTexNode.value = r), this._compiled ||= (this.renderer.compute(this._gradientNode), this.renderer.compute(this._temporalNodeA), this.renderer.compute(this._temporalNodeB), !0);
-		let a = this.currentMoments === 0 ? this._temporalTexB : this._temporalTexA, o = this.currentMoments === 0 ? this._prevNDTexB : this._prevNDTexA, s = this.currentMoments === 0 ? this._temporalNodeA : this._temporalNodeB, c = this.currentMoments === 0 ? this._temporalTexA : this._temporalTexB;
-		this._readTemporalTexNode.value = a, this.renderer.compute(this._gradientNode), this._gradientReadTexNode.value = this._gradientStorageTex, this._readPrevNDTexNode.value = o, this.renderer.compute(s), e.setTexture("asvgf:output", c), e.setTexture("asvgf:temporalColor", c), this.currentMoments = 1 - this.currentMoments, this.showHeatmap && (this._heatmapRawColorTexNode.value = t, this._heatmapColorTexNode.value = c, this._heatmapTemporalTexNode.value = c, n && (this._heatmapNDTexNode.value = n), r && (this._heatmapMotionTexNode.value = r), this._heatmapGradientTexNode.value = this._gradientStorageTex, this.renderer.compute(this._heatmapComputeNode), this.renderer.copyTextureToTexture(this._heatmapStorageTex, this.heatmapTarget.texture), this.heatmapHelper.update()), this.frameCount++;
+		i && i.width > 0 && i.height > 0 && (i.width !== this._temporalTexA.image.width || i.height !== this._temporalTexA.image.height) && this.setSize(i.width, i.height), this._colorTexNode.value = t, r && (this._motionTexNode.value = r), this._compiled ||= (this.renderer.compute(this._gradientNode), this.renderer.compute(this._temporalNodeA), this.renderer.compute(this._temporalNodeB), !0);
+		let a = this.currentMoments === 0 ? this._temporalTexB : this._temporalTexA, o = this.currentMoments === 0 ? this._temporalNodeA : this._temporalNodeB, s = this.currentMoments === 0 ? this._temporalTexA : this._temporalTexB;
+		this._readTemporalTexNode.value = a, this.renderer.compute(o), e.setTexture("asvgf:output", s), e.setTexture("asvgf:temporalColor", s), this.currentMoments = 1 - this.currentMoments, this.showHeatmap && (this.renderer.compute(this._gradientNode), this._heatmapRawColorTexNode.value = t, this._heatmapColorTexNode.value = s, this._heatmapTemporalTexNode.value = s, n && (this._heatmapNDTexNode.value = n), r && (this._heatmapMotionTexNode.value = r), this._heatmapGradientTexNode.value = this._gradientStorageTex, this.renderer.compute(this._heatmapComputeNode), this.renderer.copyTextureToTexture(this._heatmapStorageTex, this.heatmapTarget.texture), this.heatmapHelper.update()), this.frameCount++;
 	}
 	toggleHeatmap(e) {
 		this.showHeatmap = e, e ? this.heatmapHelper.show() : this.heatmapHelper.hide();
@@ -9331,13 +9325,13 @@ var bc = class extends wn {
 		this.temporalEnabled = e;
 	}
 	updateParameters(e) {
-		e && (e.temporalAlpha !== void 0 && (this.temporalAlpha.value = e.temporalAlpha), e.gradientScale !== void 0 && (this.gradientScale.value = e.gradientScale), e.phiColor !== void 0 && (this.phiColor.value = e.phiColor), e.phiNormal !== void 0 && (this.phiNormal.value = e.phiNormal), e.phiDepth !== void 0 && (this.phiDepth.value = e.phiDepth), e.debugMode !== void 0 && (this.debugMode.value = e.debugMode));
+		e && (e.temporalAlpha !== void 0 && (this.temporalAlpha.value = e.temporalAlpha), e.phiColor !== void 0 && (this.phiColor.value = e.phiColor), e.debugMode !== void 0 && (this.debugMode.value = e.debugMode));
 	}
 	resetTemporalData() {
 		this.frameCount = 0, this.currentMoments = 0;
 	}
 	setSize(e, t) {
-		this._temporalTexA.setSize(e, t), this._temporalTexB.setSize(e, t), this._prevNDTexA.setSize(e, t), this._prevNDTexB.setSize(e, t), this._gradientStorageTex.setSize(e, t), this._heatmapStorageTex.setSize(e, t), this.heatmapTarget.setSize(e, t), this.heatmapTarget.texture.needsUpdate = !0, this.resW.value = e, this.resH.value = t, this._dispatchX = Math.ceil(e / 8), this._dispatchY = Math.ceil(t / 8), this._gradientNode.dispatchSize = [
+		this._temporalTexA.setSize(e, t), this._temporalTexB.setSize(e, t), this._gradientStorageTex.setSize(e, t), this._heatmapStorageTex.setSize(e, t), this.heatmapTarget.setSize(e, t), this.heatmapTarget.texture.needsUpdate = !0, this.resW.value = e, this.resH.value = t, this._dispatchX = Math.ceil(e / 8), this._dispatchY = Math.ceil(t / 8), this._gradientNode.dispatchSize = [
 			this._dispatchX,
 			this._dispatchY,
 			1
@@ -9357,7 +9351,7 @@ var bc = class extends wn {
 	}
 	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._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();
+		this._gradientNode?.dispose(), this._temporalNodeA?.dispose(), this._temporalNodeB?.dispose(), this._temporalTexA?.dispose(), this._temporalTexB?.dispose(), this._gradientStorageTex?.dispose(), this._heatmapComputeNode?.dispose(), this._heatmapStorageTex?.dispose(), this.heatmapTarget?.dispose(), this._colorTexNode?.dispose(), this._motionTexNode?.dispose(), this._readTemporalTexNode?.dispose(), this._heatmapRawColorTexNode?.dispose(), this._heatmapColorTexNode?.dispose(), this._heatmapTemporalTexNode?.dispose(), this._heatmapNDTexNode?.dispose(), this._heatmapMotionTexNode?.dispose(), this._heatmapGradientTexNode?.dispose(), this.heatmapHelper?.dispose();
 	}
 }, xc = /* @__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"), Sc = class extends wn {
 	constructor(e, t = {}) {
@@ -9434,7 +9428,7 @@ var bc = class extends wn {
 	dispose() {
 		this._computeNodeA?.dispose(), this._computeNodeB?.dispose(), this._storageTexA?.dispose(), this._storageTexB?.dispose(), this._colorTexNode?.dispose(), this._readTexNodeA?.dispose(), this._readTexNodeB?.dispose();
 	}
-}, Cc = /* @__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"), wc = class extends wn {
+}, Cc = /* @__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		// clamp dot to [0,1] not just max(., 0): miss-ray normals decode to\n		// non-unit (-1,-1,-1) with dot=3, which would saturate pow to +inf\n		// and poison output via inf*0 = NaN. See project_tsl_pitfalls memory.\n		let normW = pow( clamp( dot( centerNormal, sNormal ), 0.0, 1.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"), wc = class extends wn {
 	constructor(e, t = {}) {
 		super("BilateralFiltering", {
 			...t,