HeavenStudio/Assets/Scripts/PostProcessingVFX.cs

using Jukebox;
using System.Collections;
using System.Collections.Generic;
using UnityEngine;
using UnityEngine.Rendering.PostProcessing;

namespace HeavenStudio
{
    public class PostProcessingVFX : MonoBehaviour
    {
        private PostProcessVolume _volume;

        // events
        private List<RiqEntity> _vignettes = new();
        private List<RiqEntity> _cabbs = new();
        private List<RiqEntity> _blooms = new();
        private List<RiqEntity> _lensDs = new();
        private List<RiqEntity> _grains = new();
        private List<RiqEntity> _colorGradings = new();

        private void Awake()
        {
            _volume = GetComponent<PostProcessVolume>();
            UpdateRetroTV();
            UpdateAnalogNoise();
            UpdateSobelNeons();
            UpdatePixelizes();
        }

        private void Start()
        {
            GameManager.instance.onBeatChanged += OnBeatChanged;
        }

        public void OnBeatChanged(double beat)
        {
            _vignettes.Clear();
            _cabbs.Clear();
            _blooms.Clear();
            _lensDs.Clear();
            _grains.Clear();
            _colorGradings.Clear();
            _retroTvs.Clear();
            _scanJitters.Clear();
            _gaussBlurs.Clear();
            _analogNoises.Clear();
            _screenJumps.Clear();
            _sobelNeons.Clear();
            _pixelizeQuads.Clear();
            foreach (var entity in GameManager.instance.Beatmap.Entities)
            {
                switch (entity.datamodel)
                {
                    case "vfx/vignette":
                        _vignettes.Add(entity);
                        break;
                    case "vfx/cabb":
                        _cabbs.Add(entity);
                        break;
                    case "vfx/bloom":
                        _blooms.Add(entity);
                        break;
                    case "vfx/lensD":
                        _lensDs.Add(entity);
                        break;
                    case "vfx/grain":
                        _grains.Add(entity);
                        break;
                    case "vfx/colorGrading":
                        _colorGradings.Add(entity);
                        break;
                    case "vfx/retroTv":
                        _retroTvs.Add(entity);
                        break;
                    case "vfx/scanJitter":
                        _scanJitters.Add(entity);
                        break;
                    case "vfx/gaussBlur":
                        _gaussBlurs.Add(entity);
                        break;
                    case "vfx/analogNoise":
                        _analogNoises.Add(entity);
                        break;
                    case "vfx/screenJump":
                        _screenJumps.Add(entity);
                        break;
                    case "vfx/sobelNeon":
                        _sobelNeons.Add(entity);
                        break;
                    case "vfx/pixelQuad":
                        _pixelizeQuads.Add(entity);
                        break;
                    default: break;
                }
            }
            Update();
        }

        private void Update()
        {
            UpdateVignette();
            UpdateChromaticAbberations();
            UpdateBlooms();
            UpdateLensDistortions();
            UpdateGrain();
            UpdateColorGrading();
            UpdateRetroTV();
            UpdateScanJitter();
            UpdateGaussBlur();
            UpdateAnalogNoise();
            UpdateScreenJumps();
            UpdateSobelNeons();
            UpdatePixelizes();
        }

        private void UpdateVignette()
        {
            if (!_volume.profile.TryGetSettings<Vignette>(out var v)) return;

            v.enabled.Override(false);
            foreach (var e in _vignettes)
            {
                float normalized = Conductor.instance.GetPositionFromBeat(e.beat, e.length);
                if (normalized < 0) break;

                float clampNormal = Mathf.Clamp01(normalized);
                var func = Util.EasingFunction.GetEasingFunction((Util.EasingFunction.Ease)e["ease"]);

                float newIntensity = func(e["intenStart"], e["intenEnd"], clampNormal);

                v.enabled.Override(newIntensity != 0);
                if (!v.enabled) continue;
                v.rounded.Override(e["rounded"]);

                Color newColor = ColorEase(e["colorStart"], e["colorEnd"], clampNormal, func);

                v.color.Override(newColor);

                v.intensity.Override(newIntensity);

                float newSmoothness = func(e["smoothStart"], e["smoothEnd"], clampNormal);
                v.smoothness.Override(newSmoothness);

                float newRoundness = func(e["roundStart"], e["roundEnd"], clampNormal);
                v.roundness.Override(newRoundness);
            }
        }

