/*
 * Gearsystem - Sega Master System / Game Gear Emulator
 * Copyright (C) 2013  Ignacio Sanchez

 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * any later version.

 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.

 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see http://www.gnu.org/licenses/
 *
 */

#define GUI_DEBUG_MEMORY_IMPORT
#include "gui_debug_memory.h"

#include "gearsystem.h"
#include "imgui.h"
#include "gui_debug_memeditor.h"
#include "gui_debug_constants.h"
#include "gui_filedialogs.h"
#include "config.h"
#include "gui.h"
#include "emu.h"

static MemEditor mem_edit[MEMORY_EDITOR_MAX];
static int mem_edit_select = -1;
static int current_mem_edit = 0;
static char set_value_buffer[5] = { };

static void memory_editor_menu(void);
static void draw_tabs(void);
static void draw_single_tab(int i);

void gui_debug_memory_init(void)
{
    gui_debug_memory_reset();

    for (int i = 0; i < MEMORY_EDITOR_MAX; i++)
    {
        MemEditor::Options options;
        options.bytes_per_row = config_debug.mem_editor_bytes_per_row[i];
        options.preview_data_type = config_debug.mem_editor_preview_data_type[i];
        options.preview_endianess = config_debug.mem_editor_preview_endianess[i];
        options.uppercase_hex = config_debug.mem_editor_uppercase_hex[i];
        options.gray_out_zeros = config_debug.mem_editor_gray_out_zeros[i];
        mem_edit[i].SetOptions(options);
    }
}

void gui_debug_memory_destroy(void)
{
    for (int i = 0; i < MEMORY_EDITOR_MAX; i++)
    {
        MemEditor::Options options = mem_edit[i].GetOptions();
        config_debug.mem_editor_bytes_per_row[i] = options.bytes_per_row;
        config_debug.mem_editor_preview_data_type[i] = options.preview_data_type;
        config_debug.mem_editor_preview_endianess[i] = options.preview_endianess;
        config_debug.mem_editor_uppercase_hex[i] = options.uppercase_hex;
        config_debug.mem_editor_gray_out_zeros[i] = options.gray_out_zeros;
    }
}

void gui_debug_memory_reset(void)
{
    GearsystemCore* core = emu_get_core();
    Memory* memory = core->GetMemory();
    Cartridge* cart = core->GetCartridge();
    Video* video = core->GetVideo();

    mem_edit[MEMORY_EDITOR_FIXED_1K].Reset("FIXED 1KB", memory->GetMemoryMap(), 0x400, 0);
    bool flash = (core->GetCartridge()->GetType() == Cartridge::CartridgeIratahackMapper);
    const char* slot0_label = flash ? "FLASH0" : "ROM0";
    const char* slot1_label = flash ? "FLASH1" : "ROM1";
    const char* slot2_label = flash ? "FLASH2" : "ROM2";

    mem_edit[MEMORY_EDITOR_ROM0_SEGA].Reset(slot0_label, memory->GetCurrentRule()->GetPage(0) + 0x400, 0x4000 - 0x400, 0x400);
    mem_edit[MEMORY_EDITOR_ROM0].Reset(slot0_label, memory->GetCurrentRule()->GetPage(0), 0x4000, 0);
    mem_edit[MEMORY_EDITOR_ROM1].Reset(slot1_label, memory->GetCurrentRule()->GetPage(1), 0x4000, 0x4000);
    mem_edit[MEMORY_EDITOR_ROM2_CODEMASTERS].Reset(slot2_label, memory->GetCurrentRule()->GetPage(2), 0x2000, 0x8000);
    mem_edit[MEMORY_EDITOR_EXT_RAM].Reset("EXT RAM", memory->GetCurrentRule()->GetRamBanks(), 0x2000, 0xA000);
    mem_edit[MEMORY_EDITOR_ROM2].Reset(slot2_label, memory->GetCurrentRule()->GetPage(2), 0x4000, 0x8000);
    bool sg1000 = cart->IsSG1000();
    int ram_size = sg1000 ? 0x400 : 0x4000;
    mem_edit[MEMORY_EDITOR_RAM].Reset("RAM", memory->GetMemoryMap() + 0xC000, ram_size, 0xC000);
    mem_edit[MEMORY_EDITOR_VRAM].Reset("VRAM", video->GetVRAM(), 0x4000, 0);
    mem_edit[MEMORY_EDITOR_CRAM].Reset("CRAM", video->GetCRAM(), 0x40, 0);
    mem_edit[MEMORY_EDITOR_ROM].Reset("FULL ROM", cart->GetROM(), cart->GetROMSize(), 0);

    if (IsValidPointer(memory->GetBootrom()))
        mem_edit[MEMORY_EDITOR_BIOS].Reset("BIOS", memory->GetBootrom(), memory->GetBootromSize(), 0);

    for (int i = MEMORY_EDITOR_ROM0_8K; i <= MEMORY_EDITOR_ROM5_8K; i++)
    {
        int page = i - MEMORY_EDITOR_ROM0_8K;
        char label[16];
        snprintf(label, 16, "%s%d", flash ? "FLASH" : "ROM", page);
        mem_edit[i].Reset(label, memory->GetCurrentRule()->GetPage(page), 0x2000, 0x2000 * page);
    }
}

