| 1 | from PIL import Image |
| 2 | from os.path import join, dirname, getmtime, exists, expanduser |
| 3 | from .config import * |
| 4 | |
| 5 | import ctypes |
| 6 | import sys |
| 7 | import os |
| 8 | |
| 9 | |
| 10 | def call_c(): |
| 11 | library = expanduser('~/.image.so') |
| 12 | sauce = join(dirname(__file__), 'image.c') |
| 13 | if not exists(library) or getmtime(sauce) > getmtime(library): |
| 14 | build = "gcc -fPIC -shared -o %s %s" % (library, sauce) |
| 15 | assert os.system(build + " >/dev/null 2>&1") == 0 |
| 16 | image_c = ctypes.cdll.LoadLibrary(library) |
| 17 | image_c.init() |
| 18 | return image_c.rgb_to_ansi |
| 19 | |
| 20 | rgb2short = call_c() |
| 21 | |
| 22 | |
| 23 | def pixel_print(ansicolor): |
| 24 | sys.stdout.write('\033[48;5;%sm \033[0m' % (ansicolor)) |
| 25 | |
| 26 | |
| 27 | def image_to_display(path,start=None,length=None): |
| 28 | rows, columns = os.popen('stty size', 'r').read().split() |
| 29 | if not start: |
| 30 | start = c['IMAGE_SHIFT'] |
| 31 | if not length: |
| 32 | length = int(columns) - 2 * start |
| 33 | i = Image.open(path) |
| 34 | i = i.convert('RGBA') |
| 35 | w, h = i.size |
| 36 | i.load() |
| 37 | width = min(w, length) |
| 38 | height = int(float(h) * (float(width) / float(w))) |
| 39 | height //= 2 |
| 40 | i = i.resize((width, height), Image.ANTIALIAS) |
| 41 | height = min(height, c['IMAGE_MAX_HEIGHT']) |
| 42 | |
| 43 | for y in xrange(height): |
| 44 | sys.stdout.write(' ' * start) |
| 45 | for x in xrange(width): |
| 46 | p = i.getpixel((x, y)) |
| 47 | r, g, b = p[:3] |
| 48 | pixel_print(rgb2short(r, g, b)) |
| 49 | sys.stdout.write('\n') |
| 50 | |
| 51 | if __name__ == '__main__': |
| 52 | image_to_display(sys.argv[1]) |