diff --git a/layoutparser-service/layoutparser-service-server/src/test/java/com/knecon/fforesight/service/layoutparser/server/graph/DocumentGraphJsonWritingTest.java b/layoutparser-service/layoutparser-service-server/src/test/java/com/knecon/fforesight/service/layoutparser/server/graph/DocumentGraphJsonWritingTest.java new file mode 100644 index 0000000..a9d1521 --- /dev/null +++ b/layoutparser-service/layoutparser-service-server/src/test/java/com/knecon/fforesight/service/layoutparser/server/graph/DocumentGraphJsonWritingTest.java @@ -0,0 +1,38 @@ +package com.knecon.fforesight.service.layoutparser.server.graph; + +import java.io.File; +import java.io.FileOutputStream; + +import org.junit.jupiter.api.Disabled; +import org.junit.jupiter.api.Test; + +import com.fasterxml.jackson.databind.ObjectMapper; +import com.iqser.red.commons.jackson.ObjectMapperFactory; +import com.knecon.fforesight.service.layoutparser.internal.api.data.DocumentData; +import com.knecon.fforesight.service.layoutparser.internal.api.graph.DocumentGraph; +import com.knecon.fforesight.service.layoutparser.internal.api.mapper.DocumentDataMapper; + +import lombok.SneakyThrows; + +public class DocumentGraphJsonWritingTest extends BuildDocumentGraphTest { + + @Test + @Disabled + public void writeJsonForFileTest() { + + writeJsons("files/S-Metolachlor_RAR_01_Volume_1_2018-09-06"); + } + + @SneakyThrows + private void writeJsons(String filename) { + + DocumentGraph documentGraph = buildGraph(filename); + DocumentData documentData = DocumentDataMapper.toDocumentData(documentGraph); + ObjectMapper mapper = ObjectMapperFactory.create(); + mapper.writeValue(new FileOutputStream(File.createTempFile(filename + "_structure", "json")), documentData.getTableOfContents()); + mapper.writeValue(new FileOutputStream(File.createTempFile(filename + "_text", "json")), documentData.getTableOfContents()); + mapper.writeValue(new FileOutputStream(File.createTempFile(filename + "_positions", "json")), documentData.getTableOfContents()); + mapper.writeValue(new FileOutputStream(File.createTempFile(filename + "_pages", "json")), documentData.getTableOfContents()); + } + +} diff --git a/layoutparser-service/layoutparser-service-server/src/test/java/com/knecon/fforesight/service/layoutparser/server/graph/DocumentGraphVisualizationTest.java b/layoutparser-service/layoutparser-service-server/src/test/java/com/knecon/fforesight/service/layoutparser/server/graph/DocumentGraphVisualizationTest.java index 27a825c..b8f5d5c 100644 --- a/layoutparser-service/layoutparser-service-server/src/test/java/com/knecon/fforesight/service/layoutparser/server/graph/DocumentGraphVisualizationTest.java +++ b/layoutparser-service/layoutparser-service-server/src/test/java/com/knecon/fforesight/service/layoutparser/server/graph/DocumentGraphVisualizationTest.java @@ -33,7 +33,7 @@ public class DocumentGraphVisualizationTest extends BuildDocumentGraphTest { @Disabled public void visualizeRotatedTestDocument() { - String filename = "files/RotateTestFileWithImages"; + String filename = "files/211"; visualizePdf(filename); }