476 lines
14 KiB
Python
476 lines
14 KiB
Python
|
#!/usr/bin/env python
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
|
|||
|
|
import maya.cmds as cmds
|
|||
|
|
import maya.mel as mel
|
|||
|
|
from Core import GetBlendShapeNodes, GetWrapNodes, FilterSingleBlendShapeDeformerTargets
|
|||
|
|
|
|||
|
|
def wrap_convert_blend():
|
|||
|
|
"""
|
|||
|
|
将选中的模型的混合变形转换为新的混合变形
|
|||
|
|
"""
|
|||
|
|
# 获取选中的对象
|
|||
|
|
sel = cmds.ls(selection=True)
|
|||
|
|
if len(sel) != 2:
|
|||
|
|
cmds.warning("请选择两个模型对象")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# 获取混合变形节点
|
|||
|
|
bs_nodes = GetBlendShapeNodes(sel[1])
|
|||
|
|
if not bs_nodes:
|
|||
|
|
cmds.warning("目标模型不包含混合变形，请确保选择正确的源和目标模型")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# 获取混合变形目标列表
|
|||
|
|
bs_target_attrs = cmds.listAttr(f"{bs_nodes[0]}.w", multi=True) or []
|
|||
|
|
if not bs_target_attrs:
|
|||
|
|
cmds.warning("混合变形目标列表为空")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# 如果选择了wrap方法
|
|||
|
|
if cmds.checkBox("wrapMethod", query=True, value=True):
|
|||
|
|
cmds.select(sel[0], sel[1])
|
|||
|
|
mel.eval('CreateWrap')
|
|||
|
|
|
|||
|
|
# 创建临时组
|
|||
|
|
if cmds.objExists("WrapConvertTemp"):
|
|||
|
|
cmds.delete("WrapConvertTemp")
|
|||
|
|
temp_group = cmds.group(empty=True, name="WrapConvertTemp")
|
|||
|
|
|
|||
|
|
rb_mesh = []
|
|||
|
|
# 复制每个混合变形目标
|
|||
|
|
for bs_attr in bs_target_attrs:
|
|||
|
|
bs_weight = cmds.getAttr(f"{bs_nodes[0]}.{bs_attr}")
|
|||
|
|
is_connected = cmds.connectionInfo(f"{bs_nodes[0]}.{bs_attr}", isDestination=True)
|
|||
|
|
|
|||
|
|
if bs_weight != 1 and not is_connected:
|
|||
|
|
# 设置权重为1并复制模型
|
|||
|
|
cmds.setAttr(f"{bs_nodes[0]}.{bs_attr}", 1)
|
|||
|
|
copy_mesh = cmds.duplicate(sel[0])[0]
|
|||
|
|
cmds.parent(copy_mesh, temp_group)
|
|||
|
|
copy_mesh = cmds.rename(copy_mesh, bs_attr)
|
|||
|
|
rb_mesh.append(f":WrapConvertTemp|:{bs_attr}")
|
|||
|
|
cmds.setAttr(f"{bs_nodes[0]}.{bs_attr}", 0)
|
|||
|
|
|
|||
|
|
# 处理wrap节点
|
|||
|
|
if cmds.checkBox("wrapMethod", query=True, value=True):
|
|||
|
|
wrap_nodes = GetWrapNodes(sel[0])
|
|||
|
|
base_nodes = cmds.ls(f"{sel[1]}Bas*")
|
|||
|
|
if wrap_nodes or base_nodes:
|
|||
|
|
cmds.delete(wrap_nodes + base_nodes)
|
|||
|
|
|
|||
|
|
# 创建新的混合变形
|
|||
|
|
if rb_mesh:
|
|||
|
|
cmds.blendShape(rb_mesh, sel[0], frontOfChain=True)
|
|||
|
|
|
|||
|
|
# 处理DeltaMush
|
|||
|
|
if cmds.objExists('deltaMush') and cmds.checkBox("deltaMushTarget", query=True, value=True):
|
|||
|
|
delta_value = cmds.intSliderGrp("deltaIntensity", query=True, value=True)
|
|||
|
|
_process_delta_mush(sel[0], delta_value)
|
|||
|
|
|
|||
|
|
# 清理
|
|||
|
|
cmds.delete("WrapConvertTemp")
|
|||
|
|
print("混合变形处理完成")
|
|||
|
|
|
|||
|
|
def _process_delta_mush(target, delta_value):
|
|||
|
|
"""
|
|||
|
|
处理DeltaMush相关的操作
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
target (str): 目标模型
|
|||
|
|
delta_value (int): DeltaMush强度值
|
|||
|
|
"""
|
|||
|
|
# 创建DeltaMush变形器
|
|||
|
|
dl_names = cmds.deltaMush(
|
|||
|
|
target,
|
|||
|
|
smoothingIterations=delta_value,
|
|||
|
|
smoothingStep=1.0,
|
|||
|
|
pinBorderVertices=1,
|
|||
|
|
envelope=1
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 获取混合变形节点和目标
|
|||
|
|
bs_nodes = GetBlendShapeNodes(target)
|
|||
|
|
bs_target_attrs = cmds.listAttr(f"{bs_nodes[0]}.w", multi=True) or []
|
|||
|
|
|
|||
|
|
# 创建临时组
|
|||
|
|
if cmds.objExists("WrapConvertTemp"):
|
|||
|
|
cmds.delete("WrapConvertTemp")
|
|||
|
|
temp_group = cmds.group(empty=True, name="WrapConvertTemp")
|
|||
|
|
|
|||
|
|
rb_mesh = []
|
|||
|
|
# 处理每个混合变形目标
|
|||
|
|
for bs_attr in bs_target_attrs:
|
|||
|
|
bs_weight = cmds.getAttr(f"{bs_nodes[0]}.{bs_attr}")
|
|||
|
|
if bs_weight != 1:
|
|||
|
|
# 设置权重并创建关键帧
|
|||
|
|
cmds.setAttr(f"{bs_nodes[0]}.{bs_attr}", 0)
|
|||
|
|
cmds.currentTime(-10)
|
|||
|
|
cmds.setAttr(f"{bs_nodes[0]}.{bs_attr}", 1)
|
|||
|
|
cmds.currentTime(0)
|
|||
|
|
|
|||
|
|
# 复制模型并重命名
|
|||
|
|
copy_mesh = cmds.duplicate(target)[0]
|
|||
|
|
cmds.parent(copy_mesh, temp_group)
|
|||
|
|
copy_mesh = cmds.rename(copy_mesh, bs_attr)
|
|||
|
|
rb_mesh.append(f":WrapConvertTemp|:{bs_attr}")
|
|||
|
|
|
|||
|
|
# 重置权重
|
|||
|
|
cmds.setAttr(f"{bs_nodes[0]}.{bs_attr}", 0)
|
|||
|
|
cmds.currentTime(-10)
|
|||
|
|
cmds.currentTime(0)
|
|||
|
|
|
|||
|
|
# 清理并创建新的混合变形
|
|||
|
|
cmds.delete(dl_names + bs_nodes)
|
|||
|
|
if rb_mesh:
|
|||
|
|
cmds.blendShape(rb_mesh, target, frontOfChain=True)
|
|||
|
|
cmds.delete("WrapConvertTemp")
|
|||
|
|
print("DeltaMush优化完成")
|
|||
|
|
|
|||
|
|
def clone_blend_shape_ui():
|
|||
|
|
"""
|
|||
|
|
创建混合变形克隆工具的UI界面
|
|||
|
|
"""
|
|||
|
|
window_name = "wrapChangeBS"
|
|||
|
|
if cmds.window(window_name, exists=True):
|
|||
|
|
cmds.deleteUI(window_name)
|
|||
|
|
|
|||
|
|
# 创建主窗口
|
|||
|
|
window = cmds.window(
|
|||
|
|
window_name,
|
|||
|
|
title="BlendShapeBake",
|
|||
|
|
width=300,
|
|||
|
|
height=300,
|
|||
|
|
menuBar=True
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 创建主布局
|
|||
|
|
main_layout = cmds.columnLayout(
|
|||
|
|
adjustableColumn=1,
|
|||
|
|
columnAttach=["both", 5],
|
|||
|
|
rowSpacing=2,
|
|||
|
|
columnWidth=150
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
cmds.separator(height=10, style="in")
|
|||
|
|
|
|||
|
|
# 创建Wrap方法选项
|
|||
|
|
cmds.checkBox(
|
|||
|
|
"wrapMethod",
|
|||
|
|
label="Use Wrap Method",
|
|||
|
|
value=1,
|
|||
|
|
changeCommand="sparkly_wrap_convert_blend_check_return"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 创建DeltaMush选项（如果可用）
|
|||
|
|
if cmds.pluginInfo("deltaMush", query=True, loaded=True):
|
|||
|
|
cmds.checkBox(
|
|||
|
|
"deltaMushTarget",
|
|||
|
|
label="DeltaMushTarget",
|
|||
|
|
value=0,
|
|||
|
|
changeCommand="sparkly_wrap_convert_blend_check_return"
|
|||
|
|
)
|
|||
|
|
cmds.intSliderGrp(
|
|||
|
|
"deltaIntensity",
|
|||
|
|
field=True,
|
|||
|
|
minValue=0,
|
|||
|
|
maxValue=100,
|
|||
|
|
value=10,
|
|||
|
|
width=50
|
|||
|
|
)
|
|||
|
|
else:
|
|||
|
|
cmds.text(
|
|||
|
|
label="低版本DeltaMush不可用",
|
|||
|
|
font="boldLabelFont"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
cmds.separator(height=10, style="in")
|
|||
|
|
|
|||
|
|
# 创建功能按钮
|
|||
|
|
cmds.button(label="Cloning", command="wrap_convert_blend()")
|
|||
|
|
cmds.button(label="Copy", command="sg_copy_blend()")
|
|||
|
|
cmds.button(label="Mirror", command="mirror_blend()")
|
|||
|
|
|
|||
|
|
cmds.separator(height=10, style="in")
|
|||
|
|
|
|||
|
|
# 调整窗口大小
|
|||
|
|
cmds.window(
|
|||
|
|
window_name,
|
|||
|
|
edit=True,
|
|||
|
|
width=260,
|
|||
|
|
height=85
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
cmds.showWindow()
|
|||
|
|
|
|||
|
|
def sparkly_wrap_convert_blend_check_return():
|
|||
|
|
"""
|
|||
|
|
检查框回调函数（保持为空以保持兼容性）
|
|||
|
|
"""
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
def sg_copy_blend():
|
|||
|
|
"""
|
|||
|
|
复制混合变形目标
|
|||
|
|
"""
|
|||
|
|
# 获取选中的混合变形目标
|
|||
|
|
sel_bs_name_list = mel.eval('getShapeEditorTreeviewSelection(14)')
|
|||
|
|
|
|||
|
|
# 解析目标索引
|
|||
|
|
tmp_a = sel_bs_name_list[0].split('.')
|
|||
|
|
tmp_b = sel_bs_name_list[1].split('.')
|
|||
|
|
|
|||
|
|
target_a = int(tmp_a[1])
|
|||
|
|
target_b = int(tmp_b[1])
|
|||
|
|
|
|||
|
|
# 重新生成目标并设置连接
|
|||
|
|
mesh_a = cmds.sculptTarget(
|
|||
|
|
tmp_a[0],
|
|||
|
|
edit=True,
|
|||
|
|
regenerate=True,
|
|||
|
|
target=target_a
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
mesh_shape_a = cmds.listRelatives(mesh_a[0], shapes=True)
|
|||
|
|
|
|||
|
|
# 断开原有连接并建立新连接
|
|||
|
|
cmds.disconnectAttr(
|
|||
|
|
f"{mesh_shape_a[0]}.worldMesh[0]",
|
|||
|
|
f"{tmp_a[0]}.inputTarget[0].inputTargetGroup[{target_a}].inputTargetItem[6000].inputGeomTarget"
|
|||
|
|
)
|
|||
|
|
cmds.connectAttr(
|
|||
|
|
f"{mesh_shape_a[0]}.worldMesh[0]",
|
|||
|
|
f"{tmp_a[0]}.inputTarget[0].inputTargetGroup[{target_b}].inputTargetItem[6000].inputGeomTarget",
|
|||
|
|
force=True
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 删除临时网格
|
|||
|
|
cmds.delete(mesh_a)
|
|||
|
|
|
|||
|
|
def mirror_blend():
|
|||
|
|
"""
|
|||
|
|
镜像混合变形目标
|
|||
|
|
"""
|
|||
|
|
# 获取当前对称设置
|
|||
|
|
original_symmetry = cmds.symmetricModelling(query=True, symmetry=True)
|
|||
|
|
symmetry_space = None
|
|||
|
|
symmetry_axis = None
|
|||
|
|
|
|||
|
|
if original_symmetry:
|
|||
|
|
symmetry_space = cmds.symmetricModelling(query=True, about=True)
|
|||
|
|
if symmetry_space == "topo":
|
|||
|
|
symmetry_axis = mel.eval('blendShapeGetTopoSymmetryEdge()')
|
|||
|
|
else:
|
|||
|
|
symmetry_axis = cmds.symmetricModelling(query=True, axis=True)
|
|||
|
|
|
|||
|
|
# 获取选中的混合变形目标
|
|||
|
|
sel_bs_name_list = mel.eval('getShapeEditorTreeviewSelection(14)')
|
|||
|
|
|
|||
|
|
# 解析目标索引
|
|||
|
|
tmp_a = sel_bs_name_list[0].split('.')
|
|||
|
|
tmp_b = sel_bs_name_list[1].split('.')
|
|||
|
|
|
|||
|
|
target_a = int(tmp_a[1])
|
|||
|
|
target_b = int(tmp_b[1])
|
|||
|
|
|
|||
|
|
# 重新生成目标并设置连接
|
|||
|
|
mesh_a = cmds.sculptTarget(
|
|||
|
|
tmp_a[0],
|
|||
|
|
edit=True,
|
|||
|
|
regenerate=True,
|
|||
|
|
target=target_a
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
mesh_shape_a = cmds.listRelatives(mesh_a[0], shapes=True)
|
|||
|
|
|
|||
|
|
# 断开原有连接并建立新连接
|
|||
|
|
cmds.disconnectAttr(
|
|||
|
|
f"{mesh_shape_a[0]}.worldMesh[0]",
|
|||
|
|
f"{tmp_a[0]}.inputTarget[0].inputTargetGroup[{target_a}].inputTargetItem[6000].inputGeomTarget"
|
|||
|
|
)
|
|||
|
|
cmds.connectAttr(
|
|||
|
|
f"{mesh_shape_a[0]}.worldMesh[0]",
|
|||
|
|
f"{tmp_a[0]}.inputTarget[0].inputTargetGroup[{target_b}].inputTargetItem[6000].inputGeomTarget",
|
|||
|
|
force=True
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 删除临时网格
|
|||
|
|
cmds.delete(mesh_a)
|
|||
|
|
|
|||
|
|
# 翻转目标
|
|||
|
|
cmds.blendShape(
|
|||
|
|
tmp_b[0],
|
|||
|
|
edit=True,
|
|||
|
|
flipTarget=True,
|
|||
|
|
geometryIndex=0,
|
|||
|
|
targetIndex=target_b,
|
|||
|
|
symmetryAxis="x",
|
|||
|
|
symmetrySpace=1
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 重置反射模式
|
|||
|
|
mel.eval('reflectionSetMode none')
|
|||
|
|
|
|||
|
|
# 恢复对称设置
|
|||
|
|
if not original_symmetry or not symmetry_axis:
|
|||
|
|
mel.eval('symmetricModelling -s 0')
|
|||
|
|
elif symmetry_space == "topo":
|
|||
|
|
mel.eval(f'symmetricModelling -e -about {symmetry_space} -s 1 {symmetry_axis}')
|
|||
|
|
else:
|
|||
|
|
mel.eval(f'symmetricModelling -e -about {symmetry_space} -axis {symmetry_axis} -s 1')
|
|||
|
|
|
|||
|
|
def get_shape_editor_treeview_selection(scope):
|
|||
|
|
"""
|
|||
|
|
获取形状编辑器树视图中的选择
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
scope (int): 选择范围:
|
|||
|
|
0-8: 简单范围
|
|||
|
|
9: 单个BSD中的目标
|
|||
|
|
10-16: 纯类型选择
|
|||
|
|
20: 最后选择的项目
|
|||
|
|
21,24: 纯类型及其组类型
|
|||
|
|
30: 引用项目检查
|
|||
|
|
31: 第一个选择项
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
list: 根据范围返回的字符串数组
|
|||
|
|
"""
|
|||
|
|
# 定义类型上限
|
|||
|
|
type_roof = 6 # 注意：添加新树项类型时更新上限
|
|||
|
|
|
|||
|
|
# 检查选择变量是否存在
|
|||
|
|
if not cmds.optionVar(exists="blendShapeEditorTreeViewSelection"):
|
|||
|
|
return []
|
|||
|
|
|
|||
|
|
# 获取选择数据
|
|||
|
|
ov = cmds.optionVar(query="blendShapeEditorTreeViewSelection")
|
|||
|
|
|
|||
|
|
# 处理不同范围的选择
|
|||
|
|
if -1 < scope < type_roof + 3: # 范围 0-8
|
|||
|
|
return ov[scope].split("/") if ov[scope] else []
|
|||
|
|
|
|||
|
|
if scope == 9: # 范围 9
|
|||
|
|
return filter_single_bsd(ov[8].split("/") if ov[8] else [])
|
|||
|
|
|
|||
|
|
if 9 < scope < type_roof + 11: # 范围 10-16
|
|||
|
|
selection = ov[scope-10].split("/") if ov[scope-10] else []
|
|||
|
|
if not selection:
|
|||
|
|
return []
|
|||
|
|
|
|||
|
|
# 跳过相关类型检查
|
|||
|
|
type_array = list(range(type_roof))
|
|||
|
|
new_type_array = [x for x in type_array if x not in [scope-10, 2, 5]]
|
|||
|
|
|
|||
|
|
if scope in [12, 15]:
|
|||
|
|
new_type_array = [x for x in new_type_array if x != scope-12]
|
|||
|
|
|
|||
|
|
# 检查混合选择
|
|||
|
|
for i in new_type_array:
|
|||
|
|
if ov[i] and ov[i].split("/"):
|
|||
|
|
return []
|
|||
|
|
return selection
|
|||
|
|
|
|||
|
|
if scope == 20: # 范围 20
|
|||
|
|
return ov[9].split("/") if ov[9] else []
|
|||
|
|
|
|||
|
|
if scope in [21, 24]: # 范围 21,24
|
|||
|
|
selection_out = ov[scope-20].split("/") if ov[scope-20] else []
|
|||
|
|
selection_in = ov[scope-19].split("/") if ov[scope-19] else []
|
|||
|
|
selection = list(set(selection_out + selection_in))
|
|||
|
|
|
|||
|
|
if not selection:
|
|||
|
|
return []
|
|||
|
|
|
|||
|
|
# 跳过相关类型检查
|
|||
|
|
type_array = list(range(type_roof))
|
|||
|
|
new_type_array = [x for x in type_array if x not in [scope-20, scope-19, scope-21]]
|
|||
|
|
|
|||
|
|
# 检查混合选择
|
|||
|
|
for i in new_type_array:
|
|||
|
|
if ov[i] and ov[i].split("/"):
|
|||
|
|
return []
|
|||
|
|
return selection
|
|||
|
|
|
|||
|
|
if scope == 30: # 范围 30
|
|||
|
|
return ov[10].split("/") if ov[10] else []
|
|||
|
|
|
|||
|
|
if scope == 31: # 范围 31
|
|||
|
|
return ov[11].split("/") if ov[11] else []
|
|||
|
|
|
|||
|
|
return [] # 无效范围
|
|||
|
|
|
|||
|
|
def filter_single_bsd(items):
|
|||
|
|
"""
|
|||
|
|
从单个混合变形变形器中返回选定的目标（组）
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
items (list): 要过滤的项目列表
|
|||
|
|
返回:
|
|||
|
|
list: 过滤后的项目列表，如果来自多个BSD则返回空列表
|
|||
|
|
"""
|
|||
|
|
result = []
|
|||
|
|
one_bsd = ""
|
|||
|
|
|
|||
|
|
for item in items:
|
|||
|
|
sub_strings = item.split(".")
|
|||
|
|
if not one_bsd:
|
|||
|
|
one_bsd = sub_strings[0]
|
|||
|
|
if one_bsd != sub_strings[0]:
|
|||
|
|
return []
|
|||
|
|
result.append(item)
|
|||
|
|
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
def is_shape_editor_reference_item_selected():
|
|||
|
|
"""
|
|||
|
|
检查是否在形状编辑器树视图中选择了引用项
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
bool: 如果选择了引用项则返回True，否则返回False
|
|||
|
|
"""
|
|||
|
|
result = get_shape_editor_treeview_selection(30)
|
|||
|
|
if len(result) == 1:
|
|||
|
|
return result[0] == "1"
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def blend_shape_editor_selected_ancestor(bsd_name, tgt_index, selected_targets):
|
|||
|
|
"""
|
|||
|
|
获取选定的祖先（如果存在）
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
bsd_name (str): 混合变形变形器名称
|
|||
|
|
tgt_index (int): 目标索引（>=0为目标，<0为目标目录）
|
|||
|
|
selected_targets (list): 选定的目标列表
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
int: 最高选定的父目录索引，如果没有选定的祖先则返回0
|
|||
|
|
"""
|
|||
|
|
highest_parent_directory = 0
|
|||
|
|
parent_dir = -1
|
|||
|
|
|
|||
|
|
if tgt_index >= 0:
|
|||
|
|
# 这是一个目标，获取父目录
|
|||
|
|
temp_attr = f"{bsd_name}.parentDirectory[{tgt_index}]"
|
|||
|
|
parent_dir = cmds.getAttr(temp_attr)
|
|||
|
|
|
|||
|
|
if parent_dir <= -1:
|
|||
|
|
return 0 # 默认值
|
|||
|
|
if parent_dir == 0:
|
|||
|
|
return 0 # 目标项直接在混合变形节点下
|
|||
|
|
else:
|
|||
|
|
# 这是一个目标目录，获取父目录
|
|||
|
|
temp_attr = f"{bsd_name}.targetDirectory[{-tgt_index}].parentIndex"
|
|||
|
|
parent_dir = cmds.getAttr(temp_attr)
|
|||
|
|
|
|||
|
|
while parent_dir > 0:
|
|||
|
|
# 检查此目录是否被选中
|
|||
|
|
dir_name_encoded = f"{bsd_name}.{-parent_dir}"
|
|||
|
|
if dir_name_encoded in selected_targets:
|
|||
|
|
highest_parent_directory = parent_dir
|
|||
|
|
|
|||
|
|
# 获取父目录的父目录
|
|||
|
|
temp_attr = f"{bsd_name}.targetDirectory[{parent_dir}].parentIndex"
|
|||
|
|
parent_dir = cmds.getAttr(temp_attr)
|
|||
|
|
|
|||
|
|
return highest_parent_directory
|