`gjObUdZddlmZddlZddlZddlZddlmZddlm Z m Z eje Z e Ze ZgdZded<eeZed eDZeezZeezZeDcic] \}}}|| c}}}Zd ed <d Zd ZdZdZddZd?d@dZ dZ!ddddAdZ"dZ#dZ$dZ%dZ&dBdZ'dZ(d=dZ)dCdZ*dDd Z+dEd!Z,d"d#dFd$Z-dGd%Z.d"d#dFd&Z/d'd(dHd)Z0dId*Z1dJd+Z2dKd,Z3dd-d.d/ dLd0Z4dMd1Z5dNd2Z6ed3dOd4Z7d.d5dPd6Z8d7Z9dQd8Z:dRd9Z;dSd:Zr8s 1 1ee 1szdict[str, int] FRAME_COUNTS g ףp= ?cntj||dz dz||dz dzz||dz dzzS)Nrr r )mathsqrt)rgbkeys r_color_distancer*NsB 99a#a&jQ&!c!f*)::a#a&jQ=NN OOc|jdj}|dtkDry|jdj}t |dtdz}|j |j z}||dzkDS)z:True if the strip already carries a real alpha background.ArFNr g?) getchannel getextrema _ALPHA_FLOOR histogramsumwidthheight)imageextremahist transparenttotals r_has_transparencyr:Rs{s#..0GqzL    C * * ,Dd-\A-./K KK%,, &E  %%r+c,ddlm}|j|j}}|j }|}d|dz dfd|dz f|dz |dz ffD]+\}}|||f\}} } } | t kDs||| | fxxdz cc<-|sy|j dddS)z@Sample the four corners and return the most common opaque color.r)Counterrrr )rr) collectionsr<r3r4loadr0 most_common) r5r<whpxcounterxyr&r'r(as r_dominant_corner_colorrI^s# ;; qA ByG!a%aQZ!a%Q@$11X 1a |  Q1I ! # $    q !! $Q ''r+c  ddlm}|j\}}}}|\ |j|j|j fd|j fd|j fdS)a)An ``L`` mask, 255 where a pixel is within *tol* per-channel of *key*. Tight on purpose: it only marks near-pure backdrop so trapped chroma pockets seed the flood, while chroma-*tinted* character pixels stay outside it. Built with channel point-ops (fast C), no per-pixel Python. r) ImageChopsc.t|z krdSdSNr>rabs)vkrtols rz _near_key_mask..{SR[C%7cQr+c.t|z krdSdSrMrN)rPkgrRs rrSz _near_key_mask..|rTr+c.t|z krdSdSrMrN)rPkbrRs rrSz _near_key_mask..~sQV!3#r+)PILrKsplitdarkerpoint) r5r)rRrKr&r'r(_arXrVrQs ` @@@r_near_key_maskr^nsp++-KAq!RJBB    GG> ? GG> ?  :;  r+cddlm}|j|jdj |j d|S)uShave the 1px antialiased edge ring left after keying. Chroma keying can't catch the antialiased band where the sprite meets the backdrop — those pixels are a key/sprite blend, too far from the key to be removed, so they ring the cutout in magenta/green. Erode the alpha by one pixel (a 3x3 min filter) to drop that contaminated ring; the sprite's own thick dark outline keeps the silhouette intact. Built on a C-level filter, no per-pixel Python. r) ImageFilterr-r )rYr`putalphar.filter MinFilter)rgbar`s r _defringeres7 MM$//#&--k.C.CA.FGH Kr+gV@) chroma_key thresholdc bddlm}ddlm}m}|j d}t |r t|S|xs t||j|j}}|jd fd } ttz dk\rxt|} |jdj!d} |j#| | } |j%|j'd|j(d || } t+| St-||z}t-||z}|}t/|D]B}d|d z fD]6}| ||s |||z|zrd |||z|z<|j1||f8Dt/|D]B}d|d z fD]6}| ||s |||z|zrd |||z|z<|j1||f8Dttz dk\rRt3t|j5D],\}} | s ||rd ||<|j1||z||zf.|r|j7\}}d |||z|z<|d z|f|d z |f||d zf||d z ffD]S\}}d|cxkr|ksnd|cxkr|ks!n$||z|z}||r2d ||<| ||sA|j1||fU|r|j9d ||ft;|j!d }t+|j%|j'd|j(d ||S)uReturn *image* (RGBA) with its flat background keyed out to transparent. If the strip already has a transparent background we leave it alone; else we key out *chroma_key* (or the dominant corner color when not given) via a **border flood-fill**: only background-coloured pixels *connected to an edge* are removed. A global color match (the old approach) punched holes in the pet wherever an interior highlight happened to match the backdrop — e.g. a pug's light belly against a near-white background — which then showed through as the window behind. Flood-fill keeps those interior pixels because they aren't reachable from the border without crossing the (non-background) pet. rdeque)ImagerKRGBAcT||f\}}}}|tkDxrt|||kSr)r0r*) rFrGr&r'r(rHr)rDrgs r_is_bgz!remove_background.._is_bgs81X 1a<NOAq!S$AY$NNr+xr-c|tkDrdSdSrMr0)rHs rrSz#remove_background..sQ=McSTr+rrrrr Lc|rdSdSrMr)rPs rrSz#remove_background..s asUVr+rFintrGrvreturnbool)r?rjrYrkrKconvertr:_repair_internal_alpha_holesrIr3r4r@maxminr^r.r\r[ compositenewsizere bytearrayrangeappend enumerategetdatapopleft frombytesbytes)r5rfrgrjrkrKrdrBrCrnnearopaque remove_maskkeyedvisitedremovequeuerFrGinxnyidxmaskr)rDs ` @@rremove_backgroundrs;"% == D+D11  4.t4C ::t{{qA BO 3x#c(c!dC(%++,TU ''f5  &$))\ JDR]^AGq1u F$)GE1X%QU %Aa|GAEAI$6%&A " aV$ %% 1X%QU %Aa|GAEAI$6%&A " aV$ %% 3x#c(c! c!:!B!B!DE .GAtGAJ  a!eQ!V_- . }}1q1uqyAqzAE1:1q5zAq1u:F /FBB{{qB{{1frks|#$GCLb"~ b"X.  /  ??3Af 6 < <=V WD U__UYYvtyy,%OQUW[\ ]]r+ch ddlm|jd}|j\|j t zd fd d fd }t D]-}ddz fD]!} ||s |z|zr|||#/t D]-}ddz fD]!} ||s |z|zr|||#/d fd }d fd }tD]A\}}|r |z}|z} ||s |||} || } | D] \} } | | | f<C|S) aFill transparent islands fully enclosed by opaque sprite pixels. Some providers return "transparent" PNGs with swiss-cheese alpha inside the character. Border flood-fill cannot see those because there is no opaque backdrop to key, so repair the alpha mask itself: transparent components that touch an image edge remain background; transparent components enclosed by the sprite are filled with the average color of their opaque neighbours. rrirlc&||fdtkS)Nr rq)rFrGrDs r_is_transparentz5_repair_internal_alpha_holes.._is_transparents!Q$x{l**r+cD ||fg}d | z|z<|r|j\}}|dz|f|dz |f||dzf||dz ffD]S\}}d|cxkr ksnd|cxkr ks!n$| z|z} |r2||s._mark_border_components(-Bxj(9 Q ==?DAqE1:AqzAq1u:1q5zJ /B;Q;1;Q;q&2+C"3._collect_holes(-Bxj(9 Q (*==?DAq MM1a& !E1:AqzAq1u:1q5zJ /B;Q;1;Q;q&2+C"3c3&K|] }|d ywrNrrcs rrzD_repair_internal_alpha_holes.._fill_color..,,qad,c3&K|] }|d ywr Nrrs rrzD_repair_internal_alpha_holes.._fill_color..-rrc3&K|] }|d ywr Nrrs rrzD_repair_internal_alpha_holes.._fill_color...rrr>)setr0rroundr2len)holesamplesseenrFrGrrr&r'r(rHrCrDrBs r _fill_colorz1_repair_internal_alpha_holes.._fill_color s.04y 2DAqE1:AqzAq1u:1q5zJ 2B;Q;1;Q;B843G!#BFJAq!Q<'1ay1  2 2 ! #,G,,s7|; < #,G,,s7|; < #,G,,s7|; <    r+ru)rrvrrvrwNone)rrvrrvrwlist[tuple[int, int]])rrrwztuple[int, int, int, int])r?rjryrr@rrr)r5rdrrFrGrrstartrrcolorhxhyrrjrCrDrrBs @@@@@@rrzrzsx" == D 99DAq BAG+ / /1X-QU -Aq!$WQUQY-?&q!, --1X-QU -Aq!$WQUQY-?&q!, --   $g& q 5>  AI QJq!$ Q"D! FBBr2vJ   Kr+c ddlm}|jdttfd}t |}|j }||S|j|}ttz }ttz }t||jz ||jz d}|dk7rh|jtdt|j|ztdt|j|zf|jj }t|jz dz}t|jz dz} |j#||| f|S)zGCrop to content, scale to fit a padded cell, and center on transparent.rrkrlrrg?r r )rYrkr~ CELL_WIDTH CELL_HEIGHT_drop_side_bleedgetbboxcrop _CELL_PADr|r3r4resizer{r ResamplingNEARESTalpha_composite) r5rktargetbboxspritemax_wmax_hscalelefttops r _fit_to_cellrCs YYv K8, GF U #E ==?D | ZZ F  "E ) #E  $efmm&;S AE |  E&,,./ 0#av}}u?T9U2V W    $ $   %! +D  &1 ,C 6D#;/ Mr+c ddlm}|jd}|j\}}t |}t |}t |dkr|S|Dcgc]\}}t|||} }}t| tz} t|| D cgc] \} } | | k\s | } } } t | t |k(r|S|j}|jd||fdd}}| D]3\}}||kDr'|j|j|d||f|df|}5||kr'|j|j|d||f|df|Scc}}wcc} } w)aRemove tiny separated left/right lobes before fitting a frame. Frogger showed the failure mode: a good centered pose plus a thin vertical sliver from the neighbouring pose. By the time it reaches a cell, that sliver may be close enough to the subject that component extraction already grouped it. A horizontal alpha projection still reveals it as a small side lobe with a low mass compared to the main silhouette. Drop only those low-mass lobes; keep large lobes so wide poses and real limbs survive. rrrlr rr)rYrkryr_column_profile _content_runsrr2r{_SIDE_LOBE_RATIOzipcopyr~paster)r5rkrdrBrCprofilerunslr&masses keep_massrunmkeepcutprevrrights rrr`sS == D 99DAqd#G  !D 4y1} ,0 1DAqc'!A, 1F 1F ..I!$/ BFC1 >C BD B 4yCI  99;D &1a&,7C e $; JJsxxq$ 23dAY ? ax 388T1aO,tQi8 K2 BsE  EEc ddlm}|jdj}|j\}}|j dd d}t |Dcgc](tfdt |D|dzk\r*}}t |Dcgc](tfdt |D|dzk\r*} }|jd|jd } ||D],\} } |j| jd| || fd| f.|| D],\} }|j| j| d||f| df.|Scc}wcc}w) ajRemove thin slot-spanning guide/floor/divider lines. Gemini will sometimes satisfy "baseline" / "cell" language by drawing literal horizontal floors or vertical panel dividers. They survive chroma keying and connect otherwise clean poses. Drop only *thin* rows/columns that span nearly the whole slot; thick sprite body rows are left alone. rrrlr-cg}d}d}|D]?}||x}} | ||dzk(r|}|#|!||z dzdkr|j||dzf|x}}A|#|!||z dzdkr|j||dzf|S)Nr r)r)indicesgroupsrrrs r _thin_groupsz,_erase_long_axis_lines.._thin_groupss(*  C}""C4!8O T%5$,:Ja:O udQh/0 ED   !1dUlQ6F!6K MM5$(+ , r+c3XK|]!}j|ftkDsd#ywrgetpixelr0)rrFalpharGs rrz)_erase_long_axis_lines..%LQennaV&<|&KqL**g333333?c3XK|]!}j|ftkDsd#ywrr)rrGrrFs rrz)_erase_long_axis_lines..rrrr)r list[int]rwr) rYrkryrrr.rr2r~rr)r5rkrdrBrCrrG wide_rowsrF tall_colsclearrbottomrrrs ` ` @r_erase_long_axis_linesrs` == % % 'D 99DAq OOC E(q L%(L LPQTXPX X Iq L%(L LPQTXPX X I IIfdii 6E#I.> V 5::q#q&12QH=>#I.? e 5::tQq12T1I>? K! s -E-Ecnddlm}|jd}|j}|gS|\}}}}||z ||z } }|j dj } t || z} g} t|| zD]7} | | r | |z| |z}}||z||z}}d| | <| ||ftkr2|||fg}|x}}|x}}d}|r|j\}}|dz }t||t||}}t||t||}}|dz|f|dz |f||dzf||dz ffD]^\}}d|cxkr|ksnd|cxkr| ks!n$||z|z}| |r2d| |<| ||z||zftkDsL|j||f`|r| j||z||z||zdz||zdzf|f:| S)a(Connected opaque components as ``[(bbox, mass)]``. A full ML segmenter would be overkill here: after chroma keying, "the pet" is the dominant connected alpha component inside each known slot. Tiny detached sparkles, tears, UI dots, and neighbour slivers are separate components. rrirlr-r ) r?rjryrr.r@rrr0rr|r{r)r5rjrdrl0t0r0b0rBrCrroutrrraxayrrrrrmassrFrGrrrs r_component_boxesrs" == D <<>D | NBB 7BGqA OOC % % 'EAG79Cq1uS 5> EQJBb"r'B R=L ( (-Bxj(9uf==?DAq AIDdA,E1 %Dc1+s61~CE1:AqzAq1u:1q5zJ 3B;Q;1;Q;q&2+C"3<'(  b"r'!12\A!LL"b2  3  R$YS"u*q."v+/JDQR3S4 Jr+c nddlm}t|}t|}|s|St |d\}}|\}}}} t d||z } g} |D]\} } | |k(r| j | | \}}}}t dt ||t ||z }||zdz }|| dzz |cxkxr || dzzknc}| t d|d zk\s~|| d zk\s|s| j | |jd |jd }| D]+} |j|j| | d| df-|S) z:Keep the slot's real subject; drop detached effects/noise.rrc |dSNr ritems rrSz'_isolate_slot_subject..s d1gr+r)r r g?gQ?g333333?rlrr) rYrkrrr{rr|r~rrr)r5rkrdcompsmain_box main_massmlmtmrmbmwrboxrr_topr_bottomoverlapcenter_x near_mainrs r_isolate_slot_subjectrsU !% (D T "E  e)=>HiNBB QRB,.D  T (? KK  %("dE7aUB#dB-785LA%"t)^DR"t)^D  3r9u,- -7b3h3F) KK   ))FDII| 4C> DIIcNSVSV,<=> Jr+c |j}|y|j\}}|\}}}}tdtdt |dz}tdtdt |dz} ||k\xr|| k\xr||z |k\xr||z | k\S)z8True when content has empty room on all four slot edges.Fr g?r {Gz?)rrr{r|r) r5rrBrCrrrrmin_xmin_ys r_has_slot_paddingrs ==?D | ::DAq#D#uf 3r5U+, -E 3r5T?+ ,E 5= XSE\ Xa%i5.@ XQZSXEXXr+ct|Dcgc]'}t||z|z t|dz|z|z f)c}Scc}wr)rr)r3 frame_countrs r _slot_boundsrsN{#   q5y;& 'A/L)MN  s,=cX|sgStd|D}td|t|dzdz}g}g}t|dD]}|d|dzdz }t|D]S\}}t ||z |ks||j |t d ||Dt||z ||<q|j |g|j |tt||d d D cgc]\} } |  } } } | D]} | jd | Scc} } w)u?Group component boxes into visual rows, then sort left→right.c3FK|]}td|d|dz yw)r r N)r{rr(s rrz(_group_component_rows..s#8QSAaD1Q4K(8s!rr g?c|d|dzdz S)Nr r r rr(s rrSz'_group_component_rows..#sAaD1Q4K1+<r+rr r c38K|]}|d|dzdz yw)r r r Nrrs rrz(_group_component_rows..(s! Dq!A$1+!2 DsF)strictc |dS)Nrrrs rrSz'_group_component_rows..-s aefgahr+c|d|dzdz S)Nrr rrs rrSz'_group_component_rows../s!qt q0r+) sortedr{rrrOrr2rsort) boxesheightsrow_tolrowscentersr cyrcenter_centerrowordereds r_group_component_rowsr0sA  8%88G"gc'la/0478G24DGe!<= !fs1vo ""7+ IAv2;7*Qs# DDG DDs4PQ7|S    KK  NN2  (.c'4.NTh'ij|wsjGj2 012 Nks; D&c t|}d}|rVd}g}dgt|z}t|D].\}}||r |\}}}} d||<t|dzt|D]} || r || \} } } }t dt | |t || z }t dt | |z || z }t dt || t || z }t dt ||z | | z }||dzk\s|t d|dzkst || t || t || t | |f\}}}} d|| <d}|j |||| f1|}|rV|S)abMerge disconnected parts that clearly belong to one subject. Capes, tails, horns, and held props sometimes key as separate components. Merge components on the same visual row when their vertical spans overlap and the horizontal gap is tiny compared with the component size. Do not bridge the much larger gaps between separate poses. TFr rg?r"g)\(?)listrrrr{r|r)r&changedmergedusedrrHalatarabjblbtbrbb v_overlapmin_hgapmin_ws r_merge_related_boxesrC3s KEG 24wU#e$ ,DAqAwNBBDG1q5#e*- #7!&qBB3r2;R#<= As27BG45!SR[3r2;67As27BG45 ,B 8M1M%(R[#b"+s2r{CPRTVK%WNBB"DG"G # MM2r2r* +# ,$- . Lr+Frequire_paddingcZddlmdfd }||xs|t|S)aExtract frame subjects as connected non-background objects. This is the robust path for models that ignore "one horizontal row" and emit a 2D sprite grid. We count real opaque subject components, discard tiny detached effects, sort in reading order, and return exactly *frame_count* frames. Slot slicing is only a fallback when object detection can't satisfy the contract. rrc pt|}|sytd|D}t|Dcgc]\}}|td|dzk\s|c}}}t|kryt |}|Dcgc] }|D]}| c}}d}t|kryrtdt dt |jdz} tdt dt |jdz} |D]7\} } } }| | ks*| | ks%|j| z | ks|j|z | ks7yt|d kD}g}|D]f\} } } }td t | | z d z}td t || z d z}|rOtd | |z td | |z t |j| |zt |j||zf}nTd k(rd d |j|jf}n4td | |z d t |j| |z|jf}jd |d|d z |d|d z fd}| |d z | |d z | |d z ||d z f}|j|j| | | |f|d |d f|j|i|Scc}}wcc}}w)Nc3&K|] \}}| ywrr)r_boxrs rrz4_component_crops..attempt..gs.WT1q.r@Q?rrg{Gz?r gQ?r r g{Gz?rrlr r rr) rr{rCrr0r|rr3r4r~rrr)sourcermax_massr rsubjectsr)r.r/rrrrrrmultirowframespad_xpad_ycrop_boxframerelrkrrEs rattemptz!_component_crops..attemptbs (...'e(hdtsSUW_bfWfOgGg(hi x=; &$X."&63#633636| D w<+ % 3r5)<#=>?E3r5)>#?@AE,3 (c5&%<3;&,,2F2NRXR_R_bhRhkpRp t9q=(/ ! $D#uf5%$,$!678E5&3,$!678E4%<(3;' eem4 v~6  !q&,, > 4%<0!Suu}5UW]WdWdeIIfx{Xa['@(1+PXYZP[B[&\^jkE(1+%sXa['8%(1+:MvX`abXcOcdC  ! !&++tS%.H"ICPQFTWXYTZK[ \ MM% - !. S)i 7sJ, J, 2J2)rw list | None)rYrkr)striprrErVrks `` @r_component_cropsrYWs*/b 5> CW%;E%BCCr+c |dkr|S|j}|j}|j|z }tdt dt |dz}t d|D]}}t ||z}td||z }t |j||zdz} t || D]3} t |jD]} || | f\} } }}| | |df|| | f<5|S)aCut thin vertical gutters at expected frame boundaries before labeling. Generated rows often have a shared shadow, glow, motion smear, or 1px bridge that connects neighbouring poses. Component detection then sees one giant blob and either fails or falls back to slot slicing. We know the requested frame count, so cut a very narrow transparent band at each expected boundary before connected-component labeling. If a pose truly overlaps the boundary, losing a few pixels is better than exporting merged frames. r r gQ?r)rr@r3r{r|rrr4)rXrrrDslothalfrrFrrgxgyr&r'r(r]s r_sever_expected_guttersr`sa **,C B 99{ "D q#b%t ,- .D 1k "* !d(O1a$hCIIq4x!|,e$ *BCJJ' * Rj 1aAq\2r6  * * * Jr+c |j}g}t|j|D]L\}}tt |j |d||f}|r t |sy|j|N|S)aSlice *strip* into *frame_count* uniform columns (one coordinate space). Equal-width columns keep every frame in a single shared coordinate frame, so a later union-crop + shared placement (:func:`normalize_cells`) preserves the row's real motion without the per-frame re-centering that makes a pet visibly slide. Each slot is cleaned independently so detached effects, floors, dividers, and neighbour slivers do not become "frames". rN)r4rr3rrrrr)rXrrErCrPrrr\s r _slot_cropsrbst  A F#EKK= e 5ejj$5RSAT6U VW #4T#: d  Mr+r )rgcg}d}tt|dgzD](\}}||kDr||}||j||fd}*|S)zContiguous column spans whose alpha mass exceeds *threshold*. A column-projection of the alpha mask: empty (background) columns separate one pose from the next, so the runs ARE the candidate frames. Nr)rr2r)rrgrrrFrPs rrrsd #%DE$w-1#-.1 y=}   KK #E  Kr+c 6 t|}t|}|sy|Dcgc]\}}t|||}}}t|dz}t ||D cgc] \}} | |k\s |}}} t ||kry|Dcgc] \}}||g c}} t |kDrKt tt dz  fd} | dzd | d< | dz=t |kDrK Dcgc] \}}||f c}}Scc}}wcc} }wcc}}wcc}}w)uQPer-frame ``(left, right)`` column ranges from the row's empty gutters. The standard sprite-sheet slice — once poses are separated by real gaps (which generation now enforces), splitting is just "find the empty columns": * spans == frames → one span per frame. * spans > frames → merge across the smallest gaps. A detached halo/ear sits a tiny gap from its body, while the inter-pose gutter is the big gap that survives — so over-segmentation (and any over-eager gutter sever) repairs itself by collapsing only the small internal gaps. * spans < frames → poses are touching; not separable by gutters (the caller raises for ``components`` or falls back to even slots for ``auto``). Ranges span content only; the caller crops full cell height, so tall ears / halos are never cut. Nrr c,|dzd|dz Srr)rrs rrSz!_frame_x_ranges..s#va!e}Q7G&QR)TU,7Vr+r)rrr2r{rrr|r) rXrrrrr&rfloorrrgirs @r_frame_x_rangesrhs""e$G  !D -1 1DAqc'!A, 1F 1 K$ E!$/ >FC1:C >D > 4y;!% &Aq!f &F f+ # s6{Q'-V WrAvq)r 1 26N f+ # & &tq!QF &&2 >' 'sD D &D D1Dc t|}|sgStd|D}t|Dcgc]\}}|td|dzk\s|c}}Scc}}w)Nc3&K|] \}}| ywrr)rrIrs rrz-_significant_subject_boxes..s0JD$40r rK)rr{rC)r5rrMr rs r_significant_subject_boxesrlsT U #E  0%00H e `dts2xZ^G_?_ ` aa `s A A ct||k7rtd|dt|g}t|D]_\}}|j}|td|dt |}t|dk\rtd|d|j |a|dkryt d |D}t d |D}td|t|d z} td|t|d z} t|D]A\}\} } } }| | z }|| z }|t| d z| d zkDs+|| dzks4td|dy)awReject rows where one "frame" is really multiple poses. A bad provider roll can collapse a strip into tiny repeated poses. If we let that through, normalization sees a huge motion envelope and shrinks the entire pet to postage-stamp size. Catch the row here so hatch can regenerate it instead of saving a technically non-empty but visually broken atlas. expected z frames, got Nzframe z is emptyr z% contains multiple separated subjectsr c32K|]}|d|dz yw)r rNrrs rrz-_validate_extracted_frames..s/AAaD1Q4K/c32K|]}|d|dz yw)r r Nrrs rrz-_validate_extracted_frames..s0QQqTAaD[0rpr @`皙?z is a multi-pose width outlier)r ValueErrorrrrlrr$r{)rPrr&rrTrrNwidthsr'med_wmed_hrrrrr3r4s r_validate_extracted_framesryst 6{k!9[Ms6{mLMM Ef%5}} <vaS 23 3-e4 x=A vaS(MNO O Ta // /F0%00G 6#f+*+ ,E 73w<1,- .E)25)9I% %D#uf # 3us{EBJ/ /Feck4IvaS(FGH HIr+autoT)rfmethodfitcddlm}t|ttfr,|j |5}|j d}dddn|j d}t||}t||d}|t||d}|"|dk(rtd |d t||d }||}t||} | t||}t||} | t||d xsg}n|j} td td t!|j"td|z dz} | D cgc]P\} } t%t'|j)td| | z dt|j"| | z| fR}} } t+|||r|Dcgc] }t-|c}S|S#1swYcxYwcc} } wcc}w)uTurn one generated row strip into *frame_count* frames. The background is keyed out, then strict extraction treats the requested frame count as the source of truth: slice known equal slots, isolate the real subject in each slot, and require empty padding on X and Y. Empty chroma gutters are only a lenient salvage fallback. Each frame is cropped at full cell height so tall ears / halos are never clipped; detached effects and neighbour slivers are dropped per slot. When a pose does not have required space around it, ``components`` raises and ``auto`` falls back to best-effort slicing. *fit* (default) fits+centers each frame into a 192x208 cell — the standalone contract for callers that don't normalize. Hatching passes ``fit=False`` to keep raw, coordinate-aligned columns for :func:`normalize_cells`, which lays one shared scale + baseline across the whole pet (no slide, no size pulse). rrrlN)rfTrD componentszcould not segment z padded sprites from stripFr r r g{Gz?)rYrk isinstancestrropenryrrYrbrurhr`r4r{r|rr3rrrryr)rXrrfr{r|rkopenedrPrLrangesrCpadrrfs rextract_strip_framesr*s2%#t% ZZ  +&NN6*E + + f% e ;E e[$ GF ~UKF ~ \ !1+>XYZ Z "%eL ~ 5 >,UK@F$V[9F > eLRPRF AaR s1k7J(Jd'R!STUC$*D%!!6v{{C4RU:DVXY[^_e_k_kmruxmx[y{|C}7~!AFv{303V ,LO ,??O + +D -sF32AGG3F=cddlm}t|jdj |j df|j jS)uMPer-column alpha mass — collapse the frame to a 1px-tall strip (fast in C).rrr-r )rYrkr2r.rr3BILINEARr)r5rks rrrps>   %,,ekk1-=u~~NVVX YYr+c t|}d}d}t| |dzD]I}d}ttd|t|||zD]}||||||z zz }|||kDsF|}|}K|S)aInteger dx that best aligns *prof* onto *ref* by cross-correlation. This is 1-D phase correlation: the body is the dominant mass in the column profile, so the peak overlap locks onto the body and a flipping arm/cape (a small secondary bump) doesn't move the match. Proven on the jitter case to cut body drift from ~9px to ~1px where a centroid/bbox anchor cannot. Nrr )rrr{r|) refprofwindown best_scorebestdscorerFs r _best_shiftrws CA#J D F7FQJ 's1ay#aQ-0 *A SVd1q5k) )E *  !3JD  Kr+)rc 2.ddlm..fd}d}i}i}t|z }t|z }|j D]7\}} | D cgc]} | j d} } t d| Ds| D cgc] } | c} ||<Ptd| D} td| D}g}| D]M} | j| |fk7r).jd| |fd }|j| d |} |j| O|D cgc] } t| }} t| Dcgc]%tfd |Dt|d z'}}td | dz}|}g}t!||D]S\} }.jd| d |zz|fd }|j| |t#|||zdf|j|Ud|DDcgc]}|s| }}t%d|D}t%d|D}td|D}td|D}|||||f||Dcgc] }|d |dz c}||Dcgc] }|d|dz c}ff||<:|s|S|}|j'D]E\} \}}}}\}!}"||z ||z }$}#t%|||"ztd|$z ||"ztd|#z }G|j D]\}\}\}}}}\}!}"||z ||z }$}#|td|"z }%tdt)|#|%ztdt)|$|%z}'}&t)t|&z d z t)t|d zz |'z })}(g}*|D]|}+|+j+||||f},|,j|&|'fk7r(|,j-|&|'f.j.j0},|}-|-j|,|(|)f|*j|-~|*||<|Scc} wcc} wcc} wcc}wcc}wcc}wcc}w)udRegister every frame into a 192x208 cell — the deterministic anti-jitter math. A per-frame "crop→scale→center" pipeline jitters because a moving limb/cape shifts the bbox (or even the centroid) and a per-frame scale pulses the size. The rigorous fix, matching image-registration practice (phase correlation) and AI-sprite pipelines (perfectpixel-studio / sprite-gen): 1. **Cross-correlate** each frame's column profile against the per-state *median* profile to find the integer shift that locks the **body** in place — robust to limbs/cape because the body dominates the profile. 2. **Union-crop** through one shared state window, then scale every state by a single global factor keyed to its median pose height, so the character is the same on-screen size in every row while a jump's lift still fits. rrc>jdttfdS)Nrlrr)r~rrrsrrSz!normalize_cells..sEIIfz;&?Nr+c6t|t|dzS)Nr )r$r)vss rrSz!normalize_cells..sVBZB1 -r+rlc3<K|]}|jywrrrrs rrz"normalize_cells..s-1199;-c34K|]}|jywr)r3rs rrz"normalize_cells..s'Q'c34K|]}|jywr)r4rs rrz"normalize_cells..s(a(rrrr=c3(K|] }| ywrr)rprFs rrz"normalize_cells..s-qad-sr r rc3<K|]}|jywrr)rrHs rrz"normalize_cells..s:QQYY[:rc3&K|] }|d ywrrrs rrz"normalize_cells..s'A1Q4'rc3&K|] }|d ywrrrs rrz"normalize_cells..s&1!A$&rc3&K|] }|d ywrrrs rrz"normalize_cells..s(QAaD(rc3&K|] }|d yw)r Nrrs rrz"normalize_cells..s)aQqT)rr r )rYrkrritemsryanyr{rr~rrrrr$rrrr|valuesrrrrr)/frames_by_staterblankmedrpreparedtarget_wtarget_hstaterPrrdrw0h0canvasrprofilesrFrrmarginalignedrshiftedr(r&rrrrK_aligned_pose_wpose_huwuhrswshrDpycellsrHrcellrks/ ` @rnormalize_cellsrsl NE -CCSUHCHS H(..0' v+12a &!22---+12a%'2CJ '$' ' (4( ( Avv"b!IIfr2h =!!!V, MM!   1771OA&77KPQS9Uav-H--c(mq.@AUUQa68, $GAtiia&j"(=|LG  # #AS$1O(OQR'S T NN7 # $;':@qa@@'''&&&(%(()5))  3v & u-!!A$1+- .%4PQQqTAaD[4P0Q R G' R  AEM__EVSA-D#uf/@v|B 8f$s1bz18f3Ds1bz3Q RSLT>>K[GG3$UF5Fgvv|BC6N"Qb5j)*C5e3D,EB R1,-ukC1H6LPR5R/SB A664eV45DyyRH${{B8U-=-=-E-EF7D  Bx 0 LL  E  JI328UA.4Ps0O62O;P%*PP P 3PP)r|cddlm}t|ttfr+|j |5}|j d}dddt|}|r t|St|S#1swY,xYw)a&One frame from a standalone image (e.g. the base look). Used as an idle fallback so a pet always renders even if the idle row generation failed. *fit* yields a finished 192x208 cell; ``fit=False`` yields the raw keyed sprite for :func:`normalize_cells` to place with the rest. rrrlN) rYrkrrrrryrrr)r5r|rkrrs r single_framersh%#t% ZZ  +&NN6*E + e $E"%< B+;E+BB + +s A++A4c&ddlm}|jd}t|j }t dt |dD]#}||dzdk(sdx||<x||dz<||dz<%|jd|jt|S)zCZero the RGB of fully-transparent pixels (no colored-halo residue).rrrlrr r r ) rYrkryrtobytesrrrrr)r5rkrddatars r_clear_transparent_rgbrs == D T\\^ $D 1c$i #4 A;! 23 3DG 3d1q5kDQK4 ??6499eDk ::r+cddlm}t|d|j}|Dcgc]"}|j dj |$c}Scc}w)ulHorizontally flip each frame *in place* (RGBA-safe). Used to derive ``running-left`` from an approved ``running-right`` row. The flip is per-frame so the leftward loop preserves the rightward loop's frame order and timing — this is NOT a whole-strip reverse (which would play the animation backwards), matching the petdex/Codex mirror rule. rr Transposerl)rYrkgetattrFLIP_LEFT_RIGHTry transpose)rPrkfliprTs r mirror_framesrsC 5+u - = =D?E FeEMM& ! + +D 1 FF Fs'A cddlm}|jdttfd}t D]\}}}|j |xsg}t|d|D]\\}}|jd} | jttfk7r t| } |j| |tz|tzf^t|S)zPack per-state frame lists into the Hermes atlas (RGBA, residue-cleared). Missing/short states leave their trailing cells transparent; extra frames beyond a state's spec are dropped. rrrlrrN)rYrkr~ ATLAS_WIDTH ATLAS_HEIGHTrgetrryrrrrrr) rrkatlasrr.rrPcolrTrs r compose_atlasrs  IIf{L9< HE&OsE $$U+1r#F6EN3 OJC==(DyyZ55#D)  ! !$z)93;L(M N  OO "% ((r+cxtj}|j|ddddd|jS)zFEncode an atlas image to lossless WebP bytes (the on-disk pet format).WEBPTdr)formatlosslessqualityr{exact)ioBytesIOsavegetvalue)rbufs ratlas_to_webp_bytesr/s1 **,C JJs6D#atJT <<>r+c2ddlm}t|ttfr,|j |5}|j d}dddn|j d}g}g}|jttfk7rV|jdtdtd|jd|jd|j|j||gd Sg}i}tD]\}}} d} g} t| D]} | tz} |t z}|j#| || tz|t zf}t%|j'd j)d d}| |z } |j+}|{| j|| dkDr|j|| ||<|jd |d |s|jdt-d|j/D}t-d|j/D}d}d}|r\|rZ|t1|dz}|t1|dz}|}t3dt5t dz}||kr|jd|d|j7D] \}} t1| d krt-d| D}t-d| D}t3d |t1|dz}t3d |t1|dz}|d}|d}|t3|dz|dzkDr||dzkr|jd |d|s|st3dt5|dz}t3dt5|d z}||ks||ks|jd |d!|d|d"|d|d |j9} d}!tdt1| d#D])}"| |"d$zdk(s| |"s| |"d zs | |"dzs%|!d z }!+|!r|j|!d%| |j|j|||d S#1swYxYw)&u3Check geometry, per-cell occupancy, and transparency invariants. Returns ``{ok, width, height, errors, warnings, filled_states}``. Errors are blockers (wrong size, empty used cell, opaque/dirty transparency); warnings are soft (a whole state row blank — generation likely dropped a row). rrrlNrnrFz, got F)okr3r4errorswarnings filled_statesr-r zstate 'z' has no framesu/atlas is empty — no state produced any framesc3>K|]}|D]\}}}}||z ywrr)rr&rr rr s rrz!validate_atlas..fs9 */ 'D$w  c3>K|]}|D]\}}}}||z ywrr)rr&_leftr_rightrs rrz!validate_atlas..ks9 */ 'E3  rr 8gQ?zEatlas sprites are too small after normalization (median frame height zpx)c30K|]\}}}}||z ywrr)rrr rr s rrz!validate_atlas..}sM)CtUG Mc30K|]\}}}}||z ywrr)rrrrrs rrz!validate_atlas..~sN*D%ff#Nrrrrsrtz%' contains a multi-pose frame outlierrkgzG?(g?z' appears collapsed (median zpx, global median rr z& transparent pixels retain RGB residue)rYrkrrrrryrrrrr3r4rrrrrr2r.r1rr$rrr{rrr)#rrkrrrrcell_boxes_by_staterr.r row_pixelsr&rrrrnonblankr all_widths all_heights global_med_w global_med_hmedian_hr@rvr'rwrxrr min_state_w min_state_hrresiduers# rvalidate_atlasr6sz%#t% ZZ  +&NN6*E + + f%FH zzk<00 +a ~VEKK=PQRWR^R^Q_`aekkU\\U[iqEGH H!MFH&>sE 13< #C#D #C::tS$*;S;=NOPD4??3/99;AB?@H ( "J<<>D T" # >   ').  & OOgeWO< =!>$  GH (//1J (//1K LLk!#j/Q"67 s;/145 BkD012 e  MMabjakkno p+113 u u:? MuMMNNNAvc&kQ./0Aws7|q012r   3us{EBJ/ /EUS[4H MMGE7*OP Q Lb% t(;"<=Kb% t(;"<=K{"ek&9 eW$@qOabnaoopq}p~BC#, ==?DG 1c$i # A;! aDQK4A; qLG  !GHIj,,&  q + +s P  P) r&rvr'rvr(rvr)tuple[int, int, int]rwfloat)rwrx)rwr)0)r)rrRrv)rftuple[int, int, int] | Nonergr)rwz+list[tuple[tuple[int, int, int, int], int]])r3rvrrvrwr)r&list[tuple[int, int, int, int]]rwz%list[list[tuple[int, int, int, int]]])r&rrwr)rrvrErxrwrW)rrv)rrrgrvrwr)rrvrwzlist[tuple[int, int]] | None)rwr)rPr2rrvrwr) rrvrfrr{rr|rxrwr2)rwr)rrrrrrvrwrv)rdict[str, list]rrvrwr)r|rx)rPr2rwr2)rr)rwr)rwdict)>__doc__ __future__rrloggingr$pathlibragent.pet.constantsrr getLogger__name__loggerrrr__annotations__rROWSr{COLUMNSrrrr0r_NORMALIZE_PADrr*r:rIr^rerrzrrrrrrrr0rCrYr`rbrrhrlryrrrrrrrrrr)rrrs000rrs.#  0   8 $    ) %  9~ 1y 1 1 " k! DMNN5u N nN   P & ( (  KOcgS^lUv:"J0f-`@ Y0!HJO<D~8EJ&;<$"'Jb#IT/3 C@C@, C@  C@  C@ C@LZ*ES`L(, C ; G)&iu Os*D*