d81ba55ce4
The lazy-connect feature of node wrangler uses the built-in `connect_sockets` function which automatically handles virtual sockets in group input and output nodes already. However, it does not handle virtual sockets in other nodes. The fix is to generalize this behavior. For that, a new `handle_dynamic_sockets` boolean input is added to `tree.links.new`. When enabled, virtual sockets are handled properly by internally calling the `bNodeType.insert_link` methods. The new behavior is turned of by default for compatibility reasons. Pull Request: https://projects.blender.org/blender/blender/pulls/126282
87 lines
2.5 KiB
Python
87 lines
2.5 KiB
Python
# SPDX-FileCopyrightText: 2009-2023 Blender Authors
|
|
#
|
|
# SPDX-License-Identifier: GPL-2.0-or-later
|
|
|
|
__all__ = (
|
|
"find_node_input",
|
|
)
|
|
|
|
|
|
def find_base_socket_type(socket):
|
|
"""
|
|
Find the base class of the socket.
|
|
|
|
Sockets can have a subtype such as NodeSocketFloatFactor,
|
|
but only the base type is allowed, e. g. NodeSocketFloat
|
|
"""
|
|
if socket.type == 'CUSTOM':
|
|
# Custom socket types are used directly
|
|
return socket.bl_idname
|
|
if socket.type == 'VALUE':
|
|
return 'NodeSocketFloat'
|
|
if socket.type == 'INT':
|
|
return 'NodeSocketInt'
|
|
if socket.type == 'BOOLEAN':
|
|
return 'NodeSocketBoolean'
|
|
if socket.type == 'VECTOR':
|
|
return 'NodeSocketVector'
|
|
if socket.type == 'ROTATION':
|
|
return 'NodeSocketRotation'
|
|
if socket.type == 'STRING':
|
|
return 'NodeSocketString'
|
|
if socket.type == 'RGBA':
|
|
return 'NodeSocketColor'
|
|
if socket.type == 'SHADER':
|
|
return 'NodeSocketShader'
|
|
if socket.type == 'OBJECT':
|
|
return 'NodeSocketObject'
|
|
if socket.type == 'IMAGE':
|
|
return 'NodeSocketImage'
|
|
if socket.type == 'GEOMETRY':
|
|
return 'NodeSocketGeometry'
|
|
if socket.type == 'COLLECTION':
|
|
return 'NodeSocketCollection'
|
|
if socket.type == 'TEXTURE':
|
|
return 'NodeSocketTexture'
|
|
if socket.type == 'MATERIAL':
|
|
return 'NodeSocketMaterial'
|
|
|
|
|
|
def connect_sockets(input, output):
|
|
"""
|
|
Connect sockets in a node tree.
|
|
|
|
This is useful because the links created through the normal Python API are
|
|
invalid when one of the sockets is a virtual socket (grayed out sockets in
|
|
Group Input and Group Output nodes).
|
|
|
|
It replaces node_tree.links.new(input, output)
|
|
"""
|
|
import bpy
|
|
|
|
# Swap sockets if they are not passed in the proper order
|
|
if input.is_output and not output.is_output:
|
|
input, output = output, input
|
|
|
|
input_node = output.node
|
|
output_node = input.node
|
|
|
|
if input_node.id_data is not output_node.id_data:
|
|
print("Sockets do not belong to the same node tree")
|
|
return
|
|
|
|
if type(input) == type(output) == bpy.types.NodeSocketVirtual:
|
|
print("Cannot connect two virtual sockets together")
|
|
return
|
|
|
|
return input_node.id_data.links.new(input, output, handle_dynamic_sockets=True)
|
|
|
|
|
|
# XXX Names are not unique. Returns the first match.
|
|
def find_node_input(node, name):
|
|
for input in node.inputs:
|
|
if input.name == name:
|
|
return input
|
|
|
|
return None
|