diff options
| -rwxr-xr-x | pixelterm.py | 26 |
1 files changed, 24 insertions, 2 deletions
diff --git a/pixelterm.py b/pixelterm.py index 2cc1553..7882f61 100755 --- a/pixelterm.py +++ b/pixelterm.py @@ -8,11 +8,13 @@ import os, sys, argparse from pygments.formatters import terminal256 from PIL import Image, PngImagePlugin +formatter = terminal256.Terminal256Formatter() +reset_sequence = terminal256.EscapeSequence(fg=formatter._closest_color(0,0,0), bg=formatter._closest_color(0,0,0)).reset_string() + def termify_pixels(img): sx, sy = img.size out = '' - formatter = terminal256.Terminal256Formatter() fg,bg = None,None fgd,bgd = {},{} def bgescape(color): @@ -37,14 +39,34 @@ def termify_pixels(img): fgd[color]= terminal256.EscapeSequence(fg=formatter._closest_color(r,g,b)).color_string() return fgd[color] #NOTE: This ignores the last line if there is an odd number of lines. + def balloon(x,y): + if img.getpixel((x+1, y)) != (0,255,0,127): + w = 1 + while img.getpixel((x-w, y)) == (0,255,0,127): + w += 1 + if x-w<0: + break + return '$balloon{}$'.format(w) + return '' + for y in range(0, sy-1, 2): for x in range(sx): coltop = img.getpixel((x, y)) colbot = img.getpixel((x, y+1)) + + if coltop[3] == 127: #Control colors + out += reset_sequence + out += {(255, 0, 0, 127): lambda x,y:'$\\$', + (0, 0, 255, 127): lambda x,y:'$/$', + (0, 255, 0, 127): balloon + }.get(coltop, lambda x,y:' ')(x,y) + continue + if coltop[3] != 255: coltop = (0,0,0,0) if colbot[3] != 255: colbot = (0,0,0,0) + #Da magicks: ▀█▄ c,cf = '▀','█' te,be = fgescape,bgescape @@ -54,7 +76,7 @@ def termify_pixels(img): c,te,be = cf,te,te out += te(coltop) + be(colbot) + c out = out.rstrip() + '\n' - out += terminal256.EscapeSequence(fg=formatter._closest_color(0,0,0), bg=formatter._closest_color(0,0,0)).reset_string() + out += reset_sequence return out if __name__ == '__main__': |