More Tools

- Added Apply Transforms
- Added Separate by materials and loose parts.

functions/additional_tools.py

@@ -0,0 +1,37 @@
+import bpy
+from bpy.types import Context, Operator
+from ..core.register import register_wrap
+from ..core.common import get_selected_armature, is_valid_armature, get_all_meshes
+from ..functions.translations import t
+
+@register_wrap
+class ApplyTransforms(Operator):
+    bl_idname = "avatar_toolkit.apply_transforms"
+    bl_label = t("Tools.apply_transforms.label")
+    bl_description = t("Tools.apply_transforms.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.apply_transforms.invalid_armature"))
+            return {'CANCELLED'}
+
+        bpy.ops.object.mode_set(mode='OBJECT')
+        bpy.ops.object.select_all(action='DESELECT')
+
+        armature.select_set(True)
+        context.view_layer.objects.active = armature
+
+        meshes = get_all_meshes(context)
+        for mesh in meshes:
+            mesh.select_set(True)
+
+        bpy.ops.object.transform_apply(location=True, rotation=True, scale=True)
+
+        self.report({'INFO'}, t("Tools.apply_transforms.success"))
+        return {'FINISHED'}

functions/seperate_by.py

@@ -0,0 +1,45 @@
+import bpy
+from bpy.types import Context, Operator
+from ..core.register import register_wrap
+from ..core.common import get_selected_armature, is_valid_armature, select_current_armature
+from ..functions.translations import t
+
+@register_wrap
+class SeparateByMaterials(Operator):
+    bl_idname = "avatar_toolkit.separate_by_materials"
+    bl_label = t("Tools.separate_by_materials.label")
+    bl_description = t("Tools.separate_by_materials.desc")
+    bl_options = {'REGISTER', 'UNDO'}
+
+    @classmethod
+    def poll(cls, context: Context) -> bool:
+        return context.active_object and context.active_object.type == 'MESH'
+
+    def execute(self, context: Context) -> set[str]:
+        obj = context.active_object
+        bpy.ops.object.mode_set(mode='EDIT')
+        bpy.ops.mesh.select_all(action='SELECT')
+        bpy.ops.mesh.separate(type='MATERIAL')
+        bpy.ops.object.mode_set(mode='OBJECT')
+        self.report({'INFO'}, t("Tools.separate_by_materials.success"))
+        return {'FINISHED'}
+
+@register_wrap
+class SeparateByLooseParts(Operator):
+    bl_idname = "avatar_toolkit.separate_by_loose_parts"
+    bl_label = t("Tools.separate_by_loose_parts.label")
+    bl_description = t("Tools.separate_by_loose_parts.desc")
+    bl_options = {'REGISTER', 'UNDO'}
+
+    @classmethod
+    def poll(cls, context: Context) -> bool:
+        return context.active_object and context.active_object.type == 'MESH'
+
+    def execute(self, context: Context) -> set[str]:
+        obj = context.active_object
+        bpy.ops.object.mode_set(mode='EDIT')
+        bpy.ops.mesh.select_all(action='SELECT')
+        bpy.ops.mesh.separate(type='LOOSE')
+        bpy.ops.object.mode_set(mode='OBJECT')
+        self.report({'INFO'}, t("Tools.separate_by_loose_parts.success"))
+        return {'FINISHED'}