GoaLitiuM/FlaxEngine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add SPIR-V compression with LZ4 of Vulkan shaders (35% avg smaller)

Browse Source
This commit is contained in:
Wojtek Figat
2026-03-07 23:24:40 +01:00
parent 3038c56af4
commit c4342b0a20
4 changed files with 44 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
Source/Engine/ShadersCompilation/Vulkan/ShaderCompilerVulkan.cpp
View File

@@ -9,6 +9,7 @@
#include "Engine/Serialization/MemoryWriteStream.h"
#include "Engine/Graphics/Config.h"
#include "Engine/GraphicsDevice/Vulkan/Types.h"
#include <ThirdParty/LZ4/lz4.h>
// Use glslang for HLSL to SPIR-V translation
// Source: https://github.com/KhronosGroup/glslang
@@ -939,8 +940,6 @@ bool ShaderCompilerVulkan::OnCompileBegin()
//_globalMacros.Add({ "VULKAN", "1" }); // glslang compiler adds VULKAN define if EShMsgVulkanRules flag is specified
// TODO: handle options->TreatWarningsAsErrors
return false;
}
@@ -967,9 +966,22 @@ void ShaderCompilerVulkan::InitCodegen(ShaderCompilationContext* context, glslan
bool ShaderCompilerVulkan::Write(ShaderCompilationContext* context, ShaderFunctionMeta& meta, int32 permutationIndex, const ShaderBindings& bindings, struct SpirvShaderHeader& header, std::vector<unsigned int>& spirv)
{
int32 spirvBytesCount = (int32)spirv.size() * sizeof(unsigned);
// Compress
const int32 srcSize = (int32)spirv.size() * sizeof(unsigned);
const int32 maxSize = LZ4_compressBound(srcSize);
Array<byte> spirvCompressed;
spirvCompressed.Resize(maxSize + sizeof(int32));
const int32 dstSize = LZ4_compress_default((const char*)&spirv[0], (char*)spirvCompressed.Get() + sizeof(int32), srcSize, maxSize);
if (dstSize > 0 && dstSize < (int32)(srcSize * 0.8f)) // Expect 20% or more compression ratio to use it (to avoid decompressing if the gain is not big enough)
{
spirvCompressed.Resize(dstSize + sizeof(int32));
*(int32*)spirvCompressed.Get() = srcSize; // Store original size in the beginning to decompress it
header.Type = SpirvShaderHeader::Types::SPIRV_LZ4;
return WriteShaderFunctionPermutation(_context, meta, permutationIndex, bindings, &header, sizeof(header), spirvCompressed.Get(), spirvCompressed.Count());
}
header.Type = SpirvShaderHeader::Types::SPIRV;
return WriteShaderFunctionPermutation(_context, meta, permutationIndex, bindings, &header, sizeof(header), &spirv[0], spirvBytesCount);
return WriteShaderFunctionPermutation(_context, meta, permutationIndex, bindings, &header, sizeof(header), &spirv[0], srcSize);
}
#endif