/*!
 * \file src/UI.cpp
 * \brief UI/Event Manager
 *
 * \author xythobuz
 */

#include "imgui/imgui.h"

#include "global.h"
#include "Camera.h"
#include "Console.h"
#include "Game.h"
#include "Log.h"
#include "Menu.h"
#include "Render.h"
#include "RunTime.h"
#include "Selector.h"
#include "SoundManager.h"
#include "TextureManager.h"
#include "World.h"
#include "system/Sound.h"
#include "system/Window.h"
#include "utils/time.h"
#include "UI.h"

#include <glbinding/gl/gl.h>
#include <glm/gtc/matrix_transform.hpp>

#define OFFSETOF(TYPE, ELEMENT) (&(static_cast<TYPE *>(nullptr)->ELEMENT))

Shader UI::imguiShader;
bool UI::visible = false;
unsigned int UI::fontTex;
std::string UI::iniFilename;
std::string UI::logFilename;
bool UI::metaKeyIsActive = false;
unsigned int UI::vboHandle = 0;
unsigned int UI::elementHandle = 0;

std::list<std::tuple<KeyboardButton, bool>> UI::keyboardEvents;
std::list<std::tuple<unsigned int, unsigned int, KeyboardButton, bool>> UI::clickEvents;
std::list<std::tuple<int, int, int, int>> UI::motionEvents;
std::list<std::tuple<int, int>> UI::scrollEvents;

void UI::setSize(glm::i32vec2 s) {
    ImGuiIO& io = ImGui::GetIO();
    io.DisplaySize = ImVec2(s.x, s.y);
}

static int attribPos, attribUV, attribCol;

