Update

2025/scripts/animation_tools/mgpicker/__init__.py

@@ -0,0 +1,231 @@
+#!/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']