MRRibbonMenu.h

Go to the documentation of this file.

#pragma once
#include "ImGuiMenu.h"
#include "MRRibbonMenuItem.h"
#include "MRRibbonMenuSearch.h"
#include "MRRibbonFontManager.h"
#include "MRRibbonButtonDrawer.h"
#include "MRAsyncTimer.h"
#include "MRRibbonSchema.h"
#include "MRToolbar.h"
#include "MRMesh/MRSignal.h"
#include "MRRibbonNotification.h"
#include <type_traits>
#include <array>
 
namespace MR
{
 
class Object;
 
 
enum class RibbonLayoutMode
{
    None, // no menu present
    SceneTree, // only scene tree is present
    All // both scene tree and top panel are present
};
 
// Class to control and render ribbon-style menu
// stores menu items,
// menu structure is provided by `menuItemsStructure.json` file (parsed on init)
class MRVIEWER_CLASS RibbonMenu : public ImGuiMenu
{
    struct CustomContextMenuCheckbox
    {
        using Setter = std::function<void( std::shared_ptr<Object> object, ViewportId id, bool checked )>;
        using Getter = std::function<bool( std::shared_ptr<Object> object, ViewportId id )>;
        Setter setter;
        Getter getter;
        // display a checkBox when only these types of objects are selected
        // by default, it is always hidden
        SelectedTypesMask selectedMask = SelectedTypesMask( -1 );
    };
 
public:
    // adds a custom checkBox to the context menu
    // it is applied to the selected objects
    MRVIEWER_API void setCustomContextCheckbox(
        const std::string& name,
        CustomContextMenuCheckbox customContextMenuCheckbox );
 
    MRVIEWER_API virtual void init( MR::Viewer* _viewer ) override;
 
    MRVIEWER_API virtual void shutdown() override;
 
    MRVIEWER_API void openToolbarCustomize();
 
    MRVIEWER_API virtual void load_font( int font_size = 13 ) override;
 
    MRVIEWER_API virtual std::filesystem::path getMenuFontPath() const override;
 
    // set top panel pinned and unpinned
    MRVIEWER_API virtual void pinTopPanel( bool on );
    MRVIEWER_API bool isTopPannelPinned() const;
    // this functions allow you to get top panel height (before scaling)
    int getTopPanelOpenedHeight() const { return topPanelOpenedHeight_; }
    int getTopPanelHiddenHeight() const { return topPanelHiddenHeight_; }
    int getTopPanelCurrentHeight() const { return currentTopPanelHeight_; }
    // set maximum wait time (in seconds) before top panel is closed after mouse leaves it (only when not pinned)
    // minimum value is 0 seconds, panel will close immediately after mouse leaves it
    void setTopPanelMaxOpenedTimer( float sec ) { openedMaxSecs_ = std::max( 0.0f, sec ); }
 
    MRVIEWER_API virtual void setQuickAccessListVersion( int version );
    MRVIEWER_API virtual void readQuickAccessList( const Json::Value& root );
 
    MRVIEWER_API void resetQuickAccessList();
 
    Vector2i getSceneSize() { return Vector2i( int( sceneSize_.x ), int( sceneSize_.y ) ); };
 
    MRVIEWER_API void setSceneSize( const Vector2i& size );
 
    bool hasActiveBlockingItem() const { return bool( activeBlockingItem_.item ); }
    bool hasAnyActiveItem() const { return bool( activeBlockingItem_.item ) || !activeNonBlockingItems_.empty(); }
 
    MRVIEWER_API void updateItemStatus( const std::string& itemName );
 
    int getActiveTabIndex() const { return activeTabIndex_; }
 
 
    RibbonFontManager& getFontManager() { return fontManager_; };
 
    RibbonButtonDrawer& getRibbonButtonDrawer() { return buttonDrawer_; }
 
    Toolbar& getToolbar() { return toolbar_; }
 
    RibbonNotifier& getRibbonNotifier() { return notifier_; };
 
    void setActiveListPos( const ImVec2& pos ) { activeListPos_ = pos; }
    
