#include "XCEngine/RHI/Vulkan/VulkanResourceView.h"

namespace XCEngine {
namespace RHI {

VulkanResourceView::~VulkanResourceView() {
    Shutdown();
}

bool VulkanResourceView::InitializeAsRenderTarget(VkDevice device, VulkanTexture* texture, const ResourceViewDesc& desc) {
    if (device == VK_NULL_HANDLE || texture == nullptr || texture->GetImage() == VK_NULL_HANDLE) {
        return false;
    }

    m_device = device;
    m_texture = texture;
    m_viewType = ResourceViewType::RenderTarget;
    m_dimension = desc.dimension != ResourceViewDimension::Unknown ? desc.dimension : ResourceViewDimension::Texture2D;
    m_format = desc.format != 0 ? static_cast<Format>(desc.format) : texture->GetFormat();

    VkImageViewCreateInfo viewInfo = {};
    viewInfo.sType = VK_STRUCTURE_TYPE_IMAGE_VIEW_CREATE_INFO;
    viewInfo.image = texture->GetImage();
    viewInfo.viewType = VK_IMAGE_VIEW_TYPE_2D;
    viewInfo.format = texture->GetVkFormat();
    viewInfo.subresourceRange.aspectMask = GetImageAspectMask(texture->GetFormat());
    viewInfo.subresourceRange.baseMipLevel = desc.mipLevel;
    viewInfo.subresourceRange.levelCount = 1;
    viewInfo.subresourceRange.baseArrayLayer = desc.firstArraySlice;
    viewInfo.subresourceRange.layerCount = 1;

    return vkCreateImageView(device, &viewInfo, nullptr, &m_imageView) == VK_SUCCESS;
}

void VulkanResourceView::Shutdown() {
    if (m_imageView != VK_NULL_HANDLE && m_device != VK_NULL_HANDLE) {
        vkDestroyImageView(m_device, m_imageView, nullptr);
    }
    m_imageView = VK_NULL_HANDLE;
    m_device = VK_NULL_HANDLE;
    m_texture = nullptr;
    m_format = Format::Unknown;
    m_dimension = ResourceViewDimension::Unknown;
}

} // namespace RHI
} // namespace XCEngine