U 8j@sdZddlmZddlZddlmZddlmZddlZ ddl m Z m Z dd d d d d dddddddddddddZ ddddZdd d!d"d#Zddd$d%Zddd&d'ZdS)(zBUtilities: preset loading, validation, synthetic preset bootstrap.) annotationsN)Path)Dict)ARKIT_52_NAMES NAME_TO_IDX)eyeLookDownLefteyeLookDownRight) eyeLookUpLefteyeLookUpRight) eyeLookInLefteyeLookInRight)eyeLookOutLefteyeLookOutRight) eyeBlinkLeft eyeBlinkRight) eyeWideLeft eyeWideRight) eyeSquintLefteyeSquintRight)mouthSmileLeftmouthSmileRight)mouthFrownLeftmouthFrownRight)mouthDimpleLeftmouthDimpleRight)mouthPressLeftmouthPressRight)mouthStretchLeftmouthStretchRight)mouthLowerDownLeftmouthLowerDownRight)mouthUpperUpLeftmouthUpperUpRight) browDownLeft browDownRight)browOuterUpLeftbrowOuterUpRight)cheekSquintLeftcheekSquintRight) noseSneerLeftnoseSneerRight)Z eyesLookDownZ eyesLookUpZ eyesLookInZ eyesLookOutZ eyesBlinkZeyesWideZ eyesSquintZ mouthSmileZ mouthFrownZ mouthDimpleZ mouthPressZ mouthStretchZmouthLowerDownZ mouthUpperUpZbrowDownZ browOuterUpZ cheekSquintZ noseSneerz np.ndarray)returncCs@tj|tjd}|jdkr*td|jt|ddtjS)z.Ensure VAD is shape (3,), float32, in [-1, 1].dtypezVAD shape must be (3,), got gg?)npasarrayfloat32shape ValueErrorclipastype)vadr9Z/home/ubuntu/service/kemix-engine/package/face/animasync-face-v3/scripts/compiler/utils.py validate_vad's r;zDict[str, dict]None)presetsr,cCs|D]\}}d|ks d|kr0td|dt|d}t|d}|jdkrjt|d|j|jdkrt|d|jt|d k|d kBrt|d qd S) z!Raise if any preset is malformed.r8bszpreset z missing 'vad' or 'bs'r/z: vad shape must be (3,), got )4z: bs shape must be (52,), got g-C6gqh?z: bs outside [0, 1]N)itemsr5r1r2r4any)r=namedatar8r>r9r9r:validate_presets/s  rDcCst|}t|jdd}|d|}ttjd}|ddd}t|jddd}i}|D]*\}}|D]} | d ||t | d f<qvqji} |D]\} } | dp| d d } t | d d}|| |fdddg}| dp| dpi}t j dt j d}|D]$\}}|tkrt||t|<q|D]B\}}|tkrFt|d krFt|D]}t||t|<qnqFt||d| | <qt| | 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-8encodingr=rCemotionemotion_vad_anchors.jsonanchorsr8level_r intensityr0gvaluesZ blendshapesr?r-r8r>)rjsonloads read_textget__file__resolveparentsr@intsplitr1zerosr3rfloatEDITOR_META_CHANNELSlistrD)pathrawZ presets_dict project_root anchor_path anchors_rawZ vad_lookupemoentrieseoutrBrCrIrNr8Z values_mapr>Zbs_namevZarkr9r9r:load_presets_from_json>s6     rhc Csddlm}m}ttjd}|ddd}t|j ddd }i}| D]d\}}|D]V}t |d } t j |d t jd } || } || | } |d | } | | d|| <q`qTt||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_resolutionrHrCrIrJrErFrKrLr8r-Z_LrP) parametricrirjrrUrVrWrQrRrSr@rXr1r2r3tolistrD) rirjr`rarbrfrcrdrerLr8r>rBr9r9r:build_synthetic_presetsus  rm)__doc__ __future__rrQpathlibrtypingrnumpyr1 constantsrrr\r;rDrhrmr9r9r9r:s:   7