int UI::initialize() {
    if (imguiShader.compile(imguiShaderVertex, imguiShaderFragment) < 0)
        return -1;
    if (imguiShader.addUniform("screen") < 0)
        return -2;
    if (imguiShader.addUniform("textureSampler") < 0)
        return -3;

    attribPos = imguiShader.getAttrib("vertexPosition_screen");
    attribUV = imguiShader.getAttrib("vertexUV");
    attribCol = imguiShader.getAttrib("vertexColor");

    iniFilename = RunTime::getBaseDir() + "/imgui.ini";
    logFilename = RunTime::getBaseDir() + "/imgui_log.txt";

    ImGuiIO& io = ImGui::GetIO();
    io.DisplaySize = ImVec2(Window::getSize().x, Window::getSize().y);
    io.DeltaTime = 1.0f / 60.0f;

    io.IniFilename = iniFilename.c_str();
    io.LogFilename = logFilename.c_str();

    io.KeyMap[ImGuiKey_Tab] = tabKey;
    io.KeyMap[ImGuiKey_LeftArrow] = leftKey;
    io.KeyMap[ImGuiKey_RightArrow] = rightKey;
    io.KeyMap[ImGuiKey_UpArrow] = upKey;
    io.KeyMap[ImGuiKey_DownArrow] = downKey;
    io.KeyMap[ImGuiKey_Home] = homeKey;
    io.KeyMap[ImGuiKey_End] = endKey;
    io.KeyMap[ImGuiKey_Delete] = delKey;
    io.KeyMap[ImGuiKey_Backspace] = backspaceKey;
    io.KeyMap[ImGuiKey_Enter] = enterKey;
    io.KeyMap[ImGuiKey_Escape] = escapeKey;
    io.KeyMap[ImGuiKey_A] = aKey;
    io.KeyMap[ImGuiKey_C] = cKey;
    io.KeyMap[ImGuiKey_V] = vKey;
    io.KeyMap[ImGuiKey_X] = xKey;
    io.KeyMap[ImGuiKey_Y] = yKey;
    io.KeyMap[ImGuiKey_Z] = zKey;

    io.RenderDrawListsFn = UI::renderImGui;
    io.GetClipboardTextFn = Window::getClipboard;
    io.SetClipboardTextFn = Window::setClipboard;
    io.ImeSetInputScreenPosFn = Window::inputPositionCallback;

    // Load font texture
    //! \fixme allow loading other TTF fonts
    unsigned char* pixels;
    int width, height;
    io.Fonts->GetTexDataAsRGBA32(&pixels, &width, &height);
    fontTex = TextureManager::loadBufferSlot(pixels, width, height, ColorMode::RGBA, 32,
              TextureStorage::SYSTEM, -1, true);
    auto bm = TextureManager::getBufferManager(fontTex, TextureStorage::SYSTEM);
    io.Fonts->TexID = bm;

    gl::glGenBuffers(1, &vboHandle);
    gl::glGenBuffers(1, &elementHandle);

    // Set up OpenRaider style
    /*
    ImGuiStyle& style = ImGui::GetStyle();
    style.Colors[ImGuiCol_Text]                 = ImVec4(0.90f, 0.90f, 0.90f, 1.00f);
    style.Colors[ImGuiCol_WindowBg]             = ImVec4(0.00f, 0.00f, 0.00f, 1.00f);
    style.Colors[ImGuiCol_Border]               = ImVec4(1.00f, 1.00f, 1.00f, 1.00f);
    style.Colors[ImGuiCol_BorderShadow]         = ImVec4(0.00f, 0.00f, 0.00f, 0.60f);
    style.Colors[ImGuiCol_FrameBg]              = ImVec4(0.80f, 0.80f, 0.80f, 0.30f);
    style.Colors[ImGuiCol_FrameBgHovered]       = ImVec4(1.00f, 0.40f, 0.40f, 0.45f);
    style.Colors[ImGuiCol_FrameBgActive]        = ImVec4(0.65f, 0.50f, 0.50f, 0.55f);
    style.Colors[ImGuiCol_TitleBg]              = ImVec4(0.50f, 0.70f, 1.00f, 0.45f);
    style.Colors[ImGuiCol_TitleBgCollapsed]     = ImVec4(0.40f, 0.65f, 1.00f, 0.20f);
    style.Colors[ImGuiCol_ScrollbarBg]          = ImVec4(0.40f, 0.65f, 1.00f, 0.15f);
    style.Colors[ImGuiCol_ScrollbarGrab]        = ImVec4(0.40f, 0.65f, 1.00f, 0.30f);
    style.Colors[ImGuiCol_ScrollbarGrabHovered] = ImVec4(0.40f, 0.65f, 1.00f, 0.40f);
    style.Colors[ImGuiCol_ScrollbarGrabActive]  = ImVec4(1.00f, 0.16f, 0.16f, 0.40f);
    style.Colors[ImGuiCol_ComboBg]              = ImVec4(0.20f, 0.20f, 0.20f, 0.99f);
    style.Colors[ImGuiCol_CheckMark]            = ImVec4(0.90f, 0.90f, 0.90f, 0.50f);
    style.Colors[ImGuiCol_SliderGrab]           = ImVec4(1.00f, 1.00f, 1.00f, 0.30f);
    style.Colors[ImGuiCol_SliderGrabActive]     = ImVec4(1.00f, 0.20f, 0.22f, 1.00f);
    style.Colors[ImGuiCol_Button]               = ImVec4(1.00f, 0.20f, 0.20f, 0.60f);
    style.Colors[ImGuiCol_ButtonHovered]        = ImVec4(1.00f, 0.20f, 0.18f, 1.00f);
    style.Colors[ImGuiCol_ButtonActive]         = ImVec4(1.00f, 0.00f, 0.00f, 1.00f);
    style.Colors[ImGuiCol_Header]               = ImVec4(0.21f, 0.54f, 1.00f, 0.45f);
    style.Colors[ImGuiCol_HeaderHovered]        = ImVec4(0.22f, 0.56f, 1.00f, 0.80f);
    style.Colors[ImGuiCol_HeaderActive]         = ImVec4(0.00f, 0.31f, 1.00f, 1.00f);
    style.Colors[ImGuiCol_Column]               = ImVec4(1.00f, 1.00f, 1.00f, 1.00f);
    style.Colors[ImGuiCol_ColumnHovered]        = ImVec4(0.60f, 0.40f, 0.40f, 1.00f);
    style.Colors[ImGuiCol_ColumnActive]         = ImVec4(0.80f, 0.50f, 0.50f, 1.00f);
    style.Colors[ImGuiCol_ResizeGrip]           = ImVec4(1.00f, 1.00f, 1.00f, 0.30f);
    style.Colors[ImGuiCol_ResizeGripHovered]    = ImVec4(1.00f, 1.00f, 1.00f, 0.60f);
    style.Colors[ImGuiCol_ResizeGripActive]     = ImVec4(1.00f, 1.00f, 1.00f, 0.90f);
    style.Colors[ImGuiCol_CloseButton]          = ImVec4(0.21f, 0.56f, 1.00f, 0.50f);
    style.Colors[ImGuiCol_CloseButtonHovered]   = ImVec4(0.21f, 0.32f, 1.00f, 0.60f);
    style.Colors[ImGuiCol_CloseButtonActive]    = ImVec4(0.70f, 0.70f, 0.70f, 1.00f);
    style.Colors[ImGuiCol_PlotLines]            = ImVec4(1.00f, 1.00f, 1.00f, 1.00f);
    style.Colors[ImGuiCol_PlotLinesHovered]     = ImVec4(0.90f, 0.70f, 0.00f, 1.00f);
    style.Colors[ImGuiCol_PlotHistogram]        = ImVec4(0.90f, 0.70f, 0.00f, 1.00f);
    style.Colors[ImGuiCol_PlotHistogramHovered] = ImVec4(1.00f, 0.60f, 0.00f, 1.00f);
    style.Colors[ImGuiCol_TextSelectedBg]       = ImVec4(0.00f, 0.00f, 1.00f, 0.35f);
    style.Colors[ImGuiCol_TooltipBg]            = ImVec4(0.05f, 0.05f, 0.10f, 0.90f);
    style.WindowPadding                         = ImVec2(2, 2);
    style.WindowRounding                        = 9;
    style.FramePadding                          = ImVec2(2, 1);
    style.ItemSpacing                           = ImVec2(2, 2);
    style.ItemInnerSpacing                      = ImVec2(1, 1);
    style.TouchExtraPadding                     = ImVec2(0, 0);
    style.IndentSpacing                         = 3;
    style.ScrollbarWidth                        = 10;
    */

    return 0;
}

