Fix "map" issue that was broken

megabites2013 · Dec 12, 2017 · b9bb9c7 · b9bb9c7
1 parent a7b5a02
commit b9bb9c7
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/src/main/java/net/sourceforge/plantuml/servlet/DiagramResponse.java b/src/main/java/net/sourceforge/plantuml/servlet/DiagramResponse.java
@@ -35,11 +35,12 @@
 import net.sourceforge.plantuml.BlockUml;
 import net.sourceforge.plantuml.FileFormat;
 import net.sourceforge.plantuml.FileFormatOption;
+import net.sourceforge.plantuml.NullOutputStream;
 import net.sourceforge.plantuml.SourceStringReader;
 import net.sourceforge.plantuml.StringUtils;
 import net.sourceforge.plantuml.core.DiagramDescription;
 import net.sourceforge.plantuml.core.Diagram;
-import net.sourceforge.plantuml.servlet.utility.NullOutputStream;
+import net.sourceforge.plantuml.core.ImageData;
 import net.sourceforge.plantuml.version.Version;
 import net.sourceforge.plantuml.PSystemError;
 import net.sourceforge.plantuml.ErrorUml;
@@ -108,12 +109,13 @@ void sendMap(String uml) throws IOException {
         if (StringUtils.isDiagramCacheable(uml)) {
             addHeaderForCache(blockUml);
         }
-        String map = reader.outputImage(new NullOutputStream(),
-           new FileFormatOption(FileFormat.PNG, false)).getDescription();
-        String[] mapLines = map.split("[\\r\\n]");
-        PrintWriter httpOut = response.getWriter();
-        for (int i = 2; (i + 1) < mapLines.length; i++) {
-            httpOut.print(mapLines[i]);
+        final Diagram diagram = blockUml.getDiagram();
+        ImageData map = diagram.exportDiagram(new NullOutputStream(), 0,
+                new FileFormatOption(FileFormat.PNG, false));
+        if (map.containsCMapData()) {
+            PrintWriter httpOut = response.getWriter();
+            final String cmap = map.getCMapData("plantuml");
+            httpOut.print(cmap);
         }
    }