@volter/blender-engine 0.1.147 → 0.1.149

package/browser/addons/vrm/editor/vrm0/__init__.py

@@ -0,0 +1 @@
+# SPDX-License-Identifier: MIT OR GPL-3.0-or-later

package/browser/addons/vrm/editor/vrm0/gizmo_group.py

@@ -0,0 +1,63 @@
+# SPDX-License-Identifier: MIT OR GPL-3.0-or-later
+
+from typing import ClassVar
+
+from bpy.types import Armature, Context, GizmoGroup
+
+from ..extension_accessor import get_armature_extension
+
+
+# https://gist.github.com/FujiSunflower/09fdabc7ca991f8292657abc4ef001b0
+class Vrm0FirstPersonBoneOffsetGizmoGroup(GizmoGroup):
+    bl_idname = "VRM_GGT_vrm0_first_person_bone_offset"
+    bl_label = "First Person Bone Offset Gizmo"
+    bl_space_type = "VIEW_3D"
+    bl_region_type = "WINDOW"
+    bl_options: ClassVar = {"3D", "PERSISTENT"}
+
+    @classmethod
+    def poll(cls, context: Context) -> bool:
+        active_object = context.active_object
+        if not active_object:
+            return False
+        return active_object.type == "ARMATURE"
+
+    def setup(self, context: Context) -> None:
+        active_object = context.active_object
+        if not active_object:
+            return
+        armature_data = active_object.data
+        if not isinstance(armature_data, Armature):
+            return
+        ext = get_armature_extension(armature_data)
+        first_person = ext.vrm0.first_person
+        first_person_bone = armature_data.bones[
+            first_person.first_person_bone.bone_name
+        ]
+        gizmo = self.gizmos.new("GIZMO_GT_move_3d")
+        gizmo.target_set_prop("offset", first_person, "first_person_bone_offset")
+        gizmo.matrix_basis = first_person_bone.matrix_local
+        gizmo.draw_style = "CROSS_2D"
+        gizmo.draw_options = {"ALIGN_VIEW"}
+        gizmo.color = 1.0, 0.5, 0.0
+        gizmo.alpha = 0.5
+        gizmo.color_highlight = 1.0, 0.5, 1.0
+        gizmo.alpha_highlight = 0.5
+        gizmo.scale_basis = 0.25
+
+        # self.first_person_gizmo = gizmo
+
+    def refresh(self, context: Context) -> None:
+        active_object = context.active_object
+        if not active_object:
+            return
+        armature_data = active_object.data
+        if not isinstance(armature_data, Armature):
+            return
+        # ext = get_armature_extension(armature_data)
+        # gizmo = self.first_person_gizmo
+        # first_person = ext.vrm0.first_person
+        # first_person_bone = armature_data.bones[
+        #     first_person.first_person_bone.bone_name
+        # ]
+        # gizmo.matrix_basis = first_person_bone.matrix_local

package/browser/addons/vrm/editor/vrm0/handler.py

@@ -0,0 +1,23 @@
+# SPDX-License-Identifier: MIT OR GPL-3.0-or-later
+import bpy
+from bpy.app.handlers import persistent
+
+from ...common.logger import get_logger
+from ..extension_accessor import get_armature_extension
+from .property_group import Vrm0BlendShapeGroupPropertyGroup
+
+_logger = get_logger(__name__)
+
+
+@persistent
+def frame_change_post(_unused: object) -> None:
+    context = bpy.context
+    Vrm0BlendShapeGroupPropertyGroup.apply_pending_preview_update_to_armatures(context)
+
+
+@persistent
+def save_pre(_unused: object) -> None:
+    context = bpy.context
+    for armature_data in context.blend_data.armatures:
+        vrm0 = get_armature_extension(armature_data).vrm0
+        vrm0.secondary_animation.fixup(armature_object=None)

package/browser/addons/vrm/editor/vrm0/menu.py

