si1& Z U d Z ddlmZ ddlZddlZddlZddlZddlmZ ddl m
Z
m Z d dZ ee
j Zedz Zedz Zed z Zd
Zd Zd Zd d
Zeefd dZefd dZd dZd!dZd"dZd#dZ d$dZ d%d&dZ i Z de d< d'd(dZ! d) d*dZ"d Z#e$dk( r e# yy)+ud squircle.py — converte PNG raw em squircle 512×512 com bg color correto.
Pipeline:
1. Resize+pad pra 512×512 (preservando aspect ratio, transparent pad)
2. Detecta bg color: override hex > edge-ring k-means > auto-neutral por luminance
3. Composita logo sobre bg sólido (elimina cantos transparentes)
4. Aplica squircle mask (radius ~22%)
5. Modula saturação *= 0.5
6. Salva 512×512 PNG
Usage as module:
from squircle import make_squircle
make_squircle("raw.png", "out.png", override_hex="#fafafa", desat=True)
Usage as CLI:
python squircle.py [--bg #RRGGBB] [--no-desat]
) annotationsN)Path)Image
ImageColorc
t | } | j j dk( rt j dddddt | gd }|j dk( rYt |j d kD rAt j t j |j j d
S dd l}|j t | d d d
}t j t j | j d
S t j | j d
S # t $ r }t# d| j$ d| d }~ww xY w)z[Open PNG, SVG, or other formats. SVGs rasterized via rsvg-convert (with cairosvg fallback).z.svgz rsvg-convertz-w1024z-hT)capture_outputr d RGBANi )url output_width
output_heightunsafez*Both rsvg-convert and cairosvg failed for z: )r suffixlower
subprocessrunstr
returncodelenstdoutr openioBytesIOconvertcairosvgsvg2png Exception RuntimeErrorname)pathresultr png_byteses >/root/nxt/terminal.net.br/recursos/assets/_scripts/squircle.py
load_imager& s+ :D {{f$
T64T C
!c&--&83&>::bjj 78@@H H
((ID ) I ::bjj 34<EE zcircle-mask-85pct-512.pngzcircle-mask-512.pngzsquircle-mask-512.pngi g{Gz?g333333?c | j d } | j }|rW| j | }|j d | j d dz k s"|j d | j d dz k r|S | S )u Auto-crop borda do logo: remove pixels transparentes/quase-brancos das margens.
Garante que o CONTEÚDO real (não o frame do raw) seja a unidade de medida.r r gffffff? )r getbboxcropsize)imgbboxcroppeds r%
trim_imager/ @ ss ++f
C ;;=D ((4. <<?SXXa[4 / /7<<?SXXa[SWEW3WN J c | j d } t | } | j \ }}t dt ||z }|t || z }t dt ||z t dt ||z f}| j |t j } t
j d||fd }|j | ||d z
dz ||d z
dz f| |S )u Fit img content into (size*content_pct)×(size*content_pct) area, centered on size canvas.
Trim transparent borders ANTES do resize pra garantir proporção uniforme.r r( )r r r r r )
r r/ r+ maxintresizer LANCZOSnewpaste) r, r+ content_pctwhtargetscalenew_sizecanvass r%
resize_padr@ P s ++f
C
S/C 88DAq
C{*+
,F SAY EAs1u9~&As1u9~(>?H
**Xu}}
-C
YYvd|\
:F
LLx{*q04(1+3E!2KLcR Mr0 c : | j \ }}t t |t || z d }| j }g }t t
| t t
||z
| z D ]: }t
d|d D ]( }|||f \ } }
} } | dkD s|j | |
| f * < t t
| t t
||z
| z D ]= }t
|||z
d D ]( }|||f \ } }
} } | dkD s|j | |
| f * ? |S )zESample opaque pixels from outer ring. Returns list of (R,G,B) tuples. r r2 )r+ r3 r4 minloadlistrangeappend)
r,
margin_pctr: r; marginpxpixelsyxrgbas
r% edge_pixelsrS _ s- 88DAq
Z#a)+,a
0F B
F
%-
4a&j!(<#=
= )q!Q )AAqDJAq!Q3w
q!Qi( )) %-
4a&j!(<#=
= )vq6z1- )AAqDJAq!Q3w
q!Qi( ))
Mr0 c | sy| D cg c] }|d }}| D cg c] }|d }}| D cg c] }|d }}t | t | z
t | t | z
t | t | z
g}|j t | t fd| D t | dz }| D cg c]
}| |k s | }}| D cg c]
}| |kD s | }}t | t | k\ r|n|} | s| } t | }
t d | D |
z t d | D |
z t d | D |
z f} | S c c}w c c}w c c}w c c}w c c}w ) zJSimple 2-cluster median-cut. Returns dominant cluster centroid as (R,G,B).Nr r( r2 c 3 ( K | ] }| y w)N ).0paxiss r% z median_cut_2..~ s +q1T7+s c 3 & K | ] }|d yw)r NrV rW rX s r% rZ z median_cut_2.. !QAaD ! c 3 & K | ] }|d yw)r( NrV r\ s r% rZ z median_cut_2.. r] r^ c 3 & K | ] }|d yw)r2 NrV r\ s r% rZ z median_cut_2.. r] r^ )r3 rD indexsortedr sum)
rL rX rsgsbsrangespivot cluster_a cluster_blargernavgrY s
@r% median_cut_2rn t sm 1!A$ B 1!A$ B 1!A$ B "gB R3r7!2CGc"g4E
FF <<F $D +F+ +CK1,< =E "7qag&67I7 "6qago6I6 i.C N:Y F F A !& !!Q& !& !!Q& !& !!Q& C
J#
86s' E E E
E%E%
E*&E*c L t | }t | dk ryt | S )zDetect probable bg color via edge ring sampling + 2-cluster median cut.
Returns RGB tuple or None if logo is mostly transparent at edges.r
N)rS r rn )r, rL s r% detect_bgrp s(
F
6{S r0 c 6 | \ }}}d|z d|z z d|z z dz S )z,Perceptual luminance 0..1 (Rec.709 weights).gz6?g,C ?g]m{?g o@rV )rgbrO rP rQ s r% luminancers s. GAq! QJ! #fqj 0E 99r0 c | j }| j \ }}d}d}t d|d D ]< }t d|d D ]* }|||f \ }} }
} | dkD s|t || |
f z
}|dz
}, > |dk( ry||z } | dkD rdS dS ) zNPick contrasting neutral bg based on dominant luminance of opaque logo pixels.g r rC r( ) ?)
rz rz )rE r+ rG rs )
r, rK r: r; sum_lcountrM rN rO rP rQ rR avg_ls
r% auto_neutralr~ s B 88DAq E
E
1a^ q!Q AAqDJAq!Q3wAq!9--
z EME 3;<;O;r0 c 2 t j | dd S )u #RRGGBB → (R,G,B).N )r getrgb)hex_strs r% parse_hexr s W %bq ))r0 c J | j d } | j \ }}}}t j d|||f }|j d j d }t j ||| }|j \ } }
} t j d| |
| |f S )z;Reduce saturation. factor=1.0 = unchanged, 0.0 = grayscale.r RGBL)r splitr mergeblend) r, factorrO rP rQ rR rr grayblendedrbgbbbs r%
desaturater s
++f
CJAq!Q
++eaAY
'C ;;s # #E *Dkk$V,G JBB ;;vBA //r0 zdict[str, Image.Image] _mask_cachec | t vr<| dk( rt nt }t j | j d t | < t | j
S )uQ Load (and cache) alpha mask 512×512 (L mode). shape=circle (default) | squircle.squircler )r SQUIRCLE_MASK_PATH MASK_PATHr r r copy)shaper! s r%
squircle_maskr sK K %*j%8!i "ZZ - 5 5c : E u " " $$r0 c t | }t |t }|rt | }d| }nEt | }|r|}d|d d|d d|d d}n t | }d|d d|d d|d d}t
j dt t f|d z } | j | t | }
| j |
|r t | d
} | j |
t | j j d d | j |d
d t! | t! | d|d d|d d|d d||dS )z]Apply full pipeline. shape=circle (default) | squircle. desat=False default (preserve cores).z override:z
detected:#r 02xr( r2 z neutral:#r ) ry T)parentsexist_okPNG)optimize#)inputoutputbg bg_sourcedesat)r& r@ SIZEr rp r~ r r7 alpha_compositer putalphar r parentmkdirsaver ) raw_path output_path override_hexr r r, r r detectedr? masks r%
make_squircler sp X
C
S$
C
|
$ ~. S> B$RU3K1c{2a5+FI c"B#BqE#;r!uSk"Q%EI YYvd|R&[
9F
3 D
OOD
FC( ""4$"?
KK UTK2 X k""Q%RU3K1c{3 r0 c t j d } | j dd | j dd | j dd | j d
d d gd
| j ddd | j }t |j
|j |j |j |j }t d|d d|d d|d y )Nu5 Convert raw PNG/SVG to circle (or squircle) 512×512.) descriptionr zraw PNG/SVG path)helpr zoutput PNG pathz--bgzoverride bg hex (#RRGGBB)z--shapecircler )choicesdefaultz--desat
store_truezapply 50%% desaturation)actionr )r r zOK: bg=r z (r u ) → ) argparseArgumentParser add_argument
parse_argsr r r r r r print)apargsinfos r% mainr s -d eBOOG"4O5OOH#4O5OOF!:J
KLr0 __main__)r!
str | Pathreturn Image.Image)r, r r r )r, r r+ r4 r9 floatr r )r, r rI r r list[tuple[int, int, int]])rL r r tuple[int, int, int] | None)r, r r r )rr tuple[int, int, int]r r )r, r r r )r r r r )ry )r, r r r r r )r )r r r r )NFr ) r r r r r z
str | Noner boolr r r dict)%__doc__
__future__r r r r syspathlibr PILr r r& __file__r
SCRIPT_DIRr MASK_FULL_PATHr r EDGE_MARGIN_PCT CONTENT_PCTr/ r@ rS rn rp rs r~ r r r __annotations__r r r __name__rV r0 r% r s " #
!,2 (^
"
"
4 4 33 "99
.2 7F *2 : <&*
0 ')
# (% $ ... . .
.
.b
M zF r0