import os
from PIL import ImageFont, ImageDraw, Image
import cv2
import random

def white2transparent(img):
    """
    將白色部分設(shè)置為透明
    :param img: 需要修改的圖片
    :return: 修改后的圖片
    """
    # Image讀取形式----------------------------------------------
    # img = img.convert('RGBA')
    # img = img.convert("RGBA")  # 轉(zhuǎn)換格式，確保像素包含alpha通道
    # 加一個判斷，不是四通道就推出
    # width, height = img.size  # 長度和寬度
    # for i in range(0, width):  # 遍歷所有長度的點(diǎn)
    #     for j in range(0, height):  # 遍歷所有寬度的點(diǎn)
    #         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):  # 遍歷所有長度的點(diǎn)
        for j in range(0, height):  # 遍歷所有寬度的點(diǎn)
            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: 目標(biāo)圖片，要求四通道
    :param x1: target_img在bg_img的坐標(biāo)，如果需要調(diào)節(jié)圖片大小，請在函數(shù)外resize
    :param y1:
    :return:
    """

    # 判斷jpg圖像是否已經(jīng)為4通道
    if bg_img.shape[2] == 3 or target_img.shape[2] == 3:
        print("merge image error, image must 4 channels.")
        return

    '''
    當(dāng)疊加圖像時，可能因?yàn)榀B加位置設(shè)置不當(dāng)，導(dǎo)致png圖像的邊界超過背景jpg圖像，而程序報錯
    這里設(shè)定一系列疊加位置的限制，可以滿足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     - 手機(jī)型號
        content_list    - 內(nèi)容
        out_file        - 輸出文件

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

    # 設(shè)置背景
    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設(shè)置字體
    draw_text = ImageDraw.Draw(bg_img)
    # font = ImageFont.truetype('C:\Windows\Fonts\simhei.ttf', 30) # 設(shè)置字體，系統(tǒng)內(nèi)置，也可下載，如自由字體
    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() 只能顯示英文字符，中文會出現(xiàn)亂碼問題，
        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')

    # 添加圖標(biāo)
    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)  # 范圍內(nèi)的隨機(jī)數(shù)
        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')