Add right click signal to TreeView

texus · Feb 15, 2020 · 15ef22f · 15ef22f
1 parent 5ee202f
commit 15ef22f
Show file tree

Hide file tree

Showing 4 changed files with 45 additions and 2 deletions.
diff --git a/include/TGUI/Signal.hpp b/include/TGUI/Signal.hpp
@@ -207,6 +207,7 @@ namespace tgui
             static constexpr const char* const DoubleClicked = "DoubleClicked"; ///< A leaf node was double clicked. Optional parameter: selected node
             static constexpr const char* const Expanded = "Expanded";           ///< A branch node was expanded in the tree view. Optional parameter: expanded node
             static constexpr const char* const Collapsed = "Collapsed";         ///< A branch node was collapsed in the tree view. Optional parameter: collapsed node
+            static constexpr const char* const RightClicked = "RightClicked";   ///< A node was right clicked. Optional parameter: node below mouse
         };
 
         using BitmapButton = Button;

diff --git a/include/TGUI/Widgets/TreeView.hpp b/include/TGUI/Widgets/TreeView.hpp
@@ -291,7 +291,12 @@ namespace tgui
         /// @internal
         /////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
         void leftMouseReleased(Vector2f pos) override;
-
+
+        /////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
+        /// @internal
+        /////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
+        void rightMousePressed(Vector2f pos) override;
+
         /////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
         /// @internal
         /////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
@@ -459,7 +464,7 @@ namespace tgui
         SignalItemHierarchy onDoubleClick = {"DoubleClicked"}; ///< A leaf node was double clicked. Optional parameter: selected node
         SignalItemHierarchy onExpand = {"Expanded"};           ///< A branch node was expanded in the tree view. Optional parameter: expanded node
         SignalItemHierarchy onCollapse = {"Collapsed"};        ///< A branch node was collapsed in the tree view. Optional parameter: collapsed node
-
+        SignalItemHierarchy onRightClick = {"RightClicked"};   ///< A node was right clicked. Optional parameter: node below mouse
 
         /////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
     protected:

diff --git a/src/TGUI/Signal.cpp b/src/TGUI/Signal.cpp
@@ -122,6 +122,7 @@ namespace tgui
         constexpr const char* const TreeView::DoubleClicked;
         constexpr const char* const TreeView::Expanded;
         constexpr const char* const TreeView::Collapsed;
+        constexpr const char* const TreeView::RightClicked;
     }
 
     /////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

diff --git a/src/TGUI/Widgets/TreeView.cpp b/src/TGUI/Widgets/TreeView.cpp
@@ -259,6 +259,7 @@ namespace tgui
         onDoubleClick                       {other.onDoubleClick},
         onExpand                            {other.onExpand},
         onCollapse                          {other.onCollapse},
+        onRightClick                        {other.onRightClick},
         m_selectedItem                      {other.m_selectedItem},
         m_hoveredItem                       {other.m_hoveredItem},
         m_itemHeight                        {other.m_itemHeight},
@@ -307,6 +308,7 @@ namespace tgui
             std::swap(onDoubleClick,                        temp.onDoubleClick);
             std::swap(onExpand,                             temp.onExpand);
             std::swap(onCollapse,                           temp.onCollapse);
+            std::swap(onRightClick,                         temp.onRightClick);
             std::swap(m_nodes,                              temp.m_nodes);
             std::swap(m_visibleNodes,                       temp.m_visibleNodes);
             std::swap(m_selectedItem,                       temp.m_selectedItem);
@@ -758,6 +760,38 @@ namespace tgui
         m_horizontalScrollbar->leftMouseReleased(childPos);
     }
 
+    /////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
+
+    void TreeView::rightMousePressed(Vector2f pos)
+    {
+        pos -= getPosition();
+
+        m_mouseDown = true;
+
+        float maxItemWidth = getInnerSize().x - m_paddingCached.getLeft() - m_paddingCached.getRight();
+        if (m_verticalScrollbar->isShown())
+            maxItemWidth -= m_verticalScrollbar->getSize().x;
+
+        if (FloatRect{m_bordersCached.getLeft() + m_paddingCached.getLeft(), m_bordersCached.getTop() + m_paddingCached.getTop(),
+                      maxItemWidth, getInnerSize().y - m_paddingCached.getTop() - m_paddingCached.getBottom()}.contains(pos))
+        {
+            pos.y -= m_bordersCached.getTop() + m_paddingCached.getTop();
+            int selectedItem = static_cast<int>(((pos.y - (m_itemHeight - (m_verticalScrollbar->getValue() % m_itemHeight))) / m_itemHeight) + (m_verticalScrollbar->getValue() / m_itemHeight) + 1);
+            if (selectedItem < static_cast<int>(m_visibleNodes.size()))
+            {
+                std::vector<sf::String> hierarchy;
+                auto* node = m_visibleNodes[selectedItem].get();
+                while (node)
+                {
+                    hierarchy.insert(hierarchy.begin(), node->text.getString());
+                    node = node->parent;
+                }
+
+                onRightClick.emit(this, hierarchy.back(), hierarchy);
+            }
+        }
+    }
+
     /////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
 
     void TreeView::mouseMoved(Vector2f pos)
@@ -854,6 +888,8 @@ namespace tgui
             return onExpand;
         else if (signalName == toLower(onCollapse.getName()))
             return onCollapse;
+        else if (signalName == toLower(onRightClick.getName()))
+            return onRightClick;
         else
             return Widget::getSignal(std::move(signalName));
     }