Removal of IK bones and etc, zero weight bones and more

functions/tools/mmd_conversion.py

@@ -7,7 +7,8 @@ from bpy.types import Operator
 from ...core.common import get_active_armature
 from ...core.translations import t
 from ...core.mmd_converter import (convert_mmd_armature, detect_mmd_armature, 
-                                   translate_mmd_everything, restructure_mmd_to_unity_bones)
+                                   translate_mmd_everything, restructure_mmd_to_unity_bones,
+                                   remove_mmd_ik_bones, remove_mmd_twist_bones, remove_mmd_zero_weight_bones)
 from ...core.logging_setup import logger
 
 
@@ -48,8 +49,12 @@ class AvatarToolkit_OT_ConvertMMDArmature(Operator):
         translate_shapekeys = toolkit.mmd_translate_shapekeys
         translate_objects = toolkit.mmd_translate_objects
         restructure_bones = toolkit.mmd_restructure_bones
+        remove_twist_bones = toolkit.mmd_remove_twist_bones
+        remove_zero_weight_bones = toolkit.mmd_remove_zero_weight_bones
         
-        logger.info(f"Conversion settings - Make parent: {make_parent}, Rename: {rename_armature}, Restructure: {restructure_bones}")
+        logger.info(f"Conversion settings - Make parent: {make_parent}, Rename: {rename_armature}, " +
+                   f"Restructure: {restructure_bones}")
+        logger.info(f"Bone cleanup - IK: True (automatic), Twist: {remove_twist_bones}, Zero weight: {remove_zero_weight_bones}")
         logger.info(f"Translation settings - Enabled: {translate_names}, Bones: {translate_bones}, " +
                    f"Materials: {translate_materials}, Shapekeys: {translate_shapekeys}, Objects: {translate_objects}")
         
@@ -66,7 +71,36 @@ class AvatarToolkit_OT_ConvertMMDArmature(Operator):
         for msg in messages:
             self.report({'INFO'}, msg)
         
-        # Step 2: Translation (if enabled)
+        # Step 2: Remove IK bones BEFORE translation (always automatic)
+        logger.info("Starting IK bone removal (before translation)")
+        self.report({'INFO'}, "Removing IK bones...")
+        
+        ik_success, ik_messages = remove_mmd_ik_bones(armature)
+        
+        if ik_success:
+            logger.info("IK bone removal completed successfully")
+        else:
+            logger.warning("IK bone removal completed with errors")
+        
+        for msg in ik_messages:
+            self.report({'INFO'}, msg)
+        
+        # Step 3: Remove twist bones BEFORE translation (if enabled)
+        if remove_twist_bones:
+            logger.info("Starting twist bone removal (before translation)")
+            self.report({'INFO'}, "Removing twist bones...")
+            
+            twist_success, twist_messages = remove_mmd_twist_bones(armature)
+            
+            if twist_success:
+                logger.info("Twist bone removal completed successfully")
+            else:
+                logger.warning("Twist bone removal completed with errors")
+            
+            for msg in twist_messages:
+                self.report({'INFO'}, msg)
+        
+        # Step 4: Translation (if enabled)
         if translate_names:
             logger.info("Starting MMD name translation")
             self.report({'INFO'}, t("MMD.translation_starting"))
@@ -87,7 +121,7 @@ class AvatarToolkit_OT_ConvertMMDArmature(Operator):
             for msg in trans_messages:
                 self.report({'INFO'}, msg)
         
-        # Step 3: Restructure bones to Unity format (if enabled)
+        # Step 5: Restructure bones to Unity format (if enabled)
         if restructure_bones:
             logger.info("Starting bone restructuring to Unity format")
             self.report({'INFO'}, t("MMD.restructure_starting"))
@@ -102,4 +136,19 @@ class AvatarToolkit_OT_ConvertMMDArmature(Operator):
             for msg in struct_messages:
                 self.report({'INFO'}, msg)
         
+        # Step 6: Remove zero weight bones (if enabled)
+        if remove_zero_weight_bones:
+            logger.info("Starting zero weight bone removal")
+            self.report({'INFO'}, "Removing zero weight bones...")
+            
+            zero_success, zero_messages = remove_mmd_zero_weight_bones(armature)
+            
+            if zero_success:
+                logger.info("Zero weight bone removal completed successfully")
+            else:
+                logger.warning("Zero weight bone removal completed with errors")
+            
+            for msg in zero_messages:
+                self.report({'INFO'}, msg)
+        
         return {'FINISHED'}