Yusarina
232 lines
8.5 KiB
Python
232 lines
8.5 KiB
Python
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')
|