Restart
This commit is contained in:
File diff suppressed because it is too large
Load Diff
@@ -1,154 +0,0 @@
|
|||||||
"""
|
|
||||||
MMD Conversion Operator
|
|
||||||
Converts MMD armatures to standard Blender format
|
|
||||||
"""
|
|
||||||
import bpy
|
|
||||||
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,
|
|
||||||
remove_mmd_ik_bones, remove_mmd_twist_bones, remove_mmd_zero_weight_bones)
|
|
||||||
from ...core.logging_setup import logger
|
|
||||||
|
|
||||||
|
|
||||||
class AvatarToolkit_OT_ConvertMMDArmature(Operator):
|
|
||||||
"""Convert MMD armature to standard Blender format"""
|
|
||||||
bl_idname = "avatar_toolkit.convert_mmd_armature"
|
|
||||||
bl_label = t("MMD.convert_armature.label")
|
|
||||||
bl_description = t("MMD.convert_armature.desc")
|
|
||||||
bl_options = {'REGISTER', 'UNDO'}
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def poll(cls, context):
|
|
||||||
armature = get_active_armature(context)
|
|
||||||
return armature is not None
|
|
||||||
|
|
||||||
def execute(self, context):
|
|
||||||
armature = get_active_armature(context)
|
|
||||||
if not armature:
|
|
||||||
logger.warning("No active armature found for MMD conversion")
|
|
||||||
self.report({'ERROR'}, t("MMD.no_armature_selected"))
|
|
||||||
return {'CANCELLED'}
|
|
||||||
|
|
||||||
logger.info(f"Starting MMD conversion for armature: {armature.name}")
|
|
||||||
|
|
||||||
# Check if it's an MMD armature
|
|
||||||
if not detect_mmd_armature(armature):
|
|
||||||
logger.warning(f"Armature '{armature.name}' does not appear to be an MMD armature")
|
|
||||||
self.report({'WARNING'}, t("MMD.not_mmd_armature"))
|
|
||||||
return {'CANCELLED'}
|
|
||||||
|
|
||||||
# Get conversion settings
|
|
||||||
toolkit = context.scene.avatar_toolkit
|
|
||||||
make_parent = toolkit.mmd_make_parent
|
|
||||||
rename_armature = toolkit.mmd_rename_armature
|
|
||||||
translate_names = toolkit.mmd_translate_names
|
|
||||||
translate_bones = toolkit.mmd_translate_bones
|
|
||||||
translate_materials = toolkit.mmd_translate_materials
|
|
||||||
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}, " +
|
|
||||||
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}")
|
|
||||||
|
|
||||||
# Step 1: Basic conversion (parent and rename)
|
|
||||||
success, messages = convert_mmd_armature(armature, make_parent, rename_armature)
|
|
||||||
|
|
||||||
if not success:
|
|
||||||
logger.warning(f"MMD conversion failed: {messages}")
|
|
||||||
for msg in messages:
|
|
||||||
self.report({'WARNING'}, msg)
|
|
||||||
return {'CANCELLED'}
|
|
||||||
|
|
||||||
logger.info(f"MMD basic conversion completed successfully")
|
|
||||||
for msg in messages:
|
|
||||||
self.report({'INFO'}, msg)
|
|
||||||
|
|
||||||
# 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"))
|
|
||||||
|
|
||||||
trans_success, trans_messages = translate_mmd_everything(
|
|
||||||
armature,
|
|
||||||
translate_bones=translate_bones,
|
|
||||||
translate_materials=translate_materials,
|
|
||||||
translate_shapekeys=translate_shapekeys,
|
|
||||||
translate_objects=translate_objects
|
|
||||||
)
|
|
||||||
|
|
||||||
if trans_success:
|
|
||||||
logger.info("MMD name translation completed successfully")
|
|
||||||
else:
|
|
||||||
logger.warning("MMD name translation completed with some failures")
|
|
||||||
|
|
||||||
for msg in trans_messages:
|
|
||||||
self.report({'INFO'}, msg)
|
|
||||||
|
|
||||||
# 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"))
|
|
||||||
|
|
||||||
struct_success, struct_messages = restructure_mmd_to_unity_bones(armature)
|
|
||||||
|
|
||||||
if struct_success:
|
|
||||||
logger.info("Bone restructuring completed successfully")
|
|
||||||
else:
|
|
||||||
logger.warning("Bone restructuring completed with errors")
|
|
||||||
|
|
||||||
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'}
|
|
||||||
-125
@@ -1,125 +0,0 @@
|
|||||||
"""
|
|
||||||
MMD Converter Panel - UI for MMD conversion tools
|
|
||||||
"""
|
|
||||||
import bpy
|
|
||||||
from bpy.types import Panel, Context, UILayout
|
|
||||||
from .main_panel import AvatarToolKit_PT_AvatarToolkitPanel, CATEGORY_NAME
|
|
||||||
from .panel_layout import get_panel_order, should_open_by_default
|
|
||||||
from ..core.translations import t
|
|
||||||
from ..core.common import get_active_armature
|
|
||||||
from ..core.mmd_converter import detect_mmd_armature
|
|
||||||
from ..functions.tools.mmd_conversion import AvatarToolkit_OT_ConvertMMDArmature
|
|
||||||
|
|
||||||
|
|
||||||
class AvatarToolKit_PT_MMDPanel(Panel):
|
|
||||||
"""Panel for MMD conversion tools"""
|
|
||||||
bl_label = t("MMD.panel.label")
|
|
||||||
bl_idname = "OBJECT_PT_avatar_toolkit_mmd"
|
|
||||||
bl_space_type = 'VIEW_3D'
|
|
||||||
bl_region_type = 'UI'
|
|
||||||
bl_category = CATEGORY_NAME
|
|
||||||
bl_parent_id = AvatarToolKit_PT_AvatarToolkitPanel.bl_idname
|
|
||||||
bl_order = get_panel_order('mmd')
|
|
||||||
bl_options = set() if not should_open_by_default('MMD') else {'DEFAULT_CLOSED'}
|
|
||||||
|
|
||||||
def draw(self, context: Context) -> None:
|
|
||||||
"""Draw the MMD conversion panel interface"""
|
|
||||||
layout: UILayout = self.layout
|
|
||||||
|
|
||||||
# MMD Conversion Tools
|
|
||||||
mmd_box: UILayout = layout.box()
|
|
||||||
col: UILayout = mmd_box.column(align=True)
|
|
||||||
col.label(text=t("MMD.converter.title"), icon='ARMATURE_DATA')
|
|
||||||
col.separator(factor=0.5)
|
|
||||||
|
|
||||||
# Check if we have an active armature
|
|
||||||
armature = get_active_armature(context)
|
|
||||||
|
|
||||||
if not armature:
|
|
||||||
col.label(text=t("MMD.no_armature_selected"), icon='ERROR')
|
|
||||||
col.label(text=t("MMD.select_armature_to_convert"))
|
|
||||||
return
|
|
||||||
|
|
||||||
# Check if the armature appears to be MMD
|
|
||||||
is_mmd = detect_mmd_armature(armature)
|
|
||||||
|
|
||||||
if is_mmd:
|
|
||||||
col.label(text=t("MMD.armature_name", name=armature.name), icon='CHECKMARK')
|
|
||||||
col.label(text=t("MMD.armature_detected"), icon='INFO')
|
|
||||||
col.separator(factor=0.3)
|
|
||||||
|
|
||||||
toolkit = context.scene.avatar_toolkit
|
|
||||||
|
|
||||||
# Basic conversion settings
|
|
||||||
col.prop(toolkit, 'mmd_make_parent', text=t("MMD.make_armature_parent"))
|
|
||||||
col.prop(toolkit, 'mmd_rename_armature', text=t("MMD.rename_to_armature"))
|
|
||||||
col.separator(factor=0.2)
|
|
||||||
|
|
||||||
# Bone restructuring
|
|
||||||
col.prop(toolkit, 'mmd_restructure_bones', text=t("MMD.restructure_bones"))
|
|
||||||
col.separator(factor=0.2)
|
|
||||||
|
|
||||||
# Bone cleanup options
|
|
||||||
col.label(text=t("MMD.bone_cleanup"), icon='BONE_DATA')
|
|
||||||
cleanup_box = col.box()
|
|
||||||
cleanup_col = cleanup_box.column(align=True)
|
|
||||||
cleanup_col.prop(toolkit, 'mmd_remove_twist_bones', text=t("MMD.remove_twist_bones"))
|
|
||||||
cleanup_col.prop(toolkit, 'mmd_remove_zero_weight_bones', text=t("MMD.remove_zero_weight_bones"))
|
|
||||||
col.separator(factor=0.2)
|
|
||||||
|
|
||||||
# Translation settings
|
|
||||||
col.prop(toolkit, 'mmd_translate_names', text=t("MMD.translate_names"))
|
|
||||||
|
|
||||||
# Translation sub-options (only show if translation is enabled)
|
|
||||||
if toolkit.mmd_translate_names:
|
|
||||||
trans_box = col.box()
|
|
||||||
trans_col = trans_box.column(align=True)
|
|
||||||
trans_col.label(text=t("MMD.translation_options"), icon='WORLD')
|
|
||||||
trans_col.prop(toolkit, 'mmd_translate_bones', text=t("MMD.translate_bones"))
|
|
||||||
trans_col.prop(toolkit, 'mmd_translate_materials', text=t("MMD.translate_materials"))
|
|
||||||
trans_col.prop(toolkit, 'mmd_translate_shapekeys', text=t("MMD.translate_shapekeys"))
|
|
||||||
trans_col.prop(toolkit, 'mmd_translate_objects', text=t("MMD.translate_objects"))
|
|
||||||
|
|
||||||
col.separator(factor=0.2)
|
|
||||||
|
|
||||||
col.operator(
|
|
||||||
AvatarToolkit_OT_ConvertMMDArmature.bl_idname,
|
|
||||||
text=t("MMD.convert_armature_button"),
|
|
||||||
icon='EXPORT'
|
|
||||||
)
|
|
||||||
|
|
||||||
info_box = mmd_box.box()
|
|
||||||
info_col = info_box.column(align=True)
|
|
||||||
info_col.label(text=t("MMD.conversion_info.title"), icon='INFO')
|
|
||||||
info_col.label(text=t("MMD.conversion_info.removes_parent"))
|
|
||||||
info_col.label(text=t("MMD.conversion_info.renames_armature"))
|
|
||||||
if toolkit.mmd_restructure_bones:
|
|
||||||
info_col.label(text=t("MMD.conversion_info.restructures_bones"))
|
|
||||||
info_col.label(text=t("MMD.conversion_info.removes_ik_bones"))
|
|
||||||
if toolkit.mmd_remove_twist_bones:
|
|
||||||
info_col.label(text=t("MMD.conversion_info.removes_twist_bones"))
|
|
||||||
if toolkit.mmd_remove_zero_weight_bones:
|
|
||||||
info_col.label(text=t("MMD.conversion_info.removes_zero_weight_bones"))
|
|
||||||
info_col.label(text=t("MMD.conversion_info.maintains_hierarchy"))
|
|
||||||
if toolkit.mmd_translate_names:
|
|
||||||
info_col.label(text=t("MMD.conversion_info.translates_names"))
|
|
||||||
|
|
||||||
else:
|
|
||||||
col.label(text=t("MMD.armature_name", name=armature.name), icon='ERROR')
|
|
||||||
col.label(text=t("MMD.no_mmd_bones_detected"), icon='CANCEL')
|
|
||||||
col.separator(factor=0.3)
|
|
||||||
|
|
||||||
row = col.row()
|
|
||||||
row.enabled = False
|
|
||||||
row.operator(
|
|
||||||
AvatarToolkit_OT_ConvertMMDArmature.bl_idname,
|
|
||||||
text=t("MMD.convert_armature_button"),
|
|
||||||
icon='CANCEL'
|
|
||||||
)
|
|
||||||
|
|
||||||
help_box = mmd_box.box()
|
|
||||||
help_col = help_box.column(align=True)
|
|
||||||
help_col.label(text=t("MMD.detection_failed.title"), icon='QUESTION')
|
|
||||||
help_col.label(text=t("MMD.detection_failed.not_mmd_format"))
|
|
||||||
help_col.label(text=t("MMD.detection_failed.need_mmd_bones"))
|
|
||||||
help_col.label(text=t("MMD.detection_failed.check_bone_names"))
|
|
||||||
Reference in New Issue
Block a user