void gui_debug_window_memory(void)
{
    for (int i = 0; i < MEMORY_EDITOR_MAX; i++)
    {
        mem_edit[i].SetGuiFont(gui_roboto_font);
        mem_edit[i].WatchPopup();
        mem_edit[i].BookMarkPopup();
    }

    ImGui::PushStyleVar(ImGuiStyleVar_WindowRounding, 8.0f);
    ImGui::SetNextWindowPos(ImVec2(60, 60), ImGuiCond_FirstUseEver);
    ImGui::SetNextWindowSize(ImVec2(670, 330), ImGuiCond_FirstUseEver);

    ImGui::Begin("Memory Editor", &config_debug.show_memory, ImGuiWindowFlags_MenuBar);

    memory_editor_menu();

    GearsystemCore* core = emu_get_core();
    Memory* memory = core->GetMemory();
    Cartridge* cart = core->GetCartridge();

    ImGui::PushFont(gui_default_font);
    ImGui::TextColored(green, "  BANKS: ");ImGui::SameLine();

    bool flash = (cart->GetType() == Cartridge::CartridgeIratahackMapper);
    const char* slot_prefix = flash ? "FLASH" : "ROM";

    if (memory->GetCurrentRule()->Has8kBanks())
    {
        for (int i = 0; i < 6; i++)
        {
            ImGui::TextColored(cyan, "%s%d", slot_prefix, i);ImGui::SameLine();
            ImGui::Text("$%02X", memory->GetCurrentRule()->GetBank(i));
            if (i != 5)
                ImGui::SameLine();
        }
    }
    else
    {
        for (int i = 0; i < 3; i++)
        {
            ImGui::TextColored(cyan, "%s%d", slot_prefix, i);ImGui::SameLine();
            ImGui::Text("$%02X", memory->GetCurrentRule()->GetBank(i));
            if (i != 2)
                ImGui::SameLine();
        }
    }

    if (cart->GetType() == Cartridge::CartridgeSegaMapper)
    {
        ImGui::SameLine();
        ImGui::TextColored(cyan, "  RAM");ImGui::SameLine();
        ImGui::Text("$%02X", memory->GetCurrentRule()->GetRamBank());
    }
    ImGui::PopFont();

    if (ImGui::BeginTabBar("##memory_tabs", ImGuiTabBarFlags_None))
    {
        draw_tabs();
        ImGui::EndTabBar();
    }

    ImGui::End();
    ImGui::PopStyleVar();
}

void gui_debug_memory_search_window(void)
{
    for (int i = 0; i < MEMORY_EDITOR_MAX; i++)
    {
        ImGui::PushFont(gui_default_font);
        mem_edit[i].DrawSearchWindow();
        ImGui::PopFont();
    }
}

void gui_debug_memory_find_bytes_window(void)
{
    for (int i = 0; i < MEMORY_EDITOR_MAX; i++)
    {
        ImGui::PushFont(gui_default_font);
        mem_edit[i].DrawFindBytesWindow();
        ImGui::PopFont();
    }
}

void gui_debug_memory_watches_window(void)
{
    for (int i = 0; i < MEMORY_EDITOR_MAX; i++)
    {
        ImGui::PushFont(gui_default_font);
        mem_edit[i].DrawWatchWindow();
        ImGui::PopFont();
    }
}

void gui_debug_memory_step_frame(void)
{
    for (int i = 0; i < MEMORY_EDITOR_MAX; i++)
    {
        mem_edit[i].StepFrame();
    }
}

void gui_debug_memory_copy(void)
{
    mem_edit[current_mem_edit].Copy();
}

void gui_debug_memory_paste(void)
{
    mem_edit[current_mem_edit].Paste();
}

