#!/usr/bin/env python
# -*- coding: utf-8 -*-

import maya.cmds as cmds
from scripts.utils.Core import GetMeshes, ProgressBar, MetaHumanNormals

def repair_normals(index):
    """
    修复模型法线
    
    参数:
        index (int): 模型索引，目前只支持0（头部模型）
    """
    # 只处理头部模型
    if index != 0:
        return
        
    # 获取顶点对应关系
    vtxs = MetaHumanNormals()
    
    # 获取头部和身体模型
    head = GetMeshes(m=0)
    body = GetMeshes(m=50)
    
    # 计算需要处理的顶点对数量
    count = len(vtxs) // 2
    
    # 初始化进度条
    ProgressBar(sp=True)
    ProgressBar(max_value=count)
    ProgressBar(t="Repair Normals...")
    
    # 处理每对对应的顶点
    for i in range(count):
        # 更新进度条
        ProgressBar(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]))
    
    # 结束进度条
    ProgressBar(ep=True)
    
    # 清除选择并选择身体模型
    cmds.select(clear=True)
    cmds.select(body, replace=True)
    
    # 烘焙历史记录
    cmds.BakeAllNonDefHistory()

# 如果直接运行此脚本
if __name__ == '__main__':
    pass