GameRig/ui.py at main · Arminando/GameRig · 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
import bpy

from rigify.ui import DATA_PT_rigify
from rigify.utils.misc import verify_armature_obj
from rigify.utils.rig import get_rigify_type

from .utils.ui import is_gamerig_metarig
from .operators.upgrade_metarig_types import outdated_types as outdated_game_types

def draw_gamerig_rigify_button(self, context):
    C = context

    layout = self.layout
    obj = verify_armature_obj(C.object)


    # if not is_gamerig_metarig(context.object):
    #     return

    layout.operator("pose.gamerig_generate", text="Generate GameRig", icon='GHOST_ENABLED')


    show_upgrade_game_face = False
    show_upgrade_game_types = False

    old_rig = ''
    old_bone = ''

    for b in obj.pose.bones:
        old_rig = get_rigify_type(b)
        if b.rigify_type == 'game.faces.super_face':
            show_upgrade_game_face = True
        elif b.rigify_type in outdated_game_types.keys():
            show_upgrade_game_types = True

    if show_upgrade_game_face:
        layout.label(text="This metarig uses the old game face rig.", icon='INFO')
        layout.operator("pose.gamerig_upgrade_game_face")
    if show_upgrade_game_types:
        layout.label(text="This metarig uses the old game types.", icon='ERROR')
        layout.operator("pose.gamerig_upgrade_types")

    armature_id_store = C.object.data

    has_non_game_rigs = False
    is_game_metarig = is_gamerig_metarig(obj)

    for b in obj.pose.bones:
        if b.rigify_type and "game" not in b.rigify_type:
            has_non_game_rigs = True
            break

    if has_non_game_rigs and is_game_metarig:
        layout.label(text="Non Game types detected", icon='ERROR')


classes = [

]


def register():
    from bpy.utils import register_class

    bpy.types.DATA_PT_rigify.prepend(draw_gamerig_rigify_button)

    # Classes.
    for c in classes:
        register_class(c)


def unregister():
    from bpy.utils import unregister_class

    bpy.types.DATA_PT_rigify.remove(draw_gamerig_rigify_button)

    # Classes.
    for c in classes:
        unregister_class(c)