void gui_debug_memory_select_all(void)
{
    mem_edit[current_mem_edit].SelectAll();
}

void gui_debug_memory_goto(int editor, int address)
{
    mem_edit_select = editor;
    mem_edit[mem_edit_select].JumpToAddress(address);
}

void gui_debug_memory_save_dump(const char* file_path, bool binary)
{
    if (binary)
        mem_edit[current_mem_edit].SaveToBinaryFile(file_path);
    else
        mem_edit[current_mem_edit].SaveToTextFile(file_path);
}

void gui_debug_memory_load_dump(const char* file_path)
{
    mem_edit[current_mem_edit].LoadFromBinaryFile(file_path);
}

static void draw_tabs(void)
{
    GearsystemCore* core = emu_get_core();
    Cartridge* cart = core->GetCartridge();
    Memory* memory = core->GetMemory();

    if (memory->GetCurrentRule()->Has8kBanks())
    {
        for (int i = MEMORY_EDITOR_ROM0_8K; i <= MEMORY_EDITOR_ROM5_8K; i++)
        {
            draw_single_tab(i);
        }
    }
    else
    {
        if ((cart->GetType() == Cartridge::CartridgeSegaMapper) || (cart->GetType() == Cartridge::CartridgeEeprom93C46Mapper))
        {
            draw_single_tab(MEMORY_EDITOR_FIXED_1K);
            draw_single_tab(MEMORY_EDITOR_ROM0_SEGA);
        }
        else
        {
            draw_single_tab(MEMORY_EDITOR_ROM0);
        }

        draw_single_tab(MEMORY_EDITOR_ROM1);

        if ((cart->GetType() == Cartridge::CartridgeCodemastersMapper) && IsValidPointer(memory->GetCurrentRule()->GetRamBanks()))
        {
            draw_single_tab(MEMORY_EDITOR_ROM2_CODEMASTERS);
            draw_single_tab(MEMORY_EDITOR_EXT_RAM);
        }
        else
        {
            draw_single_tab(MEMORY_EDITOR_ROM2);
        }
    }

    draw_single_tab(MEMORY_EDITOR_RAM);
    draw_single_tab(MEMORY_EDITOR_VRAM);
    draw_single_tab(MEMORY_EDITOR_CRAM);

    if (IsValidPointer(cart->GetROM()))
        draw_single_tab(MEMORY_EDITOR_ROM);

    if (IsValidPointer(memory->GetBootrom()))
        draw_single_tab(MEMORY_EDITOR_BIOS);
}

static void draw_single_tab(int i)
{
    if (ImGui::BeginTabItem(mem_edit[i].GetTitle(), NULL, mem_edit_select == i ? ImGuiTabItemFlags_SetSelected : ImGuiTabItemFlags_None))
    {
        ImGui::PushFont(gui_default_font);
        if (mem_edit_select == i)
            mem_edit_select = -1;
        current_mem_edit = i;
        mem_edit[i].Draw();
        ImGui::PopFont();
        ImGui::EndTabItem();
    }
}

