549 lines
18 KiB
C#
549 lines
18 KiB
C#
using System;
|
|
using System.Diagnostics;
|
|
using System.Drawing;
|
|
using Switch_Toolbox.Library;
|
|
|
|
namespace Switch_Toolbox.Library
|
|
{
|
|
public class DDSCompressor
|
|
{
|
|
//Huge thanks to gdkchan and AbooodXD for the method of decomp BC5/BC4.
|
|
|
|
//Todo. Add these to DDS code and add in methods to compress and decode more formats
|
|
//BC7 also needs to be decompressed properly since OpenTK can't decompress those
|
|
|
|
//BC4 actually breaks a bit with artifacts so i'll need to go back and fix
|
|
|
|
private static byte[] BCnDecodeTile(byte[] Input, int Offset, bool IsBC1)
|
|
{
|
|
Color[] CLUT = new Color[4];
|
|
|
|
int c0 = Get16(Input, Offset + 0);
|
|
int c1 = Get16(Input, Offset + 2);
|
|
|
|
CLUT[0] = DecodeRGB565(c0);
|
|
CLUT[1] = DecodeRGB565(c1);
|
|
CLUT[2] = CalculateCLUT2(CLUT[0], CLUT[1], c0, c1, IsBC1);
|
|
CLUT[3] = CalculateCLUT3(CLUT[0], CLUT[1], c0, c1, IsBC1);
|
|
|
|
int Indices = Get32(Input, Offset + 4);
|
|
|
|
int IdxShift = 0;
|
|
|
|
byte[] Output = new byte[4 * 4 * 4];
|
|
|
|
int OOffset = 0;
|
|
|
|
for (int TY = 0; TY < 4; TY++)
|
|
{
|
|
for (int TX = 0; TX < 4; TX++)
|
|
{
|
|
int Idx = (Indices >> IdxShift) & 3;
|
|
|
|
IdxShift += 2;
|
|
|
|
Color Pixel = CLUT[Idx];
|
|
|
|
Output[OOffset + 0] = Pixel.B;
|
|
Output[OOffset + 1] = Pixel.G;
|
|
Output[OOffset + 2] = Pixel.R;
|
|
Output[OOffset + 3] = Pixel.A;
|
|
|
|
OOffset += 4;
|
|
}
|
|
}
|
|
return Output;
|
|
}
|
|
private static Color DecodeRGB565(int Value)
|
|
{
|
|
int B = ((Value >> 0) & 0x1f) << 3;
|
|
int G = ((Value >> 5) & 0x3f) << 2;
|
|
int R = ((Value >> 11) & 0x1f) << 3;
|
|
|
|
return Color.FromArgb(
|
|
R | (R >> 5),
|
|
G | (G >> 6),
|
|
B | (B >> 5));
|
|
}
|
|
private static Color CalculateCLUT2(Color C0, Color C1, int c0, int c1, bool IsBC1)
|
|
{
|
|
if (c0 > c1 || !IsBC1)
|
|
{
|
|
return Color.FromArgb(
|
|
(2 * C0.R + C1.R) / 3,
|
|
(2 * C0.G + C1.G) / 3,
|
|
(2 * C0.B + C1.B) / 3);
|
|
}
|
|
else
|
|
{
|
|
return Color.FromArgb(
|
|
(C0.R + C1.R) / 2,
|
|
(C0.G + C1.G) / 2,
|
|
(C0.B + C1.B) / 2);
|
|
}
|
|
}
|
|
private static Color CalculateCLUT3(Color C0, Color C1, int c0, int c1, bool IsBC1)
|
|
{
|
|
if (c0 > c1 || !IsBC1)
|
|
{
|
|
return
|
|
Color.FromArgb(
|
|
(2 * C1.R + C0.R) / 3,
|
|
(2 * C1.G + C0.G) / 3,
|
|
(2 * C1.B + C0.B) / 3);
|
|
}
|
|
|
|
return Color.Transparent;
|
|
}
|
|
public static Bitmap DecompressBC1(Byte[] data, int width, int height, bool IsSRGB)
|
|
{
|
|
int W = (width + 3) / 4;
|
|
int H = (height + 3) / 4;
|
|
|
|
byte[] Output = new byte[W * H * 64];
|
|
|
|
for (int Y = 0; Y < H; Y++)
|
|
{
|
|
for (int X = 0; X < W; X++)
|
|
{
|
|
int IOffs = (Y * W + X) * 8;
|
|
|
|
byte[] Tile = BCnDecodeTile(data, IOffs, true);
|
|
|
|
int TOffset = 0;
|
|
|
|
for (int TY = 0; TY < 4; TY++)
|
|
{
|
|
for (int TX = 0; TX < 4; TX++)
|
|
{
|
|
int OOffset = (X * 4 + TX + (Y * 4 + TY) * W * 4) * 4;
|
|
|
|
Output[OOffset + 0] = Tile[TOffset + 0];
|
|
Output[OOffset + 1] = Tile[TOffset + 1];
|
|
Output[OOffset + 2] = Tile[TOffset + 2];
|
|
Output[OOffset + 3] = Tile[TOffset + 3];
|
|
|
|
TOffset += 4;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return BitmapExtension.GetBitmap(Output, W * 4, H * 4);
|
|
}
|
|
public static Bitmap DecompressBC3(Byte[] data, int width, int height, bool IsSRGB)
|
|
{
|
|
int W = (width + 3) / 4;
|
|
int H = (height + 3) / 4;
|
|
|
|
byte[] Output = new byte[W * H * 64];
|
|
|
|
for (int Y = 0; Y < H; Y++)
|
|
{
|
|
for (int X = 0; X < W; X++)
|
|
{
|
|
int IOffs = (Y * W + X) * 16;
|
|
|
|
byte[] Tile = BCnDecodeTile(data, IOffs + 8, false);
|
|
|
|
byte[] Alpha = new byte[8];
|
|
|
|
Alpha[0] = data[IOffs + 0];
|
|
Alpha[1] = data[IOffs + 1];
|
|
|
|
CalculateBC3Alpha(Alpha);
|
|
|
|
int AlphaLow = Get32(data, IOffs + 2);
|
|
int AlphaHigh = Get16(data, IOffs + 6);
|
|
|
|
ulong AlphaCh = (uint)AlphaLow | (ulong)AlphaHigh << 32;
|
|
|
|
int TOffset = 0;
|
|
|
|
for (int TY = 0; TY < 4; TY++)
|
|
{
|
|
for (int TX = 0; TX < 4; TX++)
|
|
{
|
|
int OOffset = (X * 4 + TX + (Y * 4 + TY) * W * 4) * 4;
|
|
|
|
byte AlphaPx = Alpha[(AlphaCh >> (TY * 12 + TX * 3)) & 7];
|
|
|
|
Output[OOffset + 0] = Tile[TOffset + 0];
|
|
Output[OOffset + 1] = Tile[TOffset + 1];
|
|
Output[OOffset + 2] = Tile[TOffset + 2];
|
|
Output[OOffset + 3] = AlphaPx;
|
|
|
|
TOffset += 4;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return BitmapExtension.GetBitmap(Output, W * 4, H * 4);
|
|
}
|
|
|
|
public static Bitmap DecompressBC4(Byte[] data, int width, int height, bool IsSNORM)
|
|
{
|
|
int W = (width + 3) / 4;
|
|
int H = (height + 3) / 4;
|
|
|
|
byte[] Output = new byte[W * H * 64];
|
|
|
|
for (int Y = 0; Y < H; Y++)
|
|
{
|
|
for (int X = 0; X < W; X++)
|
|
{
|
|
int IOffs = (Y * W + X) * 8;
|
|
|
|
byte[] Red = new byte[8];
|
|
|
|
Red[0] = data[IOffs + 0];
|
|
Red[1] = data[IOffs + 1];
|
|
|
|
CalculateBC3Alpha(Red);
|
|
|
|
int RedLow = Get32(data, IOffs + 2);
|
|
int RedHigh = Get16(data, IOffs + 6);
|
|
|
|
ulong RedCh = (uint)RedLow | (ulong)RedHigh << 32;
|
|
|
|
int TOffset = 0;
|
|
int TW = Math.Min(width - X * 4, 4);
|
|
int TH = Math.Min(height - Y * 4, 4);
|
|
|
|
for (int TY = 0; TY < 4; TY++)
|
|
{
|
|
for (int TX = 0; TX < 4; TX++)
|
|
{
|
|
int OOffset = (X * 4 + TX + (Y * 4 + TY) * W * 4) * 4;
|
|
|
|
byte RedPx = Red[(RedCh >> (TY * 12 + TX * 3)) & 7];
|
|
|
|
Output[OOffset + 0] = RedPx;
|
|
Output[OOffset + 1] = RedPx;
|
|
Output[OOffset + 2] = RedPx;
|
|
Output[OOffset + 3] = 255;
|
|
|
|
TOffset += 4;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return BitmapExtension.GetBitmap(Output, W * 4, H * 4);
|
|
}
|
|
public static byte[] DecompressBC5(Byte[] data, int width, int height, bool IsSNORM, bool IsByteArray)
|
|
{
|
|
int W = (width + 3) / 4;
|
|
int H = (height + 3) / 4;
|
|
|
|
|
|
byte[] Output = new byte[W * H * 64];
|
|
|
|
for (int Y = 0; Y < H; Y++)
|
|
{
|
|
for (int X = 0; X < W; X++)
|
|
|
|
{
|
|
int IOffs = (Y * W + X) * 16;
|
|
byte[] Red = new byte[8];
|
|
byte[] Green = new byte[8];
|
|
|
|
Red[0] = data[IOffs + 0];
|
|
Red[1] = data[IOffs + 1];
|
|
|
|
Green[0] = data[IOffs + 8];
|
|
Green[1] = data[IOffs + 9];
|
|
|
|
if (IsSNORM == true)
|
|
{
|
|
CalculateBC3AlphaS(Red);
|
|
CalculateBC3AlphaS(Green);
|
|
}
|
|
else
|
|
{
|
|
CalculateBC3Alpha(Red);
|
|
CalculateBC3Alpha(Green);
|
|
}
|
|
|
|
int RedLow = Get32(data, IOffs + 2);
|
|
int RedHigh = Get16(data, IOffs + 6);
|
|
|
|
int GreenLow = Get32(data, IOffs + 10);
|
|
int GreenHigh = Get16(data, IOffs + 14);
|
|
|
|
ulong RedCh = (uint)RedLow | (ulong)RedHigh << 32;
|
|
ulong GreenCh = (uint)GreenLow | (ulong)GreenHigh << 32;
|
|
|
|
int TW = Math.Min(width - X * 4, 4);
|
|
int TH = Math.Min(height - Y * 4, 4);
|
|
|
|
|
|
if (IsSNORM == true)
|
|
{
|
|
for (int TY = 0; TY < TH; TY++)
|
|
{
|
|
for (int TX = 0; TX < TW; TX++)
|
|
{
|
|
|
|
int Shift = TY * 12 + TX * 3;
|
|
int OOffset = ((Y * 4 + TY) * width + (X * 4 + TX)) * 4;
|
|
|
|
byte RedPx = Red[(RedCh >> Shift) & 7];
|
|
byte GreenPx = Green[(GreenCh >> Shift) & 7];
|
|
|
|
if (IsSNORM == true)
|
|
{
|
|
RedPx += 0x80;
|
|
GreenPx += 0x80;
|
|
}
|
|
|
|
float NX = (RedPx / 255f) * 2 - 1;
|
|
float NY = (GreenPx / 255f) * 2 - 1;
|
|
float NZ = (float)Math.Sqrt(1 - (NX * NX + NY * NY));
|
|
|
|
Output[OOffset + 0] = Clamp((NX + 1) * 0.5f);
|
|
Output[OOffset + 1] = Clamp((NY + 1) * 0.5f);
|
|
Output[OOffset + 2] = Clamp((NZ + 1) * 0.5f);
|
|
Output[OOffset + 3] = 0xff;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
for (int TY = 0; TY < TH; TY++)
|
|
{
|
|
for (int TX = 0; TX < TW; TX++)
|
|
{
|
|
|
|
int Shift = TY * 12 + TX * 3;
|
|
int OOffset = ((Y * 4 + TY) * width + (X * 4 + TX)) * 4;
|
|
|
|
byte RedPx = Red[(RedCh >> Shift) & 7];
|
|
byte GreenPx = Green[(GreenCh >> Shift) & 7];
|
|
|
|
Output[OOffset + 0] = RedPx;
|
|
Output[OOffset + 1] = GreenPx;
|
|
Output[OOffset + 2] = 255;
|
|
Output[OOffset + 3] = 255;
|
|
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return Output;
|
|
}
|
|
public static Bitmap DecompressBC5(Byte[] data, int width, int height, bool IsSNORM)
|
|
{
|
|
int W = (width + 3) / 4;
|
|
int H = (height + 3) / 4;
|
|
|
|
|
|
byte[] Output = new byte[W * H * 64];
|
|
|
|
for (int Y = 0; Y < H; Y++)
|
|
{
|
|
for (int X = 0; X < W; X++)
|
|
|
|
{
|
|
int IOffs = (Y * W + X) * 16;
|
|
byte[] Red = new byte[8];
|
|
byte[] Green = new byte[8];
|
|
|
|
Red[0] = data[IOffs + 0];
|
|
Red[1] = data[IOffs + 1];
|
|
|
|
Green[0] = data[IOffs + 8];
|
|
Green[1] = data[IOffs + 9];
|
|
|
|
if (IsSNORM == true)
|
|
{
|
|
CalculateBC3AlphaS(Red);
|
|
CalculateBC3AlphaS(Green);
|
|
}
|
|
else
|
|
{
|
|
CalculateBC3Alpha(Red);
|
|
CalculateBC3Alpha(Green);
|
|
}
|
|
|
|
int RedLow = Get32(data, IOffs + 2);
|
|
int RedHigh = Get16(data, IOffs + 6);
|
|
|
|
int GreenLow = Get32(data, IOffs + 10);
|
|
int GreenHigh = Get16(data, IOffs + 14);
|
|
|
|
ulong RedCh = (uint)RedLow | (ulong)RedHigh << 32;
|
|
ulong GreenCh = (uint)GreenLow | (ulong)GreenHigh << 32;
|
|
|
|
int TW = Math.Min(width - X * 4, 4);
|
|
int TH = Math.Min(height - Y * 4, 4);
|
|
|
|
|
|
if (IsSNORM == true)
|
|
{
|
|
for (int TY = 0; TY < TH; TY++)
|
|
{
|
|
for (int TX = 0; TX < TW; TX++)
|
|
{
|
|
|
|
int Shift = TY * 12 + TX * 3;
|
|
int OOffset = ((Y * 4 + TY) * width + (X * 4 + TX)) * 4;
|
|
|
|
byte RedPx = Red[(RedCh >> Shift) & 7];
|
|
byte GreenPx = Green[(GreenCh >> Shift) & 7];
|
|
|
|
if (IsSNORM == true)
|
|
{
|
|
RedPx += 0x80;
|
|
GreenPx += 0x80;
|
|
}
|
|
|
|
float NX = (RedPx / 255f) * 2 - 1;
|
|
float NY = (GreenPx / 255f) * 2 - 1;
|
|
float NZ = (float)Math.Sqrt(1 - (NX * NX + NY * NY));
|
|
|
|
Output[OOffset + 0] = Clamp((NZ + 1) * 0.5f);
|
|
Output[OOffset + 1] = Clamp((NY + 1) * 0.5f);
|
|
Output[OOffset + 2] = Clamp((NX + 1) * 0.5f);
|
|
Output[OOffset + 3] = 0xff;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
for (int TY = 0; TY < TH; TY++)
|
|
{
|
|
for (int TX = 0; TX < TW; TX++)
|
|
{
|
|
|
|
int Shift = TY * 12 + TX * 3;
|
|
int OOffset = ((Y * 4 + TY) * width + (X * 4 + TX)) * 4;
|
|
|
|
byte RedPx = Red[(RedCh >> Shift) & 7];
|
|
byte GreenPx = Green[(GreenCh >> Shift) & 7];
|
|
|
|
Output[OOffset + 0] = 255;
|
|
Output[OOffset + 1] = GreenPx;
|
|
Output[OOffset + 2] = RedPx;
|
|
Output[OOffset + 3] = 255;
|
|
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return BitmapExtension.GetBitmap(Output, W * 4, H * 4);
|
|
}
|
|
|
|
|
|
public static byte[] DecompressBlock(Byte[] data, int width, int height, DDS.DXGI_FORMAT format)
|
|
{
|
|
return DirectXTex.ImageCompressor.Decompress(data, width, height, (int)format);
|
|
}
|
|
public static byte[] CompressBlock(Byte[] data, int width, int height, DDS.DXGI_FORMAT format)
|
|
{
|
|
return DirectXTex.ImageCompressor.Compress(data, width, height, (int)format);
|
|
}
|
|
public static byte[] EncodePixelBlock(Byte[] data, int width, int height, DDS.DXGI_FORMAT format)
|
|
{
|
|
return DirectXTex.ImageConverter.Convert(data, width, height,(int)DDS.DXGI_FORMAT.DXGI_FORMAT_R8G8B8A8_UNORM, (int)format);
|
|
}
|
|
public static byte[] DecodePixelBlock(Byte[] data, int width, int height, DDS.DXGI_FORMAT format)
|
|
{
|
|
return DirectXTex.ImageConverter.Convert(data, width, height, (int)format, (int)DDS.DXGI_FORMAT.DXGI_FORMAT_R8G8B8A8_UNORM);
|
|
}
|
|
|
|
public static Bitmap DecompressBlock(Byte[] data, int width, int height, DDS.DXGI_FORMAT format, bool GetBitmap)
|
|
{
|
|
return BitmapExtension.GetBitmap(DirectXTex.ImageCompressor.Decompress(data, width, height, (int)format), width, height);
|
|
}
|
|
|
|
public static int Get16(byte[] Data, int Address)
|
|
{
|
|
return
|
|
Data[Address + 0] << 0 |
|
|
Data[Address + 1] << 8;
|
|
}
|
|
|
|
public static int Get32(byte[] Data, int Address)
|
|
{
|
|
return
|
|
Data[Address + 0] << 0 |
|
|
Data[Address + 1] << 8 |
|
|
Data[Address + 2] << 16 |
|
|
Data[Address + 3] << 24;
|
|
}
|
|
|
|
private static byte Clamp(float Value)
|
|
{
|
|
if (Value > 1)
|
|
{
|
|
return 0xff;
|
|
}
|
|
else if (Value < 0)
|
|
{
|
|
return 0;
|
|
}
|
|
else
|
|
{
|
|
return (byte)(Value * 0xff);
|
|
}
|
|
}
|
|
|
|
private static void CalculateBC3Alpha(byte[] Alpha)
|
|
{
|
|
for (int i = 2; i < 8; i++)
|
|
{
|
|
if (Alpha[0] > Alpha[1])
|
|
{
|
|
Alpha[i] = (byte)(((8 - i) * Alpha[0] + (i - 1) * Alpha[1]) / 7);
|
|
}
|
|
else if (i < 6)
|
|
{
|
|
Alpha[i] = (byte)(((6 - i) * Alpha[0] + (i - 1) * Alpha[1]) / 7);
|
|
}
|
|
else if (i == 6)
|
|
{
|
|
Alpha[i] = 0;
|
|
}
|
|
else /* i == 7 */
|
|
{
|
|
Alpha[i] = 0xff;
|
|
}
|
|
}
|
|
}
|
|
private static void CalculateBC3AlphaS(byte[] Alpha)
|
|
{
|
|
for (int i = 2; i < 8; i++)
|
|
{
|
|
if ((sbyte)Alpha[0] > (sbyte)Alpha[1])
|
|
{
|
|
Alpha[i] = (byte)(((8 - i) * (sbyte)Alpha[0] + (i - 1) * (sbyte)Alpha[1]) / 7);
|
|
}
|
|
else if (i < 6)
|
|
{
|
|
Alpha[i] = (byte)(((6 - i) * (sbyte)Alpha[0] + (i - 1) * (sbyte)Alpha[1]) / 7);
|
|
}
|
|
else if (i == 6)
|
|
{
|
|
Alpha[i] = 0x80;
|
|
}
|
|
else /* i == 7 */
|
|
{
|
|
Alpha[i] = 0x7f;
|
|
}
|
|
}
|
|
}
|
|
|
|
public static byte[] DecodeBC7(int X, int Y, int block)
|
|
{
|
|
byte[] result = null;
|
|
|
|
//Alright so BC7 decompression as multple modes
|
|
|
|
return result;
|
|
}
|
|
}
|
|
}
|