package com.mojang.blaze3d.opengl;

import com.mojang.blaze3d.buffers.GpuBuffer;
import java.nio.ByteBuffer;
import java.util.Set;
import org.jspecify.annotations.Nullable;
import org.lwjgl.opengl.ARBBufferStorage;
import org.lwjgl.opengl.ARBDirectStateAccess;
import org.lwjgl.opengl.GL33C;
import org.lwjgl.opengl.GLCapabilities;

public abstract class DirectStateAccess {
	public static DirectStateAccess create(final GLCapabilities capabilities, final Set<String> enabledExtensions, final GlHeuristics heuristics) {
		if (capabilities.GL_ARB_direct_state_access && GlDevice.USE_GL_ARB_direct_state_access && !heuristics.isGlOnDx12()) {
			enabledExtensions.add("GL_ARB_direct_state_access");
			return new DirectStateAccess.Core();
		} else {
			return new DirectStateAccess.Emulated();
		}
	}

	public abstract int createBuffer();

	public abstract void bufferData(int buffer, long size, @GpuBuffer.Usage int usage);

	public abstract void bufferData(int buffer, ByteBuffer data, @GpuBuffer.Usage int usage);

	public abstract void bufferSubData(int buffer, long offset, ByteBuffer data, @GpuBuffer.Usage int usage);

	public abstract void bufferStorage(int buffer, long size, @GpuBuffer.Usage int usage);

	public abstract void bufferStorage(int buffer, ByteBuffer data, @GpuBuffer.Usage int usage);

	@Nullable
	public abstract ByteBuffer mapBufferRange(int buffer, long offset, long length, int access, @GpuBuffer.Usage int usage);

	public abstract void unmapBuffer(int buffer, @GpuBuffer.Usage int usage);

	public abstract int createFrameBufferObject();

	public abstract void bindFrameBufferTextures(int fbo, int[] color, int[] colorMipLevels, int depth, int depthMipLevel, int bindSlot);

	public void bindFrameBufferTextures(final int fbo, final int color, final int depth, final int mipLevel, final int bindSlot) {
		this.bindFrameBufferTextures(fbo, new int[]{color}, new int[]{mipLevel}, depth, mipLevel, bindSlot);
	}

	public abstract void blitFrameBuffers(
		int source, int dest, int srcX0, int srcY0, int srcX1, int srcY1, int dstX0, int dstY0, int dstX1, int dstY1, int mask, int filter
	);

	public abstract void flushMappedBufferRange(final int handle, final long offset, final long length, @GpuBuffer.Usage final int usage);

	public abstract void copyBufferSubData(int source, int target, long sourceOffset, long targetOffset, long length);

	private static class Core extends DirectStateAccess {
		@Override
		public int createBuffer() {
			GlStateManager.incrementTrackedBuffers();
			return ARBDirectStateAccess.glCreateBuffers();
		}

		@Override
		public void bufferData(final int buffer, final long size, @GpuBuffer.Usage final int usage) {
			ARBDirectStateAccess.glNamedBufferData(buffer, size, GlConst.bufferUsageToGlEnum(usage));
		}

		@Override
		public void bufferData(final int buffer, final ByteBuffer data, @GpuBuffer.Usage final int usage) {
			ARBDirectStateAccess.glNamedBufferData(buffer, data, GlConst.bufferUsageToGlEnum(usage));
		}

		@Override
		public void bufferSubData(final int buffer, final long offset, final ByteBuffer data, @GpuBuffer.Usage final int usage) {
			ARBDirectStateAccess.glNamedBufferSubData(buffer, offset, data);
		}

		@Override
		public void bufferStorage(final int buffer, final long size, @GpuBuffer.Usage final int usage) {
			ARBDirectStateAccess.glNamedBufferStorage(buffer, size, GlConst.bufferUsageToGlFlag(usage));
		}

		@Override
		public void bufferStorage(final int buffer, final ByteBuffer data, @GpuBuffer.Usage final int usage) {
			ARBDirectStateAccess.glNamedBufferStorage(buffer, data, GlConst.bufferUsageToGlFlag(usage));
		}

