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

#ifndef GBA_SPRITE_ENGINE_PALETTE_MANAGER_H
#define GBA_SPRITE_ENGINE_PALETTE_MANAGER_H

#include <libgba-bitmap-engine/gba/tonc_memmap.h>
#include <libgba-bitmap-engine/gba/tonc_types.h>
#include "combined_palette.h"

#define PALETTE_BANK_SIZE 16
#define PALETTE_MAX_SIZE 256

int getBits(int number, int k, int p);

class CombinedPalette;

class PaletteManager {
protected:
    const COLOR *data;
    int size;

    virtual void* paletteAddress() = 0;
    virtual PALBANK* paletteBank() = 0;
public:
    PaletteManager();
    PaletteManager(const COLOR paletteData[]) : PaletteManager(paletteData, PALETTE_MAX_SIZE) {}
    PaletteManager(const COLOR paletteData[], int size);

    CombinedPalette* operator+(const PaletteManager& other);

    void persist();
    void persistToBank(int bank);
    COLOR change(int bank, int index, COLOR newColor);
    COLOR get(int bank, int index) { return paletteBank()[bank][index]; }
    void increaseBrightness(u32 intensity);

    static COLOR modify(COLOR color, u32 intensity);
    static COLOR color(u32 r, u32 g, u32 b);
    static u32 red(COLOR r);
    static u32 green(COLOR r);
    static u32 blue(COLOR r);
};

class BackgroundPaletteManager : public PaletteManager {
protected:
    void *paletteAddress() override {
        return pal_bg_mem;
    }

    PALBANK *paletteBank() override {
        return pal_bg_bank;
    }

public:
    BackgroundPaletteManager() : PaletteManager() {}
    BackgroundPaletteManager(const COLOR paletteData[]) : PaletteManager(paletteData) {}
    BackgroundPaletteManager(const COLOR paletteData[], int size) : PaletteManager(paletteData, size) {}
};


class ForegroundPaletteManager : public PaletteManager {
protected:
    void *paletteAddress() override {
        return pal_obj_mem;
    }

    PALBANK *paletteBank() override {
        return pal_obj_bank;
    }

public:
    ForegroundPaletteManager() : PaletteManager() {}
    ForegroundPaletteManager(const COLOR paletteData[]) : PaletteManager(paletteData) {}
    ForegroundPaletteManager(const COLOR paletteData[], int size) : PaletteManager(paletteData, size) {}
};

#endif //GBA_SPRITE_ENGINE_PALETTE_MANAGER_H