Marmelator/layout-parser
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 289 Wiki Activity

TAAS-89: fixed weird bug with empty sections

Browse Source
This commit is contained in:
Kilian Schuettler 2023-08-31 11:41:22 +02:00
parent de6760abc1
commit f87e2d75b5
2 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/model/graph/nodes/TableCell.java
View File

@ -76,16 +76,22 @@ public class TableCell implements GenericSemanticNode {
} }
if (textBlock == null) { if (textBlock == null) {
textBlock = streamAllSubNodes().filter(SemanticNode::isLeaf).map(SemanticNode::getLeafTextBlock).collect(new TextBlockCollector()); textBlock = buildTextBlock();
} }
return textBlock; return textBlock;
} }
private TextBlock buildTextBlock() {
return streamAllSubNodes().filter(SemanticNode::isLeaf).map(SemanticNode::getLeafTextBlock).collect(new TextBlockCollector());
}
@Override @Override
public String toString() { public String toString() {
return treeId + ": " + NodeType.TABLE_CELL + ": " + this.getTextBlock().buildSummary(); return treeId + ": " + NodeType.TABLE_CELL + ": " + this.buildTextBlock().buildSummary();
} }
} }

2
layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/model/table/TablePageBlock.java
View File

@ -45,7 +45,7 @@ public class TablePageBlock extends AbstractPageBlock {
@Override @Override
public boolean isEmpty() { public boolean isEmpty() {
return unrotatedColCount == 0 || unrotatedRowCount == 0; return getColCount() == 0 || getRowCount() == 0;
} }
public List<List<Cell>> getRows() { public List<List<Cell>> getRows() {