like-sci-figures 0.0.0-stage → 1.0.0

package/scripts/generate_samples.py

@@ -0,0 +1,322 @@
+#!/usr/bin/env python3
+"""Generate product preview assets: samples/ and showcases/.
+
+samples/
+    36 standalone PNG files — one per chart type, rendered with
+    palette=wx01_blue_red and style=nature. These are the per-chart
+    sample images used in the product page / docs.
+
+showcases/
+    34 full-grid PNG files — one per palette, each showing all 36
+    chart types rendered with that palette. These are the per-palette
+    showcase images used in the palette gallery.
+
+Usage
+-----
+    generate_samples.py                       # generate both
+    generate_samples.py --samples-only        # only samples/
+    generate_samples.py --showcases-only      # only showcases/
+    generate_samples.py --force               # overwrite existing
+    generate_samples.py --style ieee          # different style
+    generate_samples.py --dry-run             # print plan, no renders
+
+Notes
+-----
+* Reuses render_utils.render_chart() and render_utils.compose_grid()
+  so there is zero duplicated rendering code.
+* ThreadPoolExecutor for parallel renders (default 4 workers).
+* Requires Pillow.
+"""
+import argparse
+import json
+import math
+import os
+import re
+import sys
+import tempfile
+import time
+from concurrent.futures import ThreadPoolExecutor
+
+SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
+sys.path.insert(0, SCRIPT_DIR)
+
+from registry import CHART_EXAMPLES, CHART_TYPES, example_path, ZH_NAME
+from render_utils import render_chart, compose_grid, load_fonts
+from palettes import list_palettes, palette_info
+
+SKILL_DIR = os.path.dirname(SCRIPT_DIR)
+SAMPLES_DIR = os.path.join(SKILL_DIR, "samples")
+SHOWCASES_DIR = os.path.join(SKILL_DIR, "showcases")
+
+# Sample images: higher resolution for standalone viewing
+SAMPLE_W = 6.0
+SAMPLE_H = 4.0
+SAMPLE_DPI = 200
+
+# Showcase thumbnails: compact grid
+THUMB_W = 3.2
+THUMB_H = 2.4
+THUMB_DPI = 120
+
+
+# ── samples/ generation ────────────────────────────────────────────────────
+
+
+def _render_sample(chart_type, style, palette, out_dir):
+    """Render one chart type as a standalone sample PNG."""
+    data = example_path(chart_type)
+    if not data:
+        return chart_type, None, "no example file"
+    out = os.path.join(out_dir, "%s.png" % chart_type)
+    png, err, dt = render_chart(
+        chart_type, data, out,
+        style=style, palette=palette,
+        width=SAMPLE_W, height=SAMPLE_H, dpi=SAMPLE_DPI,
+    )
+    return chart_type, png, err, dt
+
+
+def generate_samples(style="nature", palette="wx01_blue_red", workers=4,
+                     force=False, dry_run=False):
+    """Render all 36 chart types as individual sample PNGs."""
+    chart_order = sorted(CHART_EXAMPLES)
+    n = len(chart_order)
+
+    if dry_run:
+        print("[samples] would render %d charts -> samples/" % n)
+        for ct in chart_order:
+            print("   %-20s -> samples/%s.png" % (ct, ct))
+        return 0
+
+    os.makedirs(SAMPLES_DIR, exist_ok=True)
+
+    results = {}
+    errors = []
+    t0 = time.time()
+
+    with ThreadPoolExecutor(max_workers=workers) as pool:
+        for ct, png, err, dt in pool.map(
+            lambda ct: _render_sample(ct, style, palette, SAMPLES_DIR),
+            chart_order
+        ):
+            if png:
+                results[ct] = png
+            else:
+                errors.append("%s: %s" % (ct, err))
+
+    elapsed = time.time() - t0
+    ok = len(results)
+
+    # Write manifest
+    manifest = {
+        "generated_at": time.strftime("%Y-%m-%d %H:%M:%S"),
+        "style": style,
+        "palette": palette,
+        "width_inch": SAMPLE_W,
+        "height_inch": SAMPLE_H,
+        "dpi": SAMPLE_DPI,
+        "charts": [],
+    }
+    for ct in chart_order:
+        path = os.path.join(SAMPLES_DIR, "%s.png" % ct)
+        entry = {
+            "type": ct,
+            "zh": ZH_NAME.get(ct, ct),
+            "file": "%s.png" % ct,
+            "status": "ok" if ct in results else "failed",
+        }
+        if os.path.exists(path):
+            entry["bytes"] = os.path.getsize(path)
+        manifest["charts"].append(entry)
+
+    manifest_path = os.path.join(SAMPLES_DIR, "manifest.json")
+    with open(manifest_path, "w", encoding="utf-8") as fh:
+        json.dump(manifest, fh, ensure_ascii=False, indent=2)
+        fh.write("\n")
+
+    print("Samples: %d ok, %d failed (%.1fs) -> %s"
+          % (ok, len(errors), elapsed, SAMPLES_DIR))
+    if errors:
+        for e in errors:
+            print("   FAIL %s" % e, file=sys.stderr)
+
+    return 0 if not errors else 1
+
+
+# ── showcases/ generation ──────────────────────────────────────────────────
+
+
+def _render_palette_showcase(palette_slug, style, workers=4,
+                             force=False, dry_run=False):
+    """Render one palette's full 36-chart grid."""
+    info = palette_info(palette_slug)
+    if not info:
+        return palette_slug, None, "unknown palette"
+
+    out_path = os.path.join(SHOWCASES_DIR, "%s.png" % palette_slug)
+    if os.path.exists(out_path) and not force:
+        print("   %-20s (exists, skipped)" % palette_slug)
+        return palette_slug, out_path, None
+
+    if dry_run:
+        print("   %-20s would render" % palette_slug)
+        return palette_slug, None, None
+
+    tmpdir = tempfile.mkdtemp(prefix="sf_show_%s_" % palette_slug)
+    chart_order = sorted(CHART_EXAMPLES)
+    thumbs = {}
+
+    def _do(ct):
+        data = example_path(ct)
+        png, err, _ = render_chart(
+            ct, data,
+            os.path.join(tmpdir, "%s.png" % ct),
+            style=style, palette=palette_slug,
+            width=THUMB_W, height=THUMB_H, dpi=THUMB_DPI,
+        )
+        return ct, png, err
+
+    with ThreadPoolExecutor(max_workers=workers) as pool:
+        for ct, png, err in pool.map(_do, chart_order):
+            if png:
+                thumbs[ct] = png
+
+    # Compose grid
+    cols = max(1, int(math.ceil(math.sqrt(len(chart_order)))))
+    n_colors = len(info.get("colors", []))
+    domain_str = "/".join(info.get("domain", ["跨学科通用"]))
+    # 去掉 zh 里的序号前缀（"13 Nature 旗舰…" → "Nature 旗舰…"），
+    # 避免与标题末尾的配色代码 wx13_… 重复
+    zh_name = re.sub(r"^\s*\d+\s*", "", info.get("zh", palette_slug))
+    canvas = compose_grid(
+        thumbnails=thumbs,
+        chart_order=chart_order,
+        # 标题行: 中文标题(配色中文名) + 配色代码(slug)
+        header_text="%s  ·  %s" % (zh_name, palette_slug),
+        # 副标题: 适用领域 + 色数
+        subtitle_text="适用领域：%s  ·  %d 色"
+                      % (domain_str, n_colors),
+        cols=cols,
+        cell_w=360, cell_h=280, label_h=28, gap=10, pad=16, header_h=72,
+        footer_text=None,
+    )
+    canvas.save(out_path, dpi=(160, 160))
+    return palette_slug, out_path, None
+
+
+def generate_showcases(style="nature", workers=4, force=False, dry_run=False):
+    """Render all 34 palettes as full 36-chart grids."""
+    palettes = list_palettes()
+    n = len(palettes)
+
+    if dry_run:
+        print("[showcases] would render %d palettes -> showcases/" % n)
+        for p in palettes:
+            print("   %-20s -> showcases/%s.png" % (p, p))
+        return 0
+
+    os.makedirs(SHOWCASES_DIR, exist_ok=True)
+
+    results = {}
+    errors = []
+    t0 = time.time()
+
+    for slug in palettes:
+        print("  Rendering palette: %s ..." % slug)
+        slug_r, png, err = _render_palette_showcase(
+            slug, style, workers=workers, force=force, dry_run=False
+        )
+        if png:
+            results[slug] = png
+        elif err:
+            errors.append("%s: %s" % (slug, err))
+
+    elapsed = time.time() - t0
+
+    # Write manifest
+    manifest = {
+        "generated_at": time.strftime("%Y-%m-%d %H:%M:%S"),
+        "style": style,
+        "palettes": [],
+    }
+    for slug in palettes:
+        info = palette_info(slug)
+        path = os.path.join(SHOWCASES_DIR, "%s.png" % slug)
+        entry = {
+            "palette": slug,
+            "zh": info.get("zh", "") if info else "",
+            "kind": info.get("kind", "") if info else "",
+            "colors": info.get("colors", []) if info else [],
+            "file": "%s.png" % slug,
+            "status": "ok" if slug in results else "failed",
+        }
+        if os.path.exists(path):
+            entry["bytes"] = os.path.getsize(path)
+        manifest["palettes"].append(entry)
+
+    manifest_path = os.path.join(SHOWCASES_DIR, "manifest.json")
+    with open(manifest_path, "w", encoding="utf-8") as fh:
+        json.dump(manifest, fh, ensure_ascii=False, indent=2)
+        fh.write("\n")
+
+    print("Showcases: %d ok, %d failed (%.1fs) -> %s"
+          % (len(results), len(errors), elapsed, SHOWCASES_DIR))
+    if errors:
+        for e in errors:
+            print("   FAIL %s" % e, file=sys.stderr)
+
+    return 0 if not errors else 1
+
+
+# ── CLI ──────────────────────────────────────────────────────────────────────
+
+
+def main():
+    ap = argparse.ArgumentParser(
+        prog="generate_samples.py",
+    description="Generate samples/ (per-chart PNGs) and showcases/ "
+                "(per-palette grids) for the LIKE Sci Figures product page.",
+    )
+    ap.add_argument("--style", default="nature",
+                    help="Journal style (default: %(default)s)")
+    ap.add_argument("--palette", default="wx01_blue_red",
+                    help="Palette for samples/ (default: %(default)s)")
+    ap.add_argument("--workers", type=int, default=4,
+                    help="Parallel render workers (default: %(default)s)")
+    ap.add_argument("--force", action="store_true",
+                    help="Overwrite existing files")
+    ap.add_argument("--samples-only", action="store_true",
+                    help="Only generate samples/")
+    ap.add_argument("--showcases-only", action="store_true",
+                    help="Only generate showcases/")
+    ap.add_argument("--dry-run", action="store_true",
+                    help="Print plan without rendering")
+    args = ap.parse_args()
+
+    rc = 0
+    do_samples = not args.showcases_only
+    do_showcases = not args.samples_only
+
+    if do_samples:
+        print("=" * 60)
+        print("GENERATING SAMPLES/")
+        print("=" * 60)
+        rc |= generate_samples(
+            style=args.style, palette=args.palette,
+            workers=args.workers, force=args.force, dry_run=args.dry_run,
+        )
+
+    if do_showcases:
+        print("=" * 60)
+        print("GENERATING SHOWCASES/")
+        print("=" * 60)
+        rc |= generate_showcases(
+            style=args.style, workers=args.workers,
+            force=args.force, dry_run=args.dry_run,
+        )
+
+    return rc
+
+
+if __name__ == "__main__":
+    sys.exit(main())

