Add IOPlugin to save AbstractPlot as SVG

maarzt · maarzt · commit 115e3d82f01c · 2017-07-12T15:15:23.000+02:00
diff --git a/pom.xml b/pom.xml
@@ -128,6 +128,11 @@
 		    <groupId>org.jfree</groupId>
 		    <artifactId>jfreechart</artifactId>
 		</dependency>
+		<dependency>
+			<groupId>org.jfree</groupId>
+			<artifactId>jfreesvg</artifactId>
+			<version>3.2</version>
+		</dependency>
 		<dependency>
 			<groupId>net.imglib2</groupId>
 			<artifactId>imglib2-ij</artifactId>
diff --git a/src/main/java/net/imagej/plot/io/AbstractPlotSvgIOPlugin.java b/src/main/java/net/imagej/plot/io/AbstractPlotSvgIOPlugin.java
@@ -0,0 +1,64 @@
+package net.imagej.plot.io;
+
+import net.imagej.plot.AbstractPlot;
+import org.jfree.chart.JFreeChart;
+import org.jfree.graphics2d.svg.SVGGraphics2D;
+import org.jfree.graphics2d.svg.SVGUtils;
+import org.scijava.convert.ConvertService;
+import org.scijava.io.AbstractIOPlugin;
+import org.scijava.io.IOPlugin;
+import org.scijava.plugin.Parameter;
+import org.scijava.plugin.Plugin;
+
+import java.awt.*;
+import java.io.File;
+import java.io.IOException;
+
+/**
+ * Plugin that can write {@link AbstractPlot} as SVG file.
+ *
+ * @author Matthias Arzt
+ */
+@Plugin(type = IOPlugin.class)
+public class AbstractPlotSvgIOPlugin extends AbstractIOPlugin<AbstractPlot> {
+
+	@Parameter
+	ConvertService convertService;
+
+	@Override
+	public boolean supportsOpen(String source) {
+		return false;
+	}
+
+	@Override
+	public boolean supportsSave(String destination) {
+		return destination.endsWith(".svg");
+	}
+
+	@Override
+	public boolean supportsSave(Object data, String destination) {
+		return supportsSave(destination) &&
+				data instanceof AbstractPlot &&
+				convertService.supports(data, JFreeChart.class);
+	}
+
+	@Override
+	public AbstractPlot open(String source) throws IOException {
+		throw new UnsupportedOperationException();
+	}
+
+	@Override
+	public void save(AbstractPlot data, String destination) throws IOException {
+		if(!supportsSave(data, destination))
+			throw new IllegalArgumentException();
+		JFreeChart chart = convertService.convert(data, JFreeChart.class);
+		SVGGraphics2D g = new SVGGraphics2D(data.getPreferredWidth(), data.getPreferredWidth());
+		chart.draw(g, new Rectangle(0, 0, g.getWidth(), g.getHeight()));
+		SVGUtils.writeToSVG(new File(destination), g.getSVGElement());
+	}
+
+	@Override
+	public Class<AbstractPlot> getDataType() {
+		return AbstractPlot.class;
+	}
+}
diff --git a/src/test/java/net/imagej/plot/PlotToSvgDemo.java b/src/test/java/net/imagej/plot/PlotToSvgDemo.java
@@ -0,0 +1,49 @@
+package net.imagej.plot;
+
+import org.scijava.Context;
+import org.scijava.io.IOService;
+import org.scijava.plugin.Parameter;
+
+import java.io.IOException;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.util.List;
+import java.util.stream.Collectors;
+import java.util.stream.IntStream;
+
+/**
+ * @author
+ */
+public class PlotToSvgDemo {
+
+	@Parameter
+	private PlotService plotService;
+
+	@Parameter
+	private IOService ioService;
+
+	public static void main(String... args) throws IOException {
+		PlotToSvgDemo demo = new PlotToSvgDemo();
+		new Context().inject(demo);
+		demo.run();
+	}
+
+	private void run() throws IOException {
+		Path path = Paths.get(System.getProperty("user.home"), "chart.svg");
+		AbstractPlot plot = getExamplePlot();
+		ioService.save(plot, path.toString());
+	}
+
+	private AbstractPlot getExamplePlot() {
+		XYPlot plot = plotService.newXYPlot();
+		plot.setTitle("Hello World!");
+		plot.xAxis().setLabel("x");
+		plot.yAxis().setLabel("y");
+		List<Double> xs = IntStream.rangeClosed(0, 100).mapToObj(x -> (double) x * 2. * Math.PI / 100.).collect(Collectors.toList());
+		List<Double> ys = xs.stream().map(Math::sin).collect(Collectors.toList());
+		XYSeries series = plot.addXYSeries();
+		series.setLabel("y = sin(x)");
+		series.setValues( xs, ys );
+		return plot;
+	}
+}
