upgrade and add new example nodes

Author: ArnoChenFx

example_auto_nodes/data_node.py

@@ -0,0 +1,124 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+from .node_base.auto_node import AutoNode
+
+
+class VectorValue(AutoNode):
+    """
+    Create a basic vector data
+    """
+
+    __identifier__ = 'Data'
+    NODE_NAME = 'Vector'
+
+    def __init__(self):
+        super(VectorValue, self).__init__()
+        value = [0.0, 0.0, 0.0]
+
+        self.add_output('out',list)
+        self.create_property("out", value)
+
+        self.add_float_input('0', 'X', value=value[0], tab='widgets')
+        self.view.widgets['0'].value_changed.connect(lambda: self.updateValue(0))
+
+        self.add_float_input('1', 'Y', value=value[1], tab='widgets')
+        self.view.widgets['1'].value_changed.connect(lambda: self.updateValue(1))
+
+        self.add_float_input('2', 'Z', value=value[2], tab='widgets')
+        self.view.widgets['2'].value_changed.connect(lambda: self.updateValue(2))
+
+        self.defaultValue = value
+
+    def updateValue(self, index):
+        self.get_property("out")[index] = self.get_property(str(index))
+        self.cook()
+
+
+class VectorSplit(AutoNode):
+    """
+    Splict a vector to x,y,z
+    """
+
+    __identifier__ = 'Data'
+    NODE_NAME = 'Vector Split'
+
+    def __init__(self):
+        super(VectorSplit, self).__init__()
+        self.defaultValue = [0.0, 0.0, 0.0]
+
+        self.add_output('x')
+        self.create_property("x", self.defaultValue[0])
+        self.add_output('y')
+        self.create_property("y", self.defaultValue[1])
+        self.add_output('z')
+        self.create_property("z", self.defaultValue[2])
+
+        self.add_input("in vector",list)
+
+    def run(self):
+        value = self.getInputData(0)
+        self.set_property("x", value[0])
+        self.set_property("y", value[1])
+        self.set_property("z", value[2])
+
+
+class VectorMaker(AutoNode):
+    """
+    Create a vector by three float value
+    """
+
+    __identifier__ = 'Data'
+    NODE_NAME = 'Vector Maker'
+
+    def __init__(self):
+        super(VectorMaker, self).__init__()
+
+        self.add_output('out')
+        self.create_property("out", [0, 0, 0])
+
+        self.add_input("x",float)
+        self.add_input("y",float)
+        self.add_input("z",float)
+
+        self.defaultValue = 0.0
+
+    def run(self):
+        self.set_property("out", [self.getInputData(0), self.getInputData(1), self.getInputData(2)])
+
+
+class DataConvect(AutoNode):
+    """
+    Create a vector by three float value
+    """
+
+    __identifier__ = 'Data'
+    NODE_NAME = 'Data Convect'
+
+    def __init__(self):
+        super(DataConvect, self).__init__()
+
+        self.add_output('out')
+        self.create_property("out",None)
+        self.add_input("in data")
+
+        items = ["all to int"]
+        items.append("all to float")
+        items.append("all to string")
+        items.append("eval string")
+        self.add_combo_menu('method', 'Method', items=items, tab='widgets')
+        self.view.widgets['method'].value_changed.connect(self.cook)
+
+    def run(self):
+        method = self.get_property("method")
+        try:
+            if method == "all to int":
+                data = int(self.getInputData(0))
+            elif method == "all to float":
+                data = float(self.getInputData(0))
+            elif method == "all to string":
+                data = str(self.getInputData(0))
+            elif method == "eval string":
+                data = eval(self.getInputData(0))
+            self.set_property("out", data)
+        except Exception as error:
+            self.error(error)

example_auto_nodes/input_nodes.py

@@ -2,21 +2,6 @@
 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.
@@ -97,7 +82,7 @@ class TextInputNode(AutoNode):
     __identifier__ = 'Inputs'
 
     # initial default node name.
-    NODE_NAME = 'text'
+    NODE_NAME = 'Text'
 
     def __init__(self):
         super(TextInputNode, self).__init__()

example_auto_nodes/module_nodes.py

