new-horizons/NewHorizons/Components/Props/NHSupernovaPlanetEffectController.cs

using NewHorizons.Components.SizeControllers;
using NewHorizons.Handlers;
using UnityEngine;
using static SupernovaPlanetEffectController;

namespace NewHorizons.Components.Props
{
    public class NHSupernovaPlanetEffectController : MonoBehaviour
    {
        public Light[] _ambientLight;
        public float[] _ambientLightOrigIntensity;
        public LODGroup _atmosphere;
        public Renderer _atmosphereRenderer;
        public float _atmosphereOrigSunIntensity = 1;
        public PlanetaryFogController _fog;
        public Renderer _fogImpostor;
        public Color _fogOrigTint;

        public MeshRenderer _shockLayer;
        public static readonly Color _shockLayerDefaultColor = new Color(0.3569f, 0.7843f, 1f, 1f);

        public Color _shockLayerColor = _shockLayerDefaultColor;

        public float shockLayerStartRadius = 1000f;
        public float shockLayerFullRadius = 10000f;
        public float shockLayerTrailLength = 300f;
        public float shockLayerTrailFlare = 100f;

        private LOD[] _atmosphereLODs;

        // Only one of StarEvolutionController or SunController can be set at once
        public StarEvolutionController StarEvolutionController
        {
            get => _starEvolutionController;
            set
            {
                _starEvolutionController = value;
                _sunController = null;
            }
        }
        private StarEvolutionController _starEvolutionController;

        public SunController SunController
        {
            get => _sunController;
            set
            {
                _sunController = value;
                _starEvolutionController = null;
            }
        }
        private SunController _sunController;

        public void Awake()
        {
            if (s_matPropBlock_Atmosphere == null)
            {
                s_matPropBlock_Atmosphere = new MaterialPropertyBlock();
                s_propID_SunIntensity = Shader.PropertyToID("_SunIntensity");
                s_propID_Tint = Shader.PropertyToID("_Tint");
                s_matPropBlock_ShockLayer = new MaterialPropertyBlock();
                s_propID_Color = Shader.PropertyToID("_Color");
                s_propID_WorldToLocalShockMatrix = Shader.PropertyToID("_WorldToShockLocalMatrix");
                s_propID_Dir = Shader.PropertyToID("_Dir");
                s_propID_Length = Shader.PropertyToID("_Length");
                s_propID_Flare = Shader.PropertyToID("_Flare");
                s_propID_TrailFade = Shader.PropertyToID("_TrailFade");
                s_propID_GradientLerp = Shader.PropertyToID("_GradientLerp");
                s_propID_MainTex_ST = Shader.PropertyToID("_MainTex_ST");
            }
        }

        public void Start()
        {
            SupernovaEffectHandler.RegisterPlanetEffect(this);
            if (_atmosphere != null) _atmosphereLODs = _atmosphere.GetLODs();
            if (_fog != null)
            {
                _fogOrigTint = _fog.fogTint;
                _fogImpostor = _fog.fogImpostor;
            }
            else if (_fogImpostor != null)
            {
                _fogOrigTint = _fogImpostor.material.GetColor(s_propID_Tint);
            }
            if (_shockLayer != null) _shockLayer.enabled = false;
        }

        public void OnDestroy()
        {
            SupernovaEffectHandler.UnregisterPlanetEffect(this);
        }

        public void Enable()
        {
            enabled = true;
        }

        public void Disable()
        {
            enabled = false;
            if (_shockLayer != null) _shockLayer.enabled = false;
        }

