529 lines
25 KiB
Python
529 lines
25 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
|
||
"""
|
||
Rigging UI Module for Plugin
|
||
绑定系统UI模块 - 负责显示骨骼绑定编辑界面和基础操作
|
||
基本功能：
|
||
- DNA浏览器
|
||
- 根据DNA导入骨骼
|
||
- 根据DNA生成身体
|
||
- DNA校准
|
||
- 骨骼位置校准
|
||
- 创建绑定
|
||
- 复制蒙皮
|
||
"""
|
||
#========================================= IMPORT =========================================
|
||
from Qt import QtWidgets, QtCore, QtGui
|
||
from Qt.QtCompat import wrapInstance
|
||
from maya import OpenMayaUI as omui
|
||
import maya.cmds as cmds
|
||
import maya.mel as mel
|
||
import maya.utils as utils
|
||
import webbrowser
|
||
import subprocess
|
||
import importlib
|
||
import traceback
|
||
import locale
|
||
import sys
|
||
import os
|
||
from scripts.ui import ui_utils
|
||
from scripts.utils import utils_rigging
|
||
#========================================== CONFIG ========================================
|
||
import config
|
||
TOOL_NAME = config.TOOL_NAME
|
||
TOOL_VERSION = config.TOOL_VERSION
|
||
TOOL_AUTHOR = config.TOOL_AUTHOR
|
||
TOOL_YEAR = config.TOOL_YEAR
|
||
TOOL_MOD_FILENAME = config.TOOL_MOD_FILENAME
|
||
TOOL_LANG = config.TOOL_LANG
|
||
TOOL_WSCL_NAME = config.TOOL_WSCL_NAME
|
||
TOOL_HELP_URL = config.TOOL_HELP_URL
|
||
TOOL_PATH = config.TOOL_PATH
|
||
SCRIPTS_PATH = config.SCRIPTS_PATH
|
||
TOOL_MAIN_SCRIPT = config.TOOL_MAIN_SCRIPT
|
||
UI_PATH = config.UI_PATH
|
||
STYLE_FILE = config.STYLE_FILE
|
||
ICONS_PATH = config.ICONS_PATH
|
||
TOOL_ICON = config.TOOL_ICON
|
||
ASSETS_PATH = config.ASSETS_PATH
|
||
DNA_FILE_PATH = config.DNA_FILE_PATH
|
||
DNA_IMG_PATH = config.DNA_IMG_PATH
|
||
TOOL_COMMAND_ICON = config.TOOL_COMMAND_ICON
|
||
TOOL_WIDTH = config.TOOL_WIDTH
|
||
TOOL_HEIGHT = config.TOOL_HEIGHT
|
||
#========================================= LOCATION =======================================
|
||
from scripts.ui import localization
|
||
LANG = localization.LANG
|
||
get_text = localization.get_text
|
||
|
||
class RiggingUI(ui_utils.BaseUI):
|
||
"""
|
||
绑定系统UI类 - 负责显示骨骼绑定编辑界面和基础操作
|
||
继承自BaseUI类，实现绑定系统相关的UI功能
|
||
"""
|
||
#========================================== INIT ========================================
|
||
def __init__(self):
|
||
"""
|
||
初始化绑定系统UI
|
||
创建主控件和布局，并连接信号和槽
|
||
"""
|
||
super(RiggingUI, self).__init__()
|
||
|
||
# 初始化字典
|
||
self.controls = {}
|
||
self.layouts = {}
|
||
self.buttons = {}
|
||
self.splitters = {}
|
||
self.inputs = {}
|
||
self.labels = {}
|
||
|
||
# 创建主控件
|
||
self.main_widget = QtWidgets.QWidget()
|
||
self.main_widget.setObjectName("riggingMainWidget")
|
||
|
||
# 初始化UI
|
||
self.create_widgets()
|
||
self.create_layouts()
|
||
self.create_connections()
|
||
|
||
#========================================= WIDGET =======================================
|
||
def create_widgets(self):
|
||
"""
|
||
创建绑定系统UI控件
|
||
包括按钮、标签、列表等
|
||
"""
|
||
# 标题标签 - 使用HTML格式化标题
|
||
title_text = f"<h4 style='margin:0;padding:5px;'>{get_text('rigging_title', '骨骼绑定')}</h4>"
|
||
self.controls["title_label"] = QtWidgets.QLabel(title_text)
|
||
self.controls["title_label"].setObjectName("riggingTitleLabel")
|
||
self.controls["title_label"].setAlignment(QtCore.Qt.AlignCenter)
|
||
self.controls["title_label"].setMaximumHeight(30) # 限制标题高度
|
||
|
||
# 创建主分割器
|
||
self.splitters["main_splitter"] = QtWidgets.QSplitter(QtCore.Qt.Vertical)
|
||
self.splitters["main_splitter"].setObjectName("riggingMainSplitter")
|
||
|
||
# 1. Presets 面板
|
||
self.controls["presets_panel"] = QtWidgets.QWidget()
|
||
self.controls["presets_panel"].setObjectName("presetsPanel")
|
||
|
||
# Presets 组
|
||
self.controls["presets_group"] = QtWidgets.QGroupBox(get_text("presets", "Presets"))
|
||
self.controls["presets_group"].setObjectName("presetsGroup")
|
||
|
||
# 创建预设显示区域
|
||
self.controls["presets_scroll_area"] = QtWidgets.QScrollArea()
|
||
self.controls["presets_scroll_area"].setWidgetResizable(True)
|
||
self.controls["presets_scroll_area"].setObjectName("presetsScrollArea")
|
||
|
||
# 预设内容区域
|
||
self.controls["presets_content"] = QtWidgets.QWidget()
|
||
self.controls["presets_content"].setObjectName("presetsContent")
|
||
|
||
# 预设流布局
|
||
self.controls["presets_flow_layout"] = QtWidgets.QGridLayout(self.controls["presets_content"])
|
||
self.controls["presets_flow_layout"].setObjectName("presetsFlowLayout")
|
||
self.controls["presets_flow_layout"].setContentsMargins(5, 5, 5, 5)
|
||
self.controls["presets_flow_layout"].setSpacing(10)
|
||
|
||
# 添加测试预设项
|
||
for i in range(6): # 添加6个预设项，如图中所示
|
||
col = i % 6
|
||
row = i // 6
|
||
|
||
preset_widget = QtWidgets.QWidget()
|
||
preset_widget.setObjectName(f"preset_{i}")
|
||
preset_widget.setMinimumSize(120, 150) # 设置预设项大小
|
||
preset_widget.setMaximumSize(120, 150)
|
||
|
||
# 预设布局
|
||
preset_layout = QtWidgets.QVBoxLayout(preset_widget)
|
||
preset_layout.setContentsMargins(0, 0, 0, 0)
|
||
preset_layout.setSpacing(2)
|
||
|
||
# 预设图片
|
||
preset_image = QtWidgets.QLabel()
|
||
preset_image.setObjectName(f"preset_image_{i}")
|
||
preset_image.setMinimumSize(120, 120)
|
||
preset_image.setMaximumSize(120, 120)
|
||
preset_image.setScaledContents(True)
|
||
preset_image.setStyleSheet("background-color: #333333; border: 1px solid #555555;")
|
||
|
||
# 加载测试图片
|
||
pixmap = QtGui.QPixmap(os.path.join(ASSETS_PATH, "metahuman_placeholder.png"))
|
||
if not pixmap.isNull():
|
||
preset_image.setPixmap(pixmap)
|
||
|
||
# 预设标签
|
||
preset_label = QtWidgets.QLabel("METAHUMAN")
|
||
preset_label.setObjectName(f"preset_label_{i}")
|
||
preset_label.setAlignment(QtCore.Qt.AlignCenter)
|
||
preset_label.setStyleSheet("color: white; background-color: rgba(0, 0, 0, 128);")
|
||
|
||
# 添加到布局
|
||
preset_layout.addWidget(preset_image)
|
||
preset_layout.addWidget(preset_label)
|
||
|
||
# 添加到流布局
|
||
self.controls["presets_flow_layout"].addWidget(preset_widget, row, col)
|
||
|
||
# 设置滚动区域内容
|
||
self.controls["presets_scroll_area"].setWidget(self.controls["presets_content"])
|
||
|
||
# 预设底部滑块和按钮
|
||
self.controls["presets_slider_layout"] = QtWidgets.QHBoxLayout()
|
||
self.controls["presets_slider_layout"].setContentsMargins(5, 5, 5, 5)
|
||
self.controls["presets_slider_layout"].setSpacing(5)
|
||
|
||
# 数量显示
|
||
self.controls["presets_count_label"] = QtWidgets.QLabel("99")
|
||
self.controls["presets_count_label"].setObjectName("presetsCountLabel")
|
||
self.controls["presets_count_label"].setAlignment(QtCore.Qt.AlignCenter)
|
||
self.controls["presets_count_label"].setStyleSheet("font-weight: bold;")
|
||
|
||
# 滑块
|
||
self.controls["presets_slider"] = QtWidgets.QSlider(QtCore.Qt.Horizontal)
|
||
self.controls["presets_slider"].setObjectName("presetsSlider")
|
||
self.controls["presets_slider"].setMinimum(0)
|
||
self.controls["presets_slider"].setMaximum(100)
|
||
self.controls["presets_slider"].setValue(50)
|
||
|
||
# 导出预设按钮
|
||
self.buttons["export_presets"] = QtWidgets.QPushButton(get_text("export_presets", "导出预设"))
|
||
self.buttons["export_presets"].setObjectName("exportPresetsButton")
|
||
self.buttons["export_presets"].setIcon(ui_utils.load_icon("export"))
|
||
|
||
# 导入预设按钮
|
||
self.buttons["import_presets"] = QtWidgets.QPushButton(get_text("import_presets", "导入预设"))
|
||
self.buttons["import_presets"].setObjectName("importPresetsButton")
|
||
self.buttons["import_presets"].setIcon(ui_utils.load_icon("import"))
|
||
|
||
# 添加到滑块布局
|
||
self.controls["presets_slider_layout"].addWidget(self.controls["presets_count_label"])
|
||
self.controls["presets_slider_layout"].addWidget(self.controls["presets_slider"])
|
||
self.controls["presets_slider_layout"].addWidget(self.buttons["export_presets"])
|
||
self.controls["presets_slider_layout"].addWidget(self.buttons["import_presets"])
|
||
|
||
# 2. Assets 面板
|
||
self.controls["assets_panel"] = QtWidgets.QWidget()
|
||
self.controls["assets_panel"].setObjectName("assetsPanel")
|
||
|
||
# Assets 组
|
||
self.controls["assets_group"] = QtWidgets.QGroupBox(get_text("assets", "Assets"))
|
||
self.controls["assets_group"].setObjectName("assetsGroup")
|
||
|
||
# 项目路径标签和输入框
|
||
self.controls["project_path_label"] = QtWidgets.QLabel(get_text("project_path", "项目路径:"))
|
||
self.controls["project_path_label"].setObjectName("projectPathLabel")
|
||
|
||
self.controls["project_path_input"] = QtWidgets.QLineEdit()
|
||
self.controls["project_path_input"].setObjectName("projectPathInput")
|
||
self.controls["project_path_input"].setText("D:/Personal/Document/maya/SuperRiggingEditor/files/data/MetaHuman")
|
||
|
||
# 浏览按钮
|
||
self.buttons["browse_path"] = QtWidgets.QPushButton(" . . . ")
|
||
self.buttons["browse_path"].setObjectName("browsePathButton")
|
||
self.buttons["browse_path"].setIcon(ui_utils.load_icon("import"))
|
||
self.buttons["browse_path"].setMinimumWidth(0)
|
||
self.buttons["browse_path"].setFixedWidth(100)
|
||
|
||
# Presets DNA 标签和输入框
|
||
self.controls["presets_dna_label"] = QtWidgets.QLabel(get_text("Presets DNA:", "预设 DNA:"))
|
||
self.controls["presets_dna_label"].setObjectName("presetsDnaLabel")
|
||
|
||
self.controls["presets_dna_input"] = QtWidgets.QLineEdit()
|
||
self.controls["presets_dna_input"].setObjectName("presetsDnaInput")
|
||
|
||
# 浏览按钮
|
||
self.buttons["browse_dna"] = QtWidgets.QPushButton(" . . . ")
|
||
self.buttons["browse_dna"].setObjectName("browseDnaButton")
|
||
self.buttons["browse_dna"].setIcon(ui_utils.load_icon("import"))
|
||
self.buttons["browse_dna"].setMinimumWidth(0)
|
||
self.buttons["browse_dna"].setFixedWidth(100)
|
||
|
||
# 3. Descriptor 面板
|
||
self.controls["descriptor_panel"] = QtWidgets.QWidget()
|
||
self.controls["descriptor_panel"].setObjectName("descriptorPanel")
|
||
|
||
# Descriptor 组
|
||
self.controls["descriptor_group"] = QtWidgets.QGroupBox(get_text("descriptor", "Descriptor"))
|
||
self.controls["descriptor_group"].setObjectName("descriptorGroup")
|
||
|
||
# 名称标签和输入框
|
||
self.controls["name_label"] = QtWidgets.QLabel(get_text("name", "名称:"))
|
||
self.controls["name_label"].setObjectName("nameLabel")
|
||
|
||
self.controls["name_input"] = QtWidgets.QLineEdit()
|
||
self.controls["name_input"].setObjectName("nameInput")
|
||
|
||
# 原型标签和下拉框
|
||
self.controls["archetype_label"] = QtWidgets.QLabel(get_text("archetype", "原型:"))
|
||
self.controls["archetype_label"].setObjectName("archetypeLabel")
|
||
|
||
self.controls["archetype_combo"] = QtWidgets.QComboBox()
|
||
self.controls["archetype_combo"].setObjectName("archetypeCombo")
|
||
self.controls["archetype_combo"].addItem("asian")
|
||
self.controls["archetype_combo"].addItem("caucasian")
|
||
self.controls["archetype_combo"].addItem("african")
|
||
|
||
# 性别标签和下拉框
|
||
self.controls["gender_label"] = QtWidgets.QLabel(get_text("gender", "性别:"))
|
||
self.controls["gender_label"].setObjectName("genderLabel")
|
||
|
||
self.controls["gender_combo"] = QtWidgets.QComboBox()
|
||
self.controls["gender_combo"].setObjectName("genderCombo")
|
||
self.controls["gender_combo"].addItem("female")
|
||
self.controls["gender_combo"].addItem("male")
|
||
|
||
# 年龄标签和输入框
|
||
self.controls["age_label"] = QtWidgets.QLabel(get_text("age", "年龄:"))
|
||
self.controls["age_label"].setObjectName("ageLabel")
|
||
|
||
self.controls["age_spinner"] = QtWidgets.QSpinBox()
|
||
self.controls["age_spinner"].setObjectName("ageSpinner")
|
||
self.controls["age_spinner"].setMinimum(1)
|
||
self.controls["age_spinner"].setMaximum(100)
|
||
self.controls["age_spinner"].setValue(24)
|
||
|
||
# 平移单位标签和下拉框
|
||
self.controls["translation_unit_label"] = QtWidgets.QLabel(get_text("translation_unit", "平移单位:"))
|
||
self.controls["translation_unit_label"].setObjectName("translationUnitLabel")
|
||
|
||
self.controls["translation_unit_combo"] = QtWidgets.QComboBox()
|
||
self.controls["translation_unit_combo"].setObjectName("translationUnitCombo")
|
||
self.controls["translation_unit_combo"].addItem("cm")
|
||
self.controls["translation_unit_combo"].addItem("mm")
|
||
self.controls["translation_unit_combo"].addItem("m")
|
||
|
||
# 旋转单位标签和下拉框
|
||
self.controls["rotation_unit_label"] = QtWidgets.QLabel(get_text("rotation_unit", "旋转单位:"))
|
||
self.controls["rotation_unit_label"].setObjectName("rotationUnitLabel")
|
||
|
||
self.controls["rotation_unit_combo"] = QtWidgets.QComboBox()
|
||
self.controls["rotation_unit_combo"].setObjectName("rotationUnitCombo")
|
||
self.controls["rotation_unit_combo"].addItem("degrees")
|
||
self.controls["rotation_unit_combo"].addItem("radians")
|
||
|
||
# 坐标系统标签和下拉框
|
||
self.controls["coordinate_system_label"] = QtWidgets.QLabel(get_text("coordinate_system", "坐标系统:"))
|
||
self.controls["coordinate_system_label"].setObjectName("coordinateSystemLabel")
|
||
|
||
self.controls["coordinate_system_combo"] = QtWidgets.QComboBox()
|
||
self.controls["coordinate_system_combo"].setObjectName("coordinateSystemCombo")
|
||
self.controls["coordinate_system_combo"].addItem("YAxisUp")
|
||
self.controls["coordinate_system_combo"].addItem("ZAxisUp")
|
||
|
||
# LOD数量标签和输入框
|
||
self.controls["lod_count_label"] = QtWidgets.QLabel(get_text("lod_count", "LOD数量:"))
|
||
self.controls["lod_count_label"].setObjectName("lodCountLabel")
|
||
|
||
self.controls["lod_count_spinner"] = QtWidgets.QSpinBox()
|
||
self.controls["lod_count_spinner"].setObjectName("lodCountSpinner")
|
||
self.controls["lod_count_spinner"].setMinimum(0)
|
||
self.controls["lod_count_spinner"].setMaximum(8)
|
||
self.controls["lod_count_spinner"].setValue(0)
|
||
|
||
# 底部按钮区域
|
||
self.controls["bottom_buttons_panel"] = QtWidgets.QWidget()
|
||
self.controls["bottom_buttons_panel"].setObjectName("bottomButtonsPanel")
|
||
|
||
# 删除所有按钮
|
||
self.buttons["remove_all"] = QtWidgets.QPushButton(get_text("remove_all", "删除所有"))
|
||
self.buttons["remove_all"].setObjectName("removeAllButton")
|
||
self.buttons["remove_all"].setIcon(ui_utils.load_icon("delete.png"))
|
||
|
||
# 导入骨骼按钮
|
||
self.buttons["import_skeleton"] = QtWidgets.QPushButton(get_text("import_skeleton", "导入骨骼"))
|
||
self.buttons["import_skeleton"].setObjectName("importSkeletonButton")
|
||
self.buttons["import_skeleton"].setIcon(ui_utils.load_icon("HIKCharacterToolSkeleton.png"))
|
||
|
||
# 生成绑定按钮
|
||
self.buttons["build_rigging"] = QtWidgets.QPushButton(get_text("build_rigging", "生成绑定"))
|
||
self.buttons["build_rigging"].setObjectName("buildRiggingButton")
|
||
self.buttons["build_rigging"].setIcon(ui_utils.load_icon("HIKcreateControlRig.png"))
|
||
|
||
#========================================= LAYOUT =======================================
|
||
def create_layouts(self):
|
||
"""
|
||
创建绑定系统UI布局
|
||
组织控件的排列和层次结构
|
||
"""
|
||
# 创建主布局
|
||
main_layout = QtWidgets.QVBoxLayout(self.main_widget)
|
||
main_layout.setContentsMargins(5, 5, 5, 5)
|
||
main_layout.setSpacing(5)
|
||
|
||
# 添加标题
|
||
main_layout.addWidget(self.controls["title_label"])
|
||
|
||
# 添加主分割器到主布局
|
||
main_layout.addWidget(self.splitters["main_splitter"])
|
||
|
||
# 1. Presets 面板布局
|
||
presets_layout = QtWidgets.QVBoxLayout(self.controls["presets_panel"])
|
||
presets_layout.setContentsMargins(0, 0, 0, 0)
|
||
presets_layout.setSpacing(5)
|
||
|
||
# Presets 组布局
|
||
presets_group_layout = QtWidgets.QVBoxLayout(self.controls["presets_group"])
|
||
presets_group_layout.setContentsMargins(5, 5, 5, 5)
|
||
presets_group_layout.setSpacing(5)
|
||
|
||
# 添加预设滚动区域
|
||
presets_group_layout.addWidget(self.controls["presets_scroll_area"])
|
||
|
||
# 添加预设底部滑块布局
|
||
presets_group_layout.addLayout(self.controls["presets_slider_layout"])
|
||
|
||
# 添加Presets组到Presets面板
|
||
presets_layout.addWidget(self.controls["presets_group"])
|
||
|
||
# 2. Assets 面板布局
|
||
assets_layout = QtWidgets.QVBoxLayout(self.controls["assets_panel"])
|
||
assets_layout.setContentsMargins(0, 0, 0, 0)
|
||
assets_layout.setSpacing(5)
|
||
|
||
# Assets 组布局
|
||
assets_group_layout = QtWidgets.QGridLayout(self.controls["assets_group"])
|
||
assets_group_layout.setContentsMargins(5, 5, 5, 5)
|
||
assets_group_layout.setSpacing(5)
|
||
|
||
# 添加项目路径标签和输入框
|
||
assets_group_layout.addWidget(self.controls["project_path_label"], 0, 0)
|
||
assets_group_layout.addWidget(self.controls["project_path_input"], 0, 1)
|
||
assets_group_layout.addWidget(self.buttons["browse_path"], 0, 2)
|
||
|
||
# 添加Presets DNA标签和输入框
|
||
assets_group_layout.addWidget(self.controls["presets_dna_label"], 1, 0)
|
||
assets_group_layout.addWidget(self.controls["presets_dna_input"], 1, 1)
|
||
assets_group_layout.addWidget(self.buttons["browse_dna"], 1, 2)
|
||
|
||
# 添加Assets组到Assets面板
|
||
assets_layout.addWidget(self.controls["assets_group"])
|
||
|
||
# 3. Descriptor 面板布局
|
||
descriptor_layout = QtWidgets.QVBoxLayout(self.controls["descriptor_panel"])
|
||
descriptor_layout.setContentsMargins(0, 0, 0, 0)
|
||
descriptor_layout.setSpacing(5)
|
||
|
||
# Descriptor 组布局（优化：紧凑表单式，参考主流DCC风格）
|
||
descriptor_group_layout = QtWidgets.QGridLayout(self.controls["descriptor_group"])
|
||
descriptor_group_layout.setContentsMargins(16, 8, 16, 8) # 左右加大
|
||
descriptor_group_layout.setHorizontalSpacing(48) # 增大左右间距
|
||
descriptor_group_layout.setVerticalSpacing(10)
|
||
|
||
# 统一控件高度和字体
|
||
for key in [
|
||
"name_input", "archetype_combo", "gender_combo", "age_spinner", "lod_count_spinner",
|
||
"translation_unit_combo", "rotation_unit_combo", "coordinate_system_combo"
|
||
]:
|
||
self.controls[key].setFixedHeight(24)
|
||
self.controls[key].setStyleSheet("font-size: 13px;")
|
||
for key in [
|
||
"name_label", "archetype_label", "gender_label", "age_label", "lod_count_label",
|
||
"translation_unit_label", "rotation_unit_label", "coordinate_system_label"
|
||
]:
|
||
self.controls[key].setStyleSheet("font-size: 13px;")
|
||
self.controls[key].setAlignment(QtCore.Qt.AlignRight | QtCore.Qt.AlignVCenter)
|
||
|
||
# 设置每列最小宽度/拉伸，保证左右均衡
|
||
descriptor_group_layout.setColumnMinimumWidth(0, 80) # 左标签
|
||
descriptor_group_layout.setColumnMinimumWidth(1, 180) # 左输入
|
||
descriptor_group_layout.setColumnMinimumWidth(2, 80) # 右标签
|
||
descriptor_group_layout.setColumnMinimumWidth(3, 180) # 右输入
|
||
descriptor_group_layout.setColumnStretch(1, 2)
|
||
descriptor_group_layout.setColumnStretch(3, 2)
|
||
|
||
# 第一行：名称
|
||
descriptor_group_layout.addWidget(self.controls["name_label"], 0, 0)
|
||
descriptor_group_layout.addWidget(self.controls["name_input"], 0, 1, 1, 3)
|
||
|
||
# 第二行：原型、性别
|
||
descriptor_group_layout.addWidget(self.controls["archetype_label"], 1, 0)
|
||
descriptor_group_layout.addWidget(self.controls["archetype_combo"], 1, 1)
|
||
descriptor_group_layout.addWidget(self.controls["gender_label"], 1, 2)
|
||
descriptor_group_layout.addWidget(self.controls["gender_combo"], 1, 3)
|
||
|
||
# 第三行：年龄、LOD数量
|
||
descriptor_group_layout.addWidget(self.controls["age_label"], 2, 0)
|
||
descriptor_group_layout.addWidget(self.controls["age_spinner"], 2, 1)
|
||
descriptor_group_layout.addWidget(self.controls["lod_count_label"], 2, 2)
|
||
descriptor_group_layout.addWidget(self.controls["lod_count_spinner"], 2, 3)
|
||
|
||
# 第四行：平移单位、旋转单位
|
||
descriptor_group_layout.addWidget(self.controls["translation_unit_label"], 3, 0)
|
||
descriptor_group_layout.addWidget(self.controls["translation_unit_combo"], 3, 1)
|
||
descriptor_group_layout.addWidget(self.controls["rotation_unit_label"], 3, 2)
|
||
descriptor_group_layout.addWidget(self.controls["rotation_unit_combo"], 3, 3)
|
||
|
||
# 第五行：坐标系统
|
||
descriptor_group_layout.addWidget(self.controls["coordinate_system_label"], 4, 0)
|
||
descriptor_group_layout.addWidget(self.controls["coordinate_system_combo"], 4, 1)
|
||
# 右侧空白占位，保持对齐美观
|
||
descriptor_group_layout.addItem(QtWidgets.QSpacerItem(0, 0, QtWidgets.QSizePolicy.Expanding, QtWidgets.QSizePolicy.Minimum), 4, 2, 1, 2)
|
||
|
||
# 添加Descriptor组到Descriptor面板
|
||
descriptor_layout.addWidget(self.controls["descriptor_group"])
|
||
|
||
# 添加各个面板到主分割器
|
||
self.splitters["main_splitter"].addWidget(self.controls["presets_panel"])
|
||
self.splitters["main_splitter"].addWidget(self.controls["assets_panel"])
|
||
self.splitters["main_splitter"].addWidget(self.controls["descriptor_panel"])
|
||
|
||
# 底部按钮区域布局
|
||
bottom_buttons_layout = QtWidgets.QHBoxLayout(self.controls["bottom_buttons_panel"])
|
||
bottom_buttons_layout.setContentsMargins(5, 5, 5, 5)
|
||
bottom_buttons_layout.setSpacing(10)
|
||
|
||
# 添加底部按钮（均等宽度）
|
||
for btn in [self.buttons["remove_all"], self.buttons["import_skeleton"], self.buttons["build_rigging"]]:
|
||
btn.setMinimumWidth(0)
|
||
btn.setMaximumWidth(16777215)
|
||
btn.setSizePolicy(QtWidgets.QSizePolicy.Expanding, QtWidgets.QSizePolicy.Fixed)
|
||
bottom_buttons_layout.addWidget(btn)
|
||
|
||
# 添加底部按钮区域到主布局
|
||
main_layout.addWidget(self.controls["bottom_buttons_panel"])
|
||
|
||
# 设置分割器初始大小
|
||
self.splitters["main_splitter"].setSizes([400, 100, 200])
|
||
|
||
# 设置分割器的伸缩因子
|
||
for i in range(self.splitters["main_splitter"].count()):
|
||
self.splitters["main_splitter"].setStretchFactor(i, 1)
|
||
|
||
#======================================= FUNCTIONS ======================================
|
||
def create_connections(self):
|
||
"""
|
||
创建信号连接，设置UI控件的交互行为（风格参考definition.py）
|
||
"""
|
||
# 主要功能按钮直接连接
|
||
if "add_joint_btn" in self.buttons:
|
||
self.buttons["add_joint_btn"].clicked.connect(utils_rigging.add_joint)
|
||
if "remove_joint_btn" in self.buttons:
|
||
self.buttons["remove_joint_btn"].clicked.connect(utils_rigging.remove_joint)
|
||
if "duplicate_joint_btn" in self.buttons:
|
||
self.buttons["duplicate_joint_btn"].clicked.connect(utils_rigging.duplicate_joint)
|
||
if "add_controller_btn" in self.buttons:
|
||
self.buttons["add_controller_btn"].clicked.connect(utils_rigging.add_controller)
|
||
if "remove_controller_btn" in self.buttons:
|
||
self.buttons["remove_controller_btn"].clicked.connect(utils_rigging.remove_controller)
|
||
if "duplicate_controller_btn" in self.buttons:
|
||
self.buttons["duplicate_controller_btn"].clicked.connect(utils_rigging.duplicate_controller)
|
||
if "import_dna_btn" in self.buttons:
|
||
self.buttons["import_dna_btn"].clicked.connect(utils_rigging.import_dna)
|
||
if "export_dna_btn" in self.buttons:
|
||
self.buttons["export_dna_btn"].clicked.connect(utils_rigging.export_dna)
|
||
if "calibrate_dna_btn" in self.buttons:
|
||
self.buttons["calibrate_dna_btn"].clicked.connect(utils_rigging.calibrate_dna)
|
||
|
||
# 其它已创建按钮
|
||
if "browse_path" in self.buttons:
|
||
self.buttons["browse_path"].clicked.connect(lambda: utils_rigging.browse_file(self, "项目路径", self.controls["project_path_input"]))
|
||
if "browse_dna" in self.buttons:
|
||
self.buttons["browse_dna"].clicked.connect(lambda: utils_rigging.browse_file(self, "DNA文件", self.controls["presets_dna_input"], "dna"))
|
||
if "export_presets" in self.buttons:
|
||
self.buttons["export_presets"].clicked.connect(utils_rigging.export_presets)
|
||
if "import_presets" in self.buttons:
|
||
self.buttons["import_presets"].clicked.connect(utils_rigging.import_presets) |