    void showActiveList() { activeListPressed_ = true; };
 
    MRVIEWER_API void pushNotification( const RibbonNotification& notification );
 
    MRVIEWER_API static void cloneTree( const std::vector<std::shared_ptr<Object>>& selectedObjects );
    MRVIEWER_API static void cloneSelectedPart( const std::shared_ptr<Object>& object );
 
    using TabChangedSignal = boost::signals2::signal<void( int prevTabId, int newTabId )>;
    TabChangedSignal tabChangedSignal;
 
    bool getAutoCloseBlockingPlugins() const { return autoCloseBlockingPlugins_; }
    void setAutoCloseBlockingPlugins( bool value ) { autoCloseBlockingPlugins_ = value; }
 
    RibbonLayoutMode getLayoutMode() const { return layoutMode_; }
    MRVIEWER_API virtual void setLayoutMode( RibbonLayoutMode mode );
 
    // ======== selected objects options drawing
    bool drawGroupUngroupButton( const std::vector<std::shared_ptr<Object>>& selected );
    bool drawSelectSubtreeButton( const std::vector<std::shared_ptr<Object>>& selected );
    bool drawCloneButton( const std::vector<std::shared_ptr<Object>>& selected );
    bool drawCustomCheckBox( const std::vector<std::shared_ptr<Object>>& selected, SelectedTypesMask selectedMask );
    bool drawCloneSelectionButton( const std::vector<std::shared_ptr<Object>>& selected );
    bool drawMergeSubtreeButton( const std::vector<std::shared_ptr<Object>>& selected );
protected:
    // draw single item
    MRVIEWER_API virtual void drawBigButtonItem_( const MenuItemInfo& item );
    // draw set of small text buttons
    MRVIEWER_API virtual void drawSmallButtonsSet_( const std::vector<std::string>& group, int setFrontIndex, int setLength,
                                                    bool withText );
 
    // Configuration of ribbon group
    struct DrawGroupConfig
    {
        int numBig{ 0 };
        int numSmallText{ 0 };
        int numSmall{ 0 };
    };
    using DrawTabConfig = std::vector<DrawGroupConfig>;
    // draw group of items
    MRVIEWER_API virtual DrawTabConfig setupItemsGroupConfig_( const std::vector<std::string>& groupsInTab, const std::string& tabName, bool centerItems );
    MRVIEWER_API virtual void setupItemsGroup_( const std::vector<std::string>& groupsInTab, const std::string& tabName );
    MRVIEWER_API virtual void drawItemsGroup_( const std::string& tabName, const std::string& groupName,
                                               DrawGroupConfig config );
    // ribbon menu item pressed
    MRVIEWER_API virtual void itemPressed_( const std::shared_ptr<RibbonMenuItem>& item, bool available );
    // this function is called when ribbon item is activated
    // this implementation only updates `searcher_`
    MRVIEWER_API virtual void onItemActivated_( const std::shared_ptr<RibbonMenuItem>& item );
 
    MRVIEWER_API virtual void drawActiveBlockingDialog_();
    MRVIEWER_API virtual void drawActiveNonBlockingDialogs_();
 
    MRVIEWER_API virtual void postResize_( int width, int height ) override;
    MRVIEWER_API virtual void postRescale_( float x, float y ) override;
 
    struct DialogItemPtr
    {
        std::shared_ptr<RibbonMenuItem> item;
        // this flag is needed to correctly fix position of UI dialog (temporary, while plugin dialogs are floating in viewport)
        bool dialogPositionFixed{ false };
    };
    DialogItemPtr activeBlockingItem_;
    std::vector<DialogItemPtr> activeNonBlockingItems_;
    MRVIEWER_API virtual void drawItemDialog_( DialogItemPtr& itemPtr );
 
    // Draw ribbon top panel
    MRVIEWER_API virtual void drawTopPanel_( bool drawTabs = true, bool centerItems = false );
    // Draw scene list window with content
    MRVIEWER_API virtual void drawRibbonSceneList_();
    // Draw vertical line at the right border of scene to enable resize of scene list
    // returns size of scene window
    MRVIEWER_API virtual Vector2f drawRibbonSceneResizeLine_();
    // Draw viewport id and projection type for all viewporrts
    MRVIEWER_API virtual void drawRibbonViewportsLabels_();
 