@@ -0,0 +1,218 @@
+from example_auto_nodes.node_base.module_node import (ModuleNode,
+                                                      get_functions_from_module,
+                                                      get_functions_from_type)
+
+import math
+import os
+import sys
+import random
+import numpy
+import numpydoc.docscrape
+import inspect
+
+# add basic math functions to math library
+math.add = lambda x, y: x + y
+math.sub = lambda x, y: x - y
+math.mul = lambda x, y: x * y
+math.div = lambda x, y: x / y
+
+
+class MathModuleNode(ModuleNode):
+    """
+    Math functions node.
+    """
+
+    # set a unique node identifier.
+    __identifier__ = 'Module'
+
+    # set the initial default node name.
+    NODE_NAME = 'math module'
+
+    module_functions = get_functions_from_type(math)
+
+    def __init__(self):
+        super(MathModuleNode, self).__init__()
+        self.defaultValue = 0.0
+
+
+class osModuleNode(ModuleNode):
+    """
+    os module node.
+    """
+
+    # set a unique node identifier.
+    __identifier__ = 'Module'
+
+    # set the initial default node name.
+    NODE_NAME = 'os module'
+
+    module_functions = get_functions_from_module(os, max_depth=2)
+    module_functions.pop("os.abort")
+
+    def __init__(self):
+        super(osModuleNode, self).__init__()
+
+class sysModuleNode(ModuleNode):
+    """
+    sys module node.
+    """
+
+    # set a unique node identifier.
+    __identifier__ = 'Module'
+
+    # set the initial default node name.
+    NODE_NAME = 'sys module'
+
+    module_functions = get_functions_from_module(sys, max_depth=2)
+    module_functions.pop("sys.exit")
+    module_functions.pop("sys.breakpointhook")
+
+    def __init__(self):
+        super(sysModuleNode, self).__init__()
+
+
+class randomModuleNode(ModuleNode):
+    """
+    random module node.
+    """
+
+    # set a unique node identifier.
+    __identifier__ = 'Module'
+
+    # set the initial default node name.
+    NODE_NAME = 'random module'
+
+    module_functions = get_functions_from_module(random,max_depth=2)
+
+
+    def __init__(self):
+        super(randomModuleNode, self).__init__()
+
+
+class numpyModuleNode(ModuleNode):
+    """
+    numpy module node.
+    """
+
+    # set a unique node identifier.
+    __identifier__ = 'Module'
+
+    # set the initial default node name.
+    NODE_NAME = 'numpy module'
+
+    module_functions = get_functions_from_module(numpy,max_depth=2)
+
+
+    def __init__(self):
+        super(numpyModuleNode, self).__init__()
+
+    def is_function(self,obj):
+        result = super(numpyModuleNode,self).is_function(obj)
+        if result:
+            return True
+        elif type(obj).__name__ == "ufunc":
+            return True
+        return False
+
+    def get_numpy_args(self,func):
+        args = []
+        info = numpydoc.docscrape.FunctionDoc(func)
+        for i in info["Parameters"]:
+            param = str(i)
+            if "name" in param:
+                args.append(param.split("'")[1])
+        return args
+
+    def addFunction(self, prop, func):
+        """
+        Create inputs based on functions arguments.
+        """
+
+        self.func = self.module_functions[func]
+
+        args = []
+        if self.is_function(self.func):
+            try:
+                args = inspect.getfullargspec(self.func).args
+            except Exception as error:
+                try:
+                    args = self.get_numpy_args(self.func)
+                except Exception as error:
+                    if type(self.func).__name__ == "ufunc":
+                        args = ["input"+str(i+1) for i in range(self.func.nin)]
+
+        self.process_args(args)
+
+
+class StringFunctionsNode(ModuleNode):
+    """
+    String functions node.
+    """
+
+    # set a unique node identifier.
+    __identifier__ = 'Data'
+
+    # set the initial default node name.
+    NODE_NAME = 'String Functions'
+
+    module_functions = get_functions_from_type(str)
+
+    def __init__(self):
+        super(StringFunctionsNode, self).__init__()
+
+
+class ListFunctionsNode(ModuleNode):
+    """
+    List functions node.
+    """
+
+    # set a unique node identifier.
+    __identifier__ = 'Data'
+
+    # set the initial default node name.
+    NODE_NAME = 'List Functions'
+
+    module_functions = get_functions_from_type(list)
+
+    def __init__(self):
+        super(ListFunctionsNode, self).__init__()
+
+
+class DictFunctionsNode(ModuleNode):
+    """
+    DIct functions node.
+    """
+
+    # set a unique node identifier.
+    __identifier__ = 'Data'
+
+    # set the initial default node name.
+    NODE_NAME = 'Dict Functions'
+
+    module_functions = get_functions_from_type(list)
+
+    def __init__(self):
+        super(DictFunctionsNode, self).__init__()
+
+
+class TupleFunctionsNode(ModuleNode):
+    """
+    Tuple functions node.
+    """
+
+    # set a unique node identifier.
+    __identifier__ = 'Data'
+
+    # set the initial default node name.
+    NODE_NAME = 'Tuple Functions'
+
+    module_functions = get_functions_from_type(tuple)
+
+    def __init__(self):
+        super(TupleFunctionsNode, self).__init__()
+
+
+if __name__ == "__main__":
+    s = "[2,2]"
+    print(numpy.zeros(eval(s)))
+    pass