Marmelator/redaction-service
1
0
Fork 0
Code Issues Pull Requests 14 Actions Packages Projects Releases 622 Wiki Activity

RED-6009: Document Tree Structure

Browse Source 
* optimized table access functions
This commit is contained in:
Kilian Schuettler 2023-04-26 20:06:58 +02:00
parent 0caa6f571d
commit 560ac1080b
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/layoutparsing/document/graph/nodes/TableNode.java
View File

@ -119,13 +119,13 @@ public class TableNode implements SemanticNode {
public Stream<TableCellNode> streamCol(int col) { public Stream<TableCellNode> streamCol(int col) {
return streamTableCells().filter(tableCellNode -> tableCellNode.getCol() == col); return IntStream.range(0, numberOfRows).boxed().map(row -> getCell(row, col));
} }
public Stream<TableCellNode> streamRow(int row) { public Stream<TableCellNode> streamRow(int row) {
return streamTableCells().filter(tableCellNode -> tableCellNode.getRow() == row); return IntStream.range(0, numberOfCols).boxed().map(col -> getCell(row, col));
} }
@ -137,7 +137,7 @@ public class TableNode implements SemanticNode {
public Stream<TableCellNode> streamHeadersForCell(int row, int col) { public Stream<TableCellNode> streamHeadersForCell(int row, int col) {
return streamHeaders().filter(cell -> cell.getRow() == row || cell.getCol() == col); return Stream.concat(streamRow(row), streamCol(col)).filter(TableCellNode::isHeader);
} }