213 lines
6.7 KiB
Python
213 lines
6.7 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
Skin API UI Functions
|
|
提供权重导出、导入和解绑的用户界面函数
|
|
"""
|
|
|
|
# 确保 Maya 环境已初始化
|
|
try:
|
|
import maya.standalone
|
|
maya.standalone.initialize()
|
|
except:
|
|
pass
|
|
|
|
try:
|
|
import pymel.core as pm
|
|
except ImportError:
|
|
pm = None
|
|
|
|
import maya.cmds as cmds
|
|
import maya.mel as mel
|
|
|
|
try:
|
|
from . import apiVtxAttribs
|
|
except ImportError:
|
|
import rigging_tools.skin_api.apiVtxAttribs as apiVtxAttribs
|
|
|
|
|
|
def WeightExport():
|
|
"""
|
|
导出选中物体的蒙皮权重
|
|
Export skin weights for selected objects
|
|
"""
|
|
try:
|
|
# 检查是否有选中物体
|
|
if pm:
|
|
selectedNodes = pm.ls(sl=True)
|
|
if not selectedNodes:
|
|
pm.warning("请选择至少一个蒙皮物体 / Please select at least one skinned object")
|
|
return
|
|
else:
|
|
selectedNodes = cmds.ls(sl=True)
|
|
if not selectedNodes:
|
|
cmds.warning("请选择至少一个蒙皮物体 / Please select at least one skinned object")
|
|
return
|
|
|
|
# 使用 ApiVtxAttribs 类导出权重
|
|
api = apiVtxAttribs.ApiVtxAttribs()
|
|
msg = api.exportSkinWeights(selected=True, saveJointInfo=True)
|
|
|
|
if msg:
|
|
print(msg)
|
|
if pm:
|
|
pm.confirmDialog(
|
|
title="Export Success",
|
|
message="Skin weights exported successfully!",
|
|
button=["OK"]
|
|
)
|
|
else:
|
|
cmds.confirmDialog(
|
|
title="Export Success",
|
|
message="Skin weights exported successfully!",
|
|
button=["OK"]
|
|
)
|
|
else:
|
|
print("Export cancelled")
|
|
|
|
except Exception as e:
|
|
error_msg = f"Failed to export skin weights: {str(e)}"
|
|
if pm:
|
|
pm.error(error_msg)
|
|
else:
|
|
cmds.error(error_msg)
|
|
|
|
|
|
def WeightImport():
|
|
"""
|
|
导入蒙皮权重到选中物体或场景中匹配的物体
|
|
Import skin weights to selected or matching objects in scene
|
|
"""
|
|
try:
|
|
# 检查是否有选中物体
|
|
if pm:
|
|
selectedNodes = pm.ls(sl=True)
|
|
else:
|
|
selectedNodes = cmds.ls(sl=True)
|
|
useSelection = len(selectedNodes) > 0
|
|
|
|
print(f"Import mode: {'selected objects' if useSelection else 'all matching objects'}")
|
|
|
|
# 使用 ApiVtxAttribs 类导入权重
|
|
api = apiVtxAttribs.ApiVtxAttribs()
|
|
msg = api.importSkinWeights(selected=useSelection, stripJointNamespaces=False, addNewToHierarchy=True)
|
|
|
|
print(f"Import result: {repr(msg)}")
|
|
|
|
if msg and msg != False:
|
|
print(msg)
|
|
if "No valid objects found" in str(msg):
|
|
warning_msg = "No valid objects found in scene!\nMake sure:\n1. Object names match\n2. Vertex counts match\n3. Objects exist in scene"
|
|
if pm:
|
|
pm.warning(warning_msg)
|
|
pm.confirmDialog(
|
|
title="Import Failed",
|
|
message=warning_msg,
|
|
button=["OK"]
|
|
)
|
|
else:
|
|
cmds.warning(warning_msg)
|
|
cmds.confirmDialog(
|
|
title="Import Failed",
|
|
message=warning_msg,
|
|
button=["OK"]
|
|
)
|
|
elif "Could not find" in str(msg):
|
|
if pm:
|
|
pm.warning(msg)
|
|
else:
|
|
cmds.warning(msg)
|
|
else:
|
|
# 成功导入
|
|
if pm:
|
|
pm.confirmDialog(
|
|
title="Import Complete",
|
|
message="Skin weights imported successfully!",
|
|
button=["OK"]
|
|
)
|
|
else:
|
|
cmds.confirmDialog(
|
|
title="Import Complete",
|
|
message="Skin weights imported successfully!",
|
|
button=["OK"]
|
|
)
|
|
else:
|
|
print("Import cancelled or no file selected")
|
|
|
|
except Exception as e:
|
|
import traceback
|
|
error_msg = f"Failed to import skin weights: {str(e)}\n{traceback.format_exc()}"
|
|
print(error_msg)
|
|
if pm:
|
|
pm.error(error_msg)
|
|
else:
|
|
cmds.error(error_msg)
|
|
|
|
|
|
def UnbindSkin():
|
|
"""
|
|
解绑选中物体的蒙皮
|
|
Unbind skin from selected objects
|
|
"""
|
|
try:
|
|
# 获取选中的物体
|
|
if pm:
|
|
selectedNodes = pm.ls(sl=True)
|
|
if not selectedNodes:
|
|
pm.warning("请选择至少一个物体 / Please select at least one object")
|
|
return
|
|
else:
|
|
selectedNodes = cmds.ls(sl=True)
|
|
if not selectedNodes:
|
|
cmds.warning("请选择至少一个物体 / Please select at least one object")
|
|
return
|
|
|
|
# 确认对话框
|
|
if pm:
|
|
result = pm.confirmDialog(
|
|
title="Unbind Skin",
|
|
message=f"确定要解绑 {len(selectedNodes)} 个物体的蒙皮吗？\nAre you sure you want to unbind skin from {len(selectedNodes)} object(s)?",
|
|
button=["Yes", "No"],
|
|
defaultButton="Yes",
|
|
cancelButton="No",
|
|
dismissString="No"
|
|
)
|
|
else:
|
|
result = cmds.confirmDialog(
|
|
title="Unbind Skin",
|
|
message=f"确定要解绑 {len(selectedNodes)} 个物体的蒙皮吗？\nAre you sure you want to unbind skin from {len(selectedNodes)} object(s)?",
|
|
button=["Yes", "No"],
|
|
defaultButton="Yes",
|
|
cancelButton="No",
|
|
dismissString="No"
|
|
)
|
|
|
|
if result != "Yes":
|
|
print("Unbind cancelled")
|
|
return
|
|
|
|
# 使用 MEL 命令解绑蒙皮
|
|
mel.eval('doDetachSkin "2" { "1","1" };')
|
|
print("已取消蒙皮！/ Skin unbound!")
|
|
|
|
if pm:
|
|
pm.confirmDialog(
|
|
title="Unbind Complete",
|
|
message="Skin unbound successfully!",
|
|
button=["OK"]
|
|
)
|
|
else:
|
|
cmds.confirmDialog(
|
|
title="Unbind Complete",
|
|
message="Skin unbound successfully!",
|
|
button=["OK"]
|
|
)
|
|
|
|
except Exception as e:
|
|
error_msg = f"Failed to unbind skin: {str(e)}"
|
|
if pm:
|
|
pm.error(error_msg)
|
|
else:
|
|
cmds.error(error_msg)
|