From c157a80630323aa0d654559636b7246e1901b598 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dominique=20Eifl=C3=A4nder?= Date: Fri, 16 Apr 2021 09:41:02 +0200 Subject: [PATCH] Fixed endless loop on documents that contains no text --- .../v1/server/segmentation/SectionsBuilderService.java | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/segmentation/SectionsBuilderService.java b/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/segmentation/SectionsBuilderService.java index 41a8cf6f..68d5574e 100644 --- a/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/segmentation/SectionsBuilderService.java +++ b/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/segmentation/SectionsBuilderService.java @@ -121,6 +121,12 @@ public class SectionsBuilderService { } } + if (paragraphMap.isEmpty()) { + Paragraph paragraph = new Paragraph(); + document.getParagraphs().add(paragraph); + paragraphMap.computeIfAbsent(1, x -> new TreeSet<>()).add(paragraph); + } + for (Page page : document.getPages()) { for (PdfImage image : page.getImages()) { SortedSet paragraphsOnPage = paragraphMap.get(page.getPageNumber());