X-Git-Url: https://vcs.fsf.org/?a=blobdiff_plain;f=rainbowstream%2Fc_image.py;h=94299ecbe2b01d6e293bdd8c4a9741b9191bb06f;hb=37cf396abd4e6bd3e0337d0e0125308d3df89a19;hp=79226232f5a785314743e423c7fab46c58414a61;hpb=c3bab4ef2c05950ac280571d9b3575beb5203b31;p=rainbowstream.git diff --git a/rainbowstream/c_image.py b/rainbowstream/c_image.py index 7922623..94299ec 100644 --- a/rainbowstream/c_image.py +++ b/rainbowstream/c_image.py @@ -1,6 +1,7 @@ from PIL import Image from os.path import join, dirname, getmtime, exists, expanduser from .config import * +from .py3patch import * import ctypes import sys @@ -8,10 +9,13 @@ import os def call_c(): + """ + Call the C program for converting RGB to Ansi colors + """ library = expanduser('~/.image.so') sauce = join(dirname(__file__), 'image.c') if not exists(library) or getmtime(sauce) > getmtime(library): - build = "gcc -fPIC -shared -o %s %s" % (library, sauce) + build = "cc -fPIC -shared -o %s %s" % (library, sauce) assert os.system(build + " >/dev/null 2>&1") == 0 image_c = ctypes.cdll.LoadLibrary(library) image_c.init() @@ -21,10 +25,16 @@ rgb2short = call_c() def pixel_print(ansicolor): + """ + Print a pixel with given Ansi color + """ sys.stdout.write('\033[48;5;%sm \033[0m' % (ansicolor)) -def image_to_display(path,start=None,length=None): +def image_to_display(path, start=None, length=None): + """ + Display an image + """ rows, columns = os.popen('stty size', 'r').read().split() if not start: start = c['IMAGE_SHIFT'] @@ -48,5 +58,9 @@ def image_to_display(path,start=None,length=None): pixel_print(rgb2short(r, g, b)) sys.stdout.write('\n') + +""" +For direct using purpose +""" if __name__ == '__main__': image_to_display(sys.argv[1])