from PIL import Image, ImageDraw, ImageFont
import os

def gen_text_img(text, font_size=20, font_path=None):
  # args：文字內(nèi)容，文字字體大小，字體路徑
  font = ImageFont.truetype(font_path, font_size) if font_path is not None else None
  (width, length) = font.getsize(text) # 獲取文字大小
  text_img = Image.new('RGBA', (width, length))
  draw = ImageDraw.Draw(text_img)
  # 第一個tuple表示未知(left,up)，之后是文字，然后顏色，最后設置字體
  draw.text((0, 0), text, fill=(0, 0, 0), font=font)
  text_img.save('./temp_pic.jpg')
  return text_img

def trans_alpha(img, pixel):
  '''
  根據(jù)rgba的pixel調(diào)節(jié)img的透明度
  這里傳進來的pixel是一個四元組（r,g,b,alpha）
  '''
  _, _, _, alpha = img.split()
  alpha = alpha.point(lambda i: pixel[-1]*10)
  img.putalpha(alpha)  #Part2有介紹
  return img

def picture_wall_mask(text_img, edge_len, pic_dir="./user"):
  # 根據(jù)文字圖gen_text_img像生成對應的照片墻，輸入：文字圖像，各個照片邊長，照片所在路徑
  new_img = Image.new('RGBA', (text_img.size[0] * edge_len, text_img.size[1] * edge_len))
  file_list = os.listdir(pic_dir)
  img_index = 0
  for x in range(0, text_img.size[0]):
    for y in range(0, text_img.size[1]):
      pixel = text_img.getpixel((x, y))#Part2有介紹
      file_name = file_list[img_index % len(file_list)]
      try:
        img = Image.open(os.path.join(pic_dir, file_name)).convert('RGBA')#Part2有介紹
        img = img.resize((edge_len, edge_len))
        img = trans_alpha(img, pixel)
        new_img.paste(img, (x * edge_len, y * edge_len)) #指定區(qū)域替換，Part2有介紹
        img_index += 1
      except Exception as e:
        print(f"open file {file_name} failed! {e}")
  return new_img

def main(text='', font_size = 20, edge_len = 60,pic_dir = "./user", out_dir = "./out/", font_path = './demo.ttf'):
  '''
  生成照片墻
  :param text: Text of picture wall, if not defined this will generage a rectangle picture wall
  :param font_size: font size of a clear value
  :param edge_len: sub picture's egde length
  '''
  if len(text) >= 1:
    text_ = ' '.join(text)#將字符串用空格分隔開，提高展示效果
    #text_ = text
    print(f"generate text wall for '{text_}' with picture path:{pic_dir}")
    text_img = gen_text_img(text_, font_size, font_path)
    # text_img.show()
    img_ascii = picture_wall_mask(text_img, edge_len, pic_dir)
    # img_ascii.show()
    img_ascii.save(out_dir + os.path.sep + '_'.join(text) + '.jpg')

if __name__ == '__main__':
  main(text='python')

import matplotlib.pyplot as plt # plt 用于顯示圖片
import matplotlib.image as mpimg # mpimg 用于讀取圖片

lena = mpimg.imread('temp_pic.png') # 讀取和代碼處于同一目錄下的 lena.png
# 此時 lena 就已經(jīng)是一個 np.array 了，可以對它進行任意處理
lena.shape #(512, 512, 3)
plt.imshow(lena) # 顯示圖片
plt.axis('off') # 不顯示坐標軸
plt.show()