add : toggle auto cook logic

Author: ArnoChenFx

NodeGraphQt/widgets/actions.py

@@ -18,7 +18,7 @@ def __init__(self, *args, **kwargs):
     #         if hasattr(a, 'node_id'):
     #             a.node_id = None
 
-    def get_menu(self, name,node_id=None):
+    def get_menu(self, name, node_id=None):
         for action in self.actions():
             menu = action.menu()
             if not menu:

example_auto_nodes.py

@@ -10,6 +10,9 @@
 import sys
 import inspect
 import importlib
+from example_auto_nodes.node_base.auto_node import AutoNode
+from example_auto_nodes.node_base.module_node import ModuleNode
+
 
 def GetNodesFromFolder(FolderPath):
     path, FolderName = os.path.split(FolderPath)
@@ -31,6 +34,17 @@ def GetNodesFromFolder(FolderPath):
     return nodes
 
 
+def cook_node(graph,node):
+    node.cook()
+
+
+def print_functions(graph,node):
+    for func in node.module_functions:
+        print(func)
+
+def toggle_auto_cook(graph,node):
+    node.autoCook = not node.autoCook
+
 if __name__ == '__main__':
     app = QtWidgets.QApplication([])
 
@@ -68,19 +82,24 @@ def show_nodes_list(node):
     for n in reg_nodes:
         graph.register_node(n)
 
-    mathNodeA = graph.create_node('Math.MathFunctionsNode',
+    node_menu = graph.context_nodes_menu()
+    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)
+
+    mathNodeA = graph.create_node('Module.MathModuleNode',
                                 name='Math Functions A',
                                 color='#0a1e20',
                                 text_color='#feab20',
                                 pos=[-250, 70])
 
-    mathNodeB = graph.create_node('Math.MathFunctionsNode',
+    mathNodeB = graph.create_node('Module.MathModuleNode',
                                 name='Math Functions B',
                                 color='#0a1e20',
                                 text_color='#feab20',
                                 pos=[-250, -70])
 
-    mathNodeC = graph.create_node('Math.MathFunctionsNode',
+    mathNodeC = graph.create_node('Module.MathModuleNode',
                                   name='Math Functions C',
                                   color='#0a1e20',
                                   text_color='#feab20',

example_auto_nodes/node_base/auto_node.py

@@ -19,15 +19,33 @@ class AutoNode(BaseNode):
     def __init__(self,defaultInputType=None,defaultOutputType=None):
         super(AutoNode, self).__init__()
         self.needCook = True
+        self._autoCook = True
         self._error = False
         self.matchTypes = [[float, int]]
         self.errorColor = (200, 50, 50)
+        self.stopCookColor = (200,200,200)
 
         self.defaultColor = self.get_property("color")
         self.defaultValue = None
         self.defaultInputType = defaultInputType
         self.defaultOutputType = defaultOutputType
 
+    @property
+    def autoCook(self):
+        return self._autoCook
+
+    @autoCook.setter
+    def autoCook(self,mode):
+        if mode is self._autoCook:
+            return
+
+        self._autoCook = mode
+        if mode:
+            self.set_property('color',self.defaultColor)
+        else:
+            self.defaultColor = self.get_property("color")
+            self.set_property('color', self.stopCookColor)
+
     def cookNextNode(self):
         for nodeList in self.connected_output_nodes().values():
             for n in nodeList:
@@ -52,7 +70,10 @@ def getInputData(self, port):
             data = from_port.node().get_property(from_port.name())
             return data
 
-    def cook(self):
+    def cook(self, forceCook = False):
+        if not self._autoCook and forceCook is not True:
+            return
+
         if self.disabled():
             num = len(self.input_ports())
             for index, out_port in enumerate(self.output_ports()):