欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

用python生成一张壁纸实例代码

程序员文章站 2023-12-28 20:39:52
目录前言代码效果总结前言之前在csdn上看见用python写春联的,这次突发奇想用python制作一张壁纸,其元素包括背景、文字、图片。知识点 用pil创建一张纯色背景image.new;图片上添加文...

前言

之前在csdn上看见用python写春联的,这次突发奇想用python制作一张壁纸,其元素包括背景、文字、图片。

知识点 用pil创建一张纯色背景image.new;图片上添加文字imagedraw,cv2只能显示英文;转换图片的透明度;图片上叠加图片;jpg与png的区别。

代码

import os
from pil import imagefont, imagedraw, image
import cv2
import random

def white2transparent(img):
    """
    将白色部分设置为透明
    :param img: 需要修改的图片
    :return: 修改后的图片
    """
    # image读取形式----------------------------------------------
    # img = img.convert('rgba')
    # img = img.convert("rgba")  # 转换格式,确保像素包含alpha通道
    # 加一个判断,不是四通道就推出
    # width, height = img.size  # 长度和宽度
    # for i in range(0, width):  # 遍历所有长度的点
    #     for j in range(0, height):  # 遍历所有宽度的点
    #         data = img.getpixel((i, j))  # 获取一个像素
    #         if (data.count(0) == 4):  # rgba都是255,改成透明色
    #             img.putpixel((i, j), (255, 255, 255, 255))
    # return img
    # cv2读取形式----------------------------------------------
    width, height = img.shape[:2]  # 长度和宽度
    for i in range(0, width):  # 遍历所有长度的点
        for j in range(0, height):  # 遍历所有宽度的点
            if (img[i, j, 0], img[i, j, 1], img[i, j, 2]) == (255, 255, 255):
                img[i, j, 3] = 0
    return img


def merge_img(bg_img, target_img, x1, y1):
    """
    将图片与背景图片融合
    :param bg_img: 背景图片,要求4通道
    :param target_img: 目标图片,要求四通道
    :param x1: target_img在bg_img的坐标,如果需要调节图片大小,请在函数外resize
    :param y1:
    :return:
    """

    # 判断jpg图像是否已经为4通道
    if bg_img.shape[2] == 3 or target_img.shape[2] == 3:
        print("merge image error, image must 4 channels.")
        return

    '''
    当叠加图像时,可能因为叠加位置设置不当,导致png图像的边界超过背景jpg图像,而程序报错
    这里设定一系列叠加位置的限制,可以满足png图像超出jpg图像范围时,依然可以正常叠加
    '''
    x2 = x1 + target_img.shape[1]
    y2 = y1 + target_img.shape[0]

    xx1 = 0
    yy1 = 0
    xx2 = target_img.shape[1]
    yy2 = target_img.shape[0]

    if x1 >= bg_img.shape[1]:
        x1 -= bg_img.shape[1]
    if y1 >= bg_img.shape[0]:
        y1 -= bg_img.shape[0]

    if x2 >= bg_img.shape[1]:
        xx2 = target_img.shape[1] - (x2 - bg_img.shape[1])
        x2 = bg_img.shape[1]
    if y2 >= bg_img.shape[0]:
        yy2 = target_img.shape[0] - (y2 - bg_img.shape[0])
        y2 = bg_img.shape[0]

    # 获取要覆盖图像的alpha值,将像素值除以255,使值保持在0-1之间
    alpha_target = target_img[yy1:yy2, xx1:xx2, 3] / 255.0
    alpha_bg = 1 - alpha_target

    # 开始叠加
    for c in range(0, 3):
        bg_img[y1:y2, x1:x2, c] = ((alpha_bg * bg_img[y1:y2, x1:x2, c]) + (alpha_target * target_img[yy1:yy2, xx1:xx2, c]))

    return bg_img


def set_wallpaper(phone_model, content_list, out_file=none):
    """
        phone_model     - 手机型号
        content_list    - 内容
        out_file        - 输出文件

    """
    resolution = (0, 0)  # 分辨率
    if phone_model == 'iphone se2':
        resolution = (750, 1334)

    # 设置背景
    if resolution != (0, 0):
        bg_img = image.new('rgba', resolution, '#fbedb2')
        # bg_im = cv2.cvtcolor(np.asarray(bg_im), cv2.color_rgb2bgr)
    else:
        print("don't have this version" )
        return

    # 添加语句
    # font = cv2.font_hershey_duplex  # cv2设置字体
    draw_text = imagedraw.draw(bg_img)
    # font = imagefont.truetype('c:\windows\fonts\simhei.ttf', 30) # 设置字体,系统内置,也可下载,如*字体
    font = imagefont.truetype(r'c:\users\administrator\appdata\local\microsoft\windows\fonts\千图小兔体.ttf', 30)
    i = 0
    for content in content_list:
        # cv2.puttext(bg_im, content, (100, 100), font, 5, (254, 67, 101))  # cv2.puttext() 只能显示英文字符,中文会出现乱码问题,
        offset = 120 * i
        draw_text.text((100, 120 + offset), content, font=font, fill=(0, 0, 0))
        i += 1
    # 将包含字体的读取为cv2形式,以便添加图片
    bg_img.save('./background.png')
    bg_img = cv2.imread('./background.png', cv2.imread_unchanged)
    os.remove('./background.png')

    # 添加图标
    icon_img = cv2.imread('gold.png', cv2.imread_unchanged)
    white2transparent(icon_img)
    for j in range(1, len(content_list)+1):
        offset = 120 * j
        bg_img = merge_img(bg_img, icon_img, 66, offset)

    # 添加装饰
    decorate_milktea_img = cv2.imread('milktea.png', cv2.imread_unchanged)
    decorate_fruit_img = cv2.imread('fruit.png', cv2.imread_unchanged)
    decorate_img_list = [decorate_milktea_img, decorate_fruit_img]
    for decorate_img in decorate_img_list:
        white2transparent(decorate_img)
        random_pos_x = random.randint(0,resolution[0]-300)
        random_pos_y = random.randint(740, resolution[1]-200)  # 范围内的随机数
        bg_img = merge_img(bg_img, decorate_img, random_pos_x, random_pos_y)

    # 保存图片
    if out_file:
        cv2.imwrite('./wallpaper.png', bg_img)

    cv2.imshow('bgim', bg_img)
    cv2.waitkey(0)
    cv2.destroyallwindows()


if __name__ == '__main__':
    content_list = [
        '天天开心',
        '心想事成'
    ]
    set_wallpaper('iphone se2', content_list, './wallpaper.png')
    

效果

用python生成一张壁纸实例代码

总结

到此这篇关于用python生成一张壁纸实例代码的文章就介绍到这了,更多相关python生成壁纸内容请搜索以前的文章或继续浏览下面的相关文章希望大家以后多多支持!

相关标签: python 壁纸

上一篇:

下一篇: