package com.mojang.blaze3d.vulkan;

import com.mojang.blaze3d.textures.GpuTextureView;
import java.nio.LongBuffer;
import org.lwjgl.system.MemoryStack;
import org.lwjgl.vulkan.VK12;
import org.lwjgl.vulkan.VkImageSubresourceRange;
import org.lwjgl.vulkan.VkImageViewCreateInfo;

public class VulkanGpuTextureView extends GpuTextureView implements Destroyable {
	private final VulkanDevice device;
	private final long vkImageView;
	private boolean closed;

	protected VulkanGpuTextureView(final VulkanDevice device, final VulkanGpuTexture texture, final int baseMipLevel, final int mipLevels) {
		super(texture, baseMipLevel, mipLevels);
		this.device = device;

		try (MemoryStack stack = MemoryStack.stackPush()) {
			boolean isCubemap = (texture.usage() & 16) != 0;
			VkImageViewCreateInfo imageViewCreateInfo = VkImageViewCreateInfo.calloc(stack).sType$Default();
			imageViewCreateInfo.image(texture.vkImage());
			imageViewCreateInfo.viewType(isCubemap ? 3 : 1);
			imageViewCreateInfo.format(VulkanConst.toVk(texture.getFormat()));
			VkImageSubresourceRange subresourceRange = imageViewCreateInfo.subresourceRange();
			subresourceRange.aspectMask(texture.getFormat().hasColorAspect() ? 1 : 2);
			subresourceRange.baseMipLevel(baseMipLevel);
			subresourceRange.levelCount(mipLevels);
			subresourceRange.baseArrayLayer(0);
			subresourceRange.layerCount(isCubemap ? 6 : 1);
			LongBuffer handlePtr = stack.callocLong(1);
			VulkanUtils.crashIfFailure(device, VK12.vkCreateImageView(device.vkDevice(), imageViewCreateInfo, null, handlePtr), "Failed to create VkImageView");
			this.vkImageView = handlePtr.get(0);
			device.instance().debug().setObjectName(device.vkDevice(), 14, this.vkImageView, texture.getLabel());
		}

		texture.addViews();
	}

	@Override
	public void destroy() {
		VK12.vkDestroyImageView(this.device.vkDevice(), this.vkImageView, null);
	}

	@Override
	public void close() {
		if (!this.closed) {
			this.closed = true;
			this.device.createCommandEncoder().queueForDestroy(this);
			this.texture().removeViews();
		}
	}

	@Override
	public boolean isClosed() {
		return this.closed;
	}

	public VulkanGpuTexture texture() {
		return (VulkanGpuTexture)super.texture();
	}

	public long vkImageView() {
		return this.vkImageView;
	}
}