rayzee 8.5.0 → 8.6.0

package/dist/ToneMapGPU-DzqSyzTx.js

@@ -1,341 +0,0 @@
-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 };