o [iq@sUdZddlmZddlZddlmZddlmZmZm Z ddl Z ddl m Z ddlmZdd lmZeejd d d d ZdZgdZdaded<d(ddZeedfd)d!d"Zeedfd*d&d'ZdS)+uExpressive compile: blend user-authored L1/L3/L5 presets within emotion family. Philosophy: when a turn has an emotion label, the user's authored presets ARE the ground truth. VAD determines intensity level (L1 mild, L5 extreme) via RBF distance weighting on anchor VAD — NOT by blending with parametric rules or other emotion families. Result: full expressivity of authored L5 poses at extreme VAD, faithful fall-off to L3/L1 at milder VAD. Emotion label is dominant. ) annotationsN)Path)DictListOptional) LIPSYNC_ONLY)parametric_layer) validate_vaddataemotionzemotion_vad_anchors.jsongffffff?)+,z*Optional[Dict[str, Dict[int, np.ndarray]]] _anchor_cachereturn Dict[str, Dict[int, np.ndarray]]cCsRtdurtSttjddd}i}|D] \}}dd|D||<q|a|S)Nzutf-8)encodinganchorscSs*i|]}t|dtj|dtjdqS)levelvaddtype)intnpasarrayfloat32).0er%S/dataset/kemix-engine/package/face/animasync-face-v3/scripts/compiler/expressive.py 4s*z!_load_anchors..)rjsonloads _ANCHOR_PATH read_textitems)rawoutemoentriesr%r%r& _load_anchors-sr1? Optional[str]r np.ndarraypresetsDict[str, dict]sigmafloatparametric_overlay_channelsOptional[List[int]]parametric_overlay_intensitycst|dus |dkr,|d}|dur$ttj|dtjdddStjdtjdSt|}|dur>tjdtjdSg}g} g} d D]-} |d | } | |vsW| |vrXqF|| | || | tj|| dtjdqF| s~tjdtjdStj fd d | Dtjd} t | d  d|d }|| }tj | dd}|dddf|j dd}|rt }tj|tjd}t|||||||<t|ddtjS)uReturn 52-dim blendshape by blending authored L1/L3/L5 presets of emotion, with optional channel-masked parametric overlay (Option E). Args: emotion: emotion family name (e.g. 'crying'). None → neutral. vad: (3,) target VAD. presets: dict with keys like 'crying_L1', 'crying_L3', 'crying_L5'. sigma: RBF bandwidth for level blending. parametric_overlay_channels: list of channel indices on which to add the parametric layer's output via max() merge. Defaults to mouth/cheek "valence-coloring" channels. Pass None or [] to disable. parametric_overlay_intensity: scalar α applied to the parametric output before max-merge (1.0 = pure max, lower = milder coloring). Option E rationale: emotion + within-emotion RBF defines the structural pose (eyes, brows, jaw). Valence then continuously colors the mouth/cheek so happy-surprise (V+) and sad-surprise (V−) become visually distinct without re-authoring presets per V quadrant. Nneutral neutral_L3bsrrr4)r_Lcsg|] }tj|qSr%)r linalgnorm)r#avrr%r& psz&compile_expressive..r g@)axisgr2)r getr clipr!r"zerosr1appendarrayexpsumstackr int64maximumastype)r rr5r7r9r;r<rlevels anchor_vads bs_vectorsLkeydistsweightsbs_stackr.parammaskr%rFr&compile_expressive9s>     r^emotionsList[Optional[str]]vadsc CsP|jd}tj|dftjd}t|D]}t||||||||d||<q|S)z-Batch version. vads: (N, 3). Returns (N, 52).rr?r)r7r9r;)shaper emptyr"ranger^) r_rar5r7r9r;Nr.ir%r%r&compile_expressive_batchs   rg)rr)r r3rr4r5r6r7r8r9r:r;r8rr4)r_r`rar4r5r6r7r8r9r:r;r8rr4)__doc__ __future__rr(pathlibrtypingrrrnumpyr constantsr parametricr utilsr __file__resolveparentsr*EXPRESSIVE_SIGMAMOUTH_CHEEK_OVERLAY_CHANNELSr__annotations__r1r^rgr%r%r%r&s,        O