// Crest Water System
// Copyright © 2024 Wave Harmonic. All rights reserved.

#if UNITY_EDITOR

using UnityEngine;
using UnityEngine.Rendering;

namespace WaveHarmonic.Crest
{
    partial class SurfaceRenderer
    {
        RenderTexture _WaterLevelDepthTexture;
        internal RenderTexture WaterLevelDepthTexture => _WaterLevelDepthTexture;
        RenderTargetIdentifier _WaterLevelDepthTarget;
        Material _WaterLevelDepthMaterial;

        const string k_WaterLevelDepthTextureName = "Crest Water Level Depth Texture";

        void ExecuteWaterLevelDepthTexture(Camera camera, CommandBuffer buffer)
        {
            // Currently, only used for painting which means only when mouse is over the view.
            if (!_Water._CurrentSceneCameraHasMouseHover)
            {
                return;
            }

            if (_WaterLevelDepthTexture == null)
            {
                _WaterLevelDepthTexture = new(0, 0, 0);
            }

            WaterLevelDepthTexture.name = k_WaterLevelDepthTextureName;

            if (_WaterLevelDepthMaterial == null)
            {
                _WaterLevelDepthMaterial = new(Shader.Find("Hidden/Crest/Editor/Water Level (Depth)"));
            }

            var descriptor = new RenderTextureDescriptor(camera.pixelWidth, camera.pixelHeight)
            {
                graphicsFormat = UnityEngine.Experimental.Rendering.GraphicsFormat.None,
                depthBufferBits = 32,
            };

            // Depth buffer.
            buffer.GetTemporaryRT(Helpers.ShaderIDs.s_MainTexture, descriptor);
            CoreUtils.SetRenderTarget(buffer, Helpers.ShaderIDs.s_MainTexture, ClearFlag.Depth);

            Render(camera, buffer, _WaterLevelDepthMaterial);

            // Depth texture.
            // Always release to handle screen size changes.
            WaterLevelDepthTexture.Release();
            descriptor.graphicsFormat = UnityEngine.Experimental.Rendering.GraphicsFormat.R32_SFloat;
            descriptor.depthBufferBits = 0;
            WaterLevelDepthTexture.descriptor = descriptor;
            WaterLevelDepthTexture.Create();

            _WaterLevelDepthTarget = new
            (
                WaterLevelDepthTexture,
                mipLevel: 0,
                CubemapFace.Unknown,
                depthSlice: -1 // Bind all XR slices.
            );

            // Convert.
            Helpers.Blit(buffer, _WaterLevelDepthTarget, Rendering.BIRP.UtilityMaterial, (int)Rendering.BIRP.UtilityPass.Copy);

            buffer.ReleaseTemporaryRT(Helpers.ShaderIDs.s_MainTexture);
        }

        void EnableWaterLevelDepthTexture()
        {
            if (Application.isPlaying) return;

#if d_UnityURP
            if (RenderPipelineHelper.IsUniversal)
            {
                WaterLevelDepthTextureURP.Enable(_Water, this);
            }
#endif

#if d_UnityHDRP
            if (RenderPipelineHelper.IsHighDefinition)
            {
                WaterLevelDepthTextureHDRP.Enable(_Water, this);
            }
#endif
        }

        void DisableWaterLevelDepthTexture()
        {
            if (Application.isPlaying) return;

#if d_UnityHDRP
            WaterLevelDepthTextureHDRP.Disable();
#endif
        }
    }
}

#endif