package/scripts/palette_picker.py

@@ -0,0 +1,244 @@
+#!/usr/bin/env python3
+"""Build the colour-palette comparison poster shown at first skill invocation.
+
+Why this exists
+---------------
+This skill ships 34 curated palettes, but the *user* should decide which one
+to use (the workflow in SKILL.md asks for it the first time the skill is used
+in a task). Judging many palettes from a text table is unreliable, so this
+script renders each palette on the same handful of representative charts and
+stitches the results into one poster:
+
+    references/palette-picker.png
+
+Rows  = the palettes (name, kind, colour count, full swatch strip, domain)
+Columns = representative chart types, all rendered from the same examples/ data
+
+Every thumbnail is produced by calling gen_figure.py for real, so what the
+user sees in the poster is byte-for-byte what the skill will produce in the
+task.
+
+Usage
+-----
+    palette_picker.py                       # write references/palette-picker.png
+    palette_picker.py --out /tmp/pk.png     # write somewhere else
+    palette_picker.py --charts bar,scatter,heatmap
+    palette_picker.py --style ieee --force
+    palette_picker.py --dry-run             # render thumbnails, no poster
+
+Notes
+-----
+* The poster is a build artefact: it is skipped when the target already exists
+  (regenerate with --force), so an agent can present it immediately without
+  waiting for ~50 figure renders.
+* Requires Pillow for the compositing step. Thumbnails themselves need only
+  matplotlib (via gen_figure.py).
+* Chart-type → example mapping lives in registry.py; shared rendering helpers
+  live in render_utils.py. This module only owns the picker-specific layout.
+"""
+import argparse
+import os
+import sys
+import tempfile
+from concurrent.futures import ThreadPoolExecutor
+
+SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
+sys.path.insert(0, SCRIPT_DIR)
+
+from registry import example_path, ZH_NAME
+from render_utils import render_chart, load_fonts
+
+SKILL_DIR = os.path.dirname(SCRIPT_DIR)
+DEFAULT_OUT = os.path.join(SKILL_DIR, "references", "palette-picker.png")
+
+# Representative charts: categorical, trend, grouped points, proportion, and a
+# continuous colour band (the heatmap shows how a palette reads as a colormap).
+DEFAULT_CHARTS = ["bar", "line", "scatter", "pie", "heatmap"]
+
+KIND_ZH = {"qualitative": "定性", "sequential": "顺序", "diverging": "发散"}
+KIND_RGB = {"qualitative": (0, 102, 204),
+            "sequential": (46, 125, 50),
+            "diverging": (183, 28, 28)}
+
+# ── poster geometry ────────────────────────────────────────────────────────
+HEADER_H = 106
+COL_HEAD_H = 44
+PAD = 16
+LABEL_W = 420
+CELL_W = 300
+CELL_H = 210
+GAP = 14
+SWATCH_CELL = 46
+
+
+def _render_thumbnail(chart, style, palette, tmpdir):
+    """Render one (chart, palette) thumbnail via render_utils."""
+    out = os.path.join(tmpdir, "%s_%s.png" % (palette, chart))
+    data = example_path(chart)
+    if not data:
+        return palette, chart, None, "no example file for %s" % chart
+    png, err, _ = render_chart(chart, data, out, style=style, palette=palette,
+                               width=4.0, height=2.8, dpi=120)
+    return palette, chart, png, err
+
+
+def generate(out_path, charts=DEFAULT_CHARTS, style="nature", workers=4,
+             overwrite=False, dry_run=False):
+    """Render every (palette x chart) thumbnail and compose the poster."""
+    from palettes import list_palettes, palette_info
+    from PIL import Image, ImageDraw, ImageFont
+
+    palettes = [s for s in list_palettes() if palette_info(s)]
+
+    if dry_run:
+        print("would render %d thumbnails (%d palettes x %d charts)"
+              % (len(palettes) * len(charts), len(palettes), len(charts)))
+        return 0
+    if out_path and os.path.exists(out_path) and not overwrite:
+        print("(poster already exists, skipped — use --force to rebuild)")
+        print(out_path)
+        return 0
+
+    tmpdir = tempfile.mkdtemp(prefix="sf_picker_")
+    todo = [(slug, chart) for slug in palettes for chart in charts]
+    thumbs = {}
+    errors = []
+    with ThreadPoolExecutor(max_workers=workers) as pool:
+        for slug, chart, png, err in pool.map(
+                lambda p: _render_thumbnail(p[1], style, p[0], tmpdir),
+                todo):
+            if png:
+                thumbs[(slug, chart)] = png
+            else:
+                errors.append("%s/%s: %s" % (slug, chart, err))
+
+    # ── geometry ─────────────────────────────────────────────────────────
+    width = LABEL_W + len(charts) * CELL_W + (len(charts) + 1) * GAP + PAD
+    row_h = CELL_H + GAP
+    height = PAD + HEADER_H + COL_HEAD_H + len(palettes) * row_h + PAD
+    canvas = Image.new("RGB", (width, height), "white")
+    d = ImageDraw.Draw(canvas)
+    fonts, font_used = load_fonts((30, 26, 20, 17, 14))
+    f30 = fonts.get(30) or ImageFont.load_default()
+    f20 = fonts.get(20) or ImageFont.load_default()
+    f17 = fonts.get(17) or ImageFont.load_default()
+    f14 = fonts.get(14) or ImageFont.load_default()
+
+    # ── title ────────────────────────────────────────────────────────────
+    d.rectangle([0, 0, width, HEADER_H], fill="#0B6E4F")
+    d.text((PAD + 2, 12), "LIKE Sci Figures — 配色方案选择",
+           fill="white", font=f30)
+    d.text((PAD + 2, 54),
+           "%d 套精选科研配色 × %d 类代表图表  ·  风格: %s  ·  适合多领域学术出版"
+           % (len(palettes), len(charts), style),
+           fill="#D7E1EB", font=f20)
+
+    # LIKE watermark on the right side of header
+    tag = "LIKE"
+    bbox = d.textbbox((0, 0), tag, font=f20)
+    tag_w = bbox[2] - bbox[0]
+    d.text((width - PAD - tag_w - 2, 16), tag,
+           fill="#D7E1EB", font=f20)
+
+    # ── column headers ───────────────────────────────────────────────────
+    y0 = PAD + HEADER_H
+    for i, chart in enumerate(charts):
+        x = LABEL_W + i * (CELL_W + GAP) + GAP
+        d.text((x + 8, y0 + 11), ZH_NAME.get(chart, chart),
+               fill="#111111", font=f20)
+    hy = y0 + COL_HEAD_H
+
+    # ── one row per palette ──────────────────────────────────────────────
+    for r, slug in enumerate(palettes):
+        info = palette_info(slug)
+        colors = list(info.get("colors") or [])
+        y = hy + r * row_h
+        d.rectangle([PAD, y, PAD + width - 2 * PAD, y + CELL_H],
+                    fill="#FAFBFC", outline="#E4E7EB")
+
+        lx = PAD + GAP
+        d.text((lx, y + 10), slug, fill="#111111", font=f17)
+        kind = info.get("kind") or ""
+        badge_w = 62
+        d.rounded_rectangle([lx, y + 36, lx + badge_w, y + 36 + 24],
+                            radius=4, fill=KIND_RGB.get(kind, (90, 90, 90)))
+        d.text((lx + 8, y + 40), KIND_ZH.get(kind, kind),
+               fill="white", font=f14)
+        d.text((lx + badge_w + 10, y + 40), "%d 色" % len(colors),
+               fill="#555555", font=f14)
+
+        # Domain tags
+        domain_list = info.get("domain") or []
+        domain_text = "适用:" + "/".join(domain_list[:3])
+        d.text((lx, y + 66), domain_text[:40],
+               fill="#777777", font=f14)
+
+        for i, hexc in enumerate(colors):
+            cx = lx + (i % SWATCH_CELL) * (CELL_W / float(SWATCH_CELL))
+            cy = y + 90 + (i // SWATCH_CELL) * 30
+            d.rectangle([cx, cy, cx + CELL_W / float(SWATCH_CELL), cy + 26],
+                        fill=hexc)
+            if CELL_W / float(SWATCH_CELL) >= 34:
+                d.text((cx + 4, cy + 5), hexc, fill="#000000", font=f14)
+
+        d.text((lx, y + 148), (info.get("zh") or "")[:18],
+               fill="#444444", font=f14)
+        d.text((lx, y + 172), (info.get("desc") or "")[:28],
+               fill="#777777", font=f14)
+
+        for i, chart in enumerate(charts):
+            px = LABEL_W + i * (CELL_W + GAP) + GAP
+            png = thumbs.get((slug, chart))
+            if png and os.path.exists(png):
+                im = Image.open(png).convert("RGB")
+                if im.width != CELL_W or im.height != CELL_H:
+                    im = im.resize((CELL_W, CELL_H), Image.LANCZOS)
+                d.rectangle([px, y, px + CELL_W, y + CELL_H],
+                            fill="white", outline="#DDDDDD")
+                canvas.paste(im, (px, y))
+            else:
+                d.rectangle([px, y, px + CELL_W, y + CELL_H],
+                            fill="#F1F3F5", outline="#DDDDDD")
+                d.text((px + 10, y + 95), "渲染失败", fill="#999999",
+                       font=f14)
+
+    os.makedirs(os.path.dirname(out_path) or ".", exist_ok=True)
+    canvas.save(out_path, dpi=(160, 160))
+    if errors:
+        print("WARNING: %d thumbnail(s) failed to render:" % len(errors),
+              file=sys.stderr)
+        for e in errors[:10]:
+            print("   " + e, file=sys.stderr)
+    print("Palette picker poster: %s  (%d x %d px, %d bytes, font=%s)"
+          % (out_path, width, height, os.path.getsize(out_path),
+             font_used or "default"))
+    print("Charts shown: " + ", ".join(ZH_NAME.get(c, c) for c in charts))
+    return 0 if not errors else 1
+
+
+def main():
+    ap = argparse.ArgumentParser(
+        prog="palette_picker.py",
+        description="Build the palette comparison poster shown to the user "
+                    "when the skill is first used in a task.")
+    ap.add_argument("--out", default=DEFAULT_OUT,
+                    help="Poster output path (default: references/palette-picker.png)")
+    ap.add_argument("--charts", default=",".join(DEFAULT_CHARTS),
+                    help="Comma separated representative chart types "
+                         "(default: %(default)s)")
+    ap.add_argument("--style", default="nature",
+                    help="Journal style used for every thumbnail (default: %(default)s)")
+    ap.add_argument("--workers", type=int, default=4, help="Parallel renders")
+    ap.add_argument("--force", action="store_true",
+                    help="Rebuild even if the poster already exists")
+    ap.add_argument("--dry-run", action="store_true",
+                    help="Print what would be rendered and exit")
+    args = ap.parse_args()
+    charts = [c.strip() for c in args.charts.split(",") if c.strip()]
+    return generate(args.out, charts=charts, style=args.style,
+                    workers=args.workers, overwrite=args.force,
+                    dry_run=args.dry_run) or 0
+
+
+if __name__ == "__main__":
+    sys.exit(main())