ScriptableRenderPipeline/ScriptableRenderPipeline/Core/Volume/VolumeManager.cs

using System;
using System.Collections.Generic;
using System.Linq;
using UnityEngine.Assertions;

namespace UnityEngine.Experimental.Rendering
{
    public sealed class VolumeManager
    {
        //>>> System.Lazy<T> is broken in Unity (legacy runtime) so we'll have to do it ourselves :|
        static volatile VolumeManager s_Instance;
        static object s_LockObj = new Object();

        public static VolumeManager instance
        {
            get
            {
                // Double-lock checking
                if (s_Instance == null)
                {
                    lock (s_LockObj) // Lock on a separate object to avoid deadlocks
                    {
                        if (s_Instance == null)
                            s_Instance = new VolumeManager();
                    }
                }

                return s_Instance;
            }
        }
        //<<<

        // Max amount of layers available in Unity
        const int k_MaxLayerCount = 32;

        // List of all volumes (sorted by priority) per layer
        readonly List<Volume>[] m_Volumes;

        // Keep track of sorting states for all layers
        readonly bool[] m_SortNeeded;

        // Internal state of all component types
        readonly Dictionary<Type, VolumeComponent> m_Components;

        // Internal list of default state for each component type - this is used to reset component
        // states on update instead of having to implement a Reset method on all components (which
        // would be error-prone)
        readonly List<VolumeComponent> m_ComponentsDefaultState;

        // Recycled list used for volume traversal
        readonly List<Collider> m_TempColliders;

        // In the editor, when entering play-mode, it will call the constructor and OnEditorReload()
        // which in turn will call ReloadBaseTypes() twice, so we need to keep track of the reloads
        // to avoid wasting any more CPU than required
        static bool s_StopReloads = false;

        VolumeManager()
        {
            m_Volumes = new List<Volume>[k_MaxLayerCount];
            m_SortNeeded = new bool[k_MaxLayerCount];
            m_TempColliders = new List<Collider>(8);
            m_Components = new Dictionary<Type, VolumeComponent>();
            m_ComponentsDefaultState = new List<VolumeComponent>();

            ReloadBaseTypes();
        }

#if UNITY_EDITOR
        // Called every time Unity recompiles scripts in the editor. We need this to keep track of
        // any new custom component the user might add to the project.
        [UnityEditor.Callbacks.DidReloadScripts]
        static void OnEditorReload()
        {
            if (!s_StopReloads)
                instance.ReloadBaseTypes();

            s_StopReloads = false;
        }
#endif

        // This will be called only once at runtime and everytime script reload kicks-in in the
        // editor as we need to keep track of any compatible component in the project
        void ReloadBaseTypes()
        {
            // Clean component map & default states
            foreach (var component in m_Components)
                CoreUtils.Destroy(component.Value);

            foreach (var component in m_ComponentsDefaultState)
                CoreUtils.Destroy(component);

            m_Components.Clear();
            m_ComponentsDefaultState.Clear();

            // Rebuild it from scratch
            var types = AppDomain.CurrentDomain.GetAssemblies()
                            .SelectMany(
                                a => a.GetTypes()
                                .Where(
                                    t => t.IsSubclassOf(typeof(VolumeComponent))
                                )
                            );

            foreach (var type in types)
            {
                // We need two instances, one for global state tracking and another one to keep a
                // default state that will act as the lowest priority global volume (so that we have
                // a state to fallback to when exiting volumes)
                var inst = (VolumeComponent)ScriptableObject.CreateInstance(type);
                m_Components.Add(type, inst);

                inst = (VolumeComponent)ScriptableObject.CreateInstance(type);
                inst.SetAllOverridesTo(true);
                m_ComponentsDefaultState.Add(inst);
            }

            s_StopReloads = true;
        }

        public T GetComponent<T>()
            where T : VolumeComponent
        {
            var comp = GetComponent(typeof(T));
            return (T)comp;
        }

        public VolumeComponent GetComponent(Type type)
        {
            VolumeComponent comp;
            m_Components.TryGetValue(type, out comp);
            Assert.IsNotNull(comp, "Component map is corrupted, \"" + type + "\" not found");
            return comp;
        }

        public void Register(Volume volume, int layer)
        {
            var volumes = m_Volumes[layer];

            if (volumes == null)
            {
                volumes = new List<Volume>();
                m_Volumes[layer] = volumes;
            }

            Assert.IsFalse(volumes.Contains(volume), "Volume has already been registered");
            volumes.Add(volume);
            SetLayerDirty(layer);
        }

        public void Unregister(Volume volume, int layer)
        {
            var volumes = m_Volumes[layer];

            if (volumes == null)
                return;

            volumes.Remove(volume);
        }

        internal void SetLayerDirty(int layer)
        {
            Assert.IsTrue(layer >= 0 && layer <= k_MaxLayerCount, "Invalid layer bit");
            m_SortNeeded[layer] = true;
        }

