Files
sbox-public/engine/Sandbox.Tools/Assets/ResourceCompile/TextureCompiler.cs
Carson Kompon 1e03e9ee7d PNG mipmap fix (#5045)
* PNG formats no longer incorrectly try to save/load mips it shouldn't

* Added [Hide] to `TextureGenerator.ImageFormat`
2026-06-10 07:14:33 -04:00

110 lines
3.2 KiB
C#

using Sandbox.Resources;
namespace Editor;
[Expose]
[ResourceIdentity( "texture" )]
[ResourceIdentity( "vtex" )]
public class TextureResourceCompiler : ResourceCompiler
{
public TextureResourceCompiler()
{
}
/// <summary>
/// We found an embedded resource definition.
/// 1. Find the TextureGenerator
/// 2. Create a child texture resource with a deterministic name
/// 3. Put the provided compile data in that and let it compile
/// 4. Store a reference to the compiled version in the json
/// </summary>
protected override bool CompileEmbedded( ref EmbeddedResource embed )
{
return CompileEmbeddedResource<Texture>( ref embed, "textures", "vtex", FileSystem.Transient );
}
override protected async Task<bool> Compile()
{
if ( !TryParseEmbeddedResource( out var serialized ) || !serialized.HasValue )
return false;
var generator = ResourceGenerator.Create<Texture>( serialized.Value );
if ( generator is null || !generator.CacheToDisk ) return false;
var texture = await generator.CreateAsync( new ResourceGenerator.Options { ForDisk = true, Compiler = this }, default );
if ( texture is null ) return false;
Context.ResourceVersion = 1;
var desc = texture.Desc;
int width = desc.m_nWidth;
int height = desc.m_nHeight;
int depth = desc.m_nDepth;
int mipCount = desc.m_nNumMipLevels;
var formatOverride = generator is TextureGenerator texGen ? texGen.FormatOverride : null;
var writer = new VTexWriter();
VTexWriter.VTEX_Format_t diskFormat;
if ( formatOverride.HasValue )
diskFormat = VTexWriter.RuntimeToVTEX_Format( formatOverride.Value ) ?? VTexWriter.VTEX_Format_t.VTEX_FORMAT_RGBA8888;
else
diskFormat = default; // will be calculated after SetTexture
// PNG format only supports a single mip level
bool isPngFormat = diskFormat == VTexWriter.VTEX_Format_t.VTEX_FORMAT_PNG_RGBA8888
|| diskFormat == VTexWriter.VTEX_Format_t.VTEX_FORMAT_PNG_DXT5;
if ( isPngFormat )
mipCount = 1;
for ( var mip = 0; mip < mipCount; mip++ )
{
var bitmap = texture.GetBitmap( mip );
writer.SetTexture( bitmap, mip );
}
writer.Header.Width = (ushort)width;
writer.Header.Height = (ushort)height;
writer.Header.Depth = (ushort)depth;
writer.Header.MipCount = (byte)mipCount;
var flags = VTexWriter.VTEX_Flags_t.NONE;
if ( desc.m_nFlags.HasFlag( NativeEngine.RuntimeTextureSpecificationFlags.TSPEC_CUBE_TEXTURE ) )
{
flags |= VTexWriter.VTEX_Flags_t.VTEX_FLAG_CUBE_TEXTURE;
}
if ( desc.m_nFlags.HasFlag( NativeEngine.RuntimeTextureSpecificationFlags.TSPEC_VOLUME_TEXTURE ) )
{
flags |= VTexWriter.VTEX_Flags_t.VTEX_FLAG_VOLUME_TEXTURE;
}
if ( desc.m_nFlags.HasFlag( NativeEngine.RuntimeTextureSpecificationFlags.TSPEC_TEXTURE_ARRAY ) )
{
flags |= VTexWriter.VTEX_Flags_t.VTEX_FLAG_TEXTURE_ARRAY;
}
if ( desc.m_nFlags.HasFlag( NativeEngine.RuntimeTextureSpecificationFlags.TSPEC_NO_LOD ) || isPngFormat )
{
flags |= VTexWriter.VTEX_Flags_t.VTEX_FLAG_NO_LOD;
}
writer.Header.Flags = flags;
if ( formatOverride.HasValue )
writer.Header.Format = diskFormat;
else
writer.CalculateFormat();
Context.Data.Write( writer.GetData() );
Context.StreamingData.Write( writer.GetStreamingData() );
return true;
}
}