#include <gint/display.h>
#include <gint/display-cg.h>
#include <math.h>

#include "mutation_engine.h"
#include "display_engine.h"

void display_background(const int background)
{
    extern const bopti_image_t img_map;
    extern const bopti_image_t img_selectedmenu;
    extern const bopti_image_t img_infos;
    extern const bopti_image_t img_grid;
    extern const bopti_image_t img_message;

    switch(background) {
        case 1:
            dimage(0, 0, &img_map);
            break;
        case 2:
            dimage(0, 0, &img_selectedmenu);
            break;
        case 3:
            dimage(0, 0, &img_infos);
            break;
        case 4:
            dimage(0, 0, &img_grid);
            break;
        case 5:
            dimage(0, 0, &img_message);
            break;
    }
}


void display_foreground(const int background, const struct game *current_game, const int mutation_menu)
{
    extern const bopti_image_t img_mutations;
    extern const bopti_image_t img_planes;
    extern const uint8_t world[184][396];


    extern const 
    GUNUSED int length;

    switch (background)
    {
        case 1:
            // Disease propagation
            for (int i = 0; i < current_game->grid.width; i ++)
            {
                for (int j = 0; j < current_game->grid.height; j ++)
                {
                    if (current_game->grid.data[i + j * current_game->grid.width] == 1 && world[j][i] != 0) dpixel(i, j, C_RED);
                    if (current_game->grid.data[i + j * current_game->grid.width] == 3 && world[j][i] != 0) dpixel(i, j, C_BLACK);
                    if (current_game->grid.data[i + j * current_game->grid.width] == 2 && world[j][i] != 0) dpixel(i, j, C_RGB(0,255,255));
                }
            }

            // Planes animations
            for (int i = 0; current_game->planes[i]; i++)
            {
                if (current_game->planes[i]->y + 8 < 184) dsubimage(current_game->planes[i]->x - 4, current_game->planes[i]->y - 4, &img_planes, 0, 8 * (current_game->planes[i]->direction - 1), 8, 8, DIMAGE_NONE);
            }

            // Research bar 
            int length = 77 * current_game->research / current_game->limit;
            dprint(30, 200, C_WHITE, "%d", current_game->dna);

            drect(86, 200, 86 + length, 203, C_RGB(0,255,255));

            // Stats bar
            for (int i = 0; i < 4; i ++)
            {
                int color = C_BLACK;
                switch (i)
                {
                    case 0:
                        color = C_GREEN;
                        break;
                    case 1:
                        color = C_RED;
                        break;
                    case 2:
                        color = C_RGB(0,255,255);
                        break;
                    case 3:
                        color = C_BLACK;
                        break;
                }

                length = 166 * current_game->humans[i] / ((current_game->grid.width * current_game->grid.height) - BLANK_CASES);
                dline(224, i*6 + 200, 224 + length, i*6 + 200, color);
                dline(224, i*6 + 201, 224 + length, i*6 + 201, color);
                dline(224, i*6 + 202, 224 + length, i*6 + 202, color);
                dline(224, i*6 + 203, 224 + length, i*6 + 203, color);
            }

            // Display if boost is activated
            if (current_game->boost) dprint(0, 0, C_WHITE, "+");

            break;

        case 2:
            drect(15 + 99 * (mutation_menu - 1), 29, 82 + 99 * (mutation_menu - 1), 30, C_WHITE);

            dprint(155, 123, C_BLACK, "ADN : %d", current_game->dna);
            
            length = 247 * current_game->contagion / 100;
            drect(122, 160, 122+length, 165, C_RGB(252, 0, 252));
            
            length = 247 * current_game->severity / 100;
            drect(122, 180, 122+length, 185, C_RGB(246, 246, 0));
            
            length = 247 * current_game->lethality / 100;
            drect(122, 200, 122+length, 205, C_RGB(140, 0, 160));

            if (current_game->mutations_selected[0]) dsubimage(78, 67, &img_mutations, 0, 44 * (current_game->mutations_selected[0] - 1), 44, 44, 0);
            if (current_game->mutations_selected[1]) dsubimage(176, 67, &img_mutations, 44, 44 * (current_game->mutations_selected[1] - 1), 44, 44, 0);
            if (current_game->mutations_selected[2]) dsubimage(273, 67, &img_mutations, 88, 44 * (current_game->mutations_selected[2] - 1), 44, 44, 0);
            break;
    }
}


