diff --git a/functions/additional_tools.py b/functions/additional_tools.py index 9745945..241cac1 100644 --- a/functions/additional_tools.py +++ b/functions/additional_tools.py @@ -89,3 +89,33 @@ class AvatarToolKit_OT_ConnectBones(Operator): def draw(self, context): layout = self.layout layout.prop(self, "min_distance") + +@register_wrap +class AvatarToolKit_OT_DeleteBoneConstraints(Operator): + bl_idname = "avatar_toolkit.delete_bone_constraints" + bl_label = t("Tools.delete_bone_constraints.label") + bl_description = t("Tools.delete_bone_constraints.desc") + bl_options = {'REGISTER', 'UNDO'} + + @classmethod + def poll(cls, context: Context) -> bool: + return get_selected_armature(context) is not None + + def execute(self, context: Context) -> set[str]: + armature = get_selected_armature(context) + if not is_valid_armature(armature): + self.report({'ERROR'}, t("Tools.delete_bone_constraints.invalid_armature")) + return {'CANCELLED'} + + bpy.ops.object.mode_set(mode='POSE') + + constraints_removed = 0 + for bone in armature.pose.bones: + while bone.constraints: + bone.constraints.remove(bone.constraints[0]) + constraints_removed += 1 + + bpy.ops.object.mode_set(mode='OBJECT') + + self.report({'INFO'}, t("Tools.delete_bone_constraints.success").format(constraints_removed=constraints_removed)) + return {'FINISHED'} diff --git a/resources/translations/en_US.json b/resources/translations/en_US.json index 09cabc8..1460dcc 100644 --- a/resources/translations/en_US.json +++ b/resources/translations/en_US.json @@ -174,6 +174,10 @@ "Tools.connect_bones.min_distance.label": "Minimum Distance", "Tools.connect_bones.min_distance.desc": "Minimum distance between bones to connect them", "Tools.connect_bones.success": "Connected {bones_connected} bones successfully", + "Tools.delete_bone_constraints.label": "Delete Bone Constraints", + "Tools.delete_bone_constraints.desc": "Remove all constraints from bones in the armature", + "Tools.delete_bone_constraints.invalid_armature": "Invalid armature selected", + "Tools.delete_bone_constraints.success": "Removed {constraints_removed} constraints from bones", "Tools.convert_rigify_to_unity.label": "Convert Rigify to Unity", "Tools.convert_rigify_to_unity.desc": "Prepare Rigify armature for use in Unity", "Tools.convert_rigify_to_unity.success": "Rigify armature successfully converted for Unity", diff --git a/ui/tools.py b/ui/tools.py index d274348..e2c0eff 100644 --- a/ui/tools.py +++ b/ui/tools.py @@ -8,7 +8,7 @@ from ..functions.translations import t from ..core.common import get_selected_armature from ..functions.mesh_tools import AvatarToolkit_OT_RemoveUnusedShapekeys from ..functions.seperate_by import AvatarToolKit_OT_SeparateByMaterials, AvatarToolKit_OT_SeparateByLooseParts -from ..functions.additional_tools import AvatarToolKit_OT_ApplyTransforms, AvatarToolKit_OT_ConnectBones +from ..functions.additional_tools import AvatarToolKit_OT_ApplyTransforms, AvatarToolKit_OT_ConnectBones, AvatarToolKit_OT_DeleteBoneConstraints from ..functions.armature_modifying import AvatarToolkit_OT_RemoveZeroWeightBones, AvatarToolkit_OT_MergeBonesToActive, AvatarToolkit_OT_MergeBonesToParents from ..functions.rigify_functions import AvatarToolKit_OT_ConvertRigifyToUnity @@ -50,6 +50,9 @@ class AvatarToolkit_PT_ToolsPanel(bpy.types.Panel): row.operator(AvatarToolkit_OT_MergeBonesToParents.bl_idname, text=t("Tools.merge_bones_to_parents.label"), icon='BONE_DATA') row = layout.row(align=True) row.operator(AvatarToolKit_OT_ConnectBones.bl_idname, text=t("Tools.connect_bones.label"), icon='BONE_DATA') + row = layout.row(align=True) + row.operator(AvatarToolKit_OT_DeleteBoneConstraints.bl_idname, text=t("Tools.delete_bone_constraints.label"), icon='CONSTRAINT_BONE') + row = layout.row(align=True) row.operator(AvatarToolKit_OT_ConvertRigifyToUnity.bl_idname, text=t("Tools.convert_rigify_to_unity.label"), icon='ARMATURE_DATA') row = layout.row() row.prop(context.scene, "merge_twist_bones")