        internal void UpdateVolumeLayer(Volume volume, int prevLayer, int newLayer)
        {
            Assert.IsTrue(prevLayer >= 0 && prevLayer <= k_MaxLayerCount, "Invalid layer bit");
            Unregister(volume, prevLayer);
            Register(volume, newLayer);
        }

        // Go through all listed components and lerp overriden values in the global state
        void OverrideData(List<VolumeComponent> components, float interpFactor)
        {
            foreach (var component in components)
            {
                if (!component.active)
                    continue;

                var target = GetComponent(component.GetType());
                int count = component.parameters.Count;

                for (int i = 0; i < count; i++)
                {
                    var toParam = component.parameters[i];
                    if (toParam.overrideState)
                    {
                        var fromParam = target.parameters[i];
                        fromParam.Interp(fromParam, toParam, interpFactor);
                    }
                }
            }
        }

        // Faster version of OverrideData to force replace values in the global state
        void ReplaceData(List<VolumeComponent> components)
        {
            foreach (var component in components)
            {
                var target = GetComponent(component.GetType());
                int count = component.parameters.Count;

                for (int i = 0; i < count; i++)
                    target.parameters[i].SetValue(component.parameters[i]);
            }
        }

        // Update the global state - should be called once per frame in the update loop before
        // anything else
        public void Update(Transform trigger, LayerMask layerMask)
        {
#if UNITY_EDITOR
            // Editor specific hack to work around serialization doing funky things when exiting
            // play mode -> re-create the world when bad things happen
            if (m_ComponentsDefaultState == null
            || (m_ComponentsDefaultState.Count > 0 && m_ComponentsDefaultState[0] == null))
            {
                ReloadBaseTypes();
            }
            else
#endif
            {
                // Start by resetting the global state to default values
                ReplaceData(m_ComponentsDefaultState);
            }

            // Do magic
            bool onlyGlobal = trigger == null;
            int mask = layerMask.value;
            var triggerPos = onlyGlobal ? Vector3.zero : trigger.position;

            for (int i = 0; i < k_MaxLayerCount; i++)
            {
                // Skip layers not in the mask
                if ((mask & (1 << i)) == 0)
                    continue;

                // Skip empty layers
                var volumes = m_Volumes[i];

                if (volumes == null)
                    continue;

                // Sort the volume list if needed
                if (m_SortNeeded[i])
                {
                    SortByPriority(volumes);
                    m_SortNeeded[i] = false;
                }

                // Traverse all volumes
                foreach (var volume in volumes)
                {
                    // Skip disabled volumes and volumes without any data or weight
                    if (!volume.enabled || volume.weight <= 0f)
                        continue;

                    var components = volume.components;

                    // Global volumes always have influence
                    if (volume.isGlobal)
                    {
                        OverrideData(components, Mathf.Clamp01(volume.weight));
                        continue;
                    }

                    if (onlyGlobal)
                        continue;

                    // If volume isn't global and has no collider, skip it as it's useless
                    var colliders = m_TempColliders;
                    volume.GetComponents(colliders);
                    if (colliders.Count == 0)
                        continue;

                    // Find closest distance to volume, 0 means it's inside it
                    float closestDistanceSqr = float.PositiveInfinity;

                    foreach (var collider in colliders)
                    {
                        if (!collider.enabled)
                            continue;

                        var closestPoint = collider.ClosestPoint(triggerPos);
                        var d = (closestPoint - triggerPos).sqrMagnitude;

                        if (d < closestDistanceSqr)
                            closestDistanceSqr = d;
                    }

                    colliders.Clear();
                    float blendDistSqr = volume.blendDistance * volume.blendDistance;

                    // Volume has no influence, ignore it
                    // Note: Volume doesn't do anything when `closestDistanceSqr = blendDistSqr` but
                    //       we can't use a >= comparison as blendDistSqr could be set to 0 in which
                    //       case volume would have total influence
                    if (closestDistanceSqr > blendDistSqr)
                        continue;

                    // Volume has influence
                    float interpFactor = 1f;

                    if (blendDistSqr > 0f)
                        interpFactor = 1f - (closestDistanceSqr / blendDistSqr);

                    // No need to clamp01 the interpolation factor as it'll always be in [0;1[ range
                    OverrideData(components, interpFactor * Mathf.Clamp01(volume.weight));
                }
            }
        }

        // Stable insertion sort. Faster than List<T>.Sort() for our needs.
        static void SortByPriority(List<Volume> volumes)
        {
            Assert.IsNotNull(volumes, "Trying to sort volumes of non-initialized layer");

            for (int i = 1; i < volumes.Count; i++)
            {
                var temp = volumes[i];
                int j = i - 1;

                // Sort order is ascending
                while (j >= 0 && volumes[j].priority > temp.priority)
                {
                    volumes[j + 1] = volumes[j];
                    j--;
                }

                volumes[j + 1] = temp;
            }
        }
    }
}