-package org.simantics.diagram.export;\r
-\r
-import java.awt.Point;\r
-import java.awt.geom.AffineTransform;\r
-import java.awt.image.BufferedImage;\r
-import java.io.File;\r
-import java.util.concurrent.Semaphore;\r
-import java.util.concurrent.atomic.AtomicReference;\r
-\r
-import org.eclipse.core.runtime.IProgressMonitor;\r
-import org.eclipse.core.runtime.SubMonitor;\r
-import org.simantics.Simantics;\r
-import org.simantics.db.ReadGraph;\r
-import org.simantics.db.Resource;\r
-import org.simantics.db.common.ResourceArray;\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.management.ISessionContext;\r
-import org.simantics.diagram.elements.DiagramNodeUtil;\r
-import org.simantics.g2d.canvas.ICanvasContext;\r
-import org.simantics.g2d.canvas.impl.CanvasContext;\r
-import org.simantics.g2d.chassis.ICanvasChassis;\r
-import org.simantics.g2d.chassis.SWTChassis;\r
-import org.simantics.g2d.participant.TransformUtil;\r
-import org.simantics.g2d.scenegraph.ICanvasSceneGraphProvider;\r
-import org.simantics.g2d.scenegraph.SceneGraphConstants;\r
-import org.simantics.modeling.ModelingResources;\r
-import org.simantics.scenegraph.g2d.nodes.NavigationNode;\r
-import org.simantics.scenegraph.utils.NodeUtil;\r
-import org.simantics.structural2.StructuralVariables;\r
-import org.simantics.utils.DataContainer;\r
-import org.simantics.utils.datastructures.Pair;\r
-import org.simantics.utils.threads.AWTThread;\r
-import org.simantics.utils.threads.IThreadWorkQueue;\r
-import org.simantics.utils.threads.ThreadUtils;\r
-import org.simantics.utils.threads.WorkerThread;\r
-\r
-public class ImagePrinter {\r
-\r
- public static class ImageExportPlan {\r
- public File exportLocation;\r
- public String name;\r
- public Resource diagram;\r
-\r
- // use dpi or size, not both.\r
- public Double dpi;\r
- public Point size;\r
- public double margin;\r
- }\r
-\r
- /**\r
- * @param monitor\r
- * the progress monitor to use for reporting progress to the\r
- * user. It is the caller's responsibility to call done() on the\r
- * given monitor. Accepts <code>null</code>, indicating that no\r
- * progress should be reported and that the operation cannot be\r
- * cancelled.\r
- * \r
- * @param exportPath\r
- * @throws Exception\r
- */\r
- public static BufferedImage printToImage(IProgressMonitor monitor, ImageExportPlan exportPlan) throws Exception {\r
- \r
-\r
- SubMonitor progress = SubMonitor.convert(monitor, "Export to Image", 1);\r
-\r
- WorkerThread workerThread = new WorkerThread("Diagram Image Painter");\r
- workerThread.start();\r
- String loc;\r
- if (exportPlan.exportLocation == null)\r
- loc = "clipboard";\r
- else\r
- loc = exportPlan.exportLocation.getAbsolutePath();\r
- progress.beginTask("Writing " + exportPlan.name + " to " + loc, 1);\r
- \r
- return render(workerThread, Simantics.getSessionContext(), exportPlan);\r
- \r
-\r
- }\r
-\r
- /**\r
- * Renders diagram to BufferedImage\r
- * \r
- * @param thread\r
- * @param sessionContext\r
- * @param exportPlan\r
- * @return\r
- * @throws Exception\r
- */\r
- public static BufferedImage render(\r
- final IThreadWorkQueue thread,\r
- final ISessionContext sessionContext,\r
- final ImageExportPlan exportPlan)\r
- throws Exception\r
- {\r
- final DataContainer<BufferedImage> result = new DataContainer<BufferedImage>(null);\r
- final DataContainer<Exception> exception = new DataContainer<Exception>();\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, exportPlan.diagram ), resolveRVI(graph, exportPlan.diagram) );\r
- }\r
- });\r
-\r
- ICanvasSceneGraphProvider provider = DiagramNodeUtil.loadSceneGraphProvider(ctx, modelAndRVI.first, exportPlan.diagram, modelAndRVI.second);\r
- sgProvider.set( provider );\r
- \r
- ThreadUtils.asyncExec(thread, new Runnable() {\r
- @Override\r
- public void run() {\r
- try {\r
- ImageBuilder chassis = new ImageBuilder(exportPlan.exportLocation,exportPlan.dpi,exportPlan.size,exportPlan.margin);\r
- \r
- result.set(chassis.paint(ctx));\r
- } catch (Exception e) {\r
- exception.set(e);\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
- /**\r
- * Renders diagram to BufferedImage using existing CanvasContext.\r
- * \r
- * Note: While the method tries to restore setting as they were, there may be visible side effects in the diagram. \r
- * \r
- * @param context\r
- * @param chassis\r
- * @param exportPlan\r
- * @return\r
- * @throws Exception\r
- */\r
- public static BufferedImage renderLocal(final ICanvasContext context, final ICanvasChassis chassis, final ImageExportPlan exportPlan) throws Exception {\r
- final Semaphore done = new Semaphore(0);\r
- final DataContainer<BufferedImage> result = new DataContainer<BufferedImage>(null);\r
- final DataContainer<Exception> exception = new DataContainer<Exception>(null);\r
- ThreadUtils.asyncExec(AWTThread.getThreadAccess(), new Runnable() {\r
-\r
- @Override\r
- public void run() {\r
- try {\r
- result.set(renderLocal(context, exportPlan, chassis));\r
- } catch (Exception e) {\r
- exception.set(e);\r
- } finally {\r
- done.release();\r
- }\r
- }\r
- });\r
- done.acquire();\r
- if (exception.get() != null)\r
- throw exception.get();\r
- return result.get();\r
- }\r
- \r
- private static BufferedImage renderLocal(ICanvasContext context, ImagePrinter.ImageExportPlan exportPlan, ICanvasChassis chassis) throws Exception {\r
- TransformUtil util = context.getSingleItem(TransformUtil.class);\r
- AffineTransform at = util.getTransform();\r
- final NavigationNode node = NodeUtil.findNodeById(context.getSceneGraph(), SceneGraphConstants.NAVIGATION_NODE_PATH);\r
- boolean adapt = true;\r
- if (node != null) {\r
- adapt = node.getAdaptViewportToResizedControl();\r
- // prevent NavigationNode from re-scaling when it detects canvas size change. \r
- node.setAdaptViewportToResizedControl(false);\r
- }\r
-\r
- ImageBuilder builder = new ImageBuilder(exportPlan.exportLocation,exportPlan.dpi,exportPlan.size,exportPlan.margin);\r
- BufferedImage image = builder.paint(context);\r
- \r
- util.setTransform(at);\r
-\r
- \r
- if (node != null && adapt != false) {\r
- if (chassis instanceof SWTChassis) {\r
- ((SWTChassis) chassis).getAWTComponent().repaint();\r
- }\r
- // restore re-scaling setting (need to be scheduled, so that Diagram canvas is re-painted once.\r
- final boolean b = adapt;\r
- ThreadUtils.asyncExec(AWTThread.getThreadAccess(), new Runnable() {\r
- @Override\r
- public void run() {\r
- node.setAdaptViewportToResizedControl(b);\r
- }\r
- });\r
- }\r
- \r
- return image;\r
- }\r
- \r
- \r
-\r
- private static Resource resolveModel(ReadGraph graph, Resource diagram) throws DatabaseException {\r
- ModelingResources mod = ModelingResources.getInstance(graph);\r
- Resource composite = graph.getSingleObject(diagram, mod.DiagramToComposite);\r
- Resource model = graph.syncRequest(new PossibleIndexRoot(composite));\r
- if (model == null)\r
- throw new ValidationException("no model found for composite " + NameUtils.getSafeName(graph, composite));\r
- return model;\r
- }\r
-\r
-\r
-\r
- private static String resolveRVI(ReadGraph graph, Resource diagram) throws DatabaseException {\r
- ModelingResources mod = ModelingResources.getInstance(graph);\r
- Resource composite = graph.getSingleObject(diagram, mod.DiagramToComposite);\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
-}\r
+package org.simantics.diagram.export;
+
+import java.awt.Point;
+import java.awt.geom.AffineTransform;
+import java.awt.image.BufferedImage;
+import java.io.File;
+import java.util.concurrent.Semaphore;
+import java.util.concurrent.atomic.AtomicReference;
+
+import org.eclipse.core.runtime.IProgressMonitor;
+import org.eclipse.core.runtime.SubMonitor;
+import org.simantics.Simantics;
+import org.simantics.db.ReadGraph;
+import org.simantics.db.Resource;
+import org.simantics.db.common.ResourceArray;
+import org.simantics.db.common.request.PossibleIndexRoot;
+import org.simantics.db.common.request.UniqueRead;
+import org.simantics.db.common.utils.NameUtils;
+import org.simantics.db.exception.DatabaseException;
+import org.simantics.db.exception.ValidationException;
+import org.simantics.db.management.ISessionContext;
+import org.simantics.diagram.elements.DiagramNodeUtil;
+import org.simantics.g2d.canvas.ICanvasContext;
+import org.simantics.g2d.canvas.impl.CanvasContext;
+import org.simantics.g2d.chassis.ICanvasChassis;
+import org.simantics.g2d.chassis.SWTChassis;
+import org.simantics.g2d.participant.TransformUtil;
+import org.simantics.g2d.scenegraph.ICanvasSceneGraphProvider;
+import org.simantics.g2d.scenegraph.SceneGraphConstants;
+import org.simantics.modeling.ModelingResources;
+import org.simantics.scenegraph.g2d.nodes.NavigationNode;
+import org.simantics.scenegraph.utils.NodeUtil;
+import org.simantics.structural2.StructuralVariables;
+import org.simantics.utils.DataContainer;
+import org.simantics.utils.datastructures.Pair;
+import org.simantics.utils.threads.AWTThread;
+import org.simantics.utils.threads.IThreadWorkQueue;
+import org.simantics.utils.threads.ThreadUtils;
+import org.simantics.utils.threads.WorkerThread;
+
+public class ImagePrinter {
+
+ public static class ImageExportPlan {
+ public File exportLocation;
+ public String name;
+ public Resource diagram;
+
+ // use dpi or size, not both.
+ public Double dpi;
+ public Point size;
+ public double margin;
+ }
+
+ /**
+ * @param monitor
+ * the progress monitor to use for reporting progress to the
+ * user. It is the caller's responsibility to call done() on the
+ * given monitor. Accepts <code>null</code>, indicating that no
+ * progress should be reported and that the operation cannot be
+ * cancelled.
+ *
+ * @param exportPath
+ * @throws Exception
+ */
+ public static BufferedImage printToImage(IProgressMonitor monitor, ImageExportPlan exportPlan) throws Exception {
+
+
+ SubMonitor progress = SubMonitor.convert(monitor, "Export to Image", 1);
+
+ WorkerThread workerThread = new WorkerThread("Diagram Image Painter");
+ workerThread.start();
+ String loc;
+ if (exportPlan.exportLocation == null)
+ loc = "clipboard";
+ else
+ loc = exportPlan.exportLocation.getAbsolutePath();
+ progress.beginTask("Writing " + exportPlan.name + " to " + loc, 1);
+
+ return render(workerThread, Simantics.getSessionContext(), exportPlan);
+
+
+ }
+
+ /**
+ * Renders diagram to BufferedImage
+ *
+ * @param thread
+ * @param sessionContext
+ * @param exportPlan
+ * @return
+ * @throws Exception
+ */
+ public static BufferedImage render(
+ final IThreadWorkQueue thread,
+ final ISessionContext sessionContext,
+ final ImageExportPlan exportPlan)
+ throws Exception
+ {
+ final DataContainer<BufferedImage> result = new DataContainer<BufferedImage>(null);
+ final DataContainer<Exception> exception = new DataContainer<Exception>();
+
+ final CanvasContext ctx = new CanvasContext(thread);
+ final AtomicReference<ICanvasSceneGraphProvider> sgProvider = new AtomicReference<ICanvasSceneGraphProvider>();
+
+ try {
+ final Semaphore done = new Semaphore(0);
+ // IMPORTANT: Load diagram in a different thread than the canvas context thread!
+ ThreadUtils.getBlockingWorkExecutor().execute(new Runnable() {
+ @Override
+ public void run() {
+ try {
+ Pair<Resource, String> modelAndRVI = sessionContext.getSession().syncRequest(new UniqueRead<Pair<Resource, String>>() {
+ @Override
+ public Pair<Resource, String> perform(ReadGraph graph) throws DatabaseException {
+ return new Pair<Resource, String>( resolveModel(graph, exportPlan.diagram ), resolveRVI(graph, exportPlan.diagram) );
+ }
+ });
+
+ ICanvasSceneGraphProvider provider = DiagramNodeUtil.loadSceneGraphProvider(ctx, modelAndRVI.first, exportPlan.diagram, modelAndRVI.second);
+ sgProvider.set( provider );
+
+ ThreadUtils.asyncExec(thread, new Runnable() {
+ @Override
+ public void run() {
+ try {
+ ImageBuilder chassis = new ImageBuilder(exportPlan.exportLocation,exportPlan.dpi,exportPlan.size,exportPlan.margin);
+
+ result.set(chassis.paint(ctx));
+ } catch (Exception e) {
+ exception.set(e);
+ } finally {
+ done.release();
+ }
+ }
+ });
+ } catch (DatabaseException e) {
+ done.release();
+ exception.set(e);
+ } catch (Throwable e) {
+ done.release();
+ exception.set(new DatabaseException(e));
+ } finally {
+ done.release();
+ }
+ }
+ });
+
+ done.acquire(2);
+ if (exception.get() != null)
+ throw exception.get();
+ return result.get();
+ } finally {
+ if (sgProvider.get() != null)
+ sgProvider.get().dispose();
+ ctx.dispose();
+ }
+ }
+
+ /**
+ * Renders diagram to BufferedImage using existing CanvasContext.
+ *
+ * Note: While the method tries to restore setting as they were, there may be visible side effects in the diagram.
+ *
+ * @param context
+ * @param chassis
+ * @param exportPlan
+ * @return
+ * @throws Exception
+ */
+ public static BufferedImage renderLocal(final ICanvasContext context, final ICanvasChassis chassis, final ImageExportPlan exportPlan) throws Exception {
+ final Semaphore done = new Semaphore(0);
+ final DataContainer<BufferedImage> result = new DataContainer<BufferedImage>(null);
+ final DataContainer<Exception> exception = new DataContainer<Exception>(null);
+ ThreadUtils.asyncExec(AWTThread.getThreadAccess(), new Runnable() {
+
+ @Override
+ public void run() {
+ try {
+ result.set(renderLocal(context, exportPlan, chassis));
+ } catch (Exception e) {
+ exception.set(e);
+ } finally {
+ done.release();
+ }
+ }
+ });
+ done.acquire();
+ if (exception.get() != null)
+ throw exception.get();
+ return result.get();
+ }
+
+ private static BufferedImage renderLocal(ICanvasContext context, ImagePrinter.ImageExportPlan exportPlan, ICanvasChassis chassis) throws Exception {
+ TransformUtil util = context.getSingleItem(TransformUtil.class);
+ AffineTransform at = util.getTransform();
+ final NavigationNode node = NodeUtil.findNodeById(context.getSceneGraph(), SceneGraphConstants.NAVIGATION_NODE_PATH);
+ boolean adapt = true;
+ if (node != null) {
+ adapt = node.getAdaptViewportToResizedControl();
+ // prevent NavigationNode from re-scaling when it detects canvas size change.
+ node.setAdaptViewportToResizedControl(false);
+ }
+
+ ImageBuilder builder = new ImageBuilder(exportPlan.exportLocation,exportPlan.dpi,exportPlan.size,exportPlan.margin);
+ BufferedImage image = builder.paint(context);
+
+ util.setTransform(at);
+
+
+ if (node != null && adapt != false) {
+ if (chassis instanceof SWTChassis) {
+ ((SWTChassis) chassis).getAWTComponent().repaint();
+ }
+ // restore re-scaling setting (need to be scheduled, so that Diagram canvas is re-painted once.
+ final boolean b = adapt;
+ ThreadUtils.asyncExec(AWTThread.getThreadAccess(), new Runnable() {
+ @Override
+ public void run() {
+ node.setAdaptViewportToResizedControl(b);
+ }
+ });
+ }
+
+ return image;
+ }
+
+
+
+ private static Resource resolveModel(ReadGraph graph, Resource diagram) throws DatabaseException {
+ ModelingResources mod = ModelingResources.getInstance(graph);
+ Resource composite = graph.getSingleObject(diagram, mod.DiagramToComposite);
+ Resource model = graph.syncRequest(new PossibleIndexRoot(composite));
+ if (model == null)
+ throw new ValidationException("no model found for composite " + NameUtils.getSafeName(graph, composite));
+ return model;
+ }
+
+
+
+ private static String resolveRVI(ReadGraph graph, Resource diagram) throws DatabaseException {
+ ModelingResources mod = ModelingResources.getInstance(graph);
+ Resource composite = graph.getSingleObject(diagram, mod.DiagramToComposite);
+ final ResourceArray compositePath = StructuralVariables.getCompositeArray(graph, composite);
+ final ResourceArray variablePath = compositePath.removeFromBeginning(1);
+ return StructuralVariables.getRVI(graph, variablePath);
+ }
+
+
+}