1 package org.simantics.diagram.svg.export;
\r
4 import java.awt.Point;
\r
5 import java.util.concurrent.Semaphore;
\r
6 import java.util.concurrent.atomic.AtomicReference;
\r
8 import org.apache.batik.svggen.SVGGraphics2D;
\r
9 import org.simantics.Simantics;
\r
10 import org.simantics.db.ReadGraph;
\r
11 import org.simantics.db.Resource;
\r
12 import org.simantics.db.common.ResourceArray;
\r
13 import org.simantics.db.common.request.PossibleIndexRoot;
\r
14 import org.simantics.db.common.request.UniqueRead;
\r
15 import org.simantics.db.common.utils.NameUtils;
\r
16 import org.simantics.db.exception.DatabaseException;
\r
17 import org.simantics.db.exception.ValidationException;
\r
18 import org.simantics.db.management.ISessionContext;
\r
19 import org.simantics.db.request.Read;
\r
20 import org.simantics.diagram.elements.DiagramNodeUtil;
\r
21 import org.simantics.diagram.export.ImagePrinter;
\r
22 import org.simantics.diagram.stubs.DiagramResource;
\r
23 import org.simantics.g2d.canvas.impl.CanvasContext;
\r
24 import org.simantics.g2d.scenegraph.ICanvasSceneGraphProvider;
\r
25 import org.simantics.layer0.Layer0;
\r
26 import org.simantics.modeling.ModelingResources;
\r
27 import org.simantics.structural.stubs.StructuralResource2;
\r
28 import org.simantics.structural2.StructuralVariables;
\r
29 import org.simantics.utils.DataContainer;
\r
30 import org.simantics.utils.datastructures.Pair;
\r
31 import org.simantics.utils.page.MarginUtils.Margins;
\r
32 import org.simantics.utils.threads.ThreadUtils;
\r
33 import org.simantics.utils.threads.WorkerThread;
\r
36 public class DiagramToSVG {
\r
39 public static String diagramToSVG(Resource diagram) {
\r
42 ImagePrinter.ImageExportPlan exportPlan = new ImagePrinter.ImageExportPlan();
\r
43 exportPlan.margin = 0.05;
\r
44 exportPlan.dpi = 96.0;
\r
45 exportPlan.diagram = diagram;
\r
46 return render(diagram,exportPlan);
\r
47 } catch (Exception e) {
\r
48 e.printStackTrace();
\r
49 return "<b>error</b> "+ e.getMessage();
\r
55 public static String diagramToSVG(Resource diagram, int width, int height) {
\r
58 ImagePrinter.ImageExportPlan exportPlan = new ImagePrinter.ImageExportPlan();
\r
59 exportPlan.margin = 0.05;
\r
60 exportPlan.size = new Point(width, height);
\r
61 exportPlan.diagram = diagram;
\r
62 return render(diagram,exportPlan);
\r
63 } catch (Exception e) {
\r
64 e.printStackTrace();
\r
65 return "<b>error</b> "+ e.getMessage();
\r
71 public static String diagramToSVG(Resource diagram, double dpi) {
\r
74 ImagePrinter.ImageExportPlan exportPlan = new ImagePrinter.ImageExportPlan();
\r
75 exportPlan.margin = 0.05;
\r
76 exportPlan.dpi = dpi;
\r
77 exportPlan.diagram = diagram;
\r
78 return render(diagram,exportPlan);
\r
79 } catch (Exception e) {
\r
80 e.printStackTrace();
\r
81 return "<b>error</b> "+ e.getMessage();
\r
87 public static String diagramToSVG(Resource diagram, ImagePrinter.ImageExportPlan exportPlan, Margins margins, SVGGraphics2D svgGenerator) {
\r
90 return render(diagram,exportPlan,margins,svgGenerator);
\r
91 } catch (Exception e) {
\r
92 e.printStackTrace();
\r
93 return "<b>error</b> "+ e.getMessage();
\r
98 private static String render(final Resource input, final ImagePrinter.ImageExportPlan exportPlan) throws Exception {
\r
99 return render(input, exportPlan, null, SVGBuilder.defaultSVGGenerator());
\r
102 private static String render(final Resource input, final ImagePrinter.ImageExportPlan exportPlan, Margins margins, SVGGraphics2D svgExporter) throws Exception {
\r
103 Resource diagram = Simantics.getSession().syncRequest(new Read<Resource>() {
\r
105 public Resource perform(ReadGraph graph) throws DatabaseException {
\r
106 DiagramResource DIA = DiagramResource.getInstance(graph);
\r
107 if (graph.isInstanceOf(input, DIA.Diagram))
\r
109 StructuralResource2 SR = StructuralResource2.getInstance(graph);
\r
110 ModelingResources MOD = ModelingResources.getInstance(graph);
\r
111 Layer0 L0 = Layer0.getInstance(graph);
\r
112 if (graph.isInstanceOf(input, SR.Composite)) {
\r
113 Resource possibleDiagram = graph.getPossibleObject(input, MOD.CompositeToDiagram);
\r
114 if (possibleDiagram != null)
\r
115 return possibleDiagram;
\r
116 for (Resource r : graph.getObjects(input, L0.ConsistsOf)) {
\r
117 if (graph.isInstanceOf(r, SR.Composite)) {
\r
118 possibleDiagram = graph.getPossibleObject(input, MOD.CompositeToDiagram);
\r
119 if (possibleDiagram != null)
\r
120 return possibleDiagram;
\r
127 if (diagram == null)
\r
128 throw new DatabaseException("Input " + input + " cannot be resolved as diagram");
\r
133 final WorkerThread thread = new WorkerThread("Diagram Image Painter");
\r
136 final CanvasContext ctx = new CanvasContext(thread);
\r
137 final AtomicReference<ICanvasSceneGraphProvider> sgProvider = new AtomicReference<ICanvasSceneGraphProvider>();
\r
138 final ISessionContext sessionContext = Simantics.getSessionContext();
\r
139 final DataContainer<String> result = new DataContainer<String>(null);
\r
140 final DataContainer<Exception> exception = new DataContainer<Exception>(null);
\r
142 final Semaphore done = new Semaphore(0);
\r
143 // IMPORTANT: Load diagram in a different thread than the canvas context thread!
\r
144 ThreadUtils.getBlockingWorkExecutor().execute(new Runnable() {
\r
146 public void run() {
\r
148 Pair<Resource, String> modelAndRVI = sessionContext.getSession().syncRequest(new UniqueRead<Pair<Resource, String>>() {
\r
150 public Pair<Resource, String> perform(ReadGraph graph) throws DatabaseException {
\r
151 return new Pair<Resource, String>( resolveModel(graph, exportPlan.diagram ), resolveRVI(graph, exportPlan.diagram) );
\r
155 ICanvasSceneGraphProvider provider = DiagramNodeUtil.loadSceneGraphProvider(ctx, modelAndRVI.first, exportPlan.diagram, modelAndRVI.second);
\r
156 sgProvider.set( provider );
\r
158 ThreadUtils.asyncExec(thread, new Runnable() {
\r
160 public void run() {
\r
162 SVGBuilder chassis = margins != null ?
\r
163 new SVGBuilder(exportPlan.dpi,exportPlan.size,margins) :
\r
164 new SVGBuilder(exportPlan.dpi,exportPlan.size,exportPlan.margin);
\r
166 result.set(chassis.paint(ctx, svgExporter));
\r
167 } catch (Exception e) {
\r
174 } catch (DatabaseException e) {
\r
177 } catch (Throwable e) {
\r
178 exception.set(new DatabaseException(e));
\r
187 if (exception.get() != null)
\r
188 throw exception.get();
\r
189 return result.get();
\r
191 if (sgProvider.get() != null)
\r
192 sgProvider.get().dispose();
\r
197 private static Resource resolveModel(ReadGraph graph, Resource diagram) throws DatabaseException {
\r
198 ModelingResources mod = ModelingResources.getInstance(graph);
\r
199 Resource composite = graph.getSingleObject(diagram, mod.DiagramToComposite);
\r
200 Resource model = graph.syncRequest(new PossibleIndexRoot(composite));
\r
202 throw new ValidationException("no model found for composite " + NameUtils.getSafeName(graph, composite));
\r
208 private static String resolveRVI(ReadGraph graph, Resource diagram) throws DatabaseException {
\r
209 ModelingResources mod = ModelingResources.getInstance(graph);
\r
210 Resource composite = graph.getSingleObject(diagram, mod.DiagramToComposite);
\r
211 final ResourceArray compositePath = StructuralVariables.getCompositeArray(graph, composite);
\r
212 final ResourceArray variablePath = compositePath.removeFromBeginning(1);
\r
213 return StructuralVariables.getRVI(graph, variablePath);
\r