--- /dev/null
+package org.simantics.diagram.svg.export;\r
+\r
+\r
+import java.awt.Point;\r
+import java.util.concurrent.Semaphore;\r
+import java.util.concurrent.atomic.AtomicReference;\r
+\r
+import org.apache.batik.svggen.SVGGraphics2D;\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.db.request.Read;\r
+import org.simantics.diagram.elements.DiagramNodeUtil;\r
+import org.simantics.diagram.export.ImagePrinter;\r
+import org.simantics.diagram.stubs.DiagramResource;\r
+import org.simantics.g2d.canvas.impl.CanvasContext;\r
+import org.simantics.g2d.scenegraph.ICanvasSceneGraphProvider;\r
+import org.simantics.layer0.Layer0;\r
+import org.simantics.modeling.ModelingResources;\r
+import org.simantics.structural.stubs.StructuralResource2;\r
+import org.simantics.structural2.StructuralVariables;\r
+import org.simantics.utils.DataContainer;\r
+import org.simantics.utils.datastructures.Pair;\r
+import org.simantics.utils.page.MarginUtils.Margins;\r
+import org.simantics.utils.threads.ThreadUtils;\r
+import org.simantics.utils.threads.WorkerThread;\r
+\r
+\r
+public class DiagramToSVG {\r
+ \r
+ \r
+ public static String diagramToSVG(Resource diagram) {\r
+\r
+ try {\r
+ ImagePrinter.ImageExportPlan exportPlan = new ImagePrinter.ImageExportPlan();\r
+ exportPlan.margin = 0.05;\r
+ exportPlan.dpi = 96.0;\r
+ exportPlan.diagram = diagram;\r
+ return render(diagram,exportPlan);\r
+ } catch (Exception e) {\r
+ e.printStackTrace();\r
+ return "<b>error</b> "+ e.getMessage();\r
+ }\r
+ \r
+ \r
+ }\r
+ \r
+ public static String diagramToSVG(Resource diagram, int width, int height) {\r
+\r
+ try {\r
+ ImagePrinter.ImageExportPlan exportPlan = new ImagePrinter.ImageExportPlan();\r
+ exportPlan.margin = 0.05;\r
+ exportPlan.size = new Point(width, height);\r
+ exportPlan.diagram = diagram;\r
+ return render(diagram,exportPlan);\r
+ } catch (Exception e) {\r
+ e.printStackTrace();\r
+ return "<b>error</b> "+ e.getMessage();\r
+ }\r
+ \r
+ \r
+ }\r
+ \r
+ public static String diagramToSVG(Resource diagram, double dpi) {\r
+\r
+ try {\r
+ ImagePrinter.ImageExportPlan exportPlan = new ImagePrinter.ImageExportPlan();\r
+ exportPlan.margin = 0.05;\r
+ exportPlan.dpi = dpi;\r
+ exportPlan.diagram = diagram;\r
+ return render(diagram,exportPlan);\r
+ } catch (Exception e) {\r
+ e.printStackTrace();\r
+ return "<b>error</b> "+ e.getMessage();\r
+ }\r
+ \r
+ \r
+ }\r
+ \r
+ public static String diagramToSVG(Resource diagram, ImagePrinter.ImageExportPlan exportPlan, Margins margins, SVGGraphics2D svgGenerator) {\r
+\r
+ try {\r
+ return render(diagram,exportPlan,margins,svgGenerator);\r
+ } catch (Exception e) {\r
+ e.printStackTrace();\r
+ return "<b>error</b> "+ e.getMessage();\r
+ }\r
+ \r
+ }\r
+ \r
+ private static String render(final Resource input, final ImagePrinter.ImageExportPlan exportPlan) throws Exception {\r
+ return render(input, exportPlan, null, SVGBuilder.defaultSVGGenerator()); \r
+ }\r
+ \r
+ private static String render(final Resource input, final ImagePrinter.ImageExportPlan exportPlan, Margins margins, SVGGraphics2D svgExporter) throws Exception {\r
+ Resource diagram = Simantics.getSession().syncRequest(new Read<Resource>() {\r
+ @Override\r
+ public Resource perform(ReadGraph graph) throws DatabaseException {\r
+ DiagramResource DIA = DiagramResource.getInstance(graph);\r
+ if (graph.isInstanceOf(input, DIA.Diagram))\r
+ return input;\r
+ StructuralResource2 SR = StructuralResource2.getInstance(graph);\r
+ ModelingResources MOD = ModelingResources.getInstance(graph);\r
+ Layer0 L0 = Layer0.getInstance(graph);\r
+ if (graph.isInstanceOf(input, SR.Composite)) {\r
+ Resource possibleDiagram = graph.getPossibleObject(input, MOD.CompositeToDiagram);\r
+ if (possibleDiagram != null)\r
+ return possibleDiagram;\r
+ for (Resource r : graph.getObjects(input, L0.ConsistsOf)) {\r
+ if (graph.isInstanceOf(r, SR.Composite)) {\r
+ possibleDiagram = graph.getPossibleObject(input, MOD.CompositeToDiagram);\r
+ if (possibleDiagram != null)\r
+ return possibleDiagram;\r
+ }\r
+ }\r
+ }\r
+ return null;\r
+ }\r
+ });\r
+ if (diagram == null)\r
+ throw new DatabaseException("Input " + input + " cannot be resolved as diagram");\r
+ \r
+ \r
+ \r
+ \r
+ final WorkerThread thread = new WorkerThread("Diagram Image Painter");\r
+ thread.start();\r
+ \r
+ final CanvasContext ctx = new CanvasContext(thread);\r
+ final AtomicReference<ICanvasSceneGraphProvider> sgProvider = new AtomicReference<ICanvasSceneGraphProvider>();\r
+ final ISessionContext sessionContext = Simantics.getSessionContext();\r
+ final DataContainer<String> result = new DataContainer<String>(null);\r
+ final DataContainer<Exception> exception = new DataContainer<Exception>(null);\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
+ SVGBuilder chassis = margins != null ?\r
+ new SVGBuilder(exportPlan.dpi,exportPlan.size,margins) :\r
+ new SVGBuilder(exportPlan.dpi,exportPlan.size,exportPlan.margin);\r
+ \r
+ result.set(chassis.paint(ctx, svgExporter));\r
+ } catch (Exception e) {\r
+ exception.set(e);\r
+ } finally {\r
+ done.release();\r
+ }\r
+ }\r
+ });\r
+ } catch (DatabaseException e) {\r
+ exception.set(e);\r
+ done.release();\r
+ } catch (Throwable e) {\r
+ exception.set(new DatabaseException(e));\r
+ done.release();\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, 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