U 8jY @shdZddlmZddlmZddlZddlmZm Z m Z dddd d Z defd d d dd dddZ 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_DEEMPHASISstr) preset_namereturncCsd|kr|ddS|S)uBExtract emotion family from preset name (e.g. 'joy_L3' → 'joy')._r)split)r r ^/home/ubuntu/service/kemix-engine/package/face/animasync-face-v3/scripts/compiler/archetype.py_extract_familysrz np.ndarraydictz Optional[str]float)vadpresets emotion_hintsigmar csb|stjdtjdStjtjdg}g}|D]\}}tj|dtjd}ttj|} tt| d d||} |dk rt |} | |kr| t 9} n| dkr|dkr| t 9} | | | tj|dtjdq4tj|tjd}| d kr2fd 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 Esz#archetype_blend..r)axiszi,ij->j)rzerosr ritemsrrrexprrrappendsumvaluesintargmincopystackeinsumastype)rrrrweightsZbs_listnamedataZ anchor_vaddistwfamily distancesidxZbs_stackr r#rarchetype_blends6      r:) __doc__ __future__rtypingrnumpyr constantsrrrrr:r r r rs