import cv2
import numpy as np
from PIL import Image, ImageDraw, ImageFont


# # 加载背景图片
# bk_img = cv2.imread("C:\\Users\\Administrator\\Pictures\\test.png")
# # 在图片上添加文字信息
# cv2.putText(bk_img, "Hello World", (100, 300), cv2.FONT_HERSHEY_SIMPLEX,
#             0.7, (255, 255, 255), 1, cv2.LINE_AA)
# # 显示图片
# cv2.imshow("add_text", bk_img)
# cv2.waitKey()
# 保存图片
# cv2.imwrite("add_text.jpg", bk_img)

def writeImgBank(path, text, x=0, y=0, tx=100, ty=100):
    """将文字写入图片指定位置"""
    # 加载背景图片
    bk_img = cv2.imread(path)
    bk_img = cv2.resize(bk_img, (tx, ty))
    # 在图片上添加文字信息
    cv2.putText(bk_img, text, (x, y), cv2.FONT_ITALIC,
                1, (0, 0, 0), 1, cv2.LINE_AA)
    return bk_img


def writeImgLicense(path, text, x=0, y=0, tx=100, ty=100):
    """将文字写入图片指定位置"""
    # 加载背景图片
    bk_img = cv2.imread(path)
    bk_img = cv2.resize(bk_img, (tx, ty))
    # 在图片上添加文字信息
    cv2.putText(bk_img, text, (x, y), cv2.FONT_ITALIC,
                0.7, (0, 0, 0), 2, cv2.LINE_AA)
    return bk_img



def writeImgId(path, text, x=0, y=0):
    """将文字写入图片指定位置"""
    # 加载背景图片
    bk_img = cv2.imread(path)
    # 在图片上添加文字信息
    cv2.putText(bk_img, text, (x, y), cv2.FONT_ITALIC,
                4, (0, 0, 0), 10, cv2.LINE_AA)
    return bk_img


def paint_chinese(im, chinese, pos=(0, 0), color=(255, 255, 255)):
    img_PIL = Image.fromarray(cv2.cvtColor(im, cv2.COLOR_BGR2RGB))
    font = ImageFont.truetype('simsun.ttc', 20)
    position = pos  # 位置
    if not isinstance(chinese, str):
        chinese = chinese.decode('utf-8')
    draw = ImageDraw.Draw(img_PIL)
    draw.text(position, chinese, font=font, fill=color)
    img = cv2.cvtColor(np.asarray(img_PIL), cv2.COLOR_RGB2BGR)
    return img


def paint_chinese_id(im, chinese, pos=(0, 0), color=(0, 0, 0)):
    img_PIL = Image.fromarray(cv2.cvtColor(im, cv2.COLOR_BGR2RGB))
    font = ImageFont.truetype('simsun.ttc', 130)
    position = pos  # 位置
    if not isinstance(chinese, str):
        chinese = chinese.decode('utf-8')
    draw = ImageDraw.Draw(img_PIL)
    draw.text(position, chinese, font=font, fill=color)
    img = cv2.cvtColor(np.asarray(img_PIL), cv2.COLOR_RGB2BGR)
    return img


def get_img(path):
    """读取图片"""
    return cv2.imread(path)