Adds freeze and unfreeze functionality

examples/calculator/MathOperationDataModel.cpp

@@ -1,5 +1,4 @@
 #include "MathOperationDataModel.hpp"
-
 #include "DecimalData.hpp"
 
 unsigned int MathOperationDataModel::nPorts(PortType portType) const
@@ -21,7 +20,9 @@ NodeDataType MathOperationDataModel::dataType(PortType, PortIndex) const
 
 std::shared_ptr<NodeData> MathOperationDataModel::outData(PortIndex)
 {
-    return std::static_pointer_cast<NodeData>(_result);
+    auto output = std::static_pointer_cast<NodeData>(_result);
+
+    return output;
 }
 
 void MathOperationDataModel::setInData(std::shared_ptr<NodeData> data, PortIndex portIndex)

examples/calculator/MathOperationDataModel.hpp

@@ -1,7 +1,6 @@
 #pragma once
 
 #include <QtNodes/NodeDelegateModel>
-
 #include <QtCore/QJsonObject>
 #include <QtCore/QObject>
 #include <QtWidgets/QLabel>

include/QtNodes/internal/BasicGraphicsScene.hpp

@@ -5,7 +5,6 @@
 #include "ConnectionIdHash.hpp"
 #include "Definitions.hpp"
 #include "Export.hpp"
-
 #include "QUuidStdHash.hpp"
 
 #include <QtCore/QUuid>
@@ -28,7 +27,7 @@ class ConnectionGraphicsObject;
 class NodeGraphicsObject;
 class NodeStyle;
 
-/// An instance of QGraphicsScene, holds connections and nodes.
+/// An instance of QGraphicsScene , holds connections and nodes.
 class NODE_EDITOR_PUBLIC BasicGraphicsScene : public QGraphicsScene
 {
     Q_OBJECT
@@ -111,6 +110,8 @@ class NODE_EDITOR_PUBLIC BasicGraphicsScene : public QGraphicsScene
      */
     virtual QMenu *createSceneMenu(QPointF const scenePos);
 
+    void freezeModelAndConnections(bool isFreeze);
+
 Q_SIGNALS:
     void modified(BasicGraphicsScene *);
     void nodeMoved(NodeId const nodeId, QPointF const &newLocation);

include/QtNodes/internal/ConnectionState.hpp

@@ -25,6 +25,7 @@ class NODE_EDITOR_PUBLIC ConnectionState
     ConnectionState(ConnectionGraphicsObject &cgo)
         : _cgo(cgo)
         , _hovered(false)
+        , _frozen(false)
     {}
 
     ConnectionState(ConnectionState const &) = delete;
@@ -42,6 +43,9 @@ class NODE_EDITOR_PUBLIC ConnectionState
     bool hovered() const;
     void setHovered(bool hovered);
 
+    bool frozen() const { return _frozen; }
+    void setFrozen(bool frozen) { _frozen = frozen; }
+
 public:
     /// Caches NodeId for further interaction.
     void setLastHoveredNode(NodeId const nodeId);
@@ -56,5 +60,7 @@ class NODE_EDITOR_PUBLIC ConnectionState
     bool _hovered;
 
     NodeId _lastHoveredNode{InvalidNodeId};
+
+    bool _frozen;
 };
 } // namespace QtNodes

include/QtNodes/internal/DataFlowGraphicsScene.hpp

@@ -1,8 +1,10 @@
 #pragma once
 
 #include "BasicGraphicsScene.hpp"
+#include "ConnectionGraphicsObject.hpp"
 #include "DataFlowGraphModel.hpp"
 #include "Export.hpp"
+#include "NodeConnectionInteraction.hpp"
 
 namespace QtNodes {
 
@@ -22,6 +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);
 
 public Q_SLOTS:
     bool save() const;

include/QtNodes/internal/NodeDelegateModel.hpp

@@ -12,6 +12,7 @@
 #include "NodeData.hpp"
 #include "NodeStyle.hpp"
 #include "Serializable.hpp"
+#include <QtGui/QColor>
 
 namespace QtNodes {
 
@@ -132,6 +133,10 @@ class NODE_EDITOR_PUBLIC NodeDelegateModel
 
     virtual bool resizable() const { return false; }
 
+    bool frozen() const { return _frozen; }
+
+    void setFrozenState(bool state) { _frozen = state; }
+
 public Q_SLOTS:
     virtual void inputConnectionCreated(ConnectionId const &) {}
     virtual void inputConnectionDeleted(ConnectionId const &) {}
@@ -185,6 +190,8 @@ public Q_SLOTS:
 private:
     NodeStyle _nodeStyle;
 
+    bool _frozen{false};
+
     NodeValidationState _nodeValidationState;
 
     NodeProcessingStatus _processingStatus{NodeProcessingStatus::NoStatus};

src/BasicGraphicsScene.cpp

@@ -3,10 +3,12 @@
 #include "AbstractNodeGeometry.hpp"
 #include "ConnectionGraphicsObject.hpp"
 #include "ConnectionIdUtils.hpp"
+#include "DataFlowGraphModel.hpp"
 #include "DefaultConnectionPainter.hpp"
 #include "DefaultHorizontalNodeGeometry.hpp"
 #include "DefaultNodePainter.hpp"
 #include "DefaultVerticalNodeGeometry.hpp"
+#include "NodeDelegateModel.hpp"
 #include "NodeGraphicsObject.hpp"
 
 #include <QUndoStack>
@@ -331,4 +333,31 @@ void BasicGraphicsScene::onModelReset()
     traverseGraphAndPopulateGraphicsObjects();
 }
 
