@volter/blender-engine 0.1.148 → 0.1.149

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

@@ -0,0 +1,246 @@
+# 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 ..extension_accessor import get_armature_extension
+from ..ops import layout_operator
+from .ops import (
+    VRM_OT_assign_spring_bone1_automatically,
+    VRM_OT_assign_spring_bone1_collider_group_collider,
+    VRM_OT_assign_spring_bone1_from_mmd,
+    VRM_OT_assign_spring_bone1_from_vrm0,
+    VRM_OT_assign_spring_bone1_spring_collider_group,
+    VRM_OT_unassign_spring_bone1_collider_group_collider,
+    VRM_OT_unassign_spring_bone1_spring_collider_group,
+)
+from .property_group import (
+    SpringBone1ColliderGroupPropertyGroup,
+    SpringBone1ColliderGroupReferencePropertyGroup,
+    SpringBone1ColliderReferencePropertyGroup,
+    SpringBone1SpringPropertyGroup,
+)
+
+
+class VRM_MT_spring_bone1_spring_bones(Menu):
+    bl_label = "Spring Bone Menu"
+    bl_idname = "VRM_MT_spring_bone1_spring_bones"
+
+    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
+
+        auto_detection_op = layout_operator(
+            layout, VRM_OT_assign_spring_bone1_automatically
+        )
+        auto_detection_op.armature_object_name = armature.name
+
+        assign_vrm0_op = layout_operator(layout, VRM_OT_assign_spring_bone1_from_vrm0)
+        assign_vrm0_op.armature_object_name = armature.name
+
+        assign_mmd_op = layout_operator(layout, VRM_OT_assign_spring_bone1_from_mmd)
+        assign_mmd_op.armature_object_name = armature.name
+
+
+class VRM_MT_spring_bone1_collider_group_collider(Menu):
+    bl_label = "Collider Assignment Menu"
+    bl_idname = "VRM_MT_spring_bone1_collider_group_collider"
+
+    CONTEXT_POINTER_COLLIDER_GROUP: Final = bl_idname + "_collider_group"
+    CONTEXT_POINTER_COLLIDER_REFERENCE: Final = bl_idname + "_collider_reference"
+
+    @classmethod
+    def draw_input_layout(
+        cls,
+        layout: UILayout,
+        collider_group: SpringBone1ColliderGroupPropertyGroup,
+        collider_reference: SpringBone1ColliderReferencePropertyGroup,
+        *,
+        icon: str = "SPHERE",
+    ) -> UILayout:
+        row = layout.row(align=True)
+        row.context_pointer_set(
+            cls.CONTEXT_POINTER_COLLIDER_GROUP,
+            collider_group,
+        )
+        row.context_pointer_set(
+            cls.CONTEXT_POINTER_COLLIDER_REFERENCE,
+            collider_reference,
+        )
+        row.menu(
+            cls.bl_idname,
+            text=collider_reference.collider_display_name or " ",
+            icon=icon,
+            translate=False,
+        )
+        return row
+
+    def draw(self, context: Context) -> None:
+        layout = self.layout
+
+        collider_group = getattr(context, self.CONTEXT_POINTER_COLLIDER_GROUP, None)
+        if not isinstance(collider_group, SpringBone1ColliderGroupPropertyGroup):
+            return
+
+        collider_reference = getattr(
+            context, self.CONTEXT_POINTER_COLLIDER_REFERENCE, None
+        )
+        if not isinstance(
+            collider_reference, SpringBone1ColliderReferencePropertyGroup
+        ):
+            return
+
+        armature_data = collider_reference.id_data
+        if not isinstance(armature_data, Armature):
+            return
+
+        if collider_group.id_data != armature_data:
+            return
+
+        assigned_collider_uuids: set[str] = {
+            collider_reference.collider_uuid
+            for collider_reference in collider_group.colliders
+            if collider_reference.collider_uuid
+        }
+
+        spring_bone1 = get_armature_extension(armature_data).spring_bone1
+        collider_reference_path = collider_reference.path_from_id()
+
+        if collider_reference.collider_uuid:
+            unassign_op = layout_operator(
+                layout,
+                VRM_OT_unassign_spring_bone1_collider_group_collider,
+                text="<Unassign>",
+                translate=True,
+                icon="REMOVE",
+            )
+            unassign_op.armature_data_name = armature_data.name
+            unassign_op.collider_reference_path = collider_reference_path
+
+        for collider in spring_bone1.colliders:
+            if not collider.uuid:
+                continue
+            if collider.uuid in assigned_collider_uuids:
+                continue
+
+            assign_op = layout_operator(
+                layout,
+                VRM_OT_assign_spring_bone1_collider_group_collider,
+                text=collider.display_name,
+                translate=False,
+                icon="SPHERE",
+            )
+            assign_op.armature_data_name = armature_data.name
+            assign_op.collider_reference_path = collider_reference_path
+            assign_op.collider_uuid = collider.uuid
+
+
+class VRM_MT_spring_bone1_spring_collider_group(Menu):
+    bl_label = "Collider Group Assignment Menu"
+    bl_idname = "VRM_MT_spring_bone1_spring_collider_group"
+
+    CONTEXT_POINTER_SPRING: Final = bl_idname + "_spring"
+    CONTEXT_POINTER_COLLIDER_GROUP_REFERENCE: Final = (
+        bl_idname + "_collider_group_reference"
+    )
+
+    @classmethod
+    def draw_input_layout(
+        cls,
+        layout: UILayout,
+        spring: SpringBone1SpringPropertyGroup,
+        collider_group_reference: SpringBone1ColliderGroupReferencePropertyGroup,
+        *,
+        icon: str = "PIVOT_INDIVIDUAL",
+    ) -> UILayout:
+        row = layout.row(align=True)
+        row.context_pointer_set(
+            cls.CONTEXT_POINTER_SPRING,
+            spring,
+        )
+        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
+
+        spring = getattr(context, self.CONTEXT_POINTER_SPRING, None)
+        if not isinstance(
+            spring,
+            SpringBone1SpringPropertyGroup,
+        ):
+            return
+
+        collider_group_reference = getattr(
+            context, self.CONTEXT_POINTER_COLLIDER_GROUP_REFERENCE, None
+        )
+        if not isinstance(
+            collider_group_reference,
+            SpringBone1ColliderGroupReferencePropertyGroup,
+        ):
+            return
+
+        armature_data = collider_group_reference.id_data
+        if not isinstance(armature_data, Armature):
+            return
+
+        if spring.id_data != armature_data:
+            return
+
+        assigned_collider_group_uuids: set[str] = {
+            collider_group_reference.collider_group_uuid
+            for collider_group_reference in spring.collider_groups
+            if collider_group_reference.collider_group_uuid
+        }
+
+        spring_bone1 = get_armature_extension(armature_data).spring_bone1
+        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_spring_bone1_spring_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 index, collider_group in enumerate(spring_bone1.collider_groups):
+            if not collider_group.uuid:
+                continue
+            if collider_group.uuid in assigned_collider_group_uuids:
+                continue
+
+            assign_op = layout_operator(
+                layout,
+                VRM_OT_assign_spring_bone1_spring_collider_group,
+                text=f"{index + 1}: " + collider_group.vrm_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

package/browser/addons/vrm/editor/spring_bone1/migration.py

@@ -0,0 +1,86 @@
+# SPDX-License-Identifier: MIT OR GPL-3.0-or-later
+from bpy.types import Armature, Context, Object
+
+from ..extension_accessor import get_armature_extension
+from .ops import (
+    assign_spring_bone1_automatically,
+)
+from .property_group import SpringBone1SpringBonePropertyGroup
+
+
+def _migrate_blender_object(armature: Armature) -> None:
+    ext = get_armature_extension(armature)
+    if tuple(ext.addon_version) >= (2, 3, 27):
+        return
+
+    for collider in ext.spring_bone1.colliders:
+        bpy_object = collider.pop("blender_object", None)
+        if isinstance(bpy_object, Object):
+            collider.bpy_object = bpy_object
+
+
+def _fixup_gravity_dir(armature: Armature) -> None:
+    ext = get_armature_extension(armature)
+    if tuple(ext.addon_version) == ext.UNMANAGED_ADDON_VERSION:
+        return
+
+    if tuple(ext.addon_version) <= (2, 14, 3):
+        for spring in ext.spring_bone1.springs:
+            for joint in spring.joints:
+                joint.gravity_dir = [
+                    joint.gravity_dir[0],
+                    joint.gravity_dir[2],
+                    joint.gravity_dir[1],
+                ]
+
+    if tuple(ext.addon_version) <= (2, 14, 10):
+        for spring in ext.spring_bone1.springs:
+            for joint in spring.joints:
+                joint.gravity_dir = [
+                    joint.gravity_dir[0],
+                    -joint.gravity_dir[1],
+                    joint.gravity_dir[2],
+                ]
+
+    if tuple(ext.addon_version) <= (2, 15, 3):
+        for spring in ext.spring_bone1.springs:
+            for joint in spring.joints:
+                gravity_dir = list(joint.gravity_dir)
+                joint.gravity_dir = (gravity_dir[0] + 1, 0, 0)  # Make a change
+                joint.gravity_dir = gravity_dir
+
+
+def is_unnecessary(
+    spring_bone1: SpringBone1SpringBonePropertyGroup,
+    *,
+    heavy_migration: bool,
+) -> bool:
+    if not heavy_migration:
+        return True
+    return not spring_bone1.initial_automatic_spring_bone_assignment
+
+
+def migrate(
+    context: Context,
+    armature: Object,
+    *,
+    heavy_migration: bool,
+) -> None:
+    armature_data = armature.data
+    if not isinstance(armature_data, Armature):
+        return
+    _migrate_blender_object(armature_data)
+    _fixup_gravity_dir(armature_data)
+
+    ext = get_armature_extension(armature_data)
+    spring_bone1 = ext.spring_bone1
+
+    if (
+        tuple(ext.addon_version) < (3, 21, 2)
+        and tuple(ext.addon_version) != ext.UNMANAGED_ADDON_VERSION
+    ):
+        spring_bone1.initial_automatic_spring_bone_assignment = False
+
+    if heavy_migration and spring_bone1.initial_automatic_spring_bone_assignment:
+        spring_bone1.initial_automatic_spring_bone_assignment = False
+        assign_spring_bone1_automatically(context, armature.name)