]> gerrit.simantics Code Review - simantics/platform.git/blobdiff - bundles/org.simantics.modeling.ui/src/org/simantics/modeling/ui/pdf/PDFBuilder.java
Migrated source code from Simantics SVN
[simantics/platform.git] / bundles / org.simantics.modeling.ui / src / org / simantics / modeling / ui / pdf / PDFBuilder.java
diff --git a/bundles/org.simantics.modeling.ui/src/org/simantics/modeling/ui/pdf/PDFBuilder.java b/bundles/org.simantics.modeling.ui/src/org/simantics/modeling/ui/pdf/PDFBuilder.java
new file mode 100644 (file)
index 0000000..a6d62e6
--- /dev/null
@@ -0,0 +1,214 @@
+/*******************************************************************************\r
+ * Copyright (c) 2007, 2010 Association for Decentralized Information Management\r
+ * in Industry THTH ry.\r
+ * All rights reserved. This program and the accompanying materials\r
+ * are made available under the terms of the Eclipse Public License v1.0\r
+ * which accompanies this distribution, and is available at\r
+ * http://www.eclipse.org/legal/epl-v10.html\r
+ *\r
+ * Contributors:\r
+ *     VTT Technical Research Centre of Finland - initial API and implementation\r
+ *******************************************************************************/\r
+package org.simantics.modeling.ui.pdf;\r
+\r
+import java.awt.Graphics2D;\r
+import java.awt.RenderingHints;\r
+import java.awt.geom.AffineTransform;\r
+import java.awt.geom.Rectangle2D;\r
+\r
+import org.simantics.g2d.canvas.ICanvasContext;\r
+import org.simantics.g2d.diagram.DiagramHints;\r
+import org.simantics.g2d.diagram.DiagramUtils;\r
+import org.simantics.g2d.diagram.IDiagram;\r
+import org.simantics.g2d.participant.TransformUtil;\r
+import org.simantics.scenegraph.g2d.G2DPDFRenderingHints;\r
+import org.simantics.scenegraph.g2d.G2DRenderingHints;\r
+import org.simantics.scenegraph.utils.QualityHints;\r
+import org.simantics.utils.page.MarginUtils;\r
+import org.simantics.utils.page.MarginUtils.Margins;\r
+import org.simantics.utils.page.PageDesc;\r
+\r
+import com.lowagie.text.Rectangle;\r
+import com.lowagie.text.pdf.FontMapper;\r
+import com.lowagie.text.pdf.PdfContentByte;\r
+import com.lowagie.text.pdf.PdfTemplate;\r
+import com.lowagie.text.pdf.PdfWriter;\r
+\r
+/**\r
+ * Renders ICanvasContext instances into PDF documents.\r
+ * \r
+ * @author Tuukka Lehtonen\r
+ */\r
+public class PDFBuilder {\r
+\r
+    private final PdfWriter  writer;\r
+\r
+    private final FontMapper mapper;\r
+\r
+    private final Rectangle  pageSize;\r
+\r
+    private final PageDesc   pageDesc;\r
+\r
+    private final boolean    fitDiagramContentsToPageMargins;\r
+\r
+    /**\r
+     * @param writer\r
+     * @param mapper\r
+     * @param pageSize\r
+     * @param pageDesc\r
+     * @param fitDiagramContentsToPageMargins see\r
+     *        {@link PDFExportPlan#fitContentToPageMargins} for a description\r
+     *        of this parameter\r
+     */\r
+    public PDFBuilder(PdfWriter writer, FontMapper mapper, Rectangle pageSize, PageDesc pageDesc, boolean fitDiagramContentsToPageMargins) {\r
+        assert writer != null;\r
+        assert mapper != null;\r
+        assert pageSize != null;\r
+        assert pageDesc != null;\r
+\r
+        this.writer = writer;\r
+        this.mapper = mapper;\r
+        this.pageSize = pageSize;\r
+        this.pageDesc = pageDesc;\r
+        this.fitDiagramContentsToPageMargins = fitDiagramContentsToPageMargins;\r
+    }\r
+\r
+    /**\r
+     * @param canvasContext the canvas context to paint\r
+     * @param writeResults <code>true</code> to actually write the resulting PDF\r
+     *        content into the {@link PdfWriter} used to construct this\r
+     *        PDFBuilder\r
+     */\r
+    public void paint(ICanvasContext canvasContext, boolean writeResults) {\r
+        // Specify rendering template size in points.\r
+        float pw = pageSize.getWidth();\r
+        float ph = pageSize.getHeight();\r
+        PdfContentByte cb = writer.getDirectContent();\r
+        PdfTemplate tp = cb.createTemplate(pw, ph);\r
+        Graphics2D g2 = tp.createGraphics(pw, ph, mapper);\r
+        g2.setRenderingHint(G2DPDFRenderingHints.KEY_PDF_WRITER, tp.getPdfWriter());\r
+        g2.setRenderingHint(G2DPDFRenderingHints.KEY_PDF_BYTECONTENT, tp);\r
+        g2.setRenderingHint(G2DPDFRenderingHints.KEY_PDF_FONTMAPPER, mapper);\r
+\r
+        double w = pageDesc.getOrientedWidth();\r
+        double h = pageDesc.getOrientedHeight();\r
+\r
+        //System.out.println("PDFBuilder.paint: page in millimeters " + w + " x " + h + ", page in points " + pw + " x " + ph);\r
+\r
+        try {\r
+            QualityHints.HIGH_QUALITY_HINTS.setQuality(g2);\r
+            g2.setRenderingHint(RenderingHints.KEY_STROKE_CONTROL, RenderingHints.VALUE_STROKE_PURE);\r
+\r
+            // Make sure the transformation is reset.\r
+            AffineTransform tr = new AffineTransform();\r
+\r
+            Margins m = pageDesc.getMargins();\r
+            double mw = pageDesc.getOrientedWidth() - m.left.diagramAbsolute - m.right.diagramAbsolute;\r
+            double mh = pageDesc.getOrientedHeight() - m.top.diagramAbsolute - m.bottom.diagramAbsolute;\r
+\r
+            Rectangle2D controlArea = new Rectangle2D.Double(m.left.diagramAbsolute, m.top.diagramAbsolute, mw, mh);\r
+\r
+            if (fitDiagramContentsToPageMargins) {\r
+                IDiagram diagram = canvasContext.getDefaultHintContext().getHint(DiagramHints.KEY_DIAGRAM);\r
+                final Rectangle2D diagramRect = DiagramUtils.getContentRect(diagram);\r
+                if (diagramRect != null) {\r
+                    canvasContext.getSingleItem(TransformUtil.class).fitArea(controlArea, diagramRect, MarginUtils.NO_MARGINS);\r
+                }\r
+            }\r
+\r
+            // We render in millimeters, but iText expects coordinates in points.\r
+            // So we scale everything to point space.\r
+            //System.out.println("PDFBuilder.paint: input transform " + tr);\r
+            tr.scale(pw / w, ph / h);\r
+            //System.out.println("PDFBuilder.paint: point space scaled transform " + tr);\r
+\r
+            g2.setTransform(tr);\r
+            //g2.setClip(new Rectangle2D.Double(0, 0, w, h));\r
+\r
+//            if (paintMargin) {\r
+//                g2.setStroke(new BasicStroke());\r
+//                g2.setColor(Color.DARK_GRAY);\r
+//                g2.draw(new Rectangle2D.Double(pageSize.getBorderWidthLeft(), pageSize.getBorderWidthTop(), ww, hh));\r
+//            }\r
+\r
+            // DEBUG: Paint page edges\r
+//            g2.setStroke(new BasicStroke(2));\r
+//            g2.setColor(Color.MAGENTA);\r
+//            g2.draw(new Rectangle2D.Double(pageDesc.getLeftEdgePos(), pageDesc.getTopEdgePos(), pageDesc.getOrientedWidth(), pageDesc.getOrientedHeight()));\r
+\r
+            // DEBUG: paint content area, i.e. the bounds that were fitted to\r
+            // meet the printed page margins.\r
+//            g2.setColor(new Color(255, 255, 0, 128));\r
+//            g2.draw(controlArea);\r
+\r
+            // TODO: test that this doesn't break anything.\r
+            g2.setRenderingHint(G2DRenderingHints.KEY_CONTROL_BOUNDS, new Rectangle2D.Double(0, 0, w, h));\r
+\r
+            if (canvasContext.isLocked())\r
+                throw new IllegalStateException("cannot render PDF, canvas context is locked: " + canvasContext);\r
+\r
+            canvasContext.getSceneGraph().render(g2);\r
+\r
+            // Code from iText examples:\r
+//            double ew = w/2;\r
+//            double eh = h/2;\r
+//            Ellipse2D.Double circle, oval, leaf, stem;\r
+//            Area circ, ov, leaf1, leaf2, st1, st2;\r
+//            circle = new Ellipse2D.Double();\r
+//            oval = new Ellipse2D.Double();\r
+//            leaf = new Ellipse2D.Double();\r
+//            stem = new Ellipse2D.Double();\r
+//            circ = new Area(circle);\r
+//            ov = new Area(oval);\r
+//            leaf1 = new Area(leaf);\r
+//            leaf2 = new Area(leaf);\r
+//            st1 = new Area(stem);\r
+//            st2 = new Area(stem);\r
+//            g2.setColor(Color.green);\r
+//\r
+//            // Creates the first leaf by filling the intersection of two Area objects created from an ellipse.\r
+//            leaf.setFrame(ew-16, eh-29, 15.0, 15.0);\r
+//            leaf1 = new Area(leaf);\r
+//            leaf.setFrame(ew-14, eh-47, 30.0, 30.0);\r
+//            leaf2 = new Area(leaf);\r
+//            leaf1.intersect(leaf2);\r
+//            g2.fill(leaf1);\r
+//\r
+//            // Creates the second leaf.\r
+//            leaf.setFrame(ew+1, eh-29, 15.0, 15.0);\r
+//            leaf1 = new Area(leaf);\r
+//            leaf2.intersect(leaf1);\r
+//            g2.fill(leaf2);\r
+//\r
+//            g2.setColor(Color.black);\r
+//\r
+//            // Creates the stem by filling the Area resulting from the subtraction of two Area objects created from an ellipse.\r
+//            stem.setFrame(ew, eh-42, 40.0, 40.0);\r
+//            st1 = new Area(stem);\r
+//            stem.setFrame(ew+3, eh-47, 50.0, 50.0);\r
+//            st2 = new Area(stem);\r
+//            st1.subtract(st2);\r
+//            g2.fill(st1);\r
+//\r
+//            g2.setColor(Color.yellow);\r
+//\r
+//            // Creates the pear itself by filling the Area resulting from the union of two Area objects created by two different ellipses.\r
+//            circle.setFrame(ew-25, eh, 50.0, 50.0);\r
+//            oval.setFrame(ew-19, eh-20, 40.0, 70.0);\r
+//            circ = new Area(circle);\r
+//            ov = new Area(oval);\r
+//            circ.add(ov);\r
+//            g2.fill(circ);\r
+        } finally {\r
+            g2.dispose();\r
+\r
+            if (writeResults) {\r
+                //System.out.println("Storing PDF template");\r
+                float x = (float) pageDesc.getLeftEdgePos();\r
+                float y = (float) pageDesc.getTopEdgePos();\r
+                cb.addTemplate(tp, x, y);\r
+            }\r
+        }\r
+    }\r
+\r
+}\r