bps-enum/enum_ex1.py at main · panthistle/bps-enum · 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
121
122
123
124
125
##############################################################################
#                                                                            #
#   Three examples of using the Enumerator Property in Blender 3.3           #
#                          Pan Thistle, 2023                                 #
#                                                                            #
#   This program is free software: you can redistribute it and/or modify     #
#   it under the terms of the GNU General Public License as published by     #
#   the Free Software Foundation, either version 3 of the License, or        #
#   (at your option) any later version.                                      #
#                                                                            #
#   This program is distributed in the hope that it will be useful,          #
#   but WITHOUT ANY WARRANTY; without even the implied warranty of           #
#   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the            #
#   GNU General Public License for more details.                             #
#                                                                            #
#   You should have received a copy of the GNU General Public License        #
#   along with this program.  If not, see <https://www.gnu.org/licenses/>.   #
#                                                                            #
##############################################################################


# Example 1: coordinate two enum properties with mutually exclusive values


import bpy


# ------------------------------------------------------------------------------
#
# ----------------------------- PROPERTIES -------------------------------------


class DENUMSYNC_props(bpy.types.PropertyGroup):
    def up_items(self, context):
        # this function will populate 'up' enum with valid entries
        # that do not clash with currently selected 'track' value
        names = ["X", "Y", "Z"]
        t = self.get("track", 1)
        items = []
        for i in range(3):
            if t in [i, i + 3]:
                continue
            items.append((names[i], names[i], names[i]))
        return items

    track: bpy.props.EnumProperty(
        name="Track",
        description="track axis",
        items=(
            ("X", "X", "X"),
            ("Y", "Y", "Y"),
            ("Z", "Z", "Z"),
            ("-X", "-X", "-X"),
            ("-Y", "-Y", "-Y"),
            ("-Z", "-Z", "-Z"),
        ),
        default="Y",
    )

    up: bpy.props.EnumProperty(
        name="Up",
        description="up axis",
        items=up_items,
    )


# ------------------------------------------------------------------------------
#
# --------------------------------- PANEL --------------------------------------


class DENUMSYNC_PT_ui(bpy.types.Panel):
    bl_space_type = "VIEW_3D"
    bl_region_type = "UI"
    bl_context = "objectmode"
    bl_category = "DENUMSYNC"
    bl_label = "DENUMSYNC"

    def draw(self, context):
        props = context.scene.denum_tu
        layout = self.layout
        box = layout.box()
        col = box.column(align=True)
        row = col.row(align=True)
        row.label(text="Track")
        row.label(text="Up")
        row = col.row(align=True)
        row.prop(props, "track", text="")
        row.prop(props, "up", text="")


# ------------------------------------------------------------------------------
#
# ------------------------------ REGISTRATION ----------------------------------

classes = (
    DENUMSYNC_props,
    DENUMSYNC_PT_ui,
)


def register():
    from bpy.utils import register_class

    for cls in classes:
        register_class(cls)
    bpy.types.Scene.denum_tu = bpy.props.PointerProperty(type=DENUMSYNC_props)


def unregister():
    from bpy.utils import unregister_class

    for cls in reversed(classes):
        unregister_class(cls)
    del bpy.types.Scene.denum_tu


# ------------------------------------------------------------------------------
#
# ------------------------------- RUN MODULE -----------------------------------

if __name__ == "__main__":
    print("-" * 30)
    register()
    unregister()