From 3db252bcc0066bf028052734bc766d2fc6f74dbe Mon Sep 17 00:00:00 2001 From: Dion Moult Date: Tue, 8 Feb 2022 15:50:55 +1100 Subject: [PATCH] Minor fix --- src/blenderbim/blenderbim/tool/geometry.py | 2 ++ src/blenderbim/test/tool/test_geometry.py | 15 +++++++++++++++ 2 files changed, 17 insertions(+) diff --git a/src/blenderbim/blenderbim/tool/geometry.py b/src/blenderbim/blenderbim/tool/geometry.py index 1b4011d4e7..26c0369cec 100644 --- a/src/blenderbim/blenderbim/tool/geometry.py +++ b/src/blenderbim/blenderbim/tool/geometry.py @@ -305,6 +305,8 @@ class Geometry(blenderbim.core.tool.Geometry): for node in slot.material.node_tree.nodes: if node.type == "TEX_COORD" and node.outputs['UV'].links: return True + elif node.type == "UVMAP" and node.outputs['UV'].links and node.uv_map: + return True return False @classmethod diff --git a/src/blenderbim/test/tool/test_geometry.py b/src/blenderbim/test/tool/test_geometry.py index 26f8aba986..005b14e53b 100644 --- a/src/blenderbim/test/tool/test_geometry.py +++ b/src/blenderbim/test/tool/test_geometry.py @@ -488,6 +488,21 @@ class TestShouldGenerateUVs(NewFile): material.node_tree.links.new(node.inputs["Vector"], coords.outputs["UV"]) assert subject.should_generate_uvs(obj) is True + def test_accepts_a_uv_map_node(self): + obj = bpy.data.objects.new("Object", bpy.data.meshes.new("Mesh")) + material = bpy.data.materials.new("Material") + obj.data.materials.append(material) + material.use_nodes = True + + bsdf = material.node_tree.nodes["Principled BSDF"] + node = material.node_tree.nodes.new(type="ShaderNodeTexImage") + material.node_tree.links.new(bsdf.inputs["Base Color"], node.outputs["Color"]) + + coords = material.node_tree.nodes.new(type="ShaderNodeUVMap") + coords.uv_map = "UVMap" + material.node_tree.links.new(node.inputs["Vector"], coords.outputs["UV"]) + assert subject.should_generate_uvs(obj) is True + class TestShouldUsePresentationStyleAssignment(NewFile): def test_run(self):