# -*- coding=utf-8 -*-
###################################
#1：import argparse

#2：parser = argparse.ArgumentParser()

#3：parser.add_argument()

#4：parser.parse_args()

#解釋：首先導(dǎo)入該模塊；然后創(chuàng)建一個(gè)解析對(duì)象；然后向該對(duì)象中添加你要關(guān)注的命令行參數(shù)和選項(xiàng)，每一個(gè)add_argument方法對(duì)應(yīng)一個(gè)你要關(guān)注的參數(shù)或選項(xiàng)；

#最后調(diào)用parse_args()方法進(jìn)行解析；解析成功之后即可使用

###################################


from PIL import Image #從PIL模塊中引入Image這個(gè)類
import argparse #引入argparse這個(gè)模塊。argparse 庫是用來管理命令行參數(shù)輸入的

#命令行輸入?yún)?shù)處理
parser = argparse.ArgumentParser()

parser.add_argument('file') #輸入文件
parser.add_argument('-o', '--output') #輸出文件
parser.add_argument('--width', type = int, default = 80) #輸出字符畫寬
parser.add_argument('--height', type = int, default = 80) #輸出字符畫高

#獲取參數(shù)
args = parser.parse_args()

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

#定義一個(gè)ascii的列表，其實(shí)就是讓圖片上的灰度與字符對(duì)應(yīng)
ascii_char = list("$@B%8WM#*oahkbdpqwmZO0QLCJUYXzcvunxrjft/\|()1{}[]?-_+~>i!lI;:,\"^`'. ")

# 將256灰度映射到70個(gè)字符上
def get_char(r,g,b,alpha = 256): #這個(gè)調(diào)用跟im.getpixel函數(shù)有關(guān)，這個(gè)函數(shù)是根據(jù)圖片的橫縱坐標(biāo)，把圖片解析成r,g,b,alpha(灰度），有關(guān)的四個(gè)參數(shù)，所以這里輸入?yún)?shù)是四個(gè)
 if alpha == 0: #如果灰度是0，說明這里沒有圖片
 return ' '
 length = len(ascii_char)#計(jì)算這些字符的長度
 gray = int(0.2126 * r + 0.7152 * g + 0.0722 * b)#把圖片的RGB值轉(zhuǎn)換成灰度值

 unit = (256.0 + 1)/length #257/length
 return ascii_char[int(gray/unit)] #這個(gè)相當(dāng)于是選出了灰度與哪個(gè)字符對(duì)應(yīng)。

if __name__ == '__main__':#如果是本程序調(diào)用，則執(zhí)行以下程序

 im = Image.open(IMG)#打開圖片
 im = im.resize((WIDTH,HEIGHT), Image.NEAREST)#更改圖片的顯示比例

 txt = ""#txt初始值為空

 for i in range(HEIGHT):#i代表縱坐標(biāo)
 for j in range(WIDTH):#j代表橫坐標(biāo)
 txt += get_char(*im.getpixel((j,i)))#把圖片按照橫縱坐標(biāo)解析成r,g,b以及alpha這幾個(gè)參數(shù)，然后調(diào)用get_char函數(shù)，把對(duì)應(yīng)的圖片轉(zhuǎn)換成灰度值，把對(duì)應(yīng)值得字符存入txt中
 txt += '\n'#每行的結(jié)尾處，自動(dòng)換行

 print(txt)#在界面打印txt文件

 #字符畫輸出到文件
 if OUTPUT:
 with open(OUTPUT,'w') as f:#文件輸出
 f.write(txt)
 else:
 with open("output.txt",'w') as f:#文件輸出
 f.write(txt)