200 lines
7.6 KiB
Python
200 lines
7.6 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
布局A: 大图在上，文字在下
|
|||
|
|
适用场景: 景点、美食、通用
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
from typing import Tuple
|
|||
|
|
from PIL import Image, ImageDraw
|
|||
|
|
|
|||
|
|
from .base import BaseLayout
|
|||
|
|
from ..schemas.content import PosterContent
|
|||
|
|
from ..schemas.theme import Theme
|
|||
|
|
from ..renderers.text import TextRenderer
|
|||
|
|
|
|||
|
|
|
|||
|
|
class HeroBottomLayout(BaseLayout):
|
|||
|
|
"""大图底部文字布局"""
|
|||
|
|
|
|||
|
|
# 字号配置
|
|||
|
|
TITLE_SIZE = 96
|
|||
|
|
SUBTITLE_SIZE = 36
|
|||
|
|
PRICE_SIZE = 88
|
|||
|
|
DETAIL_SIZE = 32
|
|||
|
|
TAG_SIZE = 24
|
|||
|
|
|
|||
|
|
def calculate_content_height(self, content: PosterContent, theme: Theme) -> int:
|
|||
|
|
"""计算内容区域高度 (与 design_samples 保持一致)"""
|
|||
|
|
height = self.PADDING
|
|||
|
|
|
|||
|
|
# 标题
|
|||
|
|
title_font = TextRenderer.get_title_font(self.TITLE_SIZE)
|
|||
|
|
_, title_h = TextRenderer.measure_text(content.title, title_font)
|
|||
|
|
height += title_h + 14
|
|||
|
|
|
|||
|
|
# 副标题 (始终计算，即使为空也保留空间)
|
|||
|
|
height += 32 + 44
|
|||
|
|
|
|||
|
|
# 装饰线
|
|||
|
|
height += 28
|
|||
|
|
|
|||
|
|
# 详情
|
|||
|
|
if content.details:
|
|||
|
|
height += len(content.details) * 44 + 20
|
|||
|
|
|
|||
|
|
# 价格
|
|||
|
|
price_font = TextRenderer.get_title_font(self.PRICE_SIZE)
|
|||
|
|
_, price_h = TextRenderer.measure_text("¥999", price_font)
|
|||
|
|
height += price_h + self.PADDING
|
|||
|
|
|
|||
|
|
return height
|
|||
|
|
|
|||
|
|
def generate(self, content: PosterContent, theme: Theme) -> Image.Image:
|
|||
|
|
"""生成海报 (与 design_samples 保持一致)"""
|
|||
|
|
# 计算内容高度
|
|||
|
|
content_height = self.calculate_content_height(content, theme)
|
|||
|
|
content_y = self.height - content_height - 40
|
|||
|
|
|
|||
|
|
# 创建画布 (淡色底)
|
|||
|
|
canvas = self.create_canvas(theme.secondary_rgb)
|
|||
|
|
draw = ImageDraw.Draw(canvas)
|
|||
|
|
|
|||
|
|
# === 上半部分: 渐变或图片 ===
|
|||
|
|
if content.image:
|
|||
|
|
# 有图片: 图片覆盖上半部分
|
|||
|
|
img = content.image.copy().convert("RGBA")
|
|||
|
|
# 保持比例，覆盖裁剪到上半部分
|
|||
|
|
target_height = content_y + 60
|
|||
|
|
img_ratio = img.width / img.height
|
|||
|
|
target_ratio = self.width / target_height
|
|||
|
|
|
|||
|
|
if img_ratio > target_ratio:
|
|||
|
|
new_height = target_height
|
|||
|
|
new_width = int(img.width * (target_height / img.height))
|
|||
|
|
img = img.resize((new_width, new_height), Image.LANCZOS)
|
|||
|
|
left = (new_width - self.width) // 2
|
|||
|
|
img = img.crop((left, 0, left + self.width, new_height))
|
|||
|
|
else:
|
|||
|
|
new_width = self.width
|
|||
|
|
new_height = int(img.height * (self.width / img.width))
|
|||
|
|
img = img.resize((new_width, new_height), Image.LANCZOS)
|
|||
|
|
img = img.crop((0, 0, self.width, target_height))
|
|||
|
|
|
|||
|
|
canvas.paste(img, (0, 0))
|
|||
|
|
else:
|
|||
|
|
# 无图片: 渐变色
|
|||
|
|
gradient = self.effect.create_gradient(
|
|||
|
|
(self.width, content_y + 60),
|
|||
|
|
theme.gradient_rgb[0],
|
|||
|
|
theme.gradient_rgb[1]
|
|||
|
|
)
|
|||
|
|
canvas.paste(gradient, (0, 0))
|
|||
|
|
|
|||
|
|
# === 底部渐变遮罩 (从底部往上渐变) ===
|
|||
|
|
# 底部完全不透明，往上越来越透明，直到标题上方
|
|||
|
|
gradient_top = content_y - 80 # 渐变起点（标题上方）
|
|||
|
|
gradient_height = self.height - gradient_top # 渐变高度
|
|||
|
|
|
|||
|
|
overlay = self.effect.create_gradient(
|
|||
|
|
(self.width, gradient_height),
|
|||
|
|
(*theme.primary_rgb, 0), # 顶部透明
|
|||
|
|
(*theme.primary_rgb, 255) # 底部完全不透明
|
|||
|
|
)
|
|||
|
|
canvas.paste(overlay, (0, gradient_top), overlay)
|
|||
|
|
|
|||
|
|
# 重新获取 draw (因为 paste 后需要)
|
|||
|
|
draw = ImageDraw.Draw(canvas)
|
|||
|
|
|
|||
|
|
# 字体
|
|||
|
|
title_font = TextRenderer.get_title_font(self.TITLE_SIZE)
|
|||
|
|
subtitle_font = TextRenderer.get_body_font(self.SUBTITLE_SIZE)
|
|||
|
|
price_font = TextRenderer.get_title_font(self.PRICE_SIZE)
|
|||
|
|
detail_font = TextRenderer.get_body_font(self.DETAIL_SIZE)
|
|||
|
|
tag_font = TextRenderer.get_body_font(self.TAG_SIZE)
|
|||
|
|
suffix_font = TextRenderer.get_body_font(28)
|
|||
|
|
|
|||
|
|
text_white = theme.text_rgb
|
|||
|
|
accent = theme.accent_rgb
|
|||
|
|
|
|||
|
|
cur_y = content_y + self.PADDING
|
|||
|
|
content_width = self.width - self.MARGIN * 2
|
|||
|
|
|
|||
|
|
# === 标题 (自适应大小，尽量单行显示) ===
|
|||
|
|
adaptive_title_font = TextRenderer.get_adaptive_title_font(
|
|||
|
|
content.title, content_width, base_size=self.TITLE_SIZE, min_size=56
|
|||
|
|
)
|
|||
|
|
title_w, title_h = TextRenderer.measure_text(content.title, adaptive_title_font)
|
|||
|
|
|
|||
|
|
# 居中显示
|
|||
|
|
title_x = self.MARGIN + (content_width - title_w) // 2
|
|||
|
|
TextRenderer.draw_text_with_shadow(
|
|||
|
|
draw, (title_x, cur_y), content.title, adaptive_title_font,
|
|||
|
|
theme.text, shadow_color=(0, 0, 0, 60), offset=(2, 2)
|
|||
|
|
)
|
|||
|
|
cur_y += title_h + 14
|
|||
|
|
|
|||
|
|
# === 副标题 (支持换行) ===
|
|||
|
|
if content.subtitle:
|
|||
|
|
_, sub_h = TextRenderer.draw_wrapped_text(
|
|||
|
|
draw, (self.MARGIN, cur_y), content.subtitle, subtitle_font,
|
|||
|
|
(*text_white, 200), content_width, line_spacing=4
|
|||
|
|
)
|
|||
|
|
cur_y += max(sub_h, 36) + 8
|
|||
|
|
|
|||
|
|
# === 装饰线 ===
|
|||
|
|
self.shape.draw_decorator_line(draw, (self.MARGIN, cur_y), 60, (*accent, 200))
|
|||
|
|
cur_y += 28
|
|||
|
|
|
|||
|
|
# === 详情 ===
|
|||
|
|
if content.details:
|
|||
|
|
for detail in content.details:
|
|||
|
|
self.shape.draw_ellipse(draw,
|
|||
|
|
(self.MARGIN, cur_y + 12, self.MARGIN + 8, cur_y + 20),
|
|||
|
|
fill=(*accent, 180))
|
|||
|
|
draw.text((self.MARGIN + 18, cur_y), detail,
|
|||
|
|
font=detail_font, fill=(*text_white, 180))
|
|||
|
|
cur_y += 44
|
|||
|
|
cur_y += 20
|
|||
|
|
|
|||
|
|
# === 价格 ===
|
|||
|
|
if content.price:
|
|||
|
|
price_w, price_h = TextRenderer.measure_text(content.price, price_font)
|
|||
|
|
|
|||
|
|
# 计算后缀宽度
|
|||
|
|
suffix = content.price_suffix or ""
|
|||
|
|
suffix_w = 0
|
|||
|
|
if suffix:
|
|||
|
|
suffix_w, _ = TextRenderer.measure_text(suffix, suffix_font)
|
|||
|
|
|
|||
|
|
# 价格背景 (半透明主题色)
|
|||
|
|
bg_width = price_w + suffix_w + 40 if suffix else price_w + 24
|
|||
|
|
self.shape.draw_rounded_rect(draw,
|
|||
|
|
(self.MARGIN - 12, cur_y - 6, self.MARGIN + bg_width, cur_y + price_h + 8),
|
|||
|
|
radius=12, fill=(*accent, 40))
|
|||
|
|
|
|||
|
|
# 价格文字 (白色，与背景形成对比)
|
|||
|
|
draw.text((self.MARGIN, cur_y), content.price, font=price_font, fill=text_white)
|
|||
|
|
|
|||
|
|
# 后缀 (如果有)
|
|||
|
|
if suffix:
|
|||
|
|
draw.text((self.MARGIN + price_w + 8, cur_y + price_h - 28), suffix,
|
|||
|
|
font=suffix_font, fill=(*text_white, 200))
|
|||
|
|
|
|||
|
|
# === 标签 ===
|
|||
|
|
if content.tags:
|
|||
|
|
tag_x = self.width - self.MARGIN
|
|||
|
|
for tag in reversed(content.tags):
|
|||
|
|
tag_text = f"#{tag}"
|
|||
|
|
tag_w, _ = TextRenderer.measure_text(tag_text, tag_font)
|
|||
|
|
tag_x -= tag_w + 22
|
|||
|
|
|
|||
|
|
self.shape.draw_rounded_rect(draw,
|
|||
|
|
(tag_x, cur_y + 20, tag_x + tag_w + 16, cur_y + 54),
|
|||
|
|
radius=17, fill=(*accent, 35))
|
|||
|
|
draw.text((tag_x + 8, cur_y + 24), tag_text,
|
|||
|
|
font=tag_font, fill=text_white)
|
|||
|
|
tag_x -= 8
|
|||
|
|
|
|||
|
|
return canvas
|