void UI::eventsFinished() {
    ImGuiIO& io = ImGui::GetIO();
    io.DisplaySize = ImVec2(Window::getSize().x, Window::getSize().y);

    static unsigned long lastTime = 0;
    io.DeltaTime = (systemTimerGet() - lastTime) / 1000.0f;
    lastTime = systemTimerGet();
    if (io.DeltaTime <= 0.0f)
        io.DeltaTime = 1.0f / 60.0f;

    ImGui::NewFrame();

    if (!(visible || Console::isVisible() || Menu::isVisible() || Selector::isVisible())) {
        while (!motionEvents.empty()) {
            auto i = motionEvents.front();
            Game::handleMouseMotion(std::get<0>(i), std::get<1>(i),
                                    std::get<2>(i), std::get<3>(i));
            motionEvents.pop_front();
        }
    }

    while (!keyboardEvents.empty()) {
        auto i = keyboardEvents.front();

        if (!(visible || Console::isVisible() || Menu::isVisible() || Selector::isVisible())) {
            for (int n = forwardAction; n < ActionEventCount; n++) {
                auto ae = static_cast<ActionEvents>(n);
                if (RunTime::getKeyBinding(ae) == std::get<0>(i))
                    Game::handleAction(ae, !std::get<1>(i));
            }
        }

        if (std::get<1>(i)) {
            if ((!io.WantCaptureKeyboard) || (!(visible || Console::isVisible() || Menu::isVisible() || Selector::isVisible()))) {
                if (!metaKeyIsActive) {
                    if (RunTime::getKeyBinding(debugAction) == std::get<0>(i)) {
                        visible = !visible;
                    }

                    if (RunTime::getKeyBinding(consoleAction) == std::get<0>(i)) {
                        Console::setVisible(!Console::isVisible());
                    }

                    if (RunTime::getKeyBinding(menuAction) == std::get<0>(i)) {
                        Menu::setVisible(!Menu::isVisible());
                    }
                }
            }
        }

        keyboardEvents.pop_front();
    }

    bool clicked = !clickEvents.empty();
    while (!clickEvents.empty()) {
        auto i = clickEvents.front();

        if (Selector::isVisible() && (!io.WantCaptureMouse)) {
            Selector::handleMouseClick(std::get<0>(i), std::get<1>(i), std::get<2>(i), std::get<3>(i));
        }

        clickEvents.pop_front();
    }

    clickEvents.clear();
    motionEvents.clear();
    scrollEvents.clear();

    // Allow clicking into the game to hide debug UI
    if ((visible || Console::isVisible()) && ((!io.WantCaptureMouse) && clicked)) {
        visible = false;
        Console::setVisible(false);
    }

    if (Window::getTextInput() != (visible || Console::isVisible() || Menu::isVisible() || Selector::isVisible()))
        Window::setTextInput(visible || Console::isVisible() || Menu::isVisible() || Selector::isVisible());

    bool input = !(visible || Console::isVisible() || Menu::isVisible() || Selector::isVisible());
    if (Window::getMousegrab() != input)
        Window::setMousegrab(input);

    io.MouseWheel = 0;
}