		@Nullable
		@Override
		public ByteBuffer mapBufferRange(final int buffer, final long offset, final long length, final int flags, @GpuBuffer.Usage final int usage) {
			return ARBDirectStateAccess.glMapNamedBufferRange(buffer, offset, length, flags);
		}

		@Override
		public void unmapBuffer(final int buffer, @GpuBuffer.Usage final int usage) {
			ARBDirectStateAccess.glUnmapNamedBuffer(buffer);
		}

		@Override
		public int createFrameBufferObject() {
			return ARBDirectStateAccess.glCreateFramebuffers();
		}

		@Override
		public void bindFrameBufferTextures(
			final int fbo, final int[] color, final int[] colorMipLevels, final int depth, final int depthMipLevel, final int bindSlot
		) {
			for (int i = 0; i < color.length; i++) {
				ARBDirectStateAccess.glNamedFramebufferTexture(fbo, 36064 + i, color[i], colorMipLevels[i]);
			}

			ARBDirectStateAccess.glNamedFramebufferTexture(fbo, 36096, depth, depthMipLevel);
			if (bindSlot != 0) {
				GlStateManager._glBindFramebuffer(bindSlot, fbo);
			}
		}

		@Override
		public void blitFrameBuffers(
			final int source,
			final int dest,
			final int srcX0,
			final int srcY0,
			final int srcX1,
			final int srcY1,
			final int dstX0,
			final int dstY0,
			final int dstX1,
			final int dstY1,
			final int mask,
			final int filter
		) {
			ARBDirectStateAccess.glBlitNamedFramebuffer(source, dest, srcX0, srcY0, srcX1, srcY1, dstX0, dstY0, dstX1, dstY1, mask, filter);
		}

		@Override
		public void flushMappedBufferRange(final int handle, final long offset, final long length, @GpuBuffer.Usage final int usage) {
			ARBDirectStateAccess.glFlushMappedNamedBufferRange(handle, offset, length);
		}

		@Override
		public void copyBufferSubData(final int source, final int target, final long sourceOffset, final long targetOffset, final long length) {
			ARBDirectStateAccess.glCopyNamedBufferSubData(source, target, sourceOffset, targetOffset, length);
		}
	}

	private static class Emulated extends DirectStateAccess {
		private static int selectBufferBindTarget(@GpuBuffer.Usage final int usage) {
			if ((usage & 32) != 0) {
				return 34962;
			} else if ((usage & 64) != 0) {
				return 34963;
			} else {
				return (usage & 128) != 0 ? 35345 : 36663;
			}
		}

		@Override
		public int createBuffer() {
			return GlStateManager._glGenBuffers();
		}

		@Override
		public void bufferData(final int buffer, final long size, @GpuBuffer.Usage final int usage) {
			int target = selectBufferBindTarget(usage);
			GlStateManager._glBindBuffer(target, buffer);
			GlStateManager._glBufferData(target, size, GlConst.bufferUsageToGlEnum(usage));
			GlStateManager._glBindBuffer(target, 0);
		}

		@Override
		public void bufferData(final int buffer, final ByteBuffer data, @GpuBuffer.Usage final int usage) {
			int target = selectBufferBindTarget(usage);
			GlStateManager._glBindBuffer(target, buffer);
			GlStateManager._glBufferData(target, data, GlConst.bufferUsageToGlEnum(usage));
			GlStateManager._glBindBuffer(target, 0);
		}

		@Override
		public void bufferSubData(final int buffer, final long offset, final ByteBuffer data, @GpuBuffer.Usage final int usage) {
			int target = selectBufferBindTarget(usage);
			GlStateManager._glBindBuffer(target, buffer);
			GlStateManager._glBufferSubData(target, offset, data);
			GlStateManager._glBindBuffer(target, 0);
		}

		@Override
		public void bufferStorage(final int buffer, final long size, @GpuBuffer.Usage final int usage) {
			int target = selectBufferBindTarget(usage);
			GlStateManager._glBindBuffer(target, buffer);
			ARBBufferStorage.glBufferStorage(target, size, GlConst.bufferUsageToGlFlag(usage));
			GlStateManager._glBindBuffer(target, 0);
		}

