首次提交

2026-08-31 16:41:12 +08:00
parent 680e028a0b
commit 9a479b6832
10 changed files with 2735 additions and 0 deletions
@@ -0,0 +1,116 @@
+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
+
+
+