    MRVIEWER_API virtual void drawRibbonSceneInformation_( const std::vector<std::shared_ptr<Object>>& selected );
 
    MRVIEWER_API virtual bool drawCollapsingHeaderTransform_() override;
    MRVIEWER_API virtual bool drawTransformContextMenu_( const std::shared_ptr<Object>& selected ) override;
 
    MRVIEWER_API virtual void addRibbonItemShortcut_( const std::string& itemName, const ShortcutKey& key, ShortcutCategory category );
 
    MRVIEWER_API virtual void setupShortcuts_() override;
 
    MRVIEWER_API virtual void drawShortcutsWindow_() override;
    // reads files with panel description
    MRVIEWER_API virtual void readMenuItemsStructure_();
 
    std::vector<std::shared_ptr<const Object>> prevFrameSelectedObjectsCache_;
 
    MRVIEWER_API virtual bool drawCollapsingHeader_( const char* label, ImGuiTreeNodeFlags flags = 0 ) override;
 
    MRVIEWER_API virtual void highlightBlocking_();
 
    // draw scene list buttons
    MRVIEWER_API virtual void drawSceneListButtons_();
 
    // updates viewport sizes with respect to ribbon top and left panels
    MRVIEWER_API virtual void fixViewportsSize_( int w, int h );
    
    // need to be called if you override windows pipeline and use ActiveListPlugin
    MRVIEWER_API void drawActiveList_();
 
    RibbonLayoutMode layoutMode_{ RibbonLayoutMode::All };
private:
    void changeTab_( int newTab );
 
    std::string getRequirements_( const std::shared_ptr<RibbonMenuItem>& item ) const;
 
    RibbonMenuSearch searcher_;
    void drawSearchButton_();
    void drawCollapseButton_();
    void drawHelpButton_();
 
    void sortObjectsRecursive_( std::shared_ptr<Object> object );
 
    // part of top panel
    void drawHeaderQuickAccess_();
    void drawHeaderPannel_();
    void drawActiveListButton_( float btnSize );
 
    ImVec2 activeListPos_{ 0,0 };
    bool activeListPressed_{ false };
 
    void beginTopPanel_();
    void endTopPanel_();
    void drawTopPanelOpened_( bool drawTabs, bool centerItems );
 
    std::string transformClipboardText_;
 
    int currentTopPanelHeight_ = 113;
    int topPanelOpenedHeight_ = 113;
    int topPanelHiddenHeight_ = 33;
 
    ImVec2 sceneSize_{ 310, 0 };
    float informationHeight_{ 0.f };
    float transformHeight_{ 0.f };
 
    // how long blocking window will blink in seconds
    float blockingHighlightTimer_{ 0.0f };
 
    // current scroll position of tabs panel
    float tabPanelScroll_{ 0.0f };
 
    bool autoCloseBlockingPlugins_{ true };
 
    enum class CollapseState
    {
        Closed,
        Opened,
        Pinned
    } collapseState_{ CollapseState::Pinned };
    // seconds to stay opened if not pinned
    float openedMaxSecs_{ 2.0f };
    float openedTimer_{ openedMaxSecs_ };
 
    int activeTabIndex_{ 0 };
    RibbonFontManager fontManager_;
    RibbonButtonDrawer buttonDrawer_;
 
    std::unordered_map<std::string, CustomContextMenuCheckbox> customCheckBox_;
 
    Toolbar toolbar_;
    RibbonNotifier notifier_;
#ifndef __EMSCRIPTEN__
    AsyncRequest asyncRequest_;
#endif // !__EMSCRIPTEN__
};
 
// Checks if RibbonMenu is available, if it is - forwards notification to RibbonNotifier. Otherwise - calls showModal() function
MRVIEWER_API void pushNotification( const RibbonNotification& notification );
 
}