//------------------------------------------------------------------------------------- // 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_ DWORD* 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(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_ DWORD convFlags) noexcept { assert(pSource && size > 0); if (!image || !image->pixels) return E_POINTER; // Compute TGA image data pitch size_t rowPitch; if (convFlags & CONV_FLAGS_EXPAND) { rowPitch = image->width * 3; } else { size_t slicePitch; HRESULT hr = ComputePitch(image->format, image->width, image->height, rowPitch, slicePitch, CP_FLAGS_NONE); if (FAILED(hr)) return hr; } auto sPtr = static_cast(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(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(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(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; HRESULT 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(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; DWORD 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; HRESULT 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_ DWORD convFlags) noexcept { assert(pSource && size > 0); if (!image || !image->pixels) return E_POINTER; // Compute TGA image data pitch size_t rowPitch; if (convFlags & CONV_FLAGS_EXPAND) { rowPitch = image->width * 3; } else { size_t slicePitch; HRESULT hr = ComputePitch(image->format, image->width, image->height, rowPitch, slicePitch, CP_FLAGS_NONE); if (FAILED(hr)) return hr; } auto sPtr = static_cast(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(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(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; HRESULT 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(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; HRESULT 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_ DWORD& 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(image.width); header.wHeight = static_cast(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(pSource); uint8_t * __restrict dPtr = static_cast(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; if (!gmtime_s(&info, &now)) { ext->wStampMonth = static_cast(info.tm_mon + 1); ext->wStampDay = static_cast(info.tm_mday); ext->wStampYear = static_cast(info.tm_year + 1900); ext->wStampHour = static_cast(info.tm_hour); ext->wStampMinute = static_cast(info.tm_min); ext->wStampSecond = static_cast(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; #if (_WIN32_WINNT >= _WIN32_WINNT_WIN8) ScopedHandle hFile(safe_handle(CreateFile2(szFile, GENERIC_READ, FILE_SHARE_READ, OPEN_EXISTING, nullptr))); #else ScopedHandle hFile(safe_handle(CreateFileW(szFile, GENERIC_READ, FILE_SHARE_READ, nullptr, OPEN_EXISTING, FILE_FLAG_SEQUENTIAL_SCAN, nullptr))); #endif if (!hFile) { return HRESULT_FROM_WIN32(GetLastError()); } // Get the file size FILE_STANDARD_INFO fileInfo; if (!GetFileInformationByHandleEx(hFile.get(), FileStandardInfo, &fileInfo, sizeof(fileInfo))) { return HRESULT_FROM_WIN32(GetLastError()); } // File is too big for 32-bit allocation, so reject read (4 GB should be plenty large enough for a valid TGA file) if (fileInfo.EndOfFile.HighPart > 0) { return HRESULT_FROM_WIN32(ERROR_FILE_TOO_LARGE); } // Need at least enough data to fill the standard header to be a valid TGA if (fileInfo.EndOfFile.LowPart < (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 (!ReadFile(hFile.get(), header, sizeof(TGA_HEADER), &bytesRead, nullptr)) { return HRESULT_FROM_WIN32(GetLastError()); } 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; DWORD 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(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(static_cast(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(static_cast(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(); #if (_WIN32_WINNT >= _WIN32_WINNT_WIN8) ScopedHandle hFile(safe_handle(CreateFile2(szFile, GENERIC_READ, FILE_SHARE_READ, OPEN_EXISTING, nullptr))); #else ScopedHandle hFile(safe_handle(CreateFileW(szFile, GENERIC_READ, FILE_SHARE_READ, nullptr, OPEN_EXISTING, FILE_FLAG_SEQUENTIAL_SCAN, nullptr))); #endif if (!hFile) { return HRESULT_FROM_WIN32(GetLastError()); } // Get the file size FILE_STANDARD_INFO fileInfo; if (!GetFileInformationByHandleEx(hFile.get(), FileStandardInfo, &fileInfo, sizeof(fileInfo))) { return HRESULT_FROM_WIN32(GetLastError()); } // File is too big for 32-bit allocation, so reject read (4 GB should be plenty large enough for a valid TGA file) if (fileInfo.EndOfFile.HighPart > 0) { return HRESULT_FROM_WIN32(ERROR_FILE_TOO_LARGE); } // Need at least enough data to fill the header to be a valid TGA if (fileInfo.EndOfFile.LowPart < sizeof(TGA_HEADER)) { return E_FAIL; } // Read the header uint8_t header[sizeof(TGA_HEADER)] = {}; DWORD bytesRead = 0; if (!ReadFile(hFile.get(), header, sizeof(TGA_HEADER), &bytesRead, nullptr)) { return HRESULT_FROM_WIN32(GetLastError()); } size_t offset; DWORD convFlags = 0; TexMetadata mdata; HRESULT hr = DecodeTGAHeader(header, bytesRead, mdata, offset, &convFlags); if (FAILED(hr)) return hr; // Read the pixels auto remaining = static_cast(fileInfo.EndOfFile.LowPart - offset); if (remaining == 0) return E_FAIL; if (offset > sizeof(TGA_HEADER)) { // Skip past the id string LARGE_INTEGER filePos = { { static_cast(offset), 0 } }; if (!SetFilePointerEx(hFile.get(), filePos, nullptr, FILE_BEGIN)) { return HRESULT_FROM_WIN32(GetLastError()); } } 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 (!ReadFile(hFile.get(), image.GetPixels(), static_cast(image.GetPixelsSize()), &bytesRead, nullptr)) { image.Release(); return HRESULT_FROM_WIN32(GetLastError()); } if (bytesRead != 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(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; } DWORD 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(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 temp(new (std::nothrow) uint8_t[remaining]); if (!temp) { image.Release(); return E_OUTOFMEMORY; } if (!ReadFile(hFile.get(), temp.get(), remaining, &bytesRead, nullptr)) { image.Release(); return HRESULT_FROM_WIN32(GetLastError()); } 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 = {}; if (SetFilePointer(hFile.get(), -static_cast(sizeof(TGA_FOOTER)), nullptr, FILE_END) == INVALID_SET_FILE_POINTER) { return HRESULT_FROM_WIN32(GetLastError()); } if (!ReadFile(hFile.get(), &footer, sizeof(TGA_FOOTER), &bytesRead, nullptr)) { image.Release(); return HRESULT_FROM_WIN32(GetLastError()); } if (bytesRead != 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)) <= fileInfo.EndOfFile.LowPart)) { LARGE_INTEGER filePos = { { static_cast(footer.dwExtensionOffset), 0 } }; if (SetFilePointerEx(hFile.get(), filePos, nullptr, FILE_BEGIN)) { TGA_EXTENSION ext = {}; if (ReadFile(hFile.get(), &ext, sizeof(TGA_EXTENSION), &bytesRead, nullptr) && bytesRead == 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 = {}; DWORD 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; if (convFlags & CONV_FLAGS_888) { rowPitch = image.width * 3; slicePitch = image.height * rowPitch; } else { hr = ComputePitch(image.format, image.width, image.height, rowPitch, slicePitch, 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(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(dPtr); SetExtension(ext, *metadata); extOffset = static_cast(dPtr - destPtr); dPtr += sizeof(TGA_EXTENSION); } // Copy TGA 2.0 footer auto footer = reinterpret_cast(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 = {}; DWORD convFlags = 0; HRESULT hr = EncodeTGAHeader(image, tga_header, convFlags); if (FAILED(hr)) return hr; // Create file and write header #if (_WIN32_WINNT >= _WIN32_WINNT_WIN8) ScopedHandle hFile(safe_handle(CreateFile2(szFile, GENERIC_WRITE, 0, CREATE_ALWAYS, nullptr))); #else ScopedHandle hFile(safe_handle(CreateFileW(szFile, GENERIC_WRITE, 0, nullptr, CREATE_ALWAYS, 0, nullptr))); #endif if (!hFile) { return HRESULT_FROM_WIN32(GetLastError()); } auto_delete_file delonfail(hFile.get()); // Determine size for TGA pixel data size_t rowPitch, slicePitch; if (convFlags & CONV_FLAGS_888) { uint64_t pitch = uint64_t(image.width) * 3u; uint64_t slice = uint64_t(image.height) * pitch; #if defined(_M_IX86) || defined(_M_ARM) || defined(_M_HYBRID_X86_ARM64) static_assert(sizeof(size_t) == 4, "Not a 32-bit platform!"); if (pitch > UINT32_MAX || slice > UINT32_MAX) return HRESULT_FROM_WIN32(ERROR_ARITHMETIC_OVERFLOW); #else static_assert(sizeof(size_t) == 8, "Not a 64-bit platform!"); #endif rowPitch = static_cast(pitch); slicePitch = static_cast(slice); } else { hr = ComputePitch(image.format, image.width, image.height, rowPitch, slicePitch, 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(blob.GetBufferSize()); DWORD bytesWritten; if (!WriteFile(hFile.get(), blob.GetBufferPointer(), bytesToWrite, &bytesWritten, nullptr)) { return HRESULT_FROM_WIN32(GetLastError()); } if (bytesWritten != bytesToWrite) { return E_FAIL; } } else { // Otherwise, write the image one scanline at a time... std::unique_ptr temp(new (std::nothrow) uint8_t[rowPitch]); if (!temp) return E_OUTOFMEMORY; // Write header DWORD bytesWritten; if (!WriteFile(hFile.get(), &tga_header, sizeof(TGA_HEADER), &bytesWritten, nullptr)) { return HRESULT_FROM_WIN32(GetLastError()); } if (bytesWritten != 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 (!WriteFile(hFile.get(), temp.get(), static_cast(rowPitch), &bytesWritten, nullptr)) { return HRESULT_FROM_WIN32(GetLastError()); } if (bytesWritten != 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 = SetFilePointer(hFile.get(), 0, nullptr, FILE_CURRENT); if (extOffset == INVALID_SET_FILE_POINTER) { return HRESULT_FROM_WIN32(GetLastError()); } if (!WriteFile(hFile.get(), &ext, sizeof(TGA_EXTENSION), &bytesWritten, nullptr)) { return HRESULT_FROM_WIN32(GetLastError()); } if (bytesWritten != 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 (!WriteFile(hFile.get(), &footer, sizeof(footer), &bytesWritten, nullptr)) { return HRESULT_FROM_WIN32(GetLastError()); } if (bytesWritten != sizeof(footer)) return E_FAIL; } delonfail.clear(); return S_OK; }