//-------------------------------------------------------------------------------------- // File: DirectXTexEXR.cpp // // DirectXTex Auxilary functions for using the OpenEXR library // // Copyright (c) Microsoft Corporation. // Licensed under the MIT License. //-------------------------------------------------------------------------------------- #include "DirectXTexP.h" #include "DirectXTexEXR.h" #include #include #include #include #include #include #include #include // // Requires the OpenEXR library and its dependencies. // #ifdef __clang__ #pragma clang diagnostic ignored "-Wswitch-enum" #pragma clang diagnostic push #pragma clang diagnostic ignored "-Wdeprecated-declarations" #pragma clang diagnostic ignored "-Wdeprecated-dynamic-exception-spec" #pragma clang diagnostic ignored "-Wfloat-equal" #pragma clang diagnostic ignored "-Wimplicit-int-conversion" #pragma clang diagnostic ignored "-Wlanguage-extension-token" #pragma clang diagnostic ignored "-Wold-style-cast" #pragma clang diagnostic ignored "-Wreserved-id-macro" #pragma clang diagnostic ignored "-Wshadow-field-in-constructor" #pragma clang diagnostic ignored "-Wsign-conversion" #pragma clang diagnostic ignored "-Wzero-as-null-pointer-constant" #endif #pragma warning(push) #pragma warning(disable : 4244 4996) #include #include #pragma warning(pop) #ifdef __clang__ #pragma clang diagnostic pop #endif static_assert(sizeof(Imf::Rgba) == 8, "Mismatch size"); using namespace DirectX; using PackedVector::XMHALF4; #ifdef _WIN32 namespace { class com_exception : public std::exception { public: com_exception(HRESULT hr) noexcept : result(hr) {} const char* what() const noexcept override { static char s_str[64] = {}; sprintf_s(s_str, "Failure with HRESULT of %08X", static_cast(result)); return s_str; } HRESULT get_result() const noexcept { return result; } private: HRESULT result; }; class InputStream : public Imf::IStream { public: InputStream(HANDLE hFile, const char fileName[]) : IStream(fileName), m_hFile(hFile) { const LARGE_INTEGER dist = {}; LARGE_INTEGER result; if (!SetFilePointerEx(m_hFile, dist, &result, FILE_END)) { throw com_exception(HRESULT_FROM_WIN32(GetLastError())); } m_EOF = result.QuadPart; if (!SetFilePointerEx(m_hFile, dist, nullptr, FILE_BEGIN)) { throw com_exception(HRESULT_FROM_WIN32(GetLastError())); } } InputStream(const InputStream&) = delete; InputStream& operator = (const InputStream&) = delete; InputStream(InputStream&&) = delete; InputStream& operator=(InputStream&&) = delete; bool read(char c[], int n) override { DWORD bytesRead; if (!ReadFile(m_hFile, c, static_cast(n), &bytesRead, nullptr)) { throw com_exception(HRESULT_FROM_WIN32(GetLastError())); } const LARGE_INTEGER dist = {}; LARGE_INTEGER result; if (!SetFilePointerEx(m_hFile, dist, &result, FILE_CURRENT)) { throw com_exception(HRESULT_FROM_WIN32(GetLastError())); } return result.QuadPart >= m_EOF; } uint64_t tellg() override { const LARGE_INTEGER dist = {}; LARGE_INTEGER result; if (!SetFilePointerEx(m_hFile, dist, &result, FILE_CURRENT)) { throw com_exception(HRESULT_FROM_WIN32(GetLastError())); } return static_cast(result.QuadPart); } void seekg(uint64_t pos) override { LARGE_INTEGER dist; dist.QuadPart = static_cast(pos); if (!SetFilePointerEx(m_hFile, dist, nullptr, FILE_BEGIN)) { throw com_exception(HRESULT_FROM_WIN32(GetLastError())); } } void clear() override { SetLastError(0); } private: HANDLE m_hFile; LONGLONG m_EOF; }; class OutputStream : public Imf::OStream { public: OutputStream(HANDLE hFile, const char fileName[]) : OStream(fileName), m_hFile(hFile) {} OutputStream(const OutputStream&) = delete; OutputStream& operator = (const OutputStream&) = delete; OutputStream(OutputStream&&) = delete; OutputStream& operator=(OutputStream&&) = delete; void write(const char c[], int n) override { DWORD bytesWritten; if (!WriteFile(m_hFile, c, static_cast(n), &bytesWritten, nullptr)) { throw com_exception(HRESULT_FROM_WIN32(GetLastError())); } } uint64_t tellp() override { const LARGE_INTEGER dist = {}; LARGE_INTEGER result; if (!SetFilePointerEx(m_hFile, dist, &result, FILE_CURRENT)) { throw com_exception(HRESULT_FROM_WIN32(GetLastError())); } return static_cast(result.QuadPart); } void seekp(uint64_t pos) override { LARGE_INTEGER dist; dist.QuadPart = static_cast(pos); if (!SetFilePointerEx(m_hFile, dist, nullptr, FILE_BEGIN)) { throw com_exception(HRESULT_FROM_WIN32(GetLastError())); } } private: HANDLE m_hFile; }; } #endif // _WIN32 //===================================================================================== // Entry-points //===================================================================================== //------------------------------------------------------------------------------------- // Obtain metadata from EXR file on disk //------------------------------------------------------------------------------------- _Use_decl_annotations_ HRESULT DirectX::GetMetadataFromEXRFile(const wchar_t* szFile, TexMetadata& metadata) { if (!szFile) return E_INVALIDARG; #ifdef _WIN32 std::string fileName; const int nameLength = WideCharToMultiByte(CP_UTF8, 0, szFile, -1, nullptr, 0, nullptr, nullptr); if (nameLength > 0) { fileName.resize(nameLength); const int result = WideCharToMultiByte(CP_UTF8, 0, szFile, -1, fileName.data(), nameLength, nullptr, nullptr); if (result <= 0) { fileName.clear(); } } #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()); } InputStream stream(hFile.get(), fileName.c_str()); #else std::wstring wFileName(szFile); std::string fileName(wFileName.cbegin(), wFileName.cend()); #endif HRESULT hr = S_OK; try { #ifdef _WIN32 Imf::RgbaInputFile file(stream); #else Imf::RgbaInputFile file(fileName.c_str()); #endif const auto dw = file.dataWindow(); const int width = dw.max.x - dw.min.x + 1; int height = dw.max.y - dw.min.y + 1; size_t arraySize = 1; if (width < 1 || height < 1) return E_FAIL; if (file.header().find("envmap") != file.header().end()) { if (width == height / 6) { height = width; arraySize = 6; } } metadata.width = static_cast(width); metadata.height = static_cast(height); metadata.depth = metadata.mipLevels = 1; metadata.arraySize = arraySize; metadata.format = DXGI_FORMAT_R16G16B16A16_FLOAT; metadata.dimension = TEX_DIMENSION_TEXTURE2D; } #ifdef _WIN32 catch (const com_exception& exc) { #ifdef _DEBUG OutputDebugStringA(exc.what()); #endif hr = exc.get_result(); } #endif #if defined(_WIN32) && defined(_DEBUG) catch (const std::exception& exc) { OutputDebugStringA(exc.what()); hr = E_FAIL; } #else catch (const std::exception&) { hr = E_FAIL; } #endif catch (...) { hr = E_UNEXPECTED; } return hr; } //------------------------------------------------------------------------------------- // Load a EXR file from disk //------------------------------------------------------------------------------------- _Use_decl_annotations_ HRESULT DirectX::LoadFromEXRFile(const wchar_t* szFile, TexMetadata* metadata, ScratchImage& image) { if (!szFile) return E_INVALIDARG; image.Release(); if (metadata) { memset(metadata, 0, sizeof(TexMetadata)); } #ifdef _WIN32 std::string fileName; const int nameLength = WideCharToMultiByte(CP_UTF8, 0, szFile, -1, nullptr, 0, nullptr, nullptr); if (nameLength > 0) { fileName.resize(nameLength); const int result = WideCharToMultiByte(CP_UTF8, 0, szFile, -1, fileName.data(), nameLength, nullptr, nullptr); if (result <= 0) { fileName.clear(); } } #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()); } InputStream stream(hFile.get(), fileName.c_str()); #else std::wstring wFileName(szFile); std::string fileName(wFileName.cbegin(), wFileName.cend()); #endif HRESULT hr = S_OK; try { #ifdef _WIN32 Imf::RgbaInputFile file(stream); #else Imf::RgbaInputFile file(fileName.c_str()); #endif auto const dw = file.dataWindow(); const int width = dw.max.x - dw.min.x + 1; int height = dw.max.y - dw.min.y + 1; size_t arraySize = 1; if (width < 1 || height < 1) return E_FAIL; if (file.header().find("envmap") != file.header().end()) { if (width == height / 6) { height = width; arraySize = 6; } } if (metadata) { metadata->width = static_cast(width); metadata->height = static_cast(height); metadata->depth = metadata->mipLevels = 1; metadata->arraySize = arraySize; metadata->format = DXGI_FORMAT_R16G16B16A16_FLOAT; metadata->dimension = TEX_DIMENSION_TEXTURE2D; } hr = image.Initialize2D(DXGI_FORMAT_R16G16B16A16_FLOAT, static_cast(width), static_cast(height), arraySize, 1u); if (FAILED(hr)) return hr; file.setFrameBuffer(reinterpret_cast(image.GetPixels()) - dw.min.x - dw.min.y * width, 1, static_cast(width)); file.readPixels(dw.min.y, dw.max.y); } #ifdef _WIN32 catch (const com_exception& exc) { #ifdef _DEBUG OutputDebugStringA(exc.what()); #endif hr = exc.get_result(); } #endif #if defined(_WIN32) && defined(_DEBUG) catch (const std::exception& exc) { OutputDebugStringA(exc.what()); hr = E_FAIL; } #else catch (const std::exception&) { hr = E_FAIL; } #endif catch (...) { hr = E_UNEXPECTED; } if (FAILED(hr)) { image.Release(); } return hr; } //------------------------------------------------------------------------------------- // Save a EXR file to disk //------------------------------------------------------------------------------------- _Use_decl_annotations_ HRESULT DirectX::SaveToEXRFile(const Image& image, const wchar_t* szFile) { if (!szFile) return E_INVALIDARG; if (!image.pixels) return E_POINTER; if (image.width > INT32_MAX || image.height > INT32_MAX) return /* HRESULT_FROM_WIN32(ERROR_NOT_SUPPORTED) */ static_cast(0x80070032L); switch (image.format) { case DXGI_FORMAT_R16G16B16A16_FLOAT: if ((image.rowPitch % 8) > 0) return E_FAIL; break; case DXGI_FORMAT_R32G32B32A32_FLOAT: case DXGI_FORMAT_R32G32B32_FLOAT: break; default: return /* HRESULT_FROM_WIN32(ERROR_NOT_SUPPORTED) */ static_cast(0x80070032L); } #ifdef _WIN32 std::string fileName; const int nameLength = WideCharToMultiByte(CP_UTF8, 0, szFile, -1, nullptr, 0, nullptr, nullptr); if (nameLength > 0) { fileName.resize(nameLength); const int result = WideCharToMultiByte(CP_UTF8, 0, szFile, -1, fileName.data(), nameLength, nullptr, nullptr); if (result <= 0) { fileName.clear(); } } // 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, FILE_ATTRIBUTE_NORMAL, nullptr))); #endif if (!hFile) { return HRESULT_FROM_WIN32(GetLastError()); } auto_delete_file delonfail(hFile.get()); OutputStream stream(hFile.get(), fileName.c_str()); #else std::wstring wFileName(szFile); std::string fileName(wFileName.cbegin(), wFileName.cend()); #endif HRESULT hr = S_OK; try { const int width = static_cast(image.width); const int height = static_cast(image.height); #ifdef _WIN32 Imf::RgbaOutputFile file(stream, Imf::Header(width, height), Imf::WRITE_RGBA); #else Imf::RgbaOutputFile file(fileName.c_str(), Imf::Header(width, height), Imf::WRITE_RGBA); #endif if (image.format == DXGI_FORMAT_R16G16B16A16_FLOAT) { file.setFrameBuffer(reinterpret_cast(image.pixels), 1, image.rowPitch / 8); file.writePixels(height); } else { const uint64_t bytes = image.width * image.height; if (bytes > static_cast(UINT32_MAX)) { return /* HRESULT_FROM_WIN32(ERROR_ARITHMETIC_OVERFLOW) */ static_cast(0x80070216L); } std::unique_ptr temp(new (std::nothrow) XMHALF4[static_cast(bytes)]); if (!temp) return E_OUTOFMEMORY; file.setFrameBuffer(reinterpret_cast(temp.get()), 1, image.width); auto sPtr = image.pixels; auto dPtr = temp.get(); if (image.format == DXGI_FORMAT_R32G32B32A32_FLOAT) { for (int j = 0; j < height; ++j) { auto srcPtr = reinterpret_cast(sPtr); auto destPtr = dPtr; for (int k = 0; k < width; ++k, ++srcPtr, ++destPtr) { const XMVECTOR v = XMLoadFloat4(srcPtr); PackedVector::XMStoreHalf4(destPtr, v); } sPtr += image.rowPitch; dPtr += width; file.writePixels(1); } } else { assert(image.format == DXGI_FORMAT_R32G32B32_FLOAT); for (int j = 0; j < height; ++j) { auto srcPtr = reinterpret_cast(sPtr); auto destPtr = dPtr; for (int k = 0; k < width; ++k, ++srcPtr, ++destPtr) { XMVECTOR v = XMLoadFloat3(srcPtr); v = XMVectorSelect(g_XMIdentityR3, v, g_XMSelect1110); PackedVector::XMStoreHalf4(destPtr, v); } sPtr += image.rowPitch; dPtr += width; file.writePixels(1); } } } } #ifdef _WIN32 catch (const com_exception& exc) { #ifdef _DEBUG OutputDebugStringA(exc.what()); #endif hr = exc.get_result(); } #endif #if defined(_WIN32) && defined(_DEBUG) catch (const std::exception& exc) { OutputDebugStringA(exc.what()); hr = E_FAIL; } #else catch (const std::exception&) { hr = E_FAIL; } #endif catch (...) { hr = E_UNEXPECTED; } if (FAILED(hr)) return hr; #ifdef _WIN32 delonfail.clear(); #endif return S_OK; }