// engine/dxt —— DDS(DXT1/3/5) + 未压缩 BGRA8 → RGBA8 软解。M1 T3。
#include "dxt.h"
#include <cstdio>
#include <cstring>

namespace mtgodot {
namespace {

inline uint32_t rd_le32(const uint8_t* p) {
    return uint32_t(p[0]) | (uint32_t(p[1]) << 8) | (uint32_t(p[2]) << 16) | (uint32_t(p[3]) << 24);
}

// 5:6:5 → r,g,b (0..255)
inline void unpack565(uint16_t c, int& r, int& g, int& b) {
    r = ((c >> 11) & 0x1f); r = (r << 3) | (r >> 2);
    g = ((c >> 5) & 0x3f);  g = (g << 2) | (g >> 4);
    b = (c & 0x1f);         b = (b << 3) | (b >> 2);
}

// 解一个 DXT1 颜色块（8B）→ 16 像素 RGB（不写 alpha；dxt1_alpha=true 时按 1-bit alpha 写）
void decode_color_block(const uint8_t* blk, uint8_t out[16][4], bool dxt1_alpha) {
    uint16_t c0 = uint16_t(blk[0] | (blk[1] << 8));
    uint16_t c1 = uint16_t(blk[2] | (blk[3] << 8));
    int r[4], g[4], b[4], a[4] = {255, 255, 255, 255};
    unpack565(c0, r[0], g[0], b[0]);
    unpack565(c1, r[1], g[1], b[1]);
    if (c0 > c1 || !dxt1_alpha) {
        r[2] = (2 * r[0] + r[1]) / 3; g[2] = (2 * g[0] + g[1]) / 3; b[2] = (2 * b[0] + b[1]) / 3;
        r[3] = (r[0] + 2 * r[1]) / 3; g[3] = (g[0] + 2 * g[1]) / 3; b[3] = (b[0] + 2 * b[1]) / 3;
    } else {
        r[2] = (r[0] + r[1]) / 2; g[2] = (g[0] + g[1]) / 2; b[2] = (b[0] + b[1]) / 2;
        r[3] = g[3] = b[3] = 0; a[3] = 0;   // 透明
    }
    uint32_t bits = rd_le32(blk + 4);
    for (int i = 0; i < 16; ++i) {
        int idx = (bits >> (i * 2)) & 3;
        out[i][0] = uint8_t(r[idx]); out[i][1] = uint8_t(g[idx]);
        out[i][2] = uint8_t(b[idx]); out[i][3] = uint8_t(a[idx]);
    }
}

// DXT3：alpha 块（8B）= 16 个 4-bit alpha，直接展开
void decode_dxt3_alpha(const uint8_t* blk, uint8_t out[16][4]) {
    for (int i = 0; i < 8; ++i) {
        int a0 = blk[i] & 0x0f, a1 = (blk[i] >> 4) & 0x0f;
        out[i * 2 + 0][3] = uint8_t(a0 * 17);   // 0..15 → 0..255
        out[i * 2 + 1][3] = uint8_t(a1 * 17);
    }
}

// DXT5：alpha 块（8B）= 2 端点 + 16×3-bit 索引
void decode_dxt5_alpha(const uint8_t* blk, uint8_t out[16][4]) {
    int a0 = blk[0], a1 = blk[1];
    int a[8];
    a[0] = a0; a[1] = a1;
    if (a0 > a1) {
        for (int i = 1; i < 7; ++i) a[i + 1] = ((7 - i) * a0 + i * a1) / 7;
    } else {
        for (int i = 1; i < 5; ++i) a[i + 1] = ((5 - i) * a0 + i * a1) / 5;
        a[6] = 0; a[7] = 255;
    }
    uint64_t bits = 0;
    for (int i = 0; i < 6; ++i) bits |= uint64_t(blk[2 + i]) << (8 * i);
    for (int i = 0; i < 16; ++i) {
        int idx = int((bits >> (i * 3)) & 7);
        out[i][3] = uint8_t(a[idx]);
    }
}

enum Fmt { F_NONE, F_DXT1, F_DXT3, F_DXT5, F_RGB };

bool valid_mask(uint32_t mask, uint32_t bits) {
    if (!mask || (bits < 32 && (mask >> bits))) return false;
    while (!(mask & 1)) mask >>= 1;
    return (mask & (mask + 1)) == 0; // contiguous channel bits
}

uint8_t channel(uint32_t pixel, uint32_t mask) {
    if (!mask) return 255;
    while (!(mask & 1)) { mask >>= 1; pixel >>= 1; }
    return uint8_t((uint64_t(pixel & mask) * 255 + mask / 2) / mask);
}

Image decode(const uint8_t* d, size_t len) {
    Image img;
    if (len < 128 || std::memcmp(d, "DDS ", 4) != 0) return img;
    uint32_t hsize = rd_le32(d + 4);
    if (hsize != 124) return img;
    uint32_t h = rd_le32(d + 12);
    uint32_t w = rd_le32(d + 16);
    uint32_t pf_flags = rd_le32(d + 80);
    const uint8_t* fourcc = d + 84;
    uint32_t rgb_bitcount = rd_le32(d + 88);

    Fmt fmt = F_NONE;
    if (pf_flags & 0x4) {   // DDPF_FOURCC
        if (!std::memcmp(fourcc, "DXT1", 4)) fmt = F_DXT1;
        else if (!std::memcmp(fourcc, "DXT3", 4)) fmt = F_DXT3;
        else if (!std::memcmp(fourcc, "DXT5", 4)) fmt = F_DXT5;
    } else if ((pf_flags & 0x40) && (rgb_bitcount == 16 || rgb_bitcount == 24 || rgb_bitcount == 32)) {
        fmt = F_RGB;
    }
    if (fmt == F_NONE || w == 0 || h == 0 || w > 8192 || h > 8192) return img;

    const uint8_t* src = d + 128;
    size_t avail = len - 128;
    img.w = uint16_t(w);
    img.h = uint16_t(h);
    img.rgba.assign(size_t(w) * h * 4, 0);

    if (fmt == F_RGB) {
        const uint32_t r = rd_le32(d + 92), g = rd_le32(d + 96), b = rd_le32(d + 100);
        const uint32_t a = (pf_flags & 1) ? rd_le32(d + 104) : 0;
        if (!valid_mask(r, rgb_bitcount) || !valid_mask(g, rgb_bitcount) || !valid_mask(b, rgb_bitcount) ||
            (a && !valid_mask(a, rgb_bitcount)) || ((pf_flags & 1) && !a) ||
            (r & g) || (r & b) || (g & b) || (a & (r | g | b))) return {};
        const size_t bytes = rgb_bitcount / 8, row_bytes = size_t(w) * bytes;
        const size_t pitch = (rd_le32(d + 8) & 8) ? rd_le32(d + 20) : row_bytes;
        if (pitch < row_bytes || avail < row_bytes || pitch > avail || (h - 1) > (avail - row_bytes) / pitch) return {};
        for (size_t y = 0; y < h; ++y) {
            for (size_t x = 0; x < w; ++x) {
                uint32_t pixel = 0;
                for (size_t k = 0; k < bytes; ++k) pixel |= uint32_t(src[y * pitch + x * bytes + k]) << (8 * k);
                auto* out = &img.rgba[(y * w + x) * 4];
                out[0] = channel(pixel, r); out[1] = channel(pixel, g);
                out[2] = channel(pixel, b); out[3] = channel(pixel, a);
            }
        }
        img.format = rgb_bitcount == 32 && r == 0xff0000 && b == 0xff ? "BGRA8" : "RGB_MASKED";
        return img;
    }

    const int block_bytes = (fmt == F_DXT1) ? 8 : 16;
    const size_t bx = (w + 3) / 4, by = (h + 3) / 4;
    if (avail < bx * by * block_bytes) { img = Image{}; return img; }

    for (size_t byi = 0; byi < by; ++byi) {
        for (size_t bxi = 0; bxi < bx; ++bxi) {
            const uint8_t* blk = src + (byi * bx + bxi) * block_bytes;
            uint8_t px[16][4];
            if (fmt == F_DXT1) {
                decode_color_block(blk, px, /*dxt1_alpha=*/true);
            } else {
                decode_color_block(blk + 8, px, /*dxt1_alpha=*/false);
                if (fmt == F_DXT3) decode_dxt3_alpha(blk, px);
                else               decode_dxt5_alpha(blk, px);
            }
            for (int py = 0; py < 4; ++py) {
                size_t y = byi * 4 + py;
                if (y >= h) break;
                for (int pxx = 0; pxx < 4; ++pxx) {
                    size_t x = bxi * 4 + pxx;
                    if (x >= w) break;
                    uint8_t* o = &img.rgba[(y * w + x) * 4];
                    const uint8_t* s = px[py * 4 + pxx];
                    o[0] = s[0]; o[1] = s[1]; o[2] = s[2]; o[3] = s[3];
                }
            }
        }
    }
    img.format = (fmt == F_DXT1) ? "DXT1" : (fmt == F_DXT3) ? "DXT3" : "DXT5";
    return img;
}

} // namespace

Image load_dds(const uint8_t* bytes, size_t len) { return decode(bytes, len); }

Image load_dds_path(const char* path) {
    FILE* f = std::fopen(path, "rb");
    if (!f) return {};
    std::fseek(f, 0, SEEK_END);
    long n = std::ftell(f);
    std::fseek(f, 0, SEEK_SET);
    std::vector<uint8_t> buf(n > 0 ? size_t(n) : 0);
    size_t rd = buf.empty() ? 0 : std::fread(buf.data(), 1, buf.size(), f);
    std::fclose(f);
    if (rd != buf.size()) return {};
    return decode(buf.data(), buf.size());
}

} // namespace mtgodot