fixed #122,#119 Unable to export large images.

Author: Vanco

CHANGELOG.md

@@ -1,6 +1,9 @@
 # SequenceDiagram Changelog
 
 ## [Unreleased]
+### Fixed
+- Unable to export large images #122, #119
+
 ## [2.1.7]
 ### Fixed
 - Support for Mermaid when export diagram.

README.md

@@ -101,7 +101,7 @@ Please try to experience it and find what happen.
 Have fun!
 
 ## Version History
-**Current Version 2.1.6**
+**Current Version 2.1.7**
 
 versions:
 [Changelog](CHANGELOG.md)

build.gradle.kts

@@ -72,4 +72,9 @@ tasks {
         // https://plugins.jetbrains.com/docs/intellij/deployment.html#specifying-a-release-channel
         channels.set(listOf(properties("pluginVersion").split('-').getOrElse(1) { "default" }.split('.').first()))
     }
+}
+
+dependencies {
+//    implementation("ar.com.hjg:pngj:2.1.0")
+    implementation("org.apache.xmlgraphics:batik-svggen:1.14")
 }

src/main/java/org/intellij/sequencer/SequencePanel.java

@@ -319,23 +319,24 @@ public ExportAction() {
 
         public void actionPerformed(@NotNull AnActionEvent event) {
             JFileChooser fileChooser = new JFileChooser();
-            fileChooser.setSelectedFile(new File(getTitleName().replaceAll("\\.", "_") + ".png"));
+            fileChooser.setSelectedFile(new File(getTitleName().replaceAll("\\.", "_") + ".svg"));
             fileChooser.setDialogType(JFileChooser.SAVE_DIALOG);
             fileChooser.setFileFilter(new FileFilter() {
                 public boolean accept(File f) {
-                    return f.isDirectory() || f.getName().endsWith("png");
+                    return f.isDirectory() || f.getName().endsWith("svg");
                 }
 
                 public String getDescription() {
-                    return "PNG Images";
+                    return "SVG Images";
                 }
             });
             try {
                 if (fileChooser.showSaveDialog(SequencePanel.this) == JFileChooser.APPROVE_OPTION) {
                     File selectedFile = fileChooser.getSelectedFile();
-                    if (!selectedFile.getName().endsWith("png"))
-                        selectedFile = new File(selectedFile.getParentFile(), selectedFile.getName() + ".png");
-                    _display.saveImageToFile(selectedFile);
+                    if (!selectedFile.getName().endsWith("svg"))
+                        selectedFile = new File(selectedFile.getParentFile(), selectedFile.getName() + ".svg");
+//                    _display.saveImageToFile(selectedFile);
+                    _display.saveImageToSvgFile(selectedFile);
                 }
             } catch (Exception e) {
                 e.printStackTrace();

src/main/java/org/intellij/sequencer/diagram/Display.java

@@ -2,8 +2,13 @@
 
 import com.intellij.ui.JBColor;
 import com.intellij.util.ui.ImageUtil;
+import kotlin.text.Charsets;
+import org.apache.batik.dom.GenericDOMImplementation;
+import org.apache.batik.svggen.SVGGraphics2D;
 import org.intellij.sequencer.config.ConfigListener;
 import org.intellij.sequencer.config.SequenceSettingsState;
+import org.w3c.dom.DOMImplementation;
+import org.w3c.dom.Document;
 
 import javax.imageio.ImageIO;
 import javax.swing.*;
@@ -12,6 +17,7 @@
 import java.awt.event.MouseEvent;
 import java.awt.image.BufferedImage;
 import java.io.File;
+import java.io.FileWriter;
 import java.io.IOException;
 import java.util.HashMap;
 
@@ -184,6 +190,14 @@ public Dimension getFullSize() {
         return new Dimension(width, height);
     }
 
+    /**
+     * Save image as png file.
+     * Known issue: when image width * height greater than Int.MAX_VALUE, will throw <code> java.lang.NegativeArraySizeException</code>.
+     * which is the limits of <code>Raster</code>
+     * @param file file to be saved
+     * @throws IOException
+     */
+    @Deprecated
     public void saveImageToFile(File file) throws IOException {
         Dimension size = getFullSize();
         BufferedImage image = ImageUtil.createImage(size.width, size.height, BufferedImage.TYPE_INT_ARGB);
@@ -205,8 +219,24 @@ public void saveImageToFile(File file) throws IOException {
         ImageIO.write(image, "png", file);
     }
 
+    public void saveImageToSvgFile(File file) throws IOException {
+        DOMImplementation domImpl = GenericDOMImplementation.getDOMImplementation();
+        String svgNS = "http://www.w3.org/2000/svg";
+        Document document = domImpl.createDocument(svgNS, "svg", null);
+
+        SVGGraphics2D svgGraphics2D = new SVGGraphics2D(document);
+
+        paintComponentWithHeader(svgGraphics2D);
+
+        FileWriter fileWriter = new FileWriter(file);
+
+        svgGraphics2D.stream(fileWriter, false);
+
+    }
+
     public void paintComponentWithHeader(Graphics2D graphics) {
         Dimension size = getFullSize();
+        graphics.setColor(JBColor.background());
         graphics.fillRect(0, 0, size.width, size.height);
         _displayHeader.paintComponent(graphics);
         graphics.translate(0, _displayHeader.getHeight());

src/test/java/org/intellij/sequencer/generator/MethodDescriptionTest.java

@@ -4,6 +4,7 @@
 import org.junit.Before;
 import org.junit.Test;
 
+import java.awt.image.DataBufferInt;
 import java.util.ArrayList;
 
 import static org.junit.Assert.*;
@@ -48,4 +49,10 @@ public void toJson() {
 
         assertEquals(m, method);
     }
+
+    @Test(expected = NegativeArraySizeException.class)
+    public void maxSize() {
+        int size = -1233295232;
+        DataBufferInt bufferInt = new DataBufferInt(size);
+    }
 }