520｜用Python绘制自定义照片墙

Posted 2022-05-23 二哥不像程序员

俗话说：做人苦，做男人更苦。
俗话又说：世间节日千千万，一个不过就完蛋。
520又到来了，今天交给大家一个绘制自定义照片墙的方法。

一、准备物料

• 照片若干
• Python+PIL库

二、绘制思路

1. 制定照片墙背景

• 制定背景的方式有很多种，简单点的我们可以用一张黑底白图的照片作为背景，在绘制的时候白色区域会被照片所填充，而黑色区域则会变成透明的状态。
• 本文我们使用的方式为指定文字（数字、英文、中文都可以）做为背景，然后进行图片填充。核心思想在于将文字写成白色，将背景涂成黑色。

2. 进行图片填充

• 接下来需要进行的就是基于上述背景的图片填充了，我们都知道每个图片都是由若干个像素点组成的，因此我们的核心操作在于给每个像素点添加一张图片，为了能够看清楚照片的内容，我们需要将每个像素点放大成能够看清的像素大小（可以自定义大小，文章中的代码设置成正方形，也可以设置成长方形）。
• 放大了像素点之后，为了美观我们可以将背景边缘进行透明化的处理，让整体照片墙有一种淡入淡出的感觉。

3. 按照路径读入照片并填充

• 准备好照片文件夹并提取填充，提取的时候可以按照顺序挨个提取，也可以使用随机选取、取模选取（计算hash值）等方法。

三、绘图代码

1. 加载模块

from PIL import Image, ImageDraw, ImageFont
# 解决读取图片报错
from PIL import ImageFile
ImageFile.LOAD_TRUNCATED_IMAGES = True
import os

2. 绘制背景

# 获取文字
def gen_text_img(text, font_size, font_path=None):
    '''
    输入：
    text：照片墙的样式文字
    font_size：字体的大小
    font_path：字体
    返回：
    文字图像
    '''
    font = ImageFont.truetype(font_path, font_size)
    (width, length) = font.getsize(text)
    text_img = Image.new('RGBA', (width, length))
    draw = ImageDraw.Draw(text_img)
    # 从左上角开始绘制
    draw.text((0, 0), text, fill=(0, 0, 0), font=font)
    return text_img

3. 设置透明度

def trans_alpha(img, pixel):
    '''
    R：红
    G：绿
    B：蓝
    A：透明
    '''
    _, _, _, alpha = img.split()
    alpha = alpha.point(lambda i: pixel[-1] * 10)
    img.putalpha(alpha)
    return img

4. 填充图片

def picture_wall_mask(text_img, edge_len, pic_dir):
    '''
    输入：
    text_img：文字图像
    edge_len：照片边长（用于扩大像素）
    pic_dir：路径
    '''
    # 像素扩大
    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))
            file_name = file_list[img_index % len(file_list)]
            try:
                # 导入图片
                img = Image.open(os.path.join(pic_dir, file_name)).convert(
                    'RGBA')
                img = img.resize((edge_len, edge_len))
                img = trans_alpha(img, pixel)
                # 进行替换
                new_img.paste(img,
                              (x * edge_len, y * edge_len))
                img_index += 1
            except Exception as e:
                print(f"文件打开失败：file_name + e")
    return new_img

5. 主函数

def main(text='',
         font_size=20,
         edge_len=60,
         pic_dir="./微信头",
         out_dir="./out",
         font_path='buzz_cloud_font.ttf'):
    '''
    生成照片墙
    :param text: 照片墙字符样式
    :param font_size: 字体大小
    :param edge_len: sub picture's egde length
    '''
    if len(text) >= 1:
        text_ = ' '.join(text)
        print(f"generate text wall for 'text_' with picture path:pic_dir")
        text_img = gen_text_img(text_, font_size, font_path)
        img_ascii = picture_wall_mask(text_img, edge_len, pic_dir)
        img_ascii.save(out_dir + os.path.sep + ''.join(text) + '.png')

# 执行函数(其余函数可以自行修改）
main(text='520')

6. 结果如下