Sollumz/sollumz_pie.py at main · Sollumz/Sollumz · 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
109
110
111
112
113
114
115
116
117
118
119
120
import bpy
from bpy.types import Menu
from .sollumz_preferences import get_addon_preferences


def find_missing_files(filepath):
    bpy.ops.file.find_missing_files(directory=filepath)
    return {'FINISHED'}


class SOLLUMZ_MT_pie_menu(Menu):
    bl_idname = "SOLLUMZ_MT_pie_menu"
    bl_label = "Sollumz Pie Menu"

    def draw(self, context):

        layout = self.layout

        if get_addon_preferences(context).legacy_import_export:
            import_op = "sollumz.import_assets_legacy"
            export_op = "sollumz.export_assets_legacy"
        else:
            import_op = "sollumz.import_assets"
            export_op = "sollumz.export_assets"

        pie = layout.menu_pie()
        # Left
        pie.operator("sollumz.autoconvertmaterials",
                     text="Convert Material", icon='NODE_MATERIAL')
        # Right
        pie.operator("sollumz.addobjasentity", icon='OBJECT_DATA')

        # Bottom
        pie.operator("sollumz.load_flag_preset",
                     text="Apply Flag Preset", icon='ALIGN_TOP')
        # Top
        pie.operator("file.find_missing_files",
                     text="Find Missing Textures", icon='VIEWZOOM')
        # Top-left
        pie.operator(import_op, icon="IMPORT")
        # Top-right
        op = pie.operator(export_op, icon="EXPORT")
        if context.scene.sollumz_export_path != "":
            op.directory = context.scene.sollumz_export_path
            op.direct_export = True

        # Bottom-left
        pie.operator("sollumz.converttodrawable", icon='CUBE')
        # Bottom-right
        pie.operator("sollumz.converttocomposite", icon='CUBE')


class SOLLUMZ_MT_view_pie_menu(Menu):
    bl_idname = "SOLLUMZ_MT_view_pie_menu"
    bl_label = "Sollumz Object Visibility"

    def draw(self, context):
        from .sollumz_ui import SOLLUMZ_UI_NAMES
        from .lods import (
            LODLevel,
            SOLLUMZ_OT_set_lod_level,
            SOLLUMZ_OT_hide_object,
            SOLLUMZ_OT_HIDE_COLLISIONS,
            SOLLUMZ_OT_HIDE_SHATTERMAPS,
            SOLLUMZ_OT_SHOW_COLLISIONS,
            SOLLUMZ_OT_SHOW_SHATTERMAPS
        )

        layout = self.layout
        scene = context.scene

        pie = layout.menu_pie()
        # Left
        pie.operator((SOLLUMZ_OT_HIDE_COLLISIONS if scene.sollumz_show_collisions else SOLLUMZ_OT_SHOW_COLLISIONS).bl_idname)
        # Right, Bottom, Top
        for lod in (LODLevel.MEDIUM, LODLevel.VERYLOW, LODLevel.VERYHIGH):
            pie.operator(SOLLUMZ_OT_set_lod_level.bl_idname, text=SOLLUMZ_UI_NAMES[lod]).lod_level = lod
        # Top-left
        pie.operator((SOLLUMZ_OT_HIDE_SHATTERMAPS if scene.sollumz_show_shattermaps else SOLLUMZ_OT_SHOW_SHATTERMAPS).bl_idname)
        # Top-right
        pie.operator(SOLLUMZ_OT_set_lod_level.bl_idname, text=SOLLUMZ_UI_NAMES[LODLevel.HIGH]).lod_level = LODLevel.HIGH
        # Bottom-left
        pie.operator(SOLLUMZ_OT_hide_object.bl_idname, text="Hide")
        # Bottom-right
        pie.operator(SOLLUMZ_OT_set_lod_level.bl_idname, text=SOLLUMZ_UI_NAMES[LODLevel.LOW]).lod_level = LODLevel.LOW


addon_keymaps = []


def register():

    # Assigns default keybinding
    wm = bpy.context.window_manager
    kc = wm.keyconfigs.addon
    if kc:
        km = kc.keymaps.new(name='3D View', space_type='VIEW_3D')
        kmi = km.keymap_items.new(
            "wm.call_menu_pie", type='V', value='PRESS', shift=False)
        kmi.properties.name = "SOLLUMZ_MT_pie_menu"

        addon_keymaps.append((km, kmi))

        kmi = km.keymap_items.new(
            "wm.call_menu_pie", type='V', value='PRESS', shift=True)
        kmi.properties.name = "SOLLUMZ_MT_view_pie_menu"

        addon_keymaps.append((km, kmi))


def unregister():

    # default keybinding
    for km, kmi in addon_keymaps:
        km.keymap_items.remove(kmi)
    addon_keymaps.clear()


if __name__ == "__main__":
    register()