banter/banter.py

57 lines
1.8 KiB
Python
Executable File

#!/bin/sh
"""'exec /usr/bin/env python3 -u "$0" "$@" #"""
__doc__ = "little script to make irc color art that will probably get you banned"
# whee, look at this sus shebang workaround to always cause
# unbuffered mode lol
import sys, time, argparse
from PIL import Image, ImageOps
from color import closestColor
def main(imgPath, delay, ASCIIWIDTH, COLORCHAR, FILLER, SHARPNESS):
im = Image.open(imgPath, "r")
im = ImageOps.scale(im, ASCIIWIDTH * SHARPNESS / im.width)
width, height = im.size
pixel_values = list(im.getdata())
currentPixel = 0
for y in range(0, height, 2 * SHARPNESS):
line = []
lastColor = 69420
for x in range(0, width, SHARPNESS):
color = closestColor(pixel_values[width * y + x])
if color == lastColor:
colorcode = ""
else:
colorcode = COLORCHAR.format(color, color)
line.append(colorcode + (FILLER[currentPixel % len(FILLER)]))
lastColor = color
currentPixel += 1
print("".join(line))
if delay:
time.sleep(delay)
if __name__ == "__main__":
parser = argparse.ArgumentParser("banter")
parser.add_argument("file")
parser.add_argument("-d", metavar="delay", default=0)
parser.add_argument("-w", metavar="width", default=80)
parser.add_argument("-s", metavar="sharpness", default=4)
parser.add_argument("--colorfmt", metavar="format", default="\\x03{},{}")
parser.add_argument("--filler", metavar="filler", default=".")
args = parser.parse_args()
main(
args.file,
float(args.d),
int(args.w),
args.colorfmt.encode().decode("unicode_escape"),
args.filler.encode().decode("unicode_escape"),
int(args.s),
)