2025-12-01

2026-03-17 14:58:51 -06:00
parent 183e865f8b
commit 4b82b57113
6846 changed files with 954887 additions and 162606 deletions
@@ -0,0 +1,103 @@
+# SPDX-FileCopyrightText: 2025 Blender Studio Tools Authors
+#
+# SPDX-License-Identifier: GPL-3.0-or-later
+
+import bpy
+from bpy.types import Operator, Menu
+
+from bpy.props import EnumProperty, StringProperty, BoolProperty
+from ..fcurve_utils import get_fcurves
+
+
+class POSE_OT_curves_set_boolean(Operator):
+    """Set lock state of all selected curves"""
+    bl_idname = "pose.curves_set_boolean"
+    bl_label = "Set a boolean on curves"
+    bl_options = {'REGISTER', 'UNDO', 'INTERNAL'}
+
+    curve_set: EnumProperty(
+        name="Affected Curves",
+        items=[
+            ('ACTIVE', 'Active Curve', 'Active Curve'),
+            ('SELECTED', 'Selected Curves', 'Selected Curves'),
+            ('UNSELECTED', 'Unselected Curves', 'Unselected Curves'),
+            ('ALL', 'All Curves', 'All Curves'),
+        ],
+        default='SELECTED'
+    )
+    prop_name: StringProperty(
+        name="Property Name",
+        description="Name of the boolean property to set",
+        default='lock'
+    )
+    prop_value: BoolProperty(
+        name="Value",
+        description="Value to set",
+        default=False
+    )
+
+    @classmethod
+    def poll(cls, context):
+        # Only works in Graph Editor, when there is an active curve.
+        return context.pose_object and context.pose_object.animation_data and context.pose_object.animation_data.action
+
+    def execute(self, context):
+        action = context.object.animation_data.action
+        affected_fcurves = get_fcurves(context, action, self.curve_set)
+
+        for fc in affected_fcurves:
+            setattr(fc, self.prop_name, self.prop_value)
+
+        return {'FINISHED'}
+
+
+class GRAPH_MT_channel_lock(Menu):
+    bl_label = "Lock"
+
+    def draw(self, context):
+        layout = self.layout
+
+        op = layout.operator(POSE_OT_curves_set_boolean.bl_idname,
+                             text="Lock Selected", icon='LOCKED')
+        op.prop_name = "lock"
+        op.prop_value = True
+        op.curve_set = 'SELECTED'
+
+        op = layout.operator(POSE_OT_curves_set_boolean.bl_idname,
+                             text="Lock Unselected", icon='LOCKED')
+        op.prop_name = "lock"
+        op.prop_value = True
+        op.curve_set = 'UNSELECTED'
+
+        layout.separator()
+
+        op = layout.operator(POSE_OT_curves_set_boolean.bl_idname,
+                             text="Unlock Selected", icon='UNLOCKED')
+        op.prop_name = "lock"
+        op.prop_value = False
+        op.curve_set = 'SELECTED'
+
+        op = layout.operator(POSE_OT_curves_set_boolean.bl_idname,
+                             text="Unlock All", icon='UNLOCKED')
+        op.prop_name = "lock"
+        op.prop_value = False
+        op.curve_set = 'ALL'
+
+
+def draw_curves_lock_menu(self, context):
+    layout = self.layout
+    layout.menu("GRAPH_MT_channel_lock", icon='LOCKED')
+
+
+registry = [
+    POSE_OT_curves_set_boolean,
+    GRAPH_MT_channel_lock
+]
+
+
+def register():
+    bpy.types.GRAPH_MT_channel.append(draw_curves_lock_menu)
+
+
+def unregister():
+    bpy.types.GRAPH_MT_channel.remove(draw_curves_lock_menu)