Marmelator/layout-parser
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 289 Wiki Activity

RED-9760: change compareDouble to something sensible

Browse Source
This commit is contained in:
Kilian Schuettler 2024-08-12 15:49:49 +02:00
parent 86997c880b
commit 15b7c149b5
2 changed files with 18 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/LayoutParsingPipeline.java
View File

@ -276,7 +276,7 @@ public class LayoutParsingPipeline {
List<TextPositionSequence> words = stripper.getTextPositionSequences(); List<TextPositionSequence> words = stripper.getTextPositionSequences();
if (layoutParsingType.equals(LayoutParsingType.DOCUMINE_OLD)) { if (layoutParsingType.equals(LayoutParsingType.DOCUMINE_OLD)) {
var lines = TextPositionOperations.groupByLine(new HashSet<>(words)); var lines = TextPositionOperations.groupByLine(new HashSet<>(words));
classificationDocument.getLayoutDebugLayer().addLineVisualizations2(lines, pageNumber); classificationDocument.getLayoutDebugLayer().addLineVisualizationsFromNestedTextPosition(lines, pageNumber);
words = TextPositionOperations.sortLines(lines); words = TextPositionOperations.sortLines(lines);
} }
classificationDocument.getLayoutDebugLayer().addTextVisualizations(words, pageNumber); classificationDocument.getLayoutDebugLayer().addTextVisualizations(words, pageNumber);

34
layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/visualization/LayoutDebugLayer.java
View File

@ -8,7 +8,6 @@ import java.util.Collection;
import java.util.List; import java.util.List;
import java.util.Set; import java.util.Set;
import java.util.concurrent.atomic.AtomicInteger; import java.util.concurrent.atomic.AtomicInteger;
import java.util.stream.Collectors;
import org.apache.pdfbox.pdmodel.documentinterchange.markedcontent.PDMarkedContent; import org.apache.pdfbox.pdmodel.documentinterchange.markedcontent.PDMarkedContent;
@ -152,6 +151,23 @@ public class LayoutDebugLayer extends LayoutDebugLayerConfig {
} }
public void addLineVisualizationsFromNestedTextPosition(Collection<Set<TextPositionSequence>> 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<TextPageBlock> textPageBlocks, int page) { public void addTextBlockVisualizations(List<TextPageBlock> textPageBlocks, int page) {
if (!active) { if (!active) {
@ -238,20 +254,4 @@ public class LayoutDebugLayer extends LayoutDebugLayerConfig {
} }
public void addLineVisualizations2(Collection<Set<TextPositionSequence>> 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());
}
} }