U 8jg @sdZddlmZddlmZmZddlZddlm Z ddl m Z m Z m Z mZmZmZddlmZmZmZdd lmZd d d d dZeddfdd dd ddd dddZdd dd dddZdS)z7Main compiler API: blend parametric + archetype layers.) annotations)ListOptionalN)archetype_blend)BLEND_NEUTRAL_THRESHOLDBLEND_NEUTRAL_WEIGHTBLEND_EXTREME_THRESHOLDBLEND_EXTREME_WEIGHTBLEND_DEFAULT_WEIGHT RBF_SIGMA)parametric_layerapply_conflict_resolutionapply_channel_mask) validate_vadz np.ndarrayfloat)vadreturncCs,ttj|}|tkrtS|tkr(tStS)zLPer-query blend weight: neutral favors parametric, extreme favors archetype.) rnpZlinalgZnormrrr r r )rZ magnituderZ/home/ubuntu/service/kemix-engine/package/face/animasync-face-v3/scripts/compiler/blend.py_dynamic_blend_weights rTz Optional[str]dictboolzOptional[float])emotionrpresetssigmaapply_lipsync_maskoverride_blend_weightrc Cstt|}t|}t||||d}|dk r,|nt|}|||d|} t| |} |r^t| } t| ddtj S)uCompute 52 blendshape values for given (emotion, VAD). Args: emotion: emotion family name (e.g. 'joy') for archetype boosting; None → pure VAD vad: (3,) in [-1, 1] presets: dict of archetype presets sigma: RBF bandwidth (archetype layer) apply_lipsync_mask: zero out LIPSYNC_ONLY channels override_blend_weight: if set, use this instead of dynamic Returns: (52,) float32 in [0, 1] )Z emotion_hintrNg?g) rr rrrrrZclipZastypefloat32) rrrrrrparamZarchwoutrrrcompile_blendshapes#s r#zList[Optional[str]])emotionsvadsrrcKsp|jd}t||kr.tdt|d|tj|dftjd}t|D] }t|||||f|||<qJ|S)z-Batch version. vads: (N, 3). Returns (N, 52).rzemotions length z != vads 4)Zdtype)shapelen ValueErrorremptyrranger#)r$r%rkwargsNr"irrr compile_batchNs   r/)__doc__Z __future__rtypingrrZnumpyrZ archetyper constantsrrr r r r Z parametricr rrutilsrrr#r/rrrrs    +