Add set_dynamic_port method to BaseNode && Optimilize AutoNode and SubGraph

Author: ArnoChenFx

NodeGraphQt/base/node.py

@@ -1104,6 +1104,16 @@ def set_editable(self, state):
         [wid.setEnabled(state) for wid in self.view._widgets.values()]
         self.view.text_item.setEnabled(state)
 
+    def set_dynamic_port(self, state):
+        """
+        Set whether the node will delete/add port after node has been created.
+
+        Args:
+            state(bool): If True, all port data will be serialized with the node,
+                         when the node is been deserialized, all ports will restore.
+        """
+        self.model.dynamic_port = state
+
 
 class BackdropNode(NodeObject):
     """

example_auto_nodes.py

@@ -1,15 +1,12 @@
 #!/usr/bin/python
 # -*- coding: utf-8 -*-
-import example_auto_nodes
-from NodeGraphQt import NodeGraph, setup_context_menu
-from NodeGraphQt import QtWidgets, QtCore, PropertiesBinWidget, \
-    NodeTreeWidget, BackdropNode, NodePublishWidget
-import os
-import sys
-import inspect
+from NodeGraphQt import NodeGraph, setup_context_menu, \
+    QtWidgets, QtCore, PropertiesBinWidget, BackdropNode
+from example_auto_nodes import Publish, RootNode, update_nodes, setup_node_menu
 import importlib
-from example_auto_nodes import AutoNode, ModuleNode, \
-    SubGraphNode, Publish, RootNode, update_nodes
+import inspect
+import sys
+import os
 
 
 def get_nodes_from_folder(folder_path):
@@ -49,58 +46,16 @@ def get_published_nodes_from_folder(folder_path):
     return nodes
 
 
-def cook_node(graph, node):
-    node.update_stream(forceCook=True)
-
-
-def print_functions(graph, node):
-    for func in node.module_functions:
-        print(func)
-
-
-def toggle_auto_cook(graph, node):
-    node.auto_cook = not node.auto_cook
-
-
-def enter_node(graph, node):
-    graph.set_node_space(node)
-
-
-def allow_edit(graph, node):
-    node.set_property('published', False)
-
-
-def print_path(graph, node):
-    print(node.path())
-
-
-def find_node_by_path(graph, node):
-    print(graph.get_node_by_path(node.path()))
-
-
-def print_children(graph, node):
-    children = node.children()
-    print(len(children), children)
-
-
-def publish_node(graph, node):
-    wid = NodePublishWidget(node=node)
-    wid.show()
-
-
-def cook_nodes(nodes):
-    update_nodes(nodes)
-
-
 if __name__ == '__main__':
-    app = QtWidgets.QApplication([])
+    app = QtWidgets.QApplication()
 
     # create node graph.
     graph = NodeGraph()
     graph.use_OpenGL()
 
     # set up default menu and commands.
     setup_context_menu(graph)
+    setup_node_menu(graph, Publish)
 
     # show the properties bin when a node is "double clicked" in the graph.
     properties_bin = PropertiesBinWidget(node_graph=graph)
@@ -111,40 +66,20 @@ def show_prop_bin(node):
             properties_bin.show()
     graph.node_double_clicked.connect(show_prop_bin)
 
-    # show the nodes list when a node is "double clicked" in the graph.
-    node_tree = NodeTreeWidget(node_graph=graph)
-
-    def show_nodes_list(node):
-        if not node_tree.isVisible():
-            node_tree.update()
-            node_tree.show()
-    graph.node_double_clicked.connect(show_nodes_list)
-
     # register nodes
     reg_nodes = get_nodes_from_folder(os.getcwd() + "/example_auto_nodes")
     BackdropNode.__identifier__ = 'Utility::Backdrop'
     reg_nodes.append(BackdropNode)
     reg_nodes.extend(get_published_nodes_from_folder(os.getcwd() + "/example_auto_nodes/published_nodes"))
     [graph.register_node(n) for n in reg_nodes]
 
-    # setup node menu
-    node_menu = graph.context_nodes_menu()
-    node_menu.add_command('Allow Edit', allow_edit, node_class=Publish)
-    node_menu.add_command('Enter Node', enter_node, node_class=SubGraphNode)
-    node_menu.add_command('Publish Node', publish_node, node_class=SubGraphNode)
-    node_menu.add_command('Print Children', print_children, node_class=SubGraphNode)
-    node_menu.add_command('Print Functions', print_functions, node_class=ModuleNode)
-    node_menu.add_command('Cook Node', cook_node, node_class=AutoNode)
-    node_menu.add_command('Toggle Auto Cook', toggle_auto_cook, node_class=AutoNode)
-    node_menu.add_command('Print Path', print_path, node_class=AutoNode)
-    node_menu.add_command('Find Node By Path', find_node_by_path, node_class=AutoNode)
-
     # create root node
+    # if we want to use sub graph system, root node is must.
     graph.add_node(RootNode())
 
     # create test nodes
     graph.load_session(r'example_auto_nodes/networks/example_SubGraph.json')
-    cook_nodes(graph.root_node().children())
+    update_nodes(graph.root_node().children())
 
     # widget used for the node graph.
     graph_widget = graph.widget

example_auto_nodes/__init__.py

@@ -1,5 +1,3 @@
-from .node_base.auto_node import AutoNode
-from .node_base.module_node import ModuleNode
-from .node_base.subgraph_node import SubGraphNode, RootNode
+from .node_base import AutoNode, ModuleNode, SubGraphNode, RootNode
 from .subgraph_nodes import Publish
-from .node_base.utils import update_node_down_stream, update_nodes
+from .node_base.utils import update_node_down_stream, update_nodes, setup_node_menu

example_auto_nodes/basic_nodes.py

@@ -1,4 +1,4 @@
-from .node_base.auto_node import AutoNode
+from .node_base import AutoNode
 
 
 class FooNode(AutoNode):

example_auto_nodes/data_node.py

@@ -1,6 +1,6 @@
 #!/usr/bin/python
 # -*- coding: utf-8 -*-
-from .node_base.auto_node import AutoNode
+from .node_base import AutoNode
 
 
 class VectorSplit(AutoNode):

example_auto_nodes/input_nodes.py

@@ -3,7 +3,7 @@
                                    NODE_PROP_VECTOR3,
                                    NODE_PROP_VECTOR4)
 
-from .node_base.auto_node import AutoNode
+from .node_base import AutoNode
 import os
 
 

example_auto_nodes/logic_nodes.py

@@ -1,4 +1,4 @@
-from .basic_nodes import AutoNode
+from .node_base import AutoNode
 
 
 class IfNode(AutoNode):

example_auto_nodes/module_nodes.py

@@ -1,6 +1,6 @@
-from .node_base.module_node import (ModuleNode,
-                                    get_functions_from_module,
-                                    get_functions_from_type)
+from .node_base import (ModuleNode,
+                        get_functions_from_module,
+                        get_functions_from_type)
 
 import os
 import sys
@@ -156,7 +156,6 @@ class StringFunctionsNode(ModuleNode):
 
     module_functions = get_functions_from_type(_str)
 
-
     def __init__(self):
         super(StringFunctionsNode, self).__init__()
 

example_auto_nodes/node_base/__init__.py

@@ -0,0 +1,3 @@
+from .auto_node import AutoNode
+from .module_node import ModuleNode, get_functions_from_module, get_functions_from_type
+from .subgraph_node import SubGraphNode, SubGraphInputNode, SubGraphOutputNode, RootNode

example_auto_nodes/node_base/subgraph_node.py

@@ -1,4 +1,5 @@
 from .auto_node import AutoNode
+from .utils import update_node_down_stream
 from NodeGraphQt import SubGraph
 import json
 from NodeGraphQt import topological_sort_by_down, BackdropNode
@@ -18,11 +19,11 @@ def __init__(self, defaultInputType=None, defaultOutputType=None, dynamic_port=T
         self.create_property('graph_rect', None)
         self.create_property('published', False)
         if dynamic_port:
-            self.model.dynamic_port = True
+            self.set_dynamic_port(True)
             self.add_int_input('input count', 'input count', 0)
             self.add_int_input('output count', 'output count', 0)
         else:
-            self.model.dynamic_port = False
+            self.set_dynamic_port(False)
             self.create_property('input count', 0)
             self.create_property('output count', 0)
         self._marked_ports = []
@@ -413,6 +414,22 @@ def get_data(self, port=None):
         for from_port in from_ports:
             return from_port.node().get_data(from_port)
 
+    def run(self):
+        parent = self.parent()
+        if parent is None or not parent.auto_cook:
+            return
+
+        port = parent.get_output(self.get_property('output index'))
+        if not port:
+            return
+
+        to_ports = port.connected_ports()
+        if not to_ports:
+            return
+
+        nodes = [p.node() for p in to_ports]
+        update_node_down_stream(nodes)
+
 
 class RootNode(SubGraphNode):
     """
@@ -439,7 +456,6 @@ def cook(self):
     def run(self):
         pass
 
-    def error(self, message=None):
-        if message is None:
-            return False
-        self._error = False
+    @property
+    def has_error(self):
+        return False