mirror of
https://github.com/Shu-Ying/main_plugin_farm.git
synced 2026-10-27 20:50:06 +08:00
60 lines
2.3 KiB
Python
60 lines
2.3 KiB
Python
"""离线生成四季与蜂巢底图，以及同一农场状态的主题对照。"""
|
|
|
|
import argparse
|
|
from pathlib import Path
|
|
import sys
|
|
|
|
from PIL import Image, ImageDraw, ImageFont
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
sys.path.insert(0, str(ROOT / "src"))
|
|
|
|
from ci_font import choose_font
|
|
from farm_core import Identity
|
|
from farm_render import FarmRenderer, RenderOptions
|
|
from farm_render.canvas import Canvas
|
|
from render_demo import demonstration
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description=__doc__)
|
|
parser.add_argument("--output", type=Path, required=True)
|
|
parser.add_argument("--font", type=Path)
|
|
args = parser.parse_args()
|
|
args.output.mkdir(parents=True, exist_ok=True)
|
|
renderer = FarmRenderer(args.font or choose_font())
|
|
who = Identity("preview", "local", "qq", "seasons")
|
|
snapshot, now = demonstration()
|
|
themes = (("spring", "春季"), ("summer", "夏季"), ("autumn", "秋季"),
|
|
("winter", "冬季"), ("bee", "蜂巢 · 仅预览"))
|
|
sheet = Image.new("RGB", (1536, 3 * 510), "#edf0eb")
|
|
draw = ImageDraw.Draw(sheet)
|
|
font = ImageFont.truetype(str(renderer.font_path), 24)
|
|
for index, (theme, label) in enumerate(themes):
|
|
options = RenderOptions(width=3072, theme=theme)
|
|
image = renderer.assets.image(
|
|
f"background/decorateIcon/decorate_{theme}_1.jpg", (3072, 1872),
|
|
)
|
|
canvas = Canvas(image)
|
|
renderer._decorations(canvas, options, "background")
|
|
renderer._decorations(canvas, options, "foreground")
|
|
canvas.image.convert("RGB").save(
|
|
args.output / f"background-{theme}.jpg", quality=92,
|
|
)
|
|
thumb = canvas.image.convert("RGB").resize((768, 468),
|
|
Image.Resampling.LANCZOS)
|
|
x, y = (index % 2) * 768, (index // 2) * 510
|
|
draw.text((x+15, y+8), label, font=font, fill="#203e2c")
|
|
sheet.paste(thumb, (x, y+40))
|
|
result = renderer.render(
|
|
who, snapshot, now,
|
|
RenderOptions.high(theme=theme, format="PNG"),
|
|
)
|
|
(args.output / f"farm-{theme}-1536.png").write_bytes(result.content)
|
|
sheet.save(args.output / "all-themes.jpg", quality=92)
|
|
print(f"Generated 5 backgrounds, 5 farm previews: {args.output}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|