using OpenTK.Graphics.OpenGL;
using Ryujinx.Graphics.GAL;
using System;

namespace Ryujinx.Graphics.OpenGL
{
    static class Buffer
    {
        public static void Clear(BufferHandle destination, int offset, int size, uint value)
        {
            GL.BindBuffer(BufferTarget.CopyWriteBuffer, destination.ToInt32());

            unsafe
            {
                uint* valueArr = stackalloc uint[1];

                valueArr[0] = value;

                GL.ClearBufferSubData(
                    BufferTarget.CopyWriteBuffer,
                    PixelInternalFormat.Rgba8ui,
                    (nint)offset,
                    (nint)size,
                    PixelFormat.RgbaInteger,
                    PixelType.UnsignedByte,
                    (nint)valueArr);
            }
        }

        public static BufferHandle Create()
        {
            return Handle.FromInt32<BufferHandle>(GL.GenBuffer());
        }

        public static BufferHandle Create(int size)
        {
            int handle = GL.GenBuffer();

            GL.BindBuffer(BufferTarget.CopyWriteBuffer, handle);
            GL.BufferData(BufferTarget.CopyWriteBuffer, size, nint.Zero, BufferUsageHint.DynamicDraw);

            return Handle.FromInt32<BufferHandle>(handle);
        }

        public static BufferHandle CreatePersistent(int size)
        {
            int handle = GL.GenBuffer();

            GL.BindBuffer(BufferTarget.CopyWriteBuffer, handle);
            GL.BufferStorage(BufferTarget.CopyWriteBuffer, size, nint.Zero,
                BufferStorageFlags.MapPersistentBit |
                BufferStorageFlags.MapCoherentBit |
                BufferStorageFlags.ClientStorageBit |
                BufferStorageFlags.MapReadBit);

            return Handle.FromInt32<BufferHandle>(handle);
        }

        public static void Copy(BufferHandle source, BufferHandle destination, int srcOffset, int dstOffset, int size)
        {
            GL.BindBuffer(BufferTarget.CopyReadBuffer, source.ToInt32());
            GL.BindBuffer(BufferTarget.CopyWriteBuffer, destination.ToInt32());

            GL.CopyBufferSubData(
                BufferTarget.CopyReadBuffer,
                BufferTarget.CopyWriteBuffer,
                (nint)srcOffset,
                (nint)dstOffset,
                (nint)size);
        }

        public static unsafe PinnedSpan<byte> GetData(OpenGLRenderer renderer, BufferHandle buffer, int offset, int size)
        {
            // Data in the persistent buffer and host array is guaranteed to be available
            // until the next time the host thread requests data.

            if (renderer.PersistentBuffers.TryGet(buffer, out nint ptr))
            {
                return new PinnedSpan<byte>(nint.Add(ptr, offset).ToPointer(), size);
            }
            else if (HwCapabilities.UsePersistentBufferForFlush)
            {
                return PinnedSpan<byte>.UnsafeFromSpan(renderer.PersistentBuffers.Default.GetBufferData(buffer, offset, size));
            }
            else
            {
                nint target = renderer.PersistentBuffers.Default.GetHostArray(size);

                GL.BindBuffer(BufferTarget.CopyReadBuffer, buffer.ToInt32());

                GL.GetBufferSubData(BufferTarget.CopyReadBuffer, (nint)offset, size, target);

                return new PinnedSpan<byte>(target.ToPointer(), size);
            }
        }

        public static void Resize(BufferHandle handle, int size)
        {
            GL.BindBuffer(BufferTarget.CopyWriteBuffer, handle.ToInt32());
            GL.BufferData(BufferTarget.CopyWriteBuffer, size, nint.Zero, BufferUsageHint.StreamCopy);
        }

        public static void SetData(BufferHandle buffer, int offset, ReadOnlySpan<byte> data)
        {
            GL.BindBuffer(BufferTarget.CopyWriteBuffer, buffer.ToInt32());

            unsafe
            {
                fixed (byte* ptr = data)
                {
                    GL.BufferSubData(BufferTarget.CopyWriteBuffer, (nint)offset, data.Length, (nint)ptr);
                }
            }
        }

        public static void Delete(BufferHandle buffer)
        {
            GL.DeleteBuffer(buffer.ToInt32());
        }
    }
}