add AutoNode class evaluate the node pipline

Author: ArnoChenFx

example_auto_nodes.py

@@ -0,0 +1,101 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+import example_auto_nodes
+
+from NodeGraphQt import (NodeGraph,
+                         BaseNode,
+                         setup_context_menu)
+from NodeGraphQt import QtWidgets, QtCore, PropertiesBinWidget, NodeTreeWidget
+import os
+import sys
+import inspect
+import importlib
+
+def GetNodesFromFolder(FolderPath):
+    path, FolderName = os.path.split(FolderPath)
+    if path not in sys.path:
+        sys.path.append(path)
+
+    nodes = []
+    for i in os.listdir(FolderPath):
+        if not i.endswith(".py") or i.startswith("_"):
+            continue
+
+        filename = i[:-3]
+        module_name = FolderName + "." + filename
+
+        for name, obj in inspect.getmembers(importlib.import_module(module_name)):
+            if inspect.isclass(obj) and filename in str(obj):
+                if len(inspect.getmembers(obj)) > 0:
+                    nodes.append(obj)
+    return nodes
+
+
+if __name__ == '__main__':
+    app = QtWidgets.QApplication([])
+
+    # create node graph.
+    graph = NodeGraph()
+
+    # set up default menu and commands.
+    setup_context_menu(graph)
+
+    # widget used for the node graph.
+    graph_widget = graph.widget
+    graph_widget.resize(1100, 800)
+    graph_widget.show()
+
+    # show the properties bin when a node is "double clicked" in the graph.
+    properties_bin = PropertiesBinWidget(node_graph=graph)
+    properties_bin.setWindowFlags(QtCore.Qt.Tool)
+
+    def show_prop_bin(node):
+        if not properties_bin.isVisible():
+            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)
+
+    reg_nodes = GetNodesFromFolder(os.getcwd() + "/example_auto_nodes")
+
+    for n in reg_nodes:
+        graph.register_node(n)
+
+    mathNodeA = graph.create_node('Math.MathFunctionsNode',
+                                name='Math Functions A',
+                                color='#0a1e20',
+                                text_color='#feab20',
+                                pos=[-250, 70])
+
+    mathNodeB = graph.create_node('Math.MathFunctionsNode',
+                                name='Math Functions B',
+                                color='#0a1e20',
+                                text_color='#feab20',
+                                pos=[-250, -70])
+
+    mathNodeC = graph.create_node('Math.MathFunctionsNode',
+                                  name='Math Functions C',
+                                  color='#0a1e20',
+                                  text_color='#feab20',
+                                  pos=[0, 0])
+
+    inputANode = graph.create_node('Inputs.FloatInputNode',
+                                   name='Input A',
+                                   pos=[-500, -50])
+
+    inputBNode = graph.create_node('Inputs.FloatInputNode',
+                                   name='Input B',
+                                   pos=[-500, 50])
+
+    outputNode = graph.create_node('Viewers.DataViewerNode',
+                                   name='Output',
+                                   pos=[250, 0])
+
+    app.exec_()

example_auto_nodes/__init__.py

@@ -0,0 +1 @@
+

example_auto_nodes/basic_nodes.py

@@ -0,0 +1,49 @@
+from .node_base.auto_node import AutoNode
+
+class FooNode(AutoNode):
+    """
+    A node class with 2 inputs and 2 outputs.
+    """
+
+    # unique node identifier.
+    __identifier__ = 'com.chantasticvfx'
+
+    # initial default node name.
+    NODE_NAME = 'foo node'
+
+    def __init__(self):
+        super(FooNode, self).__init__()
+
+        # create node inputs.
+        self.add_input('in A')
+        self.add_input('in B')
+
+        # create node outputs.
+        self.add_output('out A')
+        self.add_output('out B')
+
+
+class BarNode(AutoNode):
+    """
+    A node class with 3 inputs and 3 outputs.
+    The last input and last output can take in multiple pipes.
+    """
+
+    # unique node identifier.
+    __identifier__ = 'com.chantasticvfx'
+
+    # initial default node name.
+    NODE_NAME = 'bar'
+
+    def __init__(self):
+        super(BarNode, self).__init__()
+
+        # create node inputs
+        self.add_input('single in 1')
+        self.add_input('single in 2')
+        self.add_input('multi in', multi_input=True)
+
+        # create node outputs
+        self.add_output('single out 1', multi_output=False)
+        self.add_output('single out 2', multi_output=False)
+        self.add_output('multi out')

example_auto_nodes/input_nodes.py

@@ -0,0 +1,85 @@
+from NodeGraphQt import QtCore
+from.node_base.auto_node import AutoNode
+import os
+
+class DataInputNode(AutoNode):
+    """
+    Input node data.
+    """
+
+    __identifier__ = 'Inputs'
+    NODE_NAME = 'Basic Input'
+
+    def __init__(self):
+        super(DataInputNode, self).__init__()
+        self.add_output('out')
+        self.add_text_input('out', 'Data Output', text='0.4', tab='widgets')
+        self.view.widgets['out'].value_changed.connect(self.cook)
+
+
+class FloatInputNode(AutoNode):
+    """
+    Input float data.
+    """
+
+    __identifier__ = 'Inputs'
+    NODE_NAME = 'Float'
+
+    def __init__(self):
+        super(FloatInputNode, self).__init__()
+        self.defaultValue = 0.0
+
+        self.output = self.add_output('out',float)
+        self.add_float_input('out', 'Float Value', value=self.defaultValue, tab='widgets')
+        self.view.widgets['out'].value_changed.connect(self.cook)
+
+
+class TickTimeNode(AutoNode):
+    """
+    Every second that passes adds a tick.
+    """
+
+    __identifier__ = 'Inputs'
+    NODE_NAME = 'Tick Time'
+
+    def __init__(self):
+        super(TickTimeNode, self).__init__()
+        self.add_output('out',float)
+        self.add_float_input('out', 'Data Input', value=0.0, tab='widgets')
+        self.view.widgets['out'].value_changed.connect(self.cook)
+
+        self.timer = QtCore.QTimer()
+        self.timer.timeout.connect(self.tick)
+        self.timer.start(1000)
+
+    def tick(self):
+        if not self.disabled():
+            current = self.get_property('out')
+            current += 1
+            self.set_property("out",current)
+
+
+class TextFileInputNode(AutoNode):
+    """
+    Text File Input node data.
+    """
+
+    __identifier__ = 'Inputs'
+    NODE_NAME = 'Text File'
+
+    def __init__(self):
+        super(TextFileInputNode, self).__init__()
+        self.add_output('out',str)
+        self.create_property('out', "")
+        self.add_text_input('path', 'Text File Path', text='', tab='widgets')
+        self.view.widgets['path'].value_changed.connect(self.cook)
+
+    def run(self):
+        path = self.get_property('path')
+        if os.path.exists(path):
+            with open(path, 'r') as fread:
+                data = fread.read()
+                self.set_property('output', data)
+        else:
+            print('No existe %s' % path)
+            self.set_property('output', '')