1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56

import pyeml.core as pm
from maya import OpenMaya
PAINT_CTX = "artAttrColorPerVertexContext"
color_set_representation = {
    "R": "RGB",
    "G": "RGB",
    "B": "RGB",
    "A": "A",
}
def get_color_sets(node):
    color_sets = pm.polyColorSet(node, q=1, allColorSets=1)
    return color_sets or pm.polyColorSet(node, create=1)

def filter_color(color, index, source_color=None):
    if index > 3:
        return color
    is_color = isinstance(source_color, OpenMaya.MColor)
    color_list = list(source_color) if is_color else [0, 0, 0, 1]
    color_list[index] = color[index]
    return OpenMaya.MColor(*color_list)


for node in set(pm.artAttrPaintVertexCtx(PAINT_CTX, q=1, pna=1).split()):
    node = pm.PyNode(node)
    node.displayColors.set(1)
    color_sets = get_color_sets(node)
    main_color_set = color_sets[0]

    mesh = node.__apimfn__()
    
    
    color_array = OpenMaya.MColorArray()
    mesh.getVertexColors(color_array, main_color_set)

    
    vtx_array = OpenMaya.MIntArray()
    for array_index in range(color_array.length()):
        vtx_array.append(array_index)

    final_colors = OpenMaya.MColorArray()
    for channel_index, color_channel in enumerate(cls.CHANNELS):
        
        color_set = "VertexColor{0}".format(color_channel)
        if color_set not in color_sets:
            rpt = color_set_representation.get(color_channel)
            pm.polyColorSet(node, create=1, rpt=rpt, colorSet=color_set)

        mesh.setCurrentColorSetName(color_set)
        final_colors.clear()
        for array_index in range(color_array.length()):
            full_color = color_array[array_index]
            color = filter_color(full_color, index=channel_index)
            final_colors.append(color)
        
        mesh.setVertexColors(final_colors, vtx_array)
    mesh.setCurrentColorSetName(main_color_set)

1
2
3
4
5
6
7
8
9
10

rgb = pm.colorSliderGrp("colorPerVertexColor", q=1, rgb=1)
alpha = pm.floatSliderGrp("colorPerVertexAlpha", q=1, value=1)
rgb.append(alpha)


sel = pm.radioButtonGrp(SINGLE_CONTROL, q=1, sl=1)

color = filter_color(rgb, index=sel)
pm.artAttrPaintVertexCtx(PAINT_CTX, e=1, cl4=tuple(color))

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49

def apply_color_channel(cls):
    index = pm.radioButtonGrp(cls.SINGLE_CONTROL, q=1, sl=1)
    mode = cls.OPTION_ITEMS[index + 1]
    is_rgb = mode == "RGB"
    for node in cls.get_paint_nodes():
        dag_path = node.__apimdagpath__()
        mesh = OpenMaya.MFnMesh(dag_path)
        color_sets = cls.get_color_sets(node)
        main_color_set = color_sets[0]

        current_color_set = mesh.currentColorSetName()

        main_colors = OpenMaya.MColorArray()
        mesh.getVertexColors(main_colors, main_color_set)
        vtx_array = cls.vertex_color_data[node.fullPathName()]
        final_colors = OpenMaya.MColorArray()

        
        if is_rgb:
            
            for channel_index, color_channel in enumerate(cls.CHANNELS):
                final_colors.clear()
                color_set = "VertexColor{0}".format(color_channel)
                mesh.setCurrentColorSetName(color_set)
                for vtx_index in vtx_array:
                    main_color = main_colors[vtx_index]
                    color = cls.filter_color(main_color, channel_index)
                    final_colors.append(color)
                mesh.setVertexColors(final_colors, vtx_array)
        else:
            mode_index = cls.OPTION_ITEMS.index(mode) - 2
            channel_colors = OpenMaya.MColorArray()
            fix_colors = OpenMaya.MColorArray()
            color_set = "VertexColor{0}".format(mode)
            mesh.getVertexColors(channel_colors, color_set)
            
            for vtx_index in vtx_array:
                channel_color = channel_colors[vtx_index]
                main_color = main_colors[vtx_index]
                color = cls.filter_color(channel_color, mode_index, main_color)
                final_colors.append(color)
                fix_color = cls.filter_color(channel_color, mode_index)
                fix_colors.append(fix_color)

            mesh.setVertexColors(fix_colors, vtx_array)
            mesh.setCurrentColorSetName(main_color_set)
            mesh.setVertexColors(final_colors, vtx_array)

        mesh.setCurrentColorSetName(current_color_set)

1
2

from maya import cmds
cmds.radioButtonGrp('artAttrColorChannelChoices',e=1,gbc=[255,0,0])

1
2
3
4
5
6
7
8
9
10

from maya import cmds
parent = cmds.radioButtonGrp('artAttrColorChannelChoices',q=1,parent=1)
print(parent)

window = cmds.window()
column_layout = cmds.columnLayout()
for control in cmds.layout(parent,q=1,childArray=1):
    cmds.control(control, e=1, p=column_layout)
    
cmds.showWindow(window)

1
2
3
4
5
6
7
8
9
10
11

from maya import cmds
window = cmds.window()
cmds.setUITemplate("OptionsTemplate", pushTemplate=1)

column_layout = cmds.columnLayout()
parent = cmds.radioButtonGrp('artAttrColorChannelChoices',q=1,parent=1)
for control in cmds.layout(parent,q=1,childArray=1):
    cmds.control(control, e=1, p=column_layout)
    
cmds.setUITemplate(popTemplate=1)
cmds.showWindow(window)

1
2
3
4
5
6
7
8
9
10
11
12
13

from maya import cmds
window = cmds.window()
cmds.setUITemplate("OptionsTemplate", pushTemplate=1)

column_layout = cmds.columnLayout()
parent = cmds.radioButtonGrp('artAttrColorChannelChoices',q=1,parent=1)
for control in cmds.layout(parent,q=1,childArray=1):
    cmds.control(control, e=1, p=column_layout)
    if control == "artAttrColorChannelChoices":
        cmds.button(label="click me")
    
cmds.setUITemplate(popTemplate=1)
cmds.showWindow(window)

1
2
3
4
5
6
7
8
9
10
11
12

from maya import cmds
parent = cmds.radioButtonGrp('artAttrColorChannelChoices',q=1,parent=1)
cmds.setParent(parent)
cmds.setUITemplate("OptionsTemplate", pushTemplate=1)
column_layout = cmds.columnLayout()

for control in cmds.layout(parent,q=1,childArray=1):
    cmds.control(control, e=1, p=column_layout)
    if control == "artAttrColorChannelChoices":
        cmds.button(label="click me")
    
cmds.setUITemplate(popTemplate=1)

1
2
3
4
5
6
7
8
9
10

import pymel.core as pm
from Qt import QtWidgets
widget = pm.uitypes.toQtObject("artAttrColorChannelChoices")
parent = widget.parent()
print(parent.objectName())

layout = parent.layout()
print(layout)

layout.addWidget(QtWidgets.QPushButton("asd"))

1
2
3
4
5
6
7
8
9
10

import pymel.core as pm
from Qt import QtWidgets
widget = pm.uitypes.toQtObject("artAttrColorChannelChoices")
parent = widget.parent()
layout = parent.layout()
index = layout.indexOf(widget)
widget_list = [layout.takeAt(0).widget() for _ in range(layout.count())]
widget_list.insert(index,QtWidgets.QPushButton("click me"))
for widget in widget_list:
    layout.addWidget(widget)