#!/usr/bin/env python
# -*- coding: utf-8 -*-

import maya.cmds as cmds

def sg_set_color(object_name, rgb, r, g, b, index, enabled):
    """
    设置对象的显示颜色
    Args:
        object_name (str): 对象名称
        rgb (int): 是否使用RGB颜色模式
        r (int): 红色值 (0-255)
        g (int): 绿色值 (0-255)
        b (int): 蓝色值 (0-255)
        index (int): 颜色索引值
        enabled (int): 是否启用颜色覆盖
    """
    # 检查对象是否存在
    if cmds.objExists(object_name):
        # 设置颜色覆盖属性
        cmds.setAttr(f"{object_name}.ove", enabled)
        cmds.setAttr(f"{object_name}.overrideColor", index)
        cmds.setAttr(f"{object_name}.ovrgbf", rgb)
        
        # 将RGB值转换为0-1范围
        red = r / 255.0
        green = g / 255.0
        blue = b / 255.0
        
        # 设置RGB颜色
        cmds.setAttr(f"{object_name}.ovrgb", red, green, blue, type="float3")

# 如果直接运行此脚本
if __name__ == '__main__':
    pass