--- /dev/null
+/*******************************************************************************\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.util.concurrent.Semaphore;\r
+import java.util.concurrent.atomic.AtomicReference;\r
+\r
+import org.simantics.db.ReadGraph;\r
+import org.simantics.db.Resource;\r
+import org.simantics.db.common.request.PossibleIndexRoot;\r
+import org.simantics.db.common.request.UniqueRead;\r
+import org.simantics.db.common.utils.NameUtils;\r
+import org.simantics.db.exception.DatabaseException;\r
+import org.simantics.db.exception.ValidationException;\r
+import org.simantics.db.layer0.variable.Variable;\r
+import org.simantics.db.layer0.variable.Variables;\r
+import org.simantics.db.management.ISessionContext;\r
+import org.simantics.diagram.elements.DiagramNodeUtil;\r
+import org.simantics.diagram.stubs.DiagramResource;\r
+import org.simantics.g2d.canvas.Hints;\r
+import org.simantics.g2d.canvas.impl.CanvasContext;\r
+import org.simantics.g2d.scenegraph.ICanvasSceneGraphProvider;\r
+import org.simantics.modeling.requests.Node;\r
+import org.simantics.structural.stubs.StructuralResource2;\r
+import org.simantics.utils.DataContainer;\r
+import org.simantics.utils.datastructures.Pair;\r
+import org.simantics.utils.page.PageDesc;\r
+import org.simantics.utils.threads.IThreadWorkQueue;\r
+import org.simantics.utils.threads.ThreadUtils;\r
+\r
+import com.lowagie.text.Rectangle;\r
+import com.lowagie.text.pdf.FontMapper;\r
+import com.lowagie.text.pdf.PdfWriter;\r
+\r
+/**\r
+ * Rasterizes diagram into a PDF document using the com.lowagie.text plug-in.\r
+ * \r
+ * @author Tuukka Lehtonen\r
+ */\r
+public class PDFPainter {\r
+\r
+ public static boolean render(\r
+ final IThreadWorkQueue thread,\r
+ final ISessionContext sessionContext,\r
+ PDFExportPlan exportModel,\r
+ final Node node,\r
+ final PdfWriter writer,\r
+ final FontMapper mapper,\r
+ final Rectangle pageSize,\r
+ final PageDesc pageDesc,\r
+ final boolean fitDiagramContentsToPageMargins,\r
+ long timeout)\r
+ throws InterruptedException, DatabaseException\r
+ {\r
+ final DataContainer<Boolean> result = new DataContainer<Boolean>(false);\r
+ final DataContainer<DatabaseException> exception = new DataContainer<DatabaseException>();\r
+\r
+ final CanvasContext ctx = new CanvasContext(thread);\r
+ final AtomicReference<ICanvasSceneGraphProvider> sgProvider = new AtomicReference<ICanvasSceneGraphProvider>();\r
+\r
+ try {\r
+ final Semaphore done = new Semaphore(0);\r
+ // IMPORTANT: Load diagram in a different thread than the canvas context thread!\r
+ ThreadUtils.getBlockingWorkExecutor().execute(new Runnable() {\r
+ @Override\r
+ public void run() {\r
+ try {\r
+ Pair<Resource, String> modelAndRVI = sessionContext.getSession().syncRequest(new UniqueRead<Pair<Resource, String>>() {\r
+ @Override\r
+ public Pair<Resource, String> perform(ReadGraph graph) throws DatabaseException {\r
+ return new Pair<Resource, String>( resolveModel(graph, node), resolveRVI(graph, node) );\r
+ }\r
+ });\r
+ \r
+ final Boolean isSymbol = sessionContext.getSession().syncRequest(new UniqueRead<Boolean>() {\r
+ @Override\r
+ public Boolean perform(ReadGraph graph) throws DatabaseException {\r
+ StructuralResource2 STR = StructuralResource2.getInstance(graph);\r
+ DiagramResource DIA = DiagramResource.getInstance(graph);\r
+ Resource possibleSymbol = graph.getPossibleObject(node.getDiagramResource(), STR.Defines);\r
+ return possibleSymbol != null && graph.isInstanceOf(possibleSymbol, DIA.ElementClass);\r
+ }\r
+ }); \r
+ \r
+ ICanvasSceneGraphProvider provider = DiagramNodeUtil.loadSceneGraphProvider(ctx, modelAndRVI.first, node.getDiagramResource(), modelAndRVI.second, 5000);\r
+ sgProvider.set( provider );\r
+ ctx.getDefaultHintContext().setHint(Hints.KEY_PAGE_DESC, pageDesc);\r
+\r
+// StringBuilder b = new StringBuilder();\r
+// NodeUtil.printTreeNodes(ctx.getCanvasNode(), b);\r
+// System.err.println(b.toString());\r
+\r
+ ThreadUtils.asyncExec(thread, new Runnable() {\r
+ @Override\r
+ public void run() {\r
+ try {\r
+ PDFBuilder chassis = new PDFBuilder(writer, mapper, pageSize, pageDesc, fitDiagramContentsToPageMargins || isSymbol);\r
+ \r
+ chassis.paint(ctx, true);\r
+ } finally {\r
+ done.release();\r
+ }\r
+ }\r
+ });\r
+ } catch (DatabaseException e) {\r
+ done.release();\r
+ exception.set(e);\r
+ } catch (Throwable e) {\r
+ done.release();\r
+ exception.set(new DatabaseException(e));\r
+ } finally {\r
+ done.release();\r
+ }\r
+ }\r
+ });\r
+\r
+ done.acquire(2);\r
+ if (exception.get() != null)\r
+ throw exception.get();\r
+ return result.get();\r
+ } finally {\r
+ if (sgProvider.get() != null)\r
+ sgProvider.get().dispose();\r
+ ctx.dispose();\r
+ }\r
+ }\r
+\r
+ private static Resource resolveModel(ReadGraph graph, Node node) throws DatabaseException {\r
+ Resource composite = node.getDefiningResources().head();\r
+ Resource model = graph.syncRequest(new PossibleIndexRoot(composite));\r
+// Resource model = StructuralVariables.getModel(graph, composite);\r
+ if (model == null)\r
+ throw new ValidationException("no model found for composite " + NameUtils.getSafeName(graph, composite));\r
+ return model;\r
+ }\r
+\r
+// private static String resolveModelURI(ReadGraph graph, final Node node) throws DatabaseException {\r
+// return graph.getURI(resolveModel(graph, node));\r
+// }\r
+\r
+ private static String resolveRVI(ReadGraph graph, final Node node) throws DatabaseException {\r
+ String RVI = node.getRVI();\r
+ if(RVI != null) return RVI;\r
+ Resource composite = node.getDefiningResources().head();\r
+ Variable var = Variables.getVariable(graph, composite);\r
+ org.simantics.db.layer0.variable.RVI rvi = var.getPossibleRVI(graph);\r
+ if(rvi == null) return null;\r
+ return rvi.toString();\r
+// final ResourceArray compositePath = StructuralVariables.getCompositeArray(graph, composite);\r
+// final ResourceArray variablePath = compositePath.removeFromBeginning(1);\r
+// return StructuralVariables.getRVI(graph, variablePath);\r
+ }\r
+\r
+}\r