# -*- coding=utf-8 -*-
#导入图片处理模块
from PIL import Image
#导入操作命令行参数处理模块
import argparse
#import matplotlib.image as mpimg

#命令行输入参数处理
parser = argparse.ArgumentParser()

parser.add_argument('file')     #输入文件
#也可直接指定高和宽
#parser.add_argument('--width', type = int, default =80) #输出字符画宽
#parser.add_argument('--height', type = int, default = 80) #输出字符画高

#获取参数
args = parser.parse_args()

IMG = args.file
#直接指定高和宽
#WIDTH = args.width
#HEIGHT = args.height
WIDTH=80
HEIGHT=80


#输入70个字符，代替不同颜色
ascii_char = list("$@B%8&WM#*oahkbdpqwmZO0QLCJUYXzcvunxrjft/\|()1{}[]?-_+~<>i!lI;:,\"^`'. ")

# 将256灰度映射到70个字符上，256种颜色映射70个字符，因此需要除以一个参数
def get_char(r,g,b,alpha = 256):
    if alpha == 0:
        return ' '
    length = len(ascii_char)
    gray = int(0.2126 * r + 0.7152 * g + 0.0722 * b)

    unit = (256.0 + 1)/length
#除以一个参数，满足256种颜色到70个字符的映射
    return ascii_char[int(gray/unit)]

if __name__ == '__main__':

    im = Image.open(IMG)
    im = im.resize((WIDTH,HEIGHT), Image.NEAREST)

    txt = ""

    for i in range(HEIGHT):
        for j in range(WIDTH):
            txt += get_char(*im.getpixel((j,i)))
        txt += '\n'

    print(txt)