@@ -0,0 +1,145 @@
+# SPDX-License-Identifier: MIT OR GPL-3.0-or-later
+from collections.abc import Callable
+from typing import Final
+
+from bpy.types import Armature, Context, Menu, Object, UILayout
+
+from ...common.logger import get_logger
+from ..extension_accessor import get_armature_extension
+from ..ops import layout_operator
+from .ops import (
+    VRM_OT_assign_vrm0_secondary_animation_group_collider_group,
+    VRM_OT_restore_vrm0_blend_shape_group_bind_object,
+    VRM_OT_unassign_vrm0_secondary_animation_group_collider_group,
+)
+from .property_group import (
+    Vrm0SecondaryAnimationColliderGroupReferencePropertyGroup,
+    Vrm0SecondaryAnimationGroupPropertyGroup,
+)
+
+_logger = get_logger(__name__)
+
+
+class VRM_MT_vrm0_blend_shape_master(Menu):
+    bl_label = "Blend Shape Proxy Menu"
+    bl_idname = "VRM_MT_vrm0_blend_shape_master"
+
+    CONTEXT_POINTER_ARMATURE: Final = bl_idname + "_armature"
+
+    @classmethod
+    def layout_context_pointer_set(cls, armature: Object) -> Callable[[UILayout], None]:
+        return lambda layout: layout.context_pointer_set(
+            cls.CONTEXT_POINTER_ARMATURE, armature
+        )
+
+    def draw(self, context: Context) -> None:
+        layout = self.layout
+
+        armature = getattr(context, self.CONTEXT_POINTER_ARMATURE, None)
+        if not isinstance(armature, Object):
+            return
+
+        op = layout_operator(layout, VRM_OT_restore_vrm0_blend_shape_group_bind_object)
+        op.armature_object_name = armature.name
+
+
+class VRM_MT_vrm0_secondary_animation_group_collider_group(Menu):
+    bl_label = "Collider Group Assignment Menu"
+    bl_idname = "VRM_MT_vrm0_secondary_animation_group_collider_group"
+
+    CONTEXT_POINTER_BONE_GROUP: Final = bl_idname + "_bone_group"
+    CONTEXT_POINTER_COLLIDER_GROUP_REFERENCE: Final = (
+        bl_idname + "_collider_group_reference"
+    )
+
+    @classmethod
+    def draw_input_layout(
+        cls,
+        layout: UILayout,
+        bone_group: Vrm0SecondaryAnimationGroupPropertyGroup,
+        collider_group_reference: (
+            Vrm0SecondaryAnimationColliderGroupReferencePropertyGroup
+        ),
+        *,
+        icon: str = "PIVOT_INDIVIDUAL",
+    ) -> UILayout:
+        row = layout.row(align=True)
+        row.context_pointer_set(
+            cls.CONTEXT_POINTER_BONE_GROUP,
+            bone_group,
+        )
+        row.context_pointer_set(
+            cls.CONTEXT_POINTER_COLLIDER_GROUP_REFERENCE,
+            collider_group_reference,
+        )
+        row.menu(
+            cls.bl_idname,
+            text=collider_group_reference.collider_group_display_name or " ",
+            icon=icon,
+            translate=False,
+        )
+        return row
+
+    def draw(self, context: Context) -> None:
+        layout = self.layout
+
+        bone_group = getattr(context, self.CONTEXT_POINTER_BONE_GROUP, None)
+        if not isinstance(
+            bone_group,
+            Vrm0SecondaryAnimationGroupPropertyGroup,
+        ):
+            return
+
+        collider_group_reference = getattr(
+            context, self.CONTEXT_POINTER_COLLIDER_GROUP_REFERENCE, None
+        )
+        if not isinstance(
+            collider_group_reference,
+            Vrm0SecondaryAnimationColliderGroupReferencePropertyGroup,
+        ):
+            return
+
+        armature_data = collider_group_reference.id_data
+        if not isinstance(armature_data, Armature):
+            return
+
+        if bone_group.id_data != armature_data:
+            return
+
+        assigned_uuids: set[str] = {
+            collider_group_reference.collider_group_uuid
+            for collider_group_reference in bone_group.collider_groups
+            if collider_group_reference.collider_group_uuid
+        }
+
+        vrm0 = get_armature_extension(armature_data).vrm0
+        secondary_animation = vrm0.secondary_animation
+        collider_group_reference_path = collider_group_reference.path_from_id()
+
+        if collider_group_reference.collider_group_uuid:
+            unassign_op = layout_operator(
+                layout,
+                VRM_OT_unassign_vrm0_secondary_animation_group_collider_group,
+                text="<Unassign>",
+                translate=True,
+                icon="REMOVE",
+            )
+            unassign_op.armature_data_name = armature_data.name
+            unassign_op.collider_group_reference_path = collider_group_reference_path
+
+        for collider_group in secondary_animation.collider_groups:
+            if not collider_group.uuid:
+                continue
+            if collider_group.uuid in assigned_uuids:
+                continue
+
+            assign_op = layout_operator(
+                layout,
+                VRM_OT_assign_vrm0_secondary_animation_group_collider_group,
+                text=collider_group.display_name,
+                translate=False,
+                icon="PIVOT_INDIVIDUAL",
+            )
+            assign_op.armature_data_name = armature_data.name
+            assign_op.collider_group_reference_path = collider_group_reference_path
+            assign_op.collider_group_uuid = collider_group.uuid