menu_builder.h

Go to the documentation of this file.

#ifndef YAZE_APP_EDITOR_MENU_MENU_BUILDER_H_
#define YAZE_APP_EDITOR_MENU_MENU_BUILDER_H_
 
#include <functional>
#include <string>
#include <vector>
 
// Must define before including imgui.h
#ifndef IMGUI_DEFINE_MATH_OPERATORS
#define IMGUI_DEFINE_MATH_OPERATORS
#endif
 
#include "absl/container/flat_hash_map.h"
#include "absl/strings/string_view.h"
#include "app/gui/core/icons.h"
#include "imgui/imgui.h"
 
namespace yaze {
namespace editor {
 
class MenuBuilder {
  friend class ProjectUiTestPeer;
 
 public:
  using Callback = std::function<void()>;
  using EnabledCheck = std::function<bool()>;
 
  MenuBuilder() = default;
 
  MenuBuilder& BeginMenu(const char* label, const char* icon = nullptr);
 
  MenuBuilder& BeginSubMenu(const char* label, const char* icon = nullptr,
                            EnabledCheck enabled = nullptr);
 
  MenuBuilder& EndMenu();
 
  MenuBuilder& Item(const char* label, const char* icon, Callback callback,
                    const char* shortcut = nullptr,
                    EnabledCheck enabled = nullptr,
                    EnabledCheck checked = nullptr);
 
  MenuBuilder& Item(const char* label, Callback callback,
                    const char* shortcut = nullptr,
                    EnabledCheck enabled = nullptr);
 
  MenuBuilder& Separator();
 
  MenuBuilder& DisabledItem(const char* label, const char* icon = nullptr);
 
  MenuBuilder& CustomMenu(const char* label, Callback draw_callback);
 
  MenuBuilder& CustomItems(Callback draw_callback);
 
  void Draw();
 
  void Clear();
 
 private:
  struct MenuItem {
    enum class Type {
      kItem,
      kSubMenuBegin,
      kSubMenuEnd,
      kSeparator,
      kDisabled,
      kCustom
    };
 
    Type type;
    std::string label;
    std::string icon;
    std::string shortcut;
    Callback callback;
    EnabledCheck enabled;
    EnabledCheck checked;
  };
 
  struct Menu {
    std::string label;
    std::string icon;
    std::vector<MenuItem> items;
    Callback custom_draw;  // For custom menus with dynamic content
    bool is_custom = false;
  };
 
  std::vector<Menu> menus_;
  Menu* current_menu_ = nullptr;
 
  // Track which submenus are actually open during drawing
  mutable std::vector<bool> submenu_stack_;
  mutable int skip_depth_ =
      0;  // Track nesting depth when skipping closed submenus
 
  void DrawMenuItem(const MenuItem& item);
};
 
}  // namespace editor
}  // namespace yaze
 
#endif  // YAZE_APP_EDITOR_MENU_MENU_BUILDER_H_