2025-12-01

scripts/addons/anim_cupboard/fcurve_utils.py

@@ -0,0 +1,43 @@
+# SPDX-FileCopyrightText: 2025 Blender Studio Tools Authors
+#
+# SPDX-License-Identifier: GPL-3.0-or-later
+
+from typing import List
+from bpy.types import Action, PoseBone, FCurve
+
+
+def get_fcurves_of_bone(action: Action, bone_name: str) -> List[FCurve]:
+    good_curves = []
+
+    for fc in action.fcurves:
+        if 'pose.bones' not in fc.data_path:
+            continue
+        bone_of_curve = fc.data_path.split('pose.bones["')[1].split('"]')[0]
+        if bone_name == bone_of_curve:
+            good_curves.append(fc)
+
+    return good_curves
+
+
+def get_fcurves_of_bones(action: Action, pose_bones: List[PoseBone]) -> List[FCurve]:
+
+    fcurves = []
+    for pb in pose_bones:
+        fcurves.extend(get_fcurves_of_bone(action, pb.name))
+
+    return fcurves
+
+
+def get_fcurves(context, action: Action, set="ALL") -> List[FCurve]:
+    """Return a list of FCurves in the given action."""
+
+    fcurves = action.fcurves
+    if set == 'ALL':
+        return fcurves
+    elif set == 'ACTIVE':
+        return context.active_editable_fcurve
+    elif set == 'SELECTED':
+        # Differs from selected_editable_fcurves because locked curves can be selected but aren't editable.
+        return [fc for fc in fcurves if fc.select]
+    elif set == 'UNSELECTED':
+        return [fc for fc in fcurves if not fc.select]