Update
This commit is contained in:
212
2023/scripts/rigging_tools/skin_api/ui.py
Normal file
212
2023/scripts/rigging_tools/skin_api/ui.py
Normal file
@@ -0,0 +1,212 @@
|
||||
#!/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)
|
||||
Reference in New Issue
Block a user