adds option to color attribute to override the attribute

This commit is contained in:
Michel Fedde 2024-01-30 17:20:59 +01:00
parent 9765e28786
commit 53366cc71d
2 changed files with 19 additions and 3 deletions

View file

@ -29,6 +29,11 @@ class BakeToIDProperties(PropertyGroup):
name="Color Attribute",
default="ID_MASK",
)
target_vertex_color_override_attribute: BoolProperty(
name="Override Color Attribute",
default=True,
description="If set true, the attribute will be deleted and recreated, if it already exists. If set false, the data will just be overwritten."
)
adv_total_hues: IntProperty(
name="Total Hues",

View file

@ -3,7 +3,8 @@ name = 'Color Attribute / Vertex Color'
description = 'Bakes the ID onto a color attribute (previously known as Vertex Color)'
connected_properties = [
'target_vertex_color_attribute_name'
'target_vertex_color_attribute_name',
'target_vertex_color_override_attribute'
]
@ -21,12 +22,22 @@ def paint_targets(props, targets, colors):
layer_name = props.target_vertex_color_attribute_name
for mesh in sorted_targets:
if layer_name in mesh.attributes:
mesh.attributes.remove(mesh.attributes[layer_name])
color_attribute = mesh.attributes.new(name=layer_name, type='FLOAT_COLOR', domain='CORNER')
color_attribute = get_color_attribute(props, mesh.attributes, layer_name)
for (indecies, color) in sorted_targets[mesh]:
for polygon in indecies:
for idx in polygon.loop_indices:
color_attribute.data[idx].color = (color[0], color[1], color[2], 1.0)
def get_color_attribute(props, attributes, layer_name):
if layer_name in attributes:
if not props.target_vertex_color_override_attribute:
return attributes[layer_name]
attributes.remove(attributes[layer_name])
return attributes.new(name=layer_name, type='FLOAT_COLOR', domain='CORNER')