#include <stdint.h>
#include <stdio.h>
#include "z80.h"
#include "z80dis.h"
#include "z80debug.h"
#include "zx_ula.h"
#include "zx_screen.h"
#include "zx_tape.h"
#include <SDL2/SDL.h>
#include <string.h>
#include "ui.h"
#include "ui_menu.h"
#include "z80analyze.h"
#include "ui_window.h"

uint8_t memory[65536];
uint32_t time = 0;
uint32_t t_states = 0;

namespace actions
{
    void exitMenu()
    {
        const uint8_t dt = z80::step();
        z80debug::cont();
        zxscreen::refresh(dt);
    }

    int fastload(int value)
    {
        zx_tape::toggleOption(ZXTAPE_OPTION_FAST_LOAD);
        return zx_tape::getOption(ZXTAPE_OPTION_FAST_LOAD);
    }

    int stopatend(int value)
    {
        zx_tape::toggleOption(ZXTAPE_OPTION_STOP_AT_END);
        return zx_tape::getOption(ZXTAPE_OPTION_STOP_AT_END);
    }

    int decZoom(int value)
    {
        zxscreen::decZoom();
        return 0;
    }

    int incZoom(int value)
    {
        zxscreen::incZoom();
        return 0;
    }

    int fullscreen(int value)
    {
        zxscreen::toggleFullscreen();
        return zxscreen::getFullscreen();
    }

    int showAnalyzer(int value)
    {
        z80analyze::show();
        return 0;
    }
}

int main(int argc, char *argv[])
{
    FILE* f = fopen("48.rom", "rb");
    fread(memory, 1024, 16, f);
    fclose(f);
    
    z80dis::loadSymbols();
    z80::reset(memory);
    z80::connect_port(0xfe, zx_ula::port_in, zx_ula::port_out);

    SDL_Init(SDL_INIT_EVERYTHING);
    z80debug::init();
    zxscreen::init();

    ui::menu::init();
    ui::menu::setexitcallback(actions::exitMenu);

    int menu = ui::menu::addsubmenu("FILE");
    ui::menu::addoption(menu, "LOAD TAPE", nullptr);
    ui::menu::addoption(menu, "SAVE TAPE", nullptr);
    ui::menu::addseparator(menu);
    ui::menu::addoption(menu, "LOAD STATE", nullptr);
    ui::menu::addoption(menu, "SAVE STATE", nullptr);

    menu = ui::menu::addsubmenu("TAPE");
    ui::menu::addbooloption(menu, "BERSERK MODE", zx_tape::getOption(ZXTAPE_OPTION_FAST_LOAD), actions::fastload);
    ui::menu::addbooloption(menu, "STOP AT END", zx_tape::getOption(ZXTAPE_OPTION_STOP_AT_END), actions::stopatend);

    menu = ui::menu::addsubmenu("SCREEN");
    ui::menu::addoption(menu, "DEC ZOOM", actions::decZoom);
    ui::menu::addoption(menu, "INC ZOOM", actions::incZoom);
    ui::menu::addbooloption(menu, "FULLSCREEN", zxscreen::getFullscreen(), actions::fullscreen);

    menu = ui::menu::addsubmenu("EMULATION");
    ui::menu::addbooloption(menu, "STOP ON INVALID OP", z80::getOption(Z80_OPTION_STOP_ON_INVALID), actions::decZoom);
    ui::menu::addoption(menu, "SHOW ANALYZER", actions::showAnalyzer);

    zx_ula::sound_init();

    zx_tape::load("fernandomartin.tap");

    if (argc==3) { z80debug::loadngo(argv[1], argv[2]); }

    bool should_exit = false;
    SDL_Event e;

    time = SDL_GetTicks();
    t_states = 0;

    while (!should_exit)
    {
        while (SDL_PollEvent(&e))
        {
            bool result = true;

            if (e.type == SDL_QUIT) { should_exit=true; break; }
            if (e.type == SDL_MOUSEBUTTONUP) result = ui::window::sendEvent(e.button.windowID, &e);
            if (e.type == SDL_MOUSEMOTION) result = ui::window::sendEvent(e.motion.windowID, &e);
            if (e.type == SDL_WINDOWEVENT) result = ui::window::sendEvent(e.window.windowID, &e);
            if (e.type == SDL_MOUSEWHEEL) result = ui::window::sendEvent(e.wheel.windowID, &e);
            if (e.type == SDL_KEYDOWN) result = ui::window::sendEvent(e.key.windowID, &e);
            if (e.type == SDL_TEXTINPUT) result = ui::window::sendEvent(e.text.windowID, &e);

            if (e.type == SDL_MOUSEBUTTONUP && e.button.button==1) ui::setClicked(true);

            if (!result)
                should_exit = true; break;
        }

        if (!z80debug::debugging() && !z80debug::paused()) {
            if (z80debug::isbreak(z80::getPC(), 9)) {
                z80debug::stop();
                zxscreen::redraw();
            } else {
                //if (z80::getPC()==0x05C8) zx_tape::go_berserk();
                bool fastload=false;
                if (zx_tape::getplaying() && zx_tape::getOption(ZXTAPE_OPTION_FAST_LOAD)) { fastload=true; time = SDL_GetTicks(); }
                while (zx_tape::getplaying() && zx_tape::getOption(ZXTAPE_OPTION_FAST_LOAD))
                {
//                    zx_tape::update(z80::step());
                    uint8_t dt = z80::step();
                    t_states += dt;
                    zx_tape::update(dt);
                    if (SDL_GetTicks()-time>=1000) {
                        time = SDL_GetTicks();
                        zx_tape::report();
                    }
                    //zx_ula::sound_update(dt);
                    //zxscreen::refresh(dt);
                }
                if (fastload) { printf("%i\n", SDL_GetTicks()-time); t_states=0; }
                // En cada bucle fem 10 pasos de la CPU, sino s'ofega
                for (int i=0;i<5;++i) {
                    uint8_t dt = z80::step();
                    t_states += dt;
                    zx_tape::update(dt);
                    
                    zx_ula::sound_update(dt);
                    zxscreen::refresh(dt);
                }

                if (t_states>=350000)
                {
                    //if (SDL_GetTicks()>=time+1000)
                        //printf("%i\n", SDL_GetTicks()-(time+1000));
                    //else
                    //    printf("%i\n", SDL_GetTicks()-(time+1000));
                    //t_states = 0;
                    //printf("%i: %i\n", SDL_GetTicks()-(time+1000), t_states);
                    while (SDL_GetTicks()<time+100) {}
                    t_states -= 350000;
                    time = SDL_GetTicks();
                    z80analyze::refresh();
                }
            }
        } else if (!z80debug::debugging() && z80debug::paused()) {
            zxscreen::redraw(false);
            ui::menu::show();
            zxscreen::present();
        }
        ui::setClicked(false);
    }

    return 0;
}