using System; using System.Collections.Generic; using System.Linq; using System.Text; using System.Threading.Tasks; using Toolbox.Library.IO; namespace Toolbox.Library { public class CTR_3DS { //From https://github.com/gdkchan/SPICA/blob/42c4181e198b0fd34f0a567345ee7e75b54cb58b/SPICA/PICA/Converters/TextureConverter.cs public enum PICASurfaceFormat { RGBA8, RGB8, RGBA5551, RGB565, RGBA4, LA8, HiLo8, L8, A8, LA4, L4, A4, ETC1, ETC1A4 } public static TEX_FORMAT ConvertPICAToGenericFormat(PICASurfaceFormat format) { switch (format) { case PICASurfaceFormat.RGB565: return TEX_FORMAT.B5G6R5_UNORM; case PICASurfaceFormat.RGB8: return TEX_FORMAT.R8G8_UNORM; case PICASurfaceFormat.RGBA8: return TEX_FORMAT.R8G8B8A8_UNORM; case PICASurfaceFormat.RGBA5551: return TEX_FORMAT.B5G5R5A1_UNORM; case PICASurfaceFormat.RGBA4: return TEX_FORMAT.B4G4R4A4_UNORM; case PICASurfaceFormat.LA8: return TEX_FORMAT.LA8; case PICASurfaceFormat.HiLo8: return TEX_FORMAT.HIL08; case PICASurfaceFormat.L8: return TEX_FORMAT.L8; case PICASurfaceFormat.A8: return TEX_FORMAT.A8_UNORM; case PICASurfaceFormat.LA4: return TEX_FORMAT.LA4; case PICASurfaceFormat.L4: return TEX_FORMAT.L4; case PICASurfaceFormat.A4: return TEX_FORMAT.A4; case PICASurfaceFormat.ETC1: return TEX_FORMAT.ETC1_UNORM; case PICASurfaceFormat.ETC1A4: return TEX_FORMAT.ETC1_A4; default: throw new NotImplementedException("Unsupported format! " + format); } } public static PICASurfaceFormat ConvertToPICAFormat(TEX_FORMAT GenericFormat) { switch (GenericFormat) { case TEX_FORMAT.B5G6R5_UNORM: return PICASurfaceFormat.RGB565; case TEX_FORMAT.R8G8_UNORM: return PICASurfaceFormat.RGB8; case TEX_FORMAT.R8G8B8A8_UNORM: return PICASurfaceFormat.RGBA8; case TEX_FORMAT.B5G5R5A1_UNORM: return PICASurfaceFormat.RGBA5551; case TEX_FORMAT.B4G4R4A4_UNORM: return PICASurfaceFormat.RGBA4; case TEX_FORMAT.LA8: return PICASurfaceFormat.LA8; case TEX_FORMAT.HIL08: return PICASurfaceFormat.HiLo8; case TEX_FORMAT.L8: return PICASurfaceFormat.L8; case TEX_FORMAT.A8_UNORM: return PICASurfaceFormat.A8; case TEX_FORMAT.LA4: return PICASurfaceFormat.LA4; case TEX_FORMAT.L4: return PICASurfaceFormat.L4; case TEX_FORMAT.A4: return PICASurfaceFormat.A4; case TEX_FORMAT.ETC1_UNORM: return PICASurfaceFormat.ETC1; case TEX_FORMAT.ETC1_A4: return PICASurfaceFormat.ETC1A4; default: throw new NotImplementedException("Unsupported format! " + GenericFormat); } } private static int[] FmtBPP = new int[] { 32, 24, 16, 16, 16, 16, 16, 8, 8, 8, 4, 4, 4, 8 }; public static int[] SwizzleLUT = { 0, 1, 8, 9, 2, 3, 10, 11, 16, 17, 24, 25, 18, 19, 26, 27, 4, 5, 12, 13, 6, 7, 14, 15, 20, 21, 28, 29, 22, 23, 30, 31, 32, 33, 40, 41, 34, 35, 42, 43, 48, 49, 56, 57, 50, 51, 58, 59, 36, 37, 44, 45, 38, 39, 46, 47, 52, 53, 60, 61, 54, 55, 62, 63 }; public static System.Drawing.Bitmap DecodeBlockToBitmap(byte[] Input, int Width, int Height, PICASurfaceFormat picaFormat) { return BitmapExtension.GetBitmap(STGenericTexture.ConvertBgraToRgba(DecodeBlock(Input, Width, Height, picaFormat)), Width, Height, System.Drawing.Imaging.PixelFormat.Format32bppArgb); } public static byte[] DecodeBlock(byte[] Input, int Width, int Height, TEX_FORMAT Format) { return DecodeBlock(Input, Width, Height, ConvertToPICAFormat(Format)); } public static byte[] DecodeBlock(byte[] Input, int Width, int Height, PICASurfaceFormat picaFormat) { if (picaFormat == PICASurfaceFormat.ETC1 || picaFormat == PICASurfaceFormat.ETC1A4) return ETC1.ETC1Decompress(Input, Width, Height, picaFormat == PICASurfaceFormat.ETC1A4); byte[] Output = new byte[Width * Height * 4]; int Increment = FmtBPP[(int)picaFormat] / 8; if (Increment == 0) Increment = 1; int IOffset = 0; for (int TY = 0; TY < Height; TY += 8) { for (int TX = 0; TX < Width; TX += 8) { for (int Px = 0; Px < 64; Px++) { int X = SwizzleLUT[Px] & 7; int Y = (SwizzleLUT[Px] - X) >> 3; int OOffet = (TX + X + ((Height - 1 - (TY + Y)) * Width)) * 4; switch (picaFormat) { case PICASurfaceFormat.RGBA8: Output[OOffet + 0] = Input[IOffset + 3]; Output[OOffet + 1] = Input[IOffset + 2]; Output[OOffet + 2] = Input[IOffset + 1]; Output[OOffet + 3] = Input[IOffset + 0]; break; case PICASurfaceFormat.RGB8: Output[OOffet + 0] = Input[IOffset + 2]; Output[OOffet + 1] = Input[IOffset + 1]; Output[OOffet + 2] = Input[IOffset + 0]; Output[OOffet + 3] = 0xff; break; case PICASurfaceFormat.RGBA5551: DecodeRGBA5551(Output, OOffet, GetUShort(Input, IOffset)); break; case PICASurfaceFormat.RGB565: DecodeRGB565(Output, OOffet, GetUShort(Input, IOffset)); break; case PICASurfaceFormat.RGBA4: DecodeRGBA4(Output, OOffet, GetUShort(Input, IOffset)); break; case PICASurfaceFormat.LA8: Output[OOffet + 0] = Input[IOffset + 1]; Output[OOffet + 1] = Input[IOffset + 1]; Output[OOffet + 2] = Input[IOffset + 1]; Output[OOffet + 3] = Input[IOffset + 0]; break; case PICASurfaceFormat.HiLo8: Output[OOffet + 0] = Input[IOffset + 1]; Output[OOffet + 1] = Input[IOffset + 0]; Output[OOffet + 2] = 0; Output[OOffet + 3] = 0xff; break; case PICASurfaceFormat.L8: Output[OOffet + 0] = Input[IOffset]; Output[OOffet + 1] = Input[IOffset]; Output[OOffet + 2] = Input[IOffset]; Output[OOffet + 3] = 0xff; break; case PICASurfaceFormat.A8: Output[OOffet + 0] = 0xff; Output[OOffet + 1] = 0xff; Output[OOffet + 2] = 0xff; Output[OOffet + 3] = Input[IOffset]; break; case PICASurfaceFormat.LA4: Output[OOffet + 0] = (byte)((Input[IOffset] >> 4) | (Input[IOffset] & 0xf0)); Output[OOffet + 1] = (byte)((Input[IOffset] >> 4) | (Input[IOffset] & 0xf0)); Output[OOffet + 2] = (byte)((Input[IOffset] >> 4) | (Input[IOffset] & 0xf0)); Output[OOffet + 3] = (byte)((Input[IOffset] << 4) | (Input[IOffset] & 0x0f)); break; case PICASurfaceFormat.L4: int L = (Input[IOffset >> 1] >> ((IOffset & 1) << 2)) & 0xf; Output[OOffet + 0] = (byte)((L << 4) | L); Output[OOffet + 1] = (byte)((L << 4) | L); Output[OOffet + 2] = (byte)((L << 4) | L); Output[OOffet + 3] = 0xff; break; case PICASurfaceFormat.A4: int A = (Input[IOffset >> 1] >> ((IOffset & 1) << 2)) & 0xf; Output[OOffet + 0] = 0xff; Output[OOffet + 1] = 0xff; Output[OOffet + 2] = 0xff; Output[OOffet + 3] = (byte)((A << 4) | A); break; } IOffset += Increment; } } } return Output; } public static byte[] EncodeBlock(byte[] Input, int Width, int Height, TEX_FORMAT Format) { return EncodeBlock(Input, Width, Height, ConvertToPICAFormat(Format)); } //Much help from encoding thanks to this // https://github.com/Cruel/3dstex/blob/master/src/Encoder.cpp public static byte[] EncodeBlock(byte[] Input, int Width, int Height, PICASurfaceFormat PicaFormat) { int ImageSize = CalculateLength(Width, Height, PicaFormat); if (PicaFormat == PICASurfaceFormat.ETC1) return SmashForge.RG_ETC1.encodeETC(BitmapExtension.GetBitmap(Input, Width, Height)); else if (PicaFormat == PICASurfaceFormat.ETC1A4) return SmashForge.RG_ETC1.encodeETCa4(BitmapExtension.GetBitmap(Input, Width, Height)); var mem = new System.IO.MemoryStream(); using (var writer = new FileWriter(mem)) { for (int TY = 0; TY < Height; TY += 8) { for (int TX = 0; TX < Width; TX += 8) { for (int Px = 0; Px < 64; Px++) { int X = SwizzleLUT[Px] & 7; int Y = (SwizzleLUT[Px] - X) >> 3; int IOffs = (TX + X + ((TY + Y) * Width)) * 4; if (PicaFormat == PICASurfaceFormat.RGBA8) { writer.Write(Input[IOffs + 3]); writer.Write(Input[IOffs + 0]); writer.Write(Input[IOffs + 1]); writer.Write(Input[IOffs + 2]); } else if (PicaFormat == PICASurfaceFormat.RGB8) { writer.Write(Input[IOffs + 0]); writer.Write(Input[IOffs + 1]); writer.Write(Input[IOffs + 2]); } else if (PicaFormat == PICASurfaceFormat.A8) { writer.Write(Input[IOffs]); } else if (PicaFormat == PICASurfaceFormat.L8) { writer.Write(ConvertBRG8ToL( new byte[] { Input[IOffs + 0], Input[IOffs + 1], Input[IOffs + 2] })); } else if (PicaFormat == PICASurfaceFormat.LA8) { writer.Write(Input[IOffs + 3]); writer.Write(ConvertBRG8ToL( new byte[] { Input[IOffs + 0], Input[IOffs + 1], Input[IOffs + 2] })); } else if (PicaFormat == PICASurfaceFormat.RGB565) { ushort R = (ushort)(Convert8To5(Input[IOffs + 0])); ushort G = (ushort)(Convert8To6(Input[IOffs + 1]) << 5); ushort B = (ushort)(Convert8To5(Input[IOffs + 2]) << 11); writer.Write((ushort)(R | G | B)); } else if (PicaFormat == PICASurfaceFormat.RGBA4) { ushort R = (ushort)(Convert8To4(Input[IOffs]) << 4); ushort G = (ushort)(Convert8To4(Input[IOffs + 1]) << 8); ushort B = (ushort)(Convert8To4(Input[IOffs + 2]) << 12); ushort A = (ushort)(Convert8To4(Input[IOffs + 3])); writer.Write((ushort)(R | G | B | A)); } else if (PicaFormat == PICASurfaceFormat.RGBA5551) { ushort R = (ushort)(Convert8To5(Input[IOffs + 0]) << 1); ushort G = (ushort)(Convert8To5(Input[IOffs + 1]) << 6); ushort B = (ushort)(Convert8To5(Input[IOffs + 2]) << 11); ushort A = (ushort)(Convert8To1(Input[IOffs + 3])); writer.Write((ushort)(R | G | B | A)); } else if (PicaFormat == PICASurfaceFormat.LA4) { byte A = Input[IOffs + 3]; byte L = ConvertBRG8ToL( new byte[] { Input[IOffs + 0], Input[IOffs + 1], Input[IOffs + 2] }); writer.Write((byte)((A >> 4) | (L & 0xF0))); } else if (PicaFormat == PICASurfaceFormat.L4) { byte L1 = ConvertBRG8ToL( new byte[] { Input[IOffs + 0], Input[IOffs + 1], Input[IOffs + 2] }); byte L2 = ConvertBRG8ToL( new byte[] { Input[IOffs + 3], Input[IOffs + 4], Input[IOffs + 5] }); writer.Write((byte)((L1 >> 4) | (L2 & 0xF0))); Px++; } else if (PicaFormat == PICASurfaceFormat.A4) { //Todo this has issues byte A1 = (byte)(Input[IOffs] >> 4); byte A2 = (byte)(Input[IOffs + 3] & 0xF0); writer.Write((byte)(A1 | A2)); Px++; } else if (PicaFormat == PICASurfaceFormat.HiLo8) { writer.Write(Input[IOffs]); writer.Write(Input[IOffs + 1]); } } } } } byte[] newOutput = mem.ToArray(); // if (newOutput.Length != ImageSize) // throw new Exception($"Invalid image size! Expected {ImageSize} got {newOutput.Length}"); if (newOutput.Length > 0) return newOutput; else return new byte[CalculateLength(Width, Height, PicaFormat)]; } // Convert helpers from Citra Emulator (citra/src/common/color.h) private static byte Convert8To1(byte val) { return (byte)(val == 0 ? 0 : 1); } private static byte Convert8To4(byte val) { return (byte)(val >> 4); } private static byte Convert8To5(byte val) { return (byte)(val >> 3); } private static byte Convert8To6(byte val) { return (byte)(val >> 2); } private static byte ConvertBRG8ToL(byte[] bytes) { byte L = (byte)(bytes[0] * 0.0722f); L += (byte)(bytes[1] * 0.7152f); L += (byte)(bytes[2] * 0.2126f); return L; } private static void DecodeRGB565(byte[] Buffer, int Address, ushort Value) { int R = ((Value >> 0) & 0x1f) << 3; int G = ((Value >> 5) & 0x3f) << 2; int B = ((Value >> 11) & 0x1f) << 3; SetColor(Buffer, Address, 0xff, B | (B >> 5), G | (G >> 6), R | (R >> 5)); } private static void DecodeRGBA4(byte[] Buffer, int Address, ushort Value) { int R = (Value >> 4) & 0xf; int G = (Value >> 8) & 0xf; int B = (Value >> 12) & 0xf; SetColor(Buffer, Address, (Value & 0xf) | (Value << 4), B | (B << 4), G | (G << 4), R | (R << 4)); } private static void DecodeRGBA5551(byte[] Buffer, int Address, ushort Value) { int R = ((Value >> 1) & 0x1f) << 3; int G = ((Value >> 6) & 0x1f) << 3; int B = ((Value >> 11) & 0x1f) << 3; SetColor(Buffer, Address, (Value & 1) * 0xff, B | (B >> 5), G | (G >> 5), R | (R >> 5)); } private static void SetColor(byte[] Buffer, int Address, int A, int B, int G, int R) { Buffer[Address + 0] = (byte)B; Buffer[Address + 1] = (byte)G; Buffer[Address + 2] = (byte)R; Buffer[Address + 3] = (byte)A; } private static ushort GetUShort(byte[] Buffer, int Address) { return (ushort)( Buffer[Address + 0] << 0 | Buffer[Address + 1] << 8); } public static int CalculateLength(int Width, int Height, PICASurfaceFormat Format) { int Length = (Width * Height * FmtBPP[(int)Format]) / 8; if ((Length & 0x7f) != 0) { Length = (Length & ~0x7f) + 0x80; } return Length; } } }