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 "MRRibbonMenuUIConfig.h"
#include "MRMesh/MRSignal.h"
#include "MRRibbonNotification.h"
#include <type_traits>
#include <array>
 
namespace MR
{
 
 
class Object;

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;
        SelectedTypesMask selectedMask = SelectedTypesMask( -1 );
    };
 
public:
    MRVIEWER_API RibbonMenu();
    MRVIEWER_API ~RibbonMenu();

    MRVIEWER_API static std::shared_ptr<RibbonMenu> instance();

    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;
 
    virtual void drawViewerWindow() override {}
 
    MRVIEWER_API  virtual void drawAdditionalWindows() override;

    MRVIEWER_API void openToolbarCustomize();
 
    MRVIEWER_API virtual void loadFonts( int font_size = 13 ) override;
 
    MRVIEWER_API virtual std::filesystem::path getMenuFontPath() const override;

    MRVIEWER_API virtual void pinTopPanel( bool on );
    MRVIEWER_API bool isTopPannelPinned() const;
    int getTopPanelOpenedHeight() const { return topPanelOpenedHeight_; }
    int getTopPanelHiddenHeight() const { return topPanelHiddenHeight_; }
    int getTopPanelCurrentHeight() const { return currentTopPanelHeight_; }
    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 );

    MRVIEWER_API virtual void addRibbonItemShortcut( const std::string& itemName, const Shortcut& shortcut );

    [[deprecated( "use addRibbonItemShortcut( itemName, Shortcut )" )]]
    void addRibbonItemShortcut( const std::string& itemName, const ShortcutKey& key, ShortcutCategory category )
        { addRibbonItemShortcut( itemName, { key, category } ); }

    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 virtual 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; }

    const RibbonMenuUIConfig& getMenuUIConfig() const { return menuUIConfig_; }
    MRVIEWER_API virtual void setMenuUIConfig( const RibbonMenuUIConfig& newConfig );

    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:
    MRVIEWER_API virtual void drawBigButtonItem_( const MenuItemInfo& item );
    MRVIEWER_API virtual void drawSmallButtonsSet_( const std::vector<std::string>& group, int setFrontIndex, int setLength,
                                                    bool withText );

    struct DrawGroupConfig
    {
        int numBig{ 0 };
        int numSmallText{ 0 };
        int numSmall{ 0 };
    };
    using DrawTabConfig = std::vector<DrawGroupConfig>;
    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, bool centerItems );
    MRVIEWER_API virtual void drawItemsGroup_( const std::string& tabName, const std::string& groupName,
                                               DrawGroupConfig config );
    MRVIEWER_API virtual bool itemPressed_( const std::shared_ptr<RibbonMenuItem>& item, const std::string& requiremetnsHint = {} );

    MRVIEWER_API virtual std::string getRequirements_( const std::shared_ptr<RibbonMenuItem>& item ) const;
 
    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;
    };
    DialogItemPtr activeBlockingItem_;
    std::vector<DialogItemPtr> activeNonBlockingItems_;
    MRVIEWER_API virtual void drawItemDialog_( DialogItemPtr& itemPtr );

    MRVIEWER_API virtual void drawTopPanel_( bool drawTabs = true, bool centerItems = false );
    MRVIEWER_API virtual void drawRibbonSceneList_();
    MRVIEWER_API virtual Vector2f drawRibbonSceneResizeLine_();
    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::vector<std::shared_ptr<Object>>& selected ) override;
 
    MRVIEWER_API virtual void setupShortcuts_() override;

    MRVIEWER_API virtual HashSet<std::string> allowedShortcutTags_() const;

    MRVIEWER_API void registerItemsShortcuts_( const HashSet<std::string>& allowedTags );
 
    MRVIEWER_API virtual void drawShortcutsWindow_() override;
    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_();

    MRVIEWER_API virtual void drawSceneListButtons_();

    MRVIEWER_API virtual void fixViewportsSize_( int w, int h );

    MRVIEWER_API void drawActiveList_();

    MRVIEWER_API virtual void drawNotifications_();

    MRVIEWER_API virtual void updateTopPanelSize_( bool drawTabs );

    MRVIEWER_API virtual void drawHeaderQuickAccess_();

    MRVIEWER_API virtual float drawHeaderHelpers_( float requiredTabSize );

    MRVIEWER_API virtual void drawActiveListButton_( float btnSize );
    MRVIEWER_API virtual void drawSearchButton_();
    MRVIEWER_API virtual void drawCollapseButton_();
    MRVIEWER_API virtual void drawHelpButton_( const std::string& url );
    MRVIEWER_API virtual void drawLanguageButton_();
 
    RibbonMenuSearch searcher_;
 
private:
    void changeTab_( int newTab );
 
    void sortObjectsRecursive_( std::shared_ptr<Object> object );

    void drawHeaderPannel_();
 
    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 };
 
    RibbonMenuUIConfig menuUIConfig_;

    float blockingHighlightTimer_{ 0.0f };

    float tabPanelScroll_{ 0.0f };
 
    bool autoCloseBlockingPlugins_{ true };
 
    enum class CollapseState
    {
        Closed,
        Opened,
        Pinned
    } collapseState_{ CollapseState::Pinned };
    float openedMaxSecs_{ 2.0f };
    float openedTimer_{ openedMaxSecs_ };
 
    int activeTabIndex_{ 0 };
    RibbonFontManager fontManager_;
    RibbonButtonDrawer buttonDrawer_;
 
    std::unordered_map<std::string, CustomContextMenuCheckbox> customCheckBox_;
 
    std::unique_ptr<Toolbar> toolbar_;
    RibbonNotifier notifier_;
#ifndef __EMSCRIPTEN__
    AsyncRequest asyncRequest_;
#endif // !__EMSCRIPTEN__
};

MRVIEWER_API void pushNotification( const RibbonNotification& notification );
 
}