static ezResult DecompressZStd(ezArrayPtr<const ezUInt8> pCompressedData, ezDynamicArray<ezUInt8>& out_Data)
{
size_t uiSize = ZSTD_findDecompressedSize(pCompressedData.GetPtr(), pCompressedData.GetCount());
if (uiSize == ZSTD_CONTENTSIZE_ERROR)
{
ezLog::Error("Can't decompress since it wasn't compressed with ZStd");
return EZ_FAILURE;
}
else if (uiSize == ZSTD_CONTENTSIZE_UNKNOWN)
{
ezLog::Error("Can't decompress since the original size can't be determined, was the data compressed using the streaming variant?");
return EZ_FAILURE;
}
if (uiSize > std::numeric_limits<ezUInt32>::max())
{
ezLog::Error("Can't compress since the output container can't hold enough elements ({0})", static_cast<ezUInt64>(uiSize));
return EZ_FAILURE;
}
out_Data.SetCountUninitialized(static_cast<ezUInt32>(uiSize));
size_t const uiActualSize = ZSTD_decompress(out_Data.GetData(), uiSize, pCompressedData.GetPtr(), pCompressedData.GetCount());
if (uiActualSize != uiSize)
{
ezLog::Error("Error during ZStd decompression: '{0}'.", ZSTD_getErrorName(uiActualSize));
return EZ_FAILURE;
}
return EZ_SUCCESS;
}
static ezResult CompressZStd(ezArrayPtr<const ezUInt8> pUncompressedData, ezDynamicArray<ezUInt8>& out_Data)
{
size_t uiSizeBound = ZSTD_compressBound(pUncompressedData.GetCount());
if(uiSizeBound > std::numeric_limits<ezUInt32>::max())
{
ezLog::Error("Can't compress since the output container can't hold enough elements ({0})", static_cast<ezUInt64>(uiSizeBound));
return EZ_FAILURE;
}
out_Data.SetCountUninitialized(static_cast<ezUInt32>(uiSizeBound));
size_t const cSize = ZSTD_compress(out_Data.GetData(), uiSizeBound, pUncompressedData.GetPtr(), pUncompressedData.GetCount(), 1);
if (ZSTD_isError(cSize))
{
ezLog::Error("Compression failed with error: '{0}'.", ZSTD_getErrorName(cSize));
return EZ_FAILURE;
}
out_Data.SetCount(static_cast<ezUInt32>(cSize));
return EZ_SUCCESS;
}
ezResult CompileDXShader(const char* szFile, const char* szSource, bool bDebug, const char* szProfile, const char* szEntryPoint,
ezDynamicArray<ezUInt8>& out_ByteCode)
{
out_ByteCode.Clear();
ID3DBlob* pResultBlob = nullptr;
ID3DBlob* pErrorBlob = nullptr;
const char* szCompileSource = szSource;
ezStringBuilder sDebugSource;
UINT flags1 = 0;
if (bDebug)
{
flags1 = D3DCOMPILE_DEBUG | D3DCOMPILE_PREFER_FLOW_CONTROL | D3DCOMPILE_SKIP_OPTIMIZATION | D3DCOMPILE_ENABLE_STRICTNESS;
// In debug mode we need to remove '#line' as any shader debugger won't work with them.
sDebugSource = szSource;
sDebugSource.ReplaceAll("#line ", "//ine ");
szCompileSource = sDebugSource;
}
if (FAILED(D3DCompile(
szCompileSource, strlen(szCompileSource), szFile, nullptr, nullptr, szEntryPoint, szProfile, flags1, 0, &pResultBlob, &pErrorBlob)))
{
if (bDebug)
{
// Try again with '#line' intact to get correct error messages with file and line info.
pErrorBlob->Release();
pErrorBlob = nullptr;
EZ_VERIFY(FAILED(D3DCompile(
szSource, strlen(szSource), szFile, nullptr, nullptr, szEntryPoint, szProfile, flags1, 0, &pResultBlob, &pErrorBlob)),
"Debug compilation with commented out '#line' failed but original version did not.");
}
const char* szError = static_cast<const char*>(pErrorBlob->GetBufferPointer());
EZ_LOG_BLOCK("Shader Compilation Failed", szFile);
ezLog::Error("Could not compile shader '{0}' for profile '{1}'", szFile, szProfile);
ezLog::Error("{0}", szError);
pErrorBlob->Release();
return EZ_FAILURE;
}
if (pErrorBlob != nullptr)
{
const char* szError = static_cast<const char*>(pErrorBlob->GetBufferPointer());
EZ_LOG_BLOCK("Shader Compilation Error Message", szFile);
ezLog::Dev("{0}", szError);
pErrorBlob->Release();
}
if (pResultBlob != nullptr)
{
out_ByteCode.SetCountUninitialized((ezUInt32)pResultBlob->GetBufferSize());
ezMemoryUtils::Copy(out_ByteCode.GetData(), static_cast<ezUInt8*>(pResultBlob->GetBufferPointer()), out_ByteCode.GetCount());
pResultBlob->Release();
}
return EZ_SUCCESS;
}
