@volter/blender-engine 0.1.148 → 0.1.150

package/browser/addons/vrm/common/ops/wm.py

@@ -0,0 +1,167 @@
+# This code is auto generated.
+# To regenerate, run the `uv run tools/property_typing.py` command.
+
+from collections.abc import Mapping, Sequence
+from typing import Optional, Union
+
+import bpy
+
+
+# This code is auto generated.
+# To regenerate, run the `uv run tools/property_typing.py` command.
+def vrm_gltf2_addon_disabled_warning(
+    execution_context: str = "EXEC_DEFAULT",
+) -> set[str]:
+    return bpy.ops.wm.vrm_gltf2_addon_disabled_warning(  # type: ignore[attr-defined, no-any-return]
+        execution_context,
+    )
+
+
+# This code is auto generated.
+# To regenerate, run the `uv run tools/property_typing.py` command.
+def vrm_export_human_bones_assignment(
+    execution_context: str = "EXEC_DEFAULT",
+    /,
+    *,
+    armature_object_name: str = "",
+) -> set[str]:
+    return bpy.ops.wm.vrm_export_human_bones_assignment(  # type: ignore[attr-defined, no-any-return]
+        execution_context,
+        armature_object_name=armature_object_name,
+    )
+
+
+# This code is auto generated.
+# To regenerate, run the `uv run tools/property_typing.py` command.
+def vrm_export_confirmation(
+    execution_context: str = "EXEC_DEFAULT",
+    /,
+    *,
+    errors: Optional[Sequence[Mapping[str, Union[str, int, float, bool]]]] = None,
+    armature_object_name: str = "",
+    export_anyway: bool = False,
+) -> set[str]:
+    return bpy.ops.wm.vrm_export_confirmation(  # type: ignore[attr-defined, no-any-return]
+        execution_context,
+        errors=errors if errors is not None else [],
+        armature_object_name=armature_object_name,
+        export_anyway=export_anyway,
+    )
+
+
+# This code is auto generated.
+# To regenerate, run the `uv run tools/property_typing.py` command.
+def vrm_export_armature_selection(
+    execution_context: str = "EXEC_DEFAULT",
+    /,
+    *,
+    armature_object_name: str = "",
+    armature_object_name_candidates: Optional[
+        Sequence[Mapping[str, Union[str, int, float, bool]]]
+    ] = None,
+) -> set[str]:
+    return bpy.ops.wm.vrm_export_armature_selection(  # type: ignore[attr-defined, no-any-return]
+        execution_context,
+        armature_object_name=armature_object_name,
+        armature_object_name_candidates=armature_object_name_candidates
+        if armature_object_name_candidates is not None
+        else [],
+    )
+
+
+# This code is auto generated.
+# To regenerate, run the `uv run tools/property_typing.py` command.
+def vrma_export_prerequisite(
+    execution_context: str = "EXEC_DEFAULT",
+    /,
+    *,
+    armature_object_name: str = "",
+    armature_object_name_candidates: Optional[
+        Sequence[Mapping[str, Union[str, int, float, bool]]]
+    ] = None,
+) -> set[str]:
+    return bpy.ops.wm.vrma_export_prerequisite(  # type: ignore[attr-defined, no-any-return]
+        execution_context,
+        armature_object_name=armature_object_name,
+        armature_object_name_candidates=armature_object_name_candidates
+        if armature_object_name_candidates is not None
+        else [],
+    )
+
+
+# This code is auto generated.
+# To regenerate, run the `uv run tools/property_typing.py` command.
+def vrm_error_dialog(
+    execution_context: str = "EXEC_DEFAULT",
+    /,
+    *,
+    title: str = "",
+    lines: Optional[Sequence[Mapping[str, Union[str, int, float, bool]]]] = None,
+    active_line_index: int = 0,
+) -> set[str]:
+    return bpy.ops.wm.vrm_error_dialog(  # type: ignore[attr-defined, no-any-return]
+        execution_context,
+        title=title,
+        lines=lines if lines is not None else [],
+        active_line_index=active_line_index,
+    )
+
+
+# This code is auto generated.
+# To regenerate, run the `uv run tools/property_typing.py` command.
+def vrm_license_warning(
+    execution_context: str = "EXEC_DEFAULT",
+    /,
+    *,
+    filepath: str = "",
+    license_confirmations: Optional[
+        Sequence[Mapping[str, Union[str, int, float, bool]]]
+    ] = None,
+    import_anyway: bool = False,
+    thumbnail_image_name: str = "",
+    extract_textures_into_folder: bool = False,
+    make_new_texture_folder: bool = False,
+    set_shading_type_to_material_on_import: bool = False,
+    set_view_transform_to_standard_on_import: bool = False,
+    set_armature_display_to_wire: bool = False,
+    set_armature_display_to_show_in_front: bool = False,
+    set_armature_bone_shape_to_default: bool = False,
+    enable_mtoon_outline_preview: bool = False,
+) -> set[str]:
+    return bpy.ops.wm.vrm_license_warning(  # type: ignore[attr-defined, no-any-return]
+        execution_context,
+        filepath=filepath,
+        license_confirmations=license_confirmations
+        if license_confirmations is not None
+        else [],
+        import_anyway=import_anyway,
+        thumbnail_image_name=thumbnail_image_name,
+        extract_textures_into_folder=extract_textures_into_folder,
+        make_new_texture_folder=make_new_texture_folder,
+        set_shading_type_to_material_on_import=set_shading_type_to_material_on_import,
+        set_view_transform_to_standard_on_import=set_view_transform_to_standard_on_import,
+        set_armature_display_to_wire=set_armature_display_to_wire,
+        set_armature_display_to_show_in_front=set_armature_display_to_show_in_front,
+        set_armature_bone_shape_to_default=set_armature_bone_shape_to_default,
+        enable_mtoon_outline_preview=enable_mtoon_outline_preview,
+    )
+
+
+# This code is auto generated.
+# To regenerate, run the `uv run tools/property_typing.py` command.
+def vrma_import_prerequisite(
+    execution_context: str = "EXEC_DEFAULT",
+    /,
+    *,
+    armature_object_name: str = "",
+    armature_object_name_candidates: Optional[
+        Sequence[Mapping[str, Union[str, int, float, bool]]]
+    ] = None,
+) -> set[str]:
+    return bpy.ops.wm.vrma_import_prerequisite(  # type: ignore[attr-defined, no-any-return]
+        execution_context,
+        armature_object_name=armature_object_name,
+        armature_object_name_candidates=armature_object_name_candidates
+        if armature_object_name_candidates is not None
+        else [],
+    )

package/browser/addons/vrm/common/preferences.py

@@ -0,0 +1,362 @@
+# SPDX-License-Identifier: MIT OR GPL-3.0-or-later
+from collections.abc import Sequence
+from typing import TYPE_CHECKING, Final, Protocol, TypedDict, Union
+
+from bpy.app.translations import pgettext
+from bpy.props import BoolProperty, IntVectorProperty
+from bpy.types import AddonPreferences, Context, Operator, UILayout
+
+from . import version
+from .logger import get_logger
+from .shader import (
+    add_mtoon1_auto_setup_shader_node_group,
+    remove_mtoon1_auto_setup_shader_node_group,
+)
+
+_logger = get_logger(__name__)
+
+ADDON_PACKAGE_NAME: Final[str] = ".".join(__name__.split(".")[:-2])
+
+
+class ImportPreferencesProtocol(Protocol):
+    extract_textures_into_folder: bool
+    make_new_texture_folder: bool
+    set_shading_type_to_material_on_import: bool
+    set_view_transform_to_standard_on_import: bool
+    set_armature_display_to_wire: bool
+    set_armature_display_to_show_in_front: bool
+    set_armature_bone_shape_to_default: bool
+    enable_mtoon_outline_preview: bool
+
+
+class ImportPreferencesDict(TypedDict):
+    extract_textures_into_folder: bool
+    make_new_texture_folder: bool
+    set_shading_type_to_material_on_import: bool
+    set_view_transform_to_standard_on_import: bool
+    set_armature_display_to_wire: bool
+    set_armature_display_to_show_in_front: bool
+    set_armature_bone_shape_to_default: bool
+    enable_mtoon_outline_preview: bool
+
+
+def create_import_preferences_dict(
+    source: ImportPreferencesProtocol,
+) -> ImportPreferencesDict:
+    return {
+        "extract_textures_into_folder": source.extract_textures_into_folder,
+        "make_new_texture_folder": source.make_new_texture_folder,
+        "set_shading_type_to_material_on_import": (
+            source.set_shading_type_to_material_on_import
+        ),
+        "set_view_transform_to_standard_on_import": (
+            source.set_view_transform_to_standard_on_import
+        ),
+        "set_armature_display_to_wire": (source.set_armature_display_to_wire),
+        "set_armature_display_to_show_in_front": (
+            source.set_armature_display_to_show_in_front
+        ),
+        "set_armature_bone_shape_to_default": source.set_armature_bone_shape_to_default,
+        "enable_mtoon_outline_preview": source.enable_mtoon_outline_preview,
+    }
+
+
+def copy_import_preferences(
+    *, source: ImportPreferencesProtocol, destination: ImportPreferencesProtocol
+) -> None:
+    (
+        destination.extract_textures_into_folder,
+        destination.make_new_texture_folder,
+        destination.set_shading_type_to_material_on_import,
+        destination.set_view_transform_to_standard_on_import,
+        destination.set_armature_display_to_wire,
+        destination.set_armature_display_to_show_in_front,
+        destination.set_armature_bone_shape_to_default,
+        destination.enable_mtoon_outline_preview,
+    ) = (
+        source.extract_textures_into_folder,
+        source.make_new_texture_folder,
+        source.set_shading_type_to_material_on_import,
+        source.set_view_transform_to_standard_on_import,
+        source.set_armature_display_to_wire,
+        source.set_armature_display_to_show_in_front,
+        source.set_armature_bone_shape_to_default,
+        source.enable_mtoon_outline_preview,
+    )
+
+
+def draw_import_preferences_layout(
+    preferences: ImportPreferencesProtocol, layout: UILayout
+) -> None:
+    if not isinstance(preferences, (AddonPreferences, Operator)):
+        return
+
+    layout.prop(preferences, "extract_textures_into_folder")
+    if preferences.extract_textures_into_folder:
+        layout.prop(preferences, "make_new_texture_folder")
+    layout.prop(preferences, "set_shading_type_to_material_on_import")
+    layout.prop(preferences, "set_view_transform_to_standard_on_import")
+    layout.prop(preferences, "set_armature_display_to_wire")
+    layout.prop(preferences, "set_armature_display_to_show_in_front")
+    layout.prop(preferences, "set_armature_bone_shape_to_default")
+    layout.prop(preferences, "enable_mtoon_outline_preview")
+
+
+class ExportPreferencesProtocol(Protocol):
+    export_invisibles: bool
+    export_only_selections: bool
+    enable_advanced_preferences: bool
+    export_all_influences: bool
+    export_lights: bool
+    export_gltf_animations: bool
+    export_try_sparse_sk: bool
+
+
+def copy_export_preferences(
+    *, source: ExportPreferencesProtocol, destination: ExportPreferencesProtocol
+) -> None:
+    (
+        destination.export_invisibles,
+        destination.export_only_selections,
+        destination.enable_advanced_preferences,
+        destination.export_all_influences,
+        destination.export_lights,
+        destination.export_gltf_animations,
+        destination.export_try_sparse_sk,
+    ) = (
+        source.export_invisibles,
+        source.export_only_selections,
+        source.enable_advanced_preferences,
+        source.export_all_influences,
+        source.export_lights,
+        source.export_gltf_animations,
+        source.export_try_sparse_sk,
+    )
+
+
+def _draw_advanced_options_description(
+    preferences: Union[AddonPreferences, Operator],
+    property_name: str,
+    layout: UILayout,
+    description: str,
+) -> None:
+    column = layout.box().column(align=True)
+    column.prop(preferences, property_name)
+    description_column = column.box().column(align=True)
+    for i, line in enumerate(description.splitlines()):
+        icon = "ERROR" if i == 0 else "NONE"
+        description_column.label(text=line, translate=False, icon=icon)
+
+
+def draw_export_preferences_layout(
+    preferences: ExportPreferencesProtocol,
+    layout: UILayout,
+    *,
+    show_vrm1_options: bool,
+) -> None:
+    if not isinstance(preferences, (AddonPreferences, Operator)):
+        return
+
+    layout.prop(preferences, "export_invisibles")
+    layout.prop(preferences, "export_only_selections")
+
+    if not show_vrm1_options:
+        return
+
+    layout.prop(preferences, "enable_advanced_preferences")
+    if not preferences.enable_advanced_preferences:
+        return
+
+    advanced_options_column = layout.box().column()
+
+    # UniVRM 0.115.0 doesn't support `export_try_sparse_sk`
+    # https://github.com/saturday06/VRM-Addon-for-Blender/issues/381#issuecomment-1838365762
+    _draw_advanced_options_description(
+        preferences,
+        "export_try_sparse_sk",
+        advanced_options_column,
+        pgettext(
+            "The file size will be reduced,\n"
+            + "but it will no longer be readable by\n"
+            + "older apps with UniVRM 0.115.0 or\n"
+            + "earlier."
+        ),
+    )
+
+    # The upstream says that Models may appear incorrectly in many viewers.
+    # https://github.com/KhronosGroup/glTF-Blender-IO/blob/356b3dda976303d3ecce8b3bd1591245e576db38/addons/io_scene_gltf2/__init__.py#L760
+    _draw_advanced_options_description(
+        preferences,
+        "export_all_influences",
+        advanced_options_column,
+        pgettext(
+            "By default, 4 bone influences\n"
+            + "are exported for each vertex. Many\n"
+            + "apps truncate to 4. Increasing it\n"
+            + "may cause jagged meshes."
+        ),
+    )
+
+    _draw_advanced_options_description(
+        preferences,
+        "export_lights",
+        advanced_options_column,
+        pgettext(
+            "There is no consensus on how\n"
+            + "to handle lights in VRM, so it is\n"
+            + "impossible to predict what the\n"
+            + "outcome will be."
+        ),
+    )
+
+    _draw_advanced_options_description(
+        preferences,
+        "export_gltf_animations",
+        advanced_options_column,
+        pgettext(
+            "UniVRM does not export\n"
+            + "glTF Animations, so it is disabled\n"
+            + "by default. Please consider using\n"
+            + "VRM Animation."
+        ),
+    )
+
+
+class VrmAddonPreferences(AddonPreferences):
+    bl_idname = ADDON_PACKAGE_NAME
+
+    UNMANAGED_ADDON_VERSION: tuple[int, int, int] = (0, 0, 0)
+
+    addon_version: IntVectorProperty(  # type: ignore[valid-type]
+        size=3,
+        default=UNMANAGED_ADDON_VERSION,
+    )
+
+    def _update_add_mtoon_shader_node_group(self, context: Context) -> None:
+        if self.add_mtoon_shader_node_group:
+            add_mtoon1_auto_setup_shader_node_group(context)
+        else:
+            remove_mtoon1_auto_setup_shader_node_group(context)
+
+    add_mtoon_shader_node_group: BoolProperty(  # type: ignore[valid-type]
+        name="Add MToon shader node group",
+        default=True,
+        update=_update_add_mtoon_shader_node_group,
+    )
+
+    extract_textures_into_folder: BoolProperty(  # type: ignore[valid-type]
+        name="Extract texture images into the folder",
+        default=False,
+    )
+    make_new_texture_folder: BoolProperty(  # type: ignore[valid-type]
+        name="Don't overwrite existing texture image folder",
+        default=True,
+    )
+    set_shading_type_to_material_on_import: BoolProperty(  # type: ignore[valid-type]
+        name='Set shading type to "Material"',
+        default=True,
+    )
+    set_view_transform_to_standard_on_import: BoolProperty(  # type: ignore[valid-type]
+        name='Set view transform to "Standard"',
+        default=True,
+    )
+    set_armature_display_to_wire: BoolProperty(  # type: ignore[valid-type]
+        name='Set an imported armature display to "Wire"',
+        default=True,
+    )
+    set_armature_display_to_show_in_front: BoolProperty(  # type: ignore[valid-type]
+        name='Set an imported armature display to show "In-Front"',
+        default=True,
+    )
+    set_armature_bone_shape_to_default: BoolProperty(  # type: ignore[valid-type]
+        name="Set an imported bone shape to default",
+        default=True,
+    )
+
+    enable_mtoon_outline_preview: BoolProperty(  # type: ignore[valid-type]
+        name="Enable MToon Outline Preview",
+        default=True,
+    )
+
+    export_invisibles: BoolProperty(  # type: ignore[valid-type]
+        name="Export Invisible Objects",
+    )
+    export_only_selections: BoolProperty(  # type: ignore[valid-type]
+        name="Export Only Selections",
+    )
+    enable_advanced_preferences: BoolProperty(  # type: ignore[valid-type]
+        name="Enable Advanced Options",
+    )
+    export_all_influences: BoolProperty(  # type: ignore[valid-type]
+        name="Export All Bone Influences",
+    )
+    export_lights: BoolProperty(  # type: ignore[valid-type]
+        name="Export Lights",
+    )
+    export_gltf_animations: BoolProperty(  # type: ignore[valid-type]
+        name="Export glTF Animations",
+    )
+    export_try_sparse_sk: BoolProperty(  # type: ignore[valid-type]
+        name="Use Sparse Accessors",
+    )
+
+    def draw(self, _context: Context) -> None:
+        layout = self.layout
+
+        warning_message = version.preferences_warning_message()
+        if warning_message:
+            box = layout.box()
+            warning_column = box.column()
+            for index, warning_line in enumerate(warning_message.splitlines()):
+                warning_column.label(
+                    text=warning_line,
+                    translate=False,
+                    icon="NONE" if index else "ERROR",
+                )
+
+        layout.prop(self, "add_mtoon_shader_node_group")
+
+        import_box = layout.box()
+        import_box.label(text="Import", icon="IMPORT")
+        draw_import_preferences_layout(self, import_box)
+
+        export_box = layout.box()
+        export_box.label(text="Export", icon="EXPORT")
+        draw_export_preferences_layout(self, export_box, show_vrm1_options=True)
+
+    if TYPE_CHECKING:
+        # This code is auto generated.
+        # To regenerate, run the `uv run tools/property_typing.py` command.
+        addon_version: Sequence[int]  # type: ignore[no-redef]
+        add_mtoon_shader_node_group: bool  # type: ignore[no-redef]
+        extract_textures_into_folder: bool  # type: ignore[no-redef]
+        make_new_texture_folder: bool  # type: ignore[no-redef]
+        set_shading_type_to_material_on_import: bool  # type: ignore[no-redef]
+        set_view_transform_to_standard_on_import: bool  # type: ignore[no-redef]
+        set_armature_display_to_wire: bool  # type: ignore[no-redef]
+        set_armature_display_to_show_in_front: bool  # type: ignore[no-redef]
+        set_armature_bone_shape_to_default: bool  # type: ignore[no-redef]
+        enable_mtoon_outline_preview: bool  # type: ignore[no-redef]
+        export_invisibles: bool  # type: ignore[no-redef]
+        export_only_selections: bool  # type: ignore[no-redef]
+        enable_advanced_preferences: bool  # type: ignore[no-redef]
+        export_all_influences: bool  # type: ignore[no-redef]
+        export_lights: bool  # type: ignore[no-redef]
+        export_gltf_animations: bool  # type: ignore[no-redef]
+        export_try_sparse_sk: bool  # type: ignore[no-redef]
+
+
+def get_preferences(context: Context) -> VrmAddonPreferences:
+    addon = context.preferences.addons.get(ADDON_PACKAGE_NAME)
+    if not addon:
+        message = f"No add-on preferences for {ADDON_PACKAGE_NAME}"
+        raise AssertionError(message)
+
+    preferences = addon.preferences
+    if not isinstance(preferences, VrmAddonPreferences):
+        raise TypeError(
+            f"Add-on preferences for {ADDON_PACKAGE_NAME} is not a VrmAddonPreferences"
+            + f" but {type(preferences)}"
+        )
+
+    return preferences

