Tetra/Minesweeper/Assets/Krivodeling/UI/Effects/Blur/Scripts/UIBlur.cs

using System;
using System.Collections;
using UnityEditor;
using UnityEngine;
using UnityEngine.Events;
using UnityEngine.UI;

namespace Krivodeling.UI.Effects
{
    public class UIBlur : MonoBehaviour
    {
        public Color Color { get => _color; set { _color = value; UpdateColor(); } }
#if UNITY_EDITOR
        /// <summary>
        /// (Editor Only).
        /// </summary>
        public FlipMode EditorFlipMode { get => _editorFlipMode; set { _editorFlipMode = value; UpdateFlipMode(); } }
#endif
        public FlipMode BuildFlipMode { get => _buildFlipMode; set { _buildFlipMode = value; UpdateFlipMode(); } }
        public float Intensity { get => _intensity; set { _intensity = Mathf.Clamp01(value); UpdateIntensity(); } }
        public float Multiplier { get => _multiplier; set { _multiplier = Mathf.Clamp01(value); UpdateMultiplier(); } }
        public UnityEvent OnBeginBlur { get => _onBeginBlur; set => _onBeginBlur = value; }
        public UnityEvent OnEndBlur { get => _onEndBlur; set => _onEndBlur = value; }
        public BlurChangedEvent OnBlurChanged { get => _onBlurChanged; set => _onBlurChanged = value; }

        [SerializeField]
        private Color _color = Color.white;
#if UNITY_EDITOR
        [SerializeField]
        private FlipMode _editorFlipMode;
#endif
        [SerializeField]
        private FlipMode _buildFlipMode;
        [SerializeField, Range(0f, 1f)]
        private float _intensity;
        [SerializeField, Range(0f, 1f)]
        private float _multiplier = 0.15f;
        [SerializeField]
        private UnityEvent _onBeginBlur;
        [SerializeField]
        private UnityEvent _onEndBlur;
        [SerializeField]
        private BlurChangedEvent _onBlurChanged;

        private Material _material;
        private int _colorId;
        private int _flipXId;
        private int _flipYId;
        private int _intensityId;
        private int _multiplierId;

        public void UpdateBlur()
        {
            SetBlur(Color, Intensity, Multiplier);
            UpdateFlipMode();
        }

        public void SetBlur(Color color, float intensity, float multiplier)
        {
            Color = color;
            Intensity = intensity;
            Multiplier = multiplier;
        }

        public void BeginBlur(float speed)
        {
            StopAllCoroutines();
            StartCoroutine(BeginBlurCoroutine(speed));
        }

        public void EndBlur(float speed)
        {
            StopAllCoroutines();
            StartCoroutine(EndBlurCoroutine(speed));
        }

        private void Start()
        {
            SetComponents();
            SetBlur(Color, Intensity, _multiplier);
        }

        private void SetComponents()
        {
            _material = FindMaterial();
            _colorId = Shader.PropertyToID("_Color");
            _flipXId = Shader.PropertyToID("_FlipX");
            _flipYId = Shader.PropertyToID("_FlipY");
            _intensityId = Shader.PropertyToID("_Intensity");
            _multiplierId = Shader.PropertyToID("_Multiplier");
        }

        private Material FindMaterial()
        {
            Material material = GetComponent<Image>().material;

            if (material == null)
                material = GetComponent<Renderer>().material;

            if (material == null)
                throw new NullReferenceException("Material not found");

            return material;
        }

        private void UpdateColor()
        {
            _material.SetColor(_colorId, Color);
        }

        private void UpdateIntensity()
        {
            _material.SetFloat(_intensityId, Intensity);
        }

        private void UpdateMultiplier()
        {
            _material.SetFloat(_multiplierId, Multiplier);
        }

        private void UpdateFlipMode()
        {
#if UNITY_EDITOR
            if (EditorFlipMode.HasFlag(FlipMode.X))
                _material.SetFloat(_flipXId, 1f);
            else
                _material.SetFloat(_flipXId, 0f);

            if (EditorFlipMode.HasFlag(FlipMode.Y))
                _material.SetFloat(_flipYId, 1f);
            else
                _material.SetFloat(_flipYId, 0f);
#else
            if (BuildFlipMode.HasFlag(FlipMode.X))
                _material.SetFloat(_flipXId, 1f);
            else
                _material.SetFloat(_flipXId, 0f);

            if (BuildFlipMode.HasFlag(FlipMode.Y))
                _material.SetFloat(_flipYId, 1f);
            else
                _material.SetFloat(_flipYId, 0f);
#endif
        }

        private IEnumerator BeginBlurCoroutine(float speed)
        {
            OnBeginBlur?.Invoke();

            while (Intensity < 1f)
            {
                Intensity += speed * Time.deltaTime;

                UpdateIntensity();

                OnBlurChanged.Invoke(Intensity);

                yield return null;
            }
        }

        private IEnumerator EndBlurCoroutine(float speed)
        {
            while (Intensity > 0f)
            {
                Intensity -= speed * Time.deltaTime;

                UpdateIntensity();

                OnBlurChanged.Invoke(Intensity);

                yield return null;
            }

            OnEndBlur?.Invoke();
        }

        [Serializable]
        public class BlurChangedEvent : UnityEvent<float> { }

        #region Editor
#if UNITY_EDITOR
        private void OnValidate()
        {
            UpdateBlurInEditor();
        }

        private void UpdateBlurInEditor()
        {
            Material material = FindMaterial();

            material.SetColor("_Color", Color);
            material.SetFloat("_FlipX", EditorFlipMode.HasFlag(FlipMode.X) ? 1f : 0f);
            material.SetFloat("_FlipY", EditorFlipMode.HasFlag(FlipMode.Y) ? 1f : 0f);
            material.SetFloat("_Intensity", Intensity);
            material.SetFloat("_Multiplier", Multiplier);

            EditorUtility.SetDirty(material);
        }
#endif
        #endregion
    }
}