33 lines
898 B
Python
33 lines
898 B
Python
|
# 忽略区域
|
|||
|
|
|
|||
|
|
from .tbpu import Tbpu
|
|||
|
|
|
|||
|
|
|
|||
|
|
class IgnoreArea(Tbpu):
|
|||
|
|
def __init__(self, areaList):
|
|||
|
|
self.tbpuName = "忽略区域"
|
|||
|
|
self.areaList = areaList
|
|||
|
|
|
|||
|
|
def run(self, textBlocks):
|
|||
|
|
# 返回是否矩形框 a 包含 b
|
|||
|
|
def isInBox(a, b):
|
|||
|
|
return (
|
|||
|
|
a[0][0] <= b[0][0]
|
|||
|
|
and a[0][1] <= b[0][1]
|
|||
|
|
and a[2][0] >= b[2][0]
|
|||
|
|
and a[2][1] >= b[2][1]
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
newList = []
|
|||
|
|
for b in textBlocks:
|
|||
|
|
flag = True # True 为没有被忽略
|
|||
|
|
# 检测当前文块 b 是否在任何一个检测块 a 内
|
|||
|
|
for a in self.areaList:
|
|||
|
|
if isInBox(a, b["box"]):
|
|||
|
|
flag = False # 踩到任何一个块，GG
|
|||
|
|
break
|
|||
|
|
if flag: # 没有被忽略
|
|||
|
|
newList.append(b)
|
|||
|
|
|
|||
|
|
return newList
|