describe.py

import re
import os
import numpy as np

import terrain
from grammar import parser

grammar = open("description.grammar").read()
desc = parser(grammar)


@desc.function("word", str)
def word(director, key):
    m = director.kwargs["mapgrid"]
    return "*" + m.lang.word(key) + "*"


@desc.function("name", str)
def name(director, key):
    m = director.kwargs["mapgrid"]
    return m.lang.name(key)


def direction(m, p1, p2):
    x, y = m.vxs[p2, :] - m.vxs[p1, :]
    angle = int(4 * np.arctan2(y, x) / np.pi + 4.5) % 8
    return [
        "west",
        "south-west",
        "south",
        "south-east",
        "east",
        "north-east",
        "north",
        "north-west",
    ][angle]


def describe(m, city, last_city=None):
    paras = []
    direc = ""
    far = False
    near = False
    land = True
    sea = False
    if last_city is not None:
        direc = direction(m, city, last_city)
        path, _ = m.shortest_path(last_city, city)
        dist = terrain.distance(m.vxs[city, :], m.vxs[last_city, :])
        far = bool(dist > 0.3)
        near = bool(dist < 0.15)
        land = bool(np.mean(m.elevation[path] > 0) > 0.4)
        sea = bool(np.mean(m.elevation[path] <= 0) > 0.4)
    kwargs = {
        "lastcity": m.city_names.get(last_city, ""),
        "city": m.city_names[city],
        "region": m.region_names[city],
        "direction": direc,
        "far": far,
        "near": near,
        "islandjourney": land,
        "isseajourney": sea,
        "mapgrid": m,
    }
    paras.extend(desc("description", **kwargs).split("\n"))
    return paras


def mdize(name):
    name = re.sub("`", "\u02bb", name)
    return name


def get_map(directory, mode):
    try:
        m = terrain.load(directory + "/map.pickle")
    except:
        m = terrain.MapGrid(mode=mode)
        m.save(directory + "/map.pickle")
    return m


def draw_map(directory, m):
    m.plot(directory + "/map.png", dpi=200)


def write_description(directory, m):
    with open(directory + "/map.md", "w") as f:
        last_city = None
        cities = m.ordered_cities()
        reg1 = m.region_names[cities[0]]
        reg2 = m.region_names[cities[-1]]
        title = f"{reg1} to {reg2}"
        f.write(f"# {mdize(title)}\n\n")
        for city in cities:
            name = mdize(m.city_names[city])
            f.write(f"## {name}\n")
            for para in describe(m, city, last_city=last_city):
                f.write(f"{mdize(para)}\n\n")
            last_city = city


def process_directory(directory, mode="shore"):
    print("PROCESSING", directory, mode)
    m = get_map(directory, mode)
    draw_map(directory, m)
    write_description(directory, m)


def choose(lst, p):
    n = len(lst) - 1
    return lst[sum(np.random.random() < p for _ in range(n))]


def do_novel(directory="tests/full", n=100):
    modes = ["shore", "island", "mountain", "desert"]
    last_mode = "shore"
    for i in range(n):
        mode = choose(modes, i / (n - 1.0))
        if mode == last_mode:
            mapmode = mode
        else:
            mapmode = last_mode + "/" + mode

        direc = f"{directory}/{i:02d}"
        os.makedirs(direc, exist_ok=True)
        process_directory(direc, mode=mapmode)
        last_mode = mode