add frozen menu flag

Author: g-abilio

examples/calculator/RandomNumber.hpp

@@ -15,6 +15,10 @@
 class RandomNumberModel : public MathOperationDataModel
 {
 public:
+    RandomNumberModel() {
+        setFrozenMenu(true);
+    }
+
     virtual ~RandomNumberModel() {}
 
 public:

include/QtNodes/internal/DataFlowGraphicsScene.hpp

@@ -24,15 +24,7 @@ class NODE_EDITOR_PUBLIC DataFlowGraphicsScene : public BasicGraphicsScene
 public:
     std::vector<NodeId> selectedNodes() const;
     QMenu *createSceneMenu(QPointF const scenePos) override;
-    void updateConnectionGraphics(const std::unordered_set<ConnectionId> &connections, bool state)
-    {
-        for (auto const &c : connections) {
-            if (auto *cgo = connectionGraphicsObject(c)) {
-                cgo->connectionState().setFrozen(state);
-                cgo->update();
-            }
-        }
-    }
+    void updateConnectionGraphics(const std::unordered_set<ConnectionId> &connections, bool state);
 
 public Q_SLOTS:
     bool save() const;

include/QtNodes/internal/NodeDelegateModel.hpp

@@ -5,6 +5,7 @@
 #include "NodeData.hpp"
 #include "NodeStyle.hpp"
 #include "Serializable.hpp"
+#include <QtGui/QColor>
 
 #include <QtWidgets/QWidget>
 
@@ -80,6 +81,10 @@ class NODE_EDITOR_PUBLIC NodeDelegateModel
 
     void setFrozenState(bool state) { _frozen = state; }
 
+    bool frozenMenu() const { return _frozenMenu; }
+
+    void setFrozenMenu(bool state) { _frozenMenu = state; }
+
 public Q_SLOTS:
     virtual void inputConnectionCreated(ConnectionId const &) {}
     virtual void inputConnectionDeleted(ConnectionId const &) {}
@@ -124,6 +129,8 @@ public Q_SLOTS:
     NodeStyle _nodeStyle;
 
     bool _frozen{false};
+
+    bool _frozenMenu{false};
 };
 
 } // namespace QtNodes

src/DataFlowGraphicsScene.cpp

@@ -190,4 +190,15 @@ bool DataFlowGraphicsScene::load()
     return true;
 }
 
+void DataFlowGraphicsScene::updateConnectionGraphics(
+    const std::unordered_set<ConnectionId> &connections, bool state)
+{
+    for (auto const &c : connections) {
+        if (auto *cgo = connectionGraphicsObject(c)) {
+            cgo->connectionState().setFrozen(state);
+            cgo->update();
+        }
+    }
+}
+
 } // namespace QtNodes

src/GraphicsView.cpp

@@ -2,6 +2,8 @@
 
 #include "BasicGraphicsScene.hpp"
 #include "ConnectionGraphicsObject.hpp"
+#include "DataFlowGraphModel.hpp"
+#include "NodeDelegateModel.hpp"
 #include "NodeGraphicsObject.hpp"
 #include "StyleCollection.hpp"
 #include "UndoCommands.hpp"
@@ -23,7 +25,10 @@
 #include <cmath>
 
 using QtNodes::BasicGraphicsScene;
+using QtNodes::DataFlowGraphModel;
 using QtNodes::GraphicsView;
+using QtNodes::NodeDelegateModel;
+using QtNodes::NodeGraphicsObject;
 
 GraphicsView::GraphicsView(QWidget *parent)
     : QGraphicsView(parent)
@@ -167,11 +172,21 @@ void GraphicsView::centerScene()
 void GraphicsView::contextMenuEvent(QContextMenuEvent *event)
 {
     QGraphicsView::contextMenuEvent(event);
-    QMenu *menu;
+    QMenu *menu = nullptr;
 
-    if (itemAt(event->pos())) {
+    bool isFrozenMenu;
+
+    if (auto *dfModel = dynamic_cast<DataFlowGraphModel *>(&nodeScene()->graphModel())) {
+        if (auto n = qgraphicsitem_cast<NodeGraphicsObject *>(itemAt(event->pos()))) {
+            if (auto *delegate = dfModel->delegateModel<NodeDelegateModel>(n->nodeId())) {
+                isFrozenMenu = delegate->frozenMenu();
+            }
+        }
+    }
+
+    if (itemAt(event->pos()) && isFrozenMenu) {
         menu = nodeScene()->createFreezeMenu();
-    } else {
+    } else if (!itemAt(event->pos())) {
         menu = nodeScene()->createSceneMenu(mapToScene(event->pos()));
     }