1 package org.simantics.modeling;
3 import java.awt.Dimension;
4 import java.awt.RenderingHints;
5 import java.awt.RenderingHints.Key;
6 import java.awt.geom.AffineTransform;
7 import java.awt.geom.Rectangle2D;
8 import java.io.ByteArrayOutputStream;
9 import java.io.OutputStreamWriter;
10 import java.util.ArrayList;
11 import java.util.Collection;
12 import java.util.HashMap;
13 import java.util.List;
15 import java.util.UUID;
17 import javax.xml.transform.OutputKeys;
18 import javax.xml.transform.Transformer;
19 import javax.xml.transform.TransformerFactory;
20 import javax.xml.transform.dom.DOMSource;
21 import javax.xml.transform.stream.StreamResult;
23 import org.apache.batik.dom.GenericDOMImplementation;
24 import org.apache.batik.svggen.SVGGeneratorContext;
25 import org.apache.batik.svggen.SVGGraphics2D;
26 import org.simantics.db.Resource;
27 import org.simantics.db.exception.DatabaseException;
28 import org.simantics.diagram.elements.DiagramNodeUtil;
29 import org.simantics.diagram.elements.TextGridNode;
30 import org.simantics.diagram.elements.TextNode;
31 import org.simantics.g2d.canvas.ICanvasContext;
32 import org.simantics.g2d.diagram.DiagramHints;
33 import org.simantics.g2d.diagram.IDiagram;
34 import org.simantics.g2d.diagram.handler.DataElementMap;
35 import org.simantics.g2d.diagram.participant.Selection;
36 import org.simantics.g2d.element.IElement;
37 import org.simantics.g2d.scenegraph.ICanvasSceneGraphProvider;
38 import org.simantics.g2d.utils.CanvasUtils;
39 import org.simantics.scenegraph.ParentNode;
40 import org.simantics.scenegraph.g2d.G2DParentNode;
41 import org.simantics.scenegraph.g2d.G2DRenderingHints;
42 import org.simantics.scenegraph.g2d.G2DSceneGraph;
43 import org.simantics.scenegraph.g2d.IG2DNode;
44 import org.simantics.scenegraph.g2d.IG2DNodeVisitor;
45 import org.simantics.scenegraph.g2d.events.command.Commands;
46 import org.simantics.scenegraph.g2d.nodes.BackgroundNode;
47 import org.simantics.scenegraph.g2d.nodes.BoundsNode;
48 import org.simantics.scenegraph.g2d.nodes.ConnectionNode;
49 import org.simantics.scenegraph.g2d.nodes.DataNode;
50 import org.simantics.scenegraph.g2d.nodes.DecorationSVGNode;
51 import org.simantics.scenegraph.g2d.nodes.NavigationNode;
52 import org.simantics.scenegraph.g2d.nodes.SVGNode;
53 import org.simantics.scenegraph.g2d.nodes.SingleElementNode;
54 import org.simantics.scenegraph.g2d.nodes.spatial.RTreeNode;
55 import org.simantics.scenegraph.utils.NodeUtil;
56 import org.simantics.trend.impl.ItemNode;
57 import org.simantics.utils.threads.ThreadUtils;
58 import org.slf4j.Logger;
59 import org.slf4j.LoggerFactory;
60 import org.w3c.dom.DOMImplementation;
61 import org.w3c.dom.Document;
62 import org.w3c.dom.Element;
63 import org.w3c.dom.NodeList;
65 public class SCLScenegraph {
67 private static final Logger LOGGER = LoggerFactory.getLogger(SCLScenegraph.class);
69 public static ICanvasSceneGraphProvider getICanvasSceneGraphProvider(Resource model, Resource diagram, String diagramRVI) throws DatabaseException, InterruptedException {
70 ICanvasSceneGraphProvider provider = DiagramNodeUtil.loadSceneGraphProvider(model, diagram, diagramRVI);
74 public static void disposeSceneGraphProvider(ICanvasSceneGraphProvider provider) {
78 public static String getNodeTransform(ICanvasContext ctx, String name) {
80 Set<TextNode> texts = NodeUtil.collectNodes(ctx.getSceneGraph(), TextNode.class);
81 for (TextNode text : texts) {
82 String nodeName = NodeUtil.getNodeName(text);
83 if (nodeName.equals(name)) {
84 String transform = text.getTransform().toString();
88 return "No node found";
91 public static String getNodeText(ICanvasContext ctx, String name) {
93 Set<TextNode> texts = NodeUtil.collectNodes(ctx.getSceneGraph(), TextNode.class);
94 for (TextNode text : texts) {
95 String nodeName = NodeUtil.getNodeName(text);
96 if (nodeName.equals(name)) {
97 String texti = text.getText();
101 return "No node found";
104 public static String getNodeCount(ICanvasContext ctx) {
105 G2DSceneGraph g2 = ctx.getSceneGraph();
106 int amount = NodeUtil.countTreeNodes(g2);
107 return "Node count: " + amount;
110 public static String getAllNodes (ICanvasContext ctx) {
112 Set<G2DSceneGraph> g2 = NodeUtil.collectNodes(ctx.getSceneGraph(), G2DSceneGraph.class);
113 int amount = g2.size() +1;
114 return "All nodes: " + amount;
117 public static String getBoundsNodes (ICanvasContext ctx) {
119 Set<BoundsNode> bn = NodeUtil.collectNodes(ctx.getSceneGraph(), BoundsNode.class);
120 int amount = bn.size();
121 return "BoundsNodes: " + amount;
124 public static String getBackgroundNodes (ICanvasContext ctx) {
126 Set<BackgroundNode> bg = NodeUtil.collectNodes(ctx.getSceneGraph(), BackgroundNode.class);
127 int amount = bg.size();
128 return "BackgroundNodes: " + amount;
131 public static String getDataNodes (ICanvasContext ctx) {
133 Set<DataNode> dn = NodeUtil.collectNodes(ctx.getSceneGraph(), DataNode.class);
134 int amount = dn.size();
135 return "DataNodes: " + amount;
138 public static String getNavigationNodes (ICanvasContext ctx) {
140 Set<NavigationNode> g2 = NodeUtil.collectNodes(ctx.getSceneGraph(), NavigationNode.class);
141 int amount = g2.size();
142 return "NavigationNodes: " + amount;
145 public static String getParentNodes (ICanvasContext ctx) {
147 Set<G2DParentNode> g2 = NodeUtil.collectNodes(ctx.getSceneGraph(), G2DParentNode.class);
148 int amount = g2.size();
149 return "ParentNodes: " + amount;
152 public static String getDecorationNodes (ICanvasContext ctx) {
154 Set<DecorationSVGNode> deco = NodeUtil.collectNodes(ctx.getSceneGraph(), DecorationSVGNode.class);
155 int amount = deco.size();
156 return "DecorationNodes: " + amount;
159 public static String getSingleElementNodes (ICanvasContext ctx) {
161 Set<SingleElementNode> g2 = NodeUtil.collectNodes(ctx.getSceneGraph(), SingleElementNode.class);
162 int amount = g2.size();
163 return "SingleElementNodes: " + amount;
166 public static String getConnectionNodes (ICanvasContext ctx) {
168 Set<ConnectionNode> g2 = NodeUtil.collectNodes(ctx.getSceneGraph(), ConnectionNode.class);
169 int amount = g2.size();
170 return "ConnectionNodes: " + amount;
173 public static String getTextNodes (ICanvasContext ctx) {
175 Set<TextNode> tn = NodeUtil.collectNodes(ctx.getSceneGraph(), TextNode.class);
176 Set<TextGridNode> tgn = NodeUtil.collectNodes(ctx.getSceneGraph(), TextGridNode.class);
177 int amount = tn.size() + tgn.size();
178 return "TextNodes: " + amount;
181 public static String getItemNodes (ICanvasContext ctx) {
183 Set<ItemNode> item = NodeUtil.collectNodes(ctx.getSceneGraph(), ItemNode.class);
184 int amount = item.size();
185 return "ItemNodes: " + amount;
188 public static String editNodeText (ICanvasContext ctx, String module, String previous_value, String new_value) {
190 Set<TextNode> textGridNodes = NodeUtil.collectNodes(ctx.getSceneGraph(), TextNode.class);
191 for (TextNode modulenode : textGridNodes) {
192 if (module.equals(modulenode.getText())) {
193 //System.out.println("Module what we were looking for: " + module);
194 //System.out.println("Modulenode: " + modulenode.getText());
196 ParentNode<?> parentnode = modulenode.getParent();
197 //System.out.println("Parentnode: " + parentnode);
199 Collection<TextNode> textnodes = (Collection<TextNode>) parentnode.getNodes();
200 for (TextNode valuenode : textnodes) {
201 if (previous_value.equals(valuenode.getText())) {
202 //System.out.println("Value what we were looking for: " + previous_value);
203 //System.out.println("Valuenode: " + valuenode.getText());
205 //valuenode.setEditMode(true);
206 valuenode.activateEdit(0, null, ctx);
207 valuenode.setText(new_value);
208 valuenode.fireTextEditingEnded();
210 //System.out.println("valuenode modified: " + valuenode);
211 return "Modified module " + module + " with value " + new_value;
214 return "Not found module : " + module;
217 return "No nodes in scenegraph!";
220 public static String sceneGraphTest (ICanvasContext ctx, String module, String value) {
222 boolean module_founded = false;
223 boolean value_founded = false;
225 Set<G2DSceneGraph> g2 = NodeUtil.collectNodes(ctx.getSceneGraph(), G2DSceneGraph.class);
226 System.out.println("Total amount of nodes: " + g2.size() + 1);
228 Set<TextGridNode> grid = NodeUtil.collectNodes(ctx.getSceneGraph(), TextGridNode.class);
229 Integer textGridNodeAmount = grid.size();
230 System.out.println("Amount of TextGridNodes " + textGridNodeAmount);
232 Set<TextNode> texts = NodeUtil.collectNodes(ctx.getSceneGraph(), TextNode.class);
233 Integer textNodeAmount = grid.size();
234 System.out.println("Amount of TextNodes " + textNodeAmount);
236 for (TextNode node : texts) {
237 if (module.equals(node.getText())) {
238 module_founded = true;
239 System.out.println("Correct module " + module + " founded.");
241 if (value.equals(node.getText())) {
242 value_founded = true;
243 System.out.println("Correct value " + value + " founded.");
247 if (value_founded == true && module_founded == true) {
248 return "Found both correct module " + module + " and value " + value;
250 if (value_founded == false && module_founded == true) {
251 return "Found only correct module " + module + " but not value " + value;
253 if (value_founded == true && module_founded == false) {
254 return "Found only correct value " + value + " but not module " + module;
257 return "Didn't found either module " + module + " or value " + value;
261 public static boolean copyPaste (final ICanvasContext source_ctx, final ICanvasContext target_ctx, List<Resource> modules) throws DatabaseException {
263 IDiagram idiagram = source_ctx.getDefaultHintContext().getHint(DiagramHints.KEY_DIAGRAM);
265 DataElementMap dem = idiagram.getDiagramClass().getAtMostOneItemOfClass(DataElementMap.class);
267 final Collection<IElement> newSelection = new ArrayList<IElement>();
268 for (Resource module : modules) {
269 IElement element = dem.getElement(idiagram, module);
270 if (element != null) {
271 newSelection.add(element);
273 throw new DatabaseException("Could not find IElement for " + element);
277 ThreadUtils.syncExec(source_ctx.getThreadAccess(), new Runnable() {
280 if (source_ctx.isDisposed())
282 Selection selection = source_ctx.getAtMostOneItemOfClass(Selection.class);
283 if (selection != null) {
284 // This prevents workbench selection from being left over.
285 // Also prevents scene graph crap from being left on the screen.
286 selection.setSelection(0, newSelection);
288 CanvasUtils.sendCommand(source_ctx, Commands.COPY);
289 CanvasUtils.sendCommand(target_ctx, Commands.PASTE);
295 while(source_ctx.getEventQueue().size() > 0) {
298 } catch (InterruptedException e) {
299 throw new DatabaseException(e);
303 ThreadUtils.syncExec(source_ctx.getThreadAccess(), new Runnable() {
313 static class Generator extends SVGGraphics2D {
316 String newElementId = null;
317 ArrayList<Element> elements = new ArrayList<Element>();
319 public static final String svgNS = "http://www.w3.org/2000/svg";
321 public Generator(SVGGeneratorContext ctx, boolean joku) {
325 public Generator(Document document) {
330 public Element getRoot() {
331 Element root = super.getRoot();
332 for(Element e : elements) {
339 public void setRenderingHint(Key arg0, Object arg1) {
340 if(G2DRenderingHints.KEY_BEGIN_ELEMENT == arg0) {
343 if(G2DRenderingHints.KEY_ELEMENT_ID == arg0) {
345 newElementId = arg1.toString();
347 newElementId = UUID.randomUUID().toString();
349 if(G2DRenderingHints.KEY_END_ELEMENT == arg0) {
352 Element group = getDOMFactory().createElement(SVG_G_TAG);
353 //Element group = getDOMFactory().createElementNS(SVG_NAMESPACE_URI, SVG_G_TAG);
354 group.setAttributeNS(null, "id", newElementId);
355 group.setAttributeNS(null, "class", arg1.toString());
360 super.setRenderingHint(arg0, arg1);
365 public static Element renderSVGNode(IG2DNode node) {
367 // Get a DOMImplementation.
368 DOMImplementation domImpl = GenericDOMImplementation.getDOMImplementation();
370 // Create an instance of org.w3c.dom.Document.
371 String svgNS = "http://www.w3.org/2000/svg";
372 Document document = domImpl.createDocument(svgNS, "svg", null);
374 SVGGeneratorContext ctx = SVGGeneratorContext.createDefault(document);
375 ctx.setComment(null);
377 // Create an instance of the SVG Generator.
378 SVGGraphics2D svgGenerator = new Generator(ctx, false);
382 svgGenerator.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
383 svgGenerator.setRenderingHint(RenderingHints.KEY_RENDERING, RenderingHints.VALUE_RENDER_QUALITY);
384 svgGenerator.setRenderingHint(RenderingHints.KEY_TEXT_ANTIALIASING, RenderingHints.VALUE_TEXT_ANTIALIAS_ON);
386 node.render(svgGenerator);
388 } catch (Throwable t) {
389 LOGGER.error("Problems rendering scene graph to SVG", t);
392 return svgGenerator.getRoot();
395 public static String printSVGDocument(Element doce) {
397 StringBuilder result = new StringBuilder();
399 NodeList nl = doce.getChildNodes();
401 for(int i=0;i<nl.getLength();i++) {
403 ByteArrayOutputStream os = new ByteArrayOutputStream();
407 TransformerFactory tf = TransformerFactory.newInstance();
408 Transformer transformer = tf.newTransformer();
409 transformer.setOutputProperty(OutputKeys.OMIT_XML_DECLARATION, "yes");
410 transformer.setOutputProperty(OutputKeys.STANDALONE, "no");
411 transformer.setOutputProperty(OutputKeys.METHOD, "xml");
412 transformer.setOutputProperty(OutputKeys.INDENT, "yes");
413 transformer.setOutputProperty(OutputKeys.ENCODING, "UTF-8");
414 transformer.setOutputProperty("{http://xml.apache.org/xslt}indent-amount", "4");
416 transformer.transform(new DOMSource(nl.item(i)),
417 new StreamResult(new OutputStreamWriter(os, "UTF-8")));
422 } catch (Throwable t) {
423 LOGGER.error("Problems formatting SVGDocument to text.", t);
426 result.append(new String(os.toByteArray()));
430 return result.toString();
434 public static String renderSVG(ICanvasContext ctx) {
436 // Get a DOMImplementation.
437 DOMImplementation domImpl = GenericDOMImplementation.getDOMImplementation();
439 // Create an instance of org.w3c.dom.Document.
440 String svgNS = "http://www.w3.org/2000/svg";
441 Document document = domImpl.createDocument(svgNS, "svg", null);
443 // Create an instance of the SVG Generator.
444 SVGGraphics2D svgGenerator = new Generator(document);
446 StringBuilder b = new StringBuilder();
450 G2DSceneGraph sg = ctx.getSceneGraph();
451 G2DParentNode root = (G2DParentNode) sg.getRootNode();
453 // rtree is the actual content of the diagram
454 RTreeNode rtree = NodeUtil.getNearestChildByClass(root, RTreeNode.class);
455 Rectangle2D rtreeBounds = NodeUtil.getLocalBounds(rtree);
457 // nav is a node that has zooming functionalities
458 NavigationNode nav = NodeUtil.getNearestChildByClass(root, NavigationNode.class);
459 nav.setZoomEnabled(true);
461 // fit view with the contents of rtreeBounds
462 nav.zoomTo(rtreeBounds);
464 // get the bounds of the content
465 Rectangle2D content = NodeUtil.getLocalBounds(nav);
467 svgGenerator.scale(3,3);
469 // translate svgGenerator to the x and y coordinates of current content
470 svgGenerator.translate(-1 * content.getX(), (-1 * content.getY()));
472 Rectangle2D destination = new Rectangle2D.Double(0,0,1000,1000);
473 double sx = destination.getWidth() / content.getWidth();
474 double sy = destination.getHeight() / content.getHeight();
475 double scale = sx < sy ? sx : sy;
477 // Set svgCanvasSize to the given size parameters
478 svgGenerator.setSVGCanvasSize(new Dimension((int)(scale * content.getWidth()), (int)(scale * content.getHeight())));
479 svgGenerator.setClip(content);
481 double trX = -1 * content.getX();
482 double trY = -1 * content.getY();
484 b.append("<svg width=\"100%\" height=\"100%\" stroke=\"black\"><g transform=\"translate(").append(trX).append(' ').append(trY).append(")\">");
486 IG2DNodeVisitor visitor = new IG2DNodeVisitor() {
490 HashMap<IG2DNode,Integer> enters = new HashMap<>();
493 public void enter(IG2DNode node) {
495 if(node instanceof ConnectionNode) {
496 Element doc = renderSVGNode((IG2DNode)node);
497 String svg = printSVGDocument(doc);
499 } else if (node instanceof SVGNode) {
500 SVGNode svg = (SVGNode)node;
501 b.append(svg.getSVGText());
502 } else if (node instanceof G2DParentNode) {
503 AffineTransform at = node.getTransform();
504 if(node instanceof SingleElementNode) {
505 SingleElementNode sen = (SingleElementNode)node;
506 if(sen.getKey() != null) {
507 String key = sen.getKey().toString();
508 b.append("\n<g class=\"definedElement\" id=\"" + key + "\">");
511 if(!at.isIdentity()) {
512 if(at.getScaleX() == 1.0 && at.getScaleY() == 1.0 && at.getShearX() == 0.0 && at.getShearY() == 0.0) {
513 String m = "translate(" + at.getTranslateX() + " " + at.getTranslateY() + ")";
514 b.append("\n<g transform=\"" + m + "\">");
516 double[] ds = new double[6];
518 String m = "matrix(" + ds[0] + " " + ds[1] + " " + ds[2] + " " + ds[3] + " " + ds[4] + " " + ds[5] + ")";
519 b.append("\n<g transform=\"" + m + "\">");
524 enters.put(node, b.length());
529 public void leave(IG2DNode node) {
530 if(node instanceof ConnectionNode || node instanceof SVGNode) {
532 } else if (node instanceof G2DParentNode) {
533 AffineTransform at = node.getTransform();
534 if(!at.isIdentity()) {
537 if(node instanceof SingleElementNode) {
538 SingleElementNode sen = (SingleElementNode)node;
539 if(sen.getKey() != null) {
540 int enterLength = enters.get(node);
541 if(b.length() == enterLength) {
542 Element doc = renderSVGNode((IG2DNode)node);
543 String svg = printSVGDocument(doc);
556 } catch (Throwable t) {
557 LOGGER.error("Problems rendering canvas context to SVG", t);
560 b.append("</g></svg>");
561 //System.err.println(" == FINAL RESULT == ");
562 //System.err.println(b);