void display_mutation(const int table[4][8], const struct cursor c, const int mutation_menu)
{
    extern const bopti_image_t img_mutations;
    extern bopti_image_t img_cursor;

    dclear(C_WHITE);
    display_background(4);
    switch(mutation_menu) {
        case 1:
            dprint(138,14,C_WHITE, "SYMPTOMES");
            break;
        case 2:
            dprint(138,14,C_WHITE, "CAPACITES");
            break;
        case 3:
            dprint(130,14,C_WHITE, "DIFFUSIONS");
            break;
    }
    
    for (int i = 0 ; i < 4 ; i++)
    {
        for (int j = 0 ; j < 8; j++)
        {
            if (table[i][j]) dsubimage(j*44+22, i*44+46, &img_mutations, 44 * (mutation_menu - 1), 44 * (table[i][j] - 1), 44, 44, DIMAGE_NONE);
        }
    }
    dsubimage((44 * c.x)+22, (44 * c.y)+46, &img_cursor, 0, 44 * (c.display), 44, 44, DIMAGE_NONE);
    dupdate();

}


void display_mutation_buy(const struct cursor c, const int mutation_menu, const int table[4][8], const int button_selected, const struct game *current_game)
{
    extern const bopti_image_t img_mutations;
    const int id = table[c.y][c.x];

    struct mutation *mutation_data = get_mutation_data(mutation_menu, id);
    struct mutation *mutation_sel = get_mutation_data(mutation_menu, current_game->mutations_selected[mutation_menu - 1]);

    dclear(C_WHITE);
    
    display_background(3);
    dsubimage(28, 91, &img_mutations, 44 * (mutation_menu - 1), 44 * (id - 1), 44, 44, DIMAGE_NONE);

    dprint(105, 85, C_BLACK, "NOM : %s", mutation_data->name);
    if (!current_game->mutations_bought[mutation_menu - 1][id - 1]) dprint(150, 180, C_RGB(123, 8, 8), "COUT ADN : %d", mutation_data->dna);
    else dprint(195, 180, C_RGB(123, 8, 8), "ACHETEE");
    dprint(105, 105, C_BLACK, "CONTAGION : %d (%d)", mutation_data->contagion, mutation_data->contagion - mutation_sel->contagion);
    dprint(105, 125, C_BLACK, "SEVERITE  : %d (%d)", mutation_data->severity, mutation_data->severity - mutation_sel->severity);
    dprint(105, 145, C_BLACK, "LETALITE  : %d (%d)", mutation_data->lethality, mutation_data->lethality - mutation_sel->lethality);

    if (button_selected) {
        drect(294, 8, 298, 26, C_WHITE);
        drect(294, 30, 391, 26, C_WHITE);
    }
    else {
        drect(294, 36, 298, 51, C_WHITE);
        drect(294, 55, 391, 51, C_WHITE);
    }

    dupdate();
}


void display_mutation_description(const char *name, const char *description, const int mutation_menu, const int id)
{
    extern const bopti_image_t img_mutations;

    int decalage = 0;

    dclear(C_WHITE);

    display_background(3);
    dsubimage(28, 91, &img_mutations, 44 * (mutation_menu - 1), 44 * (id - 1), 44, 44, DIMAGE_NONE);
    dprint(105, 85, C_BLACK, "NOM : %s", name);
    
    for (int i = 0; i < 4; i ++)
    {
        dtext_opt(105, 115 + i * 20, C_BLACK, C_WHITE, 0 ,0, description + decalage, 19);
        
        int offset = 0;
        while (description[decalage + offset] != '\0') offset += 1;

        if (!offset) break;
        else if (offset > 19) decalage += 19;
        else decalage += offset;
    }
    dupdate();
}


void display_message(char *msg)
{
    int decalage = 0;

    dclear(C_WHITE);
    display_background(5);
    for (int i = 0; i < 5; i ++)
    {
        dtext_opt(190, 20 * i + 30, C_BLACK, C_WHITE, 0, 0, msg + decalage, 13);
        
        int offset = 0;
        while (msg[decalage + offset] != '\0') offset += 1;

        if (!offset) break;
        else if (offset > 13) decalage += 13;
        else decalage += offset;
    }
    dupdate();
}