OldBlueWater/BlueWater/Assets/NWH/Dynamic Water Physics 2/Scripts/WaterObject/WaterEffects/WaterParticleSystem.cs

using System;
using NWH.DWP2.WaterData;
using NWH.Common.Utility;
using UnityEngine;
using UnityEngine.Serialization;
using Random = UnityEngine.Random;

namespace NWH.DWP2.WaterObjects
{
    /// <summary>
    ///     Class for generating water particles based on simulation data.
    /// </summary>
    [RequireComponent(typeof(ParticleSystem))]
    public class WaterParticleSystem : MonoBehaviour
    {
        public WaterObject ReferenceWaterObject;
        
        /// <summary>
        ///     Should the particle system emit?
        /// </summary>
        [UnityEngine.Tooltip("    Should the particle system emit?")]
        public bool emit = true;
        
        /// <summary>
        ///     Render queue of the particle material.
        /// </summary>
        [UnityEngine.Tooltip("    Render queue of the particle material.")]
        public int renderQueue = 2700;
        
        /// <summary>
        ///     Elevation above water at which the particles will spawn. Used to avoid clipping.
        /// </summary>
        [Tooltip("Elevation above water at which the particles will spawn. Used to avoid clipping.")]
        [Range(0f, 0.1f)] public float surfaceElevation = 0.016f;
        
        /// <summary>
        ///     Initial size of the particle.
        /// </summary>
        [Tooltip("Initial size of the particle.")]
        [Range(0f, 64f)] public float startSize = 4f;
        
        /// <summary>
        ///     Velocity object has to have to emit particles.
        /// </summary>
        [FormerlySerializedAs("sleepTresholdVelocity")]
        [Tooltip("Velocity object has to have to emit particles.")]
        [Range(0.1f, 5f)] public float sleepThresholdVelocity = 1.5f;
        
        /// <summary>
        ///     Determines how much velocity of the object will affect initial particle speed.
        /// </summary>
        [Tooltip("Determines how much velocity of the object will affect initial particle speed.")]
        [Range(0f, 5f)] public float initialVelocityModifier = 0.01f;
        
        /// <summary>
        ///     Limit initial alpha to this value.
        /// </summary>
        [Tooltip("Limit initial alpha to this value.")]
        [Range(0f, 1f)] public float maxInitialAlpha = 0.15f;
        
        /// <summary>
        ///     Multiplies initial alpha by this value. Alpha cannot be higher than maxInitialAlpha.
        /// </summary>
        [Tooltip("Multiplies initial alpha by this value. Alpha cannot be higher than maxInitialAlpha.")]
        [Range(0f, 10f)] public float initialAlphaModifier = 0.4f;
        
        /// <summary>
        ///     How many particles should be emitted each 'emitTimeInterval' seconds.
        /// </summary>
        [Tooltip("How many particles should be emitted each 'emitTimeInterval' seconds.")]
        [Range(0f, 20f)] public int emitPerCycle = 6;
        
        /// <summary>
        ///     Determines how often the particles will be emitted.
        /// </summary>
        [Tooltip("Determines how often the particles will be emitted.")]
        [Range(0f, 0.1f)] public float emitTimeInterval = 0.04f;
        
        /// <summary>
        ///     Script will try to predict where the object will be in the next n frames.
        /// </summary>
        [UnityEngine.Tooltip("    Script will try to predict where the object will be in the next n frames.")]
        public int positionExtrapolationFrames = 4;
        
        private float                      _timeElapsed;
        private WaterObject                _targetWaterObject;
        private ParticleSystem             _particleSystem;
        private int[]                      _waterlineIndices;
        private ParticleSystem.NoiseModule _noiseModule;
        private int                        _prevTriCount;
        private int                        _waterlineCount;
        
