261 lines
10 KiB
Python
261 lines
10 KiB
Python
"""
|
|
Animation Retargeting 관련 명령 핸들러
|
|
본 매핑, 애니메이션 리타게팅 실행
|
|
"""
|
|
|
|
import bpy
|
|
from typing import Dict
|
|
from ..utils.logger import get_logger
|
|
from ..utils.bone_matching import fuzzy_match_bones, get_match_quality_report
|
|
|
|
logger = get_logger(__name__)
|
|
|
|
|
|
def auto_map_bones(source_armature: str, target_armature: str) -> Dict[str, str]:
|
|
"""
|
|
자동 본 매핑 (Mixamo -> 사용자 캐릭터)
|
|
Fuzzy matching 알고리즘 사용으로 정확도 개선
|
|
|
|
Args:
|
|
source_armature: 소스 아마추어 이름 (Mixamo)
|
|
target_armature: 타겟 아마추어 이름 (사용자 캐릭터)
|
|
|
|
Returns:
|
|
본 매핑 딕셔너리 {source_bone: target_bone}
|
|
|
|
Raises:
|
|
ValueError: 아마추어를 찾을 수 없는 경우
|
|
"""
|
|
logger.info(f"Auto-mapping bones: {source_armature} -> {target_armature}")
|
|
|
|
source = bpy.data.objects.get(source_armature)
|
|
target = bpy.data.objects.get(target_armature)
|
|
|
|
if not source or not target:
|
|
logger.error("Source or target armature not found")
|
|
raise ValueError("Source or target armature not found")
|
|
|
|
# Mixamo 표준 본 이름과 알려진 별칭 (확장: 손가락, 발가락 포함)
|
|
mixamo_bone_aliases = {
|
|
# 몸통 (6개)
|
|
"Hips": ["hips", "pelvis", "root"],
|
|
"Spine": ["spine", "spine1"],
|
|
"Spine1": ["spine1", "spine2"],
|
|
"Spine2": ["spine2", "spine3", "chest"],
|
|
"Neck": ["neck"],
|
|
"Head": ["head"],
|
|
|
|
# 왼쪽 팔 (4개)
|
|
"LeftShoulder": ["shoulder.l", "clavicle.l", "leftshoulder"],
|
|
"LeftArm": ["upper_arm.l", "leftarm", "upperarm.l"],
|
|
"LeftForeArm": ["forearm.l", "leftforearm", "lowerarm.l"],
|
|
"LeftHand": ["hand.l", "lefthand"],
|
|
|
|
# 오른쪽 팔 (4개)
|
|
"RightShoulder": ["shoulder.r", "clavicle.r", "rightshoulder"],
|
|
"RightArm": ["upper_arm.r", "rightarm", "upperarm.r"],
|
|
"RightForeArm": ["forearm.r", "rightforearm", "lowerarm.r"],
|
|
"RightHand": ["hand.r", "righthand"],
|
|
|
|
# 왼쪽 다리 (4개)
|
|
"LeftUpLeg": ["thigh.l", "leftupleg", "upperleg.l"],
|
|
"LeftLeg": ["shin.l", "leftleg", "lowerleg.l"],
|
|
"LeftFoot": ["foot.l", "leftfoot"],
|
|
"LeftToeBase": ["toe.l", "lefttoebase", "foot.l.001"],
|
|
|
|
# 오른쪽 다리 (4개)
|
|
"RightUpLeg": ["thigh.r", "rightupleg", "upperleg.r"],
|
|
"RightLeg": ["shin.r", "rightleg", "lowerleg.r"],
|
|
"RightFoot": ["foot.r", "rightfoot"],
|
|
"RightToeBase": ["toe.r", "righttoebase", "foot.r.001"],
|
|
|
|
# 왼쪽 손가락 (15개)
|
|
"LeftHandThumb1": ["thumb.01.l", "lefthandthumb1", "thumb_01.l"],
|
|
"LeftHandThumb2": ["thumb.02.l", "lefthandthumb2", "thumb_02.l"],
|
|
"LeftHandThumb3": ["thumb.03.l", "lefthandthumb3", "thumb_03.l"],
|
|
"LeftHandIndex1": ["f_index.01.l", "lefthandindex1", "index_01.l"],
|
|
"LeftHandIndex2": ["f_index.02.l", "lefthandindex2", "index_02.l"],
|
|
"LeftHandIndex3": ["f_index.03.l", "lefthandindex3", "index_03.l"],
|
|
"LeftHandMiddle1": ["f_middle.01.l", "lefthandmiddle1", "middle_01.l"],
|
|
"LeftHandMiddle2": ["f_middle.02.l", "lefthandmiddle2", "middle_02.l"],
|
|
"LeftHandMiddle3": ["f_middle.03.l", "lefthandmiddle3", "middle_03.l"],
|
|
"LeftHandRing1": ["f_ring.01.l", "lefthandring1", "ring_01.l"],
|
|
"LeftHandRing2": ["f_ring.02.l", "lefthandring2", "ring_02.l"],
|
|
"LeftHandRing3": ["f_ring.03.l", "lefthandring3", "ring_03.l"],
|
|
"LeftHandPinky1": ["f_pinky.01.l", "lefthandpinky1", "pinky_01.l"],
|
|
"LeftHandPinky2": ["f_pinky.02.l", "lefthandpinky2", "pinky_02.l"],
|
|
"LeftHandPinky3": ["f_pinky.03.l", "lefthandpinky3", "pinky_03.l"],
|
|
|
|
# 오른쪽 손가락 (15개)
|
|
"RightHandThumb1": ["thumb.01.r", "righthandthumb1", "thumb_01.r"],
|
|
"RightHandThumb2": ["thumb.02.r", "righthandthumb2", "thumb_02.r"],
|
|
"RightHandThumb3": ["thumb.03.r", "righthandthumb3", "thumb_03.r"],
|
|
"RightHandIndex1": ["f_index.01.r", "righthandindex1", "index_01.r"],
|
|
"RightHandIndex2": ["f_index.02.r", "righthandindex2", "index_02.r"],
|
|
"RightHandIndex3": ["f_index.03.r", "righthandindex3", "index_03.r"],
|
|
"RightHandMiddle1": ["f_middle.01.r", "righthandmiddle1", "middle_01.r"],
|
|
"RightHandMiddle2": ["f_middle.02.r", "righthandmiddle2", "middle_02.r"],
|
|
"RightHandMiddle3": ["f_middle.03.r", "righthandmiddle3", "middle_03.r"],
|
|
"RightHandRing1": ["f_ring.01.r", "righthandring1", "ring_01.r"],
|
|
"RightHandRing2": ["f_ring.02.r", "righthandring2", "ring_02.r"],
|
|
"RightHandRing3": ["f_ring.03.r", "righthandring3", "ring_03.r"],
|
|
"RightHandPinky1": ["f_pinky.01.r", "righthandpinky1", "pinky_01.r"],
|
|
"RightHandPinky2": ["f_pinky.02.r", "righthandpinky2", "pinky_02.r"],
|
|
"RightHandPinky3": ["f_pinky.03.r", "righthandpinky3", "pinky_03.r"],
|
|
}
|
|
|
|
# 소스 본 리스트 (실제로 존재하는 본만)
|
|
source_bones = [bone.name for bone in source.data.bones
|
|
if bone.name in mixamo_bone_aliases]
|
|
|
|
# 타겟 본 리스트
|
|
target_bones = [bone.name for bone in target.data.bones]
|
|
|
|
# Fuzzy matching 실행 (정확한 매칭 우선, 그 다음 유사도 매칭)
|
|
logger.info("Running fuzzy bone matching algorithm...")
|
|
logger.debug(f"Source bones: {len(source_bones)}, Target bones: {len(target_bones)}")
|
|
|
|
bone_map = fuzzy_match_bones(
|
|
source_bones=source_bones,
|
|
target_bones=target_bones,
|
|
known_aliases=mixamo_bone_aliases,
|
|
threshold=0.6, # 60% 이상 유사도
|
|
prefer_exact=True # 정확한 매칭 우선
|
|
)
|
|
|
|
# 매칭 품질 보고서
|
|
quality_report = get_match_quality_report(bone_map)
|
|
logger.info(f"Auto-mapped {quality_report['total_mappings']} bones")
|
|
logger.info(f"Quality: {quality_report['quality'].upper()}")
|
|
logger.info(f"Critical bones: {quality_report['critical_bones_mapped']}")
|
|
logger.debug(f"Bone mapping: {bone_map}")
|
|
|
|
# 콘솔 출력 (사용자에게 피드백)
|
|
print(f"✅ Auto-mapped {quality_report['total_mappings']} bones")
|
|
print(f" Quality: {quality_report['quality'].upper()}")
|
|
print(f" Critical bones: {quality_report['critical_bones_mapped']}")
|
|
|
|
return bone_map
|
|
|
|
|
|
def retarget_animation(
|
|
source_armature: str,
|
|
target_armature: str,
|
|
bone_map: Dict[str, str],
|
|
preserve_rotation: bool = True,
|
|
preserve_location: bool = False
|
|
) -> str:
|
|
"""
|
|
애니메이션 리타게팅 실행
|
|
|
|
Args:
|
|
source_armature: 소스 아마추어 이름
|
|
target_armature: 타겟 아마추어 이름
|
|
bone_map: 본 매핑 딕셔너리
|
|
preserve_rotation: 회전 보존 여부
|
|
preserve_location: 위치 보존 여부 (보통 루트 본만)
|
|
|
|
Returns:
|
|
결과 메시지
|
|
|
|
Raises:
|
|
ValueError: 아마추어를 찾을 수 없거나 애니메이션이 없는 경우
|
|
"""
|
|
logger.info(f"Retargeting animation: {source_armature} -> {target_armature}")
|
|
logger.debug(f"Bone mappings: {len(bone_map)}, Rotation: {preserve_rotation}, Location: {preserve_location}")
|
|
|
|
source = bpy.data.objects.get(source_armature)
|
|
target = bpy.data.objects.get(target_armature)
|
|
|
|
if not source or not target:
|
|
logger.error("Source or target armature not found")
|
|
raise ValueError("Source or target armature not found")
|
|
|
|
if not source.animation_data or not source.animation_data.action:
|
|
logger.error("Source armature has no animation")
|
|
raise ValueError("Source armature has no animation")
|
|
|
|
# 타겟 아마추어 선택
|
|
bpy.context.view_layer.objects.active = target
|
|
target.select_set(True)
|
|
|
|
# Pose 모드로 전환
|
|
bpy.ops.object.mode_set(mode='POSE')
|
|
|
|
# 각 본에 대해 컨스트레인트 생성
|
|
constraints_added = 0
|
|
for source_bone_name, target_bone_name in bone_map.items():
|
|
if source_bone_name not in source.pose.bones:
|
|
logger.debug(f"Source bone not found: {source_bone_name}")
|
|
continue
|
|
if target_bone_name not in target.pose.bones:
|
|
logger.debug(f"Target bone not found: {target_bone_name}")
|
|
continue
|
|
|
|
target_bone = target.pose.bones[target_bone_name]
|
|
|
|
# Rotation constraint
|
|
if preserve_rotation:
|
|
constraint = target_bone.constraints.new('COPY_ROTATION')
|
|
constraint.target = source
|
|
constraint.subtarget = source_bone_name
|
|
constraints_added += 1
|
|
|
|
# Location constraint (일반적으로 루트 본만)
|
|
if preserve_location and source_bone_name == "Hips":
|
|
constraint = target_bone.constraints.new('COPY_LOCATION')
|
|
constraint.target = source
|
|
constraint.subtarget = source_bone_name
|
|
constraints_added += 1
|
|
|
|
logger.info(f"Added {constraints_added} constraints")
|
|
|
|
# 컨스트레인트를 키프레임으로 베이크
|
|
logger.info("Baking constraints to keyframes...")
|
|
bpy.ops.nla.bake(
|
|
frame_start=bpy.context.scene.frame_start,
|
|
frame_end=bpy.context.scene.frame_end,
|
|
only_selected=False,
|
|
visual_keying=True,
|
|
clear_constraints=True,
|
|
bake_types={'POSE'}
|
|
)
|
|
|
|
bpy.ops.object.mode_set(mode='OBJECT')
|
|
|
|
logger.info("Animation retargeting completed successfully")
|
|
return f"Animation retargeted to {target_armature}"
|
|
|
|
|
|
def get_preset_bone_mapping(preset: str) -> Dict[str, str]:
|
|
"""
|
|
미리 정의된 본 매핑 프리셋
|
|
|
|
Args:
|
|
preset: 프리셋 이름 (예: "mixamo_to_rigify")
|
|
|
|
Returns:
|
|
본 매핑 딕셔너리
|
|
"""
|
|
logger.debug(f"Getting bone mapping preset: {preset}")
|
|
|
|
presets = {
|
|
"mixamo_to_rigify": {
|
|
"Hips": "torso",
|
|
"Spine": "spine",
|
|
"Spine1": "spine.001",
|
|
"Spine2": "spine.002",
|
|
"Neck": "neck",
|
|
"Head": "head",
|
|
"LeftShoulder": "shoulder.L",
|
|
"LeftArm": "upper_arm.L",
|
|
"LeftForeArm": "forearm.L",
|
|
"LeftHand": "hand.L",
|
|
# ... 더 많은 매핑 추가 가능
|
|
}
|
|
}
|
|
|
|
bone_map = presets.get(preset, {})
|
|
logger.info(f"Preset '{preset}' loaded with {len(bone_map)} mappings")
|
|
return bone_map
|