//
// Created by Wouter Groeneveld on 27/07/18.
//

#ifdef CODE_COMPILED_AS_PART_OF_TEST
#include <libgba-bitmap-engine/gba/tonc_core_stub.h>
#else
#include <libgba-bitmap-engine/gba/tonc_core.h>
#endif
#include <libgba-bitmap-engine/palette/palette_manager.h>

const COLOR defaultPaletteData[PALETTE_MAX_SIZE] __attribute__((aligned(4))) = {
        0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
        0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
        0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
        0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
        0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
        0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
        0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
        0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
        0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
        0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
        0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
        0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
        0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
        0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
        0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
        0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0
};

PaletteManager::PaletteManager() : PaletteManager(defaultPaletteData, PALETTE_MAX_SIZE) {
}


int getBits(int number, int k, int p) {
    return (((1 << k) - 1) & (number >> p));
}

PaletteManager::PaletteManager(const COLOR *paletteData, int size) : data(paletteData), size(size) {

}

CombinedPalette* PaletteManager::operator+(const PaletteManager &other) {
    return new CombinedPalette(*this, const_cast<PaletteManager&>(other));
}

void PaletteManager::persist() {
    dma3_cpy(this->paletteAddress(), this->data, this->size);
}

COLOR PaletteManager::change(int bank, int index, COLOR newColor) {
    auto palBank = this->paletteBank();
    COLOR oldColor = palBank[bank][index];
    palBank[bank][index] = newColor;
    return oldColor;
}

COLOR PaletteManager::color(u32 red, u32 green, u32 blue) {
    if(red > 31) red = 31;
    if(green > 31) green = 31;
    if(blue > 31) blue = 31;
    return red | (green<<5) | (blue<<10);
}

u32 PaletteManager::red(COLOR r) {
    return getBits(r, 5, 0);
}

u32 PaletteManager::green(COLOR r) {
    return getBits(r, 5, 5);
}

u32 PaletteManager::blue(COLOR r) {
    return getBits(r, 5, 10);
}

COLOR PaletteManager::modify(COLOR color, u32 intensity) {
    return PaletteManager::color(PaletteManager::red(color) + intensity,
                                      PaletteManager::green(color) + intensity, PaletteManager::blue(color) + intensity);
}

void PaletteManager::increaseBrightness(u32 intensity) {
    for(int bank = 0; bank < PALETTE_BANK_SIZE; bank++) {
        for(int index = 0; index < PALETTE_BANK_SIZE; index++) {
            auto current = get(bank, index);
            auto next = PaletteManager::modify(current, intensity);

            change(bank, index, next);
        }
    }
}

void PaletteManager::persistToBank(int bank) {
    auto palBank = this->paletteBank();
    dma3_cpy(palBank[bank], this->data, PALETTE_BANK_SIZE);
}