421 lines
14 KiB
C++
421 lines
14 KiB
C++
// Copyright (C) 2021 Jérôme "Lynix" Leclercq (lynix680@gmail.com)
|
|
// This file is part of the "Nazara Engine - Vulkan renderer"
|
|
// For conditions of distribution and use, see copyright notice in Config.hpp
|
|
|
|
#include <Nazara/VulkanRenderer/VulkanTexture.hpp>
|
|
#include <Nazara/Core/CallOnExit.hpp>
|
|
#include <Nazara/Utility/PixelFormat.hpp>
|
|
#include <Nazara/VulkanRenderer/Wrapper/CommandBuffer.hpp>
|
|
#include <Nazara/VulkanRenderer/Wrapper/QueueHandle.hpp>
|
|
#include <vma/vk_mem_alloc.h>
|
|
#include <stdexcept>
|
|
#include <Nazara/VulkanRenderer/Debug.hpp>
|
|
|
|
namespace Nz
|
|
{
|
|
VulkanTexture::VulkanTexture(Vk::Device& device, const TextureInfo& params) :
|
|
m_image(VK_NULL_HANDLE),
|
|
m_allocation(nullptr),
|
|
m_device(device),
|
|
m_params(params)
|
|
{
|
|
VkImageCreateInfo createInfo = {};
|
|
createInfo.sType = VK_STRUCTURE_TYPE_IMAGE_CREATE_INFO;
|
|
createInfo.initialLayout = VK_IMAGE_LAYOUT_UNDEFINED;
|
|
createInfo.mipLevels = params.mipmapLevel;
|
|
createInfo.samples = VK_SAMPLE_COUNT_1_BIT;
|
|
createInfo.tiling = VK_IMAGE_TILING_OPTIMAL;
|
|
createInfo.usage = ToVulkan(params.usageFlags);
|
|
|
|
VkImageViewCreateInfo createInfoView = {};
|
|
createInfoView.sType = VK_STRUCTURE_TYPE_IMAGE_VIEW_CREATE_INFO;
|
|
createInfoView.subresourceRange = {
|
|
ToVulkan(PixelFormatInfo::GetContent(params.pixelFormat)),
|
|
0,
|
|
1,
|
|
0,
|
|
1
|
|
};
|
|
|
|
InitForFormat(params.pixelFormat, createInfo, createInfoView);
|
|
|
|
switch (params.type)
|
|
{
|
|
case ImageType::E1D:
|
|
NazaraAssert(params.width > 0, "Width must be over zero");
|
|
|
|
createInfoView.viewType = VK_IMAGE_VIEW_TYPE_1D;
|
|
|
|
createInfo.imageType = VK_IMAGE_TYPE_1D;
|
|
createInfo.extent.width = params.width;
|
|
createInfo.extent.height = 1;
|
|
createInfo.extent.depth = 1;
|
|
createInfo.arrayLayers = 1;
|
|
break;
|
|
|
|
case ImageType::E1D_Array:
|
|
NazaraAssert(params.width > 0, "Width must be over zero");
|
|
NazaraAssert(params.height > 0, "Height must be over zero");
|
|
|
|
createInfoView.viewType = VK_IMAGE_VIEW_TYPE_1D_ARRAY;
|
|
|
|
createInfo.imageType = VK_IMAGE_TYPE_1D;
|
|
createInfo.extent.width = params.width;
|
|
createInfo.extent.height = 1;
|
|
createInfo.extent.depth = 1;
|
|
createInfo.arrayLayers = params.height;
|
|
break;
|
|
|
|
case ImageType::E2D:
|
|
NazaraAssert(params.width > 0, "Width must be over zero");
|
|
NazaraAssert(params.height > 0, "Height must be over zero");
|
|
|
|
createInfoView.viewType = VK_IMAGE_VIEW_TYPE_2D;
|
|
|
|
createInfo.imageType = VK_IMAGE_TYPE_2D;
|
|
createInfo.extent.width = params.width;
|
|
createInfo.extent.height = params.height;
|
|
createInfo.extent.depth = 1;
|
|
createInfo.arrayLayers = 1;
|
|
break;
|
|
|
|
case ImageType::E2D_Array:
|
|
NazaraAssert(params.width > 0, "Width must be over zero");
|
|
NazaraAssert(params.height > 0, "Height must be over zero");
|
|
NazaraAssert(params.depth > 0, "Depth must be over zero");
|
|
|
|
createInfoView.viewType = VK_IMAGE_VIEW_TYPE_2D_ARRAY;
|
|
|
|
createInfo.imageType = VK_IMAGE_TYPE_2D;
|
|
createInfo.extent.width = params.width;
|
|
createInfo.extent.height = params.height;
|
|
createInfo.extent.depth = 1;
|
|
createInfo.arrayLayers = params.depth;
|
|
break;
|
|
|
|
case ImageType::E3D:
|
|
NazaraAssert(params.width > 0, "Width must be over zero");
|
|
NazaraAssert(params.height > 0, "Height must be over zero");
|
|
NazaraAssert(params.depth > 0, "Depth must be over zero");
|
|
|
|
createInfoView.viewType = VK_IMAGE_VIEW_TYPE_3D;
|
|
|
|
createInfo.imageType = VK_IMAGE_TYPE_3D;
|
|
createInfo.extent.width = params.width;
|
|
createInfo.extent.height = params.height;
|
|
createInfo.extent.depth = params.depth;
|
|
createInfo.arrayLayers = 1;
|
|
break;
|
|
|
|
case ImageType::Cubemap:
|
|
NazaraAssert(params.width > 0, "Width must be over zero");
|
|
NazaraAssert(params.height > 0, "Height must be over zero");
|
|
|
|
createInfoView.viewType = VK_IMAGE_VIEW_TYPE_CUBE;
|
|
createInfoView.subresourceRange.layerCount = 6;
|
|
|
|
createInfo.imageType = VK_IMAGE_TYPE_2D;
|
|
createInfo.extent.width = params.width;
|
|
createInfo.extent.height = params.height;
|
|
createInfo.extent.depth = 1;
|
|
createInfo.arrayLayers = 6;
|
|
createInfo.flags |= VK_IMAGE_CREATE_CUBE_COMPATIBLE_BIT;
|
|
break;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
|
|
VmaAllocationCreateInfo allocInfo = {};
|
|
allocInfo.usage = VMA_MEMORY_USAGE_GPU_ONLY;
|
|
|
|
VkResult result = vmaCreateImage(m_device.GetMemoryAllocator(), &createInfo, &allocInfo, &m_image, &m_allocation, nullptr);
|
|
if (result != VK_SUCCESS)
|
|
throw std::runtime_error("Failed to allocate image: " + TranslateVulkanError(result));
|
|
|
|
createInfoView.image = m_image;
|
|
|
|
if (!m_imageView.Create(device, createInfoView))
|
|
{
|
|
// FIXME
|
|
vmaDestroyImage(m_device.GetMemoryAllocator(), m_image, m_allocation);
|
|
throw std::runtime_error("Failed to create image view: " + TranslateVulkanError(m_imageView.GetLastErrorCode()));
|
|
}
|
|
}
|
|
|
|
VulkanTexture::~VulkanTexture()
|
|
{
|
|
vmaDestroyImage(m_device.GetMemoryAllocator(), m_image, m_allocation);
|
|
}
|
|
|
|
bool VulkanTexture::Copy(const Texture& source, const Boxui& srcBox, const Vector3ui& dstPos)
|
|
{
|
|
const VulkanTexture& sourceTexture = static_cast<const VulkanTexture&>(source);
|
|
|
|
Vk::AutoCommandBuffer copyCommandBuffer = m_device.AllocateCommandBuffer(QueueType::Graphics);
|
|
if (!copyCommandBuffer->Begin(VK_COMMAND_BUFFER_USAGE_ONE_TIME_SUBMIT_BIT))
|
|
return false;
|
|
|
|
VkImageSubresourceLayers subresourceLayers = { //< FIXME
|
|
VK_IMAGE_ASPECT_COLOR_BIT,
|
|
0, //< mipLevel
|
|
0, //< baseArrayLayer
|
|
1 //< layerCount
|
|
};
|
|
|
|
VkImageSubresourceRange subresourceRange = { //< FIXME
|
|
VK_IMAGE_ASPECT_COLOR_BIT,
|
|
0, //< baseMipLevel
|
|
1, //< levelCount
|
|
subresourceLayers.baseArrayLayer, //< baseArrayLayer
|
|
subresourceLayers.layerCount //< layerCount
|
|
};
|
|
|
|
VkImageCopy region = {
|
|
subresourceLayers,
|
|
VkOffset3D { static_cast<Int32>(srcBox.x), static_cast<Int32>(srcBox.y), static_cast<Int32>(srcBox.z) },
|
|
subresourceLayers,
|
|
VkOffset3D { static_cast<Int32>(dstPos.x), static_cast<Int32>(dstPos.y), static_cast<Int32>(dstPos.z) },
|
|
VkExtent3D { static_cast<UInt32>(srcBox.width), static_cast<UInt32>(srcBox.height), static_cast<UInt32>(srcBox.depth) }
|
|
};
|
|
|
|
copyCommandBuffer->SetImageLayout(sourceTexture.GetImage(), VK_PIPELINE_STAGE_FRAGMENT_SHADER_BIT, VK_PIPELINE_STAGE_TRANSFER_BIT, VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL, VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL, subresourceRange);
|
|
copyCommandBuffer->SetImageLayout(m_image, VK_PIPELINE_STAGE_TOP_OF_PIPE_BIT, VK_PIPELINE_STAGE_TRANSFER_BIT, VK_IMAGE_LAYOUT_UNDEFINED, VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL, subresourceRange);
|
|
|
|
copyCommandBuffer->CopyImage(sourceTexture.GetImage(), VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL, m_image, VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL, region);
|
|
|
|
copyCommandBuffer->SetImageLayout(sourceTexture.GetImage(), VK_PIPELINE_STAGE_TRANSFER_BIT, VK_PIPELINE_STAGE_FRAGMENT_SHADER_BIT, VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL, VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL, subresourceRange);
|
|
copyCommandBuffer->SetImageLayout(m_image, VK_PIPELINE_STAGE_TRANSFER_BIT, VK_PIPELINE_STAGE_FRAGMENT_SHADER_BIT, VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL, VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL, subresourceRange);
|
|
|
|
if (!copyCommandBuffer->End())
|
|
return false;
|
|
|
|
Vk::QueueHandle transferQueue = m_device.GetQueue(m_device.GetDefaultFamilyIndex(QueueType::Graphics), 0);
|
|
if (!transferQueue.Submit(copyCommandBuffer))
|
|
return false;
|
|
|
|
transferQueue.WaitIdle();
|
|
|
|
return true;
|
|
}
|
|
|
|
PixelFormat VulkanTexture::GetFormat() const
|
|
{
|
|
return m_params.pixelFormat;
|
|
}
|
|
|
|
UInt8 VulkanTexture::GetLevelCount() const
|
|
{
|
|
return m_params.mipmapLevel;
|
|
}
|
|
|
|
Vector3ui VulkanTexture::GetSize(UInt8 level) const
|
|
{
|
|
return Vector3ui(GetLevelSize(m_params.width, level), GetLevelSize(m_params.height, level), GetLevelSize(m_params.depth, level));
|
|
}
|
|
|
|
ImageType VulkanTexture::GetType() const
|
|
{
|
|
return m_params.type;
|
|
}
|
|
|
|
bool VulkanTexture::Update(const void* ptr, const Boxui& box, unsigned int srcWidth, unsigned int srcHeight, UInt8 level)
|
|
{
|
|
std::size_t textureSize = box.width * box.height * box.depth * PixelFormatInfo::GetBytesPerPixel(m_params.pixelFormat);
|
|
if (m_params.type == ImageType::Cubemap)
|
|
textureSize *= 6;
|
|
|
|
VkBufferCreateInfo createInfo = {};
|
|
createInfo.sType = VK_STRUCTURE_TYPE_BUFFER_CREATE_INFO;
|
|
createInfo.size = textureSize;
|
|
createInfo.usage = VK_BUFFER_USAGE_TRANSFER_SRC_BIT;
|
|
|
|
VmaAllocationCreateInfo allocInfo = {};
|
|
allocInfo.flags = VMA_ALLOCATION_CREATE_MAPPED_BIT;
|
|
allocInfo.usage = VMA_MEMORY_USAGE_CPU_TO_GPU;
|
|
|
|
VmaAllocationInfo allocationInfo;
|
|
|
|
VkBuffer stagingBuffer;
|
|
VmaAllocation stagingAllocation;
|
|
|
|
VkResult result = vmaCreateBuffer(m_device.GetMemoryAllocator(), &createInfo, &allocInfo, &stagingBuffer, &stagingAllocation, &allocationInfo);
|
|
if (result != VK_SUCCESS)
|
|
{
|
|
NazaraError("Failed to allocate staging buffer: " + TranslateVulkanError(result));
|
|
return false;
|
|
}
|
|
|
|
CallOnExit freeStaging([&] {
|
|
vmaDestroyBuffer(m_device.GetMemoryAllocator(), stagingBuffer, stagingAllocation);
|
|
});
|
|
|
|
if (srcWidth == 0)
|
|
srcWidth = box.width;
|
|
|
|
if (srcHeight == 0)
|
|
srcHeight = box.height;
|
|
|
|
if (srcWidth == box.width && srcHeight == box.height)
|
|
std::memcpy(allocationInfo.pMappedData, ptr, textureSize);
|
|
else
|
|
{
|
|
unsigned int dstWidth = box.width;
|
|
unsigned int dstHeight = box.height;
|
|
|
|
unsigned int bpp = PixelFormatInfo::GetBytesPerPixel(m_params.pixelFormat);
|
|
unsigned int lineStride = box.width * bpp;
|
|
unsigned int dstLineStride = dstWidth * bpp;
|
|
unsigned int dstFaceStride = dstLineStride * dstHeight;
|
|
unsigned int srcLineStride = srcWidth * bpp;
|
|
unsigned int srcFaceStride = srcLineStride * srcHeight;
|
|
|
|
const UInt8* source = static_cast<const UInt8*>(ptr);
|
|
UInt8* destination = static_cast<UInt8*>(allocationInfo.pMappedData);
|
|
for (unsigned int i = 0; i < box.depth; ++i)
|
|
{
|
|
UInt8* dstFacePtr = destination;
|
|
const UInt8* srcFacePtr = source;
|
|
for (unsigned int y = 0; y < box.height; ++y)
|
|
{
|
|
std::memcpy(dstFacePtr, srcFacePtr, lineStride);
|
|
|
|
dstFacePtr += dstLineStride;
|
|
srcFacePtr += srcLineStride;
|
|
}
|
|
|
|
destination += dstFaceStride;
|
|
source += srcFaceStride;
|
|
}
|
|
}
|
|
|
|
Vk::AutoCommandBuffer copyCommandBuffer = m_device.AllocateCommandBuffer(QueueType::Graphics);
|
|
if (!copyCommandBuffer->Begin(VK_COMMAND_BUFFER_USAGE_ONE_TIME_SUBMIT_BIT))
|
|
return false;
|
|
|
|
VkImageSubresourceLayers subresourceLayers = { //< FIXME
|
|
VK_IMAGE_ASPECT_COLOR_BIT,
|
|
level, //< mipLevel
|
|
0, //< baseArrayLayer
|
|
UInt32((m_params.type == ImageType::Cubemap) ? 6 : 1) //< layerCount
|
|
};
|
|
|
|
VkImageSubresourceRange subresourceRange = { //< FIXME
|
|
VK_IMAGE_ASPECT_COLOR_BIT,
|
|
0, //< baseMipLevel
|
|
1, //< levelCount
|
|
subresourceLayers.baseArrayLayer, //< baseArrayLayer
|
|
subresourceLayers.layerCount //< layerCount
|
|
};
|
|
|
|
VkBufferImageCopy region = {
|
|
0,
|
|
0,
|
|
0,
|
|
subresourceLayers,
|
|
{ // imageOffset
|
|
SafeCast<Int32>(box.x), SafeCast<Int32>(box.y), SafeCast<Int32>(box.z)
|
|
},
|
|
{ // imageExtent
|
|
box.width, box.height, box.depth
|
|
}
|
|
};
|
|
|
|
copyCommandBuffer->SetImageLayout(m_image, VK_PIPELINE_STAGE_TOP_OF_PIPE_BIT, VK_PIPELINE_STAGE_TRANSFER_BIT, VK_IMAGE_LAYOUT_UNDEFINED, VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL, subresourceRange);
|
|
|
|
copyCommandBuffer->CopyBufferToImage(stagingBuffer, m_image, VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL, region);
|
|
|
|
copyCommandBuffer->SetImageLayout(m_image, VK_PIPELINE_STAGE_TRANSFER_BIT, VK_PIPELINE_STAGE_FRAGMENT_SHADER_BIT, VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL, VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL, subresourceRange);
|
|
|
|
if (!copyCommandBuffer->End())
|
|
return false;
|
|
|
|
Vk::QueueHandle transferQueue = m_device.GetQueue(m_device.GetDefaultFamilyIndex(QueueType::Graphics), 0);
|
|
if (!transferQueue.Submit(copyCommandBuffer))
|
|
return false;
|
|
|
|
transferQueue.WaitIdle();
|
|
|
|
return true;
|
|
}
|
|
|
|
void VulkanTexture::InitForFormat(PixelFormat pixelFormat, VkImageCreateInfo& createImage, VkImageViewCreateInfo& createImageView)
|
|
{
|
|
createImageView.components = {
|
|
VK_COMPONENT_SWIZZLE_R,
|
|
VK_COMPONENT_SWIZZLE_G,
|
|
VK_COMPONENT_SWIZZLE_B,
|
|
VK_COMPONENT_SWIZZLE_A
|
|
};
|
|
|
|
// TODO: Fill this switch
|
|
switch (pixelFormat)
|
|
{
|
|
// Regular formats
|
|
case PixelFormat::BGR8:
|
|
case PixelFormat::BGR8_SRGB:
|
|
case PixelFormat::BGRA8:
|
|
case PixelFormat::BGRA8_SRGB:
|
|
case PixelFormat::Depth16:
|
|
case PixelFormat::Depth16Stencil8:
|
|
case PixelFormat::Depth24Stencil8:
|
|
case PixelFormat::Depth32F:
|
|
case PixelFormat::Depth32FStencil8:
|
|
case PixelFormat::RGB8:
|
|
case PixelFormat::RGB8_SRGB:
|
|
case PixelFormat::RGBA8:
|
|
case PixelFormat::RGBA8_SRGB:
|
|
case PixelFormat::RGBA16F:
|
|
case PixelFormat::RGBA32F:
|
|
{
|
|
createImage.format = ToVulkan(pixelFormat);
|
|
createImageView.format = createImage.format;
|
|
break;
|
|
}
|
|
|
|
// "emulated" formats
|
|
case PixelFormat::A8:
|
|
{
|
|
createImage.format = VK_FORMAT_R8_UNORM;
|
|
createImageView.format = createImage.format;
|
|
createImageView.components = {
|
|
VK_COMPONENT_SWIZZLE_ONE,
|
|
VK_COMPONENT_SWIZZLE_ONE,
|
|
VK_COMPONENT_SWIZZLE_ONE,
|
|
VK_COMPONENT_SWIZZLE_R
|
|
};
|
|
break;
|
|
}
|
|
|
|
case PixelFormat::L8:
|
|
{
|
|
createImage.format = VK_FORMAT_R8_UNORM;
|
|
createImageView.format = createImage.format;
|
|
createImageView.components = {
|
|
VK_COMPONENT_SWIZZLE_R,
|
|
VK_COMPONENT_SWIZZLE_R,
|
|
VK_COMPONENT_SWIZZLE_R,
|
|
VK_COMPONENT_SWIZZLE_ONE
|
|
};
|
|
break;
|
|
}
|
|
|
|
case PixelFormat::LA8:
|
|
{
|
|
createImage.format = VK_FORMAT_R8G8_UNORM;
|
|
createImageView.format = createImage.format;
|
|
createImageView.components = {
|
|
VK_COMPONENT_SWIZZLE_R,
|
|
VK_COMPONENT_SWIZZLE_R,
|
|
VK_COMPONENT_SWIZZLE_R,
|
|
VK_COMPONENT_SWIZZLE_G
|
|
};
|
|
break;
|
|
}
|
|
|
|
default:
|
|
throw std::runtime_error("Unsupported pixel format " + PixelFormatInfo::GetName(pixelFormat));
|
|
}
|
|
}
|
|
}
|