d~idZddlmZddlZddlmZddlmZddlZ ddl m Z m Z idd d d d d ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+iZ d5d.Zd6d2Zd7d3Zd7d4ZdS)8zBUtilities: preset loading, validation, synthetic preset bootstrap.) annotationsN)Path)Dict)ARKIT_52_NAMES NAME_TO_IDX eyesLookDown)eyeLookDownLefteyeLookDownRight eyesLookUp) eyeLookUpLefteyeLookUpRight eyesLookIn) eyeLookInLefteyeLookInRight eyesLookOut)eyeLookOutLefteyeLookOutRight eyesBlink) eyeBlinkLeft eyeBlinkRighteyesWide) eyeWideLeft eyeWideRight eyesSquint) eyeSquintLefteyeSquintRight mouthSmile)mouthSmileLeftmouthSmileRight mouthFrown)mouthFrownLeftmouthFrownRight mouthDimple)mouthDimpleLeftmouthDimpleRight mouthPress)mouthPressLeftmouthPressRight mouthStretch)mouthStretchLeftmouthStretchRightmouthLowerDown)mouthLowerDownLeftmouthLowerDownRight mouthUpperUp)mouthUpperUpLeftmouthUpperUpRightbrowDown) browDownLeft browDownRight browOuterUp)browOuterUpLeftbrowOuterUpRight cheekSquint)cheekSquintLeftcheekSquintRight noseSneer) noseSneerLeftnoseSneerRightreturn np.ndarrayctj|tj}|jdkrt d|jtj|ddtjS)z.Ensure VAD is shape (3,), float32, in [-1, 1].dtypezVAD shape must be (3,), got gg?)npasarrayfloat32shape ValueErrorclipastype)vads N/dataset/kemix-engine/package/face/animasync-face-v3/scripts/compiler/utils.py validate_vadrO'sb *S + + +C yDC CCDDD 73c " " ) )"* 5 55presetsDict[str, dict]Nonec|D]\}}d|vsd|vrtd|dtj|d}tj|d}|jdkrt|d|j|jdkrt|d|jtj|d k|d kzrt|d d S) z!Raise if any preset is malformed.rMbszpreset z missing 'vad' or 'bs'rDz: vad shape must be (3,), got )4z: bs shape must be (52,), got g-C6gqh?z: bs outside [0, 1]N)itemsrJrFrGrIany)rQnamedatarMrUs rNvalidate_presetsr[/s mmoo ; ; d   D 0 0CtCCCDD Dje%% ZT # # 9  OOCIOOPP P 8u  NNBHNNOO O 62:"x-0 1 1 ;999:: : ; ; ;rPc 2t|}tj|d}|d|}tt jd}|dz dz dz }tj|dd}i}|D]*\}}|D]"} | d ||t| d f<#+i} |D]w\} } | dp| d d } t| d d}|| |fgd}| dp| dpi}tj dtj }|D]+\}}|tvrt||t|<,|D]N\}}|t vr@t|d kr-t |D]}t||t|< Ot#||d| | <yt%| | S)aeLoad presets from the JSON format exported by tools/blendshape-editor.html. Expected schema: {"presets": {"joy_L3": {"emotion": "joy", "intensity": 3, "values": {"mouthSmileLeft": 0.7, ...}}, ...}} or flat dict at top level. VAD is read from emotion_vad_anchors.json based on (emotion, intensity). utf-8encodingrQrZemotionemotion_vad_anchors.jsonanchorsrMlevel_r intensityrE)rgrgvalues blendshapesrVrBrMrU)rjsonloads read_textget__file__resolveparentsrWintsplitrFzerosrHrfloatEDITOR_META_CHANNELSlistr[)pathraw presets_dict project_root anchor_path anchors_raw vad_lookupemoentrieseoutrYrZrarfrM values_maprUbs_namevarks rNload_presets_from_jsonr>s ::D *T^^W^55 6 6C779c**L>>))++3A6L')36PPK*[22G2DDEEiPKJ#))++:: W : :A125JS7__- . . :C"((**11 d((9%%;C);a0011 nngy1???CCXXh''H488M+B+BHb Xb + + +$**,, 4 4JGQ+%%+088;w'($**,, 4 4JGQ...588a<</844C+088B{3'(( IIR00D S JrPc0ddlm}m}ttjd}|dz dz dz }tj| dd }i}| D]}\}}|D]u}t|d } tj |d tj } || } || | } |d | } | | d|| <v~t!||S)a&Bootstrap presets using the parametric layer on each anchor VAD. Use this when no user-authored preset JSON is available. Each preset's blendshape vector is computed by running the parametric layer on the anchor's VAD coordinate. This ensures anchor-reproducibility trivially. r)parametric_layerapply_conflict_resolutionr`rZrarbr]r^rcrdrMrB_Lrj) parametricrrrrorprqrkrlrmrWrrrFrGrHtolistr[) rrr{r|r}rrrrrdrMrUrYs rNbuild_synthetic_presetsrus<HGGGGGGG>>))++3A6L')36PPK*[22G2DDEEiPKC#))++88 W 8 8A' OOE*QuXRZ888C!!#&&B**2s33B$$U$$D # B77CII  8S JrP)r?r@)rQrRr?rS)r?rR)__doc__ __future__rrkpathlibrtypingrnumpyrF constantsrrrvrOr[rrrPrNrsHH"""""" 22222222?==>  <  ; =>>?>@B@< ?!"?#$=%,6666 ; ; ; ;4444nrP