232 lines
7.1 KiB
Python
232 lines
7.1 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
|
||
"""
|
||
MG-Picker Studio Module
|
||
Animation picker tool for Maya
|
||
"""
|
||
|
||
import os
|
||
import maya.cmds as cmds
|
||
import maya.mel as mel
|
||
|
||
|
||
def get_maya_version():
|
||
"""
|
||
获取当前 Maya 版本号
|
||
返回格式: '2023', '2024' 等
|
||
"""
|
||
try:
|
||
maya_version = cmds.about(version=True)
|
||
# Maya 版本格式可能是 "2023" 或 "2023.1" 等，取主版本号
|
||
return maya_version.split('.')[0]
|
||
except Exception as e:
|
||
print("Failed to get Maya version: " + str(e))
|
||
return None
|
||
|
||
|
||
def find_compatible_plugin(plugin_dir, maya_version):
|
||
"""
|
||
查找兼容的 MGPicker 插件
|
||
优先使用完全匹配的版本，如果没有则向下查找最接近的版本
|
||
|
||
Args:
|
||
plugin_dir: 插件目录路径
|
||
maya_version: Maya 版本号字符串，如 '2023'
|
||
|
||
Returns:
|
||
插件名称（不含扩展名），如果找不到则返回 None
|
||
"""
|
||
if not maya_version:
|
||
return None
|
||
|
||
try:
|
||
maya_ver_int = int(maya_version)
|
||
except ValueError:
|
||
print("Invalid Maya version format: " + str(maya_version))
|
||
return None
|
||
|
||
# 列出所有可用的插件版本
|
||
available_versions = []
|
||
if os.path.exists(plugin_dir):
|
||
for filename in os.listdir(plugin_dir):
|
||
if filename.startswith("MGPicker_") and filename.endswith(".mll"):
|
||
# 提取版本号，例如从 "MGPicker_2023x64.mll" 提取 "2023"
|
||
try:
|
||
version_str = filename.replace("MGPicker_", "").replace("x64.mll", "")
|
||
ver_int = int(version_str)
|
||
available_versions.append((ver_int, filename.replace(".mll", "")))
|
||
except ValueError:
|
||
continue
|
||
|
||
if not available_versions:
|
||
return None
|
||
|
||
# 按版本号排序
|
||
available_versions.sort(reverse=True)
|
||
|
||
# 首先尝试完全匹配
|
||
for ver, plugin_name in available_versions:
|
||
if ver == maya_ver_int:
|
||
print("Found exact match plugin for Maya " + str(maya_version) + ": " + plugin_name)
|
||
return plugin_name
|
||
|
||
# 如果没有完全匹配，使用小于等于当前版本的最高版本
|
||
for ver, plugin_name in available_versions:
|
||
if ver <= maya_ver_int:
|
||
print("Using compatible plugin for Maya " + str(maya_version) + ": " + plugin_name + " (version " + str(ver) + ")")
|
||
return plugin_name
|
||
|
||
# 如果当前版本比所有可用版本都旧，使用最旧的版本
|
||
oldest_ver, oldest_plugin = available_versions[-1]
|
||
print("Warning: Maya " + str(maya_version) + " is older than available plugins. Using oldest: " + oldest_plugin + " (version " + str(oldest_ver) + ")")
|
||
return oldest_plugin
|
||
|
||
|
||
def get_mgpicker_path():
|
||
"""Get MG-Picker Studio installation path"""
|
||
current_dir = os.path.dirname(os.path.abspath(__file__))
|
||
return current_dir.replace("\\", "/")
|
||
|
||
|
||
def load_plugin():
|
||
"""
|
||
Load MGPicker plugin
|
||
自动检测 Maya 版本并加载对应的插件
|
||
"""
|
||
# 获取当前 Maya 版本
|
||
maya_version = get_maya_version()
|
||
if not maya_version:
|
||
print("Warning: Could not determine Maya version")
|
||
return False
|
||
|
||
print("Detected Maya version: " + str(maya_version))
|
||
|
||
# 查找兼容的插件
|
||
mgpicker_path = get_mgpicker_path()
|
||
plugin_dir = os.path.join(mgpicker_path, "MGPicker_Program", "Plug-ins")
|
||
plugin_name = find_compatible_plugin(plugin_dir, maya_version)
|
||
|
||
if not plugin_name:
|
||
print("Error: No compatible MGPicker plugin found for Maya " + str(maya_version))
|
||
return False
|
||
|
||
# Check if plugin is already loaded
|
||
try:
|
||
if cmds.pluginInfo(plugin_name, query=True, loaded=True):
|
||
print("MGPicker plugin already loaded: " + plugin_name)
|
||
return True
|
||
except RuntimeError:
|
||
# Plugin not registered yet, will try to load
|
||
pass
|
||
|
||
# Get the plugin path
|
||
plugin_path = os.path.join(plugin_dir, plugin_name + ".mll")
|
||
plugin_path = plugin_path.replace("\\", "/")
|
||
|
||
if not os.path.exists(plugin_path):
|
||
print("Error: Plugin file not found: " + plugin_path)
|
||
return False
|
||
|
||
try:
|
||
cmds.loadPlugin(plugin_path)
|
||
print("MGPicker plugin loaded successfully: " + plugin_path)
|
||
return True
|
||
except Exception as e:
|
||
print("Failed to load MGPicker plugin: " + str(e))
|
||
import traceback
|
||
traceback.print_exc()
|
||
return False
|
||
|
||
|
||
def is_window_open():
|
||
"""
|
||
检查 MGPicker 窗口是否已经打开
|
||
|
||
Returns:
|
||
bool: 如果窗口已打开返回 True
|
||
"""
|
||
# MGPicker 的主窗口名称
|
||
window_names = [
|
||
'MG_PickerStudio_Window',
|
||
'MGPickerStudioWindow',
|
||
'mgPickerStudioWindow'
|
||
]
|
||
|
||
for window_name in window_names:
|
||
if cmds.window(window_name, exists=True):
|
||
return True
|
||
|
||
return False
|
||
|
||
|
||
def show_existing_window():
|
||
"""
|
||
显示已存在的 MGPicker 窗口
|
||
|
||
Returns:
|
||
bool: 如果成功显示窗口返回 True
|
||
"""
|
||
window_names = [
|
||
'MG_PickerStudio_Window',
|
||
'MGPickerStudioWindow',
|
||
'mgPickerStudioWindow'
|
||
]
|
||
|
||
for window_name in window_names:
|
||
if cmds.window(window_name, exists=True):
|
||
try:
|
||
# 尝试显示窗口
|
||
cmds.showWindow(window_name)
|
||
print("MGPicker window already open, bringing to front")
|
||
return True
|
||
except:
|
||
pass
|
||
|
||
return False
|
||
|
||
|
||
def start(mode=1, force_new=False):
|
||
"""
|
||
Start MG-Picker Studio
|
||
|
||
Args:
|
||
mode (int): 0 for designer mode, 1 for animator mode (default)
|
||
force_new (bool): 如果为 True，强制创建新窗口（不推荐）
|
||
"""
|
||
# 检查窗口是否已经打开
|
||
if not force_new and is_window_open():
|
||
if show_existing_window():
|
||
return
|
||
else:
|
||
print("Warning: MGPicker window exists but cannot be shown. Will try to create new window.")
|
||
|
||
# Load plugin first
|
||
if not load_plugin():
|
||
cmds.warning("Failed to load MGPicker plugin")
|
||
return
|
||
|
||
# Get MGPicker path
|
||
mgpicker_path = get_mgpicker_path()
|
||
program_path = os.path.join(mgpicker_path, "MGPicker_Program")
|
||
|
||
# Set global variables
|
||
mel.eval('global string $MGPicker_ProgramPath = "' + mgpicker_path + '/"')
|
||
mel.eval('global string $MGPicker_ProgramFolder = "MGPicker_Program"')
|
||
mel.eval('global string $MGPicker_ScriptEntry = "MG_PickerStudio"')
|
||
|
||
# Source the main MEL script
|
||
main_mel = os.path.join(program_path, "MGPicker_WrittenByMiguel.mel").replace("\\", "/")
|
||
|
||
try:
|
||
mel.eval('source "' + main_mel + '"')
|
||
# Open MGPicker window with mode parameter
|
||
# mode 0: designer mode, mode 1: animator mode
|
||
mel.eval('MG_PickerStudio ' + str(mode))
|
||
print("MG-Picker Studio started in " + ("designer" if mode == 0 else "animator") + " mode")
|
||
except Exception as e:
|
||
cmds.warning("Failed to start MG-Picker Studio: " + str(e))
|
||
|
||
|
||
__all__ = ['start', 'load_plugin']
|