bundles/org.simantics.modeling/src/org/simantics/modeling/SCLScenegraph.java

   1 package org.simantics.modeling;
   2
   3 import java.awt.BasicStroke;
   4 import java.awt.Color;
   5 import java.awt.Dimension;
   6 import java.awt.RenderingHints;
   7 import java.awt.RenderingHints.Key;
   8 import java.awt.geom.AffineTransform;
   9 import java.awt.geom.Rectangle2D;
  10 import java.io.ByteArrayOutputStream;
  11 import java.io.OutputStreamWriter;
  12 import java.util.ArrayList;
  13 import java.util.Arrays;
  14 import java.util.Collection;
  15 import java.util.HashMap;
  16 import java.util.HashSet;
  17 import java.util.List;
  18 import java.util.Map;
  19 import java.util.Random;
  20 import java.util.Set;
  21 import java.util.UUID;
  22 import java.util.function.Function;
  23 import java.util.stream.Collectors;
  24
  25 import javax.xml.transform.OutputKeys;
  26 import javax.xml.transform.Transformer;
  27 import javax.xml.transform.TransformerFactory;
  28 import javax.xml.transform.dom.DOMSource;
  29 import javax.xml.transform.stream.StreamResult;
  30
  31 import org.apache.batik.dom.GenericDOMImplementation;
  32 import org.apache.batik.svggen.SVGGeneratorContext;
  33 import org.apache.batik.svggen.SVGGraphics2D;
  34 import org.simantics.Simantics;
  35 import org.simantics.datatypes.literal.GUID;
  36 import org.simantics.db.ReadGraph;
  37 import org.simantics.db.Resource;
  38 import org.simantics.db.common.request.IndexRoot;
  39 import org.simantics.db.common.request.ResourceRead;
  40 import org.simantics.db.common.request.UnaryRead;
  41 import org.simantics.db.exception.DatabaseException;
  42 import org.simantics.db.exception.RuntimeDatabaseException;
  43 import org.simantics.db.layer0.variable.Variables;
  44 import org.simantics.diagram.elements.DecorationSVGNode;
  45 import org.simantics.diagram.elements.DiagramNodeUtil;
  46 import org.simantics.diagram.elements.TextGridNode;
  47 import org.simantics.diagram.elements.TextNode;
  48 import org.simantics.diagram.stubs.DiagramResource;
  49 import org.simantics.g2d.canvas.Hints;
  50 import org.simantics.g2d.canvas.ICanvasContext;
  51 import org.simantics.g2d.diagram.DiagramHints;
  52 import org.simantics.g2d.diagram.IDiagram;
  53 import org.simantics.g2d.diagram.handler.DataElementMap;
  54 import org.simantics.g2d.diagram.participant.ElementPainter.SelectionShapeNode;
  55 import org.simantics.g2d.diagram.participant.Selection;
  56 import org.simantics.g2d.element.IElement;
  57 import org.simantics.g2d.scenegraph.ICanvasSceneGraphProvider;
  58 import org.simantics.g2d.utils.CanvasUtils;
  59 import org.simantics.layer0.Layer0;
  60 import org.simantics.scenegraph.INode;
  61 import org.simantics.scenegraph.ParentNode;
  62 import org.simantics.scenegraph.g2d.G2DParentNode;
  63 import org.simantics.scenegraph.g2d.G2DRenderingHints;
  64 import org.simantics.scenegraph.g2d.G2DSceneGraph;
  65 import org.simantics.scenegraph.g2d.IG2DNode;
  66 import org.simantics.scenegraph.g2d.IG2DNodeVisitor;
  67 import org.simantics.scenegraph.g2d.events.command.Commands;
  68 import org.simantics.scenegraph.g2d.nodes.BackgroundNode;
  69 import org.simantics.scenegraph.g2d.nodes.BoundsNode;
  70 import org.simantics.scenegraph.g2d.nodes.ConnectionNode;
  71 import org.simantics.scenegraph.g2d.nodes.DataNode;
  72 import org.simantics.scenegraph.g2d.nodes.LinkNode;
  73 import org.simantics.scenegraph.g2d.nodes.NavigationNode;
  74 import org.simantics.scenegraph.g2d.nodes.SVGNode;
  75 import org.simantics.scenegraph.g2d.nodes.SelectionNode;
  76 import org.simantics.scenegraph.g2d.nodes.SingleElementNode;
  77 import org.simantics.scenegraph.g2d.nodes.connection.RouteGraphNode;
  78 import org.simantics.scenegraph.g2d.nodes.spatial.RTreeNode;
  79 import org.simantics.scenegraph.utils.NodeUtil;
  80 import org.simantics.scl.runtime.function.Function1;
  81 import org.simantics.scl.runtime.tuple.Tuple2;
  82 import org.simantics.trend.impl.ItemNode;
  83 import org.simantics.utils.threads.ThreadUtils;
  84 import org.slf4j.Logger;
  85 import org.slf4j.LoggerFactory;
  86 import org.w3c.dom.DOMImplementation;
  87 import org.w3c.dom.Document;
  88 import org.w3c.dom.Element;
  89 import org.w3c.dom.Node;
  90 import org.w3c.dom.NodeList;
  91
  92 public class SCLScenegraph {
  93
  94         private static final Logger LOGGER = LoggerFactory.getLogger(SCLScenegraph.class);
  95
  96         private static final String MAIN_SECTION = "main";
  97         private static final String SELECTION_SECTION = "selection";
  98         private static final String SELECTION_MASK_SECTION = "selectionMask";
  99
 100         private static final String[] ALL_SECTIONS = { MAIN_SECTION, SELECTION_SECTION, SELECTION_MASK_SECTION };
 101
 102         // Changed from 0.001 to 0.0001 to prevent creation of huge BufferedImage's when
 103         // generating PDF from SVG. If SVG contains any transparency then Batik uses
 104         // bitmap-rendering which remarkably slows things down
 105         // See org.apache.batik.gvt.AbstractGraphicsNode.paint(Graphics2D) where decisions are made
 106         // if AlphaComposite should be painted
 107         private static final String OPACITY = "0.0001";
 108
 109         @Deprecated
 110         public static ICanvasSceneGraphProvider getICanvasSceneGraphProvider(Resource model, Resource diagram, String diagramRVI) throws DatabaseException, InterruptedException {
 111                 ICanvasSceneGraphProvider provider = DiagramNodeUtil.loadSceneGraphProvider(model, diagram, diagramRVI);
 112                 return provider;
 113         }
 114
 115         @Deprecated
 116         public static void disposeSceneGraphProvider(ICanvasSceneGraphProvider provider) {
 117                 provider.dispose();
 118         }
 119
 120     public static <T> T doWithICanvasSceneGraphProvider(Resource diagram, Function1<ICanvasSceneGraphProvider, T> func) throws DatabaseException {
 121         return doWithICanvasSceneGraphProvider(diagram, (Function<ICanvasSceneGraphProvider, T>) provider -> func.apply(provider));
 122     }
 123
 124     public static <T> T doWithICanvasSceneGraphProvider(Resource diagram, Function<ICanvasSceneGraphProvider, T> func) throws DatabaseException {
 125         Tuple2 result = Simantics.getSession().syncRequest(new ResourceRead<Tuple2>(diagram) {
 126
 127             @Override
 128             public Tuple2 perform(ReadGraph graph) throws DatabaseException {
 129                 Resource indexRoot = graph.syncRequest(new IndexRoot(resource));
 130                 String diagramRVI = Variables.getRVI(graph, resource);
 131                 return new Tuple2(indexRoot, diagramRVI);
 132             }
 133         });
 134         ICanvasSceneGraphProvider provider = DiagramNodeUtil.loadSceneGraphProvider((Resource) result.c0, diagram, (String) result.c1);
 135         try {
 136             return func.apply(provider);
 137         } finally {
 138             provider.dispose();
 139         }
 140     }
 141
 142     public static <T> T doWithCanvasContext(Resource diagram, Function1<ICanvasContext, T> func) throws DatabaseException {
 143         return doWithCanvasContext(diagram, (Function<ICanvasContext, T>) canvasContext -> func.apply(canvasContext));
 144     }
 145
 146     public static <T> T doWithCanvasContext(Resource diagram, Function<ICanvasContext, T> func) throws DatabaseException {
 147         return doWithICanvasSceneGraphProvider(diagram, (Function<ICanvasSceneGraphProvider, T>) provider -> func.apply(provider.getCanvasContext()));
 148     }
 149
 150         public static String getNodeTransform(ICanvasContext ctx, String name) {
 151
 152                 Set<TextNode> texts = NodeUtil.collectNodes(ctx.getSceneGraph(), TextNode.class);
 153                 for (TextNode text : texts) {
 154                         String nodeName = NodeUtil.getNodeName(text);
 155                         if (nodeName.equals(name)) {
 156                                 String transform = text.getTransform().toString();
 157                                 return transform;
 158                         }
 159                 }
 160                 return "No node found";
 161         }
 162
 163         public static String getNodeText(ICanvasContext ctx, String name) {
 164
 165                 Set<TextNode> texts = NodeUtil.collectNodes(ctx.getSceneGraph(), TextNode.class);
 166                 for (TextNode text : texts) {
 167                         String nodeName = NodeUtil.getNodeName(text);
 168                         if (nodeName.equals(name)) {
 169                                 String texti = text.getText();
 170                                 return texti;
 171                         }
 172                 }
 173                 return "No node found";
 174         }
 175
 176         public static String getNodeCount(ICanvasContext ctx) {
 177                 G2DSceneGraph g2 = ctx.getSceneGraph();
 178                 int amount = NodeUtil.countTreeNodes(g2);
 179                 return "Node count: " + amount;
 180         }
 181
 182     public static String getAllNodes (ICanvasContext ctx) {
 183
 184         Set<G2DSceneGraph> g2 = NodeUtil.collectNodes(ctx.getSceneGraph(), G2DSceneGraph.class);
 185         int amount = g2.size() +1;
 186         return "All nodes: " + amount;
 187     }
 188
 189     public static String getBoundsNodes (ICanvasContext ctx) {
 190
 191         Set<BoundsNode> bn = NodeUtil.collectNodes(ctx.getSceneGraph(), BoundsNode.class);
 192         int amount = bn.size();
 193         return "BoundsNodes: " + amount;
 194     }
 195
 196     public static String getBackgroundNodes (ICanvasContext ctx) {
 197
 198         Set<BackgroundNode> bg = NodeUtil.collectNodes(ctx.getSceneGraph(), BackgroundNode.class);
 199         int amount = bg.size();
 200         return "BackgroundNodes: " + amount;
 201     }
 202
 203     public static String getDataNodes (ICanvasContext ctx) {
 204
 205         Set<DataNode> dn = NodeUtil.collectNodes(ctx.getSceneGraph(), DataNode.class);
 206         int amount = dn.size();
 207         return "DataNodes: " + amount;
 208     }
 209
 210     public static String getNavigationNodes (ICanvasContext ctx) {
 211
 212         Set<NavigationNode> g2 = NodeUtil.collectNodes(ctx.getSceneGraph(), NavigationNode.class);
 213         int amount = g2.size();
 214         return "NavigationNodes: " + amount;
 215     }
 216
 217     public static String getParentNodes (ICanvasContext ctx) {
 218
 219         Set<G2DParentNode> g2 = NodeUtil.collectNodes(ctx.getSceneGraph(), G2DParentNode.class);
 220         int amount = g2.size();
 221         return "ParentNodes: " + amount;
 222     }
 223
 224     public static String getDecorationNodes (ICanvasContext ctx) {
 225
 226         Set<DecorationSVGNode> deco = NodeUtil.collectNodes(ctx.getSceneGraph(), DecorationSVGNode.class);
 227         int amount = deco.size();
 228         return "DecorationNodes: " + amount;
 229     }
 230
 231     public static String getSingleElementNodes (ICanvasContext ctx) {
 232
 233         Set<SingleElementNode> g2 = NodeUtil.collectNodes(ctx.getSceneGraph(), SingleElementNode.class);
 234         int amount = g2.size();
 235         return "SingleElementNodes: " + amount;
 236     }
 237
 238     public static String getConnectionNodes (ICanvasContext ctx) {
 239
 240         Set<ConnectionNode> g2 = NodeUtil.collectNodes(ctx.getSceneGraph(), ConnectionNode.class);
 241         int amount = g2.size();
 242         return "ConnectionNodes: " + amount;
 243     }
 244
 245     public static String getTextNodes (ICanvasContext ctx) {
 246
 247         Set<TextNode> tn = NodeUtil.collectNodes(ctx.getSceneGraph(), TextNode.class);
 248         Set<TextGridNode> tgn = NodeUtil.collectNodes(ctx.getSceneGraph(), TextGridNode.class);
 249         int amount = tn.size() + tgn.size();
 250         return "TextNodes: " + amount;
 251     }
 252
 253     public static String getItemNodes (ICanvasContext ctx) {
 254
 255         Set<ItemNode> item = NodeUtil.collectNodes(ctx.getSceneGraph(), ItemNode.class);
 256         int amount = item.size();
 257         return "ItemNodes: " + amount;
 258     }
 259
 260     public static String editNodeText (ICanvasContext ctx, String module, String previous_value, String new_value) {
 261
 262         Set<TextNode> textGridNodes = NodeUtil.collectNodes(ctx.getSceneGraph(), TextNode.class);
 263         for (TextNode modulenode : textGridNodes) {
 264                 if (module.equals(modulenode.getText())) {
 265                         //System.out.println("Module what we were looking for: " + module);
 266                         //System.out.println("Modulenode: " + modulenode.getText());
 267
 268                         ParentNode<?> parentnode = modulenode.getParent();
 269                         //System.out.println("Parentnode: " + parentnode);
 270
 271                         Collection<TextNode> textnodes = (Collection<TextNode>) parentnode.getNodes();
 272                         for (TextNode valuenode : textnodes) {
 273                                 if (previous_value.equals(valuenode.getText())) {
 274                                         //System.out.println("Value what we were looking for: " + previous_value);
 275                                         //System.out.println("Valuenode: " + valuenode.getText());
 276
 277                                         //valuenode.setEditMode(true);
 278                                         valuenode.activateEdit(0, null, ctx);
 279                                         valuenode.setText(new_value);
 280                                         valuenode.fireTextEditingEnded();
 281
 282                                         //System.out.println("valuenode modified: " + valuenode);
 283                                         return "Modified module " + module + " with value " + new_value;
 284                                 }
 285                         }
 286                         return "Not found module : " + module;
 287                 }
 288         }
 289         return "No nodes in scenegraph!";
 290     }
 291
 292     /**
 293      * Returns background color of a canvasContext or null.
 294      * @param ctx
 295      * @return color in RGBA List<Integer> format, or null.
 296      */
 297     public static List<Integer> getBackgroundColor(ICanvasContext ctx) {
 298         Color color = ctx.getDefaultHintContext().getHint(Hints.KEY_BACKGROUND_COLOR);
 299         if (color == null)
 300                 return null;
 301         ArrayList<Integer> rgba = new ArrayList<>(4);
 302         rgba.add(color.getRed());
 303         rgba.add(color.getGreen());
 304         rgba.add(color.getBlue());
 305         rgba.add(color.getAlpha());
 306         return rgba;
 307     }
 308
 309     public static String sceneGraphTest (ICanvasContext ctx, String module, String value) {
 310
 311         boolean module_founded = false;
 312         boolean value_founded = false;
 313
 314         Set<G2DSceneGraph> g2 = NodeUtil.collectNodes(ctx.getSceneGraph(), G2DSceneGraph.class);
 315         System.out.println("Total amount of nodes: " + g2.size() + 1);
 316
 317         Set<TextGridNode> grid = NodeUtil.collectNodes(ctx.getSceneGraph(), TextGridNode.class);
 318         Integer textGridNodeAmount = grid.size();
 319         System.out.println("Amount of TextGridNodes " + textGridNodeAmount);
 320
 321         Set<TextNode> texts = NodeUtil.collectNodes(ctx.getSceneGraph(), TextNode.class);
 322         Integer textNodeAmount = grid.size();
 323         System.out.println("Amount of TextNodes " + textNodeAmount);
 324
 325         for (TextNode node : texts) {
 326             if (module.equals(node.getText())) {
 327                 module_founded = true;
 328                 System.out.println("Correct module " + module + " founded.");
 329             }
 330             if (value.equals(node.getText())) {
 331                 value_founded = true;
 332                 System.out.println("Correct value " + value + " founded.");
 333             }
 334         }
 335
 336         if (value_founded == true && module_founded == true) {
 337                 return "Found both correct module " + module + " and value " + value;
 338         }
 339         if (value_founded == false && module_founded == true) {
 340                 return "Found only correct module " + module + " but not value " + value;
 341         }
 342         if (value_founded == true && module_founded == false) {
 343                 return "Found only correct value " + value + " but not module " + module;
 344         }
 345         else {
 346                 return "Didn't found either module " + module + " or value " + value;
 347         }
 348     }
 349
 350     public static boolean copyPaste (final ICanvasContext source_ctx, final ICanvasContext target_ctx, List<Resource> modules) throws DatabaseException {
 351
 352         IDiagram idiagram = source_ctx.getDefaultHintContext().getHint(DiagramHints.KEY_DIAGRAM);
 353
 354                 DataElementMap dem = idiagram.getDiagramClass().getAtMostOneItemOfClass(DataElementMap.class);
 355                 if (dem != null) {
 356                         final Collection<IElement> newSelection = new ArrayList<IElement>();
 357                         for (Resource module : modules) {
 358                                 IElement element = dem.getElement(idiagram, module);
 359                                 if (element != null) {
 360                                         newSelection.add(element);
 361                                 } else {
 362                                         throw new DatabaseException("Could not find IElement for " + element);
 363                                 }
 364                         }
 365
 366                         ThreadUtils.syncExec(source_ctx.getThreadAccess(), new Runnable() {
 367                     @Override
 368                     public void run() {
 369                         if (source_ctx.isDisposed())
 370                             return;
 371                         Selection selection = source_ctx.getAtMostOneItemOfClass(Selection.class);
 372                         if (selection != null) {
 373                             // This prevents workbench selection from being left over.
 374                             // Also prevents scene graph crap from being left on the screen.
 375                             selection.setSelection(0, newSelection);
 376                         }
 377                                 CanvasUtils.sendCommand(source_ctx, Commands.COPY);
 378                                 CanvasUtils.sendCommand(target_ctx, Commands.PASTE);
 379                     }
 380                 });
 381
 382                 //}
 383
 384                 while(source_ctx.getEventQueue().size() > 0) {
 385                         try {
 386                                 Thread.sleep(10);
 387                         } catch (InterruptedException e) {
 388                                 throw new DatabaseException(e);
 389                         }
 390                 }
 391
 392                 ThreadUtils.syncExec(source_ctx.getThreadAccess(), new Runnable() {
 393             @Override
 394             public void run() {
 395             }
 396         });
 397
 398                 }
 399                 return true;
 400     }
 401
 402         static class Generator extends SVGGraphics2D {
 403
 404                 int elemLevel = 0;
 405                 String newElementId = null;
 406                 ArrayList<Element> elements = new ArrayList<Element>();
 407
 408                 public static final String svgNS = "http://www.w3.org/2000/svg";
 409
 410                 public Generator(SVGGeneratorContext ctx, boolean textAsShapes) {
 411                         super(ctx, textAsShapes);
 412                 }
 413
 414                 public Generator(Document document) {
 415                         super(document);
 416                         // prevent batik comment in each g-element
 417                         getGeneratorContext().setComment(null);
 418                 }
 419
 420                 @Override
 421                 public Element getRoot() {
 422                         Element root = super.getRoot();
 423                         for(Element e : elements) {
 424                                 root.appendChild(e);
 425                         }
 426                         return root;
 427                 }
 428
 429                 @Override
 430                 public void setRenderingHint(Key arg0, Object arg1) {
 431                         if(G2DRenderingHints.KEY_BEGIN_ELEMENT == arg0) {
 432                                 elemLevel++;
 433                         }
 434                         if(G2DRenderingHints.KEY_ELEMENT_ID == arg0) {
 435                                 if(arg1 != null)
 436                                         newElementId = arg1.toString();
 437                                 else
 438                                         newElementId = UUID.randomUUID().toString();
 439                         }
 440                         if(G2DRenderingHints.KEY_END_ELEMENT == arg0) {
 441                                 elemLevel--;
 442                                 if(elemLevel == 0) {
 443                                         Element group = getDOMFactory().createElement(SVG_G_TAG);
 444                                         //Element group = getDOMFactory().createElementNS(SVG_NAMESPACE_URI, SVG_G_TAG);
 445                                         group.setAttributeNS(null, "id", newElementId);
 446                                         group.setAttributeNS(null, "class", arg1.toString());
 447                                         getRoot(group);
 448                                         elements.add(group);
 449                                 }
 450                         }
 451                         super.setRenderingHint(arg0, arg1);
 452                 }
 453
 454         }
 455
 456         public static Element renderSVGNode(SVGGraphics2D svgGenerator, IG2DNode node) {
 457
 458                 try {
 459
 460                         svgGenerator.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
 461                         svgGenerator.setRenderingHint(RenderingHints.KEY_RENDERING, RenderingHints.VALUE_RENDER_QUALITY);
 462                         svgGenerator.setRenderingHint(RenderingHints.KEY_TEXT_ANTIALIASING, RenderingHints.VALUE_TEXT_ANTIALIAS_ON);
 463
 464                         node.render(svgGenerator);
 465
 466                 } catch (Throwable t) {
 467                         LOGGER.error("Problems rendering scene graph to SVG", t);
 468                 }
 469
 470                 return svgGenerator.getRoot();
 471         }
 472
 473         public static String printSVGDocument(Element doce) {
 474
 475                 StringBuilder result = new StringBuilder();
 476
 477                 NodeList nl =  doce.getChildNodes();
 478
 479                 for(int i=0;i<nl.getLength();i++) {
 480
 481                         ByteArrayOutputStream os = new ByteArrayOutputStream();
 482
 483                         try {
 484
 485                                 TransformerFactory tf = TransformerFactory.newInstance();
 486                                 Transformer transformer = tf.newTransformer();
 487                                 transformer.setOutputProperty(OutputKeys.OMIT_XML_DECLARATION, "yes");
 488                                 transformer.setOutputProperty(OutputKeys.STANDALONE, "no");
 489                                 transformer.setOutputProperty(OutputKeys.METHOD, "xml");
 490                                 transformer.setOutputProperty(OutputKeys.INDENT, "yes");
 491                                 transformer.setOutputProperty(OutputKeys.ENCODING, "UTF-8");
 492                                 transformer.setOutputProperty("{http://xml.apache.org/xslt}indent-amount", "4");
 493
 494                                 transformer.transform(new DOMSource(nl.item(i)),
 495                                                 new StreamResult(new OutputStreamWriter(os, "UTF-8")));
 496
 497                                 os.flush();
 498                                 os.close();
 499
 500                         } catch (Throwable t) {
 501                                 LOGGER.error("Problems formatting SVGDocument to text.", t);
 502                         }
 503
 504                         result.append(new String(os.toByteArray()));
 505
 506                 }
 507
 508                 return result.toString();
 509
 510         }
 511
 512         public static String renderSVG3(ICanvasContext ctx) {
 513                 return renderSVG3(ctx, -1, -1);
 514         }
 515
 516         public static String renderSVG3(ICanvasContext ctx, double width, double height) {
 517         return renderSVG0(width, height, ctx, p0 -> p0.stream().collect(Collectors.toMap(p1 -> p1, p2 -> p2)));
 518         }
 519
 520     /**
 521      * @param graph
 522      * @param component
 523      * @throws DatabaseException
 524      */
 525     private static Object[] createURIBasedL0Identifier(ReadGraph graph, Resource component) throws DatabaseException {
 526         String uri = graph.getPossibleURI(component);
 527         int hashCode = uri.hashCode();
 528         Random random = new Random(hashCode);
 529         long l1 = random.nextLong();
 530         long l2 = random.nextLong();
 531         return new Object[] { l1, l2 };
 532     }
 533
 534     /**
 535      * Default no-op mapper
 536      */
 537     private static final Function1<Set<?>, Map<?, ?>> mapper = p0 -> p0.stream().collect(Collectors.toMap(p1 -> p1, p2 -> p2));
 538
 539     public static String renderSVG(ICanvasContext ctx, double width, double height) {
 540         return renderSVG0(width, height, ctx, mapper);
 541     }
 542
 543     public static String renderSVG(ICanvasContext ctx) {
 544         return renderSVG(ctx, -1, -1);
 545     }
 546
 547     public static String renderSVGMapIdentifiers(ICanvasContext ctx) {
 548                 return renderSVGMapIdentifiers(ctx, -1, -1);
 549     }
 550
 551     /**
 552      * Renders ICanvasContext into SVG by mapping the SVG id's into URI based
 553      * GUID's
 554      *
 555      * @param ctx
 556      * @return
 557      */
 558     public static String renderSVGMapIdentifiers(ICanvasContext ctx, double width, double height) {
 559         return renderSVG0(width, height, ctx, new Function1<Set<?>, Map<?, ?>>() {
 560
 561             @Override
 562             public Map<?, ?> apply(Set<?> p0) {
 563                 try {
 564                     return Simantics.getSession().syncRequest(new UnaryRead<Set<?>, Map<?, ?>>(p0) {
 565
 566                         @Override
 567                         public Map<?, ?> perform(ReadGraph graph) throws DatabaseException {
 568                             ModelingResources MOD = ModelingResources.getInstance(graph);
 569                             DiagramResource DIA = DiagramResource.getInstance(graph);
 570                             Layer0 L0 = Layer0.getInstance(graph);
 571                             return parameter.stream().collect(Collectors.toMap(p -> p, p -> {
 572                                 try {
 573                                     if (p instanceof Resource) {
 574                                         Resource element = (Resource) p;
 575                                         if (graph.isInstanceOf(element, DIA.Element)) {
 576                                             Resource component = graph.getPossibleObject(element, MOD.ElementToComponent);
 577                                             if (component != null) {
 578                                                 GUID identifier = graph.getPossibleRelatedValue(component, L0.identifier, GUID.BINDING);
 579                                                 if (identifier != null) {
 580                                                     return Arrays.toString(createURIBasedL0Identifier(graph, component));
 581                                                 } else {
 582                                                     LOGGER.error("Component {} does not have GUID identifier!", component);
 583                                                 }
 584                                             } else if (graph.isInstanceOf(element, DIA.Connection) || graph.isInstanceOf(element, DIA.Terminal)) {
 585                                                 // Ok, lets create a hashcode for connections and terminals as they do not have identifiers
 586                                                 return graph.getURI(element).hashCode();
 587                                             } else {
 588                                                 // Ok, lets create a hashcode or either return empty string in cases where no ID is required
 589                                                 if (graph.hasStatement(element, L0.HasName))
 590                                                     return graph.getURI(element).hashCode();
 591                                                 return "";
 592                                             }
 593                                         }
 594                                     } else {
 595                                         LOGGER.error("Parameter p {} is not resource but it is {}", p, p.getClass());
 596                                     }
 597                                     return p;
 598                                 } catch (DatabaseException e) {
 599                                     throw new RuntimeDatabaseException(e);
 600                                 }
 601                             }));
 602                         }
 603                     });
 604                 } catch (DatabaseException e) {
 605                     LOGGER.error("Could not apply mappings", e);
 606                     throw new RuntimeDatabaseException(e);
 607                 }
 608             }
 609         });
 610         }
 611
 612     static class RenderSVGContext {
 613
 614         Map<String,StringBuilder> documents = new HashMap<>();
 615
 616         public void append(String[] keys, String svgText) {
 617                 for(String key : keys) append(key, svgText);
 618         }
 619
 620         public void append(String key, String svgText) {
 621                 StringBuilder builder = documents.get(key);
 622                 if(builder == null) {
 623                         builder = new StringBuilder();
 624                         documents.put(key, builder);
 625                 }
 626                 builder.append(svgText);
 627         }
 628
 629         public void append(RenderSVGContext other) {
 630                 for(String key : other.documents.keySet()) {
 631                         append(key, other.get(key));
 632                 }
 633         }
 634
 635         public String get(String key) {
 636                 StringBuilder builder = documents.get(key);
 637                 if(builder == null) return "";
 638                 else return builder.toString();
 639         }
 640
 641     }
 642
 643         private static String renderSVG0(double width, double height, ICanvasContext ctx, Function1<Set<?>, Map<?, ?>> mappingFunction) {
 644
 645                 // Get a DOMImplementation.
 646                 DOMImplementation domImpl = GenericDOMImplementation.getDOMImplementation();
 647
 648                 // Create an instance of org.w3c.dom.Document.
 649                 String svgNS = "http://www.w3.org/2000/svg";
 650                 Document document = domImpl.createDocument(svgNS, "svg", null);
 651
 652                 // Create an instance of the SVG Generator.
 653                 SVGGraphics2D svgGenerator = new Generator(document);
 654
 655                 RenderSVGContext result = new RenderSVGContext();
 656
 657                 double[] matrix = new double[6];
 658                 try {
 659
 660             Selection selection = ctx.getAtMostOneItemOfClass(Selection.class);
 661             if (selection != null) {
 662                 // This prevents workbench selection from being left over.
 663                 // Also prevents scene graph crap from being left on the screen.
 664                 IDiagram d = ctx.getDefaultHintContext().getHint(DiagramHints.KEY_DIAGRAM);
 665                 selection.setSelection(0, d.getElements());
 666             }
 667
 668                         G2DSceneGraph sg = ctx.getSceneGraph();
 669                         sg.performCleanup();
 670                         G2DParentNode root = (G2DParentNode) sg.getRootNode();
 671
 672                         // rtree is the actual content of the diagram
 673                         RTreeNode rtree = NodeUtil.getNearestChildByClass(root, RTreeNode.class);
 674                         Rectangle2D rtreeBounds = NodeUtil.getLocalBounds(rtree);
 675
 676                         // get the bounds of the content
 677                         Rectangle2D content = rtreeBounds;
 678
 679
 680                         if (content != null) {
 681                          // To account for dynamic padding of selection rectangles (5 units + stroke width)
 682                     int offset = 6;
 683
 684                     double scale = width < 0 || height < 0 ? 1.0 : Math.min((width - 2*offset) / content.getWidth(), (height - 2*offset) / content.getHeight());
 685
 686                     AffineTransform tr = new AffineTransform();
 687                     tr.translate(offset, offset);
 688                     tr.scale(scale, scale);
 689                             tr.translate(-content.getX(), -content.getY());
 690                             tr.getMatrix(matrix);
 691                             svgGenerator.setSVGCanvasSize(new Dimension((int)Math.ceil(scale * content.getWidth()) + 2*offset, (int)Math.ceil(scale * content.getHeight()) + 2*offset));
 692                         } else {
 693                             svgGenerator.setSVGCanvasSize(new Dimension(100, 100));
 694                         }
 695                         //svgGenerator.translate(offset, offset);
 696                         //svgGenerator.scale(scale, scale);
 697                         // translate svgGenerator to the x and y coordinates of current content
 698                         //svgGenerator.translate(-content.getX(), -content.getY());
 699
 700
 701                         //svgGenerator.setClip(content);
 702
 703                         result.append(MAIN_SECTION, "<g class=\"symbols\">");
 704                         result.append(SELECTION_SECTION, "<g class=\"selections\">");
 705                         result.append(SELECTION_MASK_SECTION, "<g class=\"selectionMasks\">");
 706
 707                         KeyVisitor keyVisitor = new KeyVisitor();
 708                         sg.accept(keyVisitor);
 709
 710                         Set<Object> keys = keyVisitor.getKeys();
 711
 712                         Map<?, ?> mappings = mappingFunction.apply(keys);
 713
 714                         IG2DNodeVisitor visitor = new PrintingVisitor(svgGenerator, result, mappings);
 715
 716                         sg.accept(visitor);
 717
 718                 } catch (Throwable t) {
 719                         LOGGER.error("Problems rendering canvas context to SVG", t);
 720                 }
 721
 722
 723                 result.append(ALL_SECTIONS, "</g>");
 724
 725                 StringBuilder res = new StringBuilder();
 726                 res.append("<svg width=\"100%\" height=\"100%\" stroke=\"black\">");
 727                 res.append("<g transform=\"matrix(").append(matrix[0]).append(",").append(matrix[1]).append(",").append(matrix[2]).append(",").append(matrix[3]).append(",").append(matrix[4]).append(",").append(matrix[5]).append(")\">");
 728                 res.append(result.get(MAIN_SECTION));
 729                 res.append(result.get(SELECTION_SECTION));
 730                 res.append(result.get(SELECTION_MASK_SECTION));
 731                 res.append("</g>");
 732                 res.append("</svg>");
 733
 734 //              System.err.println(" == FINAL RESULT == ");
 735 //              System.err.println(res.toString());
 736
 737                 return res.toString();
 738
 739         }
 740
 741
 742
 743         private static class KeyVisitor implements IG2DNodeVisitor {
 744
 745         private Set<Object> keys = new HashSet<>();
 746
 747         @Override
 748         public void enter(IG2DNode node) {
 749             if (node instanceof SingleElementNode) {
 750                 Object key = ((SingleElementNode) node).getKey();
 751                 if (key != null) {
 752                     keys.add(key);
 753                 }
 754             }
 755         }
 756
 757         @Override
 758         public void leave(IG2DNode node) {
 759             // Nothing to do
 760         }
 761
 762         public Set<Object> getKeys() {
 763             return keys;
 764         }
 765         }
 766
 767         private static class PrintingVisitor implements IG2DNodeVisitor {
 768
 769         int indent = 0;
 770
 771         HashMap<SingleElementNode,RenderSVGContext> senBuilders = new HashMap<>();
 772
 773         private RenderSVGContext result;
 774         private SVGGraphics2D svgGenerator;
 775
 776         private Map<?, ?> mappings;
 777
 778         public PrintingVisitor(SVGGraphics2D svgGenerator, RenderSVGContext result, Map<?, ?> mappings) {
 779             this.result = result;
 780             this.mappings = mappings;
 781             this.svgGenerator = svgGenerator;
 782         }
 783
 784         private String getKey(SingleElementNode node) {
 785             String key;
 786                 if(node.getKey() != null) {
 787                 if (mappings.containsKey(node.getKey()))
 788                     key = mappings.get(node.getKey()).toString();
 789                 else
 790                     key = node.getKey().toString();
 791                 } else {
 792                         key = Long.toString(node.getId());
 793                 }
 794             return key;
 795         }
 796
 797         @Override
 798         public void enter(IG2DNode node) {
 799
 800                 RenderSVGContext parentBuilder = getParentBuilder(node);
 801
 802             indent++;
 803             if(node instanceof ConnectionNode) {
 804
 805                 for(RouteGraphNode n : NodeUtil.collectNodes(node, RouteGraphNode.class)) {
 806                     n.setIgnoreSelection(true);
 807                 }
 808
 809                 String key = getKey((ConnectionNode) node);
 810                 parentBuilder.append(MAIN_SECTION, "\n<g class=\"connection\" id=\"" + key + "\">");
 811                 parentBuilder.append(SELECTION_SECTION, "\n<g style=\"visibility:hidden\" class=\"selection\" id=\"" + key + "\">");
 812                 parentBuilder.append(SELECTION_MASK_SECTION, "\n<g class=\"selectionMask\" opacity=\"" + OPACITY + "\" id=\"" + key + "\">");
 813
 814                 Element doc = renderSVGNode(svgGenerator, (IG2DNode)node);
 815                 String svg = printSVGDocument(doc);
 816                 parentBuilder.append(MAIN_SECTION, svg);
 817
 818                 for(RouteGraphNode n : NodeUtil.collectNodes(node, RouteGraphNode.class)) {
 819                     n.setIgnoreSelection(false);
 820                 }
 821
 822                 doc = renderSVGNode(svgGenerator, (IG2DNode)node);
 823                 svg = printSVGDocument(doc);
 824                 parentBuilder.append(SELECTION_SECTION, svg);
 825
 826                 BasicStroke bs = new BasicStroke(10f);
 827
 828                 for(RouteGraphNode n : NodeUtil.collectNodes(node, RouteGraphNode.class)) {
 829                     n.setDynamicStroke(bs);
 830                 }
 831
 832                 doc = renderSVGNode(svgGenerator, (IG2DNode)node);
 833                 svg = printSVGDocument(doc);
 834                 parentBuilder.append(SELECTION_MASK_SECTION, svg);
 835
 836                 parentBuilder.append(SELECTION_MASK_SECTION, "\n</g>");
 837                 parentBuilder.append(SELECTION_SECTION, "\n</g>");
 838                 parentBuilder.append(MAIN_SECTION, "\n</g>");
 839
 840             } else if (isSelection0(node)) {
 841
 842                 SelectionNode n = (SelectionNode)node;
 843                 SingleElementNode parentSEN = (SingleElementNode)NodeUtil.getNearestParentOfType(node, SingleElementNode.class);
 844                 if(parentSEN != null) {
 845
 846                         RenderSVGContext parentBuilder2 = getParentBuilder(parentSEN);
 847
 848                     String key = getKey(parentSEN);
 849                     n.setIgnore(false);
 850                     Element doc = renderSVGNode(svgGenerator, (IG2DNode)node);
 851                     n.setIgnore(true);
 852                     String svg = printSVGDocument(doc);
 853                     parentBuilder2.append(SELECTION_SECTION, "\n<g style=\"visibility:hidden\" class=\"selection\" id=\"" + key + "\">");
 854                     parentBuilder2.append(SELECTION_SECTION, svg);
 855                     parentBuilder2.append(SELECTION_SECTION, "\n</g>");
 856                     parentBuilder2.append(SELECTION_MASK_SECTION, "\n<g class=\"selectionMask\" id=\"" + key /*+ "\" transform=\"" + matrixString + "\"*/+ "\">");
 857                     Rectangle2D rect = n.getRect();
 858                     // NaN
 859                     if(rect.getHeight() == rect.getHeight() && rect.getWidth() == rect.getWidth()) {
 860                             parentBuilder2.append(SELECTION_MASK_SECTION,"<rect style=\"fill:#fff\" opacity=\"" + OPACITY + "\"");
 861                             parentBuilder2.append(SELECTION_MASK_SECTION," x=\"" + rect.getX() + "\" y=\"" + rect.getY() + "\"");
 862                             parentBuilder2.append(SELECTION_MASK_SECTION," width=\"" + rect.getWidth() + "\" height=\"" + rect.getHeight() + "\"");
 863                             parentBuilder2.append(SELECTION_MASK_SECTION,"></rect>");
 864                     }
 865                     parentBuilder2.append(SELECTION_MASK_SECTION,"\n</g>");
 866
 867                 }
 868             } else if (node instanceof SelectionNode) {
 869
 870                 Element doc = renderSVGNode(svgGenerator, (IG2DNode)node);
 871                 String svg = printSVGDocument(doc);
 872                 parentBuilder.append(MAIN_SECTION, "<g>");
 873                 parentBuilder.append(MAIN_SECTION, svg);
 874                 parentBuilder.append(MAIN_SECTION, "\n</g>");
 875
 876             } else if (node instanceof SVGNode) {
 877                 SVGNode svg = (SVGNode)node;
 878                 AffineTransform at = svg.getTransform();
 879                 if (!at.isIdentity()) {
 880                          if(at.getScaleX() == 1.0 && at.getScaleY() == 1.0 && at.getShearX() == 0.0 && at.getShearY() == 0.0) {
 881                          String m = "translate(" + at.getTranslateX() + " " + at.getTranslateY() + ")";
 882                          parentBuilder.append(MAIN_SECTION, "\n<g transform=\"" + m + "\">");
 883                      } else {
 884                          double[] ds = new double[6];
 885                          at.getMatrix(ds);
 886                          String m = "matrix(" + ds[0] + " " + ds[1] + " " + ds[2] + " " + ds[3] + " " + ds[4] + " " + ds[5] + ")";
 887                          parentBuilder.append(MAIN_SECTION, "\n<g transform=\"" + m + "\">");
 888                      }
 889                 }
 890                 parentBuilder.append(MAIN_SECTION, svg.getSVGText());
 891                 if (!at.isIdentity()) {
 892                         parentBuilder.append(MAIN_SECTION, "\n</g>");
 893                 }
 894             } else if (node instanceof G2DParentNode) {
 895                 AffineTransform at = node.getTransform();
 896                 if(node instanceof SingleElementNode) {
 897                     SingleElementNode sen = (SingleElementNode)node;
 898                     String key = getKey(sen);
 899                     String typeClass = sen.getTypeClass();
 900                     String clazz = "definedElement";
 901                     if(typeClass != null && !typeClass.isEmpty())
 902                         clazz = clazz + " " + typeClass;
 903
 904                     parentBuilder.append(MAIN_SECTION, "\n<g class=\""+clazz+"\" id=\"" + key + "\">");
 905                     senBuilders.put(sen, new RenderSVGContext());
 906                 }
 907                 if(!at.isIdentity()) {
 908                     if(at.getScaleX() == 1.0 && at.getScaleY() == 1.0 && at.getShearX() == 0.0 && at.getShearY() == 0.0) {
 909                         String m = "translate(" + at.getTranslateX() + " " + at.getTranslateY() + ")";
 910                         parentBuilder.append(ALL_SECTIONS, "\n<g transform=\"" + m + "\">");
 911                     } else {
 912                         double[] ds = new double[6];
 913                         at.getMatrix(ds);
 914                         String m = "matrix(" + ds[0] + " " + ds[1] + " " + ds[2] + " " + ds[3] + " " + ds[4] + " " + ds[5] + ")";
 915                         parentBuilder.append(ALL_SECTIONS, "\n<g transform=\"" + m + "\">");
 916                     }
 917                 }
 918             } else if (node instanceof TextNode) {
 919                  TextNode text = (TextNode)node;
 920
 921                  SingleElementNode parentSEN = (SingleElementNode)NodeUtil.getNearestParentOfType(node, SingleElementNode.class);
 922                  if(parentSEN != null) {
 923
 924                          text.setShowSelection(false);
 925                          Element doc = renderSVGNode(svgGenerator, (IG2DNode)node);
 926                      String svg = printSVGDocument(doc);
 927                      parentBuilder.append(MAIN_SECTION, svg);
 928
 929                          RenderSVGContext parentBuilder2 = getParentBuilder(parentSEN);
 930
 931                      String key = getKey(parentSEN);
 932                      text.setShowSelection(true);
 933                      doc = renderSVGNode(svgGenerator, (IG2DNode)node);
 934                      svg = printSVGDocument(doc);
 935
 936                      parentBuilder2.append(SELECTION_SECTION, "\n<g style=\"visibility:hidden\" class=\"selection\" id=\"" + key + "\">");
 937                      parentBuilder2.append(SELECTION_SECTION, svg);
 938                      parentBuilder2.append(SELECTION_SECTION, "\n</g>");
 939                      parentBuilder2.append(SELECTION_MASK_SECTION, "\n<g class=\"selectionMask\" id=\"" + key /*+ "\" transform=\"" + matrixString + "\"*/+ "\">");
 940                      Rectangle2D rect = text.getBounds();
 941                      // NaN
 942                      if(rect.getHeight() == rect.getHeight() && rect.getWidth() == rect.getWidth()) {
 943                             parentBuilder2.append(SELECTION_MASK_SECTION,"<rect style=\"fill:#fff\" opacity=\"" + OPACITY + "\"");
 944                             parentBuilder2.append(SELECTION_MASK_SECTION," x=\"" + rect.getX() + "\" y=\"" + rect.getY() + "\"");
 945                             parentBuilder2.append(SELECTION_MASK_SECTION," width=\"" + rect.getWidth() + "\" height=\"" + rect.getHeight() + "\"");
 946                             parentBuilder2.append(SELECTION_MASK_SECTION,"></rect>");
 947                      }
 948                      parentBuilder2.append(SELECTION_MASK_SECTION,"\n</g>");
 949
 950                  }
 951             } else if (!(node instanceof RouteGraphNode) && !(node instanceof LinkNode)){
 952                 try {
 953                         Element doc = renderSVGNode(svgGenerator, (IG2DNode)node);
 954                         NodeList gList = doc.getElementsByTagName("g");
 955                         if (gList.getLength() == 0)
 956                                 return;
 957                         boolean hasContent = false;
 958                         for (int i = 0; i < gList.getLength(); i++) {
 959                             Node gNode = gList.item(i);
 960                             if (gNode.hasChildNodes()) {
 961                                 hasContent = true;
 962                                 break;
 963                             }
 964                         }
 965                         if (!hasContent)
 966                             return;
 967                         String svg = printSVGDocument(doc);
 968                         if (node instanceof SelectionShapeNode) {
 969                                 SingleElementNode parentSEN = (SingleElementNode)NodeUtil.getNearestParentOfType(node, SingleElementNode.class);
 970                         if(parentSEN != null) {
 971                                  String key = getKey(parentSEN);
 972                                  RenderSVGContext parentBuilder2 = getParentBuilder(parentSEN);
 973                                  parentBuilder2.append(SELECTION_SECTION, "\n<g style=\"visibility:hidden\" class=\"selection\" id=\"" + key + "\">");
 974                              parentBuilder2.append(SELECTION_SECTION, svg);
 975                              parentBuilder2.append(SELECTION_SECTION, "\n</g>");
 976
 977                              parentBuilder2.append(SELECTION_MASK_SECTION, "\n<g class=\"selectionMask\" id=\"" + key /*+ "\" transform=\"" + matrixString + "\"*/+ "\">");
 978                              Rectangle2D rect = node.getBounds();
 979                              // NaN
 980                              if(rect.getHeight() == rect.getHeight() && rect.getWidth() == rect.getWidth()) {
 981                                     parentBuilder2.append(SELECTION_MASK_SECTION,"<rect style=\"fill:#fff\" opacity=\"" + OPACITY + "\"");
 982                                     parentBuilder2.append(SELECTION_MASK_SECTION," x=\"" + rect.getX() + "\" y=\"" + rect.getY() + "\"");
 983                                     parentBuilder2.append(SELECTION_MASK_SECTION," width=\"" + rect.getWidth() + "\" height=\"" + rect.getHeight() + "\"");
 984                                     parentBuilder2.append(SELECTION_MASK_SECTION,"></rect>");
 985                              }
 986                              parentBuilder2.append(SELECTION_MASK_SECTION,"\n</g>");
 987                         }
 988                         } else {
 989                                 parentBuilder.append(MAIN_SECTION, "<g class=\"" +node.getSimpleClassName() +"\">");
 990                                 parentBuilder.append(MAIN_SECTION, svg);
 991                                 parentBuilder.append(MAIN_SECTION, "\n</g>");
 992                         }
 993                 } catch (Exception e) {
 994                         // TODO: There are nodes that do not behave well when rendered to SVG. For backwards compatibility, we don't handle the exceptions.
 995                 }
 996             }
 997
 998             //enters.put(node, b.length());
 999
1000         }
1001
1002         private boolean isSelection0(IG2DNode node) {
1003
1004             if(node instanceof SelectionNode) {
1005                 SelectionNode sn = (SelectionNode)node;
1006                 return sn.getSelectionId() == 0;
1007             } else {
1008                 return false;
1009             }
1010
1011         }
1012
1013         private RenderSVGContext getParentBuilder(IG2DNode node) {
1014
1015             INode parentSEN = NodeUtil.getNearestParentOfType(node, SingleElementNode.class);
1016             if(parentSEN instanceof G2DSceneGraph) return result;
1017
1018             RenderSVGContext parentBuilder = senBuilders.get(parentSEN);
1019             if(parentBuilder == null) return result;
1020
1021             return parentBuilder;
1022
1023         }
1024
1025         @Override
1026         public void leave(IG2DNode node) {
1027
1028             if(node instanceof ConnectionNode || node instanceof SVGNode) {
1029                 // We are done
1030             } else if (node instanceof G2DParentNode) {
1031
1032                 RenderSVGContext parentBuilder = getParentBuilder(node);
1033
1034                 if(node instanceof SingleElementNode) {
1035                         SingleElementNode sen = (SingleElementNode)node;
1036                         RenderSVGContext b = senBuilders.get(sen);
1037                         String content = b.get(MAIN_SECTION);
1038                         if(content.isEmpty()) {
1039                                 if(sen.getKey() != null) {
1040
1041                                         for(SelectionNode n : NodeUtil.collectNodes(node, SelectionNode.class)) {
1042                                                 n.setIgnore(true);
1043                                         }
1044
1045                                         Element doc = renderSVGNode(svgGenerator, (IG2DNode)node);
1046                                         String svg = printSVGDocument(doc);
1047                                         parentBuilder.append(MAIN_SECTION, svg);
1048                                 }
1049                         } else {
1050                                 parentBuilder.append(b);
1051                         }
1052                 }
1053
1054                 AffineTransform at = node.getTransform();
1055                 if(!at.isIdentity()) {
1056                     parentBuilder.append(ALL_SECTIONS, "</g>");
1057                 }
1058                 if(node instanceof SingleElementNode) {
1059                     SingleElementNode sen = (SingleElementNode)node;
1060                     //if(sen.getKey() != null) {
1061                         parentBuilder.append(MAIN_SECTION, "</g>");
1062                     //}
1063                 }
1064             }
1065             indent --;
1066         }
1067         }
1068 }