Commit for upload

Author: mewo2

.gitignore

@@ -0,0 +1,2 @@
+novel
+tests

Makefile

@@ -0,0 +1,19 @@
+NOVEL = novel
+
+novel: $(NOVEL)/novel.pdf
+	
+$(NOVEL)/novel.pdf: $(NOVEL)/99/map.tex
+	cp frame.tex $(NOVEL)/novel.tex
+	cd $(NOVEL);\
+		latexmk --pdf novel.tex
+
+$(NOVEL)/99/map.tex: $(NOVEL)/99/map.png
+	for dir in $(NOVEL)/??; do \
+		cd $$dir; \
+		pandoc -t latex --chapters map.md -o map.tex; \
+		cd -; \
+	done
+
+$(NOVEL)/99/map.png:
+	python generate.py $(NOVEL)
+

describe.py

@@ -0,0 +1,111 @@
+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('`', u'\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 = "%s to %s" % (reg1, reg2)
+        f.write("# %s\n\n" % mdize(title).encode("utf8"))
+        for city in cities:
+            name = mdize(m.city_names[city])
+            f.write("## %s\n" % name.encode("utf8"))
+            for para in describe(m, city, last_city=last_city):
+                f.write("%s\n\n" % mdize(para).encode("utf8"))
+            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 xrange(n))]
+
+def do_novel(directory='tests/full', n=100):
+    modes = ["shore", "island", "mountain", "desert"]
+    last_mode = "shore"
+    for i in xrange(n):
+        mode = choose(modes, i/(n-1.))
+        if mode == last_mode:
+            mapmode = mode
+        else:
+            mapmode = last_mode + "/" + mode
+        direc = "%s/%02d" % (directory, i)
+        try:
+            os.makedirs(direc)
+        except:
+            pass
+        process_directory(direc, mode=mapmode)
+        last_mode = mode
+