topo

examples/topo.py

@@ -0,0 +1,55 @@
+from __future__ import division
+
+from itertools import groupby
+from PIL import Image
+
+Image.MAX_IMAGE_PIXELS = 1000000000
+
+import axi
+import numpy as np
+import sys
+
+WIDTH = 12
+HEIGHT = 8.5
+LANDSCAPE = False
+ROWS = 36
+
+if not LANDSCAPE:
+    WIDTH, HEIGHT = HEIGHT, WIDTH
+
+def remove_flats(path):
+    paths = []
+    for k, g in groupby(path, lambda p: p[1]):
+        if k > 0:
+            paths.append(list(g))
+    return paths
+
+def main():
+    paths = []
+    im = Image.open(sys.argv[1])
+    w, h = im.size
+    data = np.asarray(im) / 255
+    # data = data ** 0.5
+    lines_per_row = int(h / ROWS)
+    for j in range(0, ROWS, 1):
+        y0 = j * lines_per_row
+        y1 = y0 + lines_per_row
+        d = data[y0:y1]
+        for q in range(0, 101, 25):
+            print j, q
+            values = np.percentile(d, q, axis=0) * 0.9
+            path = enumerate(values)
+            for path in remove_flats(path):
+                x = np.array([p[0] for p in path]) * WIDTH / w
+                y = (j - np.array([p[1] for p in path])) * HEIGHT / ROWS
+                path = zip(x, y)
+                path = axi.simplify_paths([path], 0.001)[0]
+                paths.append(path)
+    d = axi.Drawing(paths)
+    im = d.render(
+        scale=109 * 1, line_width=0.3/25.4,
+        show_axi_bounds=False, use_axi_bounds=False)
+    im.write_to_png('out.png')
+
+if __name__ == '__main__':
+    main()