diff --git a/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/model/graph/nodes/Document.java b/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/model/graph/nodes/Document.java index e31a27c..3c71fed 100644 --- a/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/model/graph/nodes/Document.java +++ b/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/model/graph/nodes/Document.java @@ -63,6 +63,19 @@ public class Document implements GenericSemanticNode { .collect(Collectors.toList()); } + public List
getHeaderSections() { + + return streamChildrenOfType(NodeType.HEADER).map(node -> (Header) node) + .collect(Collectors.toList()); + } + + public List