CM3D2 Converter.misc_DATA_PT_vertex_groups

  1# 「プロパティ」エリア → 「メッシュデータ」タブ → 「頂点グループ」パネル
  2import re
  3import bpy
  4from . import common
  5from . import compat
  6
  7
  8# メニュー等に項目追加
  9def menu_func(self, context):
 10    ob = context.active_object
 11    if not ob or len(ob.vertex_groups) == 0 and ob.type != 'MESH':
 12        return
 13
 14    flag = False
 15    for vertex_group in ob.vertex_groups:
 16        if not flag and re.search(r'[_ ]([rRlL])[_ ]', vertex_group.name):
 17            flag = True
 18        if not flag and vertex_group.name.count('*') == 1:
 19            if re.search(r'\.([rRlL])$', vertex_group.name):
 20                flag = True
 21        if flag:
 22            col = self.layout.column(align=True)
 23            col.label(text="CM3D2用 頂点グループ名変換", icon_value=common.kiss_icon())
 24            row = col.row(align=True)
 25            row.operator('object.decode_cm3d2_vertex_group_names', icon='BLENDER', text="CM3D2 → Blender")
 26            row.operator('object.encode_cm3d2_vertex_group_names', icon_value=common.kiss_icon(), text="Blender → CM3D2")
 27            break
 28
 29
 30@compat.BlRegister()
 31class CNV_OT_decode_cm3d2_vertex_group_names(bpy.types.Operator):
 32    bl_idname = 'object.decode_cm3d2_vertex_group_names'
 33    bl_label = "頂点グループ名をCM3D2用→Blender用に変換"
 34    bl_description = "CM3D2で使われてるボーン名(頂点グループ名)をBlenderで左右対称編集できるように変換します"
 35    bl_options = {'REGISTER', 'UNDO'}
 36
 37    @classmethod
 38    def poll(cls, context):
 39        import re
 40        ob = context.active_object
 41        if ob and ob.type == 'MESH':
 42            if ob.vertex_groups.active:
 43                for vg in ob.vertex_groups:
 44                    if re.search(r'[_ ]([rRlL])[_ ]', vg.name):
 45                        return True
 46        return False
 47
 48    def execute(self, context):
 49        ob = context.active_object
 50        me = ob.data
 51        convert_count = 0
 52        context.window_manager.progress_begin(0, len(ob.vertex_groups))
 53        for vg_index, vg in enumerate(ob.vertex_groups[:]):
 54            context.window_manager.progress_update(vg_index)
 55            vg_name = common.decode_bone_name(vg.name)
 56            if vg_name != vg.name:
 57                if vg_name in ob.vertex_groups:
 58                    target_vg = ob.vertex_groups[vg_name]
 59                    for vert in me.vertices:
 60                        try:
 61                            weight = vg.weight(vert.index)
 62                        except:
 63                            weight = 0.0
 64                        try:
 65                            target_weight = target_vg.weight(vert.index)
 66                        except:
 67                            target_weight = 0.0
 68                        if 0.0 < weight + target_weight:
 69                            target_vg.add([vert.index], weight + target_weight, 'REPLACE')
 70                    ob.vertex_groups.remove(vg)
 71                else:
 72                    vg.name = vg_name
 73                convert_count += 1
 74        if convert_count == 0:
 75            self.report(type={'WARNING'}, message="変換できる名前が見つかりませんでした")
 76        else:
 77            self.report(type={'INFO'}, message="頂点グループ名をBlender用に変換しました")
 78        context.window_manager.progress_end()
 79        return {'FINISHED'}
 80
 81
 82@compat.BlRegister()
 83class CNV_OT_encode_cm3d2_vertex_group_names(bpy.types.Operator):
 84    bl_idname = 'object.encode_cm3d2_vertex_group_names'
 85    bl_label = "頂点グループ名をBlender用→CM3D2用に変換"
 86    bl_description = "CM3D2で使われてるボーン名(頂点グループ名)に戻します"
 87    bl_options = {'REGISTER', 'UNDO'}
 88
 89    @classmethod
 90    def poll(cls, context):
 91        ob = context.active_object
 92        if ob and ob.type == 'MESH' and ob.vertex_groups.active:
 93            for vg in ob.vertex_groups:
 94                if vg.name.count('*') == 1 and re.search(r'\.([rRlL])$', vg.name):
 95                    return True
 96        return False
 97
 98    def execute(self, context):
 99        ob = context.active_object
100        me = ob.data
101        convert_count = 0
102        context.window_manager.progress_begin(0, len(ob.vertex_groups))
103        for vg_index, vg in enumerate(ob.vertex_groups[:]):
104            context.window_manager.progress_update(vg_index)
105            vg_name = common.encode_bone_name(vg.name)
106            if vg_name != vg.name:
107                if vg_name in ob.vertex_groups:
108                    target_vg = ob.vertex_groups[vg_name]
109                    for vert in me.vertices:
110                        try:
111                            weight = vg.weight(vert.index)
112                        except:
113                            weight = 0.0
114
115                        try:
116                            target_weight = target_vg.weight(vert.index)
117                        except:
118                            target_weight = 0.0
119                        if 0.0 < weight + target_weight:
120                            target_vg.add([vert.index], weight + target_weight, 'REPLACE')
121                    ob.vertex_groups.remove(vg)
122                else:
123                    vg.name = vg_name
124                convert_count += 1
125        if convert_count == 0:
126            self.report(type={'WARNING'}, message="変換できる名前が見つかりませんでした")
127        else:
128            self.report(type={'INFO'}, message="頂点グループ名をCM3D2用に戻しました")
129        context.window_manager.progress_end()
130        return {'FINISHED'}