        private void UpdateChromaticAbberations()
        {
            if (!_volume.profile.TryGetSettings<ChromaticAberration>(out var c)) return;
            c.enabled.Override(false);
            foreach (var e in _cabbs)
            {
                float normalized = Conductor.instance.GetPositionFromBeat(e.beat, e.length);
                if (normalized < 0) break;

                float clampNormal = Mathf.Clamp01(normalized);
                var func = Util.EasingFunction.GetEasingFunction((Util.EasingFunction.Ease)e["ease"]);

                float newIntensity = func(e["intenStart"], e["intenEnd"], clampNormal);
                c.enabled.Override(newIntensity != 0);
                if (!c.enabled) continue;
                c.intensity.Override(newIntensity);
            }
        }

        private void UpdateBlooms()
        {
            if (!_volume.profile.TryGetSettings<Bloom>(out var b)) return;
            b.enabled.Override(false);
            foreach (var e in _blooms)
            {
                float normalized = Conductor.instance.GetPositionFromBeat(e.beat, e.length);
                if (normalized < 0) break;

                float clampNormal = Mathf.Clamp01(normalized);
                var func = Util.EasingFunction.GetEasingFunction((Util.EasingFunction.Ease)e["ease"]);

                float newIntensity = func(e["intenStart"], e["intenEnd"], clampNormal);
                b.enabled.Override(newIntensity != 0);
                if (!b.enabled) continue;
                b.intensity.Override(newIntensity);

                Color newColor = ColorEase(e["colorStart"], e["colorEnd"], clampNormal, func);

                b.color.Override(newColor);

                float newThreshold = func(e["thresholdStart"], e["thresholdEnd"], clampNormal);
                b.threshold.Override(newThreshold);

                float newSoftKnee = func(e["softKneeStart"], e["softKneeEnd"], clampNormal);
                b.softKnee.Override(newSoftKnee);

                float newAna = func(e["anaStart"], e["anaEnd"], clampNormal);
                b.anamorphicRatio.Override(newAna);
            }
        }

        private void UpdateLensDistortions()
        {
            if (!_volume.profile.TryGetSettings<LensDistortion>(out var l)) return;
            l.enabled.Override(false);
            foreach (var e in _lensDs)
            {
                float normalized = Conductor.instance.GetPositionFromBeat(e.beat, e.length);
                if (normalized < 0) break;

                float clampNormal = Mathf.Clamp01(normalized);
                var func = Util.EasingFunction.GetEasingFunction((Util.EasingFunction.Ease)e["ease"]);

                float newIntensity = func(e["intenStart"], e["intenEnd"], clampNormal);
                l.enabled.Override(newIntensity != 0);
                if (!l.enabled) continue;
                l.intensity.Override(newIntensity);

                float newX = func(e["xStart"], e["xEnd"], clampNormal);
                l.intensityX.Override(newX);

                float newY = func(e["yStart"], e["yEnd"], clampNormal);
                l.intensityY.Override(newY);
            }
        }

        private void UpdateGrain()
        {
            if (!_volume.profile.TryGetSettings<Grain>(out var g)) return;
            g.enabled.Override(false);
            foreach (var e in _grains)
            {
                float normalized = Conductor.instance.GetPositionFromBeat(e.beat, e.length);
                if (normalized < 0) break;

                float clampNormal = Mathf.Clamp01(normalized);
                var func = Util.EasingFunction.GetEasingFunction((Util.EasingFunction.Ease)e["ease"]);

                float newIntensity = func(e["intenStart"], e["intenEnd"], clampNormal);
                g.enabled.Override(newIntensity != 0);
                if (!g.enabled) continue;
                g.intensity.Override(newIntensity);
                g.colored.Override(e["colored"]);

                float newSize = func(e["sizeStart"], e["sizeEnd"], clampNormal);
                g.size.Override(newSize);
            }
        }

