#ifndef BNG_H_
#define BNG_H_

#define BNG_MAGIC 0x21474E42

#if defined(__GNUC__) || defined(__clang__)
#  define PACKED __attribute__((packed))
#else
#  warning "Packed attributes for struct is not implemented for this compiler. This may result in a program working incorrectly. Feel free to fix that and submit a Pull Request to https://github.com/tsoding/bng"
#  define PACKED
#endif

struct Bng_Pixel_Format
{
    uint8_t red_byte;           // 0
    uint8_t green_byte;         // 1
    uint8_t blue_byte;          // 2
    uint8_t alpha_byte;         // 3
} PACKED;

int bng_pixel_format_equals(struct Bng_Pixel_Format a, struct Bng_Pixel_Format b)
{
    return a.red_byte == b.red_byte &&
        a.green_byte == b.green_byte &&
        a.blue_byte == b.blue_byte &&
        a.alpha_byte == b.alpha_byte;
}

const struct Bng_Pixel_Format RGBA = {0, 1, 2, 3};
const struct Bng_Pixel_Format GRBA = {1, 0, 2, 3};
const struct Bng_Pixel_Format BGRA = {2, 1, 0, 3};
const struct Bng_Pixel_Format GBRA = {1, 2, 0, 3};
const struct Bng_Pixel_Format BRGA = {2, 0, 1, 3};
const struct Bng_Pixel_Format RBGA = {0, 2, 1, 3};
const struct Bng_Pixel_Format ABGR = {3, 2, 1, 0};
const struct Bng_Pixel_Format BAGR = {2, 3, 1, 0};
const struct Bng_Pixel_Format BGAR = {2, 1, 3, 0};
const struct Bng_Pixel_Format AGBR = {3, 1, 2, 0};
const struct Bng_Pixel_Format GABR = {1, 3, 2, 0};
const struct Bng_Pixel_Format GBAR = {1, 2, 3, 0};
const struct Bng_Pixel_Format ARGB = {3, 0, 1, 2};
const struct Bng_Pixel_Format RAGB = {0, 3, 1, 2};
const struct Bng_Pixel_Format RGAB = {0, 1, 3, 2};
const struct Bng_Pixel_Format AGRB = {3, 1, 0, 2};
const struct Bng_Pixel_Format GARB = {1, 3, 0, 2};
const struct Bng_Pixel_Format GRAB = {1, 0, 3, 2};
const struct Bng_Pixel_Format ARBG = {3, 0, 2, 1};
const struct Bng_Pixel_Format RABG = {0, 3, 2, 1};
const struct Bng_Pixel_Format RBAG = {0, 2, 3, 1};
const struct Bng_Pixel_Format ABRG = {3, 2, 0, 1};
const struct Bng_Pixel_Format BARG = {2, 3, 0, 1};
const struct Bng_Pixel_Format BRAG = {2, 0, 3, 1};

struct Bng_Pixel_Format_Name
{
    char cstr[5];
};

struct Bng_Pixel_Format_Name name_of_pixel_format(struct Bng_Pixel_Format format)
{
    struct Bng_Pixel_Format_Name name = {0};
    // BTW: if sizeof(Bng_Pixel_Format) was 1 I would not have to do these asserts, because then
    // the values of byte indices could not be physically equal or greater than 4.
    assert(format.red_byte < 4);
    name.cstr[format.red_byte] = 'R';
    assert(format.green_byte < 4);
    name.cstr[format.green_byte] = 'G';
    assert(format.blue_byte < 4);
    name.cstr[format.blue_byte] = 'B';
    assert(format.alpha_byte < 4);
    name.cstr[format.alpha_byte] = 'A';
    return name;
}

struct Bng_Pixel_Format pixel_format_by_name(const char *name)
{
    struct Bng_Pixel_Format result = {0};
    size_t n = strlen(name);
    if (n > 4) n = 4;

    for (size_t i = 0; i < n; ++i) {
        switch (name[i]) {
        case 'R':
            result.red_byte = i;
            break;
        case 'G':
            result.green_byte = i;
            break;
        case 'B':
            result.blue_byte = i;
            break;
        case 'A':
            result.alpha_byte = i;
            break;
        default: {}
        }
    }

    return result;
}

struct RLE_Pair
{
    uint32_t count;
    uint32_t pixel;
};

struct Bng
{
    uint32_t magic;
    uint32_t width;
    uint32_t height;
    struct Bng_Pixel_Format pixel_format;
    uint32_t pairs_count;
    struct RLE_Pair pairs[];
} PACKED;

// NOTE: You have to allocate `width * height` amount of pixels
void decompress_pixels(struct RLE_Pair *pairs, uint32_t pairs_count, uint32_t *pixels)
{
    size_t pixels_count = 0;
    for (uint32_t i = 0; i < pairs_count; ++i) {
        for (uint32_t j = 0; j < pairs[i].count; ++j) {
            pixels[pixels_count++] = pairs[i].pixel;
        }
    }
}

// NOTE: You have to allocate `width * height` amount of pairs
uint32_t compress_pixels(uint32_t width, uint32_t height, uint32_t *pixels, struct RLE_Pair *pairs)
{
    uint32_t pairs_count = 0;
    const uint32_t n = width * height;

    for (uint32_t i = 0; i < n; ++i) {
        if (pairs_count > 0 && pairs[pairs_count - 1].pixel == pixels[i]) {
            pairs[pairs_count - 1].count += 1;
        } else {
            pairs_count += 1;
            pairs[pairs_count - 1].count = 1;
            pairs[pairs_count - 1].pixel = pixels[i];
        }
    }

    return pairs_count;
}

#define GET_BYTE(bytes, index) ((bytes & (0xFF << ((4 - index - 1) * 8))) >> ((4 - index - 1) * 8))
#define SET_BYTE(bytes, index, byte) bytes = (bytes | (byte << ((4 - index - 1) * 8)))

uint32_t convert_pixel(uint32_t pixel,
                       struct Bng_Pixel_Format source,
                       struct Bng_Pixel_Format desired)
{
    uint32_t result = 0;
    SET_BYTE(result, desired.red_byte,   GET_BYTE(pixel, source.red_byte));
    SET_BYTE(result, desired.green_byte, GET_BYTE(pixel, source.green_byte));
    SET_BYTE(result, desired.blue_byte,  GET_BYTE(pixel, source.blue_byte));
    SET_BYTE(result, desired.alpha_byte, GET_BYTE(pixel, source.alpha_byte));
    return result;
}

#endif  // BNG_H_