Skip to content

Commit

Permalink
Refactor ModMenu
Browse files Browse the repository at this point in the history
  • Loading branch information
@doyaGu
doyaGu committed Jul 7, 2024
1 parent 30ccd19 commit d99fe70
Show file tree
Hide file tree
Showing 5 changed files with 935 additions and 691 deletions.
332 changes: 331 additions & 1 deletion include/BML/Bui.h
View file
Original file line number Diff line number Diff line change
@@ -1,6 +1,13 @@
#ifndef BML_BUI_H
#define BML_BUI_H

#include <string>
#include <stack>
#include <vector>
#include <unordered_map>
#include <functional>
#include <utility>

#include "imgui.h"

#include "BML/Export.h"
Expand All @@ -23,7 +30,7 @@ namespace Bui {
BUTTON_COUNT
};

inline ImVec2 CoordToScreenPos(const ImVec2 &coord) {
inline ImVec2 CoordToPixel(const ImVec2 &coord) {
const ImVec2 &vpSize = ImGui::GetMainViewport()->Size;
return {vpSize.x * coord.x, vpSize.y * coord.y};
}
Expand Down Expand Up @@ -95,6 +102,329 @@ namespace Bui {
BML_EXPORT bool InputTextButton(const char *label, char *buf, size_t buf_size, ImGuiInputTextFlags flags = 0, ImGuiInputTextCallback callback = nullptr, void *user_data = nullptr);
BML_EXPORT bool InputFloatButton(const char *label, float *v, float step = 0.0f, float step_fast = 0.0f, const char *format = "%.3f", ImGuiInputTextFlags flags = 0);
BML_EXPORT bool InputIntButton(const char *label, int* v, int step = 1, int step_fast = 100, ImGuiInputTextFlags flags = 0);

class Window {
public:
explicit Window(std::string name) : m_Name(std::move(name)) {}

virtual ~Window() = default;

const std::string &GetName() const { return m_Name; }

bool IsVisible() const { return m_Visible; }
bool IsHovered() const { return m_Hovered; }

void SetVisibility(bool visible) {
if (m_Visible != visible)
m_Visible = visible;
}
void ToggleVisibility() { SetVisibility(!m_Visible); }
void Show() { SetVisibility(true); }
void Hide() { SetVisibility(false); }

bool Begin() {
OnBegin();
bool keepVisible = true;
const bool notCollapsed = ImGui::Begin(m_Name.c_str(), &keepVisible, GetFlags());
if (!keepVisible)
Hide();
if (notCollapsed)
OnAfterBegin();
return notCollapsed;
}

void End() {
OnEnd();
m_Hovered = ImGui::IsWindowHovered();
ImGui::End();
}

void Render() {
if (!IsVisible())
return;

if (Begin()) {
OnDraw();
}

End();
}

virtual ImGuiWindowFlags GetFlags() { return 0; }
virtual void OnBegin() {}
virtual void OnAfterBegin() {}
virtual void OnDraw() = 0;
virtual void OnEnd() {}

protected:
std::string m_Name;
bool m_Visible = true;
bool m_Hovered = false;
};

class Page : public Window {
public:
explicit Page(std::string name) : Window(std::move(name)) { SetVisibility(false); }

int GetPage() const { return m_PageIndex; }
void SetPage(int page) {
if (page >= 0 && page < m_PageCount)
m_PageIndex = page;
}
int NextPage() { return m_PageIndex < m_PageCount ? ++m_PageIndex : m_PageCount - 1; }
int PrevPage() { return m_PageIndex > 0 ? --m_PageIndex : 0; }

int GetMaxPage() const { return m_PageCount; }
void SetMaxPage(int num) {
if (num > 0) {
if (m_PageIndex >= num)
m_PageIndex = num - 1;
m_PageCount = num;
}
}

void Open() { if (OnOpen()) Show(); }
void Close() { Hide(); OnClose(); }

ImGuiWindowFlags GetFlags() override {
return ImGuiWindowFlags_NoDecoration |
ImGuiWindowFlags_NoBackground |
ImGuiWindowFlags_NoMove |
ImGuiWindowFlags_NoScrollWithMouse |
ImGuiWindowFlags_NoBringToFrontOnFocus |
ImGuiWindowFlags_NoSavedSettings;
}

void OnBegin() override {
const ImVec2 &vpSize = ImGui::GetMainViewport()->Size;
ImGui::SetNextWindowPos(ImVec2(0.0f, 0.0f), ImGuiCond_Appearing);
ImGui::SetNextWindowSize(ImVec2(vpSize.x, vpSize.y), ImGuiCond_Appearing);
}

void OnAfterBegin() override {
if (!IsVisible())
return;

DrawCenteredText(m_Name.c_str());

if (m_PageIndex > 0 &&
LeftButton("PrevPage")) {
PrevPage();
}

if (m_PageCount > 1 && m_PageIndex < m_PageCount - 1 &&
RightButton("NextPage")) {
NextPage();
}
}

void OnEnd() override {
if (!IsVisible())
return;

if (BackButton("Back")) {
Close();
}
}

virtual bool OnOpen() { return true; }
virtual void OnClose() {}

static std::size_t DrawEntries(const std::function<bool(std::size_t index)> &onEntry,
const ImVec2 &pos = ImVec2(0.35f, 0.24f), float spacing = 0.14f, std::size_t capability = 4) {
auto oldPos = ImGui::GetCursorScreenPos();

for (std::size_t i = 0; i < capability; ++i) {
ImGui::SetCursorScreenPos(Bui::CoordToPixel(ImVec2(pos.x, pos.y + (float) i * spacing)));
if (!onEntry(i)) {
return i;
}
}

ImGui::SetCursorScreenPos(oldPos);
return capability;
}

static bool BackButton(const std::string &label, const ImVec2 &pos = ImVec2(0.4031f, 0.85f)) {
auto oldPos = ImGui::GetCursorScreenPos();
ImGui::SetCursorScreenPos(Bui::CoordToPixel(pos));
bool result = (Bui::BackButton(label.c_str()) || ImGui::IsKeyPressed(ImGuiKey_Escape));
ImGui::SetCursorScreenPos(oldPos);
return result;
}

static bool LeftButton(const std::string &label, const ImVec2 &pos = ImVec2(0.36f, 0.124f)) {
auto oldPos = ImGui::GetCursorScreenPos();
ImGui::SetCursorScreenPos(Bui::CoordToPixel(pos));
bool result = (Bui::LeftButton(label.c_str()) || ImGui::IsKeyPressed(ImGuiKey_PageUp));
ImGui::SetCursorScreenPos(oldPos);
return result;
}

static bool RightButton(const std::string &label, const ImVec2 &pos = ImVec2(0.6038f, 0.124f)) {
auto oldPos = ImGui::GetCursorScreenPos();
ImGui::SetCursorScreenPos(Bui::CoordToPixel(pos));
bool result = (Bui::RightButton(label.c_str()) || ImGui::IsKeyPressed(ImGuiKey_PageDown));
ImGui::SetCursorScreenPos(oldPos);
return result;
}

static void WrappedText(const char *text, float length, float scale = 1.0f) {
if (!text || text[0] == '\0')
return;

float oldPosX = ImGui::GetCursorPosX();

const float width = ImGui::CalcTextSize(text).x;
const float indent = (length - width) * 0.5f;
if (indent > 0) {
ImGui::SetCursorPosX(oldPosX + indent);
}

ImGui::PushTextWrapPos(ImGui::GetCursorPosX() + length);

const float oldScale = ImGui::GetFont()->Scale;
if (scale != 1.0f) {
ImGui::GetFont()->Scale *= scale;
ImGui::PushFont(ImGui::GetFont());
}
ImGui::TextUnformatted(text);
if (scale != 1.0f) {
ImGui::GetFont()->Scale = oldScale;
ImGui::PopFont();
}

ImGui::PopTextWrapPos();

ImGui::SetCursorPosX(oldPosX);
}

static void DrawCenteredText(const char *text, float y = 0.13f, float scale = 1.5f, ImU32 color = IM_COL32_WHITE) {
if (!text || text[0] == '\0')
return;

float oldScale = ImGui::GetFont()->Scale;
ImGui::GetFont()->Scale *= scale;
ImGui::PushFont(ImGui::GetFont());

const auto titleSize = ImGui::CalcTextSize(text);
const ImVec2 &vpSize = ImGui::GetMainViewport()->Size;
ImGui::GetWindowDrawList()->AddText(ImVec2((vpSize.x - titleSize.x) / 2.0f, vpSize.y * y), color, text);

ImGui::GetFont()->Scale = oldScale;
ImGui::PopFont();
}

protected:
int m_PageIndex = 0;
int m_PageCount = 1;
};

class Menu {
public:
Menu() = default;

bool AddPage(Page *page) {
if (!page)
return false;

const auto &name = page->GetName();
auto it = m_PageMap.find(name);
if (it != m_PageMap.end())
return false;

m_PageMap[name] = page;
return true;
}

bool RemovePage(Page *page) {
if (!page)
return false;

const auto &name = page->GetName();
auto it = m_PageMap.find(name);
if (it == m_PageMap.end())
return false;

HidePage();

while (!m_PageStack.empty())
m_PageStack.pop();

m_PageMap.erase(it);
return true;
}

Page *GetPage(const std::string &name) {
auto it = m_PageMap.find(name);
if (it == m_PageMap.end())
return nullptr;
return m_PageMap[name];
}

bool ShowPage(const std::string &name) {
auto *page = GetPage(name);
if (!page)
return false;

PushPage(m_CurrentPage);
m_CurrentPage = page;
m_CurrentPage->Show();
return true;
}
void ShowPrevPage() {
HidePage();
auto *page = PopPage();
m_CurrentPage = page;
if (m_CurrentPage)
m_CurrentPage->Show();
else
OnClose();
}

void HidePage() {
if (m_CurrentPage) {
m_CurrentPage->Hide();
m_CurrentPage = nullptr;
}
}

void Render() {
if (m_CurrentPage)
m_CurrentPage->Render();
}

void Open(const std::string &name) {
if (ShowPage(name))
OnOpen();
}

void Close() {
ShowPrevPage();
HidePage();
OnClose();
}

virtual void OnOpen() = 0;
virtual void OnClose() = 0;

protected:
void PushPage(Page *page) {
if (page)
m_PageStack.push(page);
}
Page *PopPage() {
if (m_PageStack.empty())
return nullptr;
auto *page = m_PageStack.top();
m_PageStack.pop();
return page;
}

Page *m_CurrentPage = nullptr;
std::stack<Page *> m_PageStack;
std::unordered_map<std::string, Page *> m_PageMap;
};
}

#endif // BML_BUI_H
Loading

0 comments on commit d99fe70

Please sign in to comment.