//pipelinedefine
#define H_URP

using HTraceWSGI.Scripts.Globals;
using UnityEngine;
using UnityEngine.Experimental.Rendering;
using UnityEngine.Rendering;
using UnityEngine.Rendering.RenderGraphModule;
using UnityEngine.Rendering.Universal;

namespace HTraceWSGI.Scripts.Wrappers
{
	public class RTWrapper
	{
		public RTHandle      rt;
#if NONE
		public RenderTexture rt;
#endif
		
		
#if NONE
		public void HTextureAlloc(string name, Vector2 resolutionWidthHeight, GraphicsFormat graphicsFormat, int volumeDepthOrSlices = -1, int depthBufferBits = 0,
			TextureDimension textureDimension = TextureDimension.Unknown,
			bool useMipMap = false, bool autoGenerateMips = false, bool enableRandomWrite = true, bool useDynamicScale = true) //useDynamicScale default = true for Upscalers switch between Hardware and Software
		{
			volumeDepthOrSlices = volumeDepthOrSlices == -1 ? HRenderer.TextureXrSlices : volumeDepthOrSlices;
			textureDimension    = textureDimension == TextureDimension.Unknown ? TextureDimension.Tex2D : textureDimension;
			
			var dscr                   = new RenderTextureDescriptor(Mathf.CeilToInt(resolutionWidthHeight.x), Mathf.CeilToInt(resolutionWidthHeight.y));
			dscr.graphicsFormat    = graphicsFormat;
			dscr.volumeDepth       = volumeDepthOrSlices;
			dscr.depthBufferBits   = depthBufferBits; //only 24 bits contains Stencil buffer
			dscr.dimension         = textureDimension;
			dscr.useMipMap         = useMipMap;
			dscr.autoGenerateMips  = autoGenerateMips;
			dscr.enableRandomWrite = enableRandomWrite;
			dscr.useDynamicScale   = useDynamicScale;
				
			rt      = new RenderTexture(dscr);
			rt.name = name;
			rt.Create();
		}
		
		public void HTextureAlloc(string name, Vector2 resolutionWidthHeight, RenderTextureFormat renderTextureFormat, int volumeDepthOrSlices = -1, int depthBufferBits = 0,
			TextureDimension textureDimension = TextureDimension.Unknown,
			bool useMipMap = false, bool autoGenerateMips = false, bool enableRandomWrite = true, bool useDynamicScale = true) //useDynamicScale default = true for Upscalers switch between Hardware and Software
		{
			volumeDepthOrSlices = volumeDepthOrSlices == -1 ? HRenderer.TextureXrSlices : volumeDepthOrSlices;
			textureDimension    = textureDimension == TextureDimension.Unknown ? TextureDimension.Tex2D : textureDimension;
			
			var dscr                   = new RenderTextureDescriptor(Mathf.CeilToInt(resolutionWidthHeight.x), Mathf.CeilToInt(resolutionWidthHeight.y));
			dscr.colorFormat       = renderTextureFormat;
			dscr.volumeDepth       = volumeDepthOrSlices;
			dscr.depthBufferBits   = depthBufferBits; //only 24 bits contains Stencil buffer
			dscr.dimension         = textureDimension;
			dscr.useMipMap         = useMipMap;
			dscr.autoGenerateMips  = autoGenerateMips;
			dscr.enableRandomWrite = enableRandomWrite;
			dscr.useDynamicScale   = useDynamicScale;
				
			rt      = new RenderTexture(dscr);
			rt.name = name;
			rt.Create();
		}
		
