o ig @sdZddlmZddlmZmZddlZddlm Z ddl m Z m Z m Z mZmZmZddlmZmZmZdd lmZd!ddZeddfd"ddZd#dd ZdS)$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_vadvad np.ndarrayreturnfloatcCs,ttj|}|tkrtS|tkrtStS)zLPer-query blend weight: neutral favors parametric, extreme favors archetype.) rnplinalgnormrrr r r )r magnituderN/dataset/kemix-engine/package/face/animasync-face-v3/scripts/compiler/blend.py_dynamic_blend_weights rTemotion Optional[str]presetsdictsigmaapply_lipsync_maskbooloverride_blend_weightOptional[float]c Cstt|}t|}t||||d}|dur|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] ) emotion_hintr Ng?g) rr rrrrrclipastypefloat32) rrrr r!r#paramarchwoutrrrcompile_blendshapes#s r-emotionsList[Optional[str]]vadscKst|jd}t||krtdt|d|tj|dftjd}t|D]}t|||||fi|||<q%|S)z-Batch version. vads: (N, 3). Returns (N, 52).rzemotions length z != vads 4)dtype)shapelen ValueErrorremptyr(ranger-)r.r0rkwargsNr,irrr compile_batchNs   "r;)rrrr)rrrrrrr rr!r"r#r$rr)r.r/r0rrrrr)__doc__ __future__rtypingrrnumpyr archetyper constantsrrr r r r parametricr rrutilsrrr-r;rrrrs      +