+void BasicGraphicsScene::freezeModelAndConnections(bool isFreeze)
+{
+    for (QGraphicsItem *item : selectedItems()) {
+        if (auto n = qgraphicsitem_cast<NodeGraphicsObject *>(item)) {
+            int portCount = graphModel().nodeData(n->nodeId(), NodeRole::OutPortCount).toInt();
+            for (int i = 0; i < portCount; i++) {
+                auto graphConnections = graphModel().connections(n->nodeId(),
+                                                                 QtNodes::PortType::Out,
+                                                                 QtNodes::PortIndex(i));
+
+                for (auto const &c : graphConnections) {
+                    if (auto *cgo = connectionGraphicsObject(c)) {
+                        cgo->connectionState().setFrozen(isFreeze);
+                        cgo->update();
+                    }
+                }
+            }
+
+            if (auto *dfModel = dynamic_cast<DataFlowGraphModel *>(&graphModel())) {
+                if (auto *delegate = dfModel->delegateModel<NodeDelegateModel>(n->nodeId())) {
+                    delegate->setFrozenState(isFreeze);
+                }
+            }
+        }
+    }
+}
+
 } // namespace QtNodes

src/DataFlowGraphModel.cpp

@@ -444,6 +444,9 @@ bool DataFlowGraphModel::setPortData(
     switch (role) {
     case PortRole::Data:
         if (portType == PortType::In) {
+            if (model->frozen())
+                return false;
+
             model->setInData(value.value<std::shared_ptr<NodeData>>(), portIndex);
 
             // Triggers repainting on the scene.

src/DataFlowGraphicsScene.cpp

@@ -26,7 +26,6 @@
 #include <stdexcept>
 #include <utility>
 
-
 namespace QtNodes {
 
 DataFlowGraphicsScene::DataFlowGraphicsScene(DataFlowGraphModel &graphModel, QObject *parent)
@@ -191,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/DefaultConnectionPainter.cpp

@@ -9,7 +9,6 @@
 
 #include <QtGui/QIcon>
 
-
 namespace QtNodes {
 
 QPainterPath DefaultConnectionPainter::cubicPath(ConnectionGraphicsObject const &connection) const
@@ -27,11 +26,12 @@ QPainterPath DefaultConnectionPainter::cubicPath(ConnectionGraphicsObject const
     return cubic;
 }
 
-void DefaultConnectionPainter::drawSketchLine(QPainter *painter, ConnectionGraphicsObject const &cgo) const
+void DefaultConnectionPainter::drawSketchLine(QPainter *painter,
+                                              ConnectionGraphicsObject const &cgo) const
 {
     ConnectionState const &state = cgo.connectionState();
 
-    if (state.requiresPort()) {
+    if (state.requiresPort() || state.frozen()) {
         auto const &connectionStyle = QtNodes::StyleCollection::connectionStyle();
 
         QPen pen;
@@ -49,7 +49,8 @@ void DefaultConnectionPainter::drawSketchLine(QPainter *painter, ConnectionGraph
     }
 }
 
-void DefaultConnectionPainter::drawHoveredOrSelected(QPainter *painter, ConnectionGraphicsObject const &cgo) const
+void DefaultConnectionPainter::drawHoveredOrSelected(QPainter *painter,
+                                                     ConnectionGraphicsObject const &cgo) const
 {
     bool const hovered = cgo.connectionState().hovered();
     bool const selected = cgo.isSelected();
@@ -74,11 +75,12 @@ void DefaultConnectionPainter::drawHoveredOrSelected(QPainter *painter, Connecti
     }
 }
 
-void DefaultConnectionPainter::drawNormalLine(QPainter *painter, ConnectionGraphicsObject const &cgo) const
+void DefaultConnectionPainter::drawNormalLine(QPainter *painter,
+                                              ConnectionGraphicsObject const &cgo) const
 {
     ConnectionState const &state = cgo.connectionState();
 
-    if (state.requiresPort())
+    if (state.requiresPort() || state.frozen())
         return;
 
     // colors
@@ -132,6 +134,7 @@ void DefaultConnectionPainter::drawNormalLine(QPainter *painter, ConnectionGraph
         painter->setBrush(Qt::NoBrush);
 
         QColor cOut = normalColorOut;
+
         if (selected)
             cOut = cOut.darker(200);
         p.setColor(cOut);
@@ -200,7 +203,8 @@ void DefaultConnectionPainter::paint(QPainter *painter, ConnectionGraphicsObject
     painter->drawEllipse(cgo.in(), pointRadius, pointRadius);
 }
 
-QPainterPath DefaultConnectionPainter::getPainterStroke(ConnectionGraphicsObject const &connection) const
+QPainterPath DefaultConnectionPainter::getPainterStroke(
+    ConnectionGraphicsObject const &connection) const
 {
     auto cubic = cubicPath(connection);
 

src/GraphicsView.cpp

@@ -3,6 +3,7 @@
 #include "BasicGraphicsScene.hpp"
 #include "ConnectionGraphicsObject.hpp"
 #include "DataFlowGraphModel.hpp"
+#include "NodeDelegateModel.hpp"
 #include "NodeGraphicsObject.hpp"
 #include "StyleCollection.hpp"
 #include "UndoCommands.hpp"
@@ -26,6 +27,7 @@
 using QtNodes::BasicGraphicsScene;
 using QtNodes::DataFlowGraphModel;
 using QtNodes::GraphicsView;
+using QtNodes::NodeDelegateModel;
 using QtNodes::NodeGraphicsObject;
 
 GraphicsView::GraphicsView(QWidget *parent)