[iqUdZddlmZddlZddlmZddlmZmZm Z ddl Z ddl m Z ddlmZdd lmZeejd d z d z d z ZdZgdZdaded<d%dZeedfd&d Zeedfd'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]]cttStjtdd}i}|D]\}}d|D||<|a|S)Nzutf-8)encodinganchorsci|]<}t|dtj|dtj=S)levelvaddtype)intnpasarrayfloat32).0es S/dataset/kemix-engine/package/face/animasync-face-v3/scripts/compiler/expressive.py z!_load_anchors..4s:]]]PQC' OORZ% %K%K%K]]])rjsonloads _ANCHOR_PATH read_textitems)rawoutemoentriess r( _load_anchorsr4-sy  *\++W+== > >y IC,.C ^^ W]]U\]]]CM Jr*? Optional[str]r np.ndarraypresetsDict[str, dict]sigmafloatparametric_overlay_channelsOptional[List[int]]parametric_overlay_intensitycBt||dkrq|d}|:tjtj|dtjddStjdtjSt|}| tjdtjSg}g} g} d D]} |d | } | |vs| |vr|| | || | tj|| dtj| s tjdtjStj fd | Dtj} tj | d z d |d zzz }|| z}tj | d}|dddf|z d}|rVt}tj|tj}tj|||||z||<tj|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_L3bsr rr4)r_LcTg|]$}tj|z %S)r#linalgnorm)r&avrs r( z&compile_expressive..ps+EEE2binnS2X..EEEr*r g@)axisgr5)r getr#clipr$r%zerosr4appendarrayexpsumstackr int64maximumastype)rrr8r:r<r>r@rlevels anchor_vads bs_vectorsLkeydistsweightsbs_stackr1parammasks ` r(compile_expressiverc9s6 s  C'Y..++l++  72:gdm2:FFF1MM Mx"*----oo!!'**Gx"*----F$&K#%J LLA g  '!1!1  a71:&&&"*WS\$%7rzJJJKKKK .x"*---- HEEEEEEERZ X X XEfuz]cEQJ&6788G w{{}}Gx +++H 111d7 h & + + + 3 3C #V %%z5RXFFFJs4y*Ft*TUUD 73S ! ! ( ( 4 44r*emotionsList[Optional[str]]vadsc |jd}tj|dftj}t |D]&}t ||||||||||<'|S)z-Batch version. vads: (N, 3). Returns (N, 52).rrCr )r:r<r>)shaper#emptyr%rangerc) rdrfr8r:r<r>Nr1is r(compile_expressive_batchrmsw 1 A (Ar7"* - - -C 1XX  # QKa'(C)E    A Jr*)rr)rr6rr7r8r9r:r;r<r=r>r;rr7)rdrerfr7r8r9r:r;r<r=r>r;rr7)__doc__ __future__rr+pathlibrtypingrrrnumpyr# constantsr parametricr utilsr __file__resolveparentsr-EXPRESSIVE_SIGMAMOUTH_CHEEK_OVERLAY_CHANNELSr__annotations__r4rcrmrHr*r(r|sn   #""""" ''''''''''######((((((tH~~%%''/2V;iGJdd    =A @@@@     $7S*- H5H5H5H5H5^$7S*- r*