AutoMech/BL_WeightRig.py at 2_90 · HuiProject/AutoMech · GitHub

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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
import bpy
import bmesh
from . BL_Tool import *


class WeightRig(bpy.types.Operator):
    bl_idname = "aw.weightrig"
    bl_label = "Weight Rig"
    bl_description = "Paint Weight for Rig,在此之前需要应用修改器"
    bl_options = {"REGISTER"}

    @classmethod
    def poll(cls, context):
        return True

    def execute(self, context):

        RigMaster = bpy.data.objects["7BindRig"]
        for Rigobj in bpy.data.collections["7BindRig"].objects:
            if Rigobj.name !='7BindRig':
            #for Rigobj in RigMaster.objects:

                RigMaster.select_set(False)
                Rigobj.select_set(True)
                bpy.context.view_layer.objects.active = Rigobj

                bpy.ops.object.mode_set(mode='EDIT')
                bpy.ops.mesh.select_mode(use_extend=False, use_expand=False, type='VERT', action='TOGGLE')

                me = Rigobj.data

                bm = bmesh.from_edit_mesh(me)
                #设置你的顶点索引值
                v_index = 0

                #遍历顶点组
                for group in Rigobj.vertex_groups:

                    #将当前顶点组设置为活动的
                    if Rigobj.name == group.name:
                        bpy.ops.object.vertex_group_set_active(group=str(group.name))

                    # 选择活动顶点组
                        bpy.ops.object.vertex_group_select()

                    #获取当前选择的所有顶点(可能不用)
                        #vertices = [v for v in bm.verts if (v.select and not v.hide)]


                        bpy.ops.mesh.select_all(action='SELECT')

                        bpy.ops.object.vertex_group_assign()
                    #顶点的迭代 vertex iteration(可能不用)
                        '''
                        for vertex in vertices:
                            if vertex.index ==  v_index:
                                print("Vertex ", v_index, " is in ", group.name)
                        '''
                    #取消选择当前顶点组
                        bpy.ops.object.vertex_group_deselect()

                bmesh.update_edit_mesh(me, True)
                bpy.ops.object.mode_set(mode='OBJECT')

                #在viewport中显示更新 show the updates in the viewport


        '''
            #if Rigobj.vertex_groups[GROUP_NAME]:
            if Rigobj.name == Rigobj.vertex_groups[GROUP_NAME].name:
                    bpy.ops.object.mode_set(mode='EDIT')
                    bpy.ops.mesh.select_mode(use_extend=False, use_expand=False, type='VERT', action='TOGGLE')
                    bpy.ops.mesh.select_all(action='SELECT')
                    RigobjVg = Rigobj.vertex_groups[GROUP_NAME].get
                    bpy.ops.object.vertex_group_assign()


            #遍历顶点组
            for group in Rigobj.vertex_groups:

                #将当前顶点组设置为活动的
                bpy.ops.object.vertex_group_set_active(group=str(group.name))

                # 选择活动顶点组
                bpy.ops.object.vertex_group_select()

                #获取当前选择的所有顶点
                vertices = [v for v in bm.verts if (v.select and not v.hide)]

                #顶点的迭代 vertex iteration
                for vertex in vertices:
                    if vertex.index ==  v_index:
                        print("Vertex ", v_index, " is in ", group.name)

                #取消选择当前顶点组
                bpy.ops.object.vertex_group_deselect()

        '''

            #gi = Rigobj.vertex_groups[GROUP_NAME].index
            #for v in Rigobj.data.vertices:
                #for g in v.groups:
            #for RigobjVg in Rigobj.vertex_groups:
            #if Rigobj.name == Rigobj.vertex_groups:

        self.report({'INFO'}, "7BindRig")
        return {"FINISHED"}