o  z$ensure_png_alpha..widthheightT)pixelsrr source_has_alpharrcSg|] \}}}}|||fqSrrrrrr_alpharrrr$cSs&g|]\}}}}|dkr|||fqS)rrrrrrr%s&css|] ^}}|dkVqdS)rNr)r_rgbrrrr 'z#ensure_png_alpha..png_already_has_alpha)rrrrr cSrrrrrrrr3r)r!rr r"has_trns bit_depth interlacerunsupported_png_format)rrrrr rr.cSg|] \}}}|||dfqSrrrrrrrrrPr&cSr2r3rr5rrrrXr&) _read_png ValueErrorstr_decode_rgba_pixels"_remove_explicit_chroma_backgroundintany_opaque_edge_background_palette!_global_chroma_background_palette!_remove_connected_edge_background_decode_rgb_pixels)rrrpngexcr! rgb_pixelsopaque_rgb_pixelsrrrensure_png_alpha s (       $       rEr!list[tuple[int, int, int, int]]rr;r r"booltuple[int, int, int]c Cst}d}d}|D]3\} } } } | dkst| | | f||kr'|d7}|dq t| | | f|r3|d7}|| | | | fq t||t||dkdd||||t|||d fS)NrrrrrTexplicit_chroma_key) rrmethodrr transparent_pixelsr"rrchroma_residue_pixels) bytearray _rgb_distanceextend_is_chroma_residue_pixel_write_rgba_pngr list) r!rr r"rrrgbatransparent_countrNrrroriginal_alpharrrr:_s,  r:c sdd|D}t|||}t|}|r8t|D]\}||r qtfdd|Dr/d||<qt|||||t}d} t|D]'\}\} } } } ||rOdn| }|dkr_| d7} |dqA|| | | |fqA| dk}t||t||dd ||| |d fS) NcSr#rrr$rrrrr&z5_remove_connected_edge_background..c3s(|]}t|dkpt|VqdS)`N)rP_is_chroma_shadow_pixelrcolorpixelrrr(s&z4_remove_connected_edge_background..TrrIrJedge_background_flood_fill)rrrLrr rMr") _find_connected_edge_backgroundr> enumerater<_remove_connected_chroma_halorOrQrSr )r!rr r"rC transparentdetached_chroma_paletteindexrUrVrrrrWrrrr\rr?s:  r?dict[str, Any]cCsR|ts tdtt}d}}}}}}}g} d} |dt|krtd|||dd} ||d|d} |d} | | }|| |}|d}| dkrctd |\}}}}}}}n| d krm| |n | d krtd } n| d kryn|dt|ks'd|||||||hvrtd|dks|dkrtd|||||d| | dS)Nnot_pngF >Ir+rr/IHDR>IIBBBBBIDATstRNSTIEND missing_ihdrunsupported_png_compression)rr r.rr0idatr,) startswith PNG_SIGNATUREr7lenstructunpackappendjoin)rpositionrr r.r compression filter_methodr0 idat_partsr,length chunk_type data_startdata_end chunk_datarrrr6sH   r6rAlist[tuple[int, int, int]]c Cst|d}t|d}d}||}t|d}g}d}t|}t|D]&} ||} |d7}t||||} ||7}t| || ||| | }q%g} |D]} tdt| dD]} | | | | | d| | dfqZqP| S)Nrr rprrIr-r;zlib decompressrOrange _unfilter_rowrvrsrArr bytes_per_pixelstriderawrowsrxprevious_ filter_typerowr!xrrrr@s,    &r@c Cst|d}t|d}d}||}t|d}g}d}t|}t|D]&} ||} |d7}t||||} ||7}t| || ||| | }q%g} |D]$} tdt| dD]} | | | | | d| | d| | dfqZqP| S) Nrr r+rprrIr-rrrrrrr9s,    0r9rrOrrrNonecCs|dkrdStt|D]h}||kr|||nd}||}||kr(|||nd}|dkr9|||d@||<q |dkrH|||d@||<q |dkr[||||dd@||<q |dkrn||t|||d@||<q td|dS)NrrIr4r-rr+unsupported_png_filter_)rrs_paethr7)rrrrrdleftup upper_leftrrrrs rrrrcCsT|||}t||}t||}t||}||kr"||kr"|S||kr(|S|SN)abs)rrrestimate left_distance up_distanceupper_left_distancerrrrs    r list[bool]c Cszt|||}dg||}|s|St}t|D]!}d|dfD]}|||}t|||r8d||<||q qt|D]%}d|dfD]}|||}||sbt|||rbd||<||qFq>|r|}||}||}|d|f|d|f||df||dffD]0\} } | dks| |ks| dks| |krq| || } || rqt|| |rd|| <|| q|sf|S)NFrrIT)_edge_background_paletterr_is_background_pixelrvpopleft) r!rr paletterbqueueryrdnext_xnext_y next_indexrrrr_!sJ       4    r_cCsg}t|D]}||||||d||qt|D]}|||||||||dq!dd|D}ddt|dDS)NrIcSg|] }t|rt|qSr_is_supported_background_pixel _quantizerr]rrrrQs z,_edge_background_palette..cSg|]\}}|qSrrrr[rrrrrVrrrvr most_common)r!rr edge_pixelsrr candidatesrrrrIs  rc Csg}t|D]#}||d||fD]}||\}}}} | dkr(||||fqqt|D]%} | || ||dfD]}||\}}}} | dkrR||||fq.cSrrrrrrrrjrrr) r!rr rrrdrrrrrrrrrr=Ys"  r=csHdd|D}|s gStdtt|dfddt|dDS)NcSrr)_is_chroma_key_likerrrrrrnrz5_global_chroma_background_palette..r+g{Gz?csg|] \}}|kr|qSrr)rr[count thresholdrrrrrr)maxr;rsrr)r!rrrrr>ms r>rbrcCst|rdnd}t|D]h}g}t|D]Q\}} ||s!t| |s"q||} ||} | d| f| d| f| | df| | dffD]$\} } | dksT| |ksT| dksT| |krUq@|| || rd||nq@q|skdS|D]}d||<qmq dS)NrIrrT)_palette_is_green_chromarr`_is_chroma_halo_pixelrv)r!rbrr rmax_iterationsrto_clearrdr]rrrrrrrraus, 4   racCs*|D]\}}}||kr||krdSqdS)NTFr)rrrrrrrrs rr]cCstdd|DS)Ncss|] }|ddVqdS)r/Nr)rvaluerrrr(z_quantize..)tupler\rrrrsrcCst|t|dkS)N)rminr\rrr_is_low_saturation_graysrcsTtrtd}d|kodkSStp)tp)tfddtDS)Nrrr4c3|] }t|dkVqdSHNrPrZr\rrr(r)z1_is_supported_background_pixel..)rsum_is_high_saturation_chroma_is_desaturated_green_chromar<_chroma_key_colors)r] brightnessrr\rrs rcs(tptptfddtDS)Nc3rrrrZr\rrr(r)z&_is_chroma_key_like..)rrr<rr\rr\rrs rc Cs|\}}}|D]Z}|\}}}||kr)||kr)|dkr)||dkr)||dkr)dS||krE||krE|dkrE|dkrE|dt||krEdS||kra||kra|dkra|dkra|dt||kradSqdS)NrgTdF)r) r]rrrrr[ color_red color_green color_bluerrrrs   ""rchromacCs|\}}}|\}}}|dkr*|dkr*|dkr*|dko)|dko)|d|ko)|d|kS|dkrJ|dkrJ|dkrJ|dkoI|dkoI|d|koI|d|kS|dkrj|dkrj|dkrj|dkoi|dkoi|d|koi|d|kSt||gS)NrP)r)r]rrrr chroma_red chroma_green chroma_bluerrrrRs  ((( rRcCs|\}}}t|t|dkrdSt|ddkrdS|dkr'|dkr'|dkpb|dko2|dko2|dkpb|dko>|dko>|dkpb|dkoJ|dkoJ|dkpb|dkoV|dkoV|d kpb|d kob|d kob|dkS) NrFr#rrrF)rrrr]rrrrrrrs rcCs:|\}}}|dko|dko|dko||dko||dkS)NrZ(<rrrrrrs 0rcstfdd|DS)Nc3s0|]}trt|dkpt|VqdSr)rrPrYrZr\rrr(s  z'_is_background_pixel..)r<)r]rrr\rrs rc s\tsdS|\}}}\}}}ddtD}ddtD} |s&dS|||ftfdd|Dr9dStfdd| DrFdStfdd|D} tfd d| Dd d } | | d krfdStd dkrpdS|dkr|dkr|dkrt||dkS|dkr|dkr|dkrt||dkS|dkr|dkr|dkrt||dkSdS)NFcSsg|] \}}|dkr|qSrrrrdrrrrrrz+_is_chroma_shadow_pixel..cSsg|] \}}|dkr|qSrrrrrrrrc3s0|]}|dkp||dkVqdS)KNrrrdrvaluesrrr(s.z*_is_chroma_shadow_pixel..c3s|] }|dkVqdS)rNrrrrrr(rc3|]}|VqdSrrrrrrr(c3rrrrrrrr(rr)default-rrrrrT)rr`r<rrrr) r]rrrrrrr high_channels low_channelshigh_minlow_maxrrrrYs4    rYrightcs tfddtdDdS)Nc3s$|] }||dVqdS)r-Nrrrrrrr( s"z _rgb_distance..rg?)rrrrrrrPs rP tuple[tuple[int, int, int], ...]cCsdS)N))r4rr4)rr4r4)rr4r)r4r4r)rrr4)r4rrrrrrr srrUc Cs|d}t}t|D]}|d|||||d|q tdtd||dddddtdtt |td d g}t d |S) Nr+rrIrirjr/rrkrlro) rOrrvrQ _png_chunkrtpackrcompressr rrrw)rr rUrrrchunksrrrrSs   rSr}datacCs6t||d@}tdt|||td|S)Nlrh)binasciicrc32rtrrs)r}rchecksumrrrr%s$r)rr rr rr r r )r!rFrr;r r;r"rGrrHrr r r ) r!rFrr;r r;r"rGr r )rr r re)rArer r)rArer rF) rrOrrOrr;rr;r r)rr;rr;rr;r r;)r!rrr;r r;r r)r!rrr;r r;r r)r!rFrr;r r;r r)r!rr r) r!rrbrrr;r r;rrr r)rrr rG)r]rHr rH)r]rHr rG)r]rHrrr rG)r]rHrrHr rG)rrHrrHr r )r r)rr;r r;rUr r r )r}r rr r r )' __future__rrrtr collectionsrrtypingrrrrEr:r?r6r@r9rrr_rr=r>rarrrrrrrRrrrrYrPrrSrrrrrsJ   R # ' '    (