From 15b7c149b5e3071c0e35affb79f714c34a6108c1 Mon Sep 17 00:00:00 2001 From: Kilian Schuettler Date: Mon, 12 Aug 2024 15:49:49 +0200 Subject: [PATCH] RED-9760: change compareDouble to something sensible --- .../processor/LayoutParsingPipeline.java | 2 +- .../visualization/LayoutDebugLayer.java | 34 +++++++++---------- 2 files changed, 18 insertions(+), 18 deletions(-) diff --git a/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/LayoutParsingPipeline.java b/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/LayoutParsingPipeline.java index 7a785c0..2f5f181 100644 --- a/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/LayoutParsingPipeline.java +++ b/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/LayoutParsingPipeline.java @@ -276,7 +276,7 @@ public class LayoutParsingPipeline { List words = stripper.getTextPositionSequences(); if (layoutParsingType.equals(LayoutParsingType.DOCUMINE_OLD)) { var lines = TextPositionOperations.groupByLine(new HashSet<>(words)); - classificationDocument.getLayoutDebugLayer().addLineVisualizations2(lines, pageNumber); + classificationDocument.getLayoutDebugLayer().addLineVisualizationsFromNestedTextPosition(lines, pageNumber); words = TextPositionOperations.sortLines(lines); } classificationDocument.getLayoutDebugLayer().addTextVisualizations(words, pageNumber); diff --git a/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/visualization/LayoutDebugLayer.java b/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/visualization/LayoutDebugLayer.java index 2d69610..619de30 100644 --- a/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/visualization/LayoutDebugLayer.java +++ b/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/visualization/LayoutDebugLayer.java @@ -8,7 +8,6 @@ import java.util.Collection; import java.util.List; import java.util.Set; import java.util.concurrent.atomic.AtomicInteger; -import java.util.stream.Collectors; import org.apache.pdfbox.pdmodel.documentinterchange.markedcontent.PDMarkedContent; @@ -152,6 +151,23 @@ public class LayoutDebugLayer extends LayoutDebugLayerConfig { } + + public void addLineVisualizationsFromNestedTextPosition(Collection> lines, int pageNumber) { + + if (!active) { + return; + } + VisualizationsOnPage visualizationsOnPage = getOrCreateVisualizationsOnPage(pageNumber, this.lines); + visualizationsOnPage.getColoredRectangles() + .addAll(lines.stream() + .map(line -> line.stream() + .map(BoundingBox::getBBoxPdf) + .collect(RectangleTransformations.collectBBox())) + .map(line -> new ColoredRectangle(line, LINES_COLOR, 0.5f)) + .toList()); + + } + public void addTextBlockVisualizations(List textPageBlocks, int page) { if (!active) { @@ -238,20 +254,4 @@ public class LayoutDebugLayer extends LayoutDebugLayerConfig { } - public void addLineVisualizations2(Collection> lines, int pageNumber) { - - if (!active) { - return; - } - VisualizationsOnPage visualizationsOnPage = getOrCreateVisualizationsOnPage(pageNumber, this.lines); - visualizationsOnPage.getColoredRectangles() - .addAll(lines.stream() - .map(line -> line.stream() - .map(BoundingBox::getBBoxPdf) - .collect(RectangleTransformations.collectBBox())) - .map(line -> new ColoredRectangle(line, LINES_COLOR, 0.5f)) - .toList()); - - } - }