#!/usr/bin/env python # -*- coding: utf-8 -*- """ Toolbar UI Module for Plugin 工具栏UI模块 - 负责显示工具栏界面和基础操作 基本功能： - 加载预设 - 保存预设 - 导入DNA - 导出DNA - 创建RL4节点（用于切换DNA编辑的状态） - 删除RL4节点（用于切换DNA编辑的状态） """ #========================================= IMPORT ========================================= from scripts.ui.Qt import QtWidgets, QtCore, QtGui from scripts.ui.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_toolbar #========================================== 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 TEXT = localization.TEXT class ToolbarUI(ui_utils.BaseUI): """ 工具栏UI类 - 负责显示工具栏界面和基础操作 继承自BaseUI类，实现工具栏相关的UI功能 """ #========================================== INIT ======================================== def __init__(self, parent=None): """ 初始化工具栏UI 创建主控件和布局，并连接信号和槽 """ super().__init__(parent) # 创建主控件 self.main_widget = QtWidgets.QWidget() self.main_widget.setObjectName("toolbarMainWidget") # 初始化UI self.create_widgets() self.create_layouts() self.create_connections() #========================================= WIDGET ======================================= def create_widgets(self): """ 创建工具栏UI控件 包括按钮、标签、输入框等 """ # 创建顶部工具栏 self.controls["toolbar_frame"] = QtWidgets.QFrame() self.controls["toolbar_frame"].setObjectName("toolbarFrame") self.controls["toolbar_frame"].setFrameShape(QtWidgets.QFrame.StyledPanel) self.controls["toolbar_frame"].setFrameShadow(QtWidgets.QFrame.Raised) # 创建工具栏按钮 # DNA相关按钮 self.buttons["Save DNA"] = self._create_tool_button("保存DNA", "save.png") self.buttons["Open DNA"] = self._create_tool_button("打开DNA", "open.png") # RL4节点相关按钮 self.buttons["Create RL4 node"] = self._create_tool_button("创建RL4节点", "create_rl4_node.png") self.buttons["Delete RL4 node"] = self._create_tool_button("删除RL4节点", "delete_rl4_node.png") # 蒙皮相关按钮 self.buttons["Import skin"] = self._create_tool_button("导入蒙皮", "import_skin.png") self.buttons["Export skin"] = self._create_tool_button("导出蒙皮", "export_skin.png") self.buttons["Copy skin"] = self._create_tool_button("复制蒙皮", "copy_skin.png") # 翻译按钮 self.buttons["Translate"] = self._create_tool_button("切换语言", "translate.png") # 帮助按钮 self.buttons["Help"] = self._create_tool_button("帮助", "help.png") def _create_tool_button(self, tooltip, icon_name, checkable=False): """ 创建工具栏按钮 参数： tooltip: 按钮提示文字 icon_name: 图标文件名，不包含路径 checkable: 是否可选中 """ button = QtWidgets.QPushButton() button.setToolTip(TEXT(tooltip, tooltip)) # 构建图标完整路径 icon_path = os.path.join(ICONS_PATH, icon_name) # 尝试设置图标，如果图标文件存在 if os.path.exists(icon_path): button.setIcon(QtGui.QIcon(icon_path)) button.setIconSize(QtCore.QSize(24, 24)) else: # 如果图标不存在，使用文字 button.setText(TEXT(tooltip, tooltip)) button.setObjectName(f"{tooltip.replace(' ', '_').lower()}_button") button.setFixedSize(32, 32) button.setCheckable(checkable) return button #========================================= LAYOUT ======================================= def create_layouts(self): """ 创建工具栏UI布局 组织控件的排列和层次结构 """ # 主布局 self.layouts["main_layout"] = QtWidgets.QVBoxLayout(self.main_widget) self.layouts["main_layout"].setContentsMargins(0, 0, 0, 0) self.layouts["main_layout"].setSpacing(0) # 添加工具栏框架 self.layouts["main_layout"].addWidget(self.controls["toolbar_frame"]) # 工具栏框架布局 self.layouts["toolbar_layout"] = QtWidgets.QVBoxLayout(self.controls["toolbar_frame"]) self.layouts["toolbar_layout"].setContentsMargins(5, 5, 5, 5) self.layouts["toolbar_layout"].setSpacing(2) # 按钮布局 self.layouts["toolbar"] = QtWidgets.QHBoxLayout() self.layouts["toolbar"].setSpacing(2) self.layouts["toolbar"].addWidget(self.buttons["Save DNA"]) self.layouts["toolbar"].addWidget(self.buttons["Open DNA"]) self.layouts["toolbar"].addWidget(self._create_separator()) self.layouts["toolbar"].addWidget(self.buttons["Create RL4 node"]) self.layouts["toolbar"].addWidget(self.buttons["Delete RL4 node"]) self.layouts["toolbar"].addWidget(self._create_separator()) self.layouts["toolbar"].addWidget(self.buttons["Import skin"]) self.layouts["toolbar"].addWidget(self.buttons["Export skin"]) self.layouts["toolbar"].addWidget(self.buttons["Copy skin"]) self.layouts["toolbar"].addWidget(self._create_separator()) self.layouts["toolbar"].addWidget(self.buttons["Translate"]) self.layouts["toolbar"].addWidget(self._create_separator()) self.layouts["toolbar"].addWidget(self.buttons["Help"]) self.layouts["toolbar"].addStretch() # 添加行到工具栏布局 self.layouts["toolbar_layout"].addLayout(self.layouts["toolbar"]) #======================================= CONNECTION ===================================== def create_connections(self): """ 连接信号和槽 设置UI控件的交互行为 """ # DNA相关按钮连接 self.buttons["Save DNA"].clicked.connect(utils_toolbar.save_dna) self.buttons["Open DNA"].clicked.connect(utils_toolbar.open_dna) # RL4节点相关按钮连接 self.buttons["Create RL4 node"].clicked.connect(utils_toolbar.create_rl4_node) self.buttons["Delete RL4 node"].clicked.connect(utils_toolbar.delete_rl4_node) # 蒙皮相关按钮连接 self.buttons["Import skin"].clicked.connect(utils_toolbar.import_skin) self.buttons["Export skin"].clicked.connect(utils_toolbar.export_skin) self.buttons["Copy skin"].clicked.connect(utils_toolbar.copy_skin) # 翻译按钮连接 self.buttons["Translate"].clicked.connect(utils_toolbar.toggle_language) # 帮助按钮连接 self.buttons["Help"].clicked.connect(utils_toolbar.show_help) #======================================== FUNCTIONS ====================================== def _create_separator(self): """ 创建分隔线 """ separator = QtWidgets.QFrame() separator.setFrameShape(QtWidgets.QFrame.VLine) separator.setFrameShadow(QtWidgets.QFrame.Sunken) separator.setFixedWidth(2) separator.setFixedHeight(24) return separator