191 lines
12 KiB
C#
191 lines
12 KiB
C#
using Echovoid.Runtime.Behavior.Rendering;
|
||
using UnityEngine;
|
||
using UnityEngine.Rendering;
|
||
using UnityEngine.Rendering.Universal;
|
||
using UnityEngine.Experimental.Rendering;
|
||
|
||
namespace SLSUtilities.Rendering.PostProcessing
|
||
{
|
||
[System.Serializable, VolumeComponentMenu("SLS/Postprocessing/Anime Bloom")]
|
||
public class AnimeBloom : ScriptablePostProcessorVolume
|
||
{
|
||
// BeforePostProcess: 在 ToneMapping 之前运行，确保 HDR bloom 颜色不被截断
|
||
// 这与 Unity 原生 Bloom 的执行阶段完全一致
|
||
public override CustomPostProcessInjectionPoint InjectionPoint => CustomPostProcessInjectionPoint.BeforePostProcess;
|
||
public override int OrderInInjectionPoint => 5;
|
||
|
||
[Header("Glow Settings")]
|
||
public ClampedFloatParameter intensity = new(1f, 0f, 10f);
|
||
|
||
// 注意：threshold 在 Gamma 空间中设置，C# 侧会转为 Linear。
|
||
// 0.9 在 gamma 空间 ≈ 0.79 在 linear 空间，这与 Unity 原生默认行为完全一致。
|
||
public MinFloatParameter threshold = new(0.9f, 0f);
|
||
|
||
[Header("Scatter")]
|
||
// scatter 在 [0,1] 范围内由用户设置。
|
||
// C# 侧映射：Mathf.Lerp(0.05f, 0.95f, scatter.value)，与 Unity 原生完全一致。
|
||
// 低值 = 光晕聚拢，高值 = 光晕向外大范围扩散。
|
||
public ClampedFloatParameter scatter = new(0.7f, 0f, 1f);
|
||
|
||
public MinFloatParameter clamp = new(65472f, 1f);
|
||
|
||
[Header("Iterations")]
|
||
// 迭代次数：Dual Kawase 每层只需 1 Pass，性能远低于原生双趟高斯。
|
||
// 4 次迭代 Dual Kawase ≈ 原生 6 次双趟高斯的扩散范围，但 Pass 数只有其 1/3。
|
||
public ClampedIntParameter diffusion = new(4, 1, 8);
|
||
|
||
[Header("Optimization (Mobile)")]
|
||
[Tooltip("初始降采样倍率（首趟大幅压缩分辨率）。\n1=原生半分辨率(1/2)；2=1/4分辨率；3=1/8分辨率。\n调大此值可用极低的迭代次数跑出巨大且柔和的光晕，大幅节省 GPU 带宽。")]
|
||
public ClampedIntParameter initialDownscaleShift = new(1, 1, 3);
|
||
|
||
[Tooltip("采样偏移跨度放大（Kernel Scale）。\n1.0=原生标准跨度。适度放大(如1.2-1.5)可拉扯光晕扩散范围，但过大会产生轻微十字星/方格马赛克。")]
|
||
public ClampedFloatParameter kernelScale = new(1.0f, 0.5f, 3.0f);
|
||
|
||
[Header("Tint")]
|
||
// tint 染色：C# 侧会归一化亮度（只保留色相/饱和度），与原生行为一致。
|
||
// 这样调整 tint 颜色不会意外改变 bloom 总亮度。
|
||
public ColorParameter tint = new(Color.white, true, true, true);
|
||
|
||
// RT 数组（仅 Down 金字塔 + Up 金字塔，数量与 diffusion 对齐）
|
||
private RTHandle[] _bloomMipDown;
|
||
private RTHandle[] _bloomMipUp;
|
||
private const int k_MaxMips = 8;
|
||
|
||
public override string GetShaderName() => "SLS/Postprocessing/AnimeBloom";
|
||
public override bool IsActive() => intensity.value > 0f;
|
||
|
||
public override void Render(CommandBuffer cmd, ref RenderingData renderingData, RTHandle source, RTHandle destination)
|
||
{
|
||
if (material == null) return;
|
||
|
||
var desc = renderingData.cameraData.cameraTargetDescriptor;
|
||
desc.msaaSamples = 1;
|
||
desc.depthBufferBits = 0;
|
||
|
||
// 【移动端极限优化】：强制降级到 32位 HDR 格式 (B10G11R11)，
|
||
// 相比主相机默认的 64位 R16G16B16A16，带宽消耗直接砍半，且肉眼几乎无损！
|
||
if (SystemInfo.IsFormatSupported(GraphicsFormat.B10G11R11_UFloatPack32, FormatUsage.Linear | FormatUsage.Render))
|
||
{
|
||
desc.graphicsFormat = GraphicsFormat.B10G11R11_UFloatPack32;
|
||
}
|
||
|
||
// ─────────────────────────────────────────────────────────────
|
||
// 1. 参数打包（完全对齐 Unity SetupBloom 的计算逻辑）
|
||
// ─────────────────────────────────────────────────────────────
|
||
|
||
// scatter: 用户 [0,1] → shader [0.05, 0.95]（与原生 Lerp 映射一致）
|
||
float scatterMapped = Mathf.Lerp(0.05f, 0.95f, scatter.value);
|
||
|
||
// threshold: Gamma → Linear（原生使用 GammaToLinearSpace 转换）
|
||
float thresholdLinear = Mathf.GammaToLinearSpace(threshold.value);
|
||
float thresholdKnee = thresholdLinear * 0.5f; // 硬编码 soft knee，与原生一致
|
||
|
||
material.SetVector(InternalShaderHelpers.ID._BloomScatterParams,
|
||
new Vector4(scatterMapped, clamp.value, thresholdLinear, thresholdKnee));
|
||
|
||
material.SetFloat(InternalShaderHelpers.ID._AnimeBloom_KernelScale, kernelScale.value);
|
||
|
||
// ─────────────────────────────────────────────────────────────
|
||
// 2. Tint 归一化（原生做法：亮度归一为 1，只携带色相/饱和度）
|
||
// 这样 tint 调整颜色时不会改变 bloom 总亮度。
|
||
// ─────────────────────────────────────────────────────────────
|
||
Color tintLinear = tint.value.linear;
|
||
float luma = 0.2126f * tintLinear.r + 0.7152f * tintLinear.g + 0.0722f * tintLinear.b;
|
||
Color tintNormalized = luma > 0f ? tintLinear * (1f / luma) : Color.white;
|
||
|
||
// 将 intensity 和归一化 tint 一并打包进 _BloomParams（原生 uber 方式）
|
||
material.SetVector(InternalShaderHelpers.ID._BloomParams,
|
||
new Vector4(intensity.value, tintNormalized.r, tintNormalized.g, tintNormalized.b));
|
||
|
||
// ─────────────────────────────────────────────────────────────
|
||
// 3. RT 数组初始化
|
||
// ─────────────────────────────────────────────────────────────
|
||
int mipCount = Mathf.Clamp(diffusion.value, 1, k_MaxMips);
|
||
|
||
if (_bloomMipDown == null || _bloomMipDown.Length != k_MaxMips)
|
||
{
|
||
_bloomMipDown = new RTHandle[k_MaxMips];
|
||
_bloomMipUp = new RTHandle[k_MaxMips];
|
||
}
|
||
|
||
// ─────────────────────────────────────────────────────────────
|
||
// 4. Prefilter（提取高亮像素 + 初始激进降采样）
|
||
// 根据用户选择，起始分辨率可能是 1/2, 1/4 或 1/8。
|
||
// ─────────────────────────────────────────────────────────────
|
||
int shift = initialDownscaleShift.value;
|
||
desc.width = Mathf.Max(1, desc.width >> shift);
|
||
desc.height = Mathf.Max(1, desc.height >> shift);
|
||
|
||
RenderingUtils.ReAllocateIfNeeded(ref _bloomMipDown[0], desc, FilterMode.Bilinear,
|
||
TextureWrapMode.Clamp, name: "_BloomDown0");
|
||
|
||
// Source → Down[0] via Pass 0 (Prefilter)
|
||
Blitter.BlitCameraTexture(cmd, source, _bloomMipDown[0], material, 0);
|
||
|
||
// ─────────────────────────────────────────────────────────────
|
||
// 5. Downsample Loop：Dual Kawase 降采样金字塔（Pass 1）
|
||
// 每次分辨率减半，产生多层不同尺度的模糊纹理
|
||
// ─────────────────────────────────────────────────────────────
|
||
for (int i = 1; i < mipCount; i++)
|
||
{
|
||
desc.width = Mathf.Max(1, desc.width >> 1);
|
||
desc.height = Mathf.Max(1, desc.height >> 1);
|
||
|
||
RenderingUtils.ReAllocateIfNeeded(ref _bloomMipDown[i], desc, FilterMode.Bilinear,
|
||
TextureWrapMode.Clamp, name: "_BloomDown" + i);
|
||
|
||
// Down[i-1] → Down[i] via Pass 1 (Dual Kawase Downsample)
|
||
Blitter.BlitCameraTexture(cmd, _bloomMipDown[i - 1], _bloomMipDown[i], material, 1);
|
||
}
|
||
|
||
// ─────────────────────────────────────────────────────────────
|
||
// 6. Upsample Loop：Dual Kawase 升采样 + lerp(high, low, scatter)
|
||
//
|
||
// 与 Unity 原生完全一致的数据流向：
|
||
// _BlitTexture = highMip = Down[i] （当前层高分辨率细节）
|
||
// _SourceTexLowMip = lowMip = Up[i+1] 或 Down[last]（低分辨率扩散光晕）
|
||
// output = Up[i]
|
||
//
|
||
// 从最底层开始，逐层往上合并，最终 Up[0] 就是完整的 bloom 纹理
|
||
// ─────────────────────────────────────────────────────────────
|
||
|
||
// 分配所有 Up RT（从最大 mip 往上，分辨率逐步翻倍）
|
||
// 由于 desc 已经在 downsample 中不断减半，我们要重新从 Down[i] 读尺寸
|
||
for (int i = mipCount - 1; i >= 0; i--)
|
||
{
|
||
RenderingUtils.ReAllocateIfNeeded(ref _bloomMipUp[i],
|
||
_bloomMipDown[i].rt.descriptor, FilterMode.Bilinear,
|
||
TextureWrapMode.Clamp, name: "_BloomUp" + i);
|
||
}
|
||
|
||
// 最底层：lowMip = Down[last]，highMip 也是 Down[last]（无 low 可用，直接 lerp 自身，结果仍 = Down[last]）
|
||
// 简化做法：直接将 Down[last] 拷到 Up[last] 作为起点
|
||
Blitter.BlitCameraTexture(cmd, _bloomMipDown[mipCount - 1], _bloomMipUp[mipCount - 1]);
|
||
|
||
// 从第二底层开始向上 upsample
|
||
for (int i = mipCount - 2; i >= 0; i--)
|
||
{
|
||
// highMip：当前层降采样纹理 Down[i]，作为 Blit 的 source（_BlitTexture）
|
||
// lowMip： 上一级升采样结果 Up[i+1]，通过 SetTexture 传入
|
||
cmd.SetGlobalTexture(InternalShaderHelpers.ID._SourceTexLowMip, _bloomMipUp[i + 1]);
|
||
Blitter.BlitCameraTexture(cmd, _bloomMipDown[i], _bloomMipUp[i], material, 2);
|
||
}
|
||
|
||
// ─────────────────────────────────────────────────────────────
|
||
// 7. Composite：将 bloom 叠加回原始画面（Pass 3）
|
||
// ─────────────────────────────────────────────────────────────
|
||
material.SetTexture(InternalShaderHelpers.ID._BloomTex, _bloomMipUp[0]);
|
||
Blitter.BlitCameraTexture(cmd, source, destination, material, 3);
|
||
}
|
||
|
||
public void Dispose()
|
||
{
|
||
if (_bloomMipDown == null) return;
|
||
for (int i = 0; i < _bloomMipDown.Length; i++)
|
||
{
|
||
_bloomMipDown[i]?.Release();
|
||
_bloomMipUp[i]?.Release();
|
||
}
|
||
}
|
||
}
|
||
} |