        private void Start()
        {
            if (ReferenceWaterObject == null)
            {
                ReferenceWaterObject = GetComponentInParent<WaterObject>();
            }
            
            _targetWaterObject = transform.GetComponentInParentsOrChildren<WaterObject>(true);
            if (_targetWaterObject == null)
            {
                Debug.LogError(
                    $"{name}: WaterParticleSystem requires WaterObject attached to the same object or one of parent objects to function.");
                return;
            }
        
            _particleSystem = GetComponent<ParticleSystem>();
            if (_particleSystem == null)
            {
                Debug.LogError("No ParticleSystem found.");
            }
            
            _particleSystem.GetComponent<Renderer>().material.renderQueue = renderQueue;
            _noiseModule                                                  = _particleSystem.noise;
            
            _prevTriCount = -999;
        }
        
        
        private void LateUpdate()
        {
            if (!emit)
            {
                return;
            }

            int triCount = _targetWaterObject.triangleCount;
            if (triCount > 0 && _prevTriCount != triCount)
            {
                _waterlineIndices = new int[triCount];
            }
        
            if (_targetWaterObject.targetRigidbody.velocity.magnitude > sleepThresholdVelocity)
            {
                EmitNew();
            }
        
            _timeElapsed    += Time.deltaTime;
            _prevTriCount =  triCount;
        }
        
        
        private void OnDestroy()
        {
            if (!Application.isPlaying)
            {
                DestroyImmediate(_particleSystem);
            }
        }
        
        
        private void EmitNew()
        {
            if (_targetWaterObject == null)
            {
                return;
            }

            int triCount = _targetWaterObject.triangleCount;
            if (emit && _timeElapsed >= emitTimeInterval && triCount > 0f)
            {
                _timeElapsed = 0;
        
                int emitted = 0;
        
                // Emit allowed number of particles
                float elevation = 0;
                if (ReferenceWaterObject != null)
                {
                    elevation = ReferenceWaterObject.GetWaterHeightSingle(Vector3.zero);
                }
                else
                {
                    Debug.LogWarning("Will not emit. WaterDataProvider is not present in the scene.");
                }
        
                _waterlineCount = 0;
                for (int i = 0; i < triCount; i++)
                {
                    if (_targetWaterObject.ResultStates[i] != 1) continue;
        
                    _waterlineIndices[_waterlineCount] = i;
                    _waterlineCount++;
                }
        
                if (_waterlineCount == 0)
                {
                    return;
                }
        
                float noise = startSize > 1f ? Mathf.Sqrt(startSize) * 0.1f : startSize * 0.1f;
                _noiseModule.strengthX = noise;
                _noiseModule.strengthY = 0f;
                _noiseModule.strengthZ = noise;
        
                while (emitted < emitPerCycle)
                {
                    int i                 = Random.Range(0, _waterlineCount);
                    int waterLineTriIndex = _waterlineIndices[i];
        
                    EmitParticle(
                        _targetWaterObject.ResultP0s[waterLineTriIndex * 6 + 2],
                        _targetWaterObject.ResultP0s[waterLineTriIndex * 6 + 1],
                        elevation,
                        _targetWaterObject.ResultVelocities[waterLineTriIndex],
                        _targetWaterObject.ResultNormals[waterLineTriIndex],
                        _targetWaterObject.ResultForces[waterLineTriIndex],
                        _targetWaterObject.ResultAreas[waterLineTriIndex]);
        
                    emitted++;
                }
            }
        }
        
        
        private void OnDrawGizmosSelected()
        {
            if (_waterlineIndices == null)
            {
                return;
            }
            
            Gizmos.color = Color.magenta;
            for (int i = 0; i < _waterlineIndices.Length; i++)
            {
                if (_targetWaterObject.ResultStates[_waterlineIndices[i]] != 1) continue;
                Vector3 a = _targetWaterObject.ResultP0s[_waterlineIndices[i] * 6 + 2];
                Vector3 b = _targetWaterObject.ResultP0s[_waterlineIndices[i] * 6 + 1];
                Gizmos.DrawLine(a, b);
            }
        }
        
        
        /// <summary>
        ///     Emit a single particle
        /// </summary>
        /// <param name="p0">First point of water line</param>
        /// <param name="p1">Second point of water line</param>
        /// <param name="elevation">Water elevation</param>
        /// <param name="velocity">Triangle velocity</param>
        /// <param name="normal">Triangle normal</param>
        /// <param name="force">Triangle force</param>
        /// <param name="area">Triangle area</param>
        private void EmitParticle(Vector3 p0,    Vector3 p1, float elevation, Vector3 velocity, Vector3 normal,
            Vector3                       force, float   area)
        {
            if (area < 0.0001f)
            {
                return;
            }
        
            // Start velocity
            Vector3 startVelocity = normal * velocity.magnitude;
            startVelocity.y =  0f;
            startVelocity   *= initialVelocityModifier;
        
            // Start position
            Vector3 emissionPoint = (p0 + p1) / 2f;
            emissionPoint   += Time.deltaTime * positionExtrapolationFrames * velocity;
            emissionPoint.y =  elevation + surfaceElevation;
        
            float normalizedForce = force.magnitude / area;
            float startAlpha      = Mathf.Clamp(normalizedForce * 0.00005f * initialAlphaModifier, 0f, maxInitialAlpha);
            Color startColor      = new Color(1f, 1f, 1f, startAlpha);
            float size            = startSize;
        
            if (startAlpha < 0.001f)
            {
                return;
            }
        
            ParticleSystem.EmitParams emitParams = new ParticleSystem.EmitParams
            {
                startColor = startColor,
                position   = emissionPoint,
                velocity   = startVelocity,
                startSize  = size,
            };
            _particleSystem.Emit(emitParams, 1);
            _particleSystem.Play();
        }
    }
}