@aura3d/engine 4.1.2 → 4.2.0

package/dist/rendering/webgl2/LegacyPost.js

@@ -10,6 +10,7 @@ import { BLOOM_BRIGHT_LUT_HEIGHT, BLOOM_BRIGHT_LUT_WIDTH, BLOOM_COMPOSITE_LUT_SI
 // `u_hasFxaa` taps stay the default path.
 import { FXAA_185_FRAGMENT_GLSL } from "../post/shaders/fxaa.glsl.js";
 import { recordPostSkipped } from "../renderer/postSkipped.js";
+import { ldrFragmentSource, ldrVariantKey } from "./LdrPostprocessShader.js";
 export class WebGL2LegacyPostPipeline {
     host;
     constructor(host) {
@@ -29,7 +30,8 @@ export class WebGL2LegacyPostPipeline {
     bloomPyramidResources = null;
     depthOfFieldProgram = null;
     lastBloomDiagnostics = null;
-    ldrPostprocessProgram = null;
+    /** One compiled program per feature variant (see `ldrFragmentSource`). */
+    ldrPostprocessPrograms = new Map();
     motionBlurProgram = null;
     nativeFxaaFinalizeProgram = null;
     motionBlurVelocitySize = null;
@@ -159,7 +161,6 @@ export class WebGL2LegacyPostPipeline {
         const ssrOptions = ssrPass ? normalizeNativeSsrOptions(ssrPass.options) : undefined;
         const taaOptions = taaPass ? normalizeNativeTaaOptions(taaPass.options, source.width, source.height) : undefined;
         const outlineOptions = outlinePass ? normalizeNativeOutlineOptions(outlinePass.options) : undefined;
-        const program = this.ensureLdrPostprocessProgram();
         const vertexArray = this.ensurePresentationVertexArray();
         const sourceIsHdr = source.colorTexture.format !== "rgba8";
         // PRD-03 Phase 1: `fxaa.variant === "r185"` splits the finalize — the
@@ -243,7 +244,7 @@ export class WebGL2LegacyPostPipeline {
             if ((depthOfFieldOptions || motionBlurOptions || ssaoOptions || ssrOptions || taaOptions || outlineOptions) && bloomResources) {
                 if (tonePass || colorPass) {
                     const baseTargetIndex = 0;
-                    this.drawNativeLdrStage(ldrSourceHandle, bloomResources.framebuffers[baseTargetIndex], source.width, source.height, program, vertexArray, tonePass, colorPass, undefined, options.toneMappingDefaults);
+                    this.drawNativeLdrStage(ldrSourceHandle, bloomResources.framebuffers[baseTargetIndex], source.width, source.height, vertexArray, tonePass, colorPass, undefined, options.toneMappingDefaults);
                     ldrSourceHandle = bloomResources.textures[baseTargetIndex];
                     temporarySourceIndex = baseTargetIndex;
                 }
@@ -280,11 +281,11 @@ export class WebGL2LegacyPostPipeline {
             }
             if (fxaaSplit && bloomResources) {
                 const fxaaScratchIndex = temporarySourceIndex === 0 ? 1 : 0;
-                this.drawNativeLdrStage(ldrSourceHandle, bloomResources.framebuffers[fxaaScratchIndex], source.width, source.height, program, vertexArray, depthOfFieldOptions || motionBlurOptions || ssaoOptions || ssrOptions || taaOptions || outlineOptions ? undefined : tonePass, depthOfFieldOptions || motionBlurOptions || ssaoOptions || ssrOptions || taaOptions || outlineOptions ? undefined : colorPass, undefined, options.toneMappingDefaults);
+                this.drawNativeLdrStage(ldrSourceHandle, bloomResources.framebuffers[fxaaScratchIndex], source.width, source.height, vertexArray, depthOfFieldOptions || motionBlurOptions || ssaoOptions || ssrOptions || taaOptions || outlineOptions ? undefined : tonePass, depthOfFieldOptions || motionBlurOptions || ssaoOptions || ssrOptions || taaOptions || outlineOptions ? undefined : colorPass, undefined, options.toneMappingDefaults);
                 this.executeNativeFxaaFinalize(bloomResources.textures[fxaaScratchIndex], webglOutputTarget?.framebuffer ?? null, outputWidth, outputHeight, source.width, source.height, vertexArray);
             }
             else {
-                this.drawNativeLdrStage(ldrSourceHandle, webglOutputTarget?.framebuffer ?? null, outputWidth, outputHeight, program, vertexArray, depthOfFieldOptions || motionBlurOptions || ssaoOptions || ssrOptions || taaOptions || outlineOptions ? undefined : tonePass, depthOfFieldOptions || motionBlurOptions || ssaoOptions || ssrOptions || taaOptions || outlineOptions ? undefined : colorPass, fxaaPass, options.toneMappingDefaults);
+                this.drawNativeLdrStage(ldrSourceHandle, webglOutputTarget?.framebuffer ?? null, outputWidth, outputHeight, vertexArray, depthOfFieldOptions || motionBlurOptions || ssaoOptions || ssrOptions || taaOptions || outlineOptions ? undefined : tonePass, depthOfFieldOptions || motionBlurOptions || ssaoOptions || ssrOptions || taaOptions || outlineOptions ? undefined : colorPass, fxaaPass, options.toneMappingDefaults);
             }
             this.host.gl.flush();
         }
@@ -483,10 +484,19 @@ export class WebGL2LegacyPostPipeline {
             this.host.stateCache.invalidate();
         }
     }
-    drawNativeLdrStage(sourceTexture, framebuffer, width, height, program, vertexArray, tonePass, colorPass, fxaaPass, toneMappingDefaults) {
+    drawNativeLdrStage(sourceTexture, framebuffer, width, height, vertexArray, tonePass, colorPass, fxaaPass, toneMappingDefaults) {
         const toneOptions = { ...(toneMappingDefaults ?? {}), ...(tonePass?.options ?? {}) };
         const colorOptions = colorPass?.options ?? {};
         const fxaaOptions = fxaaPass?.options ?? {};
+        const program = this.ensureLdrPostprocessProgram({
+            toneMapping: Boolean(tonePass),
+            operator: toneMappingOperatorId(stringOption(toneOptions, "operator", "reinhard")),
+            inputSrgb: colorSpaceId(stringOption(toneOptions, "inputColorSpace", "linear")) === 1,
+            outputSrgb: colorSpaceId(stringOption(toneOptions, "outputColorSpace", "linear")) === 1,
+            grade: Boolean(colorPass),
+            sharpening: Boolean(colorPass) && numberOption(colorOptions, "sharpening", 0) > 0,
+            fxaa: Boolean(fxaaPass)
+        });
         this.host.lifecycle.bindOutput(framebuffer, width, height);
         this.host.gl.useProgram(program);
         this.host.gl.bindVertexArray(vertexArray);
@@ -1808,10 +1818,11 @@ void main() {
         }
         return program;
     }
-    ensureLdrPostprocessProgram() {
-        if (this.ldrPostprocessProgram) {
-            return this.ldrPostprocessProgram;
-        }
+    ensureLdrPostprocessProgram(variant) {
+        const key = ldrVariantKey(variant);
+        const cached = this.ldrPostprocessPrograms.get(key);
+        if (cached)
+            return cached;
         const vertex = this.host.device.compileShader(this.host.gl.VERTEX_SHADER, `#version 300 es
 precision highp float;
 out vec2 v_uv;
@@ -1821,186 +1832,7 @@ void main() {
   gl_Position = vec4(position * 2.0 - 1.0, 0.0, 1.0);
 }
 `, "webgl2-ldr-postprocess");
-        const fragment = this.host.device.compileShader(this.host.gl.FRAGMENT_SHADER, `#version 300 es
-precision highp float;
-uniform sampler2D u_source;
-uniform vec2 u_texelSize;
-uniform int u_hasToneMapping;
-uniform int u_toneOperator;
-uniform int u_inputColorSpace;
-uniform int u_outputColorSpace;
-uniform float u_exposure;
-uniform float u_whitePoint;
-uniform float u_gamma;
-uniform int u_hasColorGrade;
-uniform float u_contrast;
-uniform float u_temperature;
-uniform float u_tint;
-uniform float u_saturation;
-uniform float u_vibrance;
-uniform float u_vignette;
-uniform float u_sharpening;
-uniform int u_hasFxaa;
-uniform float u_edgeThreshold;
-uniform float u_subpixelBlend;
-in vec2 v_uv;
-out vec4 outColor;
-
-float srgbToLinear(float value) {
-  return value <= 0.04045 ? value / 12.92 : pow((value + 0.055) / 1.055, 2.4);
-}
-
-float linearToSrgb(float value) {
-  return value <= 0.0031308 ? value * 12.92 : 1.055 * pow(value, 1.0 / 2.4) - 0.055;
-}
-
-vec3 decodeColor(vec3 color) {
-  return u_inputColorSpace == 1 ? vec3(srgbToLinear(color.r), srgbToLinear(color.g), srgbToLinear(color.b)) : color;
-}
-
-vec3 encodeColor(vec3 color) {
-  vec3 linear = clamp(color, 0.0, 1.0);
-  return u_outputColorSpace == 1 ? vec3(linearToSrgb(linear.r), linearToSrgb(linear.g), linearToSrgb(linear.b)) : linear;
-}
-
-vec3 acesRrtAndOdtFit(vec3 value) {
-  vec3 numerator = value * (value + 0.0245786) - 0.000090537;
-  vec3 denominator = value * (0.983729 * value + 0.4329510) + 0.238081;
-  return numerator / denominator;
-}
-
-vec3 acesFilmic(vec3 color) {
-  // Match current Three.js ACESFilmicToneMapping. ACES is a coupled RGB
-  // transform; a scalar approximation shifts hue and washes saturated colors.
-  const mat3 acesInput = mat3(
-    vec3(0.59719, 0.07600, 0.02840),
-    vec3(0.35458, 0.90834, 0.13383),
-    vec3(0.04823, 0.01566, 0.83777)
-  );
-  const mat3 acesOutput = mat3(
-    vec3(1.60475, -0.10208, -0.00327),
-    vec3(-0.53108, 1.10813, -0.07276),
-    vec3(-0.07367, -0.00605, 1.07602)
-  );
-  vec3 acesColor = acesInput * (color / 0.6);
-  acesColor = acesRrtAndOdtFit(acesColor);
-  return clamp(acesOutput * acesColor, 0.0, 1.0);
-}
-
-float filmic(float value) {
-  float x = max(0.0, value);
-  float toe = max(0.0, x - 0.004);
-  float curve = (toe * (6.2 * toe + 0.5)) / (toe * (6.2 * toe + 1.7) + 0.06);
-  return min(curve, x * 1.08);
-}
-
-float uncharted2(float value) {
-  float a = 0.15;
-  float b = 0.5;
-  float c = 0.1;
-  float d = 0.2;
-  float e = 0.02;
-  float f = 0.3;
-  float w = 11.2;
-  float x = value * 2.0;
-  float curve = ((x * (a * x + c * b) + d * e) / (x * (a * x + b) + d * f)) - e / f;
-  float wx = w;
-  float white = ((wx * (a * wx + c * b) + d * e) / (wx * (a * wx + b) + d * f)) - e / f;
-  return clamp(curve / white, 0.0, 1.0);
-}
-
-float agx(float value) {
-  float x = max(0.0, value);
-  float encoded = log2(1.0 + x) / log2(17.0);
-  return clamp(encoded * encoded * (3.0 - 2.0 * encoded), 0.0, 1.0);
-}
-
-float neutral(float value) {
-  float x = max(0.0, value);
-  return min(1.0, (x * (1.0 + x / 7.5)) / (1.0 + x));
-}
-
-float toneMapChannel(float value) {
-  float exposed = max(0.0, value * u_exposure) / max(0.0001, u_whitePoint);
-  if (u_toneOperator == 0) return min(1.0, exposed);
-  if (u_toneOperator == 1) return exposed / (1.0 + exposed);
-  if (u_toneOperator == 2) return exposed;
-  if (u_toneOperator == 3) return filmic(exposed);
-  if (u_toneOperator == 4) return uncharted2(exposed);
-  if (u_toneOperator == 5) return agx(exposed);
-  return neutral(exposed);
-}
-
-vec3 applyToneMapping(vec3 color) {
-  if (u_hasToneMapping == 0) return color;
-  vec3 decoded = decodeColor(color);
-  vec3 mapped = u_toneOperator == 2
-    ? acesFilmic(max(vec3(0.0), decoded * u_exposure) / max(0.0001, u_whitePoint))
-    : vec3(toneMapChannel(decoded.r), toneMapChannel(decoded.g), toneMapChannel(decoded.b));
-  return encodeColor(mapped);
-}
-
-vec3 applyColorGrade(vec3 color, vec2 uv) {
-  if (u_hasColorGrade == 0) return color;
-  float contrastOffset = 0.5 - 0.5 * u_contrast;
-  float redShift = u_temperature * 0.08 - u_tint * 0.02;
-  float greenShift = u_tint * 0.06;
-  float blueShift = -u_temperature * 0.08 - u_tint * 0.02;
-  float luma = dot(color, vec3(0.2126, 0.7152, 0.0722));
-  vec3 graded = color * u_contrast + vec3(contrastOffset + redShift, contrastOffset + greenShift, contrastOffset + blueShift);
-  float distanceFromLuma = min(1.0, abs(graded.r - luma) + abs(graded.g - luma) + abs(graded.b - luma));
-  float vibranceBoost = u_vibrance == 0.0 ? 0.0 : u_vibrance * (1.0 - distanceFromLuma);
-  float saturation = u_saturation + vibranceBoost;
-  graded = vec3(luma) + (graded - vec3(luma)) * saturation;
-  vec2 centered = uv * 2.0 - 1.0;
-  float vignette = 1.0 - u_vignette * clamp((length(centered) - 0.28) / 1.12, 0.0, 1.0);
-  return clamp(graded * vignette, 0.0, 1.0);
-}
-
-vec3 baseColorAt(vec2 uv) {
-  vec2 clampedUv = clamp(uv, vec2(0.0), vec2(1.0));
-  vec3 color = texture(u_source, clampedUv).rgb;
-  return applyColorGrade(applyToneMapping(color), clampedUv);
-}
-
-vec3 finalColorAt(vec2 uv) {
-  vec3 center = baseColorAt(uv);
-  if (u_hasColorGrade == 0 || u_sharpening <= 0.0) return center;
-  vec3 blur = (
-    baseColorAt(uv + vec2(-u_texelSize.x, 0.0)) +
-    baseColorAt(uv + vec2(u_texelSize.x, 0.0)) +
-    baseColorAt(uv + vec2(0.0, -u_texelSize.y)) +
-    baseColorAt(uv + vec2(0.0, u_texelSize.y))
-  ) * 0.25;
-  return clamp(center + (center - blur) * u_sharpening, 0.0, 1.0);
-}
-
-float luma(vec3 color) {
-  return dot(color, vec3(0.2126, 0.7152, 0.0722));
-}
-
-void main() {
-  vec3 center = finalColorAt(v_uv);
-  float alpha = texture(u_source, clamp(v_uv, vec2(0.0), vec2(1.0))).a;
-  if (u_hasFxaa == 0) {
-    outColor = vec4(center, alpha);
-    return;
-  }
-  vec3 north = finalColorAt(v_uv + vec2(0.0, -u_texelSize.y));
-  vec3 south = finalColorAt(v_uv + vec2(0.0, u_texelSize.y));
-  vec3 west = finalColorAt(v_uv + vec2(-u_texelSize.x, 0.0));
-  vec3 east = finalColorAt(v_uv + vec2(u_texelSize.x, 0.0));
-  float centerLuma = luma(center);
-  float minLuma = min(centerLuma, min(min(luma(north), luma(south)), min(luma(west), luma(east))));
-  float maxLuma = max(centerLuma, max(max(luma(north), luma(south)), max(luma(west), luma(east))));
-  if (maxLuma - minLuma < u_edgeThreshold) {
-    outColor = vec4(center, alpha);
-    return;
-  }
-  vec3 average = (north + south + west + east) * 0.25;
-  outColor = vec4(mix(center, average, clamp(u_subpixelBlend, 0.0, 1.0)), alpha);
-}
-`, "webgl2-ldr-postprocess");
+        const fragment = this.host.device.compileShader(this.host.gl.FRAGMENT_SHADER, ldrFragmentSource(variant), "webgl2-ldr-postprocess");
         const program = this.host.gl.createProgram();
         if (!program) {
             this.host.gl.deleteShader(vertex);
@@ -2018,7 +1850,7 @@ void main() {
             this.host.gl.deleteProgram(program);
             throw new RenderDeviceError("WebGL2 LDR postprocess shader link failed", "SHADER_LINK_FAILED", { log });
         }
-        this.ldrPostprocessProgram = program;
+        this.ldrPostprocessPrograms.set(key, program);
         return program;
     }
 }