blender python api 将指定的顶点组（water）转换为颜色属性water_colors

1.选中物体，进入权重绘制模式

2.代码：

import bpy

# 获取当前活动的物体
obj = bpy.context.object

# 确保物体是网格类型
if obj.type != 'MESH':
    print("当前激活的对象不是网格类型。")
    #exit()  # 使用exit()来提前退出脚本

# 获取名为“water”的顶点组
vertex_group_name = "water"
vg = obj.vertex_groups.get(vertex_group_name)
if not vg:
    print(f"未找到名为 '{vertex_group_name}' 的顶点组。")
    #exit()  # 使用exit()来提前退出脚本

# 获取顶点组的权重
weights = [vg.weight(i) for i in range(len(obj.data.vertices))]

# 检查物体的属性中是否已经存在名为"water_colors"的颜色属性，如果没有则创建一个
color_attr_name = "water_colors"
if color_attr_name not in obj.data.attributes:
    color_attr = obj.data.attributes.new(
        name=color_attr_name,
        type='BYTE_COLOR',
        domain='POINT',
    )
else:
    color_attr = obj.data.attributes[color_attr_name]

# 设置顶点颜色属性
for i, vertex in enumerate(obj.data.vertices):
    # 将权重值映射到颜色上，这里简单地将权重映射到颜色的红色通道上
    # 可以根据需要调整颜色映射逻辑
    weight = weights[i]
    color = (int(weight * 255), 0, 0, 255)  # 这里设置为纯红色，根据权重值变化
    color_attr.data[i].color = color

print(f"顶点组 '{vertex_group_name}' 的权重已成功转换为顶点颜色属性。")