83 lines
2.6 KiB
Python
Executable File
83 lines
2.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""Render a static OpenStreetMap image centred on a coordinate, with a marker.
|
|
|
|
Used by the Location quad because libshumate does not render tiles in this
|
|
environment (the official shumate-demo shows the same blank map — a library/GTK
|
|
render bug), while plain tile downloads work fine. Stitches OSM raster tiles into
|
|
one PNG with Pillow and caches them.
|
|
|
|
staticmap.py <lat> <lon> <zoom> <width_px> <height_px> <out_path>
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import io
|
|
import math
|
|
import os
|
|
import sys
|
|
import urllib.request
|
|
from pathlib import Path
|
|
|
|
from PIL import Image, ImageDraw
|
|
|
|
TILE = 256
|
|
CACHE = Path(os.environ.get("XDG_CACHE_HOME", Path.home() / ".cache")) / "astal-menu" / "tiles"
|
|
UA = "astal-menu/1.0 (personal dotfiles)"
|
|
BG = (26, 26, 26)
|
|
ACCENT = (228, 0, 70)
|
|
|
|
|
|
def _center_px(lat: float, lon: float, zoom: int) -> tuple[float, float]:
|
|
n = 2 ** zoom
|
|
lat_r = math.radians(lat)
|
|
x = (lon + 180.0) / 360.0 * n
|
|
y = (1.0 - math.asinh(math.tan(lat_r)) / math.pi) / 2.0 * n
|
|
return x * TILE, y * TILE
|
|
|
|
|
|
def _tile(z: int, x: int, y: int) -> Image.Image | None:
|
|
CACHE.mkdir(parents=True, exist_ok=True)
|
|
fp = CACHE / f"{z}_{x}_{y}.png"
|
|
if fp.exists():
|
|
try:
|
|
return Image.open(fp).convert("RGB")
|
|
except Exception:
|
|
pass
|
|
url = f"https://tile.openstreetmap.org/{z}/{x}/{y}.png"
|
|
try:
|
|
req = urllib.request.Request(url, headers={"User-Agent": UA})
|
|
data = urllib.request.urlopen(req, timeout=8).read()
|
|
except Exception as exc:
|
|
print(f"tile {z}/{x}/{y}: {exc}", file=sys.stderr)
|
|
return None
|
|
fp.write_bytes(data)
|
|
return Image.open(io.BytesIO(data)).convert("RGB")
|
|
|
|
|
|
def render(lat: float, lon: float, zoom: int, w: int, h: int, out: str) -> None:
|
|
n = 2 ** zoom
|
|
cx, cy = _center_px(lat, lon, zoom)
|
|
left, top = cx - w / 2, cy - h / 2
|
|
img = Image.new("RGB", (w, h), BG)
|
|
x0, x1 = int(left // TILE), int((left + w) // TILE)
|
|
y0, y1 = int(top // TILE), int((top + h) // TILE)
|
|
for tx in range(x0, x1 + 1):
|
|
for ty in range(y0, y1 + 1):
|
|
if ty < 0 or ty >= n:
|
|
continue
|
|
tile = _tile(zoom, tx % n, ty)
|
|
if tile is None:
|
|
continue
|
|
img.paste(tile, (int(tx * TILE - left), int(ty * TILE - top)))
|
|
d = ImageDraw.Draw(img)
|
|
mx, my, r = w // 2, h // 2, 8
|
|
d.ellipse([mx - r, my - r, mx + r, my + r], fill=ACCENT, outline=(255, 255, 255), width=2)
|
|
Path(out).parent.mkdir(parents=True, exist_ok=True)
|
|
img.save(out)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
a = sys.argv
|
|
render(float(a[1]), float(a[2]), int(a[3]), int(a[4]), int(a[5]), a[6])
|
|
print(a[6])
|