30 lines
1.1 KiB
Python
30 lines
1.1 KiB
Python
|
# 排版解析-多栏-无换行
|
||
|
|
|
||
|
|
from .tbpu import Tbpu
|
||
|
|
from .parser_tools.line_preprocessing import linePreprocessing # 行预处理
|
||
|
|
from .parser_tools.gap_tree import GapTree # 间隙树排序算法
|
||
|
|
from .parser_tools.paragraph_parse import word_separator # 上下句间隔符
|
||
|
|
|
||
|
|
|
||
|
|
class MultiNone(Tbpu):
|
||
|
|
def __init__(self):
|
||
|
|
self.tbpuName = "排版解析-多栏-无换行"
|
||
|
|
|
||
|
|
# 构建算法对象，指定包围盒的元素位置
|
||
|
|
self.gtree = GapTree(lambda tb: tb["normalized_bbox"])
|
||
|
|
|
||
|
|
def run(self, textBlocks):
|
||
|
|
textBlocks = linePreprocessing(textBlocks) # 预处理
|
||
|
|
textBlocks = self.gtree.sort(textBlocks) # 构建间隙树
|
||
|
|
# 补充行尾间隔符
|
||
|
|
for i in range(len(textBlocks)):
|
||
|
|
tb = textBlocks[i]
|
||
|
|
if i < len(textBlocks) - 1:
|
||
|
|
letter1 = tb["text"][-1] # 行1结尾字母
|
||
|
|
letter2 = textBlocks[i + 1]["text"][0] # 行2开头字母
|
||
|
|
tb["end"] = word_separator(letter1, letter2) # 获取间隔符
|
||
|
|
else:
|
||
|
|
tb["end"] = "\n"
|
||
|
|
del tb["normalized_bbox"]
|
||
|
|
return textBlocks
|