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

RED-8825: general layoutparsing improvements

Browse Source 
* fix tests
This commit is contained in:
Kilian Schuettler 2024-04-30 16:06:22 +02:00
parent ae46c5f1ca
commit d61cac8b4f
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/services/factory/SearchTextWithTextPositionFactory.java
View File

@ -2,6 +2,7 @@ package com.knecon.fforesight.service.layoutparser.processor.services.factory;
import java.awt.geom.AffineTransform; import java.awt.geom.AffineTransform;
import java.awt.geom.Rectangle2D; import java.awt.geom.Rectangle2D;
import java.util.Collection;
import java.util.Collections; import java.util.Collections;
import java.util.LinkedList; import java.util.LinkedList;
import java.util.List; import java.util.List;
@ -66,12 +67,16 @@ public class SearchTextWithTextPositionFactory {
++context.stringIdx; ++context.stringIdx;
} }
assert context.stringBuilder.length() == context.stringIdxToPositionIdx.size();
List<Rectangle2D> positions = sequences.stream() List<Rectangle2D> positions = sequences.stream()
.map(TextPositionSequence::getBoundingBox) .map(TextPositionSequence::getTextPositions)
.flatMap(Collection::stream)
.map(RedTextPosition::getInitialUserSpacePosition)
.toList(); .toList();
assert context.stringBuilder.length() == context.stringIdxToPositionIdx.size();
assert positions.size() == context.stringIdxToPositionIdx.size();
return SearchTextWithTextPositionDto.builder() return SearchTextWithTextPositionDto.builder()
.searchText(context.stringBuilder.toString()) .searchText(context.stringBuilder.toString())
.lineBreaks(context.lineBreaksStringIdx) .lineBreaks(context.lineBreaksStringIdx)