1585 lines
49 KiB
C++
1585 lines
49 KiB
C++
//-------------------------------------------------------------------------------------
|
|
// DirectXTexTGA.cpp
|
|
//
|
|
// DirectX Texture Library - Targa Truevision (TGA) file format reader/writer
|
|
//
|
|
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
//
|
|
// http://go.microsoft.com/fwlink/?LinkId=248926
|
|
//-------------------------------------------------------------------------------------
|
|
|
|
#include "DirectXTexP.h"
|
|
|
|
//
|
|
// The implementation here has the following limitations:
|
|
// * Does not support files that contain color maps (these are rare in practice)
|
|
// * Interleaved files are not supported (deprecated aspect of TGA format)
|
|
// * Only supports 8-bit grayscale; 16-, 24-, and 32-bit truecolor images
|
|
// * Always writes uncompressed files (i.e. can read RLE compression, but does not write it)
|
|
//
|
|
|
|
using namespace DirectX;
|
|
|
|
namespace
|
|
{
|
|
const char g_Signature[] = "TRUEVISION-XFILE.";
|
|
// This is the official footer signature for the TGA 2.0 file format.
|
|
|
|
enum TGAImageType
|
|
{
|
|
TGA_NO_IMAGE = 0,
|
|
TGA_COLOR_MAPPED = 1,
|
|
TGA_TRUECOLOR = 2,
|
|
TGA_BLACK_AND_WHITE = 3,
|
|
TGA_COLOR_MAPPED_RLE = 9,
|
|
TGA_TRUECOLOR_RLE = 10,
|
|
TGA_BLACK_AND_WHITE_RLE = 11,
|
|
};
|
|
|
|
enum TGADescriptorFlags
|
|
{
|
|
TGA_FLAGS_INVERTX = 0x10,
|
|
TGA_FLAGS_INVERTY = 0x20,
|
|
TGA_FLAGS_INTERLEAVED_2WAY = 0x40, // Deprecated
|
|
TGA_FLAGS_INTERLEAVED_4WAY = 0x80, // Deprecated
|
|
};
|
|
|
|
enum TGAAttributesType : uint8_t
|
|
{
|
|
TGA_ATTRIBUTE_NONE = 0, // 0: no alpha data included
|
|
TGA_ATTRIBUTE_IGNORED = 1, // 1: undefined data, can be ignored
|
|
TGA_ATTRIBUTE_UNDEFINED = 2, // 2: uedefined data, should be retained
|
|
TGA_ATTRIBUTE_ALPHA = 3, // 3: useful alpha channel data
|
|
TGA_ATTRIBUTE_PREMULTIPLIED = 4, // 4: pre-multiplied alpha
|
|
};
|
|
|
|
#pragma pack(push,1)
|
|
struct TGA_HEADER
|
|
{
|
|
uint8_t bIDLength;
|
|
uint8_t bColorMapType;
|
|
uint8_t bImageType;
|
|
uint16_t wColorMapFirst;
|
|
uint16_t wColorMapLength;
|
|
uint8_t bColorMapSize;
|
|
uint16_t wXOrigin;
|
|
uint16_t wYOrigin;
|
|
uint16_t wWidth;
|
|
uint16_t wHeight;
|
|
uint8_t bBitsPerPixel;
|
|
uint8_t bDescriptor;
|
|
};
|
|
|
|
static_assert(sizeof(TGA_HEADER) == 18, "TGA 2.0 size mismatch");
|
|
|
|
struct TGA_FOOTER
|
|
{
|
|
uint32_t dwExtensionOffset;
|
|
uint32_t dwDeveloperOffset;
|
|
char Signature[18];
|
|
};
|
|
|
|
static_assert(sizeof(TGA_FOOTER) == 26, "TGA 2.0 size mismatch");
|
|
|
|
struct TGA_EXTENSION
|
|
{
|
|
uint16_t wSize;
|
|
char szAuthorName[41];
|
|
char szAuthorComment[324];
|
|
uint16_t wStampMonth;
|
|
uint16_t wStampDay;
|
|
uint16_t wStampYear;
|
|
uint16_t wStampHour;
|
|
uint16_t wStampMinute;
|
|
uint16_t wStampSecond;
|
|
char szJobName[41];
|
|
uint16_t wJobHour;
|
|
uint16_t wJobMinute;
|
|
uint16_t wJobSecond;
|
|
char szSoftwareId[41];
|
|
uint16_t wVersionNumber;
|
|
uint8_t bVersionLetter;
|
|
uint32_t dwKeyColor;
|
|
uint16_t wPixelNumerator;
|
|
uint16_t wPixelDenominator;
|
|
uint16_t wGammaNumerator;
|
|
uint16_t wGammaDenominator;
|
|
uint32_t dwColorOffset;
|
|
uint32_t dwStampOffset;
|
|
uint32_t dwScanOffset;
|
|
uint8_t bAttributesType;
|
|
};
|
|
|
|
static_assert(sizeof(TGA_EXTENSION) == 495, "TGA 2.0 size mismatch");
|
|
|
|
#pragma pack(pop)
|
|
|
|
enum CONVERSION_FLAGS
|
|
{
|
|
CONV_FLAGS_NONE = 0x0,
|
|
CONV_FLAGS_EXPAND = 0x1, // Conversion requires expanded pixel size
|
|
CONV_FLAGS_INVERTX = 0x2, // If set, scanlines are right-to-left
|
|
CONV_FLAGS_INVERTY = 0x4, // If set, scanlines are top-to-bottom
|
|
CONV_FLAGS_RLE = 0x8, // Source data is RLE compressed
|
|
|
|
CONV_FLAGS_SWIZZLE = 0x10000, // Swizzle BGR<->RGB data
|
|
CONV_FLAGS_888 = 0x20000, // 24bpp format
|
|
};
|
|
|
|
|
|
//-------------------------------------------------------------------------------------
|
|
// Decodes TGA header
|
|
//-------------------------------------------------------------------------------------
|
|
HRESULT DecodeTGAHeader(
|
|
_In_reads_bytes_(size) const void* pSource,
|
|
size_t size,
|
|
_Out_ TexMetadata& metadata,
|
|
size_t& offset,
|
|
_Inout_opt_ uint32_t* convFlags) noexcept
|
|
{
|
|
if (!pSource)
|
|
return E_INVALIDARG;
|
|
|
|
memset(&metadata, 0, sizeof(TexMetadata));
|
|
|
|
if (size < sizeof(TGA_HEADER))
|
|
{
|
|
return HRESULT_FROM_WIN32(ERROR_INVALID_DATA);
|
|
}
|
|
|
|
auto pHeader = static_cast<const TGA_HEADER*>(pSource);
|
|
|
|
if (pHeader->bColorMapType != 0
|
|
|| pHeader->wColorMapLength != 0)
|
|
{
|
|
return HRESULT_FROM_WIN32(ERROR_NOT_SUPPORTED);
|
|
}
|
|
|
|
if (pHeader->bDescriptor & (TGA_FLAGS_INTERLEAVED_2WAY | TGA_FLAGS_INTERLEAVED_4WAY))
|
|
{
|
|
return HRESULT_FROM_WIN32(ERROR_NOT_SUPPORTED);
|
|
}
|
|
|
|
if (!pHeader->wWidth || !pHeader->wHeight)
|
|
{
|
|
return HRESULT_FROM_WIN32(ERROR_INVALID_DATA);
|
|
}
|
|
|
|
switch (pHeader->bImageType)
|
|
{
|
|
case TGA_TRUECOLOR:
|
|
case TGA_TRUECOLOR_RLE:
|
|
switch (pHeader->bBitsPerPixel)
|
|
{
|
|
case 16:
|
|
metadata.format = DXGI_FORMAT_B5G5R5A1_UNORM;
|
|
break;
|
|
|
|
case 24:
|
|
metadata.format = DXGI_FORMAT_R8G8B8A8_UNORM;
|
|
metadata.SetAlphaMode(TEX_ALPHA_MODE_OPAQUE);
|
|
if (convFlags)
|
|
*convFlags |= CONV_FLAGS_EXPAND;
|
|
// We could use DXGI_FORMAT_B8G8R8X8_UNORM, but we prefer DXGI 1.0 formats
|
|
break;
|
|
|
|
case 32:
|
|
metadata.format = DXGI_FORMAT_R8G8B8A8_UNORM;
|
|
// We could use DXGI_FORMAT_B8G8R8A8_UNORM, but we prefer DXGI 1.0 formats
|
|
break;
|
|
}
|
|
|
|
if (convFlags && (pHeader->bImageType == TGA_TRUECOLOR_RLE))
|
|
{
|
|
*convFlags |= CONV_FLAGS_RLE;
|
|
}
|
|
break;
|
|
|
|
case TGA_BLACK_AND_WHITE:
|
|
case TGA_BLACK_AND_WHITE_RLE:
|
|
switch (pHeader->bBitsPerPixel)
|
|
{
|
|
case 8:
|
|
metadata.format = DXGI_FORMAT_R8_UNORM;
|
|
break;
|
|
|
|
default:
|
|
return HRESULT_FROM_WIN32(ERROR_NOT_SUPPORTED);
|
|
}
|
|
|
|
if (convFlags && (pHeader->bImageType == TGA_BLACK_AND_WHITE_RLE))
|
|
{
|
|
*convFlags |= CONV_FLAGS_RLE;
|
|
}
|
|
break;
|
|
|
|
case TGA_NO_IMAGE:
|
|
case TGA_COLOR_MAPPED:
|
|
case TGA_COLOR_MAPPED_RLE:
|
|
return HRESULT_FROM_WIN32(ERROR_NOT_SUPPORTED);
|
|
|
|
default:
|
|
return HRESULT_FROM_WIN32(ERROR_INVALID_DATA);
|
|
}
|
|
|
|
metadata.width = pHeader->wWidth;
|
|
metadata.height = pHeader->wHeight;
|
|
metadata.depth = metadata.arraySize = metadata.mipLevels = 1;
|
|
metadata.dimension = TEX_DIMENSION_TEXTURE2D;
|
|
|
|
if (convFlags)
|
|
{
|
|
if (pHeader->bDescriptor & TGA_FLAGS_INVERTX)
|
|
*convFlags |= CONV_FLAGS_INVERTX;
|
|
|
|
if (pHeader->bDescriptor & TGA_FLAGS_INVERTY)
|
|
*convFlags |= CONV_FLAGS_INVERTY;
|
|
}
|
|
|
|
offset = sizeof(TGA_HEADER);
|
|
|
|
if (pHeader->bIDLength != 0)
|
|
{
|
|
offset += pHeader->bIDLength;
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
//-------------------------------------------------------------------------------------
|
|
// Set alpha for images with all 0 alpha channel
|
|
//-------------------------------------------------------------------------------------
|
|
HRESULT SetAlphaChannelToOpaque(_In_ const Image* image) noexcept
|
|
{
|
|
assert(image);
|
|
|
|
uint8_t* pPixels = image->pixels;
|
|
if (!pPixels)
|
|
return E_POINTER;
|
|
|
|
for (size_t y = 0; y < image->height; ++y)
|
|
{
|
|
_CopyScanline(pPixels, image->rowPitch, pPixels, image->rowPitch, image->format, TEXP_SCANLINE_SETALPHA);
|
|
pPixels += image->rowPitch;
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
//-------------------------------------------------------------------------------------
|
|
// Uncompress pixel data from a TGA into the target image
|
|
//-------------------------------------------------------------------------------------
|
|
HRESULT UncompressPixels(
|
|
_In_reads_bytes_(size) const void* pSource,
|
|
size_t size,
|
|
_In_ const Image* image,
|
|
_In_ uint32_t convFlags) noexcept
|
|
{
|
|
assert(pSource && size > 0);
|
|
|
|
if (!image || !image->pixels)
|
|
return E_POINTER;
|
|
|
|
// Compute TGA image data pitch
|
|
size_t rowPitch, slicePitch;
|
|
HRESULT hr = ComputePitch(image->format, image->width, image->height, rowPitch, slicePitch,
|
|
(convFlags & CONV_FLAGS_EXPAND) ? CP_FLAGS_24BPP : CP_FLAGS_NONE);
|
|
if (FAILED(hr))
|
|
return hr;
|
|
|
|
auto sPtr = static_cast<const uint8_t*>(pSource);
|
|
const uint8_t* endPtr = sPtr + size;
|
|
|
|
bool opaquealpha = false;
|
|
|
|
switch (image->format)
|
|
{
|
|
//--------------------------------------------------------------------------- 8-bit
|
|
case DXGI_FORMAT_R8_UNORM:
|
|
for (size_t y = 0; y < image->height; ++y)
|
|
{
|
|
size_t offset = ((convFlags & CONV_FLAGS_INVERTX) ? (image->width - 1) : 0);
|
|
assert(offset < rowPitch);
|
|
|
|
uint8_t* dPtr = image->pixels
|
|
+ (image->rowPitch * ((convFlags & CONV_FLAGS_INVERTY) ? y : (image->height - y - 1)))
|
|
+ offset;
|
|
|
|
for (size_t x = 0; x < image->width; )
|
|
{
|
|
if (sPtr >= endPtr)
|
|
return E_FAIL;
|
|
|
|
if (*sPtr & 0x80)
|
|
{
|
|
// Repeat
|
|
size_t j = size_t(*sPtr & 0x7F) + 1;
|
|
if (++sPtr >= endPtr)
|
|
return E_FAIL;
|
|
|
|
for (; j > 0; --j, ++x)
|
|
{
|
|
if (x >= image->width)
|
|
return E_FAIL;
|
|
|
|
*dPtr = *sPtr;
|
|
|
|
if (convFlags & CONV_FLAGS_INVERTX)
|
|
--dPtr;
|
|
else
|
|
++dPtr;
|
|
}
|
|
|
|
++sPtr;
|
|
}
|
|
else
|
|
{
|
|
// Literal
|
|
size_t j = size_t(*sPtr & 0x7F) + 1;
|
|
++sPtr;
|
|
|
|
if (sPtr + j > endPtr)
|
|
return E_FAIL;
|
|
|
|
for (; j > 0; --j, ++x)
|
|
{
|
|
if (x >= image->width)
|
|
return E_FAIL;
|
|
|
|
*dPtr = *(sPtr++);
|
|
|
|
if (convFlags & CONV_FLAGS_INVERTX)
|
|
--dPtr;
|
|
else
|
|
++dPtr;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
break;
|
|
|
|
//-------------------------------------------------------------------------- 16-bit
|
|
case DXGI_FORMAT_B5G5R5A1_UNORM:
|
|
{
|
|
uint32_t minalpha = 255;
|
|
uint32_t maxalpha = 0;
|
|
|
|
for (size_t y = 0; y < image->height; ++y)
|
|
{
|
|
size_t offset = ((convFlags & CONV_FLAGS_INVERTX) ? (image->width - 1) : 0);
|
|
assert(offset * 2 < rowPitch);
|
|
|
|
auto dPtr = reinterpret_cast<uint16_t*>(image->pixels
|
|
+ (image->rowPitch * ((convFlags & CONV_FLAGS_INVERTY) ? y : (image->height - y - 1))))
|
|
+ offset;
|
|
|
|
for (size_t x = 0; x < image->width; )
|
|
{
|
|
if (sPtr >= endPtr)
|
|
return E_FAIL;
|
|
|
|
if (*sPtr & 0x80)
|
|
{
|
|
// Repeat
|
|
size_t j = size_t(*sPtr & 0x7F) + 1;
|
|
++sPtr;
|
|
|
|
if (sPtr + 1 >= endPtr)
|
|
return E_FAIL;
|
|
|
|
auto t = static_cast<uint16_t>(uint32_t(*sPtr) | uint32_t(*(sPtr + 1u) << 8));
|
|
|
|
uint32_t alpha = (t & 0x8000) ? 255 : 0;
|
|
minalpha = std::min(minalpha, alpha);
|
|
maxalpha = std::max(maxalpha, alpha);
|
|
|
|
sPtr += 2;
|
|
|
|
for (; j > 0; --j, ++x)
|
|
{
|
|
if (x >= image->width)
|
|
return E_FAIL;
|
|
|
|
*dPtr = t;
|
|
|
|
if (convFlags & CONV_FLAGS_INVERTX)
|
|
--dPtr;
|
|
else
|
|
++dPtr;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// Literal
|
|
size_t j = size_t(*sPtr & 0x7F) + 1;
|
|
++sPtr;
|
|
|
|
if (sPtr + (j * 2) > endPtr)
|
|
return E_FAIL;
|
|
|
|
for (; j > 0; --j, ++x)
|
|
{
|
|
if (x >= image->width)
|
|
return E_FAIL;
|
|
|
|
auto t = static_cast<uint16_t>(uint32_t(*sPtr) | uint32_t(*(sPtr + 1u) << 8));
|
|
|
|
uint32_t alpha = (t & 0x8000) ? 255 : 0;
|
|
minalpha = std::min(minalpha, alpha);
|
|
maxalpha = std::max(maxalpha, alpha);
|
|
|
|
sPtr += 2;
|
|
*dPtr = t;
|
|
|
|
if (convFlags & CONV_FLAGS_INVERTX)
|
|
--dPtr;
|
|
else
|
|
++dPtr;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// If there are no non-zero alpha channel entries, we'll assume alpha is not used and force it to opaque
|
|
if (maxalpha == 0)
|
|
{
|
|
opaquealpha = true;
|
|
hr = SetAlphaChannelToOpaque(image);
|
|
if (FAILED(hr))
|
|
return hr;
|
|
}
|
|
else if (minalpha == 255)
|
|
{
|
|
opaquealpha = true;
|
|
}
|
|
}
|
|
break;
|
|
|
|
//----------------------------------------------------------------------- 24/32-bit
|
|
case DXGI_FORMAT_R8G8B8A8_UNORM:
|
|
{
|
|
uint32_t minalpha = 255;
|
|
uint32_t maxalpha = 0;
|
|
|
|
for (size_t y = 0; y < image->height; ++y)
|
|
{
|
|
size_t offset = ((convFlags & CONV_FLAGS_INVERTX) ? (image->width - 1) : 0);
|
|
|
|
auto dPtr = reinterpret_cast<uint32_t*>(image->pixels
|
|
+ (image->rowPitch * ((convFlags & CONV_FLAGS_INVERTY) ? y : (image->height - y - 1))))
|
|
+ offset;
|
|
|
|
for (size_t x = 0; x < image->width; )
|
|
{
|
|
if (sPtr >= endPtr)
|
|
return E_FAIL;
|
|
|
|
if (*sPtr & 0x80)
|
|
{
|
|
// Repeat
|
|
size_t j = size_t(*sPtr & 0x7F) + 1;
|
|
++sPtr;
|
|
|
|
uint32_t t;
|
|
if (convFlags & CONV_FLAGS_EXPAND)
|
|
{
|
|
assert(offset * 3 < rowPitch);
|
|
|
|
if (sPtr + 2 >= endPtr)
|
|
return E_FAIL;
|
|
|
|
// BGR -> RGBA
|
|
t = uint32_t(*sPtr << 16) | uint32_t(*(sPtr + 1) << 8) | uint32_t(*(sPtr + 2)) | 0xFF000000;
|
|
sPtr += 3;
|
|
|
|
minalpha = maxalpha = 255;
|
|
}
|
|
else
|
|
{
|
|
assert(offset * 4 < rowPitch);
|
|
|
|
if (sPtr + 3 >= endPtr)
|
|
return E_FAIL;
|
|
|
|
// BGRA -> RGBA
|
|
uint32_t alpha = *(sPtr + 3);
|
|
t = uint32_t(*sPtr << 16) | uint32_t(*(sPtr + 1) << 8) | uint32_t(*(sPtr + 2)) | uint32_t(alpha << 24);
|
|
|
|
minalpha = std::min(minalpha, alpha);
|
|
maxalpha = std::max(maxalpha, alpha);
|
|
|
|
sPtr += 4;
|
|
}
|
|
|
|
for (; j > 0; --j, ++x)
|
|
{
|
|
if (x >= image->width)
|
|
return E_FAIL;
|
|
|
|
*dPtr = t;
|
|
|
|
if (convFlags & CONV_FLAGS_INVERTX)
|
|
--dPtr;
|
|
else
|
|
++dPtr;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// Literal
|
|
size_t j = size_t(*sPtr & 0x7F) + 1;
|
|
++sPtr;
|
|
|
|
if (convFlags & CONV_FLAGS_EXPAND)
|
|
{
|
|
if (sPtr + (j * 3) > endPtr)
|
|
return E_FAIL;
|
|
}
|
|
else
|
|
{
|
|
if (sPtr + (j * 4) > endPtr)
|
|
return E_FAIL;
|
|
}
|
|
|
|
for (; j > 0; --j, ++x)
|
|
{
|
|
if (x >= image->width)
|
|
return E_FAIL;
|
|
|
|
if (convFlags & CONV_FLAGS_EXPAND)
|
|
{
|
|
assert(offset * 3 < rowPitch);
|
|
|
|
if (sPtr + 2 >= endPtr)
|
|
return E_FAIL;
|
|
|
|
// BGR -> RGBA
|
|
*dPtr = uint32_t(*sPtr << 16) | uint32_t(*(sPtr + 1) << 8) | uint32_t(*(sPtr + 2)) | 0xFF000000;
|
|
sPtr += 3;
|
|
|
|
minalpha = maxalpha = 255;
|
|
}
|
|
else
|
|
{
|
|
assert(offset * 4 < rowPitch);
|
|
|
|
if (sPtr + 3 >= endPtr)
|
|
return E_FAIL;
|
|
|
|
// BGRA -> RGBA
|
|
uint32_t alpha = *(sPtr + 3);
|
|
*dPtr = uint32_t(*sPtr << 16) | uint32_t(*(sPtr + 1) << 8) | uint32_t(*(sPtr + 2)) | uint32_t(alpha << 24);
|
|
|
|
minalpha = std::min(minalpha, alpha);
|
|
maxalpha = std::max(maxalpha, alpha);
|
|
|
|
sPtr += 4;
|
|
}
|
|
|
|
if (convFlags & CONV_FLAGS_INVERTX)
|
|
--dPtr;
|
|
else
|
|
++dPtr;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// If there are no non-zero alpha channel entries, we'll assume alpha is not used and force it to opaque
|
|
if (maxalpha == 0)
|
|
{
|
|
opaquealpha = true;
|
|
hr = SetAlphaChannelToOpaque(image);
|
|
if (FAILED(hr))
|
|
return hr;
|
|
}
|
|
else if (minalpha == 255)
|
|
{
|
|
opaquealpha = true;
|
|
}
|
|
}
|
|
break;
|
|
|
|
//---------------------------------------------------------------------------------
|
|
default:
|
|
return E_FAIL;
|
|
}
|
|
|
|
return opaquealpha ? S_FALSE : S_OK;
|
|
}
|
|
|
|
|
|
//-------------------------------------------------------------------------------------
|
|
// Copies pixel data from a TGA into the target image
|
|
//-------------------------------------------------------------------------------------
|
|
HRESULT CopyPixels(
|
|
_In_reads_bytes_(size) const void* pSource,
|
|
size_t size,
|
|
_In_ const Image* image,
|
|
_In_ uint32_t convFlags) noexcept
|
|
{
|
|
assert(pSource && size > 0);
|
|
|
|
if (!image || !image->pixels)
|
|
return E_POINTER;
|
|
|
|
// Compute TGA image data pitch
|
|
size_t rowPitch, slicePitch;
|
|
HRESULT hr = ComputePitch(image->format, image->width, image->height, rowPitch, slicePitch,
|
|
(convFlags & CONV_FLAGS_EXPAND) ? CP_FLAGS_24BPP : CP_FLAGS_NONE);
|
|
if (FAILED(hr))
|
|
return hr;
|
|
|
|
auto sPtr = static_cast<const uint8_t*>(pSource);
|
|
const uint8_t* endPtr = sPtr + size;
|
|
|
|
bool opaquealpha = false;
|
|
|
|
switch (image->format)
|
|
{
|
|
//--------------------------------------------------------------------------- 8-bit
|
|
case DXGI_FORMAT_R8_UNORM:
|
|
for (size_t y = 0; y < image->height; ++y)
|
|
{
|
|
size_t offset = ((convFlags & CONV_FLAGS_INVERTX) ? (image->width - 1) : 0);
|
|
assert(offset < rowPitch);
|
|
|
|
uint8_t* dPtr = image->pixels
|
|
+ (image->rowPitch * ((convFlags & CONV_FLAGS_INVERTY) ? y : (image->height - y - 1)))
|
|
+ offset;
|
|
|
|
for (size_t x = 0; x < image->width; ++x)
|
|
{
|
|
if (sPtr >= endPtr)
|
|
return E_FAIL;
|
|
|
|
*dPtr = *(sPtr++);
|
|
|
|
if (convFlags & CONV_FLAGS_INVERTX)
|
|
--dPtr;
|
|
else
|
|
++dPtr;
|
|
}
|
|
}
|
|
break;
|
|
|
|
//-------------------------------------------------------------------------- 16-bit
|
|
case DXGI_FORMAT_B5G5R5A1_UNORM:
|
|
{
|
|
uint32_t minalpha = 255;
|
|
uint32_t maxalpha = 0;
|
|
|
|
for (size_t y = 0; y < image->height; ++y)
|
|
{
|
|
size_t offset = ((convFlags & CONV_FLAGS_INVERTX) ? (image->width - 1) : 0);
|
|
assert(offset * 2 < rowPitch);
|
|
|
|
auto dPtr = reinterpret_cast<uint16_t*>(image->pixels
|
|
+ (image->rowPitch * ((convFlags & CONV_FLAGS_INVERTY) ? y : (image->height - y - 1))))
|
|
+ offset;
|
|
|
|
for (size_t x = 0; x < image->width; ++x)
|
|
{
|
|
if (sPtr + 1 >= endPtr)
|
|
return E_FAIL;
|
|
|
|
auto t = static_cast<uint16_t>(uint32_t(*sPtr) | uint32_t(*(sPtr + 1u) << 8));
|
|
sPtr += 2;
|
|
*dPtr = t;
|
|
|
|
uint32_t alpha = (t & 0x8000) ? 255 : 0;
|
|
minalpha = std::min(minalpha, alpha);
|
|
maxalpha = std::max(maxalpha, alpha);
|
|
|
|
if (convFlags & CONV_FLAGS_INVERTX)
|
|
--dPtr;
|
|
else
|
|
++dPtr;
|
|
}
|
|
}
|
|
|
|
// If there are no non-zero alpha channel entries, we'll assume alpha is not used and force it to opaque
|
|
if (maxalpha == 0)
|
|
{
|
|
opaquealpha = true;
|
|
hr = SetAlphaChannelToOpaque(image);
|
|
if (FAILED(hr))
|
|
return hr;
|
|
}
|
|
else if (minalpha == 255)
|
|
{
|
|
opaquealpha = true;
|
|
}
|
|
}
|
|
break;
|
|
|
|
//----------------------------------------------------------------------- 24/32-bit
|
|
case DXGI_FORMAT_R8G8B8A8_UNORM:
|
|
{
|
|
uint32_t minalpha = 255;
|
|
uint32_t maxalpha = 0;
|
|
|
|
for (size_t y = 0; y < image->height; ++y)
|
|
{
|
|
size_t offset = ((convFlags & CONV_FLAGS_INVERTX) ? (image->width - 1) : 0);
|
|
|
|
auto dPtr = reinterpret_cast<uint32_t*>(image->pixels
|
|
+ (image->rowPitch * ((convFlags & CONV_FLAGS_INVERTY) ? y : (image->height - y - 1))))
|
|
+ offset;
|
|
|
|
for (size_t x = 0; x < image->width; ++x)
|
|
{
|
|
if (convFlags & CONV_FLAGS_EXPAND)
|
|
{
|
|
assert(offset * 3 < rowPitch);
|
|
|
|
if (sPtr + 2 >= endPtr)
|
|
return E_FAIL;
|
|
|
|
// BGR -> RGBA
|
|
*dPtr = uint32_t(*sPtr << 16) | uint32_t(*(sPtr + 1) << 8) | uint32_t(*(sPtr + 2)) | 0xFF000000;
|
|
sPtr += 3;
|
|
|
|
minalpha = maxalpha = 255;
|
|
}
|
|
else
|
|
{
|
|
assert(offset * 4 < rowPitch);
|
|
|
|
if (sPtr + 3 >= endPtr)
|
|
return E_FAIL;
|
|
|
|
// BGRA -> RGBA
|
|
uint32_t alpha = *(sPtr + 3);
|
|
*dPtr = uint32_t(*sPtr << 16) | uint32_t(*(sPtr + 1) << 8) | uint32_t(*(sPtr + 2)) | uint32_t(alpha << 24);
|
|
|
|
minalpha = std::min(minalpha, alpha);
|
|
maxalpha = std::max(maxalpha, alpha);
|
|
|
|
sPtr += 4;
|
|
}
|
|
|
|
if (convFlags & CONV_FLAGS_INVERTX)
|
|
--dPtr;
|
|
else
|
|
++dPtr;
|
|
}
|
|
}
|
|
|
|
// If there are no non-zero alpha channel entries, we'll assume alpha is not used and force it to opaque
|
|
if (maxalpha == 0)
|
|
{
|
|
opaquealpha = true;
|
|
hr = SetAlphaChannelToOpaque(image);
|
|
if (FAILED(hr))
|
|
return hr;
|
|
}
|
|
else if (minalpha == 255)
|
|
{
|
|
opaquealpha = true;
|
|
}
|
|
}
|
|
break;
|
|
|
|
//---------------------------------------------------------------------------------
|
|
default:
|
|
return E_FAIL;
|
|
}
|
|
|
|
return opaquealpha ? S_FALSE : S_OK;
|
|
}
|
|
|
|
|
|
//-------------------------------------------------------------------------------------
|
|
// Encodes TGA file header
|
|
//-------------------------------------------------------------------------------------
|
|
HRESULT EncodeTGAHeader(_In_ const Image& image, _Out_ TGA_HEADER& header, _Inout_ uint32_t& convFlags) noexcept
|
|
{
|
|
memset(&header, 0, sizeof(TGA_HEADER));
|
|
|
|
if ((image.width > UINT16_MAX)
|
|
|| (image.height > UINT16_MAX))
|
|
{
|
|
return HRESULT_FROM_WIN32(ERROR_NOT_SUPPORTED);
|
|
}
|
|
|
|
header.wWidth = static_cast<uint16_t>(image.width);
|
|
header.wHeight = static_cast<uint16_t>(image.height);
|
|
|
|
switch (image.format)
|
|
{
|
|
case DXGI_FORMAT_R8G8B8A8_UNORM:
|
|
case DXGI_FORMAT_R8G8B8A8_UNORM_SRGB:
|
|
header.bImageType = TGA_TRUECOLOR;
|
|
header.bBitsPerPixel = 32;
|
|
header.bDescriptor = TGA_FLAGS_INVERTY | 8;
|
|
convFlags |= CONV_FLAGS_SWIZZLE;
|
|
break;
|
|
|
|
case DXGI_FORMAT_B8G8R8A8_UNORM:
|
|
case DXGI_FORMAT_B8G8R8A8_UNORM_SRGB:
|
|
header.bImageType = TGA_TRUECOLOR;
|
|
header.bBitsPerPixel = 32;
|
|
header.bDescriptor = TGA_FLAGS_INVERTY | 8;
|
|
break;
|
|
|
|
case DXGI_FORMAT_B8G8R8X8_UNORM:
|
|
case DXGI_FORMAT_B8G8R8X8_UNORM_SRGB:
|
|
header.bImageType = TGA_TRUECOLOR;
|
|
header.bBitsPerPixel = 24;
|
|
header.bDescriptor = TGA_FLAGS_INVERTY;
|
|
convFlags |= CONV_FLAGS_888;
|
|
break;
|
|
|
|
case DXGI_FORMAT_R8_UNORM:
|
|
case DXGI_FORMAT_A8_UNORM:
|
|
header.bImageType = TGA_BLACK_AND_WHITE;
|
|
header.bBitsPerPixel = 8;
|
|
header.bDescriptor = TGA_FLAGS_INVERTY;
|
|
break;
|
|
|
|
case DXGI_FORMAT_B5G5R5A1_UNORM:
|
|
header.bImageType = TGA_TRUECOLOR;
|
|
header.bBitsPerPixel = 16;
|
|
header.bDescriptor = TGA_FLAGS_INVERTY | 1;
|
|
break;
|
|
|
|
default:
|
|
return HRESULT_FROM_WIN32(ERROR_NOT_SUPPORTED);
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
//-------------------------------------------------------------------------------------
|
|
// Copies BGRX data to form BGR 24bpp data
|
|
//-------------------------------------------------------------------------------------
|
|
#pragma warning(suppress: 6001 6101) // In the case where outSize is insufficient we do not write to pDestination
|
|
void Copy24bppScanline(
|
|
_Out_writes_bytes_(outSize) void* pDestination,
|
|
_In_ size_t outSize,
|
|
_In_reads_bytes_(inSize) const void* pSource,
|
|
_In_ size_t inSize) noexcept
|
|
{
|
|
assert(pDestination && outSize > 0);
|
|
assert(pSource && inSize > 0);
|
|
|
|
assert(pDestination != pSource);
|
|
|
|
const uint32_t * __restrict sPtr = static_cast<const uint32_t*>(pSource);
|
|
uint8_t * __restrict dPtr = static_cast<uint8_t*>(pDestination);
|
|
|
|
if (inSize >= 4 && outSize >= 3)
|
|
{
|
|
const uint8_t* endPtr = dPtr + outSize;
|
|
|
|
for (size_t count = 0; count < (inSize - 3); count += 4)
|
|
{
|
|
uint32_t t = *(sPtr++);
|
|
|
|
if (dPtr + 3 > endPtr)
|
|
return;
|
|
|
|
*(dPtr++) = uint8_t(t & 0xFF); // Blue
|
|
*(dPtr++) = uint8_t((t & 0xFF00) >> 8); // Green
|
|
*(dPtr++) = uint8_t((t & 0xFF0000) >> 16); // Red
|
|
}
|
|
}
|
|
}
|
|
|
|
//-------------------------------------------------------------------------------------
|
|
// TGA 2.0 Extension helpers
|
|
//-------------------------------------------------------------------------------------
|
|
void SetExtension(TGA_EXTENSION *ext, const TexMetadata& metadata) noexcept
|
|
{
|
|
memset(ext, 0, sizeof(TGA_EXTENSION));
|
|
|
|
ext->wSize = sizeof(TGA_EXTENSION);
|
|
|
|
memcpy(ext->szSoftwareId, "DirectXTex", sizeof("DirectXTex"));
|
|
ext->wVersionNumber = DIRECTX_TEX_VERSION;
|
|
ext->bVersionLetter = ' ';
|
|
|
|
if (IsSRGB(metadata.format))
|
|
{
|
|
ext->wGammaNumerator = 22;
|
|
ext->wGammaDenominator = 10;
|
|
}
|
|
|
|
switch (metadata.GetAlphaMode())
|
|
{
|
|
case TEX_ALPHA_MODE_UNKNOWN:
|
|
ext->bAttributesType = HasAlpha(metadata.format) ? TGA_ATTRIBUTE_UNDEFINED : TGA_ATTRIBUTE_NONE;
|
|
break;
|
|
|
|
case TEX_ALPHA_MODE_STRAIGHT:
|
|
ext->bAttributesType = TGA_ATTRIBUTE_ALPHA;
|
|
break;
|
|
|
|
case TEX_ALPHA_MODE_PREMULTIPLIED:
|
|
ext->bAttributesType = TGA_ATTRIBUTE_PREMULTIPLIED;
|
|
break;
|
|
|
|
case TEX_ALPHA_MODE_OPAQUE:
|
|
ext->bAttributesType = TGA_ATTRIBUTE_IGNORED;
|
|
break;
|
|
|
|
case TEX_ALPHA_MODE_CUSTOM:
|
|
ext->bAttributesType = TGA_ATTRIBUTE_UNDEFINED;
|
|
break;
|
|
}
|
|
|
|
// Set file time stamp
|
|
{
|
|
time_t now = {};
|
|
time(&now);
|
|
|
|
tm info;
|
|
info = *gmtime(&now);
|
|
|
|
ext->wStampMonth = static_cast<uint16_t>(info.tm_mon + 1);
|
|
ext->wStampDay = static_cast<uint16_t>(info.tm_mday);
|
|
ext->wStampYear = static_cast<uint16_t>(info.tm_year + 1900);
|
|
ext->wStampHour = static_cast<uint16_t>(info.tm_hour);
|
|
ext->wStampMinute = static_cast<uint16_t>(info.tm_min);
|
|
ext->wStampSecond = static_cast<uint16_t>(info.tm_sec);
|
|
}
|
|
}
|
|
|
|
TEX_ALPHA_MODE GetAlphaModeFromExtension(const TGA_EXTENSION *ext) noexcept
|
|
{
|
|
if (ext && ext->wSize == sizeof(TGA_EXTENSION))
|
|
{
|
|
switch (ext->bAttributesType)
|
|
{
|
|
case TGA_ATTRIBUTE_IGNORED: return TEX_ALPHA_MODE_OPAQUE;
|
|
case TGA_ATTRIBUTE_UNDEFINED: return TEX_ALPHA_MODE_CUSTOM;
|
|
case TGA_ATTRIBUTE_ALPHA: return TEX_ALPHA_MODE_STRAIGHT;
|
|
case TGA_ATTRIBUTE_PREMULTIPLIED: return TEX_ALPHA_MODE_PREMULTIPLIED;
|
|
}
|
|
}
|
|
|
|
return TEX_ALPHA_MODE_UNKNOWN;
|
|
}
|
|
}
|
|
|
|
|
|
//=====================================================================================
|
|
// Entry-points
|
|
//=====================================================================================
|
|
|
|
//-------------------------------------------------------------------------------------
|
|
// Obtain metadata from TGA file in memory/on disk
|
|
//-------------------------------------------------------------------------------------
|
|
_Use_decl_annotations_
|
|
HRESULT DirectX::GetMetadataFromTGAMemory(
|
|
const void* pSource,
|
|
size_t size,
|
|
TexMetadata& metadata) noexcept
|
|
{
|
|
if (!pSource || size == 0)
|
|
return E_INVALIDARG;
|
|
|
|
size_t offset;
|
|
return DecodeTGAHeader(pSource, size, metadata, offset, nullptr);
|
|
}
|
|
|
|
_Use_decl_annotations_
|
|
HRESULT DirectX::GetMetadataFromTGAFile(const wchar_t* szFile, TexMetadata& metadata) noexcept
|
|
{
|
|
if (!szFile)
|
|
return E_INVALIDARG;
|
|
|
|
ScopedReadFile hFile(szFile);
|
|
|
|
// Need at least enough data to fill the standard header to be a valid TGA
|
|
if (hFile.GetLength() < (sizeof(TGA_HEADER)))
|
|
{
|
|
return E_FAIL;
|
|
}
|
|
|
|
// Read the standard header (we don't need the file footer to parse the file)
|
|
uint8_t header[sizeof(TGA_HEADER)] = {};
|
|
DWORD bytesRead = 0;
|
|
if (!(bytesRead = hFile.Read(header, sizeof(TGA_HEADER))))
|
|
{
|
|
return E_FAIL;
|
|
}
|
|
|
|
size_t offset;
|
|
return DecodeTGAHeader(header, bytesRead, metadata, offset, nullptr);
|
|
}
|
|
|
|
|
|
//-------------------------------------------------------------------------------------
|
|
// Load a TGA file in memory
|
|
//-------------------------------------------------------------------------------------
|
|
_Use_decl_annotations_
|
|
HRESULT DirectX::LoadFromTGAMemory(
|
|
const void* pSource,
|
|
size_t size,
|
|
TexMetadata* metadata,
|
|
ScratchImage& image) noexcept
|
|
{
|
|
if (!pSource || size == 0)
|
|
return E_INVALIDARG;
|
|
|
|
image.Release();
|
|
|
|
size_t offset;
|
|
uint32_t convFlags = 0;
|
|
TexMetadata mdata;
|
|
HRESULT hr = DecodeTGAHeader(pSource, size, mdata, offset, &convFlags);
|
|
if (FAILED(hr))
|
|
return hr;
|
|
|
|
if (offset > size)
|
|
return E_FAIL;
|
|
|
|
const void* pPixels = static_cast<const uint8_t*>(pSource) + offset;
|
|
|
|
size_t remaining = size - offset;
|
|
if (remaining == 0)
|
|
return E_FAIL;
|
|
|
|
hr = image.Initialize2D(mdata.format, mdata.width, mdata.height, 1, 1);
|
|
if (FAILED(hr))
|
|
return hr;
|
|
|
|
if (convFlags & CONV_FLAGS_RLE)
|
|
{
|
|
hr = UncompressPixels(pPixels, remaining, image.GetImage(0, 0, 0), convFlags);
|
|
}
|
|
else
|
|
{
|
|
hr = CopyPixels(pPixels, remaining, image.GetImage(0, 0, 0), convFlags);
|
|
}
|
|
|
|
if (FAILED(hr))
|
|
{
|
|
image.Release();
|
|
return hr;
|
|
}
|
|
|
|
if (metadata)
|
|
{
|
|
memcpy(metadata, &mdata, sizeof(TexMetadata));
|
|
if (hr == S_FALSE)
|
|
{
|
|
metadata->SetAlphaMode(TEX_ALPHA_MODE_OPAQUE);
|
|
}
|
|
else if (size >= sizeof(TGA_FOOTER))
|
|
{
|
|
// Handle optional TGA 2.0 footer
|
|
auto footer = reinterpret_cast<const TGA_FOOTER*>(static_cast<const uint8_t*>(pSource) + size - sizeof(TGA_FOOTER));
|
|
|
|
if (memcmp(footer->Signature, g_Signature, sizeof(g_Signature)) == 0)
|
|
{
|
|
if (footer->dwExtensionOffset != 0
|
|
&& ((footer->dwExtensionOffset + sizeof(TGA_EXTENSION)) <= size))
|
|
{
|
|
auto ext = reinterpret_cast<const TGA_EXTENSION*>(static_cast<const uint8_t*>(pSource) + footer->dwExtensionOffset);
|
|
metadata->SetAlphaMode(GetAlphaModeFromExtension(ext));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
//-------------------------------------------------------------------------------------
|
|
// Load a TGA file from disk
|
|
//-------------------------------------------------------------------------------------
|
|
_Use_decl_annotations_
|
|
HRESULT DirectX::LoadFromTGAFile(
|
|
const wchar_t* szFile,
|
|
TexMetadata* metadata,
|
|
ScratchImage& image) noexcept
|
|
{
|
|
if (!szFile)
|
|
return E_INVALIDARG;
|
|
|
|
image.Release();
|
|
|
|
ScopedReadFile hFile(szFile);
|
|
|
|
auto length = hFile.GetLength();
|
|
|
|
// Need at least enough data to fill the standard header to be a valid TGA
|
|
if (length < (sizeof(TGA_HEADER)))
|
|
{
|
|
return E_FAIL;
|
|
}
|
|
|
|
// Read the header
|
|
uint8_t header[sizeof(TGA_HEADER)] = {};
|
|
DWORD bytesRead = 0;
|
|
if (!(bytesRead = hFile.Read(header, sizeof(TGA_HEADER))))
|
|
{
|
|
return E_FAIL;
|
|
}
|
|
|
|
size_t offset;
|
|
uint32_t convFlags = 0;
|
|
TexMetadata mdata;
|
|
HRESULT hr = DecodeTGAHeader(header, bytesRead, mdata, offset, &convFlags);
|
|
if (FAILED(hr))
|
|
return hr;
|
|
|
|
|
|
// Read the pixels
|
|
auto remaining = static_cast<DWORD>(length - offset);
|
|
if (remaining == 0)
|
|
return E_FAIL;
|
|
|
|
if (offset > sizeof(TGA_HEADER))
|
|
{
|
|
hFile.SeekBegin(offset);
|
|
}
|
|
|
|
hr = image.Initialize2D(mdata.format, mdata.width, mdata.height, 1, 1);
|
|
if (FAILED(hr))
|
|
return hr;
|
|
|
|
assert(image.GetPixels());
|
|
|
|
bool opaquealpha = false;
|
|
|
|
if (!(convFlags & (CONV_FLAGS_RLE | CONV_FLAGS_EXPAND | CONV_FLAGS_INVERTX)) && (convFlags & CONV_FLAGS_INVERTY))
|
|
{
|
|
// This case we can read directly into the image buffer in place
|
|
if (remaining < image.GetPixelsSize())
|
|
{
|
|
image.Release();
|
|
return HRESULT_FROM_WIN32(ERROR_HANDLE_EOF);
|
|
}
|
|
|
|
if (image.GetPixelsSize() > UINT32_MAX)
|
|
{
|
|
image.Release();
|
|
return HRESULT_FROM_WIN32(ERROR_ARITHMETIC_OVERFLOW);
|
|
}
|
|
|
|
if (image.GetPixelsSize() != hFile.Read(image.GetPixels(), static_cast<DWORD>(image.GetPixelsSize())))
|
|
{
|
|
image.Release();
|
|
return E_FAIL;
|
|
}
|
|
|
|
switch (mdata.format)
|
|
{
|
|
case DXGI_FORMAT_R8G8B8A8_UNORM:
|
|
{
|
|
// TGA stores 32-bit data in BGRA form, need to swizzle to RGBA
|
|
assert(image.GetImageCount() == 1);
|
|
const Image* img = image.GetImage(0, 0, 0);
|
|
if (!img)
|
|
{
|
|
image.Release();
|
|
return E_POINTER;
|
|
}
|
|
|
|
uint8_t *pPixels = img->pixels;
|
|
if (!pPixels)
|
|
{
|
|
image.Release();
|
|
return E_POINTER;
|
|
}
|
|
|
|
size_t rowPitch = img->rowPitch;
|
|
|
|
// Scan for non-zero alpha channel
|
|
uint32_t minalpha = 255;
|
|
uint32_t maxalpha = 0;
|
|
|
|
for (size_t h = 0; h < img->height; ++h)
|
|
{
|
|
auto sPtr = reinterpret_cast<const uint32_t*>(pPixels);
|
|
|
|
for (size_t x = 0; x < img->width; ++x)
|
|
{
|
|
uint32_t alpha = ((*sPtr & 0xFF000000) >> 24);
|
|
|
|
minalpha = std::min(minalpha, alpha);
|
|
maxalpha = std::max(maxalpha, alpha);
|
|
|
|
++sPtr;
|
|
}
|
|
|
|
pPixels += rowPitch;
|
|
}
|
|
|
|
uint32_t tflags = TEXP_SCANLINE_NONE;
|
|
if (maxalpha == 0)
|
|
{
|
|
opaquealpha = true;
|
|
tflags = TEXP_SCANLINE_SETALPHA;
|
|
}
|
|
else if (minalpha == 255)
|
|
{
|
|
opaquealpha = true;
|
|
}
|
|
|
|
// Swizzle scanlines
|
|
pPixels = img->pixels;
|
|
|
|
for (size_t h = 0; h < img->height; ++h)
|
|
{
|
|
_SwizzleScanline(pPixels, rowPitch, pPixels, rowPitch, mdata.format, tflags);
|
|
pPixels += rowPitch;
|
|
}
|
|
}
|
|
break;
|
|
|
|
// If we start using DXGI_FORMAT_B8G8R8X8_UNORM or DXGI_FORMAT_B8G8R8A8_UNORM we need to check for a fully 0 alpha channel
|
|
|
|
case DXGI_FORMAT_B5G5R5A1_UNORM:
|
|
{
|
|
assert(image.GetImageCount() == 1);
|
|
const Image* img = image.GetImage(0, 0, 0);
|
|
if (!img)
|
|
{
|
|
image.Release();
|
|
return E_POINTER;
|
|
}
|
|
|
|
// Scan for non-zero alpha channel
|
|
uint32_t minalpha = 255;
|
|
uint32_t maxalpha = 0;
|
|
|
|
const uint8_t *pPixels = img->pixels;
|
|
if (!pPixels)
|
|
{
|
|
image.Release();
|
|
return E_POINTER;
|
|
}
|
|
|
|
size_t rowPitch = img->rowPitch;
|
|
|
|
for (size_t h = 0; h < img->height; ++h)
|
|
{
|
|
auto sPtr = reinterpret_cast<const uint16_t*>(pPixels);
|
|
|
|
for (size_t x = 0; x < img->width; ++x)
|
|
{
|
|
uint32_t alpha = (*sPtr & 0x8000) ? 255 : 0;
|
|
|
|
minalpha = std::min(minalpha, alpha);
|
|
maxalpha = std::max(maxalpha, alpha);
|
|
|
|
++sPtr;
|
|
}
|
|
|
|
pPixels += rowPitch;
|
|
}
|
|
|
|
// If there are no non-zero alpha channel entries, we'll assume alpha is not used and force it to opaque
|
|
if (maxalpha == 0)
|
|
{
|
|
opaquealpha = true;
|
|
hr = SetAlphaChannelToOpaque(img);
|
|
if (FAILED(hr))
|
|
{
|
|
image.Release();
|
|
return hr;
|
|
}
|
|
}
|
|
else if (minalpha == 255)
|
|
{
|
|
opaquealpha = true;
|
|
}
|
|
}
|
|
break;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
else // RLE || EXPAND || INVERTX || !INVERTY
|
|
{
|
|
std::unique_ptr<uint8_t[]> temp(new (std::nothrow) uint8_t[remaining]);
|
|
if (!temp)
|
|
{
|
|
image.Release();
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
|
|
if (!(bytesRead = hFile.Read(temp.get(), remaining)))
|
|
{
|
|
image.Release();
|
|
return E_FAIL;
|
|
}
|
|
|
|
if (bytesRead != remaining)
|
|
{
|
|
image.Release();
|
|
return E_FAIL;
|
|
}
|
|
|
|
if (convFlags & CONV_FLAGS_RLE)
|
|
{
|
|
hr = UncompressPixels(temp.get(), remaining, image.GetImage(0, 0, 0), convFlags);
|
|
}
|
|
else
|
|
{
|
|
hr = CopyPixels(temp.get(), remaining, image.GetImage(0, 0, 0), convFlags);
|
|
}
|
|
|
|
if (FAILED(hr))
|
|
{
|
|
image.Release();
|
|
return hr;
|
|
}
|
|
|
|
if (hr == S_FALSE)
|
|
opaquealpha = true;
|
|
}
|
|
|
|
if (metadata)
|
|
{
|
|
memcpy(metadata, &mdata, sizeof(TexMetadata));
|
|
if (opaquealpha)
|
|
{
|
|
metadata->SetAlphaMode(TEX_ALPHA_MODE_OPAQUE);
|
|
}
|
|
else
|
|
{
|
|
// Handle optional TGA 2.0 footer
|
|
TGA_FOOTER footer = {};
|
|
|
|
hFile.SeekEnd(-static_cast<int>(sizeof(TGA_FOOTER)));
|
|
|
|
if (sizeof(TGA_FOOTER) != hFile.Read(&footer, sizeof(TGA_FOOTER)))
|
|
{
|
|
image.Release();
|
|
return E_FAIL;
|
|
}
|
|
|
|
|
|
if (memcmp(footer.Signature, g_Signature, sizeof(g_Signature)) == 0)
|
|
{
|
|
if (footer.dwExtensionOffset != 0
|
|
&& ((footer.dwExtensionOffset + sizeof(TGA_EXTENSION)) <= length))
|
|
{
|
|
hFile.SeekBegin(footer.dwExtensionOffset);
|
|
TGA_EXTENSION ext = {};
|
|
|
|
if ((sizeof(TGA_EXTENSION) == hFile.Read(&ext, sizeof(TGA_EXTENSION))))
|
|
{
|
|
metadata->SetAlphaMode(GetAlphaModeFromExtension(&ext));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
//-------------------------------------------------------------------------------------
|
|
// Save a TGA file to memory
|
|
//-------------------------------------------------------------------------------------
|
|
_Use_decl_annotations_
|
|
HRESULT DirectX::SaveToTGAMemory(const Image& image, Blob& blob, const TexMetadata* metadata) noexcept
|
|
{
|
|
if (!image.pixels)
|
|
return E_POINTER;
|
|
|
|
TGA_HEADER tga_header = {};
|
|
uint32_t convFlags = 0;
|
|
HRESULT hr = EncodeTGAHeader(image, tga_header, convFlags);
|
|
if (FAILED(hr))
|
|
return hr;
|
|
|
|
blob.Release();
|
|
|
|
// Determine memory required for image data
|
|
size_t rowPitch, slicePitch;
|
|
hr = ComputePitch(image.format, image.width, image.height, rowPitch, slicePitch,
|
|
(convFlags & CONV_FLAGS_888) ? CP_FLAGS_24BPP : CP_FLAGS_NONE);
|
|
if (FAILED(hr))
|
|
return hr;
|
|
|
|
hr = blob.Initialize(sizeof(TGA_HEADER)
|
|
+ slicePitch
|
|
+ (metadata ? sizeof(TGA_EXTENSION) : 0)
|
|
+ sizeof(TGA_FOOTER));
|
|
if (FAILED(hr))
|
|
return hr;
|
|
|
|
// Copy header
|
|
auto destPtr = static_cast<uint8_t*>(blob.GetBufferPointer());
|
|
assert(destPtr != nullptr);
|
|
|
|
uint8_t* dPtr = destPtr;
|
|
memcpy_s(dPtr, blob.GetBufferSize(), &tga_header, sizeof(TGA_HEADER));
|
|
dPtr += sizeof(TGA_HEADER);
|
|
|
|
const uint8_t* pPixels = image.pixels;
|
|
assert(pPixels);
|
|
|
|
for (size_t y = 0; y < image.height; ++y)
|
|
{
|
|
// Copy pixels
|
|
if (convFlags & CONV_FLAGS_888)
|
|
{
|
|
Copy24bppScanline(dPtr, rowPitch, pPixels, image.rowPitch);
|
|
}
|
|
else if (convFlags & CONV_FLAGS_SWIZZLE)
|
|
{
|
|
_SwizzleScanline(dPtr, rowPitch, pPixels, image.rowPitch, image.format, TEXP_SCANLINE_NONE);
|
|
}
|
|
else
|
|
{
|
|
_CopyScanline(dPtr, rowPitch, pPixels, image.rowPitch, image.format, TEXP_SCANLINE_NONE);
|
|
}
|
|
|
|
dPtr += rowPitch;
|
|
pPixels += image.rowPitch;
|
|
}
|
|
|
|
uint32_t extOffset = 0;
|
|
if (metadata)
|
|
{
|
|
// metadata is only used for writing the TGA 2.0 extension header
|
|
|
|
auto ext = reinterpret_cast<TGA_EXTENSION*>(dPtr);
|
|
SetExtension(ext, *metadata);
|
|
|
|
extOffset = static_cast<uint32_t>(dPtr - destPtr);
|
|
dPtr += sizeof(TGA_EXTENSION);
|
|
}
|
|
|
|
// Copy TGA 2.0 footer
|
|
auto footer = reinterpret_cast<TGA_FOOTER*>(dPtr);
|
|
footer->dwDeveloperOffset = 0;
|
|
footer->dwExtensionOffset = extOffset;
|
|
memcpy(footer->Signature, g_Signature, sizeof(g_Signature));
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
//-------------------------------------------------------------------------------------
|
|
// Save a TGA file to disk
|
|
//-------------------------------------------------------------------------------------
|
|
_Use_decl_annotations_
|
|
HRESULT DirectX::SaveToTGAFile(const Image& image, const wchar_t* szFile, const TexMetadata* metadata) noexcept
|
|
{
|
|
if (!szFile)
|
|
return E_INVALIDARG;
|
|
|
|
if (!image.pixels)
|
|
return E_POINTER;
|
|
|
|
TGA_HEADER tga_header = {};
|
|
uint32_t convFlags = 0;
|
|
HRESULT hr = EncodeTGAHeader(image, tga_header, convFlags);
|
|
if (FAILED(hr))
|
|
return hr;
|
|
|
|
ScopedWriteFile hFile(szFile);
|
|
|
|
// Determine size for TGA pixel data
|
|
size_t rowPitch, slicePitch;
|
|
hr = ComputePitch(image.format, image.width, image.height, rowPitch, slicePitch,
|
|
(convFlags & CONV_FLAGS_888) ? CP_FLAGS_24BPP : CP_FLAGS_NONE);
|
|
if (FAILED(hr))
|
|
return hr;
|
|
|
|
if (slicePitch < 65535)
|
|
{
|
|
// For small images, it is better to create an in-memory file and write it out
|
|
Blob blob;
|
|
|
|
hr = SaveToTGAMemory(image, blob);
|
|
if (FAILED(hr))
|
|
return hr;
|
|
|
|
// Write blob
|
|
const DWORD bytesToWrite = static_cast<DWORD>(blob.GetBufferSize());
|
|
if (! hFile.Write(blob.GetBufferPointer(), bytesToWrite))
|
|
{
|
|
return E_FAIL;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// Otherwise, write the image one scanline at a time...
|
|
std::unique_ptr<uint8_t[]> temp(new (std::nothrow) uint8_t[rowPitch]);
|
|
if (!temp)
|
|
return E_OUTOFMEMORY;
|
|
|
|
// Write header
|
|
if (!hFile.Write(&tga_header, sizeof(TGA_HEADER)))
|
|
{
|
|
return E_FAIL;
|
|
}
|
|
|
|
if (rowPitch > UINT32_MAX)
|
|
return HRESULT_FROM_WIN32(ERROR_ARITHMETIC_OVERFLOW);
|
|
|
|
// Write pixels
|
|
const uint8_t* pPixels = image.pixels;
|
|
|
|
for (size_t y = 0; y < image.height; ++y)
|
|
{
|
|
// Copy pixels
|
|
if (convFlags & CONV_FLAGS_888)
|
|
{
|
|
Copy24bppScanline(temp.get(), rowPitch, pPixels, image.rowPitch);
|
|
}
|
|
else if (convFlags & CONV_FLAGS_SWIZZLE)
|
|
{
|
|
_SwizzleScanline(temp.get(), rowPitch, pPixels, image.rowPitch, image.format, TEXP_SCANLINE_NONE);
|
|
}
|
|
else
|
|
{
|
|
_CopyScanline(temp.get(), rowPitch, pPixels, image.rowPitch, image.format, TEXP_SCANLINE_NONE);
|
|
}
|
|
|
|
pPixels += image.rowPitch;
|
|
|
|
if (!hFile.Write(temp.get(), rowPitch))
|
|
{
|
|
return E_FAIL;
|
|
}
|
|
}
|
|
|
|
uint32_t extOffset = 0;
|
|
if (metadata)
|
|
{
|
|
// metadata is only used for writing the TGA 2.0 extension header
|
|
TGA_EXTENSION ext = {};
|
|
SetExtension(&ext, *metadata);
|
|
|
|
extOffset = hFile.GetOffset();
|
|
|
|
if (!hFile.Write(&ext, sizeof(TGA_EXTENSION)))
|
|
{
|
|
return E_FAIL;
|
|
}
|
|
}
|
|
|
|
// Write TGA 2.0 footer
|
|
TGA_FOOTER footer = {};
|
|
footer.dwExtensionOffset = extOffset;
|
|
memcpy(footer.Signature, g_Signature, sizeof(g_Signature));
|
|
|
|
|
|
if (!hFile.Write(&footer, sizeof(footer)))
|
|
{
|
|
return E_FAIL;
|
|
}
|
|
}
|
|
|
|
return S_OK;
|
|
}
|