295 lines
11 KiB
Python
295 lines
11 KiB
Python
#!/usr/bin/env python3
|
||
"""从用户提供的设计稿中提取 Wasteland 边框与实际尺寸图标。"""
|
||
|
||
from pathlib import Path
|
||
import json
|
||
|
||
from PIL import Image, ImageFilter
|
||
|
||
|
||
ROOT = Path('/home/cc-web')
|
||
REFERENCE = ROOT / 'sessions/_attachments/f01c1388-b603-434a-b226-2ff5771d4925.webp'
|
||
ICON_SHEET = ROOT / 'sessions/_attachments/43bd9ed6-2ba6-4c9d-bd9d-25b068cb4711.webp'
|
||
WELCOME_MATERIAL = ROOT / 'sessions/_attachments/2b88b81d-c5a8-4d92-a518-afeea691eade.webp'
|
||
ASSET_ROOT = ROOT / 'public/assets/themes/wasteland'
|
||
FRAME_DIR = ASSET_ROOT / 'frames'
|
||
ICON_DIR = ASSET_ROOT / 'icons/ui'
|
||
TEXTURE_DIR = ASSET_ROOT / 'textures'
|
||
|
||
|
||
def gold_alpha(pixel):
|
||
r, g, b = pixel
|
||
warm = max(0, min(r - b, g - b + 8))
|
||
light = max(0, (r + g) // 2 - 24)
|
||
if r < 38 or g < 27 or warm < 8:
|
||
return 0
|
||
return max(0, min(238, warm * 4 + light * 2))
|
||
|
||
|
||
def icon_alpha(pixel):
|
||
r, g, b = pixel
|
||
gold = r > 44 and g > 31 and r - b > 15 and g - b > 7
|
||
pale = r > 118 and g > 86 and r - b > 28
|
||
green = g > 58 and g - r > 5 and g - b > 10
|
||
if not (gold or pale or green):
|
||
return 0
|
||
if green:
|
||
score = (g - min(r, b)) * 4 + g
|
||
else:
|
||
score = (r - b) * 3 + (g - b) * 2 + (r + g) // 4
|
||
return max(0, min(255, score))
|
||
|
||
|
||
def crop_frame(source, name, box, zone):
|
||
crop = source.crop(box).convert('RGB')
|
||
width, height = crop.size
|
||
rgba = Image.new('RGBA', crop.size, (0, 0, 0, 0))
|
||
src = crop.load()
|
||
dst = rgba.load()
|
||
for y in range(height):
|
||
for x in range(width):
|
||
if not zone(x, y, width, height):
|
||
continue
|
||
alpha = gold_alpha(src[x, y])
|
||
if alpha:
|
||
r, g, b = src[x, y]
|
||
dst[x, y] = (r, g, b, alpha)
|
||
rgba.save(FRAME_DIR / f'{name}.png')
|
||
return {'name': name, 'source_box': list(box), 'size': [width, height]}
|
||
|
||
|
||
def extract_welcome_card(material):
|
||
"""从烘焙棋盘格的素材板中分离欢迎卡金属框和干净暗纹理。"""
|
||
frame_box = (70, 0, 1165, 875)
|
||
crop = material.crop(frame_box).convert('RGB')
|
||
frame = Image.new('RGBA', crop.size, (0, 0, 0, 0))
|
||
src = crop.load()
|
||
dst = frame.load()
|
||
for y in range(crop.height):
|
||
source_y = y + frame_box[1]
|
||
for x in range(crop.width):
|
||
source_x = x + frame_box[0]
|
||
# 730px 以下只保留主框的中央下坠徽记，排除素材板的其他组件。
|
||
if source_y >= 730 and not 500 <= source_x <= 780:
|
||
continue
|
||
alpha = gold_alpha(src[x, y])
|
||
if alpha < 18:
|
||
continue
|
||
r, g, b = src[x, y]
|
||
dst[x, y] = (r, g, b, alpha)
|
||
frame.save(FRAME_DIR / 'welcome-card.png')
|
||
|
||
surface_box = (350, 250, 950, 600)
|
||
surface = material.crop(surface_box).convert('RGB')
|
||
surface.save(TEXTURE_DIR / 'welcome-card-surface.webp', quality=92, method=6)
|
||
return [
|
||
{
|
||
'name': 'welcome-card',
|
||
'source': WELCOME_MATERIAL.name,
|
||
'source_box': list(frame_box),
|
||
'size': list(frame.size),
|
||
'output': 'frames/welcome-card.png',
|
||
},
|
||
{
|
||
'name': 'welcome-card-surface',
|
||
'source': WELCOME_MATERIAL.name,
|
||
'source_box': list(surface_box),
|
||
'size': list(surface.size),
|
||
'output': 'textures/welcome-card-surface.webp',
|
||
},
|
||
]
|
||
|
||
|
||
def slot_box(card, variant):
|
||
x, y, width, height = card
|
||
bands = {
|
||
'16': (x + 14, x + 74),
|
||
'24': (x + 76, x + 150),
|
||
'32': (x + 146, x + width - 12),
|
||
}
|
||
sx0, sx1 = bands[variant]
|
||
return sx0, y + 30, sx1, min(y + 105, y + height - 24)
|
||
|
||
|
||
def crop_icon(sheet, name, card, variant, canvas):
|
||
sx0, sy0, sx1, sy1 = slot_box(card, variant)
|
||
region = sheet.crop((sx0, sy0, sx1, sy1)).convert('RGB')
|
||
width, height = region.size
|
||
alpha = Image.new('L', region.size, 0)
|
||
ap = alpha.load()
|
||
rp = region.load()
|
||
points = []
|
||
for y in range(height):
|
||
for x in range(width):
|
||
value = icon_alpha(rp[x, y])
|
||
if value:
|
||
ap[x, y] = value
|
||
points.append((x, y))
|
||
if not points:
|
||
raise RuntimeError(f'没有检测到图标：{name}')
|
||
left = max(0, min(p[0] for p in points) - 3)
|
||
top = max(0, min(p[1] for p in points) - 3)
|
||
right = min(width, max(p[0] for p in points) + 4)
|
||
bottom = min(height, max(p[1] for p in points) + 4)
|
||
icon = region.crop((left, top, right, bottom)).convert('RGBA')
|
||
icon.putalpha(alpha.crop((left, top, right, bottom)).filter(ImageFilter.GaussianBlur(0.18)))
|
||
|
||
target_w, target_h = canvas
|
||
scale = min((target_w - 2) / icon.width, (target_h - 2) / icon.height)
|
||
resized = icon.resize(
|
||
(max(1, round(icon.width * scale)), max(1, round(icon.height * scale))),
|
||
Image.Resampling.LANCZOS,
|
||
)
|
||
output = Image.new('RGBA', canvas, (0, 0, 0, 0))
|
||
output.alpha_composite(resized, ((target_w - resized.width) // 2, (target_h - resized.height) // 2))
|
||
|
||
# 输入区图标按有效像素重心居中，避免纸飞机等非对称轮廓在按钮里视觉偏移。
|
||
if name in {'note', 'queue', 'send', 'stop'}:
|
||
alpha = output.getchannel('A')
|
||
pixels = alpha.load()
|
||
total = weighted_x = weighted_y = 0
|
||
for py in range(target_h):
|
||
for px in range(target_w):
|
||
value = pixels[px, py]
|
||
if not value:
|
||
continue
|
||
total += value
|
||
weighted_x += px * value
|
||
weighted_y += py * value
|
||
if total:
|
||
offset_x = round((target_w - 1) / 2 - weighted_x / total)
|
||
offset_y = round((target_h - 1) / 2 - weighted_y / total)
|
||
if offset_x or offset_y:
|
||
centered = Image.new('RGBA', canvas, (0, 0, 0, 0))
|
||
centered.alpha_composite(output, (offset_x, offset_y))
|
||
output = centered
|
||
output.save(ICON_DIR / f'{name}.png')
|
||
return {
|
||
'name': name,
|
||
'variant': variant,
|
||
'slot_box': [sx0, sy0, sx1 - sx0, sy1 - sy0],
|
||
'detected_box': [sx0 + left, sy0 + top, right - left, bottom - top],
|
||
'size': [target_w, target_h],
|
||
}
|
||
|
||
|
||
def main():
|
||
FRAME_DIR.mkdir(parents=True, exist_ok=True)
|
||
ICON_DIR.mkdir(parents=True, exist_ok=True)
|
||
TEXTURE_DIR.mkdir(parents=True, exist_ok=True)
|
||
reference = Image.open(REFERENCE).convert('RGB')
|
||
sheet = Image.open(ICON_SHEET).convert('RGB')
|
||
welcome_material = Image.open(WELCOME_MATERIAL).convert('RGB')
|
||
|
||
border_zone = lambda x, y, w, h: x < 8 or x >= w - 8 or y < 8 or y >= h - 8
|
||
frames = [
|
||
crop_frame(reference, 'new-chat', (15, 9, 304, 58), lambda x, y, w, h: border_zone(x, y, w, h) or 225 <= x <= 239),
|
||
crop_frame(reference, 'search', (15, 67, 304, 103), border_zone),
|
||
crop_frame(reference, 'project', (15, 117, 304, 164), border_zone),
|
||
crop_frame(reference, 'active-session', (15, 163, 304, 209), lambda x, y, w, h: x < 12 or x >= w - 7 or y < 7 or y >= h - 7),
|
||
crop_frame(
|
||
reference,
|
||
'composer',
|
||
(557, 830, 1364, 931),
|
||
lambda x, y, w, h: (
|
||
(y < 10)
|
||
or (x < 10 and y < 77)
|
||
or (x >= w - 10 and y < 77)
|
||
or (64 <= y < 79)
|
||
or (w // 2 - 56 <= x <= w // 2 + 56 and y >= 58)
|
||
),
|
||
),
|
||
crop_frame(reference, 'composer-action', (1211, 845, 1257, 891), border_zone),
|
||
crop_frame(reference, 'tool-call', (421, 603, 1381, 664), border_zone),
|
||
]
|
||
|
||
# 设计稿输入框含两层顶线。UI 叠加时只保留内侧主蚀刻线，
|
||
# 同时对称收紧左右 5px，使主竖线贴合真实容器边界；徽记保持几何中心不变。
|
||
composer_path = FRAME_DIR / 'composer.png'
|
||
composer = Image.open(composer_path).convert('RGBA')
|
||
composer = composer.crop((5, 5, composer.width - 5, composer.height))
|
||
composer = composer.resize((807, 101), Image.Resampling.LANCZOS)
|
||
|
||
# 设计稿原始裁区的中心徽记视觉中心比画布中心向右约 8px。
|
||
# 只校正下半部装饰层，保留左右框边和顶部线的原始几何位置。
|
||
centered_composer = composer.copy()
|
||
ornament_region = (96, 50, 711, 101)
|
||
centered_composer.paste((0, 0, 0, 0), ornament_region)
|
||
centered_composer.alpha_composite(composer.crop(ornament_region), (88, 50))
|
||
composer = centered_composer
|
||
composer.save(composer_path)
|
||
|
||
# 输入框不能把 807×101 的整图同时做横纵拉伸：窄窗口会让底部徽记
|
||
# 与真实 74px 框体错层。拆出只含设计稿顶部/侧边的框壳和独立徽记，
|
||
# 由 CSS 分别控制尺寸，保证任意宽度下都只出现一条边框。
|
||
shell_height = 74
|
||
shell = Image.new('RGBA', (composer.width, shell_height), (0, 0, 0, 0))
|
||
composer_pixels = composer.load()
|
||
shell_pixels = shell.load()
|
||
for y in range(shell_height):
|
||
for x in range(composer.width):
|
||
if y < 14 or x < 14 or x >= composer.width - 14:
|
||
shell_pixels[x, y] = composer_pixels[x, y]
|
||
shell.save(FRAME_DIR / 'composer-shell.png')
|
||
|
||
ornament_box = (130, 50, 677, 101)
|
||
composer.crop(ornament_box).save(FRAME_DIR / 'composer-ornament.png')
|
||
frames.extend([
|
||
{
|
||
'name': 'composer-shell',
|
||
'derived_from': 'composer.png',
|
||
'source_box': [0, 0, composer.width, shell_height],
|
||
'size': [composer.width, shell_height],
|
||
},
|
||
{
|
||
'name': 'composer-ornament',
|
||
'derived_from': 'composer.png',
|
||
'source_box': [
|
||
ornament_box[0],
|
||
ornament_box[1],
|
||
ornament_box[2] - ornament_box[0],
|
||
ornament_box[3] - ornament_box[1],
|
||
],
|
||
'size': [ornament_box[2] - ornament_box[0], ornament_box[3] - ornament_box[1]],
|
||
},
|
||
])
|
||
frames.extend(extract_welcome_card(welcome_material))
|
||
|
||
cards = {
|
||
'search': (582, 50, 247, 137),
|
||
'dropdown': (315, 50, 247, 137),
|
||
'theme': (50, 201, 245, 138),
|
||
'ai': (315, 201, 247, 138),
|
||
'status': (582, 201, 247, 138),
|
||
'attachment': (1111, 201, 247, 138),
|
||
'send': (1377, 201, 244, 138),
|
||
'queue': (846, 201, 246, 138),
|
||
'stop': (50, 354, 245, 136),
|
||
'settings': (315, 354, 247, 136),
|
||
'user': (582, 354, 247, 136),
|
||
'terminal': (846, 354, 246, 136),
|
||
'note': (582, 506, 247, 138),
|
||
}
|
||
specs = [
|
||
('search', '16', (20, 20)),
|
||
('dropdown', '16', (14, 10)),
|
||
('theme', '24', (26, 26)),
|
||
('ai', '32', (36, 36)),
|
||
('status', '16', (16, 16)),
|
||
('attachment', '24', (26, 28)),
|
||
('send', '32', (30, 30)),
|
||
('queue', '32', (30, 30)),
|
||
('stop', '24', (26, 26)),
|
||
('settings', '16', (20, 20)),
|
||
('user', '32', (36, 36)),
|
||
('terminal', '16', (20, 18)),
|
||
('note', '32', (36, 36)),
|
||
]
|
||
icons = [crop_icon(sheet, name, cards[name], variant, canvas) for name, variant, canvas in specs]
|
||
(FRAME_DIR / 'manifest.json').write_text(json.dumps(frames, ensure_ascii=False, indent=2) + '\n', encoding='utf-8')
|
||
(ICON_DIR / 'manifest.json').write_text(json.dumps(icons, ensure_ascii=False, indent=2) + '\n', encoding='utf-8')
|
||
|
||
|
||
if __name__ == '__main__':
|
||
main()
|