mirror of
https://github.com/KillzXGaming/Switch-Toolbox
synced 2024-11-23 13:03:15 +00:00
324 lines
11 KiB
C#
324 lines
11 KiB
C#
using System;
|
|
|
|
// https://github.com/KFreon/CSharpImageLibrary
|
|
namespace CSharpImageLibrary.DDS
|
|
{
|
|
public static class Dxt
|
|
{
|
|
public static byte[] DecompressDxt1(byte[] Data, int Width, int Height)
|
|
{
|
|
var image = new byte[Height * Width * 4];
|
|
var widthBlocks = Width / 4;
|
|
var heightBlocks = Height / 4;
|
|
int pos = 0;
|
|
|
|
for (int y = 0; y < heightBlocks; y++)
|
|
{
|
|
for (int x = 0; x < widthBlocks; x++)
|
|
{
|
|
DecompressDxt1Block(Data, (int)Width, (int)Height, pos, image, x * 4, y * 4);
|
|
pos += 8;
|
|
}
|
|
}
|
|
|
|
return image;
|
|
}
|
|
|
|
public static void DecompressDxt1Block(byte[] Data,int Width, int Height, int pos, byte[] output, int xPos, int yPos)
|
|
{
|
|
if (pos >= Data.Length) return;
|
|
Color[] c = new Color[4];
|
|
|
|
var color0 = BitConverter.ToUInt16(Data, pos);
|
|
var color1 = BitConverter.ToUInt16(Data, pos + 2);
|
|
var mask = BitConverter.ToUInt32(Data, pos + 4);
|
|
|
|
Color(color0, ref c[0]);
|
|
Color(color1, ref c[1]);
|
|
c[0].A = 255;
|
|
c[1].A = 255;
|
|
|
|
if (color0 > color1)
|
|
{
|
|
c[2].R = (byte)((2 * c[0].R + c[1].R) / 3);
|
|
c[2].G = (byte)((2 * c[0].G + c[1].G) / 3);
|
|
c[2].B = (byte)((2 * c[0].B + c[1].B) / 3);
|
|
c[2].A = 255;
|
|
|
|
c[3].R = (byte)((c[0].R + 2 * c[1].R) / 3);
|
|
c[3].G = (byte)((c[0].G + 2 * c[1].G) / 3);
|
|
c[3].B = (byte)((c[0].B + 2 * c[1].B) / 3);
|
|
c[3].A = 255;
|
|
}
|
|
else
|
|
{
|
|
c[2].R = (byte)((c[0].R + c[1].R) / 2);
|
|
c[2].G = (byte)((c[0].G + c[1].G) / 2);
|
|
c[2].B = (byte)((c[0].B + c[1].B) / 2);
|
|
c[2].A = 255;
|
|
|
|
c[3].R = 0;
|
|
c[3].G = 0;
|
|
c[3].B = 0;
|
|
c[3].A = 0;
|
|
}
|
|
|
|
for (int y = 0; y < 4; y++)
|
|
{
|
|
for (int x = 0; x < 4; x++)
|
|
{
|
|
var index = mask & 3;
|
|
mask >>= 2;
|
|
var col = c[index];
|
|
|
|
var x2 = xPos + x;
|
|
var y2 = yPos + y;
|
|
|
|
var offset = (x2 + y2 * Width) * 4;
|
|
output[offset] = col.B;
|
|
output[offset + 1] = col.G;
|
|
output[offset + 2] = col.R;
|
|
output[offset + 3] = col.A;
|
|
}
|
|
}
|
|
}
|
|
|
|
public static byte[] DecompressDxt4(byte[] Data, int Width, int Height)
|
|
{
|
|
var image = new byte[Height * Width * 4];
|
|
var widthBlocks = Width / 4;
|
|
var heightBlocks = Height / 4;
|
|
int pos = 0;
|
|
|
|
for (int y = 0; y < heightBlocks; y++)
|
|
{
|
|
for (int x = 0; x < widthBlocks; x++)
|
|
{
|
|
DecompressDxt4Block(Data, Width, Height, pos, image, x * 4, y * 4);
|
|
pos += 8;
|
|
}
|
|
}
|
|
|
|
return image;
|
|
}
|
|
|
|
public static void DecompressDxt4Block(byte[] Data, int Width, int Height, int pos, byte[] output, int xPos, int yPos)
|
|
{
|
|
if (pos >= Data.Length) return;
|
|
byte[] alpha = new byte[8];
|
|
|
|
var alpha0 = Data[pos];
|
|
var alpha1 = Data[pos + 1];
|
|
var alphaMask = BitConverter.ToUInt64(Data, pos) >> 16;
|
|
|
|
alpha[0] = alpha0;
|
|
alpha[1] = alpha1;
|
|
if (alpha[0] > alpha[1])
|
|
{
|
|
alpha[2] = (byte)((6 * alpha[0] + 1 * alpha[1] + 3) / 7);
|
|
alpha[3] = (byte)((5 * alpha[0] + 2 * alpha[1] + 3) / 7);
|
|
alpha[4] = (byte)((4 * alpha[0] + 3 * alpha[1] + 3) / 7);
|
|
alpha[5] = (byte)((3 * alpha[0] + 4 * alpha[1] + 3) / 7);
|
|
alpha[6] = (byte)((2 * alpha[0] + 5 * alpha[1] + 3) / 7);
|
|
alpha[7] = (byte)((1 * alpha[0] + 6 * alpha[1] + 3) / 7);
|
|
}
|
|
else
|
|
{
|
|
alpha[2] = (byte)((4 * alpha[0] + 1 * alpha[1] + 2) / 5);
|
|
alpha[3] = (byte)((3 * alpha[0] + 2 * alpha[1] + 2) / 5);
|
|
alpha[4] = (byte)((2 * alpha[0] + 3 * alpha[1] + 2) / 5);
|
|
alpha[5] = (byte)((1 * alpha[0] + 4 * alpha[1] + 2) / 5);
|
|
alpha[6] = 0;
|
|
alpha[7] = 255;
|
|
}
|
|
|
|
for (int y = 0; y < 4; y++)
|
|
{
|
|
for (int x = 0; x < 4; x++)
|
|
{
|
|
var alphaIndex = alphaMask & 7;
|
|
alphaMask >>= 3;
|
|
|
|
var x2 = xPos + x;
|
|
var y2 = yPos + y;
|
|
|
|
var offset = (x2 + y2 * Width) * 4;
|
|
output[offset] = alpha[alphaIndex];
|
|
output[offset + 1] = alpha[alphaIndex];
|
|
output[offset + 2] = alpha[alphaIndex];
|
|
output[offset + 3] = 0xFF;
|
|
}
|
|
}
|
|
}
|
|
|
|
public static byte[] DecompressDxt5(byte[] Data, int Width, int Height)
|
|
{
|
|
var image = new byte[Height * Width * 4];
|
|
var widthBlocks = Width / 4;
|
|
var heightBlocks = Height / 4;
|
|
int pos = 0;
|
|
|
|
for (int y = 0; y < heightBlocks; y++)
|
|
{
|
|
for (int x = 0; x < widthBlocks; x++)
|
|
{
|
|
DecompressDxt5Block(Data, Width, Height, pos, image, x * 4, y * 4);
|
|
pos += 16;
|
|
}
|
|
}
|
|
|
|
return image;
|
|
}
|
|
|
|
public static void DecompressDxt5Block(byte[] Data, int Width, int Height, int pos, byte[] output, int xPos, int yPos)
|
|
{
|
|
if (pos >= Data.Length) return;
|
|
Color[] c = new Color[4];
|
|
byte[] alpha = new byte[8];
|
|
|
|
var alpha0 = Data[pos];
|
|
var alpha1 = Data[pos + 1];
|
|
var alphaMask = BitConverter.ToUInt64(Data, pos) >> 16;
|
|
|
|
var color0 = BitConverter.ToUInt16(Data, pos + 8);
|
|
var color1 = BitConverter.ToUInt16(Data, pos + 10);
|
|
var mask = BitConverter.ToUInt32(Data, pos + 12);
|
|
|
|
alpha[0] = alpha0;
|
|
alpha[1] = alpha1;
|
|
if (alpha[0] > alpha[1])
|
|
{
|
|
alpha[2] = (byte)((6 * alpha[0] + 1 * alpha[1] + 3) / 7);
|
|
alpha[3] = (byte)((5 * alpha[0] + 2 * alpha[1] + 3) / 7);
|
|
alpha[4] = (byte)((4 * alpha[0] + 3 * alpha[1] + 3) / 7);
|
|
alpha[5] = (byte)((3 * alpha[0] + 4 * alpha[1] + 3) / 7);
|
|
alpha[6] = (byte)((2 * alpha[0] + 5 * alpha[1] + 3) / 7);
|
|
alpha[7] = (byte)((1 * alpha[0] + 6 * alpha[1] + 3) / 7);
|
|
}
|
|
else
|
|
{
|
|
alpha[2] = (byte)((4 * alpha[0] + 1 * alpha[1] + 2) / 5);
|
|
alpha[3] = (byte)((3 * alpha[0] + 2 * alpha[1] + 2) / 5);
|
|
alpha[4] = (byte)((2 * alpha[0] + 3 * alpha[1] + 2) / 5);
|
|
alpha[5] = (byte)((1 * alpha[0] + 4 * alpha[1] + 2) / 5);
|
|
alpha[6] = 0;
|
|
alpha[7] = 255;
|
|
}
|
|
|
|
Color(color0, ref c[0]);
|
|
Color(color1, ref c[1]);
|
|
|
|
c[2].R = (byte)((2 * c[0].R + c[1].R) / 3);
|
|
c[2].G = (byte)((2 * c[0].G + c[1].G) / 3);
|
|
c[2].B = (byte)((2 * c[0].B + c[1].B) / 3);
|
|
|
|
c[3].R = (byte)((c[0].R + 2 * c[1].R) / 3);
|
|
c[3].G = (byte)((c[0].G + 2 * c[1].G) / 3);
|
|
c[3].B = (byte)((c[0].B + 2 * c[1].B) / 3);
|
|
|
|
for (int y = 0; y < 4; y++)
|
|
{
|
|
for (int x = 0; x < 4; x++)
|
|
{
|
|
var index = mask & 3;
|
|
mask >>= 2;
|
|
var alphaIndex = alphaMask & 7;
|
|
alphaMask >>= 3;
|
|
|
|
var col = c[index];
|
|
|
|
var x2 = xPos + x;
|
|
var y2 = yPos + y;
|
|
|
|
var offset = (x2 + y2 * Width) * 4;
|
|
output[offset] = col.B;
|
|
output[offset + 1] = col.G;
|
|
output[offset + 2] = col.R;
|
|
output[offset + 3] = alpha[alphaIndex];
|
|
}
|
|
}
|
|
}
|
|
|
|
static byte ExpandTo255(double v)
|
|
{
|
|
if (double.IsNaN(v) || v == 0)
|
|
return 128;
|
|
else
|
|
return (byte)(((v + 1d) / 2d) * 255d);
|
|
}
|
|
|
|
internal static int GetDecompressedOffset(int start, int lineLength, int pixelIndex)
|
|
{
|
|
return start + (lineLength * (pixelIndex / 4)) + (pixelIndex % 4) * 4;
|
|
}
|
|
|
|
public static byte[] DecompressBc7(byte[] Data, int Width, int Height)
|
|
{
|
|
var image = new byte [Height * Width * 4];
|
|
var widthBlocks = Width / 4;
|
|
var heightBlocks = Height / 4;
|
|
int pos = 0;
|
|
|
|
for (int y = 0; y < heightBlocks; y++)
|
|
{
|
|
for (int x = 0; x < widthBlocks; x++)
|
|
{
|
|
DecompressBC7Block(Data, Width, Height, pos, image, x * 4, y * 4);
|
|
pos += 16;
|
|
}
|
|
}
|
|
|
|
return image;
|
|
}
|
|
|
|
internal static void DecompressBC7Block(byte[] Data, int Width, int Height, int pos, byte[] output, int xPos, int yPos)
|
|
{
|
|
var colours = BC7.DecompressBC7(Data, pos);
|
|
BC7.SetColoursFromDX10(colours, output, xPos, yPos, Width);
|
|
}
|
|
|
|
public static byte[] DecompressBc6(byte[] Data, int Width, int Height, bool IsSigned)
|
|
{
|
|
var image = new byte[Height * Width * 4];
|
|
var widthBlocks = Width / 4;
|
|
var heightBlocks = Height / 4;
|
|
int pos = 0;
|
|
|
|
for (int y = 0; y < heightBlocks; y++)
|
|
{
|
|
for (int x = 0; x < widthBlocks; x++)
|
|
{
|
|
DecompressBC6Block(Data, Width, Height, IsSigned, pos, image, x * 4, y * 4);
|
|
pos += 16;
|
|
}
|
|
}
|
|
|
|
return image;
|
|
}
|
|
|
|
internal static void DecompressBC6Block(byte[] Data, int Width, int Height, bool IsSigned, int pos, byte[] output, int xPos, int yPos)
|
|
{
|
|
var colours = BC6.DecompressBC6(Data, pos, IsSigned);
|
|
BC7.SetColoursFromDX10(colours, output, xPos, yPos, Width);
|
|
}
|
|
|
|
public static void Color(ushort color, ref Color c)
|
|
{
|
|
c.R = (byte)((color >> 11) & 0x1f);
|
|
c.G = (byte)((color >> 5) & 0x3f);
|
|
c.B = (byte)(color & 0x1f);
|
|
|
|
c.R = (byte)((c.R << 3) | (c.R >> 2));
|
|
c.G = (byte)((c.G << 2) | (c.G >> 4));
|
|
c.B = (byte)((c.B << 3) | (c.B >> 2));
|
|
}
|
|
}
|
|
|
|
public struct Color
|
|
{
|
|
public byte R;
|
|
public byte G;
|
|
public byte B;
|
|
public byte A;
|
|
}
|
|
}
|