56 lines
1.4 KiB
Python
56 lines
1.4 KiB
Python
|
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()
|