bundles/org.simantics.diagram.svg/src/org/simantics/diagram/svg/export/SVGBuilder.java

   1 package org.simantics.diagram.svg.export;\r
   2 \r
   3 import java.awt.Graphics2D;\r
   4 import java.awt.Point;\r
   5 import java.awt.RenderingHints;\r
   6 import java.awt.geom.AffineTransform;\r
   7 import java.awt.geom.Rectangle2D;\r
   8 import java.io.StringWriter;\r
   9 import java.util.UUID;\r
  10 \r
  11 import org.apache.batik.dom.GenericDOMImplementation;\r
  12 import org.apache.batik.svggen.SVGGraphics2D;\r
  13 import org.simantics.g2d.canvas.ICanvasContext;\r
  14 import org.simantics.g2d.diagram.DiagramHints;\r
  15 import org.simantics.g2d.diagram.DiagramUtils;\r
  16 import org.simantics.g2d.diagram.IDiagram;\r
  17 import org.simantics.g2d.participant.TransformUtil;\r
  18 import org.simantics.scenegraph.g2d.G2DRenderingHints;\r
  19 import org.simantics.scenegraph.utils.QualityHints;\r
  20 import org.simantics.utils.page.MarginUtils;\r
  21 import org.simantics.utils.page.MarginUtils.Margins;\r
  22 import org.w3c.dom.DOMImplementation;\r
  23 import org.w3c.dom.Document;\r
  24 import org.w3c.dom.Element;\r
  25 \r
  26 \r
  27 /**\r
  28  * org.simantics.diagram.export.ImageBuilder with SVG support \r
  29  * \r
  30  * @author Marko Luukkainen <marko.luukkainen@vtt.fi>\r
  31  *\r
  32  */\r
  33 public class SVGBuilder {\r
  34         Double dpi;\r
  35         Point size;\r
  36         double margin;\r
  37         Margins margins;\r
  38         \r
  39         /**\r
  40          * \r
  41          * @param file File to write the image (optional)\r
  42          * @param dpi Dots Per Inch\r
  43          * @param size Image size in pixels\r
  44          * @param margin percentage of image width for margins. 0.05 is 5%.\r
  45          */\r
  46         public SVGBuilder(Double dpi, Point size, double margin) {\r
  47                 this.dpi = dpi;\r
  48                 this.size = size;\r
  49                 this.margin = Math.max(margin,0.0);\r
  50                 this.margins = null;\r
  51         }\r
  52 \r
  53         /**\r
  54          * \r
  55          * @param file File to write the image (optional)\r
  56          * @param dpi Dots Per Inch\r
  57          * @param size Image size in pixels\r
  58          * @param margins Margins\r
  59          */\r
  60         public SVGBuilder(Double dpi, Point size, Margins margins) {\r
  61                 this.dpi = dpi;\r
  62                 this.size = size;\r
  63                 this.margin = 0.0;\r
  64                 this.margins = margins;\r
  65         }\r
  66         \r
  67         public static SVGGraphics2D defaultSVGGenerator() {\r
  68                 DOMImplementation domImpl =  GenericDOMImplementation.getDOMImplementation();\r
  69                 \r
  70                 // Create an instance of org.w3c.dom.Document.\r
  71             String svgNS = "http://www.w3.org/2000/svg";\r
  72             Document document = domImpl.createDocument(svgNS, "svg", null);\r
  73         \r
  74             // Create an instance of the SVG Generator.\r
  75             SVGGraphics2D svgGenerator = new SVGGraphics2D(document);\r
  76             svgGenerator.getGeneratorContext().setIDGenerator(new UniqueIDGenerator(UUID.randomUUID().toString()));\r
  77             return svgGenerator;\r
  78         }\r
  79 \r
  80         /**\r
  81          * @param canvasContext\r
  82          *            the canvas context to paint\r
  83          * @param writeResults\r
  84          *            <code>true</code> to actually write the resulting Image to the file.\r
  85          */\r
  86         public String paint(ICanvasContext canvasContext) throws Exception {\r
  87                 return paint(canvasContext, defaultSVGGenerator());\r
  88         }\r
  89         \r
  90         /**\r
  91          * @param canvasContext\r
  92          *            the canvas context to paint\r
  93          *        svgGenerator \r
  94          *            the svg generator to use\r
  95          * @param writeResults\r
  96          *            <code>true</code> to actually write the resulting Image to the file.\r
  97          */\r
  98         public String paint(ICanvasContext canvasContext, SVGGraphics2D svgGenerator) throws Exception {\r
  99                 \r
 100                 Graphics2D g2 = svgGenerator;\r
 101                 \r
 102                 IDiagram diagram = canvasContext.getDefaultHintContext().getHint(DiagramHints.KEY_DIAGRAM);\r
 103                 Rectangle2D diagramRect = DiagramUtils.getContentRect(diagram);\r
 104 \r
 105                 if (diagramRect == null)\r
 106                         diagramRect = new Rectangle2D.Double(0, 0, 100, 100);\r
 107 \r
 108                 // add margins to content.\r
 109                 double off = Math.max(diagramRect.getWidth(), diagramRect.getHeight()) * margin*0.5;\r
 110                 diagramRect = new Rectangle2D.Double(diagramRect.getX() - off, diagramRect.getY() - off, diagramRect.getWidth() + off * 2.0, diagramRect.getHeight() + off * 2.0);\r
 111 \r
 112                 if (margins != null) {\r
 113                         diagramRect = org.simantics.scenegraph.utils.GeometryUtils.expandRectangle(diagramRect,\r
 114                                         margins.top.diagramAbsolute,\r
 115                                         margins.bottom.diagramAbsolute,\r
 116                                         margins.left.diagramAbsolute,\r
 117                                         margins.right.diagramAbsolute\r
 118                                         );\r
 119                 }\r
 120                 \r
 121                 // Make sure the transformation is reset.\r
 122                 AffineTransform tr = new AffineTransform();\r
 123 \r
 124                 Rectangle2D controlArea;\r
 125                 if (dpi != null) {\r
 126                         double mmToInch = 1.0 / 25.4;\r
 127                         controlArea = new Rectangle2D.Double(0, 0, diagramRect.getWidth() * mmToInch * dpi, diagramRect.getHeight() * mmToInch * dpi);\r
 128 \r
 129                         if (margins != null) {\r
 130                                 double w = controlArea.getWidth();\r
 131                                 double h = controlArea.getHeight();\r
 132                                 controlArea = org.simantics.scenegraph.utils.GeometryUtils.expandRectangle(controlArea,\r
 133                                                 margins.top.controlAbsolute + margins.top.controlRelative * h,\r
 134                                                 margins.bottom.controlAbsolute + margins.bottom.controlRelative * h,\r
 135                                                 margins.left.controlAbsolute + margins.left.controlRelative * w,\r
 136                                                 margins.right.controlAbsolute + margins.right.controlRelative * w\r
 137                                                 );\r
 138                         }\r
 139                 } else {\r
 140                         controlArea = new Rectangle2D.Double(0, 0, size.getX(), size.getY());\r
 141                         if (margins != null) {\r
 142                                 double w = controlArea.getWidth();\r
 143                                 double h = controlArea.getHeight();\r
 144                                 controlArea = org.simantics.scenegraph.utils.GeometryUtils.expandRectangle(controlArea,\r
 145                                                 -(margins.top.controlAbsolute + margins.top.controlRelative * h),\r
 146                                                 -(margins.bottom.controlAbsolute + margins.bottom.controlRelative * h),\r
 147                                                 -(margins.left.controlAbsolute + margins.left.controlRelative * w),\r
 148                                                 -(margins.right.controlAbsolute + margins.right.controlRelative * w)\r
 149                                                 );\r
 150                         }\r
 151                 }\r
 152 \r
 153                 canvasContext.getSingleItem(TransformUtil.class).fitArea(controlArea, diagramRect, MarginUtils.NO_MARGINS);\r
 154                 \r
 155                 float width = (float)controlArea.getWidth();\r
 156                 float height = (float)controlArea.getHeight();\r
 157                 \r
 158 \r
 159                 QualityHints.HIGH_QUALITY_HINTS.setQuality(g2);\r
 160                 g2.setRenderingHint(RenderingHints.KEY_STROKE_CONTROL, RenderingHints.VALUE_STROKE_PURE);\r
 161 \r
 162 \r
 163                 g2.setTransform(tr);\r
 164                 g2.setClip(new Rectangle2D.Double(0, 0, controlArea.getWidth(), controlArea.getHeight()));\r
 165 \r
 166                 g2.setRenderingHint(G2DRenderingHints.KEY_CONTROL_BOUNDS, new Rectangle2D.Double(0, 0, controlArea.getWidth(), controlArea.getHeight()));\r
 167 \r
 168                 if (canvasContext.isLocked())\r
 169                         throw new IllegalStateException("cannot render image, canvas context is locked: " + canvasContext);\r
 170 \r
 171                 canvasContext.getSceneGraph().render(g2);\r
 172                         \r
 173 \r
 174                 \r
 175                 Element root = svgGenerator.getRoot();\r
 176                 root.setAttributeNS(null, "viewBox", "0 0 " + width + " " + height); \r
 177                 root.setAttributeNS(null, "height", Float.toString(height));\r
 178                 root.setAttributeNS(null, "width", Float.toString(width));\r
 179                 \r
 180                 // Finally, stream out SVG to the standard output using\r
 181             // UTF-8 encoding.\r
 182             boolean useCSS = false; // we want to use CSS style attributes\r
 183             StringWriter writer = new StringWriter();\r
 184             //svgGenerator.stream(writer, useCSS);\r
 185             svgGenerator.stream(root,writer, useCSS, false);\r
 186             return writer.toString();\r
 187         }\r
 188 }\r