Merge pull request #4 from jchanvfx/master

merge

Author: ArnoChenFx

NodeGraphQt/base/commands.py

@@ -67,7 +67,6 @@ def undo(self):
 
     def redo(self):
         do_redo = False
-
         try:
             if self.old_val != self.new_val:
                 do_redo = True

NodeGraphQt/base/graph.py

@@ -140,7 +140,7 @@ def _insert_node(self, pipe, node_id, prev_node_pos):
         node = self.get_node_by_id(node_id)
 
         # exclude the BackdropNode
-        if not isinstance(node,BaseNode):
+        if not isinstance(node, BaseNode):
             return
 
         disconnected = [(pipe.input_port, pipe.output_port)]

NodeGraphQt/base/model.py

@@ -234,7 +234,7 @@ def to_dict(self):
         custom_props = node_dict.pop('_custom_prop', {})
 
         if custom_props:
-            # exclude the data which can not be serialed (like numpy array)
+            # exclude the data which can not be serialized (like numpy array)
             to_remove = []
             types = [float, str, int, list, dict, bool, None, complex, tuple]
             for k, v in custom_props.items():

NodeGraphQt/base/node.py

@@ -7,10 +7,10 @@
                                    NODE_PROP_QTEXTEDIT,
                                    NODE_PROP_QCOMBO,
                                    NODE_PROP_QCHECKBOX,
-                                   IN_PORT, OUT_PORT,
                                    NODE_PROP_FILE,
                                    NODE_PROP_FLOAT,
-                                   NODE_PROP_INT)
+                                   NODE_PROP_INT,
+                                   IN_PORT, OUT_PORT)
 from NodeGraphQt.errors import PortRegistrationError
 from NodeGraphQt.qgraphics.node_backdrop import BackdropNodeItem
 from NodeGraphQt.qgraphics.node_base import NodeItem
@@ -727,7 +727,7 @@ def output_ports(self):
             list[NodeGraphQt.Port]: node output ports.
         """
         return self._outputs
-    
+
     def input(self, index):
         """
         Return the input port with the matching index.
@@ -833,6 +833,38 @@ def on_input_disconnected(self, in_port, out_port):
         """
         return
 
+    def update_streams(self, *args):
+        """
+        Update all nodes joined by pipes to this.
+        """
+        nodes = []
+        trash = []
+
+        for port, nodeList in self.connected_output_nodes().items():
+            nodes.extend(nodeList)
+
+        while nodes:
+            node = nodes.pop()
+            if node.disabled():
+                continue
+            if node not in trash:
+                trash.append(node)
+
+            for port, nodeList in node.connected_output_nodes().items():
+                nodes.extend(nodeList)
+
+            if not node.connected_output_nodes():
+                try:
+                    node.run()
+                except Exception as error:
+                    print("Error Update Streams: %s" % str(error))
+
+    def run(self):
+        """
+        Node evaluation logics.
+        """
+        return
+
 
 class BackdropNode(NodeObject):
     """

NodeGraphQt/base/utils.py

@@ -35,7 +35,7 @@ def setup_context_menu(graph):
     file_menu.add_command('Open...', _open_session, QtGui.QKeySequence.Open)
     file_menu.add_command('Save...', _save_session, QtGui.QKeySequence.Save)
     file_menu.add_command('Save As...', _save_session_as, 'Ctrl+Shift+s')
-    file_menu.add_command('New', _new_session)
+    file_menu.add_command('New Session', _new_session)
 
     file_menu.add_separator()
 

NodeGraphQt/qgraphics/node_abstract.py

@@ -1,7 +1,8 @@
 #!/usr/bin/python
 from NodeGraphQt import QtCore, QtWidgets
 
-from NodeGraphQt.constants import Z_VAL_NODE, NODE_WIDTH, NODE_HEIGHT, ITEM_CACHE_MODE
+from NodeGraphQt.constants import (Z_VAL_NODE, NODE_WIDTH, NODE_HEIGHT,
+                                   ITEM_CACHE_MODE)
 
 
 class AbstractNodeItem(QtWidgets.QGraphicsItem):

NodeGraphQt/qgraphics/node_base.py

@@ -501,7 +501,7 @@ def auto_switch_mode(self):
 
         self.set_proxy_mode(width < self._porxy_mode_threshold)
 
-    def set_proxy_mode(self,mode):
+    def set_proxy_mode(self, mode):
         if mode is self._proxy_mode:
             return
 

NodeGraphQt/qgraphics/pipe.py

@@ -7,8 +7,9 @@
     PIPE_HIGHLIGHT_COLOR, PIPE_DISABLED_COLOR,
     PIPE_STYLE_DASHED, PIPE_STYLE_DEFAULT, PIPE_STYLE_DOTTED,
     PIPE_LAYOUT_STRAIGHT, PIPE_WIDTH, IN_PORT, OUT_PORT, Z_VAL_PIPE,
-    Z_VAL_NODE_WIDGET,PIPE_LAYOUT_ANGLE,
-    PIPE_LAYOUT_CURVED,ITEM_CACHE_MODE)
+    Z_VAL_NODE_WIDGET,
+    PIPE_LAYOUT_ANGLE, PIPE_LAYOUT_CURVED,
+    ITEM_CACHE_MODE)
 from NodeGraphQt.qgraphics.port import PortItem
 
 PIPE_STYLES = {

NodeGraphQt/widgets/actions.py

@@ -30,7 +30,7 @@ def get_menu(self, name, node_id=None):
                 if isinstance(node,menu.node_class):
                     return menu
 
-    def get_menus(self,node_class):
+    def get_menus(self, node_class):
         menus = []
         for action in self.actions():
             menu = action.menu()

NodeGraphQt/widgets/node_widgets.py

@@ -6,6 +6,7 @@
 from NodeGraphQt.widgets.file_dialog import file_dialog
 from NodeGraphQt.widgets.properties import _valueEdit
 
+
 class _NodeGroupBox(QtWidgets.QGroupBox):
 
     def __init__(self, label, parent=None):