        public void Update()
        {
            SupernovaEffectHandler.GetNearestStarSupernova(this);
            if (StarEvolutionController != null)
            {
                if (StarEvolutionController.HasSupernovaStarted())
                {
                    if (_shockLayer != null)
                    {
                        if (!_shockLayer.enabled) _shockLayer.enabled = true;
                        Vector3 dir = Vector3.Normalize(transform.position - StarEvolutionController.transform.position);
                        s_matPropBlock_ShockLayer.SetColor(s_propID_Color, StarEvolutionController.supernovaColour != null ? StarEvolutionController.supernovaColour.ToColor() : _shockLayerColor);
                        s_matPropBlock_ShockLayer.SetMatrix(s_propID_WorldToLocalShockMatrix, Matrix4x4.TRS(transform.position, Quaternion.LookRotation(dir, Vector3.up), Vector3.one).inverse);
                        s_matPropBlock_ShockLayer.SetVector(s_propID_Dir, dir);
                        s_matPropBlock_ShockLayer.SetFloat(s_propID_Length, shockLayerTrailLength);
                        s_matPropBlock_ShockLayer.SetFloat(s_propID_Flare, shockLayerTrailFlare);
                        s_matPropBlock_ShockLayer.SetFloat(s_propID_TrailFade, 1f - Mathf.InverseLerp(shockLayerStartRadius, shockLayerFullRadius, StarEvolutionController.GetSupernovaRadius()));
                        s_matPropBlock_ShockLayer.SetFloat(s_propID_GradientLerp, 0);
                        s_matPropBlock_ShockLayer.SetVector(s_propID_MainTex_ST, _shockLayer.sharedMaterial.GetVector(s_propID_MainTex_ST) with { w = -Time.timeSinceLevelLoad });
                        _shockLayer.SetPropertyBlock(s_matPropBlock_ShockLayer);
                    }
                }
                if (StarEvolutionController.IsCollapsing())
                {
                    float collapseProgress = StarEvolutionController.GetCollapseProgress();

                    if (_ambientLight != null)
                    {
                        for (int i = 0; i < _ambientLight.Length; i++)
                        {
                            _ambientLight[i].intensity = _ambientLightOrigIntensity[i] * (1f - collapseProgress);
                        }
                    }

                    if (_atmosphere != null)
                    {
                        s_matPropBlock_Atmosphere.SetFloat(s_propID_SunIntensity, _atmosphereOrigSunIntensity * (1f - collapseProgress));

                        foreach (var lod in _atmosphereLODs)
                            foreach (var renderer in lod.renderers)
                                renderer.SetPropertyBlock(s_matPropBlock_Atmosphere);
                    }

                    if (_atmosphereRenderer != null)
                    {
                        s_matPropBlock_Atmosphere.SetFloat(s_propID_SunIntensity, _atmosphereOrigSunIntensity * (1f - collapseProgress));

                        _atmosphereRenderer.SetPropertyBlock(s_matPropBlock_Atmosphere);
                    }

                    if (_fog != null) _fog.fogTint = Color.Lerp(_fogOrigTint, Color.black, collapseProgress);

                    if (_fogImpostor != null) _fogImpostor.material.SetColor(s_propID_Tint, Color.Lerp(_fogOrigTint, Color.black, collapseProgress));
                }
                else
                {
                    if (_shockLayer != null) _shockLayer.enabled = false;
                }
            }
            else if (SunController != null)
            {
                if (SunController.HasSupernovaStarted())
                {
                    if (_shockLayer != null)
                    {
                        if (!_shockLayer.enabled) _shockLayer.enabled = true;
                        Vector3 dir = Vector3.Normalize(transform.position - SunController.transform.position);
                        s_matPropBlock_ShockLayer.SetColor(s_propID_Color, _shockLayerColor);
                        s_matPropBlock_ShockLayer.SetMatrix(s_propID_WorldToLocalShockMatrix, Matrix4x4.TRS(transform.position, Quaternion.LookRotation(dir, Vector3.up), Vector3.one).inverse);
                        s_matPropBlock_ShockLayer.SetVector(s_propID_Dir, dir);
                        s_matPropBlock_ShockLayer.SetFloat(s_propID_Length, shockLayerTrailLength);
                        s_matPropBlock_ShockLayer.SetFloat(s_propID_Flare, shockLayerTrailFlare);
                        s_matPropBlock_ShockLayer.SetFloat(s_propID_TrailFade, 1f - Mathf.InverseLerp(shockLayerStartRadius, shockLayerFullRadius, SunController.GetSupernovaRadius()));
                        s_matPropBlock_ShockLayer.SetFloat(s_propID_GradientLerp, 0);
                        s_matPropBlock_ShockLayer.SetVector(s_propID_MainTex_ST, _shockLayer.sharedMaterial.GetVector(s_propID_MainTex_ST) with { w = -Time.timeSinceLevelLoad });
                        _shockLayer.SetPropertyBlock(s_matPropBlock_ShockLayer);
                    }
                }
                else if (SunController._collapseStarted)
                {
                    float collapseProgress = SunController.GetCollapseProgress();

                    if (_ambientLight != null && _ambientLightOrigIntensity != null)
                    {
                        for (int i = 0; i < _ambientLight.Length; i++)
                        {
                            var ambientLight = _ambientLight[i];
                            if (ambientLight != null)
                            {
                                ambientLight.intensity = _ambientLightOrigIntensity[i] * (1f - collapseProgress);
                            }
                        }
                    }

                    if (_atmosphere != null)
                    {
                        s_matPropBlock_Atmosphere.SetFloat(s_propID_SunIntensity, 1f - collapseProgress);

                        foreach (var lod in _atmosphereLODs)
                            foreach (var renderer in lod.renderers)
                                renderer.SetPropertyBlock(s_matPropBlock_Atmosphere);
                    }

                    if (_atmosphereRenderer != null)
                    {
                        s_matPropBlock_Atmosphere.SetFloat(s_propID_SunIntensity, 1f - collapseProgress);

                        _atmosphereRenderer.SetPropertyBlock(s_matPropBlock_Atmosphere);
                    }

                    if (_fog != null) _fog.fogTint = Color.Lerp(_fogOrigTint, Color.black, collapseProgress);

                    if (_fogImpostor != null) _fogImpostor.material.SetColor(s_propID_Tint, Color.Lerp(_fogOrigTint, Color.black, collapseProgress));
                }
                else
                {
                    if (_shockLayer != null) _shockLayer.enabled = false;
                }
            }
            else
            {
                if (_shockLayer != null) _shockLayer.enabled = false;

                if (_ambientLight != null && _ambientLightOrigIntensity != null)
                {
                    for (int i = 0; i < _ambientLight.Length; i++)
                    {
                        if (_ambientLight[i] != null)
                        {
                            _ambientLight[i].intensity = _ambientLightOrigIntensity[i];
                        }
                    }
                }

                if (_fog != null) _fog.fogTint = _fogOrigTint;

                if (_fogImpostor != null) _fogImpostor.material.SetColor(s_propID_Tint, _fogOrigTint);
            }
        }
    }
}