From c3b04a94d760fe72e193578ed1f2eb110e9e4f87 Mon Sep 17 00:00:00 2001
From: Greg <gregzzmail@gmail.com>
Date: Thu, 11 Aug 2016 12:19:11 +0200
Subject: [PATCH] Node Wrangler: Rewrite Copy Settings function Fixes a bug
with frames, selects only the source node afterwards, reports info. From
D2129 by @poor
---
node_wrangler.py | 143 +++++++++++++++++++++++++++++------------------
1 file changed, 89 insertions(+), 54 deletions(-)
diff --git a/node_wrangler.py b/node_wrangler.py
index 6b4a8fb0b..491058c1d 100644
--- a/node_wrangler.py
+++ b/node_wrangler.py
@@ -38,6 +38,7 @@ from math import cos, sin, pi, hypot
from os import path
from glob import glob
from copy import copy
+from itertools import chain
#################
# rl_outputs:
@@ -2286,62 +2287,96 @@ class NWCopySettings(Operator, NWBase):
return valid
def execute(self, context):
- nodes, links = get_nodes_links(context)
- selected = [n for n in nodes if n.select]
- reselect = [] # duplicated nodes will be selected after execution
- active = nodes.active
- if active.select:
- reselect.append(active)
+ node_active = context.active_node
+ node_selected = context.selected_nodes
+
+ # Error handling
+ if not (len(node_selected) > 1):
+ self.report({'ERROR'}, "2 nodes must be selected at least")
+ return {'CANCELLED'}
- for node in selected:
- if node.type == active.type and node != active:
- # duplicate active, relink links as in 'node', append copy to 'reselect', delete node
- bpy.ops.node.select_all(action='DESELECT')
- nodes.active = active
- active.select = True
- bpy.ops.node.duplicate()
- copied = nodes.active
- # Copied active should however inherit some properties from 'node'
- attributes = (
- 'hide', 'show_preview', 'mute', 'label',
- 'use_custom_color', 'color', 'width', 'width_hidden',
- )
- for attr in attributes:
- setattr(copied, attr, getattr(node, attr))
- # Handle scenario when 'node' is in frame. 'copied' is in same frame then.
- if copied.parent:
- bpy.ops.node.parent_clear()
- locx = node.location.x
- locy = node.location.y
- # get absolute node location
- parent = node.parent
- while parent:
- locx += parent.location.x
- locy += parent.location.y
- parent = parent.parent
- copied.location = [locx, locy]
- # reconnect links from node to copied
- for i, input in enumerate(node.inputs):
- if input.links:
- link = input.links[0]
- links.new(link.from_socket, copied.inputs[i])
- for out, output in enumerate(node.outputs):
- if output.links:
- out_links = output.links
- for link in out_links:
- links.new(copied.outputs[out], link.to_socket)
- bpy.ops.node.select_all(action='DESELECT')
- node.select = True
- bpy.ops.node.delete()
- reselect.append(copied)
- else: # If selected wasn't copied, need to reselect it afterwards.
- reselect.append(node)
- # clean up
- bpy.ops.node.select_all(action='DESELECT')
- for node in reselect:
- node.select = True
- nodes.active = active
+ # Check if active node is in the selection
+ selected_node_names = [n.name for n in node_selected]
+ if node_active.name not in selected_node_names:
+ self.report({'ERROR'}, "No active node")
+ return {'CANCELLED'}
+
+ # Get nodes in selection by type
+ valid_nodes = [n for n in node_selected if n.type == node_active.type]
+ if not (len(valid_nodes) > 1) and node_active:
+ self.report({'ERROR'}, "Selected nodes are not of the same type as {}".format(node_active.name))
+ return {'CANCELLED'}
+
+ if len(valid_nodes) != len(node_selected):
+ # Report nodes that are not valid
+ valid_node_names = [n.name for n in valid_nodes]
+ not_valid_names = list(set(selected_node_names) - set(valid_node_names))
+ self.report({'INFO'}, "Ignored {} (not of the same type as {})".format(", ".join(not_valid_names), node_active.name))
+
+ # Reference original
+ orig = node_active
+ #node_selected_names = [n.name for n in node_selected]
+
+ # Output list
+ success_names = []
+
+ # Deselect all nodes
+ for i in node_selected:
+ i.select = False
+
+ # Run through all other nodes
+ for node in valid_nodes[1:]:
+
+ # Check for frame node
+ parent = node.parent if node.parent else None
+ node_loc = [node.location.x, node.location.y]
+
+ # Select original to duplicate
+ orig.select = True
+
+ # Duplicate selected node
+ bpy.ops.node.duplicate()
+ new_node = context.selected_nodes[0]
+
+ # Deselect copy
+ new_node.select = False
+
+ # Properties to copy
+ node_tree = node.id_data
+ props_to_copy = 'bl_idname name location height width'.split(' ')
+
+ # Input and outputs
+ reconnections = []
+ mappings = chain.from_iterable([node.inputs, node.outputs])
+ for i in (i for i in mappings if i.is_linked):
+ for L in i.links:
+ reconnections.append([L.from_socket.path_from_id(), L.to_socket.path_from_id()])
+
+ # Properties
+ props = {j: getattr(node, j) for j in props_to_copy}
+ props_to_copy.pop(0)
+
+ for prop in props_to_copy:
+ setattr(new_node, prop, props[prop])
+
+ # Get the node tree to remove the old node
+ nodes = node_tree.nodes
+ nodes.remove(node)
+ new_node.name = props['name']
+
+ if parent:
+ new_node.parent = parent
+ new_node.location = node_loc
+
+ for str_from, str_to in reconnections:
+ node_tree.links.new(eval(str_from), eval(str_to))
+
+ success_names.append(new_node.name)
+
+ orig.select = True
+ node_tree.nodes.active = orig
+ self.report({'INFO'}, "Successfully copied attributes from {} to: {}".format(orig.name, ", ".join(success_names)))
return {'FINISHED'}
--
GitLab