Yusarina
113 lines
5.3 KiB
Python
113 lines
5.3 KiB
Python
import bpy
|
|
from typing import Set
|
|
from bpy.types import Panel, Context, UILayout, Operator, UIList
|
|
from .main_panel import AvatarToolKit_PT_AvatarToolkitPanel, CATEGORY_NAME
|
|
from ..core.translations import t
|
|
|
|
class AVATAR_TOOLKIT_UL_ZeroWeightBones(UIList):
|
|
"""UI List for displaying zero weight bones with selection options"""
|
|
def draw_item(self, context, layout, data, item, icon, active_data, active_propname):
|
|
if self.layout_type in {'DEFAULT', 'COMPACT'}:
|
|
row = layout.row(align=True)
|
|
row.prop(item, "selected", text="")
|
|
row.label(text=item.name)
|
|
if item.has_children:
|
|
row.label(text="", icon='OUTLINER_OB_ARMATURE')
|
|
if item.is_deform:
|
|
row.label(text="", icon='MOD_ARMATURE')
|
|
|
|
class AvatarToolKit_PT_ToolsPanel(Panel):
|
|
"""Panel containing various tools for avatar customization and optimization"""
|
|
bl_label: str = t("Tools.label")
|
|
bl_idname: str = "OBJECT_PT_avatar_toolkit_tools"
|
|
bl_space_type: str = 'VIEW_3D'
|
|
bl_region_type: str = 'UI'
|
|
bl_category: str = CATEGORY_NAME
|
|
bl_parent_id: str = AvatarToolKit_PT_AvatarToolkitPanel.bl_idname
|
|
bl_order: int = 2
|
|
bl_options = {'DEFAULT_CLOSED'}
|
|
|
|
def draw(self, context: Context) -> None:
|
|
"""Draw the tools panel interface"""
|
|
layout: UILayout = self.layout
|
|
toolkit = context.scene.avatar_toolkit
|
|
|
|
# General Tools
|
|
tools_box: UILayout = layout.box()
|
|
col: UILayout = tools_box.column(align=True)
|
|
col.label(text=t("Tools.general_title"), icon='TOOL_SETTINGS')
|
|
col.separator(factor=0.5)
|
|
col.operator("avatar_toolkit.convert_resonite", text=t("Tools.convert_resonite"), icon='EXPORT')
|
|
|
|
# Separation Tools
|
|
sep_box: UILayout = layout.box()
|
|
col = sep_box.column(align=True)
|
|
col.label(text=t("Tools.separate_title"), icon='MOD_EXPLODE')
|
|
col.separator(factor=0.5)
|
|
row: UILayout = col.row(align=True)
|
|
row.operator("avatar_toolkit.separate_materials", text=t("Tools.separate_materials"), icon='MATERIAL')
|
|
row.operator("avatar_toolkit.separate_loose", text=t("Tools.separate_loose"), icon='MESH_DATA')
|
|
|
|
# Bone Tools
|
|
bone_box: UILayout = layout.box()
|
|
col = bone_box.column(align=True)
|
|
col.label(text=t("Tools.bone_title"), icon='BONE_DATA')
|
|
col.separator(factor=0.5)
|
|
col.operator("avatar_toolkit.create_digitigrade", text=t("Tools.create_digitigrade"), icon='BONE_DATA')
|
|
|
|
# Standardization Tools
|
|
standardize_box: UILayout = bone_box.box()
|
|
col = standardize_box.column(align=True)
|
|
col.label(text=t("Tools.standardize_title"), icon='OUTLINER_OB_ARMATURE')
|
|
col.separator(factor=0.5)
|
|
col.operator("avatar_toolkit.standardize_armature", icon='CHECKMARK')
|
|
|
|
# Weight Tools
|
|
weight_box: UILayout = bone_box.box()
|
|
col = weight_box.column(align=True)
|
|
col.prop(toolkit, "merge_twist_bones", text=t("Tools.merge_twist_bones"))
|
|
col.prop(toolkit, "preserve_parent_bones")
|
|
col.prop(toolkit, "target_bone_type")
|
|
col.prop(toolkit, "list_only_mode")
|
|
|
|
if toolkit.list_only_mode and len(toolkit.zero_weight_bones) > 0:
|
|
box = weight_box.box()
|
|
row = box.row()
|
|
row.template_list("AVATAR_TOOLKIT_UL_ZeroWeightBones", "",
|
|
toolkit, "zero_weight_bones",
|
|
toolkit, "zero_weight_bones_index")
|
|
|
|
col = box.column(align=True)
|
|
col.operator("avatar_toolkit.remove_selected_bones",
|
|
text=t("Tools.remove_selected_bones"))
|
|
|
|
row = col.row(align=True)
|
|
row.operator("avatar_toolkit.clean_weights", text=t("Tools.clean_weights"), icon='GROUP_BONE')
|
|
row.operator("avatar_toolkit.clean_constraints", text=t("Tools.clean_constraints"), icon='CONSTRAINT_BONE')
|
|
|
|
# Merge Tools
|
|
merge_box: UILayout = layout.box()
|
|
col = merge_box.column(align=True)
|
|
col.label(text=t("Tools.merge_title"), icon='AUTOMERGE_ON')
|
|
col.separator(factor=0.5)
|
|
row = col.row(align=True)
|
|
row.operator("avatar_toolkit.merge_to_active", text=t("Tools.merge_to_active"), icon='BONE_DATA')
|
|
row.operator("avatar_toolkit.merge_to_parent", text=t("Tools.merge_to_parent"), icon='BONE_DATA')
|
|
col.operator("avatar_toolkit.connect_bones", text=t("Tools.connect_bones"), icon='BONE_DATA')
|
|
|
|
# Additional Tools
|
|
extra_box: UILayout = layout.box()
|
|
col = extra_box.column(align=True)
|
|
col.label(text=t("Tools.additional_title"), icon='TOOL_SETTINGS')
|
|
col.separator(factor=0.5)
|
|
col.operator("avatar_toolkit.apply_transforms", text=t("Tools.apply_transforms"), icon='OBJECT_DATA')
|
|
col.operator("avatar_toolkit.clean_shapekeys", text=t("Tools.clean_shapekeys"), icon='SHAPEKEY_DATA')
|
|
|
|
# Rigify Tools
|
|
rigify_box: UILayout = layout.box()
|
|
col = rigify_box.column(align=True)
|
|
col.label(text=t("Tools.rigify_title"), icon='ARMATURE_DATA')
|
|
col.separator(factor=0.5)
|
|
col.operator("avatar_toolkit.convert_rigify_to_unity", icon='ARMATURE_DATA')
|
|
col.prop(context.scene.avatar_toolkit, "merge_twist_bones")
|