		@Override
		public void bufferStorage(final int buffer, final ByteBuffer data, @GpuBuffer.Usage final int usage) {
			int target = selectBufferBindTarget(usage);
			GlStateManager._glBindBuffer(target, buffer);
			ARBBufferStorage.glBufferStorage(target, data, GlConst.bufferUsageToGlFlag(usage));
			GlStateManager._glBindBuffer(target, 0);
		}

		@Nullable
		@Override
		public ByteBuffer mapBufferRange(final int buffer, final long offset, final long length, final int access, @GpuBuffer.Usage final int usage) {
			int target = selectBufferBindTarget(usage);
			GlStateManager._glBindBuffer(target, buffer);
			ByteBuffer byteBuffer = GlStateManager._glMapBufferRange(target, offset, length, access);
			GlStateManager._glBindBuffer(target, 0);
			return byteBuffer;
		}

		@Override
		public void unmapBuffer(final int buffer, @GpuBuffer.Usage final int usage) {
			int target = selectBufferBindTarget(usage);
			GlStateManager._glBindBuffer(target, buffer);
			GlStateManager._glUnmapBuffer(target);
			GlStateManager._glBindBuffer(target, 0);
		}

		@Override
		public void flushMappedBufferRange(final int buffer, final long offset, final long length, @GpuBuffer.Usage final int usage) {
			int target = selectBufferBindTarget(usage);
			GlStateManager._glBindBuffer(target, buffer);
			GL33C.glFlushMappedBufferRange(target, offset, length);
			GlStateManager._glBindBuffer(target, 0);
		}

		@Override
		public void copyBufferSubData(final int source, final int target, final long sourceOffset, final long targetOffset, final long length) {
			GlStateManager._glBindBuffer(36662, source);
			GlStateManager._glBindBuffer(36663, target);
			GL33C.glCopyBufferSubData(36662, 36663, sourceOffset, targetOffset, length);
			GlStateManager._glBindBuffer(36662, 0);
			GlStateManager._glBindBuffer(36663, 0);
		}

		@Override
		public int createFrameBufferObject() {
			return GlStateManager.glGenFramebuffers();
		}

		@Override
		public void bindFrameBufferTextures(
			final int fbo, final int[] color, final int[] colorMipLevels, final int depth, final int depthMipLevel, final int bindSlot
		) {
			int tempBindSlot = bindSlot == 0 ? '販' : bindSlot;
			int oldFbo = GlStateManager.getFrameBuffer(tempBindSlot);
			GlStateManager._glBindFramebuffer(tempBindSlot, fbo);

			for (int i = 0; i < color.length; i++) {
				GlStateManager._glFramebufferTexture2D(tempBindSlot, 36064 + i, 3553, color[i], colorMipLevels[i]);
			}

			GlStateManager._glFramebufferTexture2D(tempBindSlot, 36096, 3553, depth, depthMipLevel);
			if (bindSlot == 0) {
				GlStateManager._glBindFramebuffer(tempBindSlot, oldFbo);
			}
		}

		@Override
		public void blitFrameBuffers(
			final int source,
			final int dest,
			final int srcX0,
			final int srcY0,
			final int srcX1,
			final int srcY1,
			final int dstX0,
			final int dstY0,
			final int dstX1,
			final int dstY1,
			final int mask,
			final int filter
		) {
			int oldRead = GlStateManager.getFrameBuffer(36008);
			int oldDraw = GlStateManager.getFrameBuffer(36009);
			GlStateManager._glBindFramebuffer(36008, source);
			GlStateManager._glBindFramebuffer(36009, dest);
			GlStateManager._glBlitFrameBuffer(srcX0, srcY0, srcX1, srcY1, dstX0, dstY0, dstX1, dstY1, mask, filter);
			GlStateManager._glBindFramebuffer(36008, oldRead);
			GlStateManager._glBindFramebuffer(36009, oldDraw);
		}
	}
}