#include <scwx/qt/ui/imgui_debug_widget.hpp>

#include <imgui.h>
#include <backends/imgui_impl_opengl3.h>
#include <backends/imgui_impl_qt.hpp>

namespace scwx
{
namespace qt
{
namespace ui
{

static const std::string logPrefix_ = "scwx::qt::ui::imgui_debug_widget";

class ImGuiDebugWidgetImpl
{
public:
   explicit ImGuiDebugWidgetImpl(ImGuiDebugWidget* self) :
       self_ {self}, context_ {ImGui::CreateContext()}
   {
      // Set ImGui Context
      ImGui::SetCurrentContext(context_);

      // ImGui Configuration
      auto& io = ImGui::GetIO();

      // Disable automatic configuration loading/saving
      io.IniFilename = nullptr;

      // Initialize ImGui Qt backend
      ImGui_ImplQt_Init();
      ImGui_ImplQt_RegisterWidget(self_);
   }

   ~ImGuiDebugWidgetImpl()
   {
      // Set ImGui Context
      ImGui::SetCurrentContext(context_);

      // Shutdown ImGui Context
      ImGui_ImplOpenGL3_Shutdown();
      ImGui_ImplQt_Shutdown();
      ImGui::DestroyContext(context_);
   }

   ImGuiDebugWidget* self_;
   ImGuiContext*     context_;

   bool firstRender_ {true};
};

ImGuiDebugWidget::ImGuiDebugWidget(QWidget* parent) :
    QOpenGLWidget(parent), p {std::make_unique<ImGuiDebugWidgetImpl>(this)}
{
   // Accept focus for keyboard events
   setFocusPolicy(Qt::StrongFocus);
}

void ImGuiDebugWidget::initializeGL()
{
   makeCurrent();

   // Initialize ImGui OpenGL3 backend
   ImGui::SetCurrentContext(p->context_);
   ImGui_ImplOpenGL3_Init();
}

void ImGuiDebugWidget::paintGL()
{
   ImGui::SetCurrentContext(p->context_);

   ImGui_ImplQt_NewFrame(this);
   ImGui_ImplOpenGL3_NewFrame();

   ImGui::NewFrame();

   if (p->firstRender_)
   {
      // Set initial position of demo window
      ImGui::SetNextWindowPos(ImVec2 {width() / 2.0f, height() / 2.0f},
                              ImGuiCond_FirstUseEver,
                              ImVec2 {0.5f, 0.5f});
      ImGui::Begin("Dear ImGui Demo");
      ImGui::End();

      p->firstRender_ = false;
   }

   ImGui::ShowDemoWindow();

   ImGui::Render();
   ImGui_ImplOpenGL3_RenderDrawData(ImGui::GetDrawData());
}

ImGuiDebugWidget::~ImGuiDebugWidget() {}

} // namespace ui
} // namespace qt
} // namespace scwx