AssetRipper.TextureDecoder/AssetRipper.TextureDecoder.ColorGenerator/Program.cs

using System.CodeDom.Compiler;

namespace AssetRipper.TextureDecoder.ColorGenerator;

internal class Program
{
	private const string OutputNamespace = "AssetRipper.TextureDecoder.Rgb.Formats";
	private const string OutputFolder = "../../../../AssetRipper.TextureDecoder/Rgb/Formats/";

	private static readonly Dictionary<Type, string> MinimumValues = new()
	{
		{ typeof(byte), "byte.MinValue" },
		{ typeof(sbyte), "sbyte.MinValue" },
		{ typeof(short), "short.MinValue" },
		{ typeof(ushort), "ushort.MinValue" },
		{ typeof(Half), "default" },
		{ typeof(float), "0f" },
		{ typeof(double), "0d" },
	};

	private static readonly Dictionary<Type, string> MaximumValues = new()
	{
		{ typeof(byte), "byte.MaxValue" },
		{ typeof(sbyte), "sbyte.MaxValue" },
		{ typeof(short), "short.MaxValue" },
		{ typeof(ushort), "ushort.MaxValue" },
		{ typeof(Half), "(Half)1" },
		{ typeof(float), "1f" },
		{ typeof(double), "1d" },
	};

	private static readonly Dictionary<Type, string> TypeNames = new()
	{
		{ typeof(byte), "byte" },
		{ typeof(sbyte), "sbyte" },
		{ typeof(short), "short" },
		{ typeof(ushort), "ushort" },
		{ typeof(Half), "Half" },
		{ typeof(float), "float" },
		{ typeof(double), "double" },
	};

	/// <summary>
	/// Name, Type, Red, Blue, Green, Alpha
	/// </summary>
	private static readonly List<(string, Type, bool, bool, bool, bool)> Colors = new()
	{
		( "ColorA8", typeof(byte), false, false, false, true ),

		( "ColorR8", typeof(byte), true, false, false, false ),
		( "ColorRG16", typeof(byte), true, true, false, false ),
		( "ColorRGB24", typeof(byte), true, true, true, false ),
		( "ColorRGBA32", typeof(byte), true, true, true, true ),

		( "ColorR8Signed", typeof(sbyte), true, false, false, false ),
		( "ColorRG16Signed", typeof(sbyte), true, true, false, false ),
		( "ColorRGB24Signed", typeof(sbyte), true, true, true, false ),
		( "ColorRGBA32Signed", typeof(sbyte), true, true, true, true ),

		( "ColorR16", typeof(ushort), true, false, false, false ),
		( "ColorRG32", typeof(ushort), true, true, false, false ),
		( "ColorRGB48", typeof(ushort), true, true, true, false ),
		( "ColorRGBA64", typeof(ushort), true, true, true, true ),

		( "ColorR16Signed", typeof(short), true, false, false, false ),
		( "ColorRG32Signed", typeof(short), true, true, false, false ),
		( "ColorRGB48Signed", typeof(short), true, true, true, false ),
		( "ColorRGBA64Signed", typeof(short), true, true, true, true ),

		( "ColorRHalf", typeof(Half), true, false, false, false ),
		( "ColorRGHalf", typeof(Half), true, true, false, false ),
		( "ColorRGBHalf", typeof(Half), true, true, true, false ),
		( "ColorRGBAHalf", typeof(Half), true, true, true, true ),

		( "ColorRSingle", typeof(float), true, false, false, false ),
		( "ColorRGSingle", typeof(float), true, true, false, false ),
		( "ColorRGBSingle", typeof(float), true, true, true, false ),
		( "ColorRGBASingle", typeof(float), true, true, true, true ),
	};

	static void Main()
	{
		foreach (var details in Colors)
		{
			WriteStruct(details);
		}
		Console.WriteLine("Done!");
	}

	private static void WriteStruct((string, Type, bool, bool, bool, bool) details)
	{
		(string name, Type type, bool hasRed, bool hasGreen, bool hasBlue, bool hasAlpha) = details;
		Console.WriteLine(name);
		using FileStream stream = File.Create($"{OutputFolder}{name}.g.cs");
		using StreamWriter streamWriter = new StreamWriter(stream);
		using IndentedTextWriter writer = new IndentedTextWriter(streamWriter, "\t");
		WriteStruct(writer, name, type, hasRed, hasGreen, hasBlue, hasAlpha);
	}

	private static void WriteStruct(IndentedTextWriter writer, string name, Type type, bool hasRed, bool hasGreen, bool hasBlue, bool hasAlpha)
	{
		string typeName = TypeNames[type];
		writer.WriteLine("//Auto-generated code. Do not modify.");
		writer.WriteLine($"namespace {OutputNamespace}");
		writer.WriteLine('{');
		writer.Indent++;

		writer.WriteLine($"public partial struct {name} : IColor<{typeName}>");
		writer.WriteLine('{');
		writer.Indent++;

		WriteProperty(writer, hasRed, typeName, 'R', MinimumValues[type]);
		writer.WriteLine();
		WriteProperty(writer, hasGreen, typeName, 'G', MinimumValues[type]);
		writer.WriteLine();
		WriteProperty(writer, hasBlue, typeName, 'B', MinimumValues[type]);
		writer.WriteLine();
		WriteProperty(writer, hasAlpha, typeName, 'A', MaximumValues[type]);
		writer.WriteLine();

		WriteGetChannels(writer, typeName);
		writer.WriteLine();
		WriteSetChannels(writer, typeName);

		writer.Indent--;
		writer.WriteLine('}');

		writer.Indent--;
		writer.WriteLine('}');
	}

	private static void WriteGetChannels(IndentedTextWriter writer, string typeName)
	{
		writer.WriteLine($"public void GetChannels(out {typeName} r, out {typeName} g, out {typeName} b, out {typeName} a)");
		writer.WriteLine('{');
		writer.Indent++;
		writer.WriteLine("DefaultColorMethods.GetChannels(this, out r, out g, out b, out a);");
		writer.Indent--;
		writer.WriteLine('}');
	}

	private static void WriteSetChannels(IndentedTextWriter writer, string typeName)
	{
		writer.WriteLine($"public void SetChannels({typeName} r, {typeName} g, {typeName} b, {typeName} a)");
		writer.WriteLine('{');
		writer.Indent++;
		writer.WriteLine("DefaultColorMethods.SetChannels(ref this, r, g, b, a);");
		writer.Indent--;
		writer.WriteLine('}');
	}

	private static void WriteProperty(IndentedTextWriter writer, bool hasColor, string typeName, char channel, string defaultValue)
	{
		if (hasColor)
		{
			writer.WriteLine($"public {typeName} {channel} {{ get; set; }}");
		}
		else
		{
			writer.WriteLine($"public {typeName} {channel} ");
			writer.WriteLine('{');
			writer.Indent++;
			writer.WriteLine($"get => {defaultValue};");
			writer.WriteLine("set { }");
			writer.Indent--;
			writer.WriteLine('}');
		}
	}
}