ilusha/blender
1
0
Fork 0
mirror of https://github.com/blender/blender synced 2026-09-29 04:37:17 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix #158280: Node Tools: Menu inputs broken since recent refactor

Browse source 
Caused by 1561c1ea4a
There were a few problems. First, the enum items generation callback
didn't get the pointer I had expected when writing this. Instead, generate
them when registering the operator. Second, the asset meta-data stored
things in a confusing way, resulting in the Python-accessible enum items
using numbers as the identifiers instead of the names like similar properties
on modifiers. Add a test to keep this all working in the future.
This commit is contained in:
Hans Goudey 2026-05-07 16:54:56 -04:00 • committed by Hans Goudey
parent 8f7d756880
commit b68a2f3921
3 changed files with 101 additions and 37 deletions
Download patch file Download diff file Expand all files Collapse all files

50
tests/python/bl_node_tool.py
View file

@ -115,6 +115,56 @@ class TestNodeTool(unittest.TestCase):
self.assertIn("attr_with_default", attribute_names)
self.assertIn("attr_without_default", attribute_names)
def test_menu_input(self):
from bpy.types import WindowManager
bpy.ops.mesh.primitive_cube_add()
cube = bpy.context.active_object
tree = create_object_mode_mesh_tool_tree(
"TestNodeToolMenu", "geometry.test_node_tool_menu"
)
tree.interface.new_socket("Geometry", in_out='INPUT', socket_type='NodeSocketGeometry')
mode_socket = tree.interface.new_socket(
"Mode", in_out='INPUT', socket_type='NodeSocketMenu')
tree.interface.new_socket("Geometry", in_out='OUTPUT', socket_type='NodeSocketGeometry')
group_input = tree.nodes.new("NodeGroupInput")
group_output = tree.nodes.new("NodeGroupOutput")
menu_switch = tree.nodes.new("GeometryNodeMenuSwitch")
menu_switch.data_type = 'VECTOR'
transform = tree.nodes.new("GeometryNodeTransform")
# The menu switch starts with default items "A" and "B". Rename and add a third so the items become X, Y, Z.
menu_switch.enum_items[0].name = "X"
menu_switch.enum_items[1].name = "Y"
menu_switch.enum_items.new("Z")
menu_switch.inputs["X"].default_value = (10.0, 0.0, 0.0)
menu_switch.inputs["Y"].default_value = (0.0, 10.0, 0.0)
menu_switch.inputs["Z"].default_value = (0.0, 0.0, 10.0)
tree.links.new(group_input.outputs["Geometry"], transform.inputs["Geometry"])
tree.links.new(group_input.outputs["Mode"], menu_switch.inputs["Menu"])
tree.links.new(menu_switch.outputs["Output"], transform.inputs["Translation"])
tree.links.new(transform.outputs["Geometry"], group_output.inputs["Geometry"])
WindowManager.register_node_group_operators()
bpy.ops.geometry.test_node_tool_menu(
'EXEC_DEFAULT',
inputs={mode_socket.identifier: {"value": "Y"}},
)
verts = cube.data.vertices
n = len(verts)
center_x = sum(v.co.x for v in verts) / n
center_y = sum(v.co.y for v in verts) / n
center_z = sum(v.co.z for v in verts) / n
self.assertAlmostEqual(center_x, 0.0, places=4)
self.assertAlmostEqual(center_y, 10.0, places=4)
self.assertAlmostEqual(center_z, 0.0, places=4)
if __name__ == "__main__":
import sys