		public void HTextureAlloc(string name, int width, int height, GraphicsFormat graphicsFormat, int volumeDepthOrSlices = -1, int depthBufferBits = 0, TextureDimension textureDimension = TextureDimension.Tex2D
				, bool useMipMap = false, bool autoGenerateMips = false, bool enableRandomWrite = true, bool useDynamicScale = true, int mipCount = 1)
		{
			volumeDepthOrSlices     = volumeDepthOrSlices == -1 ? HRenderer.TextureXrSlices : volumeDepthOrSlices;
			textureDimension        = textureDimension == TextureDimension.Unknown ? TextureDimension.Tex2D : textureDimension;
			
			var dscr                   = new RenderTextureDescriptor(width, height);
			dscr.graphicsFormat    = graphicsFormat;
			dscr.volumeDepth       = volumeDepthOrSlices;
			dscr.depthBufferBits   = depthBufferBits; //only 24 bits contains Stencil buffer
			dscr.dimension         = textureDimension;
			dscr.useMipMap         = useMipMap;
			dscr.autoGenerateMips  = autoGenerateMips;
			dscr.enableRandomWrite = enableRandomWrite;
			dscr.useDynamicScale   = useDynamicScale;
			dscr.mipCount          = mipCount;
			
			rt      = new RenderTexture(dscr);
			rt.name = name;
			rt.Create();
		}
		
		public void HTextureAlloc(string name, int width, int height, RenderTextureFormat renderTextureFormat, int volumeDepthOrSlices = -1, int depthBufferBits = 0, TextureDimension textureDimension = TextureDimension.Tex2D,
			bool useMipMap = false, bool autoGenerateMips = false, bool enableRandomWrite = true, bool useDynamicScale = true)
		{
			volumeDepthOrSlices     = volumeDepthOrSlices == -1 ? HRenderer.TextureXrSlices : volumeDepthOrSlices;
			textureDimension        = textureDimension == TextureDimension.Unknown ? TextureDimension.Tex2D : textureDimension;
      
			var dscr                   = new RenderTextureDescriptor(width, height);
			dscr.colorFormat       = renderTextureFormat;
			dscr.volumeDepth       = volumeDepthOrSlices;
			dscr.depthBufferBits   = depthBufferBits; //only 24 bits contains Stencil buffer
			dscr.dimension         = textureDimension;
			dscr.useMipMap         = useMipMap;
			dscr.autoGenerateMips  = autoGenerateMips;
			dscr.enableRandomWrite = enableRandomWrite;
			dscr.useDynamicScale   = useDynamicScale;
      
			rt      = new RenderTexture(dscr);
			rt.name = name;
			rt.Create();
		}


		public void HRelease()
		{
			if (rt != null) 
				rt.Release();
		}
#endif
		
		public void HTextureAlloc(string name, Vector2 scaleFactor, GraphicsFormat graphicsFormat, int volumeDepthOrSlices = -1, int depthBufferBits = 0,
			TextureDimension textureDimension = TextureDimension.Unknown,
			bool useMipMap = false, bool autoGenerateMips = false, bool enableRandomWrite = true, bool useDynamicScale = true) //useDynamicScale default = true for Upscalers switch between Hardware and Software
		{
			volumeDepthOrSlices = volumeDepthOrSlices == -1 ? TextureXR.slices : volumeDepthOrSlices;
			textureDimension = textureDimension == TextureDimension.Unknown ? TextureXR.dimension : textureDimension;
			
			rt = RTHandles.Alloc(scaleFactor, volumeDepthOrSlices, dimension: textureDimension, colorFormat: graphicsFormat, name: name,
				enableRandomWrite: enableRandomWrite, useMipMap: useMipMap, useDynamicScale: useDynamicScale, autoGenerateMips: autoGenerateMips,
				depthBufferBits: (DepthBits)depthBufferBits);
		}
		