static void memory_editor_menu(void)
{
    ImGui::BeginMenuBar();

    if (ImGui::BeginMenu("File"))
    {
        if (ImGui::MenuItem("Save Memory As Text..."))
        {
            gui_file_dialog_save_memory_dump(false);
        }

        if (ImGui::MenuItem("Save Memory As Binary..."))
        {
            gui_file_dialog_save_memory_dump(true);
        }

        if (ImGui::MenuItem("Load Memory From Binary..."))
        {
            gui_file_dialog_load_memory_dump();
        }

        ImGui::EndMenu();
    }

    if (ImGui::BeginMenu("Edit"))
    {
        if (ImGui::MenuItem("Copy", "Ctrl+C"))
        {
            gui_debug_memory_copy();
        }

        if (ImGui::MenuItem("Paste", "Ctrl+V"))
        {
            gui_debug_memory_paste();
        }

        ImGui::EndMenu();
    }

    if (ImGui::BeginMenu("Selection"))
    {
        if (ImGui::MenuItem("Select All", "Ctrl+A"))
        {
            mem_edit[current_mem_edit].SelectAll();
        }

        if (ImGui::MenuItem("Clear Selection"))
        {
            mem_edit[current_mem_edit].ClearSelection();
        }

        if (ImGui::BeginMenu("Set value"))
        {
            ImVec2 character_size = ImGui::CalcTextSize("X");
            int word_bytes = mem_edit[current_mem_edit].GetWordBytes();
            ImGui::SetNextItemWidth(((word_bytes * 2) + 1) * character_size.x);
            if (ImGui::InputTextWithHint("##set_value", word_bytes == 1 ? "XX" : "XXXX", set_value_buffer, IM_ARRAYSIZE(set_value_buffer), ImGuiInputTextFlags_AutoSelectAll | ImGuiInputTextFlags_EnterReturnsTrue | ImGuiInputTextFlags_CharsHexadecimal | ImGuiInputTextFlags_CharsUppercase))
            {
                u16 value = 0;
                if (parse_hex_string(set_value_buffer, strlen(set_value_buffer), &value))
                {
                    mem_edit[current_mem_edit].SetValueToSelection((int)value);
                    set_value_buffer[0] = 0;
                }
            }
            ImGui::SameLine();
            if (ImGui::Button("Set!", ImVec2(40, 0)))
            {
                u16 value = 0;
                if (parse_hex_string(set_value_buffer, strlen(set_value_buffer), &value))
                {
                    mem_edit[current_mem_edit].SetValueToSelection((int)value);
                    set_value_buffer[0] = 0;
                }
            }
            ImGui::EndMenu();
        }

        ImGui::EndMenu();
    }

    if (ImGui::BeginMenu("Bookmarks"))
    {
        if (ImGui::MenuItem("Add Bookmark"))
        {
            mem_edit[current_mem_edit].AddBookmark();
        }

        if (ImGui::MenuItem("Clear All"))
        {
            mem_edit[current_mem_edit].RemoveBookmarks();
        }

        std::vector<MemEditor::Bookmark>* bookmarks = mem_edit[current_mem_edit].GetBookmarks();

        if (bookmarks->size() > 0)
            ImGui::Separator();

        for (long unsigned int i = 0; i < bookmarks->size(); i++)
        {
            MemEditor::Bookmark* bookmark = &(*bookmarks)[i];

            char label[80];
            snprintf(label, 80, "$%04X: %s", bookmark->address, bookmark->name);

            if (ImGui::MenuItem(label))
            {
                mem_edit[current_mem_edit].JumpToAddress(bookmark->address);
            }
        }

        ImGui::EndMenu();
    }

    char label[64];

    if (ImGui::BeginMenu("Watches"))
    {
        snprintf(label, 64, "Open %s Watch Window", mem_edit[current_mem_edit].GetTitle());
        if (ImGui::MenuItem(label))
        {
            mem_edit[current_mem_edit].OpenWatchWindow();
        }

        snprintf(label, 64, "Add %s Watch", mem_edit[current_mem_edit].GetTitle());
        if (ImGui::MenuItem(label))
        {
            mem_edit[current_mem_edit].AddWatch();
        }

        snprintf(label, 64, "Clear All %s Watches", mem_edit[current_mem_edit].GetTitle());
        if (ImGui::MenuItem(label))
        {
            mem_edit[current_mem_edit].RemoveWatches();
        }

        ImGui::EndMenu();
    }

    if (ImGui::BeginMenu("Search"))
    {
        snprintf(label, 64, "Open %s Search Window", mem_edit[current_mem_edit].GetTitle());
        if (ImGui::MenuItem(label))
        {
            mem_edit[current_mem_edit].OpenSearchWindow();
        }

        snprintf(label, 64, "Find Bytes in %s...", mem_edit[current_mem_edit].GetTitle());
        if (ImGui::MenuItem(label))
        {
            mem_edit[current_mem_edit].OpenFindBytes();
        }

        ImGui::EndMenu();
    }

    ImGui::EndMenuBar();
}

bool gui_debug_memory_select_range(int editor, int start_address, int end_address)
{
    if (editor < 0 || editor >= MEMORY_EDITOR_MAX)
        return false;

    if (!mem_edit[editor].SetSelection(start_address, end_address))
        return false;

    mem_edit_select = editor;
    mem_edit[editor].ScrollToAddress(start_address);

    return true;
}

void gui_debug_memory_set_selection_value(int editor, u8 value)
{
    if (editor < 0 || editor >= MEMORY_EDITOR_MAX)
        return;

    mem_edit[editor].SetValueToSelection(value);
}

void gui_debug_memory_add_bookmark(int editor, int address, const char* name)
{
    if (editor < 0 || editor >= MEMORY_EDITOR_MAX)
        return;

    std::vector<MemEditor::Bookmark>* bookmarks = mem_edit[editor].GetBookmarks();
    MemEditor::Bookmark bookmark;
    bookmark.address = address;

    if (name && strlen(name) > 0)
    {
        snprintf(bookmark.name, sizeof(bookmark.name), "%s", name);
    }
    else
    {
        snprintf(bookmark.name, sizeof(bookmark.name), "Bookmark_%04X", address);
    }

    bookmarks->push_back(bookmark);
}

