From d61cac8b4ff19835129151fb320c55a4d9e7fd53 Mon Sep 17 00:00:00 2001 From: Kilian Schuettler Date: Tue, 30 Apr 2024 16:06:22 +0200 Subject: [PATCH] RED-8825: general layoutparsing improvements * fix tests --- .../factory/SearchTextWithTextPositionFactory.java | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/services/factory/SearchTextWithTextPositionFactory.java b/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/services/factory/SearchTextWithTextPositionFactory.java index 34cea76..a847e1b 100644 --- a/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/services/factory/SearchTextWithTextPositionFactory.java +++ b/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/services/factory/SearchTextWithTextPositionFactory.java @@ -2,6 +2,7 @@ package com.knecon.fforesight.service.layoutparser.processor.services.factory; import java.awt.geom.AffineTransform; import java.awt.geom.Rectangle2D; +import java.util.Collection; import java.util.Collections; import java.util.LinkedList; import java.util.List; @@ -66,12 +67,16 @@ public class SearchTextWithTextPositionFactory { ++context.stringIdx; } - assert context.stringBuilder.length() == context.stringIdxToPositionIdx.size(); List positions = sequences.stream() - .map(TextPositionSequence::getBoundingBox) + .map(TextPositionSequence::getTextPositions) + .flatMap(Collection::stream) + .map(RedTextPosition::getInitialUserSpacePosition) .toList(); + assert context.stringBuilder.length() == context.stringIdxToPositionIdx.size(); + assert positions.size() == context.stringIdxToPositionIdx.size(); + return SearchTextWithTextPositionDto.builder() .searchText(context.stringBuilder.toString()) .lineBreaks(context.lineBreaksStringIdx)