		public void HTextureAlloc(string name, ScaleFunc scaleFunc, GraphicsFormat graphicsFormat, int volumeDepthOrSlices = -1, int depthBufferBits = 0,
			TextureDimension textureDimension = TextureDimension.Unknown,
			bool useMipMap = false, bool autoGenerateMips = false, bool enableRandomWrite = true, bool useDynamicScale = true) //useDynamicScale default = true for Upscalers switch between Hardware and Software
		{
			volumeDepthOrSlices = volumeDepthOrSlices == -1 ? TextureXR.slices : volumeDepthOrSlices;
			textureDimension    = textureDimension == TextureDimension.Unknown ? TextureXR.dimension : textureDimension;
			
			rt = RTHandles.Alloc(scaleFunc, volumeDepthOrSlices, dimension: textureDimension, colorFormat: graphicsFormat, name: name,
				enableRandomWrite: enableRandomWrite, useMipMap: useMipMap, useDynamicScale: useDynamicScale, autoGenerateMips: autoGenerateMips,
				depthBufferBits: (DepthBits)depthBufferBits);
		}
		
		public void HTextureAlloc(string name, int width, int height, GraphicsFormat graphicsFormat, int volumeDepthOrSlices = -1, int depthBufferBits = 0,
			TextureDimension textureDimension = TextureDimension.Unknown,
			bool useMipMap = false, bool autoGenerateMips = false, bool enableRandomWrite = true, bool useDynamicScale = true) //useDynamicScale default = true for Upscalers switch between Hardware and Software
		{
			volumeDepthOrSlices = volumeDepthOrSlices == -1 ? TextureXR.slices : volumeDepthOrSlices;
			textureDimension    = textureDimension == TextureDimension.Unknown ? TextureXR.dimension : textureDimension;
			
			rt = RTHandles.Alloc(width, height, volumeDepthOrSlices, dimension: textureDimension, colorFormat: graphicsFormat, name: name, 
				enableRandomWrite: enableRandomWrite, useMipMap: useMipMap, useDynamicScale: useDynamicScale, autoGenerateMips: autoGenerateMips,
				depthBufferBits: (DepthBits)depthBufferBits);
		}
		
		public void HRelease()
		{
			RTHandles.Release(rt);
		}
		
		public void ReAllocateIfNeeded(string name, ref RenderTextureDescriptor inputDescriptor, GraphicsFormat graphicsFormat = GraphicsFormat.None,
			int width = -1, int height = -1, int volumeDepth = 1,
			TextureDimension dimension = TextureDimension.Tex2D,
			bool enableRandomWrite = true, bool useMipMap = false, bool autoGenerateMips = false,
			bool useDynamicScale = false)
		{
			if (dimension != TextureDimension.Tex2D || width > 0 || height > 0) //conditions to create new descriptor
			{
				RenderTextureDescriptor newDesc = inputDescriptor;
				
				newDesc.width = width == -1 ? inputDescriptor.width : width;
				newDesc.height = height == -1 ? inputDescriptor.height : height;
				newDesc.volumeDepth = volumeDepth;
				newDesc.dimension = dimension;
				newDesc.graphicsFormat = graphicsFormat;
				newDesc.autoGenerateMips = autoGenerateMips;
				newDesc.useMipMap = useMipMap;
				newDesc.enableRandomWrite = enableRandomWrite;
				newDesc.msaaSamples = 1;
				newDesc.useDynamicScale = false;

#if UNITY_2023_3_OR_NEWER
				RenderingUtils.ReAllocateHandleIfNeeded(ref rt, newDesc, name: name);
#else
				RenderingUtils.ReAllocateIfNeeded(ref rt, newDesc, name: name);
#endif
				return;
			}
			
			inputDescriptor.volumeDepth = volumeDepth;
			inputDescriptor.graphicsFormat = graphicsFormat;
			inputDescriptor.autoGenerateMips = autoGenerateMips;
			inputDescriptor.useMipMap = useMipMap;
			inputDescriptor.enableRandomWrite = enableRandomWrite;
			inputDescriptor.msaaSamples = 1;
			inputDescriptor.useDynamicScale = false;

#if UNITY_2023_3_OR_NEWER
			RenderingUtils.ReAllocateHandleIfNeeded(ref rt, inputDescriptor, name: name);
#else
			RenderingUtils.ReAllocateIfNeeded(ref rt, inputDescriptor, name: name);
#endif
		}
	}
}