pointer color override logic.

Author: jchanvfx

NodeGraphQt/qgraphics/pipe.py

@@ -583,8 +583,7 @@ def hoverEnterEvent(self, event):
         """
         QtWidgets.QGraphicsPathItem.hoverEnterEvent(self, event)
 
-    def draw_path(self, start_port, end_port=None, cursor_pos=None,
-                  color_mode=None):
+    def draw_path(self, start_port, end_port=None, cursor_pos=None, color=None):
         """
         re-implemented to also update the index pointer arrow position.
 
@@ -593,22 +592,20 @@ def draw_path(self, start_port, end_port=None, cursor_pos=None,
             end_port (PortItem): port used to draw the end point.
             cursor_pos (QtCore.QPointF): cursor position if specified this
                 will be the draw end point.
-            color_mode (str): arrow index pointer color mode
-                              ('accept', 'reject' or None).
+            color (list[int]): override arrow index pointer color. (r, g, b)
         """
         super(LivePipeItem, self).draw_path(start_port, end_port, cursor_pos)
-        self.draw_index_pointer(start_port, cursor_pos, color_mode)
+        self.draw_index_pointer(start_port, cursor_pos, color)
 
-    def draw_index_pointer(self, start_port, cursor_pos, color_mode=None):
+    def draw_index_pointer(self, start_port, cursor_pos, color=None):
         """
         Update the index pointer arrow position and direction when the
         live pipe path is redrawn.
 
         Args:
             start_port (PortItem): start port item.
             cursor_pos (QtCore.QPoint): cursor scene position.
-            color_mode (str): arrow index pointer color mode
-                              ('accept', 'reject' or None).
+            color (list[int]): override arrow index pointer color. (r, g, b).
         """
         text_rect = self._idx_text.boundingRect()
 
@@ -635,16 +632,13 @@ def draw_index_pointer(self, start_port, cursor_pos, color_mode=None):
 
         self._idx_pointer.setPolygon(transform.map(self._poly))
 
-        if color_mode == 'accept':
-            color = QtGui.QColor(*PipeEnum.HIGHLIGHT_COLOR.value)
-        elif color_mode == 'reject':
-            color = QtGui.QColor(*PipeEnum.DISABLED_COLOR.value)
-        else:
-            color = QtGui.QColor(*PipeEnum.ACTIVE_COLOR.value)
+        pen_color = QtGui.QColor(*PipeEnum.HIGHLIGHT_COLOR.value)
+        if isinstance(color, (list, tuple)):
+            pen_color = QtGui.QColor(*color)
 
         pen = self._idx_pointer.pen()
-        pen.setColor(color)
-        self._idx_pointer.setBrush(color.darker(300))
+        pen.setColor(pen_color)
+        self._idx_pointer.setBrush(pen_color.darker(300))
         self._idx_pointer.setPen(pen)
 
 

NodeGraphQt/widgets/viewer.py

@@ -9,6 +9,7 @@
 from NodeGraphQt.constants import (
     LayoutDirectionEnum,
     PortTypeEnum,
+    PipeEnum,
     PipeLayoutEnum,
     ViewerEnum,
     Z_VAL_PIPE,
@@ -756,7 +757,7 @@ def sceneMouseMoveEvent(self, event):
             return
 
         pos = event.scenePos()
-        color_mode = None
+        pointer_color = None
         for item in self.scene().items(pos):
             if not isinstance(item, PortItem):
                 continue
@@ -768,26 +769,25 @@ def sceneMouseMoveEvent(self, event):
             pos.setY(pos.y() + y)
             if item == self._start_port:
                 break
-            color_mode = 'accept'
-
+            pointer_color = PipeEnum.HIGHLIGHT_COLOR.value
             accept = self._validate_accept_connection(self._start_port, item)
             if not accept:
-                color_mode = 'reject'
+                pointer_color = [150, 60, 255]
                 break
             reject = self._validate_reject_connection(self._start_port, item)
             if reject:
-                color_mode = 'reject'
+                pointer_color = [150, 60, 255]
                 break
 
             if self.acyclic:
                 if item.node == self._start_port.node:
-                    color_mode = 'reject'
+                    pointer_color = PipeEnum.DISABLED_COLOR.value
                 elif item.port_type == self._start_port.port_type:
-                    color_mode = 'reject'
+                    pointer_color = PipeEnum.DISABLED_COLOR.value
             break
 
         self._LIVE_PIPE.draw_path(
-            self._start_port, cursor_pos=pos, color_mode=color_mode
+            self._start_port, cursor_pos=pos, color=pointer_color
         )
 
     def sceneMousePressEvent(self, event):