Something went wrong. Try again.
fork of prime agent with changes i want but are not upstreamable probqbly (and some fixes)
Something went wrong. Try again.
7.1 kB · 185 lines
Python
at main
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186#!/usr/bin/env python3"""Render the Prime butterfly SVG as terminal-safe text art.
Usage: uv run scripts/render-logo.py [--width 60] [--threshold 96] [--style blocks] [--svg path]
Outputs the rendered art to stdout.
Requires `cairo` on the system (brew install cairo). Uses cairosvg + pillow,both pulled via uv's --with on each invocation so there's no committed venv."""
# /// script# requires-python = ">=3.10"# dependencies = ["cairosvg", "pillow"]# ///
from __future__ import annotations
import argparseimport ioimport osimport sysfrom pathlib import Path
def render(svg_path: Path, width: int, threshold: int, style: str) -> str: # Late imports so --help works without the deps installed. import cairosvg from PIL import Image
if style == "quadrants": # Quadrant block characters encode a solid 2x2 grid. Halving the SVG's # raster height compensates for terminal cells being roughly twice as tall # as they are wide while retaining twice the horizontal edge resolution. png_bytes = cairosvg.svg2png(url=str(svg_path), output_width=width * 2) img = Image.open(io.BytesIO(png_bytes)).convert("RGBA") bg = Image.new("RGBA", img.size, (0, 0, 0, 255)) bg.paste(img, (0, 0), img) img = bg.convert("L").resize((width * 2, max(2, round(img.size[1] / 2))), Image.LANCZOS) w, h = img.size padded_h = h + (-h % 2) if padded_h != h: padded = Image.new("L", (w, padded_h), 0) padded.paste(img, (0, 0)) img = padded h = padded_h px = img.load() glyphs = (" ", "▘", "▝", "▀", "▖", "▌", "▞", "▛", "▗", "▚", "▐", "▜", "▄", "▙", "▟", "█") rows = [] for y in range(0, h, 2): row = "" for x in range(0, w, 2): mask = ( (1 if px[x, y] > threshold else 0) | (2 if px[x + 1, y] > threshold else 0) | (4 if px[x, y + 1] > threshold else 0) | (8 if px[x + 1, y + 1] > threshold else 0) ) row += glyphs[mask] rows.append(row.rstrip()) elif style == "braille": # A Braille cell is a 2 x 4 pixel grid. Terminal cells are typically twice as # tall as they are wide, so those subpixels are approximately square and the # SVG keeps its intended aspect without further vertical scaling. png_bytes = cairosvg.svg2png(url=str(svg_path), output_width=width * 2) img = Image.open(io.BytesIO(png_bytes)).convert("RGBA") bg = Image.new("RGBA", img.size, (0, 0, 0, 255)) bg.paste(img, (0, 0), img) img = bg.convert("L") w, h = img.size padded_w = w + (-w % 2) padded_h = h + (-h % 4) if (padded_w != w) or (padded_h != h): padded = Image.new("L", (padded_w, padded_h), 0) padded.paste(img, (0, 0)) img = padded w, h = img.size px = img.load() dot_bits = ( (0, 0, 0), (0, 1, 1), (0, 2, 2), (1, 0, 3), (1, 1, 4), (1, 2, 5), (0, 3, 6), (1, 3, 7), ) rows = [] for y in range(0, h, 4): row: list[str] = [] for x in range(0, w, 2): mask = 0 for dx, dy, bit in dot_bits: if px[x + dx, y + dy] > threshold: mask |= 1 << bit row.append(chr(0x2800 + mask) if mask else " ") rows.append("".join(row).rstrip()) elif style == "dots": # 1 pixel = 1 character cell. Compensate for terminal cells being ~2x taller than wide # so the rendered shape keeps the original proportions. # Use ▪ (BLACK SMALL SQUARE) so adjacent cells read as a tightly packed dot matrix — # mid-dot · is too small and reads as "spaced out" against the cell. png_bytes = cairosvg.svg2png(url=str(svg_path), output_width=width * 2) img = Image.open(io.BytesIO(png_bytes)).convert("RGBA") bg = Image.new("RGBA", img.size, (0, 0, 0, 255)) bg.paste(img, (0, 0), img) img = bg.convert("L") target_h = max(1, img.size[1] // 4) img = img.resize((width, target_h), Image.LANCZOS) px = img.load() rows: list[str] = [] for y in range(target_h): row = "".join("▪" if px[x, y] > threshold else " " for x in range(width)) rows.append(row.rstrip()) else: png_bytes = cairosvg.svg2png(url=str(svg_path), output_width=width) img = Image.open(io.BytesIO(png_bytes)).convert("RGBA") bg = Image.new("RGBA", img.size, (0, 0, 0, 255)) bg.paste(img, (0, 0), img) img = bg.convert("L") w, h = img.size if h % 2: padded = Image.new("L", (w, h + 1), 0) padded.paste(img, (0, 0)) img = padded h += 1 px = img.load() table = {(False, False): " ", (True, False): "▀", (False, True): "▄", (True, True): "█"} rows = [] for y in range(0, h, 2): rows.append( "".join( table[(px[x, y] > threshold, px[x, y + 1] > threshold)] for x in range(w) ).rstrip() )
while rows and not rows[0].strip(): rows.pop(0) while rows and not rows[-1].strip(): rows.pop() if rows: left = min(len(r) - len(r.lstrip(" ")) for r in rows if r.strip()) rows = [r[left:] for r in rows] return "\n".join(rows)
def main() -> int: repo_root = Path(__file__).resolve().parent.parent default_svg = repo_root / "assets" / "brand" / "prime-butterfly.svg"
parser = argparse.ArgumentParser(description=__doc__) parser.add_argument("--width", type=int, default=60, help="output columns (default: 60)") parser.add_argument( "--threshold", type=int, default=96, help="luminance cutoff 0-255; raise for thinner strokes (default: 96)", ) parser.add_argument( "--svg", type=Path, default=default_svg, help=f"input SVG (default: {default_svg})" ) parser.add_argument( "--style", choices=("blocks", "braille", "dots", "quadrants"), default="blocks", help="blocks = half-block; braille = 2x4 dots; dots = square matrix; quadrants = solid 2x2 cells", ) args = parser.parse_args()
if not args.svg.exists(): print(f"svg not found: {args.svg}", file=sys.stderr) return 1
# Set DYLD_FALLBACK_LIBRARY_PATH for homebrew cairo on Apple Silicon. if sys.platform == "darwin" and "DYLD_FALLBACK_LIBRARY_PATH" not in os.environ: os.environ["DYLD_FALLBACK_LIBRARY_PATH"] = "/opt/homebrew/lib:/usr/local/lib"
print(render(args.svg, args.width, args.threshold, args.style)) return 0
if __name__ == "__main__": raise SystemExit(main())