import math
from PIL import Image as pli
from PIL.Image import Image
import cv2
import numpy as np
from PIL import Image


def distance(p1, p2):
    return math.sqrt((p1[0] - p2[0]) ** 2 + (p1[1] - p2[1]) ** 2)


def find_edge_lengths(points):
    # 计算所有可能的边的长度
    edge_lengths = []
    for i in range(len(points)):
        for j in range(i + 1, len(points)):
            edge_lengths.append(distance(points[i], points[j]))
    return edge_lengths


def find_opposite_edges(edge_lengths):
    # 对边的长度应该是成对的
    edge_lengths.sort()
    return edge_lengths


async def determine_orientation(points):
    # 提取所有点的x和y坐标
    x_coords = [p[0] for p in points]
    y_coords = [p[1] for p in points]

    # 计算x和y坐标的最大值和最小值
    x_max, x_min = max(x_coords), min(x_coords)
    y_max, y_min = max(y_coords), min(y_coords)

    # 计算宽度和高度
    width = x_max - x_min
    height = y_max - y_min

    edge_lengths = find_edge_lengths(points)
    # 计算对边长度
    opposite_edges = find_opposite_edges(edge_lengths)

    # 判断竖着还是横着
    if height > width:
        return 1, opposite_edges
    else:
        return 0, opposite_edges


async def split_text_box(direction, opposite_edges, text_content, points):
    # 左上，右上，右下，左下
    point_lu_x, point_lu_y = points[0]
    point_ru_x, point_ru_y = points[1]
    point_rd_x, point_rd_y = points[2]
    point_ld_x, point_ld_y = points[3]
    if direction == 0:
        # 文本框 长度
        box_length_l = point_ru_x - point_lu_x
        box_length_r = point_ld_x - point_rd_x
        text_length = len(text_content)
        each_text_size = (box_length_l + box_length_r) / 2 / text_length
        min_size = min(opposite_edges)
        cha = min_size / 6
        print(f'差值：{cha}')
        word_box = []
        point_lu_x_sp = point_lu_x
        point_ru_x_sp = point_ru_x
        point_rd_x_sp = point_rd_x
        point_ld_x_sp = point_ld_x
        for char in text_content:
            word_const = {
                "text": char,
                "box": [[point_lu_x_sp, point_lu_y],
                        [point_ru_x_sp, point_ru_y],
                        [point_rd_x_sp, point_rd_y],
                        [point_ld_x_sp, point_ld_y]]
            }
            point_lu_x_sp += min_size + cha
            point_ru_x_sp += min_size + cha
            point_rd_x_sp += min_size + cha
            point_ld_x_sp += min_size + cha
            word_box.append(word_const)
        return word_box
    elif direction == 1:
        # 文本框 长度
        box_length_l = point_ru_y - point_lu_y
        box_length_r = point_ld_y - point_rd_y
        text_length = len(text_content)
        each_text_size = (box_length_l + box_length_r) / 2 / text_length
        min_size = min(opposite_edges)
        cha = min_size / 5
        print(f'差值：{cha}')
        word_box = []
        point_lu_y_sp = point_lu_y
        point_ru_y_sp = point_ru_y
        point_rd_y_sp = point_rd_y
        point_ld_y_sp = point_ld_y
        for char in text_content:
            word_const = {
                "text": char,
                "box": [[point_lu_x, point_lu_y_sp],
                        [point_ru_x, point_ru_y_sp],
                        [point_rd_x, point_rd_y_sp],
                        [point_ld_x, point_ld_y_sp]]
            }
            point_lu_y_sp += min_size + cha
            point_ru_y_sp += min_size + cha
            point_rd_y_sp += min_size + cha
            point_ld_y_sp += min_size + cha
            word_box.append(word_const)
        return word_box