import java.io.ByteArrayOutputStream;
import java.io.OutputStreamWriter;
import java.util.ArrayList;
+import java.util.Arrays;
import java.util.Collection;
import java.util.HashMap;
+import java.util.HashSet;
import java.util.List;
+import java.util.Map;
+import java.util.Random;
import java.util.Set;
import java.util.UUID;
+import java.util.stream.Collectors;
import javax.xml.transform.OutputKeys;
import javax.xml.transform.Transformer;
import org.apache.batik.dom.GenericDOMImplementation;
import org.apache.batik.svggen.SVGGeneratorContext;
import org.apache.batik.svggen.SVGGraphics2D;
+import org.simantics.Simantics;
+import org.simantics.datatypes.literal.GUID;
+import org.simantics.db.ReadGraph;
import org.simantics.db.Resource;
+import org.simantics.db.common.request.UnaryRead;
import org.simantics.db.exception.DatabaseException;
+import org.simantics.db.exception.RuntimeDatabaseException;
+import org.simantics.diagram.elements.DecorationSVGNode;
import org.simantics.diagram.elements.DiagramNodeUtil;
import org.simantics.diagram.elements.TextGridNode;
import org.simantics.diagram.elements.TextNode;
+import org.simantics.diagram.stubs.DiagramResource;
import org.simantics.g2d.canvas.ICanvasContext;
import org.simantics.g2d.diagram.DiagramHints;
import org.simantics.g2d.diagram.IDiagram;
import org.simantics.g2d.element.IElement;
import org.simantics.g2d.scenegraph.ICanvasSceneGraphProvider;
import org.simantics.g2d.utils.CanvasUtils;
+import org.simantics.layer0.Layer0;
import org.simantics.scenegraph.INode;
import org.simantics.scenegraph.ParentNode;
import org.simantics.scenegraph.g2d.G2DParentNode;
import org.simantics.scenegraph.g2d.nodes.BoundsNode;
import org.simantics.scenegraph.g2d.nodes.ConnectionNode;
import org.simantics.scenegraph.g2d.nodes.DataNode;
-import org.simantics.scenegraph.g2d.nodes.DecorationSVGNode;
import org.simantics.scenegraph.g2d.nodes.NavigationNode;
import org.simantics.scenegraph.g2d.nodes.SVGNode;
import org.simantics.scenegraph.g2d.nodes.SelectionNode;
import org.simantics.scenegraph.g2d.nodes.connection.RouteGraphNode;
import org.simantics.scenegraph.g2d.nodes.spatial.RTreeNode;
import org.simantics.scenegraph.utils.NodeUtil;
+import org.simantics.scl.runtime.function.Function1;
import org.simantics.trend.impl.ItemNode;
import org.simantics.utils.threads.ThreadUtils;
import org.slf4j.Logger;
public class SCLScenegraph {
private static final Logger LOGGER = LoggerFactory.getLogger(SCLScenegraph.class);
+
+ private static final String MAIN_SECTION = "main";
+ private static final String SELECTION_SECTION = "selection";
+ private static final String SELECTION_MASK_SECTION = "selectionMask";
+
+ private static final String[] ALL_SECTIONS = { MAIN_SECTION, SELECTION_SECTION, SELECTION_MASK_SECTION };
+
public static ICanvasSceneGraphProvider getICanvasSceneGraphProvider(Resource model, Resource diagram, String diagramRVI) throws DatabaseException, InterruptedException {
ICanvasSceneGraphProvider provider = DiagramNodeUtil.loadSceneGraphProvider(model, diagram, diagramRVI);
}
- public static String renderSVG(ICanvasContext ctx) {
+ public static String renderSVG3(ICanvasContext ctx) {
+ return renderSVG0(ctx, p0 -> p0.stream().collect(Collectors.toMap(p1 -> p1, p2 -> p2)));
+ }
+
+ /**
+ * @param graph
+ * @param component
+ * @throws DatabaseException
+ */
+ private static Object[] createURIBasedL0Identifier(ReadGraph graph, Resource component) throws DatabaseException {
+ String uri = graph.getPossibleURI(component);
+ int hashCode = uri.hashCode();
+ Random random = new Random(hashCode);
+ long l1 = random.nextLong();
+ long l2 = random.nextLong();
+ return new Object[] { l1, l2 };
+ }
+
+ /**
+ * Default no-op mapper
+ */
+ private static final Function1<Set<?>, Map<?, ?>> mapper = p0 -> p0.stream().collect(Collectors.toMap(p1 -> p1, p2 -> p2));
+
+ public static String renderSVG(ICanvasContext ctx) {
+ return renderSVG0(ctx, mapper);
+ }
+
+ /**
+ * Renders ICanvasContext into SVG by mapping the SVG id's into URI based
+ * GUID's
+ *
+ * @param ctx
+ * @return
+ */
+ public static String renderSVGMapIdentifiers(ICanvasContext ctx) {
+ return renderSVG0(ctx, new Function1<Set<?>, Map<?, ?>>() {
+
+ @Override
+ public Map<?, ?> apply(Set<?> p0) {
+ try {
+ return Simantics.getSession().syncRequest(new UnaryRead<Set<?>, Map<?, ?>>(p0) {
+
+ @Override
+ public Map<?, ?> perform(ReadGraph graph) throws DatabaseException {
+ ModelingResources MOD = ModelingResources.getInstance(graph);
+ DiagramResource DIA = DiagramResource.getInstance(graph);
+ Layer0 L0 = Layer0.getInstance(graph);
+ return parameter.stream().collect(Collectors.toMap(p -> p, p -> {
+ try {
+ if (p instanceof Resource) {
+ Resource element = (Resource) p;
+ if (graph.isInstanceOf(element, DIA.Element)) {
+ Resource component = graph.getPossibleObject(element, MOD.ElementToComponent);
+ if (component != null) {
+ GUID identifier = graph.getPossibleRelatedValue(component, L0.identifier, GUID.BINDING);
+ if (identifier != null) {
+ return Arrays.toString(createURIBasedL0Identifier(graph, component));
+ } else {
+ LOGGER.error("Component {} does not have GUID identifier!", component);
+ }
+ } else if (graph.isInstanceOf(element, DIA.Connection) || graph.isInstanceOf(element, DIA.Terminal)) {
+ // Ok, lets create a hashcode for connections and terminals as they do not have identifiers
+ return graph.getURI(element).hashCode();
+ } else {
+ // Ok, lets create a hashcode or either return empty string in cases where no ID is required
+ if (graph.hasStatement(element, L0.HasName))
+ return graph.getURI(element).hashCode();
+ return "";
+ }
+ }
+ } else {
+ LOGGER.error("Parameter p {} is not resource but it is {}", p, p.getClass());
+ }
+ return p;
+ } catch (DatabaseException e) {
+ throw new RuntimeDatabaseException(e);
+ }
+ }));
+ }
+ });
+ } catch (DatabaseException e) {
+ LOGGER.error("Could not apply mappings", e);
+ throw new RuntimeDatabaseException(e);
+ }
+ }
+ });
+ }
+
+ static class RenderSVGContext {
+
+ Map<String,StringBuilder> documents = new HashMap<>();
+
+ public void append(String[] keys, String svgText) {
+ for(String key : keys) append(key, svgText);
+ }
+
+ public void append(String key, String svgText) {
+ StringBuilder builder = documents.get(key);
+ if(builder == null) {
+ builder = new StringBuilder();
+ documents.put(key, builder);
+ }
+ builder.append(svgText);
+ }
+
+ public void append(RenderSVGContext other) {
+ for(String key : other.documents.keySet()) {
+ append(key, other.get(key));
+ }
+ }
+
+ public String get(String key) {
+ StringBuilder builder = documents.get(key);
+ if(builder == null) return "";
+ else return builder.toString();
+ }
+
+ }
+
+ private static String renderSVG0(ICanvasContext ctx, Function1<Set<?>, Map<?, ?>> mappingFunction) {
// Get a DOMImplementation.
DOMImplementation domImpl = GenericDOMImplementation.getDOMImplementation();
// Create an instance of the SVG Generator.
SVGGraphics2D svgGenerator = new Generator(document);
- StringBuilder result = new StringBuilder();
+ RenderSVGContext result = new RenderSVGContext();
try {
}
G2DSceneGraph sg = ctx.getSceneGraph();
+ sg.performCleanup();
G2DParentNode root = (G2DParentNode) sg.getRootNode();
// rtree is the actual content of the diagram
double trX = -1 * content.getX();
double trY = -1 * content.getY();
+
+ // NaNs
+ if(!Double.isFinite(trX)) trX = 0;
+ if(!Double.isFinite(trY)) trY = 0;
+
+ result.append(MAIN_SECTION, "<g class=\"symbols\">");
+ result.append(SELECTION_SECTION, "<g class=\"selections\">");
+ result.append(SELECTION_MASK_SECTION, "<g class=\"selectionMasks\">");
- result.append("<svg width=\"100%\" height=\"100%\" stroke=\"black\"><g transform=\"translate(").append(trX).append(' ').append(trY).append(")\">");
-
- IG2DNodeVisitor visitor = new IG2DNodeVisitor() {
-
- int indent = 0;
-
- HashMap<SingleElementNode,StringBuilder> senBuilders = new HashMap<>();
-
- @Override
- public void enter(IG2DNode node) {
-
- StringBuilder parentBuilder = getParentBuilder(node);
-
- indent++;
- if(node instanceof ConnectionNode) {
-
- for(RouteGraphNode n : NodeUtil.collectNodes(node, RouteGraphNode.class)) {
- n.setIgnoreSelection(true);
- }
-
- String key = ((ConnectionNode)node).getKey().toString();
-
- parentBuilder.append("\n<g class=\"connection\" id=\"" + key + "\">");
- Element doc = renderSVGNode((IG2DNode)node);
- String svg = printSVGDocument(doc);
- parentBuilder.append(svg);
-
- for(RouteGraphNode n : NodeUtil.collectNodes(node, RouteGraphNode.class)) {
- n.setIgnoreSelection(false);
- }
-
- parentBuilder.append("\n<g style=\"visibility:hidden\" class=\"selection\" id=\"" + key + "\">");
- doc = renderSVGNode((IG2DNode)node);
- svg = printSVGDocument(doc);
- parentBuilder.append(svg);
- parentBuilder.append("\n</g>");
-
- BasicStroke bs = new BasicStroke(5f);
-
- for(RouteGraphNode n : NodeUtil.collectNodes(node, RouteGraphNode.class)) {
- n.setDynamicStroke(bs);
- }
-
- parentBuilder.append("\n<g class=\"selectionMask\" opacity=\"0.001\" id=\"" + key + "\">");
- doc = renderSVGNode((IG2DNode)node);
- svg = printSVGDocument(doc);
- parentBuilder.append(svg);
- parentBuilder.append("\n</g>");
-
- parentBuilder.append("\n</g>");
-
- } else if (node instanceof SelectionNode) {
-
- SelectionNode n = (SelectionNode)node;
- SingleElementNode parentSEN = (SingleElementNode)NodeUtil.getNearestParentOfType(node, SingleElementNode.class);
- if(parentSEN != null && parentSEN.getKey() != null) {
-
- StringBuilder parentBuilder2 = getParentBuilder(parentSEN);
-
- String key = parentSEN.getKey().toString();
- Element doc = renderSVGNode((IG2DNode)node);
- String svg = printSVGDocument(doc);
- parentBuilder2.append("\n<g style=\"visibility:hidden\" class=\"selection\" id=\"" + key + "\">");
- parentBuilder2.append(svg);
- parentBuilder2.append("\n</g>");
- parentBuilder2.append("\n<g class=\"selectionMask\" id=\"" + key + "\">");
- Rectangle2D rect = n.getRect();
- parentBuilder2.append("<rect style=\"fill:#fff\" opacity=\"0.001\"");
- parentBuilder2.append(" x=\"" + rect.getX() + "\" y=\"" + rect.getY() + "\"");
- parentBuilder2.append(" width=\"" + rect.getWidth() + "\" height=\"" + rect.getHeight() + "\"");
- parentBuilder2.append("></rect>");
- parentBuilder2.append("\n</g>");
- }
- } else if (node instanceof SVGNode) {
- SVGNode svg = (SVGNode)node;
- parentBuilder.append(svg.getSVGText());
- } else if (node instanceof G2DParentNode) {
- AffineTransform at = node.getTransform();
- if(node instanceof SingleElementNode) {
- SingleElementNode sen = (SingleElementNode)node;
- if(sen.getKey() != null) {
- String key = sen.getKey().toString();
- parentBuilder.append("\n<g class=\"definedElement\" id=\"" + key + "\">");
- }
- senBuilders.put(sen, new StringBuilder());
- }
- if(!at.isIdentity()) {
- if(at.getScaleX() == 1.0 && at.getScaleY() == 1.0 && at.getShearX() == 0.0 && at.getShearY() == 0.0) {
- String m = "translate(" + at.getTranslateX() + " " + at.getTranslateY() + ")";
- parentBuilder.append("\n<g transform=\"" + m + "\">");
- } else {
- double[] ds = new double[6];
- at.getMatrix(ds);
- String m = "matrix(" + ds[0] + " " + ds[1] + " " + ds[2] + " " + ds[3] + " " + ds[4] + " " + ds[5] + ")";
- parentBuilder.append("\n<g transform=\"" + m + "\">");
- }
- }
- }
-
- //enters.put(node, b.length());
+ result.append(ALL_SECTIONS, "<g transform=\"translate(");
+ result.append(ALL_SECTIONS, "" + trX);
+ result.append(ALL_SECTIONS, ", ");
+ result.append(ALL_SECTIONS, "" + trY);
+ result.append(ALL_SECTIONS, ")\">");
- }
-
- private StringBuilder getParentBuilder(IG2DNode node) {
-
- INode parentSEN = NodeUtil.getNearestParentOfType(node, SingleElementNode.class);
- if(parentSEN instanceof G2DSceneGraph) return result;
-
- StringBuilder parentBuilder = senBuilders.get(parentSEN);
- if(parentBuilder == null) return result;
-
- return parentBuilder;
-
- }
-
- @Override
- public void leave(IG2DNode node) {
-
- if(node instanceof ConnectionNode || node instanceof SVGNode) {
- // We are done
- } else if (node instanceof G2DParentNode) {
-
- StringBuilder parentBuilder = getParentBuilder(node);
-
- if(node instanceof SingleElementNode) {
- SingleElementNode sen = (SingleElementNode)node;
-// if(sen.getKey() != null) {
- StringBuilder b = senBuilders.get(sen);
- String content = b.toString();
- if(content.isEmpty()) {
- if(sen.getKey() != null) {
-
- for(SelectionNode n : NodeUtil.collectNodes(node, SelectionNode.class)) {
- n.setIgnore(true);
- }
-
- Element doc = renderSVGNode((IG2DNode)node);
- String svg = printSVGDocument(doc);
- parentBuilder.append(svg);
- }
- } else {
- parentBuilder.append(content);
- }
-// }
- }
-
-
- AffineTransform at = node.getTransform();
- if(!at.isIdentity()) {
- parentBuilder.append("</g>");
- }
- if(node instanceof SingleElementNode) {
- SingleElementNode sen = (SingleElementNode)node;
- if(sen.getKey() != null) {
- parentBuilder.append("</g>");
- }
- }
- }
- indent --;
- }
+
+ KeyVisitor keyVisitor = new KeyVisitor();
+ sg.accept(keyVisitor);
+
+ Set<Object> keys = keyVisitor.getKeys();
+
+ Map<?, ?> mappings = mappingFunction.apply(keys);
- };
+ IG2DNodeVisitor visitor = new PrintingVisitor(result, mappings);
+
sg.accept(visitor);
} catch (Throwable t) {
LOGGER.error("Problems rendering canvas context to SVG", t);
}
- result.append("</g></svg>");
- //System.err.println(" == FINAL RESULT == ");
- //System.err.println(b);
- return result.toString();
+ result.append(ALL_SECTIONS, "</g></g>");
+
+ StringBuilder res = new StringBuilder();
+ res.append("<svg width=\"100%\" height=\"100%\" stroke=\"black\">");
+ res.append(result.get(MAIN_SECTION));
+ res.append(result.get(SELECTION_SECTION));
+ res.append(result.get(SELECTION_MASK_SECTION));
+ res.append(result.get("</svg>"));
+
+// System.err.println(" == FINAL RESULT == ");
+// System.err.println(res.toString());
+
+ return res.toString();
+
+ }
+
+
+
+ private static class KeyVisitor implements IG2DNodeVisitor {
+
+ private Set<Object> keys = new HashSet<>();
+
+ @Override
+ public void enter(IG2DNode node) {
+ if (node instanceof SingleElementNode) {
+ Object key = ((SingleElementNode) node).getKey();
+ if (key != null) {
+ keys.add(key);
+ }
+ }
+ }
+
+ @Override
+ public void leave(IG2DNode node) {
+ // Nothing to do
+ }
+
+ public Set<Object> getKeys() {
+ return keys;
+ }
}
+ private static class PrintingVisitor implements IG2DNodeVisitor {
+ int indent = 0;
+
+ HashMap<SingleElementNode,RenderSVGContext> senBuilders = new HashMap<>();
+
+ private RenderSVGContext result;
+
+ private Map<?, ?> mappings;
+
+ public PrintingVisitor(RenderSVGContext result, Map<?, ?> mappings) {
+ this.result = result;
+ this.mappings = mappings;
+ }
+
+ private String getKey(SingleElementNode node) {
+ String key;
+ if (mappings.containsKey(node.getKey()))
+ key = mappings.get(node.getKey()).toString();
+ else
+ key = node.getKey().toString();
+ return key;
+ }
+
+ @Override
+ public void enter(IG2DNode node) {
+
+ RenderSVGContext parentBuilder = getParentBuilder(node);
+
+ indent++;
+ if(node instanceof ConnectionNode) {
+
+ for(RouteGraphNode n : NodeUtil.collectNodes(node, RouteGraphNode.class)) {
+ n.setIgnoreSelection(true);
+ }
+
+ String key = getKey((ConnectionNode) node);
+ parentBuilder.append(MAIN_SECTION, "\n<g class=\"connection\" id=\"" + key + "\">");
+ parentBuilder.append(SELECTION_SECTION, "\n<g style=\"visibility:hidden\" class=\"selection\" id=\"" + key + "\">");
+ parentBuilder.append(SELECTION_MASK_SECTION, "\n<g class=\"selectionMask\" opacity=\"0.001\" id=\"" + key + "\">");
+
+ Element doc = renderSVGNode((IG2DNode)node);
+ String svg = printSVGDocument(doc);
+ parentBuilder.append(MAIN_SECTION, svg);
+
+ for(RouteGraphNode n : NodeUtil.collectNodes(node, RouteGraphNode.class)) {
+ n.setIgnoreSelection(false);
+ }
+
+ doc = renderSVGNode((IG2DNode)node);
+ svg = printSVGDocument(doc);
+ parentBuilder.append(SELECTION_SECTION, svg);
+
+ BasicStroke bs = new BasicStroke(10f);
+
+ for(RouteGraphNode n : NodeUtil.collectNodes(node, RouteGraphNode.class)) {
+ n.setDynamicStroke(bs);
+ }
+
+ doc = renderSVGNode((IG2DNode)node);
+ svg = printSVGDocument(doc);
+ parentBuilder.append(SELECTION_MASK_SECTION, svg);
+
+ parentBuilder.append(SELECTION_MASK_SECTION, "\n</g>");
+ parentBuilder.append(SELECTION_SECTION, "\n</g>");
+ parentBuilder.append(MAIN_SECTION, "\n</g>");
+
+ } else if (node instanceof SelectionNode) {
+
+ SelectionNode n = (SelectionNode)node;
+ SingleElementNode parentSEN = (SingleElementNode)NodeUtil.getNearestParentOfType(node, SingleElementNode.class);
+ if(parentSEN != null && parentSEN.getKey() != null) {
+
+ RenderSVGContext parentBuilder2 = getParentBuilder(parentSEN);
+
+ String key = getKey(parentSEN);
+ Element doc = renderSVGNode((IG2DNode)node);
+ String svg = printSVGDocument(doc);
+ parentBuilder2.append(SELECTION_SECTION, "\n<g style=\"visibility:hidden\" class=\"selection\" id=\"" + key + "\">");
+ parentBuilder2.append(SELECTION_SECTION, svg);
+ parentBuilder2.append(SELECTION_SECTION, "\n</g>");
+ parentBuilder2.append(SELECTION_MASK_SECTION, "\n<g class=\"selectionMask\" id=\"" + key + "\">");
+ Rectangle2D rect = n.getRect();
+ // NaN
+ if(rect.getHeight() == rect.getHeight() && rect.getWidth() == rect.getWidth()) {
+ parentBuilder2.append(SELECTION_MASK_SECTION,"<rect style=\"fill:#fff\" opacity=\"0.001\"");
+ parentBuilder2.append(SELECTION_MASK_SECTION," x=\"" + rect.getX() + "\" y=\"" + rect.getY() + "\"");
+ parentBuilder2.append(SELECTION_MASK_SECTION," width=\"" + rect.getWidth() + "\" height=\"" + rect.getHeight() + "\"");
+ parentBuilder2.append(SELECTION_MASK_SECTION,"></rect>");
+ }
+ parentBuilder2.append(SELECTION_MASK_SECTION,"\n</g>");
+ }
+ } else if (node instanceof SVGNode) {
+ SVGNode svg = (SVGNode)node;
+ parentBuilder.append(MAIN_SECTION, svg.getSVGText());
+ } else if (node instanceof G2DParentNode) {
+ AffineTransform at = node.getTransform();
+ if(node instanceof SingleElementNode) {
+ SingleElementNode sen = (SingleElementNode)node;
+ if(sen.getKey() != null) {
+ String key = getKey(sen);
+ String typeClass = sen.getTypeClass();
+ String clazz = "definedElement";
+ if(typeClass != null && !typeClass.isEmpty())
+ clazz = clazz + " " + typeClass;
+
+ parentBuilder.append(MAIN_SECTION, "\n<g class=\""+clazz+"\" id=\"" + key + "\">");
+ }
+ senBuilders.put(sen, new RenderSVGContext());
+ }
+ if(!at.isIdentity()) {
+ if(at.getScaleX() == 1.0 && at.getScaleY() == 1.0 && at.getShearX() == 0.0 && at.getShearY() == 0.0) {
+ String m = "translate(" + at.getTranslateX() + " " + at.getTranslateY() + ")";
+ parentBuilder.append(ALL_SECTIONS, "\n<g transform=\"" + m + "\">");
+ } else {
+ double[] ds = new double[6];
+ at.getMatrix(ds);
+ String m = "matrix(" + ds[0] + " " + ds[1] + " " + ds[2] + " " + ds[3] + " " + ds[4] + " " + ds[5] + ")";
+ parentBuilder.append(ALL_SECTIONS, "\n<g transform=\"" + m + "\">");
+ }
+ }
+ }
+
+ //enters.put(node, b.length());
+
+ }
+
+ private RenderSVGContext getParentBuilder(IG2DNode node) {
+
+ INode parentSEN = NodeUtil.getNearestParentOfType(node, SingleElementNode.class);
+ if(parentSEN instanceof G2DSceneGraph) return result;
+
+ RenderSVGContext parentBuilder = senBuilders.get(parentSEN);
+ if(parentBuilder == null) return result;
+
+ return parentBuilder;
+
+ }
+
+ @Override
+ public void leave(IG2DNode node) {
+
+ if(node instanceof ConnectionNode || node instanceof SVGNode) {
+ // We are done
+ } else if (node instanceof G2DParentNode) {
+
+ RenderSVGContext parentBuilder = getParentBuilder(node);
+
+ if(node instanceof SingleElementNode) {
+ SingleElementNode sen = (SingleElementNode)node;
+ RenderSVGContext b = senBuilders.get(sen);
+ String content = b.get(MAIN_SECTION);
+ if(content.isEmpty()) {
+ if(sen.getKey() != null) {
+
+ for(SelectionNode n : NodeUtil.collectNodes(node, SelectionNode.class)) {
+ n.setIgnore(true);
+ }
+
+ Element doc = renderSVGNode((IG2DNode)node);
+ String svg = printSVGDocument(doc);
+ parentBuilder.append(MAIN_SECTION, svg);
+ }
+ } else {
+ parentBuilder.append(b);
+ }
+ }
+
+ AffineTransform at = node.getTransform();
+ if(!at.isIdentity()) {
+ parentBuilder.append(ALL_SECTIONS, "</g>");
+ }
+ if(node instanceof SingleElementNode) {
+ SingleElementNode sen = (SingleElementNode)node;
+ if(sen.getKey() != null) {
+ parentBuilder.append(MAIN_SECTION, "</g>");
+ }
+ }
+ }
+ indent --;
+ }
+ }
}
\ No newline at end of file
Code Review / simantics / platform.git / blobdiff