Files
anatomia/data-pipeline/scripts/mannequin-asset/repose_and_export.py
T
Rodrigo Bertoldo de Sousa Martins 85ec49c4f0 fix corpo manequim
Signed-off-by: Rodrigo Bertoldo de Sousa Martins <rodrigo@martinsevieira.adv.br>
2026-09-14 22:52:55 -03:00

126 lines
4.8 KiB
Python

"""Reposiciona os bracos do 'GEO-body_male_realistic' (bundle Human Base
Meshes da Blender Studio, CC0) da pose relaxada original para posicao
anatomica (bracos retos ao lado do corpo), e exporta pra GLB.
Rodar de dentro do Blender, sem interface:
blender --background human_base_meshes_bundle.blend \
--python repose_and_export.py
A malha nao tem armature, entao a repose e feita rotacionando os vertices
do braco em torno de um pivo no ombro. O angulo de rotacao e MEDIDO na
propria malha (distancia/posicao do vertice mais distante do ombro dentro
da faixa de altura do braco), nao e um valor chutado, entao continua
funcionando mesmo que uma versao futura do bundle mude a pose de origem.
Uma janela suave (smoothstep) evita costura na axila.
"""
import math
import bpy
BODY_OBJECT = "GEO-body_male_realistic"
EYE_OBJECTS = ["GEO-body_male_realistic.eye.L", "GEO-body_male_realistic.eye.R"]
MULTIRES_LEVEL = 1 # ~42k triangulos: mantem a definicao muscular visivel a distancia normal de camera, bem mais leve que o nivel 3 (~677k) do arquivo original
OUTPUT_PATH = "/tmp/mannequin-realistic.glb"
SKIN_COLOR = (0.56, 0.64, 0.78, 1.0) # mesmo tom usado no Mannequin.tsx anterior
def smoothstep(edge0, edge1, x):
t = max(0.0, min(1.0, (x - edge0) / (edge1 - edge0)))
return t * t * (3 - 2 * t)
def repose_arms(obj):
mesh = obj.data
zs = [v.co.z for v in mesh.vertices]
z_min, z_max = min(zs), max(zs)
height = z_max - z_min
centerline_x = sum(v.co.x for v in mesh.vertices) / len(mesh.vertices)
shoulder_z = z_min + 0.84 * height
# A mao termina de forma bem abrupta por volta de z=0.705-0.71 nesta
# escala (medido amostrando a malha por fatias de altura — ver
# diagnose2.py no historico) — abaixo disso so existe coxa. Uma janela
# mais larga aqui (ex: baseada só numa fracao generica da altura) pega
# vertice de coxa por engano e deforma a perna inteira, porque eles
# entram na rotacao rigida em torno do pivo do OMBRO.
arm_hand_low_z = z_min + 0.417 * height # ~0.705 pra este mesh especifico
torso_half_width = 0.20 # metade da largura biacromial aproximada, nesta escala
# Angulo residual: parar EXATAMENTE na vertical faz a mao encostar (ou
# atravessar) a coxa. Deixar uma pequena abducao residual dá uma folga
# natural, como nas referências de posição anatômica.
residual = math.radians(5.0)
for side in (1, -1):
pivot_x = centerline_x + side * torso_half_width * 0.95
candidates = [
v
for v in mesh.vertices
if side * (v.co.x - centerline_x) > torso_half_width
and arm_hand_low_z < v.co.z < shoulder_z - 0.02
]
far = max(
candidates,
key=lambda v: (v.co.x - pivot_x) ** 2 + (v.co.z - shoulder_z) ** 2,
)
measured_angle = math.atan2(far.co.x - pivot_x, -(far.co.z - shoulder_z))
angle = measured_angle - side * residual
for v in mesh.vertices:
dx = v.co.x - centerline_x
if side * dx <= 0 or v.co.z >= shoulder_z + 0.03:
continue
weight_x = smoothstep(torso_half_width * 0.7, torso_half_width * 1.15, abs(dx))
# janela estreita: cheia bem em cima da mao, zero já na coxa —
# a transição real na malha é abrupta (não gradual), então uma
# rampa larga aqui é o que causava a contaminação da perna.
weight_z = smoothstep(arm_hand_low_z - 0.015, arm_hand_low_z + 0.005, v.co.z)
weight = weight_x * weight_z
if weight <= 0.001:
continue
rot = -weight * angle
cos_a, sin_a = math.cos(rot), math.sin(rot)
px, pz = v.co.x - pivot_x, v.co.z - shoulder_z
v.co.x = pivot_x + (px * cos_a - pz * sin_a)
v.co.z = shoulder_z + (px * sin_a + pz * cos_a)
mesh.update()
def main():
body = bpy.data.objects[BODY_OBJECT]
repose_arms(body)
for mod in body.modifiers:
if mod.type == "MULTIRES":
mod.levels = MULTIRES_LEVEL
mat = bpy.data.materials.new("MannequinSkin")
mat.use_nodes = True
mat.node_tree.nodes["Principled BSDF"].inputs["Base Color"].default_value = SKIN_COLOR
mat.node_tree.nodes["Principled BSDF"].inputs["Roughness"].default_value = 0.6
targets = [body] + [bpy.data.objects[n] for n in EYE_OBJECTS if n in bpy.data.objects]
for obj in targets:
obj.data.materials.clear()
obj.data.materials.append(mat)
bpy.ops.object.select_all(action="DESELECT")
for obj in targets:
obj.select_set(True)
bpy.context.view_layer.objects.active = body
bpy.ops.export_scene.gltf(
filepath=OUTPUT_PATH,
export_format="GLB",
use_selection=True,
export_apply=True,
export_yup=True,
)
print(f"Exportado para {OUTPUT_PATH}")
main()