ig dZddlmZddlmZmZddlZddlm Z ddl m Z m Z m Z mZmZmZddlmZmZmZdd lmZddZeddfddZd dZdS)!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.ndarrayreturnfloatcttj|}|tkrt S|t krtStS)zLPer-query blend weight: neutral favors parametric, extreme favors archetype.) rnplinalgnormrr r r r )r magnitudes N/dataset/kemix-engine/package/face/animasync-face-v3/scripts/compiler/blend.py_dynamic_blend_weightrsGbinnS))**I***##***## Temotion Optional[str]presetsdictsigmaapply_lipsync_maskbooloverride_blend_weightOptional[float]cNt|}t|}t||||}||nt|}||z|d|z zz} t | |} |rt | } t j| ddt j 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) rrrrrrrclipastypefloat32) rrr r"r#r%paramarchwouts rcompile_blendshapesr0#s* s  C S ! !E 3gU K K KD"7!BH]^aHbHbA !)dcAg& &C $C - -C& %% 73S ! ! ( ( 4 44remotionsList[Optional[str]]vadsc .|jd}t||kr"tdt|d|tj|dftj}t |D] }t|||||fi|||<!|S)z-Batch version. vads: (N, 3). Returns (N, 52).rzemotions length z != vads 4)dtype)shapelen ValueErrorremptyr+ranger0)r1r3r kwargsNr/is r compile_batchr?Ns 1 A 8}}GCMMGGAGGHHH (Ar7"* - - -C 1XXNN$Xa[$q'7MMfMMA Jr)rrrr)rrrrr r!r"rr#r$r%r&rr)r1r2r3rr r!rr)__doc__ __future__rtypingrrnumpyr archetyper constantsrr r r r r parametricrrrutilsrrr0r?rrrIs@==""""""!!!!!!!!&&&&&&      #-1 (5(5(5(5(5V      r