wip dynamic layout switching.

Author: jchanvfx

NodeGraphQt/base/graph.py

@@ -17,9 +17,12 @@
 from NodeGraphQt.base.node import NodeObject
 from NodeGraphQt.base.port import Port
 from NodeGraphQt.constants import (
-    NODE_LAYOUT_DIRECTION, NODE_LAYOUT_HORIZONTAL, NODE_LAYOUT_VERTICAL,
+    NODE_LAYOUT_DIRECTION,
+    NODE_LAYOUT_HORIZONTAL,
+    NODE_LAYOUT_VERTICAL,
+    URI_SCHEME,
+    URN_SCHEME,
     PipeLayoutEnum,
-    URI_SCHEME, URN_SCHEME,
     PortTypeEnum,
     ViewerEnum
 )
@@ -786,6 +789,26 @@ def set_pipe_style(self, style=PipeLayoutEnum.CURVED.value):
         style = style if 0 <= style <= pipe_max else PipeLayoutEnum.CURVED.value
         self._viewer.set_pipe_layout(style)
 
+    def set_layout_direction(self, direction):
+        """
+        Sets the node graph layout direction to horizontal or vertical.
+
+        Note:
+            By default node graph direction is set to "NODE_LAYOUT_HORIZONTAL".
+
+            Node Graph Layout Types:
+
+            * :attr:`NodeGraphQt.constants.NODE_LAYOUT_HORIZONTAL`
+            * :attr:`NodeGraphQt.constants.NODE_LAYOUT_VERTICAL`
+
+        Args:
+            direction (int): layout direction.
+        """
+        direction_types = [NODE_LAYOUT_HORIZONTAL, NODE_LAYOUT_VERTICAL]
+        if direction not in direction_types:
+            direction = NODE_LAYOUT_HORIZONTAL
+        self._viewer.set_layout_direction(direction)
+
     def fit_to_selection(self):
         """
         Sets the zoom level to fit selected nodes.
@@ -1672,7 +1695,9 @@ def auto_layout_nodes(self, nodes=None, down_stream=True, start_nodes=None):
             else:
                 rank_map[rank] = [node]
 
-        if NODE_LAYOUT_DIRECTION is NODE_LAYOUT_HORIZONTAL:
+        node_layout_direction = self._viewer.get_layout_direction()
+
+        if node_layout_direction is NODE_LAYOUT_HORIZONTAL:
             current_x = 0
             node_height = 120
             for rank in sorted(range(len(rank_map)), reverse=not down_stream):
@@ -1687,7 +1712,7 @@ def auto_layout_nodes(self, nodes=None, down_stream=True, start_nodes=None):
                     current_y += dy * 0.5 + 10
 
                 current_x += max_width * 0.5 + 100
-        elif NODE_LAYOUT_DIRECTION is NODE_LAYOUT_VERTICAL:
+        elif node_layout_direction is NODE_LAYOUT_VERTICAL:
             current_y = 0
             node_width = 250
             for rank in sorted(range(len(rank_map)), reverse=not down_stream):
@@ -1953,6 +1978,8 @@ def _build_port_nodes(self):
         Returns:
              tuple(dict, dict): input nodes, output nodes.
         """
+        node_layout_direction = self._viewer.get_layout_direction()
+
         # build the parent input port nodes.
         input_nodes = {n.name(): n for n in
                        self.get_nodes_by_type(PortInputNode.type_)}
@@ -1965,9 +1992,9 @@ def _build_port_nodes(self):
                 input_nodes[port.name()] = input_node
                 self.add_node(input_node, selected=False, push_undo=False)
                 x, y = input_node.pos()
-                if NODE_LAYOUT_DIRECTION is NODE_LAYOUT_HORIZONTAL:
+                if node_layout_direction is NODE_LAYOUT_HORIZONTAL:
                     x -= 100
-                elif NODE_LAYOUT_DIRECTION is NODE_LAYOUT_VERTICAL:
+                elif node_layout_direction is NODE_LAYOUT_VERTICAL:
                     y -= 100
                 input_node.set_property('pos', [x, y], push_undo=False)
 
@@ -1983,9 +2010,9 @@ def _build_port_nodes(self):
                 output_nodes[port.name()] = output_node
                 self.add_node(output_node, selected=False, push_undo=False)
                 x, y = output_node.pos()
-                if NODE_LAYOUT_DIRECTION is NODE_LAYOUT_HORIZONTAL:
+                if node_layout_direction is NODE_LAYOUT_HORIZONTAL:
                     x += 100
-                elif NODE_LAYOUT_DIRECTION is NODE_LAYOUT_VERTICAL:
+                elif node_layout_direction is NODE_LAYOUT_VERTICAL:
                     y += 100
                 output_node.set_property('pos', [x, y], push_undo=False)
 

NodeGraphQt/constants.py

@@ -17,38 +17,24 @@
 URI_SCHEME = 'nodegraphqt://'
 URN_SCHEME = 'nodegraphqt::'
 
-# === PATHS ===
-
+# PATHS
 BASE_PATH = os.path.dirname(os.path.abspath(__file__))
 ICON_PATH = os.path.join(BASE_PATH, 'widgets', 'icons')
 ICON_DOWN_ARROW = os.path.join(ICON_PATH, 'down_arrow.png')
 ICON_NODE_BASE = os.path.join(ICON_PATH, 'node_base.png')
 
-# === DRAW STACK ORDER ===
-
+# DRAW STACK ORDER
 Z_VAL_PIPE = -1
 Z_VAL_NODE = 1
 Z_VAL_PORT = 2
 Z_VAL_NODE_WIDGET = 3
 
-# === ITEM CACHE MODE ===
-
+# ITEM CACHE MODE
 # QGraphicsItem.NoCache
 # QGraphicsItem.DeviceCoordinateCache
 # QGraphicsItem.ItemCoordinateCache
-
 ITEM_CACHE_MODE = QtWidgets.QGraphicsItem.DeviceCoordinateCache
 
-# === NODE LAYOUT DIRECTION ===
-
-#: Mode for vertical node layout.
-NODE_LAYOUT_VERTICAL = 0
-#: Mode for horizontal node layout.
-NODE_LAYOUT_HORIZONTAL = 1
-#: Variable for setting the node layout direction.
-# NODE_LAYOUT_DIRECTION = NODE_LAYOUT_VERTICAL
-NODE_LAYOUT_DIRECTION = NODE_LAYOUT_HORIZONTAL
-
 # =================================== GLOBAL ===================================
 
 
@@ -66,6 +52,23 @@ class VersionEnum(Enum):
     #:
     PATCH = int(_v.split('.')[2])
 
+
+class LayoutDirectionEnum(Enum):
+    """
+    Node graph nodes layout direction:
+    :py:mod:`NodeGraphQt.constants.ViewerLayoutEnum`
+    """
+    #: layout nodes top to bottom.
+    VERTICAL = 0
+    #: layout nodes left to right.
+    HORIZONTAL = 1
+
+
+#: Variable for setting the node layout direction.
+# NODE_LAYOUT_DIRECTION = LayoutDirectionEnum.VERTICAL.value
+NODE_LAYOUT_DIRECTION = LayoutDirectionEnum.HORIZONTAL.value
+
+
 # =================================== VIEWER ===================================