Pull request #355: Renamed result field in research services to data
Merge in RED/redaction-service from ResearchRename to master * commit '22bb0b44db7a22f8088512570d4b2be6e742a887': Renamed result field in research services to data
This commit is contained in:
commit
3758720683
@ -16,6 +16,6 @@ import lombok.NoArgsConstructor;
|
|||||||
public class NerEntities {
|
public class NerEntities {
|
||||||
|
|
||||||
@Builder.Default
|
@Builder.Default
|
||||||
private Map<Integer, List<EntityRecogintionEntity>> result = new HashMap<>();
|
private Map<Integer, List<EntityRecogintionEntity>> data = new HashMap<>();
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|||||||
@ -10,6 +10,6 @@ public class ImageServiceResponse {
|
|||||||
|
|
||||||
private String dossierId;
|
private String dossierId;
|
||||||
private String fileId;
|
private String fileId;
|
||||||
private List<ImageMetadata> imageMetadata = new ArrayList<>();
|
private List<ImageMetadata> data = new ArrayList<>();
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|||||||
@ -249,8 +249,8 @@ public class EntityRedactionService {
|
|||||||
|
|
||||||
Set<Entity> entities = new HashSet<>();
|
Set<Entity> entities = new HashSet<>();
|
||||||
|
|
||||||
if (redactionServiceSettings.isNerServiceEnabled() && nerEntities.getResult().containsKey(sectionNumber)) {
|
if (redactionServiceSettings.isNerServiceEnabled() && nerEntities.getData().containsKey(sectionNumber)) {
|
||||||
nerEntities.getResult().get(sectionNumber).forEach(res -> {
|
nerEntities.getData().get(sectionNumber).forEach(res -> {
|
||||||
if (cellStarts == null || cellStarts.isEmpty()) {
|
if (cellStarts == null || cellStarts.isEmpty()) {
|
||||||
entities.add(new Entity(res.getValue(), res.getType(), res.getStartOffset(), res.getEndOffset(), headline, sectionNumber, false, false, Engine.NER, EntityType.RECOMMENDATION));
|
entities.add(new Entity(res.getValue(), res.getType(), res.getStartOffset(), res.getEndOffset(), headline, sectionNumber, false, false, Engine.NER, EntityType.RECOMMENDATION));
|
||||||
} else {
|
} else {
|
||||||
|
|||||||
@ -31,10 +31,12 @@ public class ImageService {
|
|||||||
public Map<Integer, List<PdfImage>> convertImages(String dossierId, String fileId){
|
public Map<Integer, List<PdfImage>> convertImages(String dossierId, String fileId){
|
||||||
|
|
||||||
var imageClassificationStream = redactionStorageService.getStoredObject(RedactionStorageService.StorageIdUtils.getStorageId(dossierId, fileId, FileType.IMAGE_INFO));
|
var imageClassificationStream = redactionStorageService.getStoredObject(RedactionStorageService.StorageIdUtils.getStorageId(dossierId, fileId, FileType.IMAGE_INFO));
|
||||||
|
|
||||||
|
|
||||||
ImageServiceResponse imageServiceResponse = objectMapper.readValue(imageClassificationStream, ImageServiceResponse.class);
|
ImageServiceResponse imageServiceResponse = objectMapper.readValue(imageClassificationStream, ImageServiceResponse.class);
|
||||||
|
|
||||||
Map<Integer, List<PdfImage>> images = new HashMap<>();
|
Map<Integer, List<PdfImage>> images = new HashMap<>();
|
||||||
imageServiceResponse.getImageMetadata().stream().forEach(imageMetadata -> {
|
imageServiceResponse.getData().stream().forEach(imageMetadata -> {
|
||||||
var classification = imageMetadata.getFilters().isAllPassed() ? ImageType.valueOf(imageMetadata.getClassification().getLabel().toUpperCase(Locale.ROOT)) : ImageType.OTHER;
|
var classification = imageMetadata.getFilters().isAllPassed() ? ImageType.valueOf(imageMetadata.getClassification().getLabel().toUpperCase(Locale.ROOT)) : ImageType.OTHER;
|
||||||
images.computeIfAbsent(imageMetadata.getPosition().getPageNumber() ,x -> new ArrayList<>())
|
images.computeIfAbsent(imageMetadata.getPosition().getPageNumber() ,x -> new ArrayList<>())
|
||||||
.add(new PdfImage(new RedRectangle2D(imageMetadata.getPosition().getX1(), imageMetadata.getPosition().getY1(), imageMetadata.getGeometry().getWidth(), imageMetadata.getGeometry().getHeight()), classification, imageMetadata.getPosition().getPageNumber()));
|
.add(new PdfImage(new RedRectangle2D(imageMetadata.getPosition().getX1(), imageMetadata.getPosition().getY1(), imageMetadata.getGeometry().getWidth(), imageMetadata.getGeometry().getHeight()), classification, imageMetadata.getPosition().getPageNumber()));
|
||||||
|
|||||||
@ -94,7 +94,7 @@ public class PdfSegmentationServiceTest {
|
|||||||
ImageServiceResponse imageServiceResponse = objectMapper.readValue(responseJson.getInputStream(), ImageServiceResponse.class);
|
ImageServiceResponse imageServiceResponse = objectMapper.readValue(responseJson.getInputStream(), ImageServiceResponse.class);
|
||||||
|
|
||||||
Map<Integer, List<PdfImage>> images = new HashMap<>();
|
Map<Integer, List<PdfImage>> images = new HashMap<>();
|
||||||
imageServiceResponse.getImageMetadata().stream().forEach(imageMetadata -> {
|
imageServiceResponse.getData().stream().forEach(imageMetadata -> {
|
||||||
images.computeIfAbsent(imageMetadata.getPosition().getPageNumber() ,x -> new ArrayList<>())
|
images.computeIfAbsent(imageMetadata.getPosition().getPageNumber() ,x -> new ArrayList<>())
|
||||||
.add(new PdfImage(new RedRectangle2D(imageMetadata.getPosition().getX1(), imageMetadata.getPosition().getY1(), imageMetadata.getGeometry().getWidth(), imageMetadata.getGeometry().getHeight()), ImageType.valueOf(imageMetadata.getClassification().getLabel().toUpperCase(Locale.ROOT)), imageMetadata.getPosition().getPageNumber()));
|
.add(new PdfImage(new RedRectangle2D(imageMetadata.getPosition().getX1(), imageMetadata.getPosition().getY1(), imageMetadata.getGeometry().getWidth(), imageMetadata.getGeometry().getHeight()), ImageType.valueOf(imageMetadata.getClassification().getLabel().toUpperCase(Locale.ROOT)), imageMetadata.getPosition().getPageNumber()));
|
||||||
});
|
});
|
||||||
|
|||||||
@ -1,7 +1,7 @@
|
|||||||
{
|
{
|
||||||
"dossierId": "f889853e-4bf8-49a9-aae5-c38605c6ef40",
|
"dossierId": "f889853e-4bf8-49a9-aae5-c38605c6ef40",
|
||||||
"fileId": "22ef63e29bb2a27db8497272336f6b32",
|
"fileId": "22ef63e29bb2a27db8497272336f6b32",
|
||||||
"imageMetadata": [
|
"data": [
|
||||||
{
|
{
|
||||||
"classification": {
|
"classification": {
|
||||||
"probabilities": {
|
"probabilities": {
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user