        private void UpdateColorGrading()
        {
            if (!_volume.profile.TryGetSettings<ColorGrading>(out var c)) return;
            c.enabled.Override(false);
            foreach (var e in _colorGradings)
            {
                c.enabled.Override(true);
                float normalized = Conductor.instance.GetPositionFromBeat(e.beat, e.length);
                if (normalized < 0) break;

                float clampNormal = Mathf.Clamp01(normalized);
                var func = Util.EasingFunction.GetEasingFunction((Util.EasingFunction.Ease)e["ease"]);

                float newTemp = func(e["tempStart"], e["tempEnd"], clampNormal);
                c.temperature.Override(newTemp);

                float newTint = func(e["tintStart"], e["tintEnd"], clampNormal);
                c.tint.Override(newTint);

                Color newColor = ColorEase(e["colorStart"], e["colorEnd"], clampNormal, func);
                c.colorFilter.Override(newColor);

                float newHue = func(e["hueShiftStart"], e["hueShiftEnd"], clampNormal);
                c.hueShift.Override(newHue);

                float newSat = func(e["satStart"], e["satEnd"], clampNormal);
                c.saturation.Override(newSat);

                float newBright = func(e["brightStart"], e["brightEnd"], clampNormal);
                c.brightness.Override(newBright);

                float newCon = func(e["conStart"], e["conEnd"], clampNormal);
                c.contrast.Override(newCon);
            }
        }

        private void UpdateRetroTV()
        {
            if (!_volume.profile.TryGetSettings<CRT>(out var t)) return;
            t.enabled.Override(false);
            foreach (var e in _retroTvs)
            {
                float normalized = Conductor.instance.GetPositionFromBeat(e.beat, e.length);
                if (normalized < 0) break;

                float clampNormal = Mathf.Clamp01(normalized);
                var func = Util.EasingFunction.GetEasingFunction((Util.EasingFunction.Ease)e["ease"]);

                float newIntensity = func(e["intenStart"], e["intenEnd"], clampNormal);
                t.enabled.Override(newIntensity != 0);
                if (!t.enabled) continue;
                t.distort.Override(newIntensity);

                float newRGBBlend = func(e["rgbStart"], e["rgbEnd"], clampNormal);
                t.RGBBlend.Override(newRGBBlend);

                float newBottomCollapse = func(e["bottomStart"], e["bottomEnd"], clampNormal);
                t.BottomCollapse.Override(newBottomCollapse);

                float newNoiseAmount = func(e["noiseStart"], e["noiseEnd"], clampNormal);
                t.NoiseAmount.Override(newNoiseAmount);
            }
        }

        private void UpdateScanJitter()
        {
            if (!_volume.profile.TryGetSettings<GlitchScanLineJitter>(out var j)) return;
            j.enabled.Override(false);
            foreach (var e in _scanJitters)
            {
                float normalized = Conductor.instance.GetPositionFromBeat(e.beat, e.length);
                if (normalized < 0) break;

                float clampNormal = Mathf.Clamp01(normalized);
                var func = Util.EasingFunction.GetEasingFunction((Util.EasingFunction.Ease)e["ease"]);

                float newIntensity = func(e["intenStart"], e["intenEnd"], clampNormal);
                j.enabled.Override(newIntensity != 0);
                if (!j.enabled) continue;
                j.JitterIndensity.Override(newIntensity);
            }
        }

        private void UpdateGaussBlur()
        {
            if (!_volume.profile.TryGetSettings<GaussianBlur>(out var g)) return;
            g.enabled.Override(false);
            foreach (var e in _gaussBlurs)
            {
                float normalized = Conductor.instance.GetPositionFromBeat(e.beat, e.length);
                if (normalized < 0) break;

                float clampNormal = Mathf.Clamp01(normalized);
                var func = Util.EasingFunction.GetEasingFunction((Util.EasingFunction.Ease)e["ease"]);

                float newIntensity = func(e["intenStart"], e["intenEnd"], clampNormal);
                g.enabled.Override(newIntensity != 0);
                if (!g.enabled) continue;
                g.BlurRadius.Override(newIntensity);
            }
        }

