projects
/
rainbowstream.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
delete and untrack pyenv file
[rainbowstream.git]
/
rainbowstream
/
c_image.py
diff --git
a/rainbowstream/c_image.py
b/rainbowstream/c_image.py
index ec0e47e5114dd278a561cf19ce1c3163913a7b25..94299ecbe2b01d6e293bdd8c4a9741b9191bb06f 100644
(file)
--- 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 PIL import Image
from os.path import join, dirname, getmtime, exists, expanduser
from .config import *
+from .py3patch import *
import ctypes
import sys
import ctypes
import sys
@@
-8,10
+9,13
@@
import os
def call_c():
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):
library = expanduser('~/.image.so')
sauce = join(dirname(__file__), 'image.c')
if not exists(library) or getmtime(sauce) > getmtime(library):
- build = "
g
cc -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()
assert os.system(build + " >/dev/null 2>&1") == 0
image_c = ctypes.cdll.LoadLibrary(library)
image_c.init()
@@
-21,28
+25,42
@@
rgb2short = call_c()
def pixel_print(ansicolor):
def pixel_print(ansicolor):
+ """
+ Print a pixel with given Ansi color
+ """
sys.stdout.write('\033[48;5;%sm \033[0m' % (ansicolor))
sys.stdout.write('\033[48;5;%sm \033[0m' % (ansicolor))
-def image_to_display(path):
+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']
+ if not length:
+ length = int(columns) - 2 * start
i = Image.open(path)
i = i.convert('RGBA')
w, h = i.size
i.load()
i = Image.open(path)
i = i.convert('RGBA')
w, h = i.size
i.load()
- rows, columns = os.popen('stty size', 'r').read().split()
- width = min(w, int(columns) - 2 * IMAGE_SHIFT)
+ width = min(w, length)
height = int(float(h) * (float(width) / float(w)))
height //= 2
i = i.resize((width, height), Image.ANTIALIAS)
height = int(float(h) * (float(width) / float(w)))
height //= 2
i = i.resize((width, height), Image.ANTIALIAS)
- height = min(height,
IMAGE_MAX_HEIGHT
)
+ height = min(height,
c['IMAGE_MAX_HEIGHT']
)
for y in xrange(height):
for y in xrange(height):
- print ' ' * IMAGE_SHIFT,
+ sys.stdout.write(' ' * start)
for x in xrange(width):
p = i.getpixel((x, y))
r, g, b = p[:3]
pixel_print(rgb2short(r, g, b))
for x in xrange(width):
p = i.getpixel((x, y))
r, g, b = p[:3]
pixel_print(rgb2short(r, g, b))
- print ''
+ sys.stdout.write('\n')
+
+"""
+For direct using purpose
+"""
if __name__ == '__main__':
image_to_display(sys.argv[1])
if __name__ == '__main__':
image_to_display(sys.argv[1])