blender/blender-addons
19
43
Fork 96
Code Issues 632 Pull Requests 27 Projects 1 Releases Wiki Activity

Node Wrangler: Improved accuracy on Align Nodes operator #104551

Open
quackarooni wants to merge 18 commits from quackarooni/blender-addons:nw_rework_align_nodes into main
pull from: quackarooni/blender-addons:nw_rework_align_nodes
merge into: blender:main

When changing the target branch, be careful to rebase the branch in your fork to match. See documentation.
Conversation 8 Commits 18 Files Changed 4 +13 -12
1 changed files with 13 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 39aba28381 - Show all commits

25
node_wrangler/operators.py
View File

@ -2339,21 +2339,22 @@ class NWAlignNodes(Operator, NWBase):
bl_options = {'REGISTER', 'UNDO'} bl_options = {'REGISTER', 'UNDO'}
margin: IntProperty(name='Margin', default=50, description='The amount of space between nodes') margin: IntProperty(name='Margin', default=50, description='The amount of space between nodes')
@classmethod
def poll(cls, context):
selection = [node for node in context.selected_nodes if node.type != 'FRAME']
if len(selection) < 2:
return False
return nw_check(context)
def execute(self, context): def execute(self, context):
nodes, links = get_nodes_links(context) selection = [node for node in context.selected_nodes if node.type != 'FRAME']
active_node = context.active_node
margin = self.margin margin = self.margin
selection = []
for node in nodes:
if node.select and node.type != 'FRAME':
selection.append(node)
# If no nodes are selected, align all nodes
active_loc = None active_loc = None
if not selection: if active_node in selection:
selection = nodes active_loc = copy(active_node.location) # make a copy, not a reference
elif nodes.active in selection:
active_loc = copy(nodes.active.location) # make a copy, not a reference
# Check if nodes should be laid out horizontally or vertically # Check if nodes should be laid out horizontally or vertically
# use dimension to get center of node, not corner # use dimension to get center of node, not corner
@ -2388,7 +2389,7 @@ class NWAlignNodes(Operator, NWBase):
# If active node is selected, center nodes around it # If active node is selected, center nodes around it
if active_loc is not None: if active_loc is not None:
active_loc_diff = active_loc - nodes.active.location active_loc_diff = active_loc - active_node.location
for node in selection: for node in selection:
node.location += active_loc_diff node.location += active_loc_diff
else: # Position nodes centered around where they used to be else: # Position nodes centered around where they used to be