        private void UpdateAnalogNoise()
        {
            if (!_volume.profile.TryGetSettings<GlitchAnalogNoise>(out var n)) return;
            n.enabled.Override(false);
            foreach (var e in _analogNoises)
            {
                float normalized = Conductor.instance.GetPositionFromBeat(e.beat, e.length);
                if (normalized < 0) break;

                float clampNormal = Mathf.Clamp01(normalized);
                var func = Util.EasingFunction.GetEasingFunction((Util.EasingFunction.Ease)e["ease"]);

                float newIntensity = func(e["intenStart"], e["intenEnd"], clampNormal);
                n.enabled.Override(newIntensity != 0);
                if (!n.enabled) continue;
                n.NoiseSpeed.Override(newIntensity);

                float newFading = func(e["fadingStart"], e["fadingEnd"], clampNormal);
                n.NoiseFading.Override(newFading);

                float newThreshold = func(e["thresholdStart"], e["thresholdEnd"], clampNormal);
                n.LuminanceJitterThreshold.Override(newThreshold);
            }
        }

        private void UpdateScreenJumps()
        {
            if (!_volume.profile.TryGetSettings<GlitchScreenJump>(out var sj)) return;
            sj.enabled.Override(false);
            foreach (var e in _screenJumps)
            {
                float normalized = Conductor.instance.GetPositionFromBeat(e.beat, e.length);
                if (normalized < 0) break;

                float clampNormal = Mathf.Clamp01(normalized);
                var func = Util.EasingFunction.GetEasingFunction((Util.EasingFunction.Ease)e["ease"]);

                float newIntensity = func(e["intenStart"], e["intenEnd"], clampNormal);
                sj.enabled.Override(newIntensity != 0);
                if (!sj.enabled) continue;
                sj.ScreenJumpIndensity.Override(newIntensity);
            }
        }

        private void UpdateSobelNeons()
        {
            if (!_volume.profile.TryGetSettings<EdgeDetectionSobelNeonV2>(out var sn)) return;
            sn.enabled.Override(false);
            foreach (var e in _sobelNeons)
            {
                float normalized = Conductor.instance.GetPositionFromBeat(e.beat, e.length);
                if (normalized < 0) break;

                float clampNormal = Mathf.Clamp01(normalized);
                var func = Util.EasingFunction.GetEasingFunction((Util.EasingFunction.Ease)e["ease"]);

                float newIntensity = func(e["intenStart"], e["intenEnd"], clampNormal);
                sn.enabled.Override(newIntensity != 0.1);
                if (!sn.enabled) continue;
                sn.EdgeNeonFade.Override(newIntensity);

                float newEdgeWidth = func(e["edgeWidthStart"], e["edgeWidthEnd"], clampNormal);
                sn.EdgeWidth.Override(newEdgeWidth);

                float newBgFade = func(e["bgFadeStart"], e["bgFadeEnd"], clampNormal);
                sn.BackgroundFade.Override(newBgFade);
                
                float newBrightness = func(e["brightnessStart"], e["brightnessEnd"], clampNormal);
                sn.Brigtness.Override(newBrightness);
            }
        }

        private void UpdatePixelizes()
        {
            if (!_volume.profile.TryGetSettings<PixelizeQuad>(out var pq)) return;
            pq.enabled.Override(false);
            foreach (var e in _pixelizeQuads)
            {
                float normalized = Conductor.instance.GetPositionFromBeat(e.beat, e.length);
                if (normalized < 0) break;

                float clampNormal = Mathf.Clamp01(normalized);
                var func = Util.EasingFunction.GetEasingFunction((Util.EasingFunction.Ease)e["ease"]);

                float newPixelSize = func(e["pixelSizeStart"], e["pixelSizeEnd"], clampNormal);
                pq.enabled.Override(newPixelSize != 0);
                if (!pq.enabled) continue;
                pq.pixelSize.Override(newPixelSize);

                float newPixelRatio = func(e["ratioStart"], e["ratioEnd"], clampNormal);
                if (!pq.enabled) continue;
                pq.pixelRatio.Override(newPixelRatio);

                float newPixelXScale = func(e["xScaleStart"], e["xScaleEnd"], clampNormal);
                if (!pq.enabled) continue;
                pq.pixelScaleX.Override(newPixelXScale);

                float newPixelYScale = func(e["yScaleStart"], e["yScaleEnd"], clampNormal);
                if (!pq.enabled) continue;
                pq.pixelScaleY.Override(newPixelYScale);
            }
        }

        private Color ColorEase(Color start, Color end, float time, Util.EasingFunction.Function func)
        {
            float newR = func(start.r, end.r, time);
            float newG = func(start.g, end.g, time);
            float newB = func(start.b, end.b, time);

            return new Color(newR, newG, newB, 1);
        }
    }
}