add boolean logics node

Author: Leandro Inocencio

example.nodes

@@ -1,74 +1,6 @@
 {
   "nodes":{
-    "0x1508f548":{
-      "type_":"Inputs.DataInputNode",
-      "icon":null,
-      "name":"Input A",
-      "color":[
-        13,
-        18,
-        23,
-        255
-      ],
-      "border_color":[
-        74,
-        84,
-        85,
-        255
-      ],
-      "text_color":[
-        255,
-        255,
-        255,
-        180
-      ],
-      "disabled":false,
-      "selected":true,
-      "width":173.0,
-      "height":91.0,
-      "pos":[
-        -417.0,
-        -160.0
-      ],
-      "custom":{
-        "out":"Hola"
-      }
-    },
-    "0x160ea048":{
-      "type_":"Inputs.DataInputNode",
-      "icon":null,
-      "name":"Input B",
-      "color":[
-        13,
-        18,
-        23,
-        255
-      ],
-      "border_color":[
-        74,
-        84,
-        85,
-        255
-      ],
-      "text_color":[
-        255,
-        255,
-        255,
-        180
-      ],
-      "disabled":false,
-      "selected":true,
-      "width":173.0,
-      "height":91.0,
-      "pos":[
-        -390.0,
-        -23.0
-      ],
-      "custom":{
-        "out":"Chau"
-      }
-    },
-    "0x160ea748":{
+    "0x15095e08":{
       "type_":"Viewers.DataViewerNode",
       "icon":null,
       "name":"Output",
@@ -95,17 +27,17 @@
       "width":170,
       "height":91.0,
       "pos":[
-        186.0,
-        -231.0
+        149.0,
+        -146.0
       ],
       "custom":{
-        "data":"Hola"
+        "data":""
       }
     },
-    "0x160f3688":{
-      "type_":"Logics.ifNode",
+    "0x15126848":{
+      "type_":"Inputs.BoolInputNode",
       "icon":null,
-      "name":"If",
+      "name":"Bool",
       "color":[
         13,
         18,
@@ -127,57 +59,20 @@
       "disabled":false,
       "selected":true,
       "width":170,
-      "height":108.0,
-      "pos":[
-        -75.0,
-        -178.0
-      ],
-      "custom":{
-        "condition":0.0,
-        "out":"Hola"
-      }
-    },
-    "0x160fad08":{
-      "type_":"Math.MathFunctionsNode",
-      "icon":null,
-      "name":"Math Functions",
-      "color":[
-        25,
-        58,
-        51,
-        255
-      ],
-      "border_color":[
-        74,
-        84,
-        85,
-        255
-      ],
-      "text_color":[
-        255,
-        255,
-        255,
-        180
-      ],
-      "disabled":false,
-      "selected":true,
-      "width":181.0,
       "height":103.0,
       "pos":[
-        -411.0,
-        -319.0
+        -413.0,
+        -212.0
       ],
       "custom":{
-        "funcs":"fmod",
-        "output":1.0,
-        "x":219.0,
-        "y":2.0
+        "out":true,
+        "combo":"True"
       }
     },
-    "0x17678bc8":{
-      "type_":"Inputs.TickTimeNode",
+    "0x15140308":{
+      "type_":"Logics.BooleanNode",
       "icon":null,
-      "name":"Tick Time",
+      "name":"Boolean",
       "color":[
         13,
         18,
@@ -198,20 +93,21 @@
       ],
       "disabled":false,
       "selected":true,
-      "width":173.0,
-      "height":91.0,
+      "width":170,
+      "height":103.0,
       "pos":[
-        -678.0,
-        -393.0
+        -143.0,
+        -149.0
       ],
       "custom":{
-        "out":"219"
+        "out":null,
+        "funcs":"and"
       }
     },
-    "0x1768efc8":{
-      "type_":"Inputs.DataInputNode",
+    "0x1514a4c8":{
+      "type_":"Inputs.BoolInputNode",
       "icon":null,
-      "name":"Basic Input",
+      "name":"Bool 1",
       "color":[
         13,
         18,
@@ -232,75 +128,46 @@
       ],
       "disabled":false,
       "selected":true,
-      "width":173.0,
-      "height":91.0,
+      "width":170,
+      "height":103.0,
       "pos":[
-        -678.0,
-        -207.0
+        -411.25,
+        -61.5
       ],
       "custom":{
-        "out":"2"
+        "out":true,
+        "combo":"True"
       }
     }
   },
   "connections":[
-    {
-      "out":[
-        "0x1508f548",
-        "out"
-      ],
-      "in":[
-        "0x160f3688",
-        "then"
-      ]
-    },
-    {
-      "out":[
-        "0x160ea048",
-        "out"
-      ],
-      "in":[
-        "0x160f3688",
-        "else"
-      ]
-    },
     {
       "in":[
-        "0x160ea748",
+        "0x15095e08",
         "data"
       ],
       "out":[
-        "0x160f3688",
+        "0x15140308",
         "out"
       ]
     },
     {
-      "in":[
-        "0x160f3688",
-        "condition"
-      ],
       "out":[
-        "0x160fad08",
-        "output"
-      ]
-    },
-    {
-      "in":[
-        "0x160fad08",
-        "x"
-      ],
-      "out":[
-        "0x17678bc8",
+        "0x15126848",
         "out"
+      ],
+      "in":[
+        "0x15140308",
+        "a"
       ]
     },
     {
       "in":[
-        "0x160fad08",
-        "y"
+        "0x15140308",
+        "b"
       ],
       "out":[
-        "0x1768efc8",
+        "0x1514a4c8",
         "out"
       ]
     }

example_nodes/logic_nodes.py

@@ -1,7 +1,7 @@
 from NodeGraphQt import BaseNode, QtCore
 
 
-class ifNode(BaseNode):
+class IfNode(BaseNode):
     """
     if node.
     """
@@ -10,7 +10,7 @@ class ifNode(BaseNode):
     NODE_NAME = 'If'
 
     def __init__(self):
-        super(ifNode, self).__init__()
+        super(IfNode, self).__init__()
         self.condition = self.add_input('condition')
         self._then = self.add_input('then')
         self._else = self.add_input('else')
@@ -39,3 +39,71 @@ def on_input_connected(self, to_port, from_port):
     def on_input_disconnected(self, to_port, from_port):
         """Override node callback method."""
         self.set_property('out', None)
+
+
+class BooleanNode(BaseNode):
+    """
+    Boolean Logic funtions node.
+    """
+
+    __identifier__ = 'Logics'
+
+    NODE_NAME = 'Boolean'
+
+    logics = {'and': 'a and b',
+              'or': 'a or b',
+              'xor': '(not a and b) or (a and not b)',
+              'not': 'not a'}
+
+    def __init__(self):
+        super(BooleanNode, self).__init__()
+        self.a = self.add_input('a')
+        self.b = self.add_input('b')
+        self.add_output('out')
+        self.create_property('out', None)
+        self.add_combo_menu('funcs',
+                            'Functions',
+                            items=list(self.logics.keys()),
+                            tab='widgets')
+        self.func = logics['and']
+        # switch math function type
+        self.view.widgets['funcs'].value_changed.connect(self.addFunction)
+        self.view.widgets['funcs'].value_changed.connect(self.update_streams)
+
+    def addFunction(self, prop, func):
+        """
+        Create inputs based on math functions arguments.
+        """
+        self.func = logics[func]
+        if self.b.visible() and not 'b' in self.func:
+            self.b.set_visible(False)
+        elif not self.b.visible():
+            self.b.set_visible(True)
+
+    def run(self):
+        a = None
+        b = None
+
+        for port in self.a.connected_ports():
+            port.node().run()
+            a = port.node().get_property(port.name())
+
+        for port in self.b.connected_ports():
+            port.node().run()
+            b = port.node().get_property(port.name())
+
+        if a is None or (b is None and 'b' in self.func):
+            raise Exception("No inputs!")
+
+        self.set_property('out', eval(self.func))
+
+    def on_input_connected(self, to_port, from_port):
+        """Override node callback method."""
+        from_port.node().run()
+        result = from_port.node().get_property(from_port.name())
+        self.set_property(to_port.name(), result)
+        self.update_streams()
+
+    def on_input_disconnected(self, to_port, from_port):
+        """Override node callback method."""
+        self.set_property('out', None)