Files
AssetRipper.TextureDecoder/TextureDecoder/Dxt/DxtDecoder.cs
T

218 lines
6.4 KiB
C#

using System;
using System.Runtime.CompilerServices;
namespace AssetRipper.TextureDecoder.Dxt
{
/// <summary>
///
/// </summary>
public static class DxtDecoder
{
/// <summary>
///
/// </summary>
/// <param name="input"></param>
/// <param name="width"></param>
/// <param name="height"></param>
/// <param name="output"></param>
public static void DecompressDXT1(byte[] input, int width, int height, byte[] output)
{
int offset = 0;
int bcw = (width + 3) / 4;
int bch = (height + 3) / 4;
int clen_last = (width + 3) % 4 + 1;
uint[] buffer = new uint[16];
int[] colors = new int[4];
for (int t = 0; t < bch; t++)
{
for (int s = 0; s < bcw; s++, offset += 8)
{
int r0, g0, b0, r1, g1, b1;
int q0 = input[offset + 0] | input[offset + 1] << 8;
int q1 = input[offset + 2] | input[offset + 3] << 8;
Rgb565(q0, out r0, out g0, out b0);
Rgb565(q1, out r1, out g1, out b1);
colors[0] = Color(r0, g0, b0, 255);
colors[1] = Color(r1, g1, b1, 255);
if (q0 > q1)
{
colors[2] = Color((r0 * 2 + r1) / 3, (g0 * 2 + g1) / 3, (b0 * 2 + b1) / 3, 255);
colors[3] = Color((r0 + r1 * 2) / 3, (g0 + g1 * 2) / 3, (b0 + b1 * 2) / 3, 255);
}
else
{
colors[2] = Color((r0 + r1) / 2, (g0 + g1) / 2, (b0 + b1) / 2, 255);
}
uint d = BitConverter.ToUInt32(input, offset + 4);
for (int i = 0; i < 16; i++, d >>= 2)
{
buffer[i] = unchecked((uint)colors[d & 3]);
}
int clen = (s < bcw - 1 ? 4 : clen_last) * 4;
for (int i = 0, y = t * 4; i < 4 && y < height; i++, y++)
{
Buffer.BlockCopy(buffer, i * 4 * 4, output, (y * width + s * 4) * 4, clen);
}
}
}
}
/// <summary>
///
/// </summary>
/// <param name="input"></param>
/// <param name="width"></param>
/// <param name="height"></param>
/// <param name="output"></param>
public static void DecompressDXT3(byte[] input, int width, int height, byte[] output)
{
int offset = 0;
int bcw = (width + 3) / 4;
int bch = (height + 3) / 4;
int clen_last = (width + 3) % 4 + 1;
uint[] buffer = new uint[16];
int[] colors = new int[4];
int[] alphas = new int[16];
for (int t = 0; t < bch; t++)
{
for (int s = 0; s < bcw; s++, offset += 16)
{
for (int i = 0; i < 4; i++)
{
int alpha = input[offset + i * 2] | input[offset + i * 2 + 1] << 8;
alphas[i * 4 + 0] = (((alpha >> 0) & 0xF) * 0x11) << 24;
alphas[i * 4 + 1] = (((alpha >> 4) & 0xF) * 0x11) << 24;
alphas[i * 4 + 2] = (((alpha >> 8) & 0xF) * 0x11) << 24;
alphas[i * 4 + 3] = (((alpha >> 12) & 0xF) * 0x11) << 24;
}
int r0, g0, b0, r1, g1, b1;
int q0 = input[offset + 8] | input[offset + 9] << 8;
int q1 = input[offset + 10] | input[offset + 11] << 8;
Rgb565(q0, out r0, out g0, out b0);
Rgb565(q1, out r1, out g1, out b1);
colors[0] = Color(r0, g0, b0, 0);
colors[1] = Color(r1, g1, b1, 0);
if (q0 > q1)
{
colors[2] = Color((r0 * 2 + r1) / 3, (g0 * 2 + g1) / 3, (b0 * 2 + b1) / 3, 0);
colors[3] = Color((r0 + r1 * 2) / 3, (g0 + g1 * 2) / 3, (b0 + b1 * 2) / 3, 0);
}
else
{
colors[2] = Color((r0 + r1) / 2, (g0 + g1) / 2, (b0 + b1) / 2, 0);
}
uint d = BitConverter.ToUInt32(input, offset + 12);
for (int i = 0; i < 16; i++, d >>= 2)
{
buffer[i] = unchecked((uint)(colors[d & 3] | alphas[i]));
}
int clen = (s < bcw - 1 ? 4 : clen_last) * 4;
for (int i = 0, y = t * 4; i < 4 && y < height; i++, y++)
{
Buffer.BlockCopy(buffer, i * 4 * 4, output, (y * width + s * 4) * 4, clen);
}
}
}
}
/// <summary>
///
/// </summary>
/// <param name="input"></param>
/// <param name="width"></param>
/// <param name="height"></param>
/// <param name="output"></param>
public static void DecompressDXT5(byte[] input, int width, int height, byte[] output)
{
int offset = 0;
int bcw = (width + 3) / 4;
int bch = (height + 3) / 4;
int clen_last = (width + 3) % 4 + 1;
uint[] buffer = new uint[16];
int[] colors = new int[4];
int[] alphas = new int[8];
for (int t = 0; t < bch; t++)
{
for (int s = 0; s < bcw; s++, offset += 16)
{
alphas[0] = input[offset + 0];
alphas[1] = input[offset + 1];
if (alphas[0] > alphas[1])
{
alphas[2] = (alphas[0] * 6 + alphas[1]) / 7;
alphas[3] = (alphas[0] * 5 + alphas[1] * 2) / 7;
alphas[4] = (alphas[0] * 4 + alphas[1] * 3) / 7;
alphas[5] = (alphas[0] * 3 + alphas[1] * 4) / 7;
alphas[6] = (alphas[0] * 2 + alphas[1] * 5) / 7;
alphas[7] = (alphas[0] + alphas[1] * 6) / 7;
}
else
{
alphas[2] = (alphas[0] * 4 + alphas[1]) / 5;
alphas[3] = (alphas[0] * 3 + alphas[1] * 2) / 5;
alphas[4] = (alphas[0] * 2 + alphas[1] * 3) / 5;
alphas[5] = (alphas[0] + alphas[1] * 4) / 5;
alphas[7] = 255;
}
for (int i = 0; i < 8; i++)
{
alphas[i] <<= 24;
}
int r0, g0, b0, r1, g1, b1;
int q0 = input[offset + 8] | input[offset + 9] << 8;
int q1 = input[offset + 10] | input[offset + 11] << 8;
Rgb565(q0, out r0, out g0, out b0);
Rgb565(q1, out r1, out g1, out b1);
colors[0] = Color(r0, g0, b0, 0);
colors[1] = Color(r1, g1, b1, 0);
if (q0 > q1)
{
colors[2] = Color((r0 * 2 + r1) / 3, (g0 * 2 + g1) / 3, (b0 * 2 + b1) / 3, 0);
colors[3] = Color((r0 + r1 * 2) / 3, (g0 + g1 * 2) / 3, (b0 + b1 * 2) / 3, 0);
}
else
{
colors[2] = Color((r0 + r1) / 2, (g0 + g1) / 2, (b0 + b1) / 2, 0);
}
ulong da = BitConverter.ToUInt64(input, offset) >> 16;
uint dc = BitConverter.ToUInt32(input, offset + 12);
for (int i = 0; i < 16; i++, da >>= 3, dc >>= 2)
{
buffer[i] = unchecked((uint)(alphas[da & 7] | colors[dc & 3]));
}
int clen = (s < bcw - 1 ? 4 : clen_last) * 4;
for (int i = 0, y = t * 4; i < 4 && y < height; i++, y++)
{
Buffer.BlockCopy(buffer, i * 4 * 4, output, (y * width + s * 4) * 4, clen);
}
}
}
}
[MethodImpl(MethodImplOptions.AggressiveInlining)]
private static void Rgb565(int c, out int r, out int g, out int b)
{
r = (c & 0xf800) >> 8;
g = (c & 0x07e0) >> 3;
b = (c & 0x001f) << 3;
r |= r >> 5;
g |= g >> 6;
b |= b >> 5;
}
[MethodImpl(MethodImplOptions.AggressiveInlining)]
private static int Color(int r, int g, int b, int a)
{
return r << 16 | g << 8 | b | a << 24;
}
}
}