1 files changed, 93 insertions, 0 deletions
diff --git a/editor/imgui-raii.cpp b/editor/imgui-raii.cpp
new file mode 100644
index 00000000..7c885114
--- /dev/null
+++ b/editor/imgui-raii.cpp
@@ -0,0 +1,93 @@
+#include "imgui-raii.hpp"
+#include <Corrade/Containers/StringView.h>
+#include <Magnum/Magnum.h>
+#include <Magnum/Math/Color.h>
+
+namespace floormat::imgui {
+
+font_saver::~font_saver()
+{
+    auto& ctx = *ImGui::GetCurrentContext();
+    ctx.FontSize = font_size;
+    ctx.FontBaseSize = font_base_size;
+}
+
+font_saver::font_saver(ImGuiContext& ctx, float size) :
+      font_size{ctx.FontSize}, font_base_size{ctx.FontBaseSize}
+{
+    ctx.FontSize = size;
+    ctx.FontBaseSize = size;
+}
+
+font_saver::font_saver(float size) : font_saver{*ImGui::GetCurrentContext(), size} {}
+style_saver::style_saver() : style{ImGui::GetStyle()} {}
+style_saver::~style_saver() { ImGui::GetStyle() = style; }
+
+void text(const char* str, std::size_t len, ImGuiTextFlags_ flags) { ImGui::TextEx(str, str + len, flags); }
+
+raii_wrapper::raii_wrapper(raii_wrapper::F fn) : dtor{fn} {}
+raii_wrapper::~raii_wrapper() { if (dtor) dtor(); }
+raii_wrapper::raii_wrapper(raii_wrapper&& other) noexcept : dtor{other.dtor} { other.dtor = nullptr; }
+raii_wrapper::operator bool() const noexcept { return dtor != nullptr; }
+
+raii_wrapper push_style_color(ImGuiCol_ var, const Color4& value)
+{
+    ImGui::PushStyleColor(var, {value[0], value[1], value[2], value[3]});
+    return {[]{ ImGui::PopStyleColor(); }};
+}
+
+raii_wrapper push_style_var(ImGuiStyleVar_ var, float value)
+{
+    ImGui::PushStyleVar(var, value);
+    return {[]{ ImGui::PopStyleVar(); }};
+}
+
+raii_wrapper push_style_var(ImGuiStyleVar_ var, Vector2 value)
+{
+    ImGui::PushStyleVar(var, {value[0], value[1]});
+    return {[]{ ImGui::PopStyleVar(); }};
+}
+
+raii_wrapper tree_node(Containers::StringView name, ImGuiTreeNodeFlags_ flags)
+{
+    if (ImGui::TreeNodeEx(name.data(), flags))
+        return {&ImGui::TreePop};
+    else
+        return {};
+}
+
+raii_wrapper begin_list_box(Containers::StringView name, ImVec2 size)
+{
+    if (ImGui::BeginListBox(name.data(), size))
+        return {&ImGui::EndListBox};
+    else
+        return {};
+}
+
+raii_wrapper begin_menu(Containers::StringView name, bool enabled)
+{
+    if (ImGui::BeginMenu(name.data(), enabled))
+        return {&ImGui::EndMenu};
+    else
+        return {};
+}
+
+raii_wrapper begin_main_menu()
+{
+    if (ImGui::BeginMainMenuBar())
+        return {&ImGui::EndMainMenuBar};
+    else
+        return {};
+}
+
+raii_wrapper begin_window(Containers::StringView name, ImGuiWindowFlags_ flags)
+{
+    if (name.isEmpty())
+        name = "floormat editor";
+    if (ImGui::Begin(name.data(), nullptr, flags))
+        return {&ImGui::End};
+    else
+        return {};
+}
+
+} // namespace floormat::imgui