Armature Merging
This commit is contained in:
Yusarina
@@ -0,0 +1,231 @@
|
||||
import bpy
|
||||
from typing import Set
|
||||
from bpy.types import Panel, Context, UILayout, Operator
|
||||
from .main_panel import AvatarToolKit_PT_AvatarToolkitPanel, CATEGORY_NAME
|
||||
from ..core.translations import t
|
||||
from ..core.common import (
|
||||
get_active_armature,
|
||||
get_all_meshes,
|
||||
validate_armature,
|
||||
get_armature_list
|
||||
)
|
||||
|
||||
class AvatarToolkit_OT_SearchMergeArmatureInto(Operator):
|
||||
bl_idname = "avatar_toolkit.search_merge_armature_into"
|
||||
bl_label = ""
|
||||
bl_description = t('CustomPanel.search_merge_into_desc')
|
||||
bl_property = "search_merge_armature_into_enum"
|
||||
|
||||
# Define the enum property within the operator class
|
||||
search_merge_armature_into_enum: bpy.props.EnumProperty(
|
||||
name=t('CustomPanel.merge_into'),
|
||||
description=t('CustomPanel.merge_into_desc'),
|
||||
items=get_armature_list
|
||||
)
|
||||
|
||||
def execute(self, context):
|
||||
context.scene.avatar_toolkit.merge_armature_into = self.search_merge_armature_into_enum
|
||||
return {'FINISHED'}
|
||||
|
||||
def invoke(self, context, event):
|
||||
context.window_manager.invoke_search_popup(self)
|
||||
return {'FINISHED'}
|
||||
|
||||
class AvatarToolkit_OT_SearchMergeArmature(Operator):
|
||||
bl_idname = "avatar_toolkit.search_merge_armature"
|
||||
bl_label = ""
|
||||
bl_description = t('CustomPanel.search_merge_desc')
|
||||
bl_property = "search_merge_armature_enum"
|
||||
|
||||
search_merge_armature_enum: bpy.props.EnumProperty(
|
||||
name=t('CustomPanel.merge_from'),
|
||||
description=t('CustomPanel.merge_from_desc'),
|
||||
items=get_armature_list
|
||||
)
|
||||
|
||||
def execute(self, context):
|
||||
context.scene.avatar_toolkit.merge_armature = self.search_merge_armature_enum
|
||||
return {'FINISHED'}
|
||||
|
||||
def invoke(self, context, event):
|
||||
context.window_manager.invoke_search_popup(self)
|
||||
return {'FINISHED'}
|
||||
|
||||
class AvatarToolkit_OT_SearchAttachMesh(Operator):
|
||||
bl_idname = "avatar_toolkit.search_attach_mesh"
|
||||
bl_label = ""
|
||||
bl_description = t('CustomPanel.search_mesh_desc')
|
||||
bl_property = "search_attach_mesh_enum"
|
||||
|
||||
search_attach_mesh_enum: bpy.props.EnumProperty(
|
||||
name=t('CustomPanel.attach_mesh'),
|
||||
description=t('CustomPanel.attach_mesh_desc'),
|
||||
items=lambda self, context: [
|
||||
(obj.name, obj.name, "")
|
||||
for obj in get_all_meshes(context)
|
||||
]
|
||||
)
|
||||
|
||||
def execute(self, context):
|
||||
context.scene.avatar_toolkit.attach_mesh = self.search_attach_mesh_enum
|
||||
return {'FINISHED'}
|
||||
|
||||
def invoke(self, context, event):
|
||||
context.window_manager.invoke_search_popup(self)
|
||||
return {'FINISHED'}
|
||||
|
||||
class AvatarToolkit_OT_SearchAttachBone(Operator):
|
||||
bl_idname = "avatar_toolkit.search_attach_bone"
|
||||
bl_label = ""
|
||||
bl_description = t('CustomPanel.search_bone_desc')
|
||||
bl_property = "search_attach_bone_enum"
|
||||
|
||||
search_attach_bone_enum: bpy.props.EnumProperty(
|
||||
name=t('CustomPanel.attach_bone'),
|
||||
description=t('CustomPanel.attach_bone_desc'),
|
||||
items=lambda self, context: [
|
||||
(bone.name, bone.name, "")
|
||||
for bone in get_active_armature(context).data.bones
|
||||
] if get_active_armature(context) else []
|
||||
)
|
||||
|
||||
def execute(self, context):
|
||||
context.scene.avatar_toolkit.attach_bone = self.search_attach_bone_enum
|
||||
return {'FINISHED'}
|
||||
|
||||
def invoke(self, context, event):
|
||||
context.window_manager.invoke_search_popup(self)
|
||||
return {'FINISHED'}
|
||||
|
||||
class AvatarToolKit_PT_CustomPanel(Panel):
|
||||
"""Panel containing tools for custom avatar creation and merging"""
|
||||
bl_label = t('CustomPanel.label')
|
||||
bl_idname = "VIEW3D_PT_avatar_toolkit_custom"
|
||||
bl_space_type = 'VIEW_3D'
|
||||
bl_region_type = 'UI'
|
||||
bl_category = CATEGORY_NAME
|
||||
bl_parent_id = AvatarToolKit_PT_AvatarToolkitPanel.bl_idname
|
||||
bl_order = 3
|
||||
bl_options = {'DEFAULT_CLOSED'}
|
||||
|
||||
def draw(self, context: Context) -> None:
|
||||
"""Draw the custom avatar tools panel interface"""
|
||||
layout: UILayout = self.layout
|
||||
toolkit = context.scene.avatar_toolkit
|
||||
|
||||
# Mode Selection Box
|
||||
mode_box: UILayout = layout.box()
|
||||
col: UILayout = mode_box.column(align=True)
|
||||
col.label(text=t('CustomPanel.merge_mode'), icon='TOOL_SETTINGS')
|
||||
col.separator(factor=0.5)
|
||||
|
||||
# Create a row for the mode buttons with increased scale
|
||||
row: UILayout = col.row(align=True)
|
||||
row.scale_y = 1.5
|
||||
row.prop(toolkit, "merge_mode", expand=True)
|
||||
|
||||
# Armature Merging Tools
|
||||
if toolkit.merge_mode == 'ARMATURE':
|
||||
self.draw_armature_tools(layout, context)
|
||||
# Mesh Attachment Tools
|
||||
else:
|
||||
self.draw_mesh_tools(layout, context)
|
||||
|
||||
def draw_armature_tools(self, layout: UILayout, context: Context) -> None:
|
||||
"""Draw the armature merging tools section"""
|
||||
toolkit = context.scene.avatar_toolkit
|
||||
|
||||
# Merge Settings Box
|
||||
settings_box: UILayout = layout.box()
|
||||
col: UILayout = settings_box.column(align=True)
|
||||
col.label(text=t('CustomPanel.mergeArmatures'), icon='ARMATURE_DATA')
|
||||
col.separator(factor=0.5)
|
||||
|
||||
if len(get_armature_list(context)) <= 1:
|
||||
col.label(text=t('CustomPanel.warn.twoArmatures'), icon='INFO')
|
||||
return
|
||||
|
||||
# Merge Options
|
||||
options_box: UILayout = layout.box()
|
||||
col: UILayout = options_box.column(align=True)
|
||||
col.label(text=t('Tools.merge_title'), icon='SETTINGS')
|
||||
col.separator(factor=0.5)
|
||||
col.prop(toolkit, "merge_all_bones")
|
||||
col.prop(toolkit, "apply_transforms")
|
||||
col.prop(toolkit, "join_meshes")
|
||||
col.prop(toolkit, "remove_zero_weights")
|
||||
col.prop(toolkit, "cleanup_shape_keys")
|
||||
|
||||
# Armature Selection Box
|
||||
selection_box: UILayout = layout.box()
|
||||
col: UILayout = selection_box.column(align=True)
|
||||
col.label(text=t('QuickAccess.select_armature'), icon='BONE_DATA')
|
||||
col.separator(factor=0.5)
|
||||
|
||||
row: UILayout = col.row(align=True)
|
||||
row.label(text=t('CustomPanel.mergeInto'))
|
||||
row.operator("avatar_toolkit.search_merge_armature_into",
|
||||
text=toolkit.merge_armature_into,
|
||||
icon='ARMATURE_DATA')
|
||||
|
||||
row: UILayout = col.row(align=True)
|
||||
row.label(text=t('CustomPanel.toMerge'))
|
||||
row.operator("avatar_toolkit.search_merge_armature",
|
||||
text=toolkit.merge_armature,
|
||||
icon='ARMATURE_DATA')
|
||||
|
||||
# Merge Button
|
||||
merge_col: UILayout = layout.column(align=True)
|
||||
merge_col.scale_y = 1.2
|
||||
merge_col.operator("avatar_toolkit.merge_armatures", icon='ARMATURE_DATA')
|
||||
|
||||
def draw_mesh_tools(self, layout: UILayout, context: Context) -> None:
|
||||
"""Draw the mesh attachment tools section"""
|
||||
toolkit = context.scene.avatar_toolkit
|
||||
|
||||
# Mesh Tools Box
|
||||
tools_box: UILayout = layout.box()
|
||||
col: UILayout = tools_box.column(align=True)
|
||||
col.label(text=t('CustomPanel.attachMesh1'), icon='MESH_DATA')
|
||||
col.separator(factor=0.5)
|
||||
|
||||
if not get_active_armature(context) or not get_all_meshes(context):
|
||||
col.label(text=t('CustomPanel.warn.noArmOrMesh1'), icon='INFO')
|
||||
col.label(text=t('CustomPanel.warn.noArmOrMesh2'))
|
||||
return
|
||||
|
||||
# Mesh Options Box
|
||||
options_box: UILayout = layout.box()
|
||||
col: UILayout = options_box.column(align=True)
|
||||
col.label(text=t('Tools.merge_title'), icon='SETTINGS')
|
||||
col.separator(factor=0.5)
|
||||
col.prop(toolkit, "join_meshes")
|
||||
|
||||
# Selection Box
|
||||
selection_box: UILayout = layout.box()
|
||||
col: UILayout = selection_box.column(align=True)
|
||||
col.label(text=t('Tools.merge_title'), icon='OBJECT_DATA')
|
||||
col.separator(factor=0.5)
|
||||
|
||||
row: UILayout = col.row(align=True)
|
||||
row.label(text=t('CustomPanel.mergeInto'))
|
||||
row.operator("avatar_toolkit.search_merge_armature_into",
|
||||
text=toolkit.merge_armature_into,
|
||||
icon='ARMATURE_DATA')
|
||||
|
||||
row: UILayout = col.row(align=True)
|
||||
row.label(text=t('CustomPanel.attachMesh2'))
|
||||
row.operator("avatar_toolkit.search_attach_mesh",
|
||||
text=toolkit.attach_mesh,
|
||||
icon='MESH_DATA')
|
||||
|
||||
row: UILayout = col.row(align=True)
|
||||
row.label(text=t('CustomPanel.attachToBone'))
|
||||
row.operator("avatar_toolkit.search_attach_bone",
|
||||
text=toolkit.attach_bone,
|
||||
icon='BONE_DATA')
|
||||
|
||||
# Attach Button
|
||||
attach_col: UILayout = layout.column(align=True)
|
||||
attach_col.scale_y = 1.2
|
||||
attach_col.operator("avatar_toolkit.attach_mesh", icon='ARMATURE_DATA')
|
||||
Reference in New Issue
Block a user