o iY @sVdZddlmZddlmZddlZddlmZm Z m Z dd d Z defdddZ dS)uArchetype layer: RBF interpolation over authored preset blendshapes. Per docs/research/vad-to-arkit-blendshape-mapping.md §6.1. ) annotations)OptionalN) RBF_SIGMARBF_EMOTION_FAMILY_BOOSTRBF_NEUTRAL_DEEMPHASIS preset_namestrreturncCsd|vr |ddS|S)uBExtract emotion family from preset name (e.g. 'joy_L3' → 'joy')._r)split)rr R/dataset/kemix-engine/package/face/animasync-face-v3/scripts/compiler/archetype.py_extract_familysrvad np.ndarraypresetsdict emotion_hint Optional[str]sigmafloatcs`|s tjdtjdStjtjdg}g}|D]W\}}tj|dtjd}ttj|} tt| d d||} |dur_t |} | |krS| t 9} n | dkr_|dkr_| t 9} | | | tj|dtjdqtj|tjd}| d krfd d |D} tt| } || S|| }tj|d d }td||tjS)aWeight-blend preset blendshapes by RBF distance in VAD space. Args: vad: (3,) float, query VAD in [-1, 1] presets: dict {name: {"vad": [V,A,D], "bs": np.ndarray(52,)}} emotion_hint: optional family name (e.g. 'joy'); boosts same-family presets sigma: RBF Gaussian bandwidth Returns: (52,) float32 weighted blend of preset blendshapes 4dtyperg@Nneutralbsgư>c s0g|]}ttjtj|dtjdqS)rr)rnplinalgnormasarrayfloat32).0drr r Es"z#archetype_blend..r)axiszi,ij->j)rzerosr"r!itemsrrr exprrrappendsumvaluesintargmincopystackeinsumastype)rrrrweightsbs_listnamedata anchor_vaddistwfamily distancesidxbs_stackr r%rarchetype_blends6       r?)rr r r ) rrrrrrrrr r) __doc__ __future__rtypingrnumpyr constantsrrrrr?r r r rs