#include <backends/imgui_impl_sdl2.h>
#include <backends/imgui_impl_sdlrenderer2.h>
#include <iostream>
#include "MainWindow.hpp"
#include "gui_exceptions.hpp"
#include "../common/ougge_format.hpp"
#include "../resources/fonts.hpp"
#include "../resources/textures.hpp"
#include "../common/math.hpp"

namespace ougge::gui {

f32 MainWindow::getDPI(){
    i32 w=0, h=0;
    SDL_GetRendererOutputSize(sdl_renderer, &w, &h);
    i32 sim_w=0, sim_h=0;
    SDL_GetWindowSize(sdl_window, &sim_w, &sim_h);
    f32 wdpi=(f32)w / sim_w;
    f32 hdpi=(f32)h / sim_h;
    f32 dpi=SDL_sqrtf(wdpi*wdpi + hdpi*hdpi);
    return dpi;
}

void MainWindow::open(const std::string& window_title){
    SDL_TRY(SDL_Init(SDL_INIT_EVERYTHING));
    SDL_version v;
    SDL_GetVersion(&v);
    std::cout<<ougge_format("SDL version: %u.%u.%u\n", v.major, v.minor, v.patch);
    
    // From 2.0.18: Enable native IME.
#ifdef SDL_HINT_IME_SHOW_UI
    SDL_SetHint(SDL_HINT_IME_SHOW_UI, "1");
#endif
    
    SDL_WindowFlags window_flags = (SDL_WindowFlags)(
        SDL_WINDOW_RESIZABLE | SDL_WINDOW_ALLOW_HIGHDPI
    );
    sdl_window = SDL_CreateWindow(window_title.c_str(),
        SDL_WINDOWPOS_CENTERED, SDL_WINDOWPOS_CENTERED,
        1280, 720, window_flags);
    if(sdl_window == nullptr)
        throw SDLException();
    
    SDL_RendererFlags renderer_flags = (SDL_RendererFlags)(
        SDL_RENDERER_ACCELERATED | SDL_RENDERER_PRESENTVSYNC
    );
    sdl_renderer = SDL_CreateRenderer(sdl_window, -1, renderer_flags);
    if(sdl_renderer == nullptr)
        throw SDLException();
    
    SDL_RendererInfo info;
    SDL_GetRendererInfo(sdl_renderer, &info);
    std::cout<<"Current SDL_Renderer driver: "<<info.name<<std::endl;

    // Setup Dear ImGui context
    IMGUI_CHECKVERSION();
    ImGui::CreateContext();
    ImGuiIO& io = ImGui::GetIO();
    io.ConfigFlags |= ImGuiConfigFlags_NavEnableKeyboard; // Enable Keyboard Controls   
    io.ConfigFlags |= ImGuiConfigFlags_NavEnableGamepad;  // Enable Gamepad Controls
    io.ConfigFlags |= ImGuiConfigFlags_DockingEnable;     // Enable Docking
    io.ConfigDockingTransparentPayload = true;
    io.ConfigInputTextCursorBlink = false;
    // io.ConfigViewportsNoTaskBarIcon = true;
    io.ConfigWindowsMoveFromTitleBarOnly = true;

    // Setup Platform/Renderer backends
    if(!ImGui_ImplSDL2_InitForSDLRenderer(sdl_window, sdl_renderer))
        throw SDLException();
    if(!ImGui_ImplSDLRenderer2_Init(sdl_renderer))
        throw SDLException();

    // Setup Dear ImGui style
    ImGui::StyleColorsDark();
    f32 dpi = getDPI();
    io.FontDefault = resources::ImFont_LoadFromResource(default_font, default_font_size, dpi);
}

void MainWindow::close(){
    ImGui_ImplSDLRenderer2_Shutdown();
    ImGui_ImplSDL2_Shutdown();
    ImGui::DestroyContext();
    SDL_DestroyRenderer(sdl_renderer);
    SDL_DestroyWindow(sdl_window);
    SDL_Quit();
}

void MainWindow::pollEvents(bool* loopRunning){
    SDL_Event event;
    while (SDL_PollEvent(&event)) {
        ImGui_ImplSDL2_ProcessEvent(&event);
        switch(event.type){
            case SDL_QUIT: {
                *loopRunning = false;
                break;
            }
            case SDL_WINDOWEVENT: {
                if(event.window.event == SDL_WINDOWEVENT_CLOSE 
                    && event.window.windowID == SDL_GetWindowID(sdl_window))
                {
                    *loopRunning = false;
                }
                break;
            }
        }
    }
}


void MainWindow::beginFrame(){
    // Start the Dear ImGui frame
    ImGui_ImplSDLRenderer2_NewFrame();
    ImGui_ImplSDL2_NewFrame();
    ImGui::NewFrame();

    ImGuiIO& io = ImGui::GetIO();
    SDL_RenderSetScale(sdl_renderer, io.DisplayFramebufferScale.x, io.DisplayFramebufferScale.y);
    SDL_SetRenderDrawColor(sdl_renderer, 
        (u8)(clear_color.x * 255),
        (u8)(clear_color.y * 255),
        (u8)(clear_color.z * 255),
        (u8)(clear_color.w * 255));
    SDL_RenderClear(sdl_renderer);

    draw_bg_window();
    draw_debug_window();
}

void MainWindow::endFrame(){
    ImGui::Render();
    ImGui_ImplSDLRenderer2_RenderDrawData(ImGui::GetDrawData(), sdl_renderer);
    // Swap buffers
    SDL_RenderPresent(sdl_renderer);
}

void MainWindow::draw_bg_window(){
    const ImGuiDockNodeFlags dockspace_flags =
            ImGuiDockNodeFlags_PassthruCentralNode;
    const ImGuiWindowFlags window_flags =
        ImGuiWindowFlags_MenuBar |
        ImGuiWindowFlags_NoDocking |
        ImGuiWindowFlags_NoScrollbar |
        ImGuiWindowFlags_NoScrollWithMouse |
        ImGuiWindowFlags_NoBringToFrontOnFocus |
        ImGuiWindowFlags_NoFocusOnAppearing |
        ImGuiWindowFlags_NoMove |
        ImGuiWindowFlags_NoResize |
        ImGuiWindowFlags_NoCollapse |
        ImGuiWindowFlags_NoTitleBar |
        ImGuiWindowFlags_NoBackground;
    // not dockable window that always bound to viewport
    ImGuiViewport* viewport = ImGui::GetWindowViewport();
    ImGui::SetNextWindowPos(viewport->Pos, ImGuiCond_Always);
    ImGui::SetNextWindowSize(viewport->Size, ImGuiCond_Always);
    ImGui::PushStyleVar(ImGuiStyleVar_WindowRounding, 0.0f);
    ImGui::PushStyleVar(ImGuiStyleVar_WindowBorderSize, 0.0f);
	ImGui::PushStyleVar(ImGuiStyleVar_WindowPadding, ImVec2(0.0f, 0.0f));
    ImGui::Begin("bg_window", nullptr, window_flags);
    ImGui::PopStyleVar(3);

    // DockSpace
    ImGuiID dockspace_id = ImGui::GetID("bg_dockspace");
    ImGui::DockSpace(dockspace_id, ImVec2(0.0f, 0.0f), dockspace_flags);
    
    // MenuBar
    if(ImGui::BeginMainMenuBar()){
        if(ImGui::BeginMenu("test")){
            if(ImGui::MenuItem("throw exception")){
                ImGui::EndMenu();
                ImGui::EndMainMenuBar();
                ImGui::End();
                throw UsefulException("example exception");
            }
            if(ImGui::MenuItem("throw const char*")){
                ImGui::EndMenu();
                ImGui::EndMainMenuBar();
                ImGui::End();
                throw "cptr";
            }
            if(ImGui::MenuItem("throw std::string")){
                ImGui::EndMenu();
                ImGui::EndMainMenuBar();
                ImGui::End();
                throw std::string("str");
            }
            if(ImGui::MenuItem("throw unknown")){
                ImGui::EndMenu();
                ImGui::EndMainMenuBar();
                ImGui::End();
                throw 111;
            }
            ImGui::EndMenu();
        }
        ImGui::EndMainMenuBar();
    }

    ImGui::End();
}

void MainWindow::draw_debug_window(){
    ImGuiIO& io = ImGui::GetIO();
    if(ImGui::Begin("Debug Options", &show_debug_window)){
        ImGui::ColorEdit3("clear_color", (float*)&clear_color);
        ImGui::InputInt("fps_max", &fps_max);
        ImGui::Text("Application average %.3f ms/frame (%.2f FPS)", 1000.0f / io.Framerate, io.Framerate);
        ImGui::Checkbox("Demo Window", &show_demo_window);
        ImGui::Checkbox("Metrics/Debug Window", &show_metrics_window);
        ImGui::End();
    }

    if (show_demo_window)
        ImGui::ShowDemoWindow(&show_demo_window);

    if (show_metrics_window)
        ImGui::ShowMetricsWindow(&show_metrics_window);
}

}