void UI::display() {
    if (RunTime::getShowFPS() && (!Menu::isVisible())) {
        if (ImGui::Begin("Debug Overlay", nullptr, ImVec2(0, 0), -1.0f,
                         ImGuiWindowFlags_NoTitleBar | ImGuiWindowFlags_NoResize
                         | ImGuiWindowFlags_NoMove | ImGuiWindowFlags_NoSavedSettings
                         | ImGuiWindowFlags_AlwaysAutoResize)) {
#ifdef DEBUG
            ImGui::Text("%d FPS  %lu CPF", RunTime::getFPS(), RunTime::getCallCount());
#else
            ImGui::Text("%d FPS", RunTime::getFPS());
#endif

            ImGui::Text("X: %.1f (%.2f)", Camera::getPosition().x, Camera::getRotation().x);
            ImGui::Text("Y: %.2f (%.2f)", Camera::getPosition().y, Camera::getRotation().y);
            ImGui::Text("Z: %.2f (%d)", Camera::getPosition().z, Camera::getRoom());

            auto window = ImGui::GetWindowSize();
            auto screen = Window::getSize();
            //ImGui::SetWindowPos(ImVec2(10, screen.y - window.y - 10));
        }
        ImGui::End();
    }

    if (Game::isLoaded())
        Camera::displayUI();

    Console::display();
    Menu::display();
    Selector::display();

    if (!visible) {
        ImGui::Render();
        return;
    }

    static bool showTestWindow = false;
    static bool showStyleWindow = false;
    if (ImGui::Begin("Engine", &visible, ImVec2(500, 600))) {
        if (Selector::isVisible()) {
            if (ImGui::Button("Disable Mouse Object Picking")) {
                Selector::setVisible(false);
            }
        } else {
            if (ImGui::Button("Enable Mouse Object Picking")) {
                Selector::setVisible(true);
            }
        }

        Render::displayUI();
        RunTime::display();
        TextureManager::display();
        SoundManager::display();
        World::displayUI();

        if (ImGui::CollapsingHeader("Library Versions")) {
            ImGui::TextWrapped("%s", VERSION);
            ImGui::Separator();
            ImGui::TextWrapped("ImGui v%s", IMGUI_VERSION);
            ImGui::TextWrapped("%s", Window::getVersion(false).c_str());
            ImGui::TextWrapped("%s", Shader::getVersion(false).c_str());
            ImGui::TextWrapped("%s", Sound::getVersion(false).c_str());
            ImGui::TextWrapped("GLM v%d.%d.%d", (GLM_VERSION / 100),
                               ((GLM_VERSION % 100) / 10), (GLM_VERSION % 10));
            ImGui::Separator();
            ImGui::TextWrapped("ImGui v%s", ImGui::GetVersion());
            ImGui::TextWrapped("%s", Window::getVersion(true).c_str());
            ImGui::TextWrapped("%s", Shader::getVersion(true).c_str());
            ImGui::TextWrapped("%s", Sound::getVersion(true).c_str());
        }

        if (ImGui::CollapsingHeader("ImGui/Debug UI Help")) {
            ImGui::ShowUserGuide();
            ImGui::Separator();
            if (ImGui::Button("Show/Hide Test Window")) {
                showTestWindow = !showTestWindow;
            }
            ImGui::SameLine();
            if (ImGui::Button("Show/Hide Style Editor")) {
                showStyleWindow = !showStyleWindow;
            }
        }

        if (ImGui::CollapsingHeader("ShaderTexture Test")) {
            static ShaderTexture* st = nullptr;
            static int index = 0;
            ImGui::SliderInt("SkeletalModel", &index, 0, World::sizeSkeletalModel() - 1);

            static glm::mat4 MVP(1.0f);
            static bool dirty = true, redraw = false;
            static float zoom = 1.0f;
            static float pos[3];
            static float rot[3];

            if (ImGui::SliderFloat3("Position", pos, -100.0f, 100.0f)) {
                //dirty = true;
            }
            if (ImGui::SliderFloat3("Rotation", rot, -6.0f, 6.0f)) {
                //dirty = true;
            }
            if (ImGui::SliderFloat("Zoom", &zoom, -1.0f, 2.0f)) {
                //dirty = true;
            }

            if (dirty) {
                static glm::mat4 projection = glm::perspective(45.0f, 1.0f, 0.1f, 2000.0f);

                glm::mat4 rotateX = glm::rotate(glm::mat4(1.0f), rot[0], glm::vec3(1.0f, 0.0f, 0.0f));
                glm::mat4 rotateY = glm::rotate(glm::mat4(1.0f), rot[1], glm::vec3(0.0f, 1.0f, 0.0f));
                glm::mat4 rotateZ = glm::rotate(glm::mat4(1.0f), rot[2], glm::vec3(0.0f, 0.0f, 1.0f));
                glm::mat4 rotate = rotateZ * rotateX * rotateY;
                glm::mat4 translate = glm::translate(glm::mat4(1.0f), glm::vec3(pos[0], pos[1], pos[2]));
                //glm::mat4 view = glm::inverse(rotate * translate);
                glm::mat4 view = rotate * translate;
                glm::mat4 scale = glm::scale(glm::mat4(1.0f), glm::vec3(zoom, zoom, zoom));
                MVP = projection * view * scale;

                redraw = true;
                dirty = false;
            }

            if ((index >= 0) && (index < World::sizeSkeletalModel())) {
                auto& sm = World::getSkeletalModel(index);
                if ((sm.size() > 0) && (sm.get(0).size() > 0)) {
                    if (ImGui::Button("(Re)Create ShaderTexture...")) {
                        if (st != nullptr) {
                            delete st;
                            st = nullptr;
                        }
                        if (st == nullptr) {
                            st = new ShaderTexture();
                            redraw = true;
                        }
                    }
                    if ((st != nullptr) && redraw) {
                        st->clear();
                        sm.display(MVP, 0, 0, st);
                        redraw = false;
                        Log::get(LOG_DEBUG) << "Rendered new ShaderTexture!" << Log::endl;
                    }
                } else {
                    ImGui::Text("Selected SkeletalModel has no animation/frame!");
                }
            } else {
                ImGui::Text("No SkeletalModels loaded!");
            }

            if (st != nullptr) {
                auto bm = TextureManager::getBufferManager(st->getTexture(), TextureStorage::SYSTEM);
                ImGui::Image(bm, ImVec2(ImGui::GetColumnWidth() * 2 / 3,
                                        ImGui::GetColumnWidth() * 2 / 3));
            }
        }
    }
    ImGui::End();

    if (showTestWindow)
        ImGui::ShowTestWindow();

    if (showStyleWindow)
        ImGui::ShowStyleEditor();

    ImGui::Render();
}