void gui_debug_memory_remove_bookmark(int editor, int address)
{
    if (editor < 0 || editor >= MEMORY_EDITOR_MAX)
        return;

    std::vector<MemEditor::Bookmark>* bookmarks = mem_edit[editor].GetBookmarks();

    for (std::vector<MemEditor::Bookmark>::iterator it = bookmarks->begin(); it != bookmarks->end(); ++it)
    {
        if (it->address == address)
        {
            bookmarks->erase(it);
            break;
        }
    }
}

bool gui_debug_memory_add_watch(int editor, int address, const char* notes, int size)
{
    if (editor < 0 || editor >= MEMORY_EDITOR_MAX)
        return false;

    int size_index = 0;
    switch (size)
    {
        case 8: size_index = 0; break;
        case 16: size_index = 1; break;
        case 24: size_index = 2; break;
        case 32: size_index = 3; break;
        default: size_index = 0; break;
    }

    return mem_edit[editor].AddWatchDirect(address, notes, size_index);
}

void gui_debug_memory_open_watch_popup(int editor, int address, const char* notes)
{
    if (editor < 0 || editor >= MEMORY_EDITOR_MAX)
        return;

    mem_edit[editor].PrepareAddWatch(address, notes);
}

void gui_debug_memory_remove_watch(int editor, int address)
{
    if (editor < 0 || editor >= MEMORY_EDITOR_MAX)
        return;

    std::vector<MemEditor::Watch>* watches = mem_edit[editor].GetWatches();

    for (std::vector<MemEditor::Watch>::iterator it = watches->begin(); it != watches->end(); ++it)
    {
        if (it->address == address)
        {
            watches->erase(it);
            break;
        }
    }
}

int gui_debug_memory_get_bookmarks(int editor, void** bookmarks_ptr)
{
    if (editor < 0 || editor >= MEMORY_EDITOR_MAX)
    {
        *bookmarks_ptr = NULL;
        return 0;
    }

    std::vector<MemEditor::Bookmark>* bookmarks = mem_edit[editor].GetBookmarks();
    *bookmarks_ptr = (void*)bookmarks;
    return (int)bookmarks->size();
}

int gui_debug_memory_get_watches(int editor, void** watches_ptr)
{
    if (editor < 0 || editor >= MEMORY_EDITOR_MAX)
    {
        *watches_ptr = NULL;
        return 0;
    }

    std::vector<MemEditor::Watch>* watches = mem_edit[editor].GetWatches();
    *watches_ptr = (void*)watches;
    return (int)watches->size();
}

void gui_debug_memory_get_selection(int editor, int* start, int* end)
{
    if (editor < 0 || editor >= MEMORY_EDITOR_MAX)
    {
        *start = -1;
        *end = -1;
        return;
    }

    mem_edit[editor].GetSelection(start, end);
}

void gui_debug_memory_search_capture(int editor)
{
    if (editor < 0 || editor >= MEMORY_EDITOR_MAX)
        return;

    mem_edit[editor].SearchCapture();
}

int gui_debug_memory_search(int editor, int op, int compare_type, int compare_value, int data_type, void** results_ptr)
{
    if (editor < 0 || editor >= MEMORY_EDITOR_MAX)
    {
        *results_ptr = NULL;
        return 0;
    }

    int count = mem_edit[editor].PerformSearch(op, compare_type, compare_value, data_type);
    std::vector<MemEditor::Search>* results = mem_edit[editor].GetSearchResults();
    *results_ptr = (void*)results;
    return count;
}

int gui_debug_memory_find_bytes(int editor, const char* hex_str, int* out_addresses, int max_results)
{
    if (editor < 0 || editor >= MEMORY_EDITOR_MAX)
        return 0;

    return mem_edit[editor].FindBytesSequence(hex_str, out_addresses, max_results);
}

void gui_debug_memory_save_settings(std::ostream& stream)
{
    for (int i = 0; i < MEMORY_EDITOR_MAX; i++)
    {
        mem_edit[i].SaveSettings(stream);
    }
}

void gui_debug_memory_load_settings(std::istream& stream)
{
    for (int i = 0; i < MEMORY_EDITOR_MAX; i++)
    {
        mem_edit[i].LoadSettings(stream);
    }
}
