//pipelinedefine
#define H_URP

using HTraceSSGI.Scripts.Globals;
using UnityEngine;
using UnityEngine.Experimental.Rendering;
using UnityEngine.Rendering;
using UnityEngine.Rendering.Universal;

namespace HTraceSSGI.Scripts.Wrappers
{
	public class RTWrapper
	{
		private RenderTextureDescriptor _dscr;
		
		public RTHandle      rt;
		
		
		
		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
		}
	}
}