#include <gint/timer.h>
#include <gint/clock.h>
#include <gint/keyboard.h>
#include <math.h>

#include "core.h"
#include "mutation_engine.h"
#include "epidemic_engine.h"
#include "display_engine.h"


int next_frame(struct game *current_game, int *vaccine)
{
    for (int i = 0; current_game->planes[i]; i++)
    {
        switch(current_game->planes[i]->direction)
        {
            case 1:
                current_game->planes[i]->y -= 1;
                break;
            case 2:
                current_game->planes[i]->x += 1;
                break;
            case 3:
                current_game->planes[i]->y += 1;
                break;
            case 4:
                current_game->planes[i]->x -= 1;
                break;
        }
        if (current_game->planes[i]->x == current_game->planes[i]->dest_x && current_game->planes[i]->y == current_game->planes[i]->dest_y)
        {
            // Infect the airport
            if (current_game->planes[i]->is_infected) current_game->grid.data[current_game->planes[i]->x + current_game->planes[i]->y * current_game->grid.width] = 1;

            // Set the new destination
            current_game->planes[i]->dest_x = current_game->planes[i]->depa_x;
            current_game->planes[i]->dest_y = current_game->planes[i]->depa_y;

            // Set the new departure
            current_game->planes[i]->depa_x = current_game->planes[i]->x;
            current_game->planes[i]->depa_y = current_game->planes[i]->y;

            // Set the new direction
            int new_dir = (current_game->planes[i]->direction + 2) % 4;
            if (!new_dir) new_dir = 4;
            current_game->planes[i]->direction = new_dir;

            // Infect the plane
            if (current_game->grid.data[current_game->planes[i]->x + current_game->planes[i]->y * current_game->grid.width] == 1  && current_game->mutations_selected[2] == 4) current_game->planes[i]->is_infected = 1;
        }
    }

    int limit_tick = LIMIT_TICK;
    if (current_game->boost) limit_tick = floor(LIMIT_TICK / 10);


    current_game->time += ENGINE_TICK;
    current_game->total_time += ENGINE_TICK;

    if (current_game->time > limit_tick)
    {
        // Reset internal clock
        current_game->time = 0;

        // Display message on research
        if (!current_game->research && current_game->priority) message("LA RECHERCHE CONTRE VOTRE VIRUS        COMMENCE !"); 
        else if (!*vaccine && (current_game->research == current_game->limit)) {*vaccine = 1; message("LE VACCIN EST TERMINE."); }

        // Update the game
        current_game->dna = current_game->dna + 1 + floor(current_game->severity / 10);
        if (current_game->dna > 30) current_game->dna = 30;

        current_game->research += current_game->priority;
        if (current_game->research > current_game->limit) current_game->research = current_game->limit;

        epidemic_simulation(current_game);

        // Check the end of the game
        if (!current_game->humans[1])
        {
            
            if (current_game->humans[3] < 4 * (current_game->humans[0] + current_game->humans[2])) message("VOUS AVEZ    PERDU.");
            else message("VOUS AVEZ GAGNE !");
            return 0;
        }
    }
    return 1;
}


int get_inputs(const int background, int *mutation_menu, int *boost)
{
    int key = rtc_key();

    if (key == KEY_ARROW) *boost = (*boost + 1) % 2;
    if (key == KEY_VARS)
    {
        *mutation_menu = 4;
        return 2;
    }

    if (key == KEY_ALPHA)
    {
        if (background == 4) return 2;
        else return 1;
    }
    if (key == KEY_EXIT && background == 1) return -1;

    if (background == 2)
    {
        switch (key)
        {
            // Symptoms
            case KEY_LEFT:
                *mutation_menu = *mutation_menu - 1;
                if (*mutation_menu <= 0) *mutation_menu = 4;
                break;

            // Abilities
            case KEY_RIGHT:
                *mutation_menu = *mutation_menu + 1;
                if (*mutation_menu >= 5) *mutation_menu = 1;
                break;

            // Validation
            case KEY_SHIFT:
                if (*mutation_menu == 4) return 1;
                else return 4;
                break;
        }
    }

    return background;
}


int rtc_key(void)
{
    int opt = GETKEY_DEFAULT & ~GETKEY_MOD_SHIFT & ~GETKEY_MOD_ALPHA & ~GETKEY_REP_ARROWS;
    int timeout = 1;
        
    key_event_t ev = getkey_opt(opt, &timeout);
    if(ev.type == KEYEV_NONE) return 0;
        
    return ev.key;
}


int callback_tick(volatile int *tick)
{
    *tick = 1;
    return TIMER_CONTINUE;
}


void message(char *msg)
{
    int opt = GETKEY_DEFAULT & ~GETKEY_MOD_SHIFT & ~GETKEY_MOD_ALPHA & ~GETKEY_REP_ARROWS;
    key_event_t ev = {0};

    display_message(msg);
    while (ev.key != KEY_ALPHA) ev = getkey_opt(opt, NULL);
}