updated draw logic in LivePipe

Author: jchanvfx

NodeGraphQt/qgraphics/pipe.py

@@ -301,9 +301,72 @@ def delete(self):
 
 class LivePipe(Pipe):
 
-    def __init__(self, input_port=None, output_port=None):
-        super(LivePipe, self).__init__(input_port, output_port)
+    def __init__(self):
+        super(LivePipe, self).__init__()
         self.setZValue(Z_VAL_NODE_WIDGET + 1)
-        self.activate()
-        self.style = PIPE_STYLE_DASHED
-        self.shift_selected = False
+
+    def paint(self, painter, option, widget):
+        """
+        Draws the connection line.
+
+        Args:
+            painter (QtGui.QPainter): painter used for drawing the item.
+            option (QtGui.QStyleOptionGraphicsItem):
+                used to describe the parameters needed to draw.
+            widget (QtWidgets.QWidget): not used.
+        """
+        color = QtGui.QColor(*PIPE_ACTIVE_COLOR)
+        pen_style = PIPE_STYLES.get(PIPE_STYLE_DASHED)
+        pen_width = PIPE_WIDTH + 0.35
+
+        pen = QtGui.QPen(color, pen_width)
+        pen.setStyle(pen_style)
+        pen.setCapStyle(QtCore.Qt.RoundCap)
+
+        painter.save()
+        painter.setPen(pen)
+        painter.setRenderHint(painter.Antialiasing, True)
+        painter.drawPath(self.path())
+
+        cen_x = self.path().pointAtPercent(0.5).x()
+        cen_y = self.path().pointAtPercent(0.5).y()
+        loc_pt = self.path().pointAtPercent(0.9)
+        tgt_pt = self.path().pointAtPercent(1.0)
+
+        dist = math.hypot(tgt_pt.x() - cen_x, tgt_pt.y() - cen_y)
+        if dist < 0.05:
+            painter.restore()
+            return
+
+        # draw circle
+        size = 10 * (dist / 100)
+        if size > 10.0:
+            size = 10.0
+        elif size < 5.0:
+            size = 5.0
+        rect = QtCore.QRectF(cen_x-(size/2), cen_y-(size/2), size, size)
+        painter.setBrush(color)
+        painter.setPen(QtGui.QPen(color.darker(130), pen_width))
+        painter.drawEllipse(rect)
+
+        # draw arrow
+        color.setAlpha(255)
+        painter.setBrush(color.darker(200))
+
+        pen_width = 0.6
+        if dist < 1.0:
+            pen_width *= (1.0 + dist)
+        painter.setPen(QtGui.QPen(color, pen_width))
+
+        transform = QtGui.QTransform()
+        transform.translate(tgt_pt.x(), tgt_pt.y())
+
+        radians = math.atan2(tgt_pt.y() - loc_pt.y(),
+                             tgt_pt.x() - loc_pt.x())
+        degrees = math.degrees(radians) + 90
+        transform.rotate(degrees)
+
+        if dist < 10.0:
+            transform.scale(0.5, 0.5)
+        painter.drawPolygon(transform.map(self._arrow))
+        painter.restore()

NodeGraphQt/widgets/viewer.py

@@ -5,7 +5,7 @@
 from NodeGraphQt import QtGui, QtCore, QtWidgets
 from NodeGraphQt.constants import (IN_PORT, OUT_PORT,
                                    PIPE_LAYOUT_CURVED,
-                                   SCENE_AREA,)
+                                   SCENE_AREA)
 from NodeGraphQt.qgraphics.node_abstract import AbstractNodeItem
 from NodeGraphQt.qgraphics.node_backdrop import BackdropNodeItem
 from NodeGraphQt.qgraphics.pipe import Pipe, LivePipe