pose_library/conversion.py

   1 # SPDX-License-Identifier: GPL-2.0-or-later
   2
   3 """
   4 Pose Library - Conversion of old pose libraries.
   5 """
   6
   7 from typing import Optional
   8 from collections.abc import Collection
   9
  10 if "pose_creation" not in locals():
  11     from . import pose_creation
  12 else:
  13     import importlib
  14
  15     pose_creation = importlib.reload(pose_creation)
  16
  17 import bpy
  18 from bpy.types import (
  19     Action,
  20     TimelineMarker,
  21 )
  22
  23
  24 def convert_old_poselib(old_poselib: Action) -> Collection[Action]:
  25     """Convert an old-style pose library to a set of pose Actions.
  26
  27     Old pose libraries were one Action with multiple pose markers. Each pose
  28     marker will be converted to an Action by itself and marked as asset.
  29     """
  30
  31     pose_assets = [
  32         action
  33         for marker in old_poselib.pose_markers
  34         if (action := convert_old_pose(old_poselib, marker))
  35     ]
  36
  37     # Mark all Actions as assets in one go. Ideally this would be done on an
  38     # appropriate frame in the scene (to set up things like the background
  39     # colour), but the old-style poselib doesn't contain such information. All
  40     # we can do is just render on the current frame.
  41     bpy.ops.asset.mark({'selected_ids': pose_assets})
  42
  43     return pose_assets
  44
  45
  46 def convert_old_pose(old_poselib: Action, marker: TimelineMarker) -> Optional[Action]:
  47     """Convert an old-style pose library pose to a pose action."""
  48
  49     frame: int = marker.frame
  50     action: Optional[Action] = None
  51
  52     for fcurve in old_poselib.fcurves:
  53         key = pose_creation.find_keyframe(fcurve, frame)
  54         if not key:
  55             continue
  56
  57         if action is None:
  58             action = bpy.data.actions.new(marker.name)
  59
  60         pose_creation.create_single_key_fcurve(action, fcurve, key)
  61
  62     return action