o j@sdZddlmZddlZddlZddlmZedZedZ edZ ed Z ed Z ed Z ed Zed ZedZedZedZedZedZedZdd#d$Zd?d%d&Z  ' ( (d@dAd-d.Z  / 0 1dBdCd5d6Z  ' / 0dDdEd:d;ZdS)FacNatural eye motion: blinks + subtle iris tremor. Final-pass post-processor for compiled blendshape targets. Adds: 1. Periodic blinks (Poisson-distributed, suppressed during emotional eye peaks) 2. Subtle iris drift (low-frequency smoothed noise on eye-look channels) Deterministic per scenario: same scenario_id always produces the same eye motion. ) annotationsN) NAME_TO_IDX eyeBlinkLeft eyeBlinkRighteyeLookDownLefteyeLookDownRight eyeLookInLefteyeLookInRighteyeLookOutLefteyeLookOutRight eyeLookUpLefteyeLookUpRight eyeSquintLefteyeSquintRight eyeWideLeft eyeWideRightseed_strstrreturnnp.random.GeneratorcCs.t|d}tjt|dddS)z;Deterministic RNG from arbitrary string (e.g. scenario_id).zutf-8N)hashlibmd5encode hexdigestnprandom default_rngint)rhr!S/dataset/kemix-engine/package/face/animasync-face-v3/scripts/compiler/eye_motion.py _seeded_rng!sr#?Trrngfreq_hzfloatfps np.ndarrayc Cstdt||}|||}tt|dd}||d|d| |}|d|}|}|dkr;||}|tjS)uGenerate length-T noise band-limited around freq_hz. Method: white noise → moving average whose window matches the period. Result drifts smoothly at ~freq_hz, no high-frequency jitter. rNgư>) maxrstandard_normalrcumsuminsertstdastypefloat32) r&r'r(r*windowrawcssmoothedsr!r!r"_smoothed_noise's  r9cCstjgdtjdS)uS5-frame triangular blink shape, peak 0.70 at center frame. Was 0.95 (ARKit-standard "fully closed eye"). Visual review on the Ready Player Me / Wolf3D avatar showed 0.95 looked like squeezing the eyes shut tightly — not a natural blink. The avatar's visible blink lives on the `eyesClosed` morph; at 0.95 that morph is near its tight- closure end of travel. Peak 0.70 gives a relaxed natural blink shape that reads correctly on this avatar. Values scaled proportionally from the original [0.30, 0.75, 0.95, 0.75, 0.30] by 0.70/0.95 = 0.737, then rounded. ))\(?皙?gffffff?r;r:)dtype)rarrayr3r!r!r!r" _blink_kernel;s r>@333333?targetmean_interval_s wide_suppresssquint_suppressc s<|jdt|d}tt}|d||}ttdtd|} t|tdtd| } d} dfd d } | kry| | r[| d7} | |} t t | d |d|} | t| 7} | ksS| dkrtd |kr| t D] }| t|rSqS)zAdd Poisson-distributed blinks. Only suppressed when the preset is holding the eyes very wide (strong surprise) or hard-squinted (peak anger/laughter). Moderate squint (anger ~0.6) still allows blinks.rz::blink333333?g@rcenterrrboolcstd|t|d}}t||tf||tf}t||tf||tf}|krz|krztD]0\}}||}d|kr[krwnqGt|tf||tf<t|tf||tf<qGdSdS)NrrTF) r-min IDX_WIDE_L IDX_WIDE_R IDX_SQUINT_L IDX_SQUINT_R enumerate IDX_BLINK_L IDX_BLINK_R)rGlohiwidesquintikfir&halfkerneloutrDrCr!r" _try_blinkas"** zadd_blinks.._try_blinkr$g @N)rGrrrH)copyshaper#r>lenr-rintegersrI exponentialr)rclip permutationarange)rArr*rBrCrDr'klen mean_frames first_maxt blink_countr\gaprGr!rXr" add_blinksKs2  (      rk~jt?Mb?rF horiz_ampvert_ampdrift_hzcCs|}|jd}t|d}t||||d|} t||||d|} t| dd} t| dd} t| dd} t| dd}t|ddtf| |ddtf<t|ddtf| |ddtf<t|ddt f| |ddt f<t|ddt f| |ddt f<t|ddt f| |ddt f<t|ddt f| |ddt f<t|ddt f||ddt f<t|ddtf||ddtf<t|ddtjS)zAdd subtle continuous iris drift on the eyeLook channels. Both eyes stay synchronized (no cross-eyed). Amplitude is intentionally small (~0.04 of [0,1] range) so it reads as life, not as looking around. rz::iris)r(r*Ngg?)r]r^r#r9rrbmaximum IDX_LOOK_IN_LIDX_LOOK_OUT_RIDX_LOOK_OUT_L IDX_LOOK_IN_R IDX_LOOK_UP_L IDX_LOOK_UP_RIDX_LOOK_DOWN_LIDX_LOOK_DOWN_Rr2r3)rArr*rnrorpr[r&r'r vh_posh_negv_posv_negr!r!r"add_iris_tremors$  $$$$$$$$rblink_interval_siris_horiz_amp iris_vert_ampcCs&t||||d}t|||||d}|S)z6Convenience wrapper: blinks + iris tremor in one pass.)r*rB)r*rnro)rkr)rArr*rrrr[r!r!r"apply_eye_motions r)rrrr)r$r%) r&rr'rr(r)r*rrr+)rr+)r%r?r@r@)rAr+rrr*rrBr)rCr)rDr)rr+)r%rlrmrF)rAr+rrr*rrnr)ror)rpr)rr+)r%r?rlrm)rAr+rrr*rrr)rr)rr)rr+)__doc__ __future__rrnumpyr constantsrrOrPrxryrrrurtrsrvrwrLrMrJrKr#r9r>rkrrr!r!r!r"sL     6 &