backdrop wrap undo block

Author: jchanvfx

NodeGraphQt/base/node.py

@@ -1,4 +1,6 @@
 #!/usr/bin/python
+from collections import OrderedDict
+
 from .commands import PropertyChangedCmd
 from .model import NodeModel
 from .port import Port
@@ -1107,7 +1109,7 @@ def connected_input_nodes(self):
         Returns:
             dict: {<input_port>: <node_list>}
         """
-        nodes = {}
+        nodes = OrderedDict()
         for p in self.input_ports():
             nodes[p] = [cp.node() for cp in p.connected_ports()]
         return nodes
@@ -1119,7 +1121,7 @@ def connected_output_nodes(self):
         Returns:
             dict: {<output_port>: <node_list>}
         """
-        nodes = {}
+        nodes = OrderedDict()
         for p in self.output_ports():
             nodes[p] = [cp.node() for cp in p.connected_ports()]
         return nodes
@@ -1212,20 +1214,54 @@ def __init__(self):
         self.create_property('backdrop_text', '',
                              widget_type=NODE_PROP_QTEXTEDIT, tab='Backdrop')
 
+    def on_backdrop_updated(self, update_prop, value=None):
+        """
+        Slot triggered by the "on_backdrop_updated" signal from
+        the node graph.
+
+        Args:
+            update_prop (str): update property type.
+            value (object): update value (optional)
+        """
+        if update_prop == 'sizer_mouse_release':
+            self.graph.begin_undo('resized "{}"'.format(self.name()))
+            self.set_property('width', value['width'])
+            self.set_property('height', value['height'])
+            self.set_pos(*value['pos'])
+            self.graph.end_undo()
+        elif update_prop == 'sizer_double_clicked':
+            self.graph.begin_undo('"{}" auto resize'.format(self.name()))
+            self.set_property('width', value['width'])
+            self.set_property('height', value['height'])
+            self.set_pos(*value['pos'])
+            self.graph.end_undo()
+
     def auto_size(self):
         """
         Auto resize the backdrop node to fit around the intersecting nodes.
         """
-        self.view.auto_resize()
+        self.graph.begin_undo('"{}" auto resize'.format(self.name()))
+        size = self.view.calc_backdrop_size()
+        self.set_property('width', size['width'])
+        self.set_property('height', size['height'])
+        self.set_pos(*size['pos'])
+        self.graph.end_undo()
 
     def wrap_nodes(self, nodes):
         """
-        Wrap backdrop size to fit around specified nodes.
+        Set the backdrop size to fit around specified nodes.
 
         Args:
             nodes (list[NodeGraphQt.NodeObject]): list of nodes.
         """
-        self.view.auto_resize([n.view for n in nodes])
+        if not nodes:
+            return
+        self.graph.begin_undo('"{}" wrap nodes'.format(self.name()))
+        size = self.view.calc_backdrop_size()
+        self.set_property('width', size['width'])
+        self.set_property('height', size['height'])
+        self.set_pos(*size['pos'])
+        self.graph.end_undo()
 
     def nodes(self):
         """

NodeGraphQt/widgets/viewer.py

@@ -38,6 +38,7 @@ class NodeViewer(QtWidgets.QGraphicsView):
     insert_node = QtCore.Signal(object, str, dict)
     need_show_tab_search = QtCore.Signal()
     node_name_changed = QtCore.Signal(str, str)
+    node_backdrop_updated = QtCore.Signal(str, str, object)
 
     # pass through signals that are translated into "NodeGraph()" signals.
     node_selected = QtCore.Signal(str)