import os

import PIL.Image as Image
from PIL import ImageDraw
import logging

def resize_by_width(infile, image_size):
    """按照宽度进行所需比例缩放"""
    im = Image.open(infile)
    if image_size != 0:
        (x, y) = im.size
        lv = round(x / image_size, 2) + 0.01
        x_s = int(x // lv)
        y_s = int(y // lv)
        print("x_s", x_s, y_s)
        out = im.resize((x_s, y_s), Image.ANTIALIAS)
        return out
    else:
        (x_s, y_s) = im.size
        print("x_s", x_s, y_s)
        out = im.resize((x_s, y_s), Image.ANTIALIAS)
        return out


def get_new_img_xy(infile, image_size):
    """返回一个图片的宽、高像素"""
    im = Image.open(infile)
    if image_size != 0:
        (x, y) = im.size
        lv = round(x / image_size, 2) + 0.01
        x_s = x // lv
        y_s = y // lv
        # print("x_s", x_s, y_s)
        # out = im.resize((x_s, y_s), Image.ANTIALIAS)
        return x_s, y_s
    else:   #等于0时按照原比例
        (x_s, y_s) = im.size
        return x_s, y_s


# 定义图像拼接函数
def image_compose(image_colnum, image_size, image_rownum, image_names, image_save_path, x_new, y_new):
    to_image = Image.new('RGB', (image_colnum * x_new, image_rownum * y_new))  # 创建一个新图
    # 循环遍历，把每张图片按顺序粘贴到对应位置上
    total_num = 0
    for y in range(1, image_rownum + 1):
        for x in range(1, image_colnum + 1):
            from_image = resize_by_width(image_names[image_colnum * (y - 1) + x - 1], image_size)
            # from_image = Image.open(image_names[image_colnum * (y - 1) + x - 1]).resize((image_size,image_size ), Image.ANTIALIAS)
            to_image.paste(from_image, ((x - 1) * x_new, (y - 1) * y_new))
            total_num += 1
            if total_num == len(image_names):
                break
    return to_image.save(image_save_path)  # 保存新图


def get_image_list_fullpath(dir_path):
    file_name_list = os.listdir(dir_path)
    image_fullpath_list = []
    for file_name_one in file_name_list:
        file_one_path = os.path.join(dir_path, file_name_one)
        if os.path.isfile(file_one_path):
            image_fullpath_list.append(file_one_path)
        else:
            img_path_list = get_image_list_fullpath(file_one_path)
            image_fullpath_list.extend(img_path_list)
    return image_fullpath_list


def merge_images(image_dir_path,image_size,image_colnum):
    logger = logging.getLogger('info')
    # 获取图片集地址下的所有图片名称
    image_fullpath_list = get_image_list_fullpath(image_dir_path)
    image_fullpath_list.sort()
    logger.info('-----------into merge_images check image_fullpath_list')
    logger.info(image_fullpath_list)
    # exit(image_fullpath_list)
    print("image_fullpath_list", len(image_fullpath_list), image_fullpath_list)

    image_save_path = r'{}.jpg'.format(image_dir_path)  # 图片转换后的地址
    print(image_save_path);
    # image_rownum = 4  # 图片间隔，也就是合并成一张图后，一共有几行
    image_rownum_yu = len(image_fullpath_list) % image_colnum
    if image_rownum_yu == 0:
        image_rownum = len(image_fullpath_list) // image_colnum
    else:
        image_rownum = len(image_fullpath_list) // image_colnum + 1

    x_list = []
    y_list = []
    for img_file in image_fullpath_list:
        img_x, img_y = get_new_img_xy(img_file, image_size)
        x_list.append(img_x)
        y_list.append(img_y)

    print("x_list", sorted(x_list))
    print("y_list", sorted(y_list))
    x_new = int(x_list[len(x_list) // 5 * 4])
    y_new = int(y_list[len(y_list) // 5 * 4])
    print(" x_new, y_new", x_new, y_new)
    image_compose(image_colnum, image_size, image_rownum, image_fullpath_list, image_save_path, x_new, y_new)  # 调用函数.
    return {'width':x_list[0],'height':sum(y_list),'num':len(y_list)}
    # for img_file in image_fullpath_list:
    #     resize_by_width(img_file,image_size)

def pic_frame(image_path):
    im = Image.open(image_path)
    (x, y) = im.size

    width_ratio =  0.5791387557983398
    height_ratio = 0.23751300573349*(y*4)/y
    # exit(height_ratio)
    left_ratio = 0.24959583580493927
    # exit(y*0.24852335453033447)
    top_ratio = (y*4*0.504405677318573 - (2*y))/y
    # exit(top_ratio)

    draw = ImageDraw.Draw(im)
    # draw.line([(10, 10), (50, 10), (50, 50), (10, 50), (10, 10)], fill=(255, 0, 0), width=5)

    x1 = x*left_ratio
    y1 = y*top_ratio

    x2 = x1 + (width_ratio*x)
    y2 = y1

    x3 = x2
    y3 = y1 + (height_ratio*y)

    x4 = x1
    y4 = y3

    draw.line([(x1, y1), (x2, y2), (x3, y3), (x4, y4), (x1, y1)], fill=(255, 0, 0), width=3)
    im.show()


if __name__ == '__main__':
    import PIL.Image as Image
    from PIL import ImageDraw
    import json

    str = '{"file_0":[{"Width":0.1756124496459961,"Height":0.19626516103744507,"Top":0.7157560586929321,"Left":0.5398021936416626},{"Width":0.07248549163341522,"Height":0.06120368093252182,"Top":0.9989035725593567,"Left":0.7263638973236084},{"Width":0.08033709228038788,"Height":0.05730639398097992,"Top":0.0012452681548893452,"Left":0.7212769985198975},{"Width":0.0820518508553505,"Height":0.0854501947760582,"Top":0.2064758688211441,"Left":0.485286146402359},{"Width":0.05153610557317734,"Height":0.07927924394607544,"Top":0.37273716926574707,"Left":0.0917201042175293},{"Width":0.0806659683585167,"Height":0.0720946192741394,"Top":0.08823304623365402,"Left":0.5998568534851074},{"Width":0.0257689468562603,"Height":0.09389368444681168,"Top":0.5453822612762451,"Left":0.6434875726699829},{"Width":0.015949059277772903,"Height":0.0916440337896347,"Top":0.5554134845733643,"Left":0.7340391874313354}],"file_1":[{"Width":0.1793510466814041,"Height":0.1985948383808136,"Top":0.7142194509506226,"Left":0.5382133722305298},{"Width":0.07374028861522675,"Height":0.06191982701420784,"Top":0.9988375902175903,"Left":0.7259079813957214},{"Width":0.08006057888269424,"Height":0.07217763364315033,"Top":0.08562219142913818,"Left":0.5998387336730957},{"Width":0.03472089767456055,"Height":0.10424872487783432,"Top":0.5323984622955322,"Left":0.712793231010437}],"file_2":[{"Width":0.17367419600486755,"Height":0.19108445942401886,"Top":0.7145259380340576,"Left":0.5396572351455688},{"Width":0.074335478246212,"Height":0.06117350235581398,"Top":0.9986345767974854,"Left":0.725648045539856},{"Width":0.08169088512659073,"Height":0.07204793393611908,"Top":0.08614397048950195,"Left":0.598838210105896},{"Width":0.08198709785938263,"Height":0.10042453557252884,"Top":0.21100950241088867,"Left":0.48813098669052124},{"Width":0.020618628710508347,"Height":0.09485914558172226,"Top":0.5609750747680664,"Left":0.5752671957015991}],"file_3":[{"Width":0.17078885436058044,"Height":0.1982082724571228,"Top":0.7157688140869141,"Left":0.5407952666282654},{"Width":0.08152589946985245,"Height":0.07218559831380844,"Top":0.08564376831054688,"Left":0.5996308326721191},{"Width":0.01685304567217827,"Height":0.09796549379825592,"Top":0.50201416015625,"Left":0.7796969413757324},{"Width":0.017284583300352097,"Height":0.09194132685661316,"Top":0.5535471439361572,"Left":0.7339507937431335}]}'
    str = json.loads(str)
    for index, labels in str.items():
        suffix = index.split('_')
        file_path = r"E:/test---------------/1647913280_"+suffix[1] + '.jpeg'
        im = Image.open(file_path)

        for label in labels:
            (x, y) = im.size
            width_ratio = label['Width']
            height_ratio = label['Height']
            # exit(height_ratio)
            left_ratio = label['Left']
            # exit(y*0.24852335453033447)
            top_ratio = label['Top']
            # exit(top_ratio)

            draw = ImageDraw.Draw(im)
            # draw.line([(10, 10), (50, 10), (50, 50), (10, 50), (10, 10)], fill=(255, 0, 0), width=5)

            x1 = x * left_ratio
            y1 = y * top_ratio

            x2 = x1 + (width_ratio * x)
            y2 = y1

            x3 = x2
            y3 = y1 + (height_ratio * y)

            x4 = x1
            y4 = y3

            draw.line([(x1, y1), (x2, y2), (x3, y3), (x4, y4), (x1, y1)], fill=(255, 0, 0), width=3)
        im.show()


    exit()




    # image_path = r'D:\devcode\ASJServer_test\static\ai\RBF474J66TLAGHW9111A\1647244844.jpg'  # 图片集地址
    image_path = r'D:\devcode\ASJServer_test\static\ai\RBF474J66TLAGHW9111A\1647244844\1647244844_0.jpg'  # 图片集地址
    pic_frame(image_path)
    exit()

    image_dir_path = r'E:\photo\test'  # 图片集地址
    image_size = 500  # 每张小图片的大小
    image_colnum = 2  # 合并成一张图后，一行有几个小图
    merge_images(image_dir_path, image_size, image_colnum)