void UI::shutdown() {
    ImGui::Shutdown();

    gl::glDeleteBuffers(1, &vboHandle);
    gl::glDeleteBuffers(1, &elementHandle);
}

void UI::handleKeyboard(KeyboardButton key, bool pressed) {
    ImGuiIO& io = ImGui::GetIO();
    io.KeysDown[key] = pressed;
    io.KeyCtrl = io.KeysDown[leftctrlKey] | io.KeysDown[rightctrlKey];
    io.KeyShift = io.KeysDown[leftshiftKey] | io.KeysDown[rightshiftKey];
    io.KeyAlt = io.KeysDown[leftaltKey] | io.KeysDown[rightaltKey];

    keyboardEvents.push_back(std::make_tuple(key, pressed));

    if ((key == leftguiKey) || (key == rightguiKey))
        metaKeyIsActive = pressed;
}

void UI::handleText(char* text, bool notFinished) {
    if (notFinished)
        return;

    ImGuiIO& io = ImGui::GetIO();
    while (*text != '\0') {
        io.AddInputCharacter(*text);
        text++;
    }
}

void UI::handleMouseClick(unsigned int x, unsigned int y, KeyboardButton button, bool released) {
    ImGuiIO& io = ImGui::GetIO();
    io.MousePos = ImVec2(x, y);
    if (button == leftmouseKey) {
        io.MouseDown[0] = !released;
    } else if (button == rightmouseKey) {
        io.MouseDown[1] = !released;
    } else if (button == middlemouseKey) {
        io.MouseDown[2] = !released;
    } else if (button == fourthmouseKey) {
        io.MouseDown[3] = !released;
    } else if (button == fifthmouseKey) {
        io.MouseDown[4] = !released;
    }

    clickEvents.push_back(std::make_tuple(x, y, button, released));
}

void UI::handleMouseMotion(int xrel, int yrel, int xabs, int yabs) {
    ImGuiIO& io = ImGui::GetIO();
    io.MousePos = ImVec2(xabs, yabs);

    motionEvents.push_back(std::make_tuple(xrel, yrel, xabs, yabs));
}

void UI::handleMouseScroll(int xrel, int yrel) {
    ImGuiIO& io = ImGui::GetIO();
    io.MouseWheel += yrel;

    scrollEvents.push_back(std::make_tuple(xrel, yrel));
}

void UI::handleControllerAxis(float value, KeyboardButton axis) {
    Game::handleControllerAxis(value, axis);
}

void UI::handleControllerButton(KeyboardButton button, bool released) {
    if (visible || Console::isVisible() || Menu::isVisible() || Selector::isVisible())
        return;

    if (Menu::isVisible()) {
        if (button == aButton) {
            handleKeyboard(enterKey, !released);
        } else if (button == padUp) {
            handleKeyboard(upKey, !released);
        } else if (button == padDown) {
            handleKeyboard(downKey, !released);
        } else if (button == padLeft) {
            handleKeyboard(leftKey, !released);
        } else if (button == padRight) {
            handleKeyboard(rightKey, !released);
        } else if (button == startButton) {
            if (!released)
                Menu::setVisible(false);
        }
    } else {
        if (button == aButton) {
            Game::handleAction(jumpAction, released);
        } else if (button == bButton) {
            Game::handleAction(crouchAction, released);
        } else if (button == xButton) {
            Game::handleAction(useAction, released);
        } else if (button == yButton) {
            Game::handleAction(holsterAction, released);
        } else if (button == padUp) {
            Game::handleAction(forwardAction, released);
        } else if (button == padDown) {
            Game::handleAction(backwardAction, released);
        } else if (button == padLeft) {
            Game::handleAction(leftAction, released);
        } else if (button == padRight) {
            Game::handleAction(rightAction, released);
        } else if (button == leftShoulder) {
            Game::handleAction(walkAction, released);
        } else if (button == startButton) {
            if (!released)
                Menu::setVisible(true);
        }
    }
}

