//------------------------------------------------------------------------------------- // 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 { 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 }; #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; }; struct TGA_FOOTER { uint16_t dwExtensionOffset; uint16_t dwDeveloperOffset; char Signature[18]; }; 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; }; #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) { 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; 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) { 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) { 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; ComputePitch(image->format, image->width, image->height, rowPitch, slicePitch, CP_FLAGS_NONE); } auto sPtr = static_cast(pSource); const uint8_t* endPtr = sPtr + size; 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: { bool nonzeroa = false; 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(unsigned(*sPtr) | (*(sPtr + 1u) << 8)); if (t & 0x8000) nonzeroa = true; 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(unsigned(*sPtr) | (*(sPtr + 1u) << 8)); if (t & 0x8000) nonzeroa = true; 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 (!nonzeroa) { HRESULT hr = SetAlphaChannelToOpaque(image); if (FAILED(hr)) return hr; } } break; //----------------------------------------------------------------------- 24/32-bit case DXGI_FORMAT_R8G8B8A8_UNORM: { bool nonzeroa = false; 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 = (*sPtr << 16) | (*(sPtr + 1) << 8) | (*(sPtr + 2)) | 0xFF000000; sPtr += 3; nonzeroa = true; } else { assert(offset * 4 < rowPitch); if (sPtr + 3 >= endPtr) return E_FAIL; // BGRA -> RGBA t = (*sPtr << 16) | (*(sPtr + 1) << 8) | (*(sPtr + 2)) | (*(sPtr + 3) << 24); if (*(sPtr + 3) > 0) nonzeroa = true; 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 = (*sPtr << 16) | (*(sPtr + 1) << 8) | (*(sPtr + 2)) | 0xFF000000; sPtr += 3; nonzeroa = true; } else { assert(offset * 4 < rowPitch); if (sPtr + 3 >= endPtr) return E_FAIL; // BGRA -> RGBA *dPtr = (*sPtr << 16) | (*(sPtr + 1) << 8) | (*(sPtr + 2)) | (*(sPtr + 3) << 24); if (*(sPtr + 3) > 0) nonzeroa = true; 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 (!nonzeroa) { HRESULT hr = SetAlphaChannelToOpaque(image); if (FAILED(hr)) return hr; } } break; //--------------------------------------------------------------------------------- default: return E_FAIL; } return 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) { 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; ComputePitch(image->format, image->width, image->height, rowPitch, slicePitch, CP_FLAGS_NONE); } auto sPtr = static_cast(pSource); const uint8_t* endPtr = sPtr + size; 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: { bool nonzeroa = false; 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(unsigned(*sPtr) | (*(sPtr + 1u) << 8)); sPtr += 2; *dPtr = t; if (t & 0x8000) nonzeroa = true; 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 (!nonzeroa) { HRESULT hr = SetAlphaChannelToOpaque(image); if (FAILED(hr)) return hr; } } break; //----------------------------------------------------------------------- 24/32-bit case DXGI_FORMAT_R8G8B8A8_UNORM: { bool nonzeroa = false; 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 = (*sPtr << 16) | (*(sPtr + 1) << 8) | (*(sPtr + 2)) | 0xFF000000; sPtr += 3; nonzeroa = true; } else { assert(offset * 4 < rowPitch); if (sPtr + 3 >= endPtr) return E_FAIL; // BGRA -> RGBA *dPtr = (*sPtr << 16) | (*(sPtr + 1) << 8) | (*(sPtr + 2)) | (*(sPtr + 3) << 24); if (*(sPtr + 3) > 0) nonzeroa = true; 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 (!nonzeroa) { HRESULT hr = SetAlphaChannelToOpaque(image); if (FAILED(hr)) return hr; } } break; //--------------------------------------------------------------------------------- default: return E_FAIL; } return S_OK; } //------------------------------------------------------------------------------------- // Encodes TGA file header //------------------------------------------------------------------------------------- HRESULT EncodeTGAHeader(_In_ const Image& image, _Out_ TGA_HEADER& header, _Inout_ DWORD& convFlags) { memset(&header, 0, sizeof(TGA_HEADER)); if ((image.width > 0xFFFF) || (image.height > 0xFFFF)) { 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) { 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 } } } } //===================================================================================== // 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) { 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) { 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) { 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)); return S_OK; } //------------------------------------------------------------------------------------- // Load a TGA file from disk //------------------------------------------------------------------------------------- _Use_decl_annotations_ HRESULT DirectX::LoadFromTGAFile( const wchar_t* szFile, TexMetadata* metadata, ScratchImage& image) { 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()); 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 (!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 bool nonzeroa = false; for (size_t h = 0; h < img->height; ++h) { auto sPtr = reinterpret_cast(pPixels); for (size_t x = 0; x < img->width; ++x) { if ((*sPtr) & 0xff000000) { nonzeroa = true; break; } ++sPtr; } if (nonzeroa) break; pPixels += rowPitch; } DWORD tflags = (!nonzeroa) ? TEXP_SCANLINE_SETALPHA : TEXP_SCANLINE_NONE; // 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 bool nonzeroa = false; 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) { if (*sPtr & 0x8000) { nonzeroa = true; break; } ++sPtr; } if (nonzeroa) break; pPixels += rowPitch; } // If there are no non-zero alpha channel entries, we'll assume alpha is not used and force it to opaque if (!nonzeroa) { hr = SetAlphaChannelToOpaque(img); if (FAILED(hr)) { image.Release(); return hr; } } } 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 (metadata) memcpy(metadata, &mdata, sizeof(TexMetadata)); return S_OK; } //------------------------------------------------------------------------------------- // Save a TGA file to memory //------------------------------------------------------------------------------------- _Use_decl_annotations_ HRESULT DirectX::SaveToTGAMemory(const Image& image, Blob& blob) { 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 { ComputePitch(image.format, image.width, image.height, rowPitch, slicePitch, CP_FLAGS_NONE); } hr = blob.Initialize(sizeof(TGA_HEADER) + slicePitch); if (FAILED(hr)) return hr; // Copy header auto dPtr = static_cast(blob.GetBufferPointer()); assert(dPtr != nullptr); 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; } return S_OK; } //------------------------------------------------------------------------------------- // Save a TGA file to disk //------------------------------------------------------------------------------------- _Use_decl_annotations_ HRESULT DirectX::SaveToTGAFile(const Image& image, const wchar_t* szFile) { 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) { rowPitch = image.width * 3; slicePitch = image.height * rowPitch; } else { ComputePitch(image.format, image.width, image.height, rowPitch, slicePitch, CP_FLAGS_NONE); } 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; // 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; } } delonfail.clear(); return S_OK; }