MetaFusion/scripts/Reference/SGRepairNormals.py

98 lines
3.0 KiB
Python
Raw Normal View History

2025-01-17 02:30:36 +08:00
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
版权所有: 深圳时光科技有限公司
联系方式: q.100@qq.com
创建日期: 2024/03/01
"""
import maya.cmds as cmds
def sg_meta_human_normals():
"""
返回MetaHuman模型的法线顶点对应关系
Returns:
list: 顶点索引列表每两个值为一组分别对应头部和身体的顶点
"""
return [
2805,9, 2806,8, 2807,7, 2808,10, 2809,6,
2810,5, 2811,2, 2812,1, 2813,0, 2814,4,
2815,12, 2816,14, 2817,15, 2818,16, 2819,17,
2820,18, 2821,19, 2822,20, 2823,21, 2824,11,
2873,22, 2876,23, 2965,3, 2997,13, 5876,3818,
5877,30, 5878,31, 5879,32, 5880,29, 5881,28,
5882,26, 5883,24, 5884,25, 5885,27, 5886,34,
5887,35, 5888,36, 5889,37, 5890,38, 5891,39,
5892,40, 5893,41, 5894,42, 5895,33, 5944,43,
5947,44, 11560,16789, 11564,16786, 11567,16795,
11569,16794, 11572,16784, 11575,7645, 11578,22814,
11582,16779, 11585,22833, 11587,22824, 11600,22801,
11795,21578, 11798,21577, 11799,21575, 11802,21573,
11803,21571, 11805,21569, 11807,21566, 11809,21564,
11811,21540, 11899,16775, 11909,16774, 11910,21542,
17607,9206, 17610,9203, 17614,9211, 17617,9207,
17619,9198, 17622,15227, 17626,15170, 17629,9195,
17632,15189, 17635,15180, 17647,15158, 17854,13968,
17856,13966, 17859,13964, 17860,13962, 17863,13960,
17865,13958, 17867,13957, 17869,13955, 17871,13927,
17962,9192, 17968,9186, 17972,13931
]
def sg_repair_normals(index):
"""
修复模型法线
Args:
index (int): 模型索引目前只支持0头部模型
"""
# 只处理头部模型
if index != 0:
return
# 获取顶点对应关系
vtxs = sg_meta_human_normals()
# 获取头部和身体模型
head = cmds.SGGetMeshes(m=0)
body = cmds.SGGetMeshes(m=50)
# 计算需要处理的顶点对数量
count = len(vtxs) // 2
# 初始化进度条
cmds.SGProgressBar(sp=True)
cmds.SGProgressBar(max=count)
cmds.SGProgressBar(t="Repair Normals...")
# 处理每对对应的顶点
for i in range(count):
# 更新进度条
cmds.SGProgressBar(apr=1)
# 获取头部和身体对应的顶点索引
h = vtxs[i*2]
b = vtxs[i*2+1]
# 构建顶点名称
head_vtx = f"{head}.vtx[{h}]"
body_vtx = f"{body}.vtx[{b}]"
# 获取头部顶点的法线
pos = cmds.polyNormalPerVertex(head_vtx, q=True, xyz=True)
# 将法线应用到身体顶点
cmds.polyNormalPerVertex(body_vtx, xyz=(pos[0], pos[1], pos[2]))
# 结束进度条
cmds.SGProgressBar(ep=True)
# 清除选择并选择身体模型
cmds.select(clear=True)
cmds.select(body, replace=True)
# 烘焙历史记录
cmds.BakeAllNonDefHistory()
# 如果直接运行此脚本
if __name__ == '__main__':
pass