101 lines
3.5 KiB
Python
101 lines
3.5 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
|
||
import maya.cmds as cmds
|
||
from scripts.utils.Core import GetMeshes, GetBlendShape, GetBlendShapes
|
||
|
||
def is_sorted(array):
|
||
"""
|
||
检查数组是否已排序
|
||
|
||
参数:
|
||
array (list): 要检查的数组
|
||
|
||
返回:
|
||
bool: 如果数组已排序返回True，否则返回False
|
||
"""
|
||
return all(array[i] >= array[i-1] for i in range(1, len(array)))
|
||
|
||
def are_indices_sorted(array_a, array_b):
|
||
"""
|
||
检查两个数组中共同元素的索引是否有序
|
||
|
||
参数:
|
||
array_a (list): 第一个数组
|
||
array_b (list): 第二个数组
|
||
|
||
返回:
|
||
bool: 如果共同元素的索引都有序返回True，否则返回False
|
||
"""
|
||
# 存储共同元素的索引
|
||
common_indices_a = []
|
||
common_indices_b = []
|
||
|
||
# 查找共同元素并记录它们的索引
|
||
for i, item_a in enumerate(array_a):
|
||
for j, item_b in enumerate(array_b):
|
||
if item_a == item_b:
|
||
common_indices_a.append(i)
|
||
common_indices_b.append(j)
|
||
break
|
||
|
||
# 检查两个索引数组是否都有序
|
||
return is_sorted(common_indices_a) and is_sorted(common_indices_b)
|
||
|
||
def reorder_blend_shapes():
|
||
"""重新排序所有网格的混合变形目标"""
|
||
# 遍历前50个网格
|
||
for mesh_index in range(50):
|
||
mesh = GetMeshes(m=mesh_index)
|
||
|
||
if cmds.objExists(mesh):
|
||
blend_shape = GetBlendShape(mesh)
|
||
|
||
if cmds.objExists(blend_shape):
|
||
attr_weight = f"{blend_shape}.weight"
|
||
|
||
# 获取目标数量和索引
|
||
nb_in_tgt = cmds.getAttr(attr_weight, size=True)
|
||
existing_indices = cmds.getAttr(attr_weight, multiIndices=True)
|
||
|
||
# 获取当前和预期的混合变形目标列表
|
||
current_blend_shape_list = cmds.listAttr(attr_weight, m=True)
|
||
blend_shape_list = GetBlendShapes()
|
||
|
||
# 检查是否需要重新排序
|
||
needs_reorder = (
|
||
nb_in_tgt - 1 != existing_indices[-1] or
|
||
not are_indices_sorted(blend_shape_list, current_blend_shape_list)
|
||
)
|
||
|
||
if needs_reorder:
|
||
# 重新生成所有目标
|
||
target_list = []
|
||
for j in existing_indices:
|
||
target = cmds.sculptTarget(
|
||
blend_shape,
|
||
e=True,
|
||
regenerate=True,
|
||
target=j
|
||
)
|
||
target_list.append(target[0])
|
||
|
||
# 删除原始混合变形器
|
||
cmds.delete(blend_shape)
|
||
|
||
# 按照预期顺序选择目标
|
||
cmds.select(clear=True)
|
||
for bs_name in blend_shape_list:
|
||
if bs_name in target_list:
|
||
cmds.select(bs_name, add=True)
|
||
|
||
# 创建新的混合变形器
|
||
cmds.select(mesh, add=True)
|
||
cmds.blendShape(automatic=True, name=blend_shape)
|
||
|
||
# 清理临时目标
|
||
cmds.delete(target_list)
|
||
|
||
# 如果直接运行此脚本
|
||
if __name__ == '__main__':
|
||
reorder_blend_shapes() |