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

#2：parser = argparse.ArgumentParser()

#3：parser.add_argument()

#4：parser.parse_args()

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

#最后調用parse_args()方法進行解析；解析成功之后即可使用

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


from PIL import Image #從PIL模塊中引入Image這個類
import argparse #引入argparse這個模塊。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

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

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

 unit = (256.0 + 1)/length #257/length
 return ascii_char[int(gray/unit)] #這個相當于是選出了灰度與哪個字符對應。

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

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

 txt = ""#txt初始值為空

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

 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)