wip pipe slicer

Author: jchanvfx

NodeGraphQt/base/graph.py

@@ -59,6 +59,7 @@ def __repr__(self):
     def _wire_signals(self):
         # internal signals.
         self._viewer.search_triggered.connect(self._on_search_triggered)
+        self._viewer.connection_sliced.connect(self._on_connection_sliced)
         self._viewer.connection_changed.connect(self._on_connection_changed)
         self._viewer.moved_nodes.connect(self._on_nodes_moved)
         self._viewer.node_double_clicked.connect(self._on_node_double_clicked)
@@ -188,6 +189,17 @@ def _on_connection_changed(self, disconnected, connected):
             port1.connect_to(port2)
         self._undo_stack.endMacro()
 
+    def _on_connection_sliced(self, ports):
+        """
+        slot when connection pipes have been sliced.
+
+        Args:
+            ports (list[list[widgets.port.PortItem]]):
+                pair list of port connections (in port, out port)
+        """
+        for in_port, out_port in ports:
+            print(in_port, out_port)
+
     @property
     def model(self):
         """

NodeGraphQt/constants.py

@@ -13,6 +13,7 @@
 PIPE_DISABLED_COLOR = (190, 20, 20, 255)
 PIPE_ACTIVE_COLOR = (70, 255, 220, 255)
 PIPE_HIGHLIGHT_COLOR = (232, 184, 13, 255)
+PIPE_SLICER_COLOR = (140, 255, 170)
 #: The draw the connection pipes as straight lines.
 PIPE_LAYOUT_STRAIGHT = 0
 #: The draw the connection pipes as curved lines.

NodeGraphQt/widgets/viewer.py

@@ -6,7 +6,9 @@
 from NodeGraphQt.constants import (IN_PORT, OUT_PORT,
                                    PIPE_LAYOUT_CURVED,
                                    PIPE_LAYOUT_STRAIGHT,
+                                   PIPE_SLICER_COLOR,
                                    PIPE_STYLE_DASHED,
+                                   Z_VAL_NODE_WIDGET,
                                    SCENE_AREA)
 from NodeGraphQt.qgraphics.node_abstract import AbstractNodeItem
 from NodeGraphQt.qgraphics.node_backdrop import BackdropNodeItem
@@ -30,6 +32,7 @@ class NodeViewer(QtWidgets.QGraphicsView):
 
     moved_nodes = QtCore.Signal(dict)
     search_triggered = QtCore.Signal(str, tuple)
+    connection_sliced = QtCore.Signal(list)
     connection_changed = QtCore.Signal(list, list)
 
     # pass through signals
@@ -63,6 +66,14 @@ def __init__(self, parent=None):
         self._rubber_band = QtWidgets.QRubberBand(
             QtWidgets.QRubberBand.Rectangle, self
         )
+        slicer_color = QtGui.QColor(*PIPE_SLICER_COLOR)
+        slicer_pen = QtGui.QPen(slicer_color, 1.5, QtCore.Qt.DashLine)
+        self._pipe_slicer = QtWidgets.QGraphicsLineItem()
+        self._pipe_slicer.setPen(slicer_pen)
+        self._pipe_slicer.setVisible(False)
+        self._pipe_slicer.setZValue(Z_VAL_NODE_WIDGET + 2)
+        self.scene().addItem(self._pipe_slicer)
+
         self._undo_stack = QtWidgets.QUndoStack(self)
         self._context_menu = QtWidgets.QMenu('main', self)
         self._context_menu.setStyleSheet(STYLE_QMENU)
@@ -126,6 +137,15 @@ def _on_search_submitted(self, node_type):
         pos = self.mapToScene(self._previous_pos)
         self.search_triggered.emit(node_type, (pos.x(), pos.y()))
 
+    def _on_pipes_sliced(self, line):
+        path = QtGui.QPainterPath()
+        path.moveTo(line.p1())
+        path.lineTo(line.p2())
+        self.connection_sliced.emit([
+            [i.input_port, i.output_port]
+            for i in self.scene().items(path) if isinstance(i, Pipe)
+        ])
+
     # --- reimplemented events ---
 
     def resizeEvent(self, event):
@@ -138,6 +158,7 @@ def contextMenuEvent(self, event):
     def mousePressEvent(self, event):
         alt_modifier = event.modifiers() == QtCore.Qt.AltModifier
         shift_modifier = event.modifiers() == QtCore.Qt.ShiftModifier
+
         if event.button() == QtCore.Qt.LeftButton:
             self.LMB_state = True
         elif event.button() == QtCore.Qt.RightButton:
@@ -152,10 +173,19 @@ def mousePressEvent(self, event):
         if self._search_widget.isVisible():
             self.tab_search_toggle()
 
+        # cursor pos.
+        map_pos = self.mapToScene(event.pos())
+
+        # pipe slicer enabled.
+        if event.modifiers() == (QtCore.Qt.AltModifier | QtCore.Qt.ShiftModifier):
+            self._pipe_slicer.setLine(QtCore.QLineF(map_pos, map_pos))
+            self._pipe_slicer.setVisible(True)
+            return
+
         if alt_modifier:
             return
 
-        items = self._items_near(self.mapToScene(event.pos()), None, 20, 20)
+        items = self._items_near(map_pos, None, 20, 20)
         nodes = [i for i in items if isinstance(i, AbstractNodeItem)]
 
         # toggle extend node selection.
@@ -188,6 +218,12 @@ def mouseReleaseEvent(self, event):
         elif event.button() == QtCore.Qt.MiddleButton:
             self.MMB_state = False
 
+        # hide pipe slicer.
+        if self._pipe_slicer.isVisible():
+            self._on_pipes_sliced(self._pipe_slicer.line())
+            self._pipe_slicer.setLine(QtCore.QLineF(0.0, 0.0, 0.0, 0.0))
+            self._pipe_slicer.setVisible(False)
+
         # hide selection marquee
         if self._rubber_band.isVisible():
             rect = self._rubber_band.rect()
@@ -211,6 +247,15 @@ def mouseReleaseEvent(self, event):
     def mouseMoveEvent(self, event):
         alt_modifier = event.modifiers() == QtCore.Qt.AltModifier
         shift_modifier = event.modifiers() == QtCore.Qt.ShiftModifier
+        if event.modifiers() == (QtCore.Qt.AltModifier | QtCore.Qt.ShiftModifier):
+            if self.LMB_state:
+                p1 = self._pipe_slicer.line().p1()
+                p2 = self.mapToScene(self._previous_pos)
+                self._pipe_slicer.setLine(QtCore.QLineF(p1, p2))
+                self._previous_pos = event.pos()
+                super(NodeViewer, self).mouseMoveEvent(event)
+                return
+
         if self.MMB_state and alt_modifier:
             pos_x = (event.x() - self._previous_pos.x())
             zoom = 0.1 if pos_x > 0 else -0.1
@@ -296,6 +341,10 @@ def sceneMousePressEvent(self, event):
             event (QtWidgets.QGraphicsScenePressEvent):
                 The event handler from the QtWidgets.QGraphicsScene
         """
+        # pipe slicer enabled.
+        if event.modifiers() == (QtCore.Qt.AltModifier | QtCore.Qt.ShiftModifier):
+            return
+        # viewer pan mode.
         if event.modifiers() == QtCore.Qt.AltModifier:
             return