rayzee 8.3.0 → 8.4.0

package/dist/ToneMapGPU-DzqSyzTx.js

@@ -0,0 +1,341 @@
+import { ACESFilmicToneMapping as e, AgXToneMapping as t, CineonToneMapping as n, LinearToneMapping as r, NeutralToneMapping as i, NoToneMapping as a, ReinhardToneMapping as o } from "three";
+//#region src/utils/Logger.js
+var s = {
+	silent: 0,
+	error: 1,
+	warn: 2,
+	info: 3,
+	debug: 4
+}, c = "info", l = "rayzeeLogLevel", u = "rayzeeLogNamespaces", d = "color:#7c8494", f = "color:inherit", p = null, m = null;
+function h(e) {
+	try {
+		return globalThis.localStorage?.getItem(e) ?? null;
+	} catch {
+		return null;
+	}
+}
+function g(e, t) {
+	try {
+		t === null ? globalThis.localStorage?.removeItem(e) : globalThis.localStorage?.setItem(e, t);
+	} catch {}
+}
+function _() {
+	if (p !== null) return p;
+	let e = globalThis.__RAYZEE_LOG_LEVEL__;
+	return e !== void 0 && e in s ? (p = s[e], p) : (p = s[h(l)] ?? s[c], p);
+}
+function v() {
+	if (m !== null) return m;
+	let e = globalThis.__RAYZEE_LOG_NS__ ?? h(u);
+	return m = e ? String(e).split(",").map((e) => e.trim()).filter(Boolean) : [], m;
+}
+function y(e, t) {
+	if (t < s.debug) return !0;
+	let n = v();
+	return n.length === 0 || n.some((t) => e === t || e.startsWith(`${t}:`));
+}
+function b(e, t, n) {
+	if (e > _() || !y(t, e)) return;
+	let r = e === s.error ? console.error : e === s.warn ? console.warn : console.log, i = `[${t}]`;
+	typeof n[0] == "string" ? r(`%c${i}%c ${n[0]}`, d, f, ...n.slice(1)) : r(`%c${i}`, d, ...n);
+}
+var x = {
+	n: (e) => Number(e).toLocaleString("en-US"),
+	ms: (e) => e < 10 ? `${Number(e).toFixed(1)} ms` : `${Math.round(e)} ms`,
+	mb: (e) => {
+		let t = e / (1024 * 1024);
+		return t < 10 ? `${t.toFixed(1)} MB` : `${Math.round(t)} MB`;
+	},
+	px: (e, t) => `${e}×${t}`,
+	count: (e, t, n = `${t}s`) => `${x.n(e)} ${Number(e) === 1 ? t : n}`,
+	list: (e) => e.filter(Boolean).join(" · ")
+}, S = {
+	levels: s,
+	setLevel(e) {
+		return e in s ? (globalThis.__RAYZEE_LOG_LEVEL__ = e, p = s[e], g(l, e === c ? null : e), e) : (console.warn(`[log] unknown level '${e}' — expected one of ${Object.keys(s).join(", ")}`), this.getLevel());
+	},
+	getLevel() {
+		let e = _();
+		return Object.keys(s).find((t) => s[t] === e);
+	},
+	isEnabled(e) {
+		return (s[e] ?? s.debug) <= _();
+	},
+	only(...e) {
+		let t = e.flat().filter(Boolean);
+		return m = t, globalThis.__RAYZEE_LOG_NS__ = t.join(","), g(u, t.length ? t.join(",") : null), t.length && this.setLevel("debug"), t;
+	},
+	refresh() {
+		return p = null, m = null, this.getLevel();
+	}
+};
+function C(e) {
+	return {
+		namespace: e,
+		error: (...t) => b(s.error, e, t),
+		warn: (...t) => b(s.warn, e, t),
+		info: (...t) => b(s.info, e, t),
+		debug: (...t) => b(s.debug, e, t),
+		isEnabled: (e) => S.isEnabled(e),
+		summary(t, n = []) {
+			let r = n.filter(Boolean);
+			if (r.length === 0 || !console.groupCollapsed) {
+				b(s.info, e, [t]), r.forEach((t) => b(s.debug, e, [t]));
+				return;
+			}
+			s.info > _() || ((_() >= s.debug && console.group ? console.group : console.groupCollapsed)(`%c[${e}]%c ${t}`, d, f), r.forEach((e) => console.log(e)), console.groupEnd());
+		}
+	};
+}
+var w = /* @__PURE__ */ new Set();
+function T(e, t, ...n) {
+	w.has(t) || (w.add(t), e.warn(...n));
+}
+function E() {
+	return S.getLevel();
+}
+//#endregion
+//#region src/AssetConfig.js
+var D = {
+	stbnScalarAtlas: "https://assets.rayzee.atulmourya.com/noise/stbn_scalar_atlas.png",
+	stbnVec2Atlas: "https://assets.rayzee.atulmourya.com/noise/stbn_vec2_atlas.png",
+	ortRuntimeUrl: "https://cdn.jsdelivr.net/npm/onnxruntime-web@1.24.3/dist/ort.webgpu.bundle.min.mjs",
+	ortWasmPaths: "https://cdn.jsdelivr.net/npm/onnxruntime-web@1.24.3/dist/",
+	dracoDecoderPath: "https://www.gstatic.com/draco/v1/decoders/",
+	ktx2TranscoderPath: "https://cdn.jsdelivr.net/npm/three@0.183.2/examples/jsm/libs/basis/",
+	oidnWeightsBaseUrl: "https://cdn.jsdelivr.net/npm/denoiser/tzas/",
+	upscalerModelBaseUrl: "https://huggingface.co/notaneimu/onnx-image-models/resolve/main/",
+	dlssRuntimeUrl: "/dlss/dlss-runtime.js",
+	dlssAssetBaseUrl: "https://pub-6e048ff014374f11bef33d76b6c2b5ef.r2.dev/v1",
+	cacheNamespace: "rayzee"
+};
+function O(e) {
+	e && Object.assign(D, e);
+}
+function k() {
+	return { ...D };
+}
+var A = `
+struct Params {
+	width: u32,
+	height: u32,
+	mode: u32,
+	flipY: u32,
+	exposure: f32,
+	saturation: f32,
+};
+
+@group(0) @binding(0) var<storage, read> src: array<u32>;
+@group(0) @binding(1) var<storage, read_write> dst: array<u32>;
+@group(0) @binding(2) var<uniform> params: Params;
+
+${`
+const TM_NONE: u32 = ${a}u;
+const TM_LINEAR: u32 = ${r}u;
+const TM_REINHARD: u32 = ${o}u;
+const TM_CINEON: u32 = ${n}u;
+const TM_ACES: u32 = ${e}u;
+const TM_AGX: u32 = ${t}u;
+const TM_NEUTRAL: u32 = ${i}u;
+
+fn tm_reinhard( c: vec3<f32> ) -> vec3<f32> {
+	return clamp( c / ( c + vec3<f32>( 1.0 ) ), vec3<f32>( 0.0 ), vec3<f32>( 1.0 ) );
+}
+
+fn tm_cineon( c0: vec3<f32> ) -> vec3<f32> {
+	let c = max( c0 - vec3<f32>( 0.004 ), vec3<f32>( 0.0 ) );
+	let v = ( c * ( 6.2 * c + vec3<f32>( 0.5 ) ) ) / ( c * ( 6.2 * c + vec3<f32>( 1.7 ) ) + vec3<f32>( 0.06 ) );
+	return pow( max( v, vec3<f32>( 0.0 ) ), vec3<f32>( 2.2 ) );
+}
+
+fn tm_aces( c0: vec3<f32> ) -> vec3<f32> {
+	let c = c0 / 0.6;
+	let m_in = mat3x3<f32>(
+		vec3<f32>( 0.59719, 0.07600, 0.02840 ),
+		vec3<f32>( 0.35458, 0.90834, 0.13383 ),
+		vec3<f32>( 0.04823, 0.01566, 0.83777 ) );
+	let v = m_in * c;
+	let a = v * ( v + vec3<f32>( 0.0245786 ) ) - vec3<f32>( 0.000090537 );
+	let b = v * ( 0.983729 * v + vec3<f32>( 0.4329510 ) ) + vec3<f32>( 0.238081 );
+	let m_out = mat3x3<f32>(
+		vec3<f32>( 1.60475, -0.10208, -0.00327 ),
+		vec3<f32>( -0.53108, 1.10813, -0.07276 ),
+		vec3<f32>( -0.07367, -0.00605, 1.07602 ) );
+	return clamp( m_out * ( a / b ), vec3<f32>( 0.0 ), vec3<f32>( 1.0 ) );
+}
+
+fn tm_agx( c0: vec3<f32> ) -> vec3<f32> {
+	let m_in = mat3x3<f32>(
+		vec3<f32>( 0.6274, 0.0691, 0.0164 ),
+		vec3<f32>( 0.3293, 0.9195, 0.0880 ),
+		vec3<f32>( 0.0433, 0.0113, 0.8956 ) );
+	let m_agx = mat3x3<f32>(
+		vec3<f32>( 0.856627153315983, 0.137318972929847, 0.11189821299995 ),
+		vec3<f32>( 0.0951212405381588, 0.761241990602591, 0.0767994186031903 ),
+		vec3<f32>( 0.0482516061458583, 0.101439036467562, 0.811302368396859 ) );
+
+	var v = m_agx * ( m_in * c0 );
+
+	let minEv = -12.47393;
+	let maxEv = 4.026069;
+	v = clamp( ( log2( max( v, vec3<f32>( 1e-10 ) ) ) - vec3<f32>( minEv ) ) / ( maxEv - minEv ),
+		vec3<f32>( 0.0 ), vec3<f32>( 1.0 ) );
+
+	let x2 = v * v;
+	let x4 = x2 * x2;
+	v = 15.5 * x4 * x2 - 40.14 * x4 * v + 31.96 * x4 - 6.868 * x2 * v + 0.4298 * x2 + 0.1191 * v
+		- vec3<f32>( 0.00232 );
+
+	let m_out = mat3x3<f32>(
+		vec3<f32>( 1.1271005818144368, -0.1413297634984383, -0.14132976349843826 ),
+		vec3<f32>( -0.11060664309660323, 1.157823702216272, -0.11060664309660294 ),
+		vec3<f32>( -0.016493938717834573, -0.016493938717834257, 1.2519364065950405 ) );
+	let o = pow( max( m_out * v, vec3<f32>( 0.0 ) ), vec3<f32>( 2.2 ) );
+
+	let m_srgb = mat3x3<f32>(
+		vec3<f32>( 1.6605, -0.1246, -0.0182 ),
+		vec3<f32>( -0.5876, 1.1329, -0.1006 ),
+		vec3<f32>( -0.0728, -0.0083, 1.1187 ) );
+	return clamp( m_srgb * o, vec3<f32>( 0.0 ), vec3<f32>( 1.0 ) );
+}
+
+fn tm_neutral( c0: vec3<f32> ) -> vec3<f32> {
+	let startCompression = 0.8 - 0.04;
+	let desaturation = 0.15;
+
+	let x = min( c0.r, min( c0.g, c0.b ) );
+	var offset = 0.04;
+	if ( x < 0.08 ) { offset = x - 6.25 * x * x; }
+	var c = c0 - vec3<f32>( offset );
+
+	let peak = max( c.r, max( c.g, c.b ) );
+	if ( peak < startCompression ) { return c; }
+
+	let d = 1.0 - startCompression;
+	let newPeak = 1.0 - d * d / ( peak + d - startCompression );
+	c = c * ( newPeak / peak );
+	let gFactor = 1.0 - 1.0 / ( desaturation * ( peak - newPeak ) + 1.0 );
+	return mix( c, vec3<f32>( newPeak ), gFactor );
+}
+
+// Three.js clamps the fragment output with max(0) before tone mapping, so the curves never see a
+// negative channel. The saturation grade drives channels below zero on much of a typical frame, and
+// AgX/Neutral mix negatives across channels instead of clipping them.
+fn rayzee_tone_curve( color: vec3<f32>, mode: u32 ) -> vec3<f32> {
+	let c = max( color, vec3<f32>( 0.0 ) );
+	if ( mode == TM_REINHARD ) { return tm_reinhard( c ); }
+	if ( mode == TM_CINEON ) { return tm_cineon( c ); }
+	if ( mode == TM_ACES ) { return tm_aces( c ); }
+	if ( mode == TM_AGX ) { return tm_agx( c ); }
+	if ( mode == TM_NEUTRAL ) { return tm_neutral( c ); }
+	return clamp( c, vec3<f32>( 0.0 ), vec3<f32>( 1.0 ) );
+}
+
+fn rayzee_tone_map( linearRGB: vec3<f32>, mode: u32, exposure: f32, saturation: f32 ) -> vec3<f32> {
+	// Three.js returns early for NoToneMapping without applying exposure, so a readback that applied
+	// it would paint brighter than the viewport it replaces.
+	var c = linearRGB * select( exposure, 1.0, mode == TM_NONE );
+	if ( saturation != 1.0 ) {
+		let luma = vec3<f32>( dot( c, vec3<f32>( 0.2126, 0.7152, 0.0722 ) ) );
+		c = luma + ( c - luma ) * saturation;
+	}
+	return rayzee_tone_curve( c, mode );
+}
+
+fn rayzee_linear_to_srgb( c: vec3<f32> ) -> vec3<f32> {
+	return select(
+		1.055 * pow( max( c, vec3<f32>( 0.0 ) ), vec3<f32>( 1.0 / 2.4 ) ) - vec3<f32>( 0.055 ),
+		12.92 * c,
+		c <= vec3<f32>( 0.0031308 ) );
+}
+
+fn rayzee_to_u8( srgb: vec3<f32> ) -> vec3<u32> {
+	return vec3<u32>( clamp( round( srgb * 255.0 + vec3<f32>( 0.5 ) ), vec3<f32>( 0.0 ), vec3<f32>( 255.0 ) ) );
+}
+`}
+
+@compute @workgroup_size(8, 8)
+fn main( @builtin(global_invocation_id) gid: vec3<u32> ) {
+	if ( gid.x >= params.width || gid.y >= params.height ) { return; }
+
+	var srcY = gid.y;
+	if ( params.flipY == 1u ) { srcY = params.height - 1u - gid.y; }
+
+	let a = ( srcY * params.width + gid.x ) * 2u;
+	let rg = unpack2x16float( src[ a ] );
+	let ba = unpack2x16float( src[ a + 1u ] );
+
+	let mapped = rayzee_tone_map( vec3<f32>( rg.x, rg.y, ba.x ), params.mode, params.exposure, params.saturation );
+	let b8 = rayzee_to_u8( rayzee_linear_to_srgb( mapped ) );
+	dst[ gid.y * params.width + gid.x ] = b8.x | ( b8.y << 8u ) | ( b8.z << 16u ) | ( 255u << 24u );
+}
+`, j = 32, M = class {
+	constructor(e, t = "rayzee:tonemap") {
+		this.device = e, this.label = t, this.width = 0, this.height = 0, this._pipeline = null, this._storage = null, this._map = null, this._params = null, this._paramData = new ArrayBuffer(j), this._paramU32 = new Uint32Array(this._paramData), this._paramF32 = new Float32Array(this._paramData), this.disposed = !1;
+	}
+	ensureSize(e, t) {
+		if (this.width === e && this.height === t && this._pipeline) return;
+		this._releaseBuffers();
+		let n = e * t * 4;
+		this._storage = this.device.createBuffer({
+			label: `${this.label}-out`,
+			size: n,
+			usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC
+		}), this._map = this.device.createBuffer({
+			label: `${this.label}-map`,
+			size: n,
+			usage: GPUBufferUsage.COPY_DST | GPUBufferUsage.MAP_READ
+		}), this._params = this.device.createBuffer({
+			label: `${this.label}-params`,
+			size: j,
+			usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST
+		}), this._pipeline ??= this.device.createComputePipeline({
+			label: this.label,
+			layout: "auto",
+			compute: {
+				module: this.device.createShaderModule({
+					label: this.label,
+					code: A
+				}),
+				entryPoint: "main"
+			}
+		}), this.width = e, this.height = t;
+	}
+	async toRGBA8(e, { exposure: t = 1, toneMapping: n = 0, saturation: r = 1, flipY: i = !1 } = {}) {
+		if (this.disposed) throw Error("PackedToneMapper: disposed");
+		if (!this._pipeline) throw Error("PackedToneMapper: call ensureSize() first");
+		let { width: a, height: o } = this;
+		this._paramU32[0] = a, this._paramU32[1] = o, this._paramU32[2] = n >>> 0, this._paramU32[3] = +!!i, this._paramF32[4] = t, this._paramF32[5] = r, this.device.queue.writeBuffer(this._params, 0, this._paramData);
+		let s = this.device.createBindGroup({
+			layout: this._pipeline.getBindGroupLayout(0),
+			entries: [
+				{
+					binding: 0,
+					resource: {
+						buffer: e,
+						size: a * o * 8
+					}
+				},
+				{
+					binding: 1,
+					resource: { buffer: this._storage }
+				},
+				{
+					binding: 2,
+					resource: { buffer: this._params }
+				}
+			]
+		}), c = this.device.createCommandEncoder({ label: this.label }), l = c.beginComputePass();
+		l.setPipeline(this._pipeline), l.setBindGroup(0, s), l.dispatchWorkgroups(Math.ceil(a / 8), Math.ceil(o / 8)), l.end(), c.copyBufferToBuffer(this._storage, 0, this._map, 0, a * o * 4), this.device.queue.submit([c.finish()]), await this._map.mapAsync(GPUMapMode.READ);
+		let u = new Uint8ClampedArray(this._map.getMappedRange().slice(0));
+		return this._map.unmap(), u;
+	}
+	_releaseBuffers() {
+		this._storage?.destroy(), this._map?.destroy(), this._params?.destroy(), this._storage = null, this._map = null, this._params = null;
+	}
+	dispose() {
+		this.disposed || (this.disposed = !0, this._releaseBuffers(), this._pipeline = null);
+	}
+};
+//#endregion
+export { S as a, T as c, s as i, E as l, O as n, C as o, k as r, x as s, M as t };