#!/usr/bin/env python # -*- coding: utf-8 -*- import maya.cmds as cmds import sys def body_transformation(gender_mesh): def add_chain_depth_attribute(joints): for joint in joints: if not cmds.attributeQuery('chainDepth', node=joint, exists=True): cmds.addAttr(joint, longName='chainDepth', attributeType='long', defaultValue=0) cmds.setAttr(joint + '.chainDepth', keyable=True) def set_chain_depth_value(joints, value): for joint in joints: cmds.setAttr(joint + '.chainDepth', value) # 获取场景中的所有关节 all_joints = cmds.ls(type="joint") # 移除"DHIhead:spine_04"层次中的关节（避免头部） exclude_joints = cmds.ls("DHIhead:spine_04", dag=True, type="joint") all_joints = list(set(all_joints) - set(exclude_joints)) add_chain_depth_attribute(all_joints) # 过滤末端关节（没有子关节的关节） end_joints = [joint for joint in all_joints if not cmds.listRelatives(joint, children=True, type='joint')] # 将chainDepth属性设置为0，所有末端关节 set_chain_depth_value(all_joints, 100) set_chain_depth_value(end_joints, 0) parents1 = [] for joint_name in end_joints: p_joint = cmds.listRelatives(joint_name, parent=True, type="joint") if p_joint: children = cmds.listRelatives(p_joint, children=True, type="joint") or [] if all(cmds.getAttr(child + ".chainDepth") == 0 for child in children): parents1.append(p_joint[0]) set_chain_depth_value(parents1, 1) #链深度添加 Attr 循环 chainDepth = 1 while parents1: chainDepth += 1 new_parents = [] for joint_name in parents1: p_joint = cmds.listRelatives(joint_name, parent=True, type="joint") if p_joint: children = cmds.listRelatives(p_joint, children=True, type="joint") or [] if all(cmds.getAttr(child + ".chainDepth") < chainDepth for child in children): new_parents.append(p_joint[0]) if new_parents: set_chain_depth_value(new_parents, chainDepth) parents1 = new_parents else: break j_name = [] j_delta_x = [] j_delta_y = [] j_delta_z = [] # DELTA X 计算 for joint_name in end_joints: mesh_name = gender_mesh # 1. 获取关节的世界空间位置 joint_pos = cmds.xform(joint_name, query=True, worldSpace=True, translation=True) # 2. 找到最近的顶点"NVertex"，位于关节位置 closest_point_node = cmds.createNode("closestPointOnMesh") cmds.connectAttr(mesh_name + ".worldMesh", closest_point_node + ".inMesh") cmds.setAttr(closest_point_node + ".inPosition", joint_pos[0], joint_pos[1], joint_pos[2]) nearest_vertex_index = cmds.getAttr(closest_point_node + ".closestVertexIndex") cmds.delete(closest_point_node) vertex = mesh_name + ".vtx[{}]".format(nearest_vertex_index) cmds.select(vertex, replace=True) # 增长选择 for i in range(1): # 获取当前选择顶点 vertices = cmds.ls(selection=True, flatten=True) # 将顶点转换为边 edges = [] for vertex in vertices: edges.extend(cmds.polyListComponentConversion(vertex, fromVertex=True, toEdge=True)) # 选择边 cmds.select(edges) # 获取当前选择边 edges = cmds.ls(selection=True, flatten=True) # 将边转换为顶点 vertices = [] for edge in edges: vertices.extend(cmds.polyListComponentConversion(edge, fromEdge=True, toVertex=True)) # 选择顶点 cmds.select(vertices) vertices = cmds.ls(selection=True, flatten=True) # 将顶点添加到列表中 vertex_position_before = [] for vertex in vertices: position = cmds.pointPosition(vertex, world=True) vertex_position_before.append(position) # 3 将目标混合形状节点设置为1 cmds.setAttr("BodyShape.B2", 1) # 4. 获取NVertex的新位置并找到新位置 vertex_position_after = [] for vertex in vertices: position = cmds.pointPosition(vertex, world=True) vertex_position_after.append(position) delta_vertex = [] for i in range(len(vertex_position_after)): sub_delta = [] for j in range(3): sub_delta.append(vertex_position_after[i][j] - vertex_position_before[i][j]) delta_vertex.append(sub_delta) average_vertex = [sum(x) / len(x) for x in zip(*delta_vertex)] # 5. 将关节移动到NVertex移动的距离 delta_x = average_vertex[0] delta_y = average_vertex[1] delta_z = average_vertex[2] if not cmds.attributeQuery("delta_x", node=joint_name, exists=True): cmds.addAttr(joint_name, ln="delta_x", at="double") if not cmds.attributeQuery("delta_y", node=joint_name, exists=True): cmds.addAttr(joint_name, ln="delta_y", at="double") if not cmds.attributeQuery("delta_z", node=joint_name, exists=True): cmds.addAttr(joint_name, ln="delta_z", at="double") cmds.setAttr(joint_name + ".delta_x", delta_x) cmds.setAttr(joint_name + ".delta_y", delta_y) cmds.setAttr(joint_name + ".delta_z", delta_z) # 6 重置 cmds.setAttr("BodyShape.B2", 0) j_name.append(joint_name) j_delta_x.append(delta_x) j_delta_y.append(delta_y) j_delta_z.append(delta_z) #cmds.move(delta_x, delta_y, delta_z, joint_name, relative=True) for i in range(len(j_name)): cmds.move(j_delta_x[i], j_delta_y[i], j_delta_z[i], j_name[i], relative=True) #修复身体部分 moving_joint = [] target_joint = [] def fix_body(moving_joint, target_joint): m_pos = cmds.xform(moving_joint, query=True, worldSpace=True, rotatePivot=True) ta_pos = cmds.xform(target_joint, query=True, worldSpace=True, rotatePivot=True) child_joints = cmds.listRelatives(moving_joint, c=True, type="joint") child_positions = [] if child_joints: for child_joint in child_joints: child_positions.append(cmds.xform(child_joint, q=True, ws=True, t=True)) # 将"FACIAL_C_LowerLipRotation"关节的平移和旋转枢轴属性设置为与"FACIAL_C_MouthUpper"相同 cmds.xform(moving_joint, translation=ta_pos, rotatePivot=m_pos, worldSpace=True) if child_joints: for i, child_joint in enumerate(child_joints): cmds.xform(child_joint, ws=True, t=child_positions[i]) m_pos = cmds.xform(moving_joint, query=True, worldSpace=True, rotatePivot=True) ta_pos = cmds.xform(target_joint, query=True, worldSpace=True, rotatePivot=True) cmds.xform(target_joint, translation=m_pos, rotatePivot=ta_pos, worldSpace=True) # 过滤chainDepth值为0的关节 CD = 1 lower_CD = 0 maxCD = 15 while CD <= maxCD: joint_CD = [joint for joint in all_joints if cmds.getAttr(joint + ".chainDepth") == CD] lower_joints = [joint for joint in all_joints if cmds.getAttr(joint + ".chainDepth") == lower_CD] for joint_name in joint_CD: first_place = cmds.xform(joint_name, q=True, ws=True, translation=True) child_joints = cmds.listRelatives(joint_name, children=True, type="joint") average_point = [] if child_joints: target_points = [] for child_joint in child_joints: if child_joint in lower_joints: delta_x = cmds.getAttr(child_joint + '.delta_x') delta_y = cmds.getAttr(child_joint + '.delta_y') delta_z = cmds.getAttr(child_joint + '.delta_z') target_points.append([delta_x, delta_y, delta_z]) if child_joint in end_joints: delta_x = cmds.getAttr(child_joint + '.delta_x') delta_y = cmds.getAttr(child_joint + '.delta_y') delta_z = cmds.getAttr(child_joint + '.delta_z') #print(cmds.getAttr(child_joint + ".chainDepth")) target_points.append([delta_x, delta_y, delta_z]) target_points.append([delta_x, delta_y, delta_z]) target_points.append([delta_x, delta_y, delta_z]) target_points.append([delta_x, delta_y, delta_z]) target_points.append([delta_x, delta_y, delta_z]) average_point = [sum(x) / len(x) for x in zip(*target_points)] if not average_point: average_point = [0, 0, 0] avg_x = average_point[0] avg_y = average_point[1] avg_z = average_point[2] cmds.move(avg_x, avg_y, avg_z, joint_name, relative=True) for child_joint in child_joints: cmds.move(-avg_x, -avg_y, -avg_z, child_joint, relative=True) pureJoint = joint_name[:-2] last_letter = joint_name[-2:] joint_name_corrective = pureJoint+"_correctiveRoot"+last_letter joint_name_half = pureJoint+"_half"+last_letter if joint_name_corrective in all_joints: fix_body(joint_name, joint_name_corrective) #print(joint_name + " fixed to the " + joint_name_corrective) if joint_name_half in all_joints: fix_body(joint_name, joint_name_half) #print(joint_name + " fixed to the " + joint_name_half) end_place = cmds.xform(joint_name, q=True, ws=True, translation=True) delta_x = end_place[0] - first_place[0] delta_y = end_place[1] - first_place[1] delta_z = end_place[2] - first_place[2] if not cmds.attributeQuery("delta_x", node=joint_name, exists=True): cmds.addAttr(joint_name, ln="delta_x", at="double") if not cmds.attributeQuery("delta_y", node=joint_name, exists=True): cmds.addAttr(joint_name, ln="delta_y", at="double") if not cmds.attributeQuery("delta_z", node=joint_name, exists=True): cmds.addAttr(joint_name, ln="delta_z", at="double") cmds.setAttr(joint_name + ".delta_x", delta_x) cmds.setAttr(joint_name + ".delta_y", delta_y) cmds.setAttr(joint_name + ".delta_z", delta_z) lower_CD += 1 CD += 1 # # # 将身体原始骨骼转换为新头部骨骼 CD = 15 lower_CD = -1 minCD = 0 # 定义关节名称和命名空间 head_joint = "DHIhead:spine_04" body_joint = "DHIbody:spine_04" # 将关节名称和命名空间分开 head_namespace, head_joint_name = head_joint.split(":") body_namespace, body_joint_name = body_joint.split(":") # 获取头部关节的子关节列表 head_child_joints = cmds.listRelatives(head_joint, children=True, allDescendents=True,type="joint") body_child_joints = cmds.listRelatives(body_joint, children=True, allDescendents=True, type="joint") while CD >= minCD: joint_CD = [joint for joint in body_child_joints if cmds.getAttr(joint + ".chainDepth") == CD] for joint_name in joint_CD: body_namespace, body_joint_name = joint_name.split(":") # 构建当前关节的关节名称 head_current_joint = "{}:{}".format(head_namespace, body_joint_name) body_current_joint = "{}:{}".format(body_namespace, body_joint_name) if cmds.objExists(head_current_joint): moving_joint = body_current_joint target_joint = head_current_joint def move2target_joint(): lower_lip_rotation_pos = cmds.xform(moving_joint, query=True, worldSpace=True, rotatePivot=True) jaw_pos = cmds.xform(target_joint, query=True, worldSpace=True, rotatePivot=True) # 获取子关节 child_joints = cmds.listRelatives(moving_joint, c=True, type="joint") # 存储子关节的初始位置 child_positions = [] if child_joints: for child_joint in child_joints: child_positions.append(cmds.xform(child_joint, q=True, ws=True, t=True)) # 将"FACIAL_C_LowerLipRotation"关节的平移和旋转枢轴属性设置为与"FACIAL_C_MouthUpper"相同 cmds.xform(moving_joint, translation=jaw_pos, rotatePivot=lower_lip_rotation_pos, worldSpace=True) if child_joints: # 将每个子关节移动回其原始位置 for i, child_joint in enumerate(child_joints): cmds.xform(child_joint, ws=True, t=child_positions[i]) move2target_joint() lower_CD -= 1 CD -= 1 fix_body("DHIbody:spine_04", "DHIhead:spine_04") def bind_skin(gender_mesh): # 身体 mesh_obj = cmds.ls(gender_mesh)[0] # 复制网格 duplicated_mesh_obj = cmds.duplicate(mesh_obj)[0] new_name = gender_mesh + "_duplicate" cmds.rename(duplicated_mesh_obj, new_name) bind_obj = cmds.ls(new_name)[0] cmds.select([bind_obj, "DHIbody:root"]) # 绑定皮肤到网格 skin_cluster = cmds.skinCluster("DHIbody:root", bind_obj)[0] # 将皮肤方法设置为"经典线性" cmds.setAttr(skin_cluster + ".skinningMethod", 1) cmds.select([mesh_obj, bind_obj]) cmds.copySkinWeights(noMirror=True, surfaceAssociation="closestPoint", influenceAssociation=["name", "oneToOne"]) cmds.delete(mesh_obj) cmds.rename(bind_obj, gender_mesh) ROOT_DIR = "c:/dna_calibration" gender=0 gender_mesh = "f_tal_nrw_body_lod0_mesh" skeleton_file_path = f"{ROOT_DIR}/data/Fem_Body_skeleton.fbx" def build_body(ROOT_DIR): #切换性别 if gender == 0: skeleton_file_path = f"{ROOT_DIR}/data/Fem_Body_skeleton.fbx" gender_mesh = "f_tal_nrw_body_lod0_mesh" elif gender == 1: skeleton_file_path = f"{ROOT_DIR}/data/Ma_Body_skeleton.fbx" gender_mesh = "m_med_nrw_body_lod0_mesh" # 将选定的对象重命名为B2 selected_object = cmds.ls(selection=True) cmds.rename(selected_object, "B2") # 导入FBX文件 cmds.file(skeleton_file_path, i=True, type="FBX", ignoreVersion=True, ra=True, mergeNamespacesOnClash=False, options="fbx") cmds.select("B2", replace=True) # 定义混合形状节点和目标形状名称 blendshape_node = "BodyShape" target_shape = "|body_rig|body_grp|body_geometry_grp|body_lod0_grp|"+gender_mesh+"|"+gender_mesh+"Shape" cmds.blendShape(gender_mesh, automatic=True) cmds.rename("blendShape1", "BodyShape") cmds.blendShape(blendshape_node, edit=True, target=[(target_shape, 0.0, "B2", 1.0)]) cmds.setAttr("BodyShape.B2", 0) #cmds.setAttr("B2" + ".visibility", 0) cmds.delete('B2') body_transformation(gender_mesh) # 复制头部网格 body_mesh = gender_mesh new_mesh = cmds.duplicate(body_mesh)[0] # 解锁平移属性 cmds.setAttr(new_mesh+'.translateX', lock=False) cmds.setAttr(new_mesh+'.translateY', lock=False) cmds.setAttr(new_mesh+'.translateZ', lock=False) # 将复制网格沿X轴移动75个单位 cmds.move(75, 0, 0, new_mesh, relative=True) # 将复制网格添加为目标混合形状节点 blend_shape_node = "BodyShape" cmds.blendShape(blend_shape_node, edit=True, target=(body_mesh, 1, new_mesh, 1.0)) # 将复制网格的目标混合形状权重设置为-1 cmds.setAttr("BodyShape."+new_mesh, -1) cmds.setAttr("BodyShape.B2", 1) cmds.delete(gender_mesh+"1") bind_skin(gender_mesh) #build_body("c:/dna_calibration")