o jl@sdZddlmZddlmZmZmZddlZddl m Z dZ dZ dZ d Zd.ddZd/ddZd0ddZd1ddZ d2d3d d!Z "d4d5d&d'Z ( d6d7d,d-ZdS)8a%Brow + eyeSquint tremor baked into training targets. Direct Python port of the runtime tremor in `tools/blendshape-player.html` so dataset `.npz` targets contain the same micro-motion that we've been judging in the viewer. Channels: - 0, 1: browDownLeft/Right (silence-gated) - 2: browInnerUp (silence-gated) - 3, 4: browOuterUpLeft/Right (silence-gated) - 18, 19: eyeSquintLeft/Right (always-on) Bilateral pairs share a single noise stream so L/R move together (orbicularis oculi tone, brow micro-furrowing). ) annotations)ListOptionalTupleN)gaussian_filter1d)r))sstrreturnintcCs&d}|D] }|t|Add@}q|S)zFNV-1a-ish 32-bit hash. Mirrors player's `_hashStr` so identical scenario IDs produce identical noise streams in Python and JS.l9i)ord)r hcrO/dataset/kemix-engine/package/face/animasync-face-v3/scripts/compiler/tremor.py _hash_strsrseedcs|d@dfdd }|S)zADeterministic uniform-[0,1) PRNG matching player's `_mulberry32`.rrfloatcsddd@}||d?Ad|Bd@}|||d?Ad|Bd@|A}|d@}||d?Ad@dS) Niy+mrr=gAr)tstaterrrand,s  z_mulberry32..randN)rrr)rr!rrr _mulberry32(s r"n np.ndarraycCsrtj|tjd}t|D]*}d}|dkr|}|dks|}ttdt|tdtj|||<q |S)zDBox-Muller on top of a uniform PRNG (matches player's `_gaussRand`).dtypegg@) npemptyfloat32rangersqrtlogcospi)rngr#outiu1u2rrr_gauss_rand_stream;s 2r5xsigmarcCsRt|tjt|dd}ttt||d}|dkr#||}|tjS)zPlayer's `_gaussianSmooth(input, sigma, true)`: Gaussian filter with edge-clamp boundary + per-stream RMS normalization to unit RMS.nearestr7modegư>r)rastyper(r*rr,mean)r6r7r1rmsrrr_smooth_rms_normalizeGs r>?T scenario_iddictc Csht|}t|dAt|dAt|dAt|dAd}i}|D]\}}t||}t||||<q!|S)zFour bilateral-correlated noise streams seeded from scenario_id. Each stream is unit-RMS after Gaussian smoothing. Keys: 'down', 'innerUp', 'outerUp', 'eyeSquint'.iiii)downinnerUpouterUp eyeSquint)rr"itemsr5r>) r@rAr7rrngsstreamskr0rawrrrgenerate_noise_streamsRs     rLwavsrfpscCstdt||}tj|tjd}d}d}t|}t|D]D} | |} t| ||} | | krC|| | } tt t | | d} nd} dtt | }||||}tt |dd|| <qt |d d d tjS) uPer-frame soft silence indicator (0=speech, 1=silent). Mirrors player's `_computeSilenceGate`: RMS per hop converted to dB, soft sigmoid around -45 dB ± 12 dB, then σ=6 Gaussian smooth. rr%gFg(@g& .>g4@r'?g@r8r9)maxrr(zerosr*lenr+minrr,r<log10cliprr;)rNrOr@rPhoprK SILENCE_DB SOFT_RANGEr#r2absegr=dbgrrrsilence_gate_from_waves   r`y&1?target silence_gateampcCs|dks|dkr |S|jd}|jd|kr;|jd|kr5||jd}t|tj||dtjdg}n|d|}t|||d}|jtjdd}||} | |d tj} | |d tj} | |d tj} ||d tj} tD]}|dd|f| |dd|f<qx|ddtf| |ddtf<t D]}|dd|f| |dd|f<qt D]}|dd|f| |dd|f<qt |dd tjS)zBake tremor into `target` (T, 52) in-place-equivalent (returns new array). Brow channels: jitter scaled by amp * silence_gate * filtered_noise. EyeSquint channels: always-on, amp * filtered_noise (no gating). Clamped to [0, 1]. r'rr%N)r7T)copyrCrDrErFrQ) shaper( concatenatefullr*rLr; BROW_DOWN_L_R BROW_INNER_UPBROW_OUTER_UP_L_REYE_SQUINT_L_RrW)rbrcrArdr7r@padrIr1r j_downj_innerj_outerj_eyechrrr apply_tremors2   " ""rt)r rrr)rr)r#rrr$)r6r$r7rrr$)r?)r@rrArr7rrrB)rM) rNr$rOrr@rrPrrr$)rar?) rbr$rcr$rArrdrr7rrr$)__doc__ __future__rtypingrrrnumpyr( scipy.ndimagerrjrkrlrmrr"r5r>rLr`rtrrrrs(