fix node item adaption when node ports change

Author: leon-thomm

ryvencore_qt/src/flows/nodes/NodeGUI.py

@@ -20,10 +20,10 @@ class NodeGUI(QObject):
     # qt signals
     updating = Signal()
     update_error = Signal(object)
-    input_added = Signal(object, object)
-    output_added = Signal(object, object)
-    input_removed = Signal(object)
-    output_removed = Signal(object)
+    input_added = Signal(object, int, object)
+    output_added = Signal(object, int, object)
+    input_removed = Signal(object, int, object)
+    output_removed = Signal(object, int, object)
     update_shape_triggered = Signal()
     hide_unconnected_ports_triggered = Signal()
     show_unconnected_ports_triggered = Signal()

ryvencore_qt/src/flows/nodes/NodeItem.py

@@ -57,10 +57,10 @@ def __init__(self, node, node_gui, flow_view, design):
         self.node_gui.update_shape_triggered.connect(self.update_shape)
         self.node_gui.hide_unconnected_ports_triggered.connect(self.hide_unconnected_ports_triggered)
         self.node_gui.show_unconnected_ports_triggered.connect(self.show_unconnected_ports_triggered)
-        self.node_gui.input_added.connect(self.add_new_input)
-        self.node_gui.output_added.connect(self.add_new_output)
-        self.node_gui.input_removed.connect(self.remove_input)
-        self.node_gui.output_removed.connect(self.remove_output)
+        self.node_gui.input_added.connect(self.on_node_input_added)
+        self.node_gui.output_added.connect(self.on_node_output_added)
+        self.node_gui.input_removed.connect(self.on_node_input_removed)
+        self.node_gui.output_removed.connect(self.on_node_output_removed)
 
         # FLAGS
         self.setFlags(
@@ -171,6 +171,10 @@ def set_tooltip(self, error_msg=None):
         self.setToolTip(html)
         self.setCursor(Qt.SizeAllCursor)
 
+    def on_node_input_added(self, node, index, inp: NodeInput):
+        insert = index if index == len(node.inputs) - 1 else None
+        self.add_new_input(inp, insert)
+
     def add_new_input(self, inp: NodeInput, insert: int = None):
 
         if inp in self.node_gui.input_widgets:
@@ -195,6 +199,9 @@ def add_new_input(self, inp: NodeInput, insert: int = None):
             self.update_shape()
             self.update()
 
+    def on_node_input_removed(self, node, index, inp: NodeInput):
+        self.remove_input(inp)
+
     def remove_input(self, inp: NodeInput):
         item = None
         for inp_item in self.inputs:
@@ -219,6 +226,10 @@ def remove_input(self, inp: NodeInput):
             self.update_shape()
             self.update()
 
+    def on_node_output_added(self, node, index, out: NodeOutput):
+        insert = index if index == len(node.outputs) - 1 else None
+        self.add_new_output(out, insert)
+
     def add_new_output(self, out: NodeOutput, insert: int = None):
 
         # create item
@@ -236,6 +247,9 @@ def add_new_output(self, out: NodeOutput, insert: int = None):
             self.update_shape()
             self.update()
 
+    def on_node_output_removed(self, node, index, out: NodeOutput):
+        self.remove_output(out)
+
     def remove_output(self, out: NodeOutput):
         item = None
         for out_item in self.outputs: