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 
* remove instanceof as much as possible
This commit is contained in:
Kilian Schuettler 2023-05-16 14:38:09 +02:00
parent 926d507583
commit f8963feac1
3 changed files with 15 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -73,7 +73,7 @@ public class Document implements GenericSemanticNode {
@Override
public Headline getHeadline() {
return streamChildrenOfType(NodeType.HEADLINE).map(node -> (Headline) node).findFirst().orElseThrow(() -> new NotFoundException("No Headlines found in this document!"));
return streamAllSubNodesOfType(NodeType.HEADLINE).map(node -> (Headline) node).findFirst().orElseThrow(() -> new NotFoundException("No Headlines found in this document!"));
}
@ -85,7 +85,7 @@ public class Document implements GenericSemanticNode {
public Stream<Image> streamAllImages() {
return streamAllSubNodes().filter(node -> node instanceof Image).map(node -> (Image) node);
return streamAllSubNodesOfType(NodeType.IMAGE).map(node -> (Image) node);
}

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

@ -55,8 +55,7 @@ public class Section implements GenericSemanticNode {
public Headline getHeadline() {
return streamChildren()//
.filter(node -> node instanceof Headline)//
return streamChildrenOfType(NodeType.HEADLINE)//
.map(node -> (Headline) node)//
.findFirst()//
.orElseGet(() -> getParent().getHeadline());

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

@ -323,7 +323,7 @@ public interface SemanticNode {
/**
* Streams all children located directly underneath this node in the DocumentTree.
* Streams all children located directly underneath this node in the DocumentTree of the provided type.
*
* @return Stream of all children
*/
@ -344,6 +344,17 @@ public interface SemanticNode {
}
/**
* Recursively streams all SemanticNodes of the provided type located underneath this node in the DocumentTree in order.
*
* @return Stream of all SubNodes
*/
default Stream<SemanticNode> streamAllSubNodesOfType(NodeType nodeType) {
return getDocumentTree().streamAllSubEntriesInOrder(getTreeId()).filter(entry -> entry.getType().equals(nodeType)).map(DocumentTree.Entry::getNode);
}
/**
* The Boundary is the start and end string offsets in the reading order of the document.
*