mirror of
https://github.com/yuzu-emu/FasTC.git
synced 2025-01-09 15:05:36 +00:00
Add 2BPP decompression
This commit is contained in:
Pavel Krajcevski
parent
b3de3f2c01
commit
1b073eb4b7
|
|
@ -69,6 +69,7 @@ namespace PVRTCC {
|
||||||
// format. The width and height must be specified in order to properly
|
// format. The width and height must be specified in order to properly
|
||||||
// decompress the data.
|
// decompress the data.
|
||||||
void Decompress(const DecompressionJob &,
|
void Decompress(const DecompressionJob &,
|
||||||
|
bool bTwoBitMode = false,
|
||||||
const EWrapMode wrapMode = eWrapMode_Clamp,
|
const EWrapMode wrapMode = eWrapMode_Clamp,
|
||||||
bool bDebugImages = false);
|
bool bDebugImages = false);
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -84,7 +84,176 @@ namespace PVRTCC {
|
||||||
return x | (y << 1);
|
return x | (y << 1);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static void Decompress4BPP(const Image &imgA, const Image &imgB,
|
||||||
|
const std::vector<Block> &blocks,
|
||||||
|
uint8 *const outBuf) {
|
||||||
|
const uint32 w = imgA.GetWidth();
|
||||||
|
const uint32 h = imgA.GetHeight();
|
||||||
|
|
||||||
|
assert(imgA.GetWidth() == imgB.GetWidth());
|
||||||
|
assert(imgA.GetHeight() == imgB.GetHeight());
|
||||||
|
|
||||||
|
for(uint32 j = 0; j < h; j++) {
|
||||||
|
for(uint32 i = 0; i < w; i++) {
|
||||||
|
const uint32 blockWidth = 4;
|
||||||
|
const uint32 blockHeight = 4;
|
||||||
|
|
||||||
|
const uint32 blockIdx =
|
||||||
|
(j/blockHeight) * (w/blockWidth) + (i/blockWidth);
|
||||||
|
const Block &b = blocks[blockIdx];
|
||||||
|
|
||||||
|
const uint32 texelIndex =
|
||||||
|
(j % blockHeight) * blockWidth + (i % blockWidth);
|
||||||
|
|
||||||
|
const Pixel &pa = imgA(i, j);
|
||||||
|
const Pixel &pb = imgB(i, j);
|
||||||
|
|
||||||
|
Pixel result;
|
||||||
|
bool punchThrough = false;
|
||||||
|
uint8 lerpVal = 0;
|
||||||
|
if(b.GetModeBit()) {
|
||||||
|
const uint8 lerpVals[3] = { 8, 4, 0 };
|
||||||
|
uint8 modVal = b.GetLerpValue(texelIndex);
|
||||||
|
|
||||||
|
if(modVal >= 2) {
|
||||||
|
if(modVal == 2) {
|
||||||
|
punchThrough = true;
|
||||||
|
}
|
||||||
|
modVal -= 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
lerpVal = lerpVals[modVal];
|
||||||
|
} else {
|
||||||
|
const uint8 lerpVals[4] = { 8, 5, 3, 0 };
|
||||||
|
lerpVal = lerpVals[b.GetLerpValue(texelIndex)];
|
||||||
|
}
|
||||||
|
|
||||||
|
for(uint32 c = 0; c < 4; c++) {
|
||||||
|
uint16 va = static_cast<uint16>(pa.Component(c));
|
||||||
|
uint16 vb = static_cast<uint16>(pb.Component(c));
|
||||||
|
|
||||||
|
uint16 res = (va * (8 - lerpVal) + vb * lerpVal) / 8;
|
||||||
|
result.Component(c) = static_cast<uint8>(res);
|
||||||
|
}
|
||||||
|
|
||||||
|
if(punchThrough) {
|
||||||
|
result.A() = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
uint32 *outPixels = reinterpret_cast<uint32 *>(outBuf);
|
||||||
|
outPixels[(j * w) + i] = result.PackRGBA();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static void Decompress2BPP(const Image &imgA, const Image &imgB,
|
||||||
|
const std::vector<Block> &blocks,
|
||||||
|
uint8 *const outBuf) {
|
||||||
|
const uint32 w = imgA.GetWidth();
|
||||||
|
const uint32 h = imgA.GetHeight();
|
||||||
|
|
||||||
|
assert(w > 0);
|
||||||
|
assert(h > 0);
|
||||||
|
assert(imgA.GetWidth() == imgB.GetWidth());
|
||||||
|
assert(imgA.GetHeight() == imgB.GetHeight());
|
||||||
|
|
||||||
|
std::vector<uint8> modValues;
|
||||||
|
modValues.reserve(w * h);
|
||||||
|
|
||||||
|
const uint32 blockWidth = 8;
|
||||||
|
const uint32 blockHeight = 4;
|
||||||
|
|
||||||
|
for(uint32 j = 0; j < h; j++) {
|
||||||
|
for(uint32 i = 0; i < w; i++) {
|
||||||
|
|
||||||
|
const uint32 blockIdx =
|
||||||
|
(j/blockHeight) * (w/blockWidth) + (i/blockWidth);
|
||||||
|
const Block &b = blocks[blockIdx];
|
||||||
|
|
||||||
|
const uint32 texelIndex =
|
||||||
|
(j % blockHeight) * blockWidth + (i % blockWidth);
|
||||||
|
|
||||||
|
uint8 lerpVal = 0;
|
||||||
|
if(b.GetModeBit()) {
|
||||||
|
uint32 texelX = texelIndex % blockWidth;
|
||||||
|
uint32 texelY = texelIndex / blockWidth;
|
||||||
|
|
||||||
|
const uint8 lerpVals[4] = { 8, 5, 3, 0 };
|
||||||
|
if(((texelX ^ texelY) & 0x1) == 0) {
|
||||||
|
uint32 lerpIdx = texelY * (blockWidth / 2) + (texelX / 2);
|
||||||
|
lerpVal = lerpVals[b.Get2BPPLerpValue(lerpIdx)];
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
lerpVal = b.Get2BPPLerpValue(texelIndex);
|
||||||
|
lerpVal = lerpVal? 0 : 8;
|
||||||
|
}
|
||||||
|
modValues.push_back(lerpVal);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
assert(modValues.size() == w * h);
|
||||||
|
|
||||||
|
for(uint32 j = 0; j < h; j++) {
|
||||||
|
for(uint32 i = 0; i < w; i++) {
|
||||||
|
|
||||||
|
const uint32 blockIdx =
|
||||||
|
(j/blockHeight) * (w/blockWidth) + (i/blockWidth);
|
||||||
|
const Block &b = blocks[blockIdx];
|
||||||
|
|
||||||
|
uint8 lerpVal = 0;
|
||||||
|
#define GET_LERP_VAL(x, y) (modValues[(y) * w + (x)])
|
||||||
|
if(b.GetModeBit() && ((i ^ j) & 0x1)) {
|
||||||
|
|
||||||
|
switch(b.Get2BPPSubMode()) {
|
||||||
|
case Block::e2BPPSubMode_Horizontal:
|
||||||
|
lerpVal += GET_LERP_VAL((i + w - 1) % w, j);
|
||||||
|
lerpVal += GET_LERP_VAL((i + w + 1) % w, j);
|
||||||
|
// lerpVal = (lerpVal + 1) / 2;
|
||||||
|
lerpVal /= 2;
|
||||||
|
break;
|
||||||
|
|
||||||
|
case Block::e2BPPSubMode_Vertical:
|
||||||
|
lerpVal += GET_LERP_VAL(i, (j + h - 1) % h);
|
||||||
|
lerpVal += GET_LERP_VAL(i, (j + h + 1) % h);
|
||||||
|
// lerpVal = (lerpVal + 1) / 2;
|
||||||
|
lerpVal /= 2;
|
||||||
|
break;
|
||||||
|
|
||||||
|
default:
|
||||||
|
case Block::e2BPPSubMode_All:
|
||||||
|
lerpVal += GET_LERP_VAL(i, (j + h - 1) % h);
|
||||||
|
lerpVal += GET_LERP_VAL(i, (j + h + 1) % h);
|
||||||
|
lerpVal += GET_LERP_VAL((i + w - 1) % w, j);
|
||||||
|
lerpVal += GET_LERP_VAL((i + w + 1) % w, j);
|
||||||
|
lerpVal = (lerpVal + 1) / 4;
|
||||||
|
// lerpVal /= 4;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
lerpVal = GET_LERP_VAL(i, j);
|
||||||
|
}
|
||||||
|
#undef GET_LERP_VAL
|
||||||
|
|
||||||
|
const Pixel &pa = imgA(i, j);
|
||||||
|
const Pixel &pb = imgB(i, j);
|
||||||
|
|
||||||
|
Pixel result;
|
||||||
|
for(uint32 c = 0; c < 4; c++) {
|
||||||
|
uint16 va = static_cast<uint16>(pa.Component(c));
|
||||||
|
uint16 vb = static_cast<uint16>(pb.Component(c));
|
||||||
|
|
||||||
|
uint16 res = (va * (8 - lerpVal) + vb * lerpVal) / 8;
|
||||||
|
result.Component(c) = static_cast<uint8>(res);
|
||||||
|
}
|
||||||
|
|
||||||
|
uint32 *outPixels = reinterpret_cast<uint32 *>(outBuf);
|
||||||
|
outPixels[(j * w) + i] = result.PackRGBA();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
void Decompress(const DecompressionJob &dcj,
|
void Decompress(const DecompressionJob &dcj,
|
||||||
|
const bool bTwoBitMode,
|
||||||
const EWrapMode wrapMode,
|
const EWrapMode wrapMode,
|
||||||
bool bDebugImages) {
|
bool bDebugImages) {
|
||||||
const uint32 w = dcj.width;
|
const uint32 w = dcj.width;
|
||||||
|
|
@ -92,14 +261,15 @@ namespace PVRTCC {
|
||||||
|
|
||||||
assert(w > 0);
|
assert(w > 0);
|
||||||
assert(h > 0);
|
assert(h > 0);
|
||||||
assert(w % 4 == 0);
|
assert(bTwoBitMode || w % 4 == 0);
|
||||||
|
assert(!bTwoBitMode || w % 8 == 0);
|
||||||
assert(h % 4 == 0);
|
assert(h % 4 == 0);
|
||||||
|
|
||||||
// First, extract all of the block information...
|
// First, extract all of the block information...
|
||||||
std::vector<Block> blocks;
|
std::vector<Block> blocks;
|
||||||
blocks.reserve(w * h);
|
blocks.reserve(w * h);
|
||||||
|
|
||||||
const uint32 blocksW = w / 4;
|
const uint32 blocksW = bTwoBitMode? (w / 8) : (w / 4);
|
||||||
const uint32 blocksH = h / 4;
|
const uint32 blocksH = h / 4;
|
||||||
const uint32 blockSz = 8;
|
const uint32 blockSz = 8;
|
||||||
|
|
||||||
|
|
@ -168,63 +338,62 @@ namespace PVRTCC {
|
||||||
}
|
}
|
||||||
|
|
||||||
// Bilinearly upscale the images.
|
// Bilinearly upscale the images.
|
||||||
imgA.BilinearUpscale(2, 2, wrapMode);
|
if(bTwoBitMode) {
|
||||||
imgB.BilinearUpscale(2, 2, wrapMode);
|
imgA.BilinearUpscale(3, 2, wrapMode);
|
||||||
|
imgB.BilinearUpscale(3, 2, wrapMode);
|
||||||
|
} else {
|
||||||
|
imgA.BilinearUpscale(2, 2, wrapMode);
|
||||||
|
imgB.BilinearUpscale(2, 2, wrapMode);
|
||||||
|
}
|
||||||
|
|
||||||
|
if(bDebugImages) {
|
||||||
|
imgA.DebugOutput("RawScaledImgA");
|
||||||
|
imgB.DebugOutput("RawScaledImgB");
|
||||||
|
}
|
||||||
|
|
||||||
// Change the bitdepth to full resolution
|
// Change the bitdepth to full resolution
|
||||||
imgA.ExpandTo8888();
|
imgA.ExpandTo8888();
|
||||||
if(bDebugImages)
|
|
||||||
imgA.DebugOutput("ScaledImgA");
|
|
||||||
imgB.ExpandTo8888();
|
imgB.ExpandTo8888();
|
||||||
if(bDebugImages)
|
|
||||||
|
if(bDebugImages) {
|
||||||
|
imgA.DebugOutput("ScaledImgA");
|
||||||
imgB.DebugOutput("ScaledImgB");
|
imgB.DebugOutput("ScaledImgB");
|
||||||
|
}
|
||||||
|
|
||||||
// Pack the pixels based on their modulation into the resulting buffer
|
// Pack the pixels based on their modulation into the resulting buffer
|
||||||
// in RGBA format...
|
// in RGBA format...
|
||||||
Image modulation(h, w);
|
if(bDebugImages && !bTwoBitMode) {
|
||||||
for(uint32 j = 0; j < h; j++) {
|
Image modulation(h, w);
|
||||||
for(uint32 i = 0; i < w; i++) {
|
for(uint32 j = 0; j < h; j++) {
|
||||||
const uint32 blockIdx = (j/4) * blocksW + (i / 4);
|
for(uint32 i = 0; i < w; i++) {
|
||||||
const Block &b = blocks[blockIdx];
|
const uint32 blockWidth = bTwoBitMode? 8 : 4;
|
||||||
|
const uint32 blockHeight = 4;
|
||||||
|
|
||||||
const uint32 texelIndex = (j % 4) * 4 + (i % 4);
|
const uint32 blockIdx = (j/blockHeight) * blocksW + (i/blockWidth);
|
||||||
const Pixel &pa = imgA(i, j);
|
const Block &b = blocks[blockIdx];
|
||||||
const Pixel &pb = imgB(i, j);
|
|
||||||
|
|
||||||
Pixel result;
|
const uint32 texelIndex =
|
||||||
bool punchThrough = false;
|
(j % blockHeight) * blockWidth + (i % blockWidth);
|
||||||
uint8 lerpVal = 0;
|
|
||||||
if(b.GetModeBit()) {
|
|
||||||
const uint8 lerpVals[3] = { 8, 4, 0 };
|
|
||||||
uint8 modVal = b.GetLerpValue(texelIndex);
|
|
||||||
|
|
||||||
if(modVal >= 2) {
|
uint8 lerpVal;
|
||||||
if(modVal == 2) {
|
if(b.GetModeBit()) {
|
||||||
punchThrough = true;
|
const uint8 lerpVals[3] = { 8, 4, 0 };
|
||||||
|
uint8 modVal = b.GetLerpValue(texelIndex);
|
||||||
|
|
||||||
|
if(modVal >= 2) {
|
||||||
|
modVal -= 1;
|
||||||
}
|
}
|
||||||
modVal -= 1;
|
|
||||||
|
lerpVal = lerpVals[modVal];
|
||||||
|
} else {
|
||||||
|
const uint8 lerpVals[4] = { 8, 5, 3, 0 };
|
||||||
|
lerpVal = lerpVals[b.GetLerpValue(texelIndex)];
|
||||||
}
|
}
|
||||||
|
|
||||||
lerpVal = lerpVals[modVal];
|
|
||||||
|
|
||||||
} else {
|
|
||||||
const uint8 lerpVals[4] = { 8, 5, 3, 0 };
|
|
||||||
lerpVal = lerpVals[b.GetLerpValue(texelIndex)];
|
|
||||||
}
|
|
||||||
|
|
||||||
for(uint32 c = 0; c < 4; c++) {
|
|
||||||
uint16 va = static_cast<uint16>(pa.Component(c));
|
|
||||||
uint16 vb = static_cast<uint16>(pb.Component(c));
|
|
||||||
|
|
||||||
uint16 res = (va * (8 - lerpVal) + vb * lerpVal) / 8;
|
|
||||||
result.Component(c) = static_cast<uint8>(res);
|
|
||||||
}
|
|
||||||
|
|
||||||
if(bDebugImages) {
|
|
||||||
Pixel iv;
|
Pixel iv;
|
||||||
const uint8 modDepth[4] = { 8, 3, 3, 3 };
|
const uint8 modDepth[4] = { 8, 3, 3, 3 };
|
||||||
iv.ChangeBitDepth(modDepth);
|
iv.ChangeBitDepth(modDepth);
|
||||||
iv.A() = punchThrough? 0 : 0xFF;
|
iv.A() = 0xFF;
|
||||||
for(int i = 1; i < 4; i++) {
|
for(int i = 1; i < 4; i++) {
|
||||||
if(lerpVal == 8) {
|
if(lerpVal == 8) {
|
||||||
iv.Component(i) = 7;
|
iv.Component(i) = 7;
|
||||||
|
|
@ -234,17 +403,14 @@ namespace PVRTCC {
|
||||||
}
|
}
|
||||||
modulation(i, j) = iv;
|
modulation(i, j) = iv;
|
||||||
}
|
}
|
||||||
|
modulation.DebugOutput("DebugModulation");
|
||||||
if(punchThrough) {
|
|
||||||
result.A() = 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
uint32 *outPixels = reinterpret_cast<uint32 *>(dcj.outBuf);
|
|
||||||
outPixels[(j * w) + i] = result.PackRGBA();
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if(bDebugImages) {
|
|
||||||
modulation.DebugOutput("DebugModulation");
|
if(bTwoBitMode) {
|
||||||
|
Decompress2BPP(imgA, imgB, blocks, dcj.outBuf);
|
||||||
|
} else {
|
||||||
|
Decompress4BPP(imgA, imgB, blocks, dcj.outBuf);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue