Skip to content
Snippets Groups Projects
conversion.py 1.69 KiB
Newer Older
  • Learn to ignore specific revisions
  • # SPDX-License-Identifier: GPL-2.0-or-later
    
    
    """
    Pose Library - Conversion of old pose libraries.
    """
    
    from typing import Optional
    from collections.abc import Collection
    
    if "pose_creation" not in locals():
        from . import pose_creation
    else:
        import importlib
    
        pose_creation = importlib.reload(pose_creation)
    
    import bpy
    from bpy.types import (
        Action,
        TimelineMarker,
    )
    
    
    def convert_old_poselib(old_poselib: Action) -> Collection[Action]:
        """Convert an old-style pose library to a set of pose Actions.
    
        Old pose libraries were one Action with multiple pose markers. Each pose
        marker will be converted to an Action by itself and marked as asset.
        """
    
        pose_assets = [
            action
            for marker in old_poselib.pose_markers
            if (action := convert_old_pose(old_poselib, marker))
        ]
    
        # Mark all Actions as assets in one go. Ideally this would be done on an
        # appropriate frame in the scene (to set up things like the background
        # colour), but the old-style poselib doesn't contain such information. All
        # we can do is just render on the current frame.
        bpy.ops.asset.mark({'selected_ids': pose_assets})
    
        return pose_assets
    
    
    def convert_old_pose(old_poselib: Action, marker: TimelineMarker) -> Optional[Action]:
        """Convert an old-style pose library pose to a pose action."""
    
        frame: int = marker.frame
        action: Optional[Action] = None
    
        for fcurve in old_poselib.fcurves:
            key = pose_creation.find_keyframe(fcurve, frame)
            if not key:
                continue
    
            if action is None:
                action = bpy.data.actions.new(marker.name)
    
            pose_creation.create_single_key_fcurve(action, fcurve, key)
    
        return action