void UI::renderImGui(ImDrawData* draw_data) {
    if (draw_data->CmdListsCount == 0)
        return;

    gl::glEnable(gl::GL_SCISSOR_TEST);
    Shader::set2DState(true);

    gl::glEnableVertexAttribArray(attribPos);
    gl::glEnableVertexAttribArray(attribUV);
    gl::glEnableVertexAttribArray(attribCol);

    gl::glBindBuffer(gl::GL_ARRAY_BUFFER, vboHandle);

    gl::glVertexAttribPointer(attribPos, 2, gl::GL_FLOAT, gl::GL_FALSE, sizeof(ImDrawVert),
                              OFFSETOF(ImDrawVert, pos));
    gl::glVertexAttribPointer(attribUV, 2, gl::GL_FLOAT, gl::GL_FALSE, sizeof(ImDrawVert),
                              OFFSETOF(ImDrawVert, uv));
    gl::glVertexAttribPointer(attribCol, 4, gl::GL_UNSIGNED_BYTE, gl::GL_TRUE, sizeof(ImDrawVert),
                              OFFSETOF(ImDrawVert, col));

    imguiShader.use();
    imguiShader.loadUniform(0, Window::getSize());

    for (int i = 0; i < draw_data->CmdListsCount; i++) {
        const ImDrawList* cmd_list = draw_data->CmdLists[i];
        const ImDrawIdx* idx_buffer_offset = 0;

        gl::glBindBuffer(gl::GL_ARRAY_BUFFER, vboHandle);
        gl::glBufferData(gl::GL_ARRAY_BUFFER, cmd_list->VtxBuffer.size() * sizeof(ImDrawVert),
                         &cmd_list->VtxBuffer.front(), gl::GL_STREAM_DRAW);

        gl::glBindBuffer(gl::GL_ELEMENT_ARRAY_BUFFER, elementHandle);
        gl::glBufferData(gl::GL_ELEMENT_ARRAY_BUFFER, cmd_list->IdxBuffer.size() * sizeof(ImDrawIdx),
                         &cmd_list->IdxBuffer.front(), gl::GL_STREAM_DRAW);

        for (const ImDrawCmd* pcmd = cmd_list->CmdBuffer.begin(); pcmd != cmd_list->CmdBuffer.end();
             pcmd++) {
            if (pcmd->UserCallback) {
                pcmd->UserCallback(cmd_list, pcmd);
                Log::get(LOG_INFO) << "renderImGui: did not draw (Callback)" << Log::endl;
            } else {
                auto bm = static_cast<BufferManager*>(pcmd->TextureId);
                orAssert(bm != nullptr);
                imguiShader.loadUniform(1, bm->getTextureID(), bm->getTextureStorage());

                gl::glScissor(pcmd->ClipRect.x,
                              Window::getSize().y - pcmd->ClipRect.w,
                              pcmd->ClipRect.z - pcmd->ClipRect.x,
                              pcmd->ClipRect.w - pcmd->ClipRect.y);

                gl::glDrawElements(gl::GL_TRIANGLES, pcmd->ElemCount, gl::GL_UNSIGNED_SHORT, idx_buffer_offset);
            }
            idx_buffer_offset += pcmd->ElemCount;
        }
    }

    gl::glDisableVertexAttribArray(attribPos);
    gl::glDisableVertexAttribArray(attribUV);
    gl::glDisableVertexAttribArray(attribCol);

    Shader::set2DState(false);
    gl::glDisable(gl::GL_SCISSOR_TEST);
}

// --------------------------------------
// *INDENT-OFF*

const char* UI::imguiShaderVertex = R"!?!(
#version 330 core

layout(location = 0) in vec2 vertexPosition_screen;
layout(location = 1) in vec2 vertexUV;
layout(location = 2) in vec4 vertexColor;

out vec2 UV;
out vec4 FragColor;

uniform vec2 screen;

void main() {
    vec2 halfScreen = screen / 2;
    vec2 vertexPosition_homogenous = (vertexPosition_screen - halfScreen) / halfScreen;

    gl_Position = vec4(vertexPosition_homogenous.x, -vertexPosition_homogenous.y, 0, 1);
    UV = vertexUV;
    FragColor = vertexColor;
}
)!?!";

const char* UI::imguiShaderFragment = R"!?!(
#version 330 core

in vec2 UV;
in vec4 FragColor;

out vec4 color;

uniform sampler2D textureSampler;

void main() {
    color = texture(textureSampler, UV) * FragColor;
}
)!?!";

// --------------------------------------
// *INDENT-ON*