docs: 添加涉密文件自检工具实施计划

UmiOCR-data/py_src/ocr/tbpu/parser_single_code.py

@@ -0,0 +1,73 @@
+# 排版解析-单栏-代码段
+
+from .parser_single_line import SingleLine
+from .parser_tools.line_preprocessing import linePreprocessing  # 行预处理
+
+from bisect import bisect_left
+
+
+class SingleCode(SingleLine):
+    def __init__(self):
+        self.tbpuName = "排版解析-单栏-代码段"
+
+    def merge_line(self, line):  # 合并一行
+        A = line[0]
+        ba = A["box"]
+        ha = ba[3][1] - ba[0][1]  # 块A行高
+        score = A["score"]
+        for i in range(1, len(line)):
+            B = line[i]
+            bb = B["box"]
+            ha = (ha + bb[3][1] - bb[0][1]) / 2
+            # 合并文字，补充与间距相同的空格数
+            space = 0
+            if bb[0][0] > ba[1][0]:
+                space = round((bb[0][0] - ba[1][0]) / ha)
+            A["text"] += "  " * space + B["text"]
+            # 合并包围盒
+            yTop = min(ba[0][1], ba[1][1], bb[0][1], bb[1][1])
+            yBottom = max(ba[2][1], ba[3][1], bb[2][1], bb[3][1])
+            xLeft = min(ba[0][0], ba[3][0], bb[0][0], bb[3][0])
+            xRight = max(ba[1][0], ba[2][0], bb[1][0], bb[2][0])
+            ba[0][1] = ba[1][1] = yTop  # y上
+            ba[2][1] = ba[3][1] = yBottom  # y下
+            ba[0][0] = ba[3][0] = xLeft  # x左
+            ba[1][0] = ba[2][0] = xRight  # x右
+            # 置信度
+            score += B["score"]
+        A["score"] = score / len(line)
+        del A["normalized_bbox"]
+        A["end"] = "\n"
+        return A
+
+    def indent(self, tbs):  # 分析所有行，构造缩进
+        lh = 0  # 平均行高
+        xMin = float("inf")  # 句首的最左、最右x值
+        xMax = float("-inf")
+        for tb in tbs:
+            b = tb["box"]
+            lh += b[3][1] - b[0][1]
+            x = b[0][0]
+            xMin = min(xMin, x)
+            xMax = max(xMax, x)
+        lh /= len(tbs)
+        lh2 = lh / 2
+        # 构建缩进层级列表
+        levelList = []
+        x = xMin
+        while x < xMax:
+            levelList.append(x)
+            x += lh
+        # 按照层级，为每行句首加上空格，并调整包围盒
+        for tb in tbs:
+            b = tb["box"]
+            level = bisect_left(levelList, b[0][0] + lh2) - 1  # 二分查找层级点
+            tb["text"] = "  " * level + tb["text"]  # 补充空格
+            b[0][0] = b[3][0] = xMin  # 左侧归零
+
+    def run(self, textBlocks):
+        textBlocks = linePreprocessing(textBlocks)  # 预处理
+        lines = self.get_lines(textBlocks)  # 获取每一行
+        tbs = [self.merge_line(line) for line in lines]  # 合并所有行
+        self.indent(tbs)  # 为每行添加句首缩进
+        return tbs