package/browser/addons/vrm/common/progress.py

@@ -0,0 +1,87 @@
+# SPDX-License-Identifier: MIT OR GPL-3.0-or-later
+import math
+from collections.abc import Generator
+from contextlib import contextmanager
+from typing import Final, Optional
+from uuid import uuid4
+
+import bpy
+from bpy.types import Context
+
+from .logger import get_logger
+
+_logger = get_logger(__name__)
+
+
+class PartialProgress:
+    def __init__(
+        self, progress: "Progress", partial_start_ratio: float, partial_end_ratio: float
+    ) -> None:
+        self.progress: Final = progress
+        self.partial_start_ratio: Final = partial_start_ratio
+        self.partial_end_ratio: Final = partial_end_ratio
+
+    def update(self, ratio: float) -> None:
+        ratio = min(max(0.0, ratio), 1.0)
+        self.progress.update(
+            self.partial_start_ratio
+            + ratio * (self.partial_end_ratio - self.partial_start_ratio)
+        )
+
+
+class Progress:
+    active_progress_uuid: Optional[str] = None
+
+    def __init__(self, context: Context, *, show_progress: bool) -> None:
+        self.context: Final = context
+        self.show_progress: Final = show_progress
+        self.uuid: Final = uuid4().hex
+        self.last_ratio = 0.0
+
+    def partial_progress(self, partial_end_ratio: float) -> PartialProgress:
+        partial_end_ratio = min(max(0.0, partial_end_ratio), 1.0)
+        return PartialProgress(self, self.last_ratio, partial_end_ratio)
+
+    def update(self, ratio: float) -> None:
+        ratio = min(max(0.0, ratio), 1.0)
+        self.last_ratio = ratio
+        if self.active_progress_uuid != self.uuid:
+            _logger.error(
+                "Progress.update() called from different progress active=%s self=%s",
+                self.active_progress_uuid,
+                self.uuid,
+            )
+            return
+
+        if not self.show_progress:
+            return
+
+        if bpy.app.version >= (5, 0):
+            progress_value = ratio * 9999
+        else:
+            # The mouse cursor becomes a four-digit number, and there is an area
+            # where values from 0 to 9999 can be displayed. However, if the
+            # progress rate is directly converted to a number from 0 to 9999, the
+            # last two digits will frequently round-trip, making the progress
+            # difficult to understand. Therefore, only the last two digits display
+            # area is used to show the progress rate as a number from 0 to 99.
+            progress_value = math.floor(ratio * 99)
+
+        self.context.window_manager.progress_update(progress_value)
+
+
+@contextmanager
+def create_progress(
+    context: Context, *, show_progress: bool = True
+) -> Generator[Progress]:
+    saved_progress_uuid = Progress.active_progress_uuid
+    try:
+        if show_progress and Progress.active_progress_uuid is None:
+            context.window_manager.progress_begin(0, 9999)
+        progress = Progress(context, show_progress=show_progress)
+        Progress.active_progress_uuid = progress.uuid
+        yield progress
+    finally:
+        Progress.active_progress_uuid = saved_progress_uuid
+        if show_progress and Progress.active_progress_uuid is None:
+            context.window_manager.progress_end()

package/browser/addons/vrm/common/rotation.py

@@ -0,0 +1,89 @@
+# SPDX-License-Identifier: MIT OR GPL-3.0-or-later
+from typing import Final, Union
+
+from bpy.types import (
+    Object,
+    PoseBone,
+)
+from mathutils import Quaternion
+
+from .logger import get_logger
+
+_logger = get_logger(__name__)
+
+
+ROTATION_MODE_QUATERNION: Final = "QUATERNION"
+ROTATION_MODE_AXIS_ANGLE: Final = "AXIS_ANGLE"
+ROTATION_MODE_EULER: Final = ("XYZ", "XZY", "YXZ", "YZX", "ZXY", "ZYX")
+
+
+def get_rotation_as_quaternion(
+    object_or_pose_bone: Union[Object, PoseBone],
+) -> Quaternion:
+    if object_or_pose_bone.rotation_mode == ROTATION_MODE_QUATERNION:
+        return object_or_pose_bone.rotation_quaternion.copy()
+
+    if object_or_pose_bone.rotation_mode == ROTATION_MODE_AXIS_ANGLE:
+        # https://projects.blender.org/blender/blender/src/tag/v2.93.18/source/blender/makesrna/intern/rna_pose.c#L270-L278
+        angle, axis_x, axis_y, axis_z = object_or_pose_bone.rotation_axis_angle
+        return Quaternion((axis_x, axis_y, axis_z), angle)
+
+    if object_or_pose_bone.rotation_mode in ROTATION_MODE_EULER:
+        return object_or_pose_bone.rotation_euler.to_quaternion()
+
+    _logger.error(
+        "Unexpected rotation mode for %s %s: %s",
+        type(object_or_pose_bone),
+        object_or_pose_bone.name,
+        object_or_pose_bone.rotation_mode,
+    )
+
+    return Quaternion()
+
+
+def set_rotation_without_mode_change(
+    object_or_pose_bone: Union[Object, PoseBone], quaternion: Quaternion
+) -> None:
+    if object_or_pose_bone.rotation_mode == ROTATION_MODE_QUATERNION:
+        object_or_pose_bone.rotation_quaternion = quaternion.copy()
+        return
+
+    if object_or_pose_bone.rotation_mode == ROTATION_MODE_AXIS_ANGLE:
+        axis, angle = quaternion.to_axis_angle()
+        # https://projects.blender.org/blender/blender/src/tag/v2.93.18/source/blender/makesrna/intern/rna_pose.c#L280-L290
+        object_or_pose_bone.rotation_axis_angle = [angle, axis.x, axis.y, axis.z]
+        return
+
+    if object_or_pose_bone.rotation_mode in ROTATION_MODE_EULER:
+        object_or_pose_bone.rotation_euler = quaternion.to_euler(
+            object_or_pose_bone.rotation_mode
+        )
+        return
+
+    _logger.error(
+        "Unexpected rotation mode for %s %s: %s",
+        type(object_or_pose_bone),
+        object_or_pose_bone.name,
+        object_or_pose_bone.rotation_mode,
+    )
+
+
+def insert_rotation_keyframe(
+    object_or_pose_bone: Union[Object, PoseBone], *, frame: int
+) -> None:
+    if object_or_pose_bone.rotation_mode == ROTATION_MODE_QUATERNION:
+        data_path = "rotation_quaternion"
+    elif object_or_pose_bone.rotation_mode == ROTATION_MODE_AXIS_ANGLE:
+        data_path = "rotation_axis_angle"
+    elif object_or_pose_bone.rotation_mode in ROTATION_MODE_EULER:
+        data_path = "rotation_euler"
+    else:
+        _logger.error(
+            "Unexpected rotation mode for %s %s: %s",
+            type(object_or_pose_bone),
+            object_or_pose_bone.name,
+            object_or_pose_bone.rotation_mode,
+        )
+        return
+
+    object_or_pose_bone.keyframe_insert(data_path, frame=frame)