295 lines
8.7 KiB
C++
295 lines
8.7 KiB
C++
/*
|
|
* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at https://mozilla.org/MPL/2.0/.
|
|
*/
|
|
|
|
#include "ImageLoaderDds.hpp"
|
|
|
|
#include <dds.hpp>
|
|
|
|
namespace OpenVulkano::Image
|
|
{
|
|
std::unique_ptr<Image> ImageLoaderDds::loadFromFile(const std::string& filePath)
|
|
{
|
|
dds::Image image;
|
|
if (dds::readFile(filePath, &image) != dds::ReadResult::Success)
|
|
{
|
|
throw std::runtime_error("Failed to load DDS texture");
|
|
}
|
|
return ExtractImage(&image);
|
|
}
|
|
|
|
std::unique_ptr<Image> ImageLoaderDds::loadFromMemory(const std::vector<uint8_t>& buffer)
|
|
{
|
|
dds::Image image;
|
|
if (dds::readImage(const_cast<uint8_t*>(buffer.data()), buffer.size(), &image) != dds::ReadResult::Success)
|
|
{
|
|
throw std::runtime_error("Failed to load DDS texture");
|
|
}
|
|
return ExtractImage(&image);
|
|
}
|
|
|
|
Math::Vector2i ImageLoaderDds::GetImageDimensions(const std::string& filename)
|
|
{
|
|
dds::Image image;
|
|
if (dds::readFile(filename, &image) != dds::ReadResult::Success)
|
|
{
|
|
throw std::runtime_error("Failed to load DDS texture");
|
|
}
|
|
|
|
Math::Vector2i result;
|
|
result.x = image.width;
|
|
result.y = image.height;
|
|
|
|
return result;
|
|
}
|
|
|
|
std::unique_ptr<Image> ImageLoaderDds::ExtractImage(dds::Image* ddsImage)
|
|
{
|
|
auto image = std::make_unique<Image>();
|
|
|
|
image->resolution.x = ddsImage->width;
|
|
image->resolution.y = ddsImage->height;
|
|
image->resolution.z = 1;
|
|
|
|
if (ddsImage->mipmaps.size() == 0)
|
|
{
|
|
throw std::runtime_error("Failed to read an image from a texture with no mipmaps");
|
|
}
|
|
|
|
dds::span texture = ddsImage->mipmaps[0];
|
|
|
|
image->data = Array<uint8_t>(texture.size_bytes());
|
|
memcpy(image->data.Data(), texture.data(), texture.size_bytes());
|
|
|
|
image->dataFormat = DataFormat::Format::UNDEFINED;
|
|
switch (ddsImage->format)
|
|
{
|
|
case DXGI_FORMAT_R32G32B32A32_FLOAT:
|
|
image->dataFormat = DataFormat::Format::R32G32B32A32_SFLOAT;
|
|
break;
|
|
case DXGI_FORMAT_R32G32B32A32_UINT:
|
|
image->dataFormat = DataFormat::Format::R32G32B32A32_UINT;
|
|
break;
|
|
case DXGI_FORMAT_R32G32B32A32_SINT:
|
|
image->dataFormat = DataFormat::Format::R32G32B32A32_SINT;
|
|
break;
|
|
case DXGI_FORMAT_R32G32B32_FLOAT:
|
|
image->dataFormat = DataFormat::Format::R32G32B32_SFLOAT;
|
|
break;
|
|
case DXGI_FORMAT_R32G32B32_UINT:
|
|
image->dataFormat = DataFormat::Format::R32G32B32_UINT;
|
|
break;
|
|
case DXGI_FORMAT_R32G32B32_SINT:
|
|
image->dataFormat = DataFormat::Format::R32G32B32_SINT;
|
|
break;
|
|
|
|
case DXGI_FORMAT_R16G16B16A16_FLOAT:
|
|
image->dataFormat = DataFormat::Format::R16G16B16A16_SFLOAT;
|
|
break;
|
|
case DXGI_FORMAT_R16G16B16A16_UNORM:
|
|
image->dataFormat = DataFormat::Format::R16G16B16A16_UNORM;
|
|
break;
|
|
case DXGI_FORMAT_R16G16B16A16_UINT:
|
|
image->dataFormat = DataFormat::Format::R16G16B16A16_UINT;
|
|
break;
|
|
case DXGI_FORMAT_R16G16B16A16_SNORM:
|
|
image->dataFormat = DataFormat::Format::R16G16B16A16_SNORM;
|
|
break;
|
|
case DXGI_FORMAT_R16G16B16A16_SINT:
|
|
image->dataFormat = DataFormat::Format::R16G16B16A16_SINT;
|
|
break;
|
|
|
|
case DXGI_FORMAT_R32G32_FLOAT:
|
|
image->dataFormat = DataFormat::Format::R32G32_SFLOAT;
|
|
break;
|
|
case DXGI_FORMAT_R32G32_UINT:
|
|
image->dataFormat = DataFormat::Format::R32G32_UINT;
|
|
break;
|
|
case DXGI_FORMAT_R32G32_SINT:
|
|
image->dataFormat = DataFormat::Format::R32G32_SINT;
|
|
break;
|
|
|
|
case DXGI_FORMAT_D32_FLOAT_S8X24_UINT:
|
|
image->dataFormat = DataFormat::Format::D32_SFLOAT_S8_UINT;
|
|
break;
|
|
|
|
case DXGI_FORMAT_R8G8B8A8_UNORM:
|
|
image->dataFormat = DataFormat::Format::R8G8B8A8_UNORM;
|
|
break;
|
|
case DXGI_FORMAT_R8G8B8A8_UNORM_SRGB:
|
|
image->dataFormat = DataFormat::Format::R8G8B8A8_SRGB;
|
|
break;
|
|
case DXGI_FORMAT_R8G8B8A8_UINT:
|
|
image->dataFormat = DataFormat::Format::R8G8B8A8_UINT;
|
|
break;
|
|
case DXGI_FORMAT_R8G8B8A8_SNORM:
|
|
image->dataFormat = DataFormat::Format::R8G8B8A8_SNORM;
|
|
break;
|
|
case DXGI_FORMAT_R8G8B8A8_SINT:
|
|
image->dataFormat = DataFormat::Format::R8G8B8A8_SINT;
|
|
break;
|
|
|
|
case DXGI_FORMAT_R16G16_FLOAT:
|
|
image->dataFormat = DataFormat::Format::R16G16_SFLOAT;
|
|
break;
|
|
case DXGI_FORMAT_R16G16_UNORM:
|
|
image->dataFormat = DataFormat::Format::R16G16_UNORM;
|
|
break;
|
|
case DXGI_FORMAT_R16G16_UINT:
|
|
image->dataFormat = DataFormat::Format::R16G16_UINT;
|
|
break;
|
|
case DXGI_FORMAT_R16G16_SNORM:
|
|
image->dataFormat = DataFormat::Format::R16G16_SNORM;
|
|
break;
|
|
case DXGI_FORMAT_R16G16_SINT:
|
|
image->dataFormat = DataFormat::Format::R16G16_SINT;
|
|
break;
|
|
|
|
case DXGI_FORMAT_D32_FLOAT:
|
|
image->dataFormat = DataFormat::Format::D32_SFLOAT;
|
|
break;
|
|
|
|
case DXGI_FORMAT_R32_FLOAT:
|
|
image->dataFormat = DataFormat::Format::R32_SFLOAT;
|
|
break;
|
|
case DXGI_FORMAT_R32_UINT:
|
|
image->dataFormat = DataFormat::Format::R32_UINT;
|
|
break;
|
|
case DXGI_FORMAT_R32_SINT:
|
|
image->dataFormat = DataFormat::Format::R32_SINT;
|
|
break;
|
|
|
|
case DXGI_FORMAT_D24_UNORM_S8_UINT:
|
|
image->dataFormat = DataFormat::Format::D24_UNORM_S8_UINT;
|
|
break;
|
|
|
|
case DXGI_FORMAT_R8G8_UNORM:
|
|
image->dataFormat = DataFormat::Format::R8G8_UNORM;
|
|
break;
|
|
case DXGI_FORMAT_R8G8_UINT:
|
|
image->dataFormat = DataFormat::Format::R8G8_UINT;
|
|
break;
|
|
case DXGI_FORMAT_R8G8_SNORM:
|
|
image->dataFormat = DataFormat::Format::R8G8_SNORM;
|
|
break;
|
|
case DXGI_FORMAT_R8G8_SINT:
|
|
image->dataFormat = DataFormat::Format::R8G8_SINT;
|
|
break;
|
|
|
|
case DXGI_FORMAT_D16_UNORM:
|
|
image->dataFormat = DataFormat::Format::D16_UNORM;
|
|
break;
|
|
case DXGI_FORMAT_R16_FLOAT:
|
|
image->dataFormat = DataFormat::Format::R16_SFLOAT;
|
|
break;
|
|
case DXGI_FORMAT_R16_UNORM:
|
|
image->dataFormat = DataFormat::Format::R16_UNORM;
|
|
break;
|
|
case DXGI_FORMAT_R16_UINT:
|
|
image->dataFormat = DataFormat::Format::R16_UINT;
|
|
break;
|
|
case DXGI_FORMAT_R16_SNORM:
|
|
image->dataFormat = DataFormat::Format::R16_SNORM;
|
|
break;
|
|
case DXGI_FORMAT_R16_SINT:
|
|
image->dataFormat = DataFormat::Format::R16_SINT;
|
|
break;
|
|
|
|
case DXGI_FORMAT_R8_UNORM:
|
|
image->dataFormat = DataFormat::Format::R8_UNORM;
|
|
break;
|
|
case DXGI_FORMAT_R8_UINT:
|
|
image->dataFormat = DataFormat::Format::R8_UINT;
|
|
break;
|
|
case DXGI_FORMAT_R8_SNORM:
|
|
image->dataFormat = DataFormat::Format::R8_SNORM;
|
|
break;
|
|
case DXGI_FORMAT_R8_SINT:
|
|
image->dataFormat = DataFormat::Format::R8_SINT;
|
|
break;
|
|
|
|
case DXGI_FORMAT_BC1_UNORM:
|
|
image->dataFormat = DataFormat::Format::BC1_RGBA_UNORM_BLOCK;
|
|
break;
|
|
case DXGI_FORMAT_BC1_UNORM_SRGB:
|
|
image->dataFormat = DataFormat::Format::BC1_RGBA_SRGB_BLOCK;
|
|
break;
|
|
|
|
case DXGI_FORMAT_BC2_UNORM:
|
|
image->dataFormat = DataFormat::Format::BC2_UNORM_BLOCK;
|
|
break;
|
|
case DXGI_FORMAT_BC2_UNORM_SRGB:
|
|
image->dataFormat = DataFormat::Format::BC2_SRGB_BLOCK;
|
|
break;
|
|
|
|
case DXGI_FORMAT_BC3_UNORM:
|
|
image->dataFormat = DataFormat::Format::BC3_UNORM_BLOCK;
|
|
break;
|
|
case DXGI_FORMAT_BC3_UNORM_SRGB:
|
|
image->dataFormat = DataFormat::Format::BC3_SRGB_BLOCK;
|
|
break;
|
|
|
|
case DXGI_FORMAT_BC4_UNORM:
|
|
image->dataFormat = DataFormat::Format::BC4_UNORM_BLOCK;
|
|
break;
|
|
case DXGI_FORMAT_BC4_SNORM:
|
|
image->dataFormat = DataFormat::Format::BC4_SNORM_BLOCK;
|
|
break;
|
|
|
|
case DXGI_FORMAT_BC5_UNORM:
|
|
image->dataFormat = DataFormat::Format::BC5_UNORM_BLOCK;
|
|
break;
|
|
case DXGI_FORMAT_BC5_SNORM:
|
|
image->dataFormat = DataFormat::Format::BC5_SNORM_BLOCK;
|
|
break;
|
|
|
|
case DXGI_FORMAT_B5G6R5_UNORM:
|
|
image->dataFormat = DataFormat::Format::B5G6R5_UNORM_PACK16;
|
|
break;
|
|
case DXGI_FORMAT_B5G5R5A1_UNORM:
|
|
image->dataFormat = DataFormat::Format::B5G5R5A1_UNORM_PACK16;
|
|
break;
|
|
|
|
case DXGI_FORMAT_B8G8R8A8_UNORM:
|
|
image->dataFormat = DataFormat::Format::B8G8R8A8_UNORM;
|
|
break;
|
|
case DXGI_FORMAT_B8G8R8X8_UNORM:
|
|
image->dataFormat = DataFormat::Format::B8G8R8A8_SNORM;
|
|
break;
|
|
|
|
case DXGI_FORMAT_B8G8R8A8_UNORM_SRGB:
|
|
image->dataFormat = DataFormat::Format::B8G8R8A8_SRGB;
|
|
break;
|
|
case DXGI_FORMAT_B8G8R8X8_UNORM_SRGB:
|
|
image->dataFormat = DataFormat::Format::B8G8R8A8_SRGB;
|
|
break;
|
|
|
|
case DXGI_FORMAT_BC6H_UF16:
|
|
image->dataFormat = DataFormat::Format::BC6H_UFLOAT_BLOCK;
|
|
break;
|
|
case DXGI_FORMAT_BC6H_SF16:
|
|
image->dataFormat = DataFormat::Format::BC6H_SFLOAT_BLOCK;
|
|
break;
|
|
|
|
case DXGI_FORMAT_BC7_UNORM:
|
|
image->dataFormat = DataFormat::Format::BC7_UNORM_BLOCK;
|
|
break;
|
|
case DXGI_FORMAT_BC7_UNORM_SRGB:
|
|
image->dataFormat = DataFormat::Format::BC7_SRGB_BLOCK;
|
|
break;
|
|
|
|
case DXGI_FORMAT_B4G4R4A4_UNORM:
|
|
image->dataFormat = DataFormat::Format::B4G4R4A4_UNORM_PACK16;
|
|
break;
|
|
|
|
default:
|
|
throw std::runtime_error("Unhandled DDS texture format: "
|
|
+ std::to_string(*reinterpret_cast<int*>(&ddsImage->format)));
|
|
}
|
|
|
|
return image;
|
|
}
|
|
}
|