实验楼——图片转字符画

参考：https://www.shiyanlou.com/courses/370/learning/?id=1191

• 自定义命令行参数：argparse模块

• 灰度值
  指黑白图像中点的颜色深度，范围一般从0到255，白色为255，黑色为0，故黑白图片也称灰度图像。

• RGB
  RGB色彩模式是工业界的一种颜色标准，是通过对红®、绿(G)、蓝(B)三个颜色通道的变化以及它们相互之间的叠加来得到各式各样的颜色

• RGB转灰度值：
  灰度值公式将像素的 RGB 值映射到灰度值（简化的 sRGB IEC61966-2.1 公式）

gray ＝ 0.2126 * r + 0.7152 * g + 0.0722 * b

• 代码：python ascii.py test_photo.png --width 120 --height 30

from PIL import Image
import argparse
import string

# 创建对象，对命令行中的参数进行处理
parser = argparse.ArgumentParser()

# 设置对应的参数名，加--则为可选参数，不加则为必须参数
parser.add_argument('file')
# 命令行参数输入时可简写为-o，但对象中的属性名是output
parser.add_argument('-o', '--output')
parser.add_argument('--width', type=int, default=80)
parser.add_argument('--height', type=int, default=80)

# 解析得到一个对象，对象中包含对应的属性
args = parser.parse_args()

IMG = args.file
OUTPUT = args.output
WIDTH = args.width
HEIGHT = args.height

# 绘图的字符列表
ascii_char = list(string.digits + string.ascii_letters + string.punctuation)

# 将0-255的灰度值映射到字符集上，第四个参数为透明度
def get_char(r, g, b,alpha=256):
    if alpha == 0:
        return ' '
    length = len(ascii_char)
    # RGB到灰度的简单换算公式
    gray = int(0.2126 * r + 0.7152 * g + 0.0722 * b)

    unit = 256 / length
    return ascii_char[int(gray/unit)]


if __name__ == '__main__':
    im = Image.open(IMG)
    # 调整图片大小对应到输出的字符画的宽度和高度，Image.NEAREST表示输出低质量的图片
    im = im.resize((WIDTH, HEIGHT), Image.NEAREST)

    txt = ''    # 记录字符

    for i in range(HEIGHT):
        for j in range(WIDTH):
            # im.getpixel 返回对应位置的像素值，元组格式，前三个对应RGB，最后一个是透明度
            txt += get_char(*im.getpixel((j, i)))
        txt += '\n'

    # or语法赋值
    filename = OUTPUT or 'output.txt'
    with open(filename, 'w') as f:
        f.write(txt)