]> gerrit.simantics Code Review - simantics/platform.git/blobdiff - bundles/org.simantics.scenegraph/src/org/simantics/scenegraph/g2d/nodes/ConnectionNode.java
Support SVG generation from scenegraph
[simantics/platform.git] / bundles / org.simantics.scenegraph / src / org / simantics / scenegraph / g2d / nodes / ConnectionNode.java
index 128b0df90f931d871846dc71faf42d32cb57de2f..d4543c6d110546c08cf8709f94f3af81d9501810 100644 (file)
-/*******************************************************************************\r
- * Copyright (c) 2007, 2010 Association for Decentralized Information Management\r
- * in Industry THTH ry.\r
- * All rights reserved. This program and the accompanying materials\r
- * are made available under the terms of the Eclipse Public License v1.0\r
- * which accompanies this distribution, and is available at\r
- * http://www.eclipse.org/legal/epl-v10.html\r
- *\r
- * Contributors:\r
- *     VTT Technical Research Centre of Finland - initial API and implementation\r
- *******************************************************************************/\r
-package org.simantics.scenegraph.g2d.nodes;\r
-\r
-import java.awt.Color;\r
-import java.awt.Composite;\r
-import java.awt.Stroke;\r
-\r
-import org.simantics.scenegraph.INode;\r
-import org.simantics.scenegraph.g2d.IG2DNode;\r
-import org.simantics.scenegraph.g2d.nodes.connection.RouteGraphNode;\r
-import org.simantics.scenegraph.utils.InitValueSupport;\r
-import org.simantics.scenegraph.utils.NodeUtil;\r
-import org.simantics.scenegraph.utils.NodeUtil.NodeProcedure;\r
-\r
-/**\r
- * @author Tuukka Lehtonen\r
- */\r
-public class ConnectionNode extends SingleElementNode implements InitValueSupport {\r
-\r
-    private static final long serialVersionUID = 8657421058765967521L;\r
-\r
-    private static class SetAlphaCompositeProcedure implements NodeProcedure<Object> {\r
-        public Composite composite;\r
-        @Override\r
-        public Object execute(INode node, String id) {\r
-            if (node instanceof EdgeNode)\r
-                ((EdgeNode) node).setAlphaComposite(composite);\r
-//          NodeUtil.setPropertyIfSupported("alpha", alphaComposite, edge);\r
-            return null;\r
-        }\r
-    };\r
-\r
-    private static class SetDynamicStrokeProcedure implements NodeProcedure<Object> {\r
-        public Stroke stroke;\r
-        @Override\r
-        public Object execute(INode node, String id) {\r
-            if (node instanceof EdgeNode)\r
-                ((EdgeNode) node).setDynamicStroke(stroke);\r
-            else if (node instanceof RouteGraphNode)\r
-                ((RouteGraphNode) node).setDynamicStroke(stroke);\r
-//          NodeUtil.setPropertyIfSupported("width", stroke, edge);\r
-            return null;\r
-        }\r
-    };\r
-\r
-    private static class SetDynamicColorProcedure implements NodeProcedure<Object> {\r
-        public Color color;\r
-        @Override\r
-        public Object execute(INode node, String id) {\r
-            if (node instanceof EdgeNode)\r
-                ((EdgeNode) node).setDynamicColor(color);\r
-            else if (node instanceof RouteGraphNode)\r
-                ((RouteGraphNode) node).setDynamicColor(color);\r
-//          NodeUtil.setPropertyIfSupported("color", color, edge);\r
-            return null;\r
-        }\r
-    };\r
-\r
-//    private static class SetValueProcedure implements NodeProcedure<Object> {\r
-//        public String key;\r
-//        public Object value;\r
-//        @Override\r
-//        public Object execute(INode node, String id) {\r
-//            NodeUtil.setPropertyIfSupported(key, value, node);\r
-////            edge.setValue(key, value);\r
-//            return null;\r
-//        }\r
-//    };\r
-\r
-    private static final SetAlphaCompositeProcedure setAlphaCompositeProcedure = new SetAlphaCompositeProcedure();\r
-    private static final SetDynamicStrokeProcedure  setDynamicStrokeProcedure  = new SetDynamicStrokeProcedure();\r
-    private static final SetDynamicColorProcedure   setDynamicColorProcedure   = new SetDynamicColorProcedure();\r
-\r
-    private static NodeProcedure<?> initValuesProcedure = new NodeProcedure<Object>() {\r
-        @Override\r
-        public Object execute(INode node, String id) {\r
-            if (node instanceof InitValueSupport)\r
-                ((InitValueSupport) node).initValues();\r
-            return null;\r
-        }\r
-    };\r
-\r
-    @PropertySetter("alpha")\r
-    public void setAlphaComposite(Composite alphaComposite) {\r
-        setAlphaCompositeProcedure.composite = alphaComposite;\r
-        forEdges(this, setAlphaCompositeProcedure);\r
-    }\r
-    \r
-    @PropertySetter("width")\r
-    public void setDynamicStroke(Stroke stroke) {\r
-        setDynamicStrokeProcedure.stroke = stroke;\r
-        forEdges(this, setDynamicStrokeProcedure);\r
-    }\r
-\r
-    @PropertySetter("color")\r
-    public void setDynamicColor(Color color) {\r
-        setDynamicColorProcedure.color = color;\r
-        forEdges(this, setDynamicColorProcedure);\r
-    }\r
-\r
-    @Override\r
-    public void initValues() {\r
-        NodeUtil.forChildren(this, initValuesProcedure, null);\r
-    }\r
-\r
-    private void forEdges(SingleElementNode node, NodeProcedure<?> procedure) {\r
-        for (String childId : node.getSortedNodesById()) {\r
-            IG2DNode child = node.getNode(childId);\r
-            if (child instanceof SingleElementNode) {\r
-                forEdges((SingleElementNode) child, procedure);\r
-            } else if (child instanceof EdgeNode) {\r
-                procedure.execute(child, childId);\r
-            } else if (child instanceof RouteGraphNode) {\r
-                procedure.execute(child, childId);\r
-            }\r
-        }\r
-    }\r
-\r
-}\r
+/*******************************************************************************
+ * Copyright (c) 2007, 2010 Association for Decentralized Information Management
+ * in Industry THTH ry.
+ * All rights reserved. This program and the accompanying materials
+ * are made available under the terms of the Eclipse Public License v1.0
+ * which accompanies this distribution, and is available at
+ * http://www.eclipse.org/legal/epl-v10.html
+ *
+ * Contributors:
+ *     VTT Technical Research Centre of Finland - initial API and implementation
+ *******************************************************************************/
+package org.simantics.scenegraph.g2d.nodes;
+
+import java.awt.Color;
+import java.awt.Composite;
+import java.awt.Graphics2D;
+import java.awt.Stroke;
+import java.awt.geom.Point2D;
+
+import org.simantics.diagram.connection.RouteGraph;
+import org.simantics.scenegraph.INode;
+import org.simantics.scenegraph.g2d.G2DRenderingHints;
+import org.simantics.scenegraph.g2d.IG2DNode;
+import org.simantics.scenegraph.g2d.events.MouseEvent;
+import org.simantics.scenegraph.g2d.nodes.connection.RouteGraphNode;
+import org.simantics.scenegraph.utils.InitValueSupport;
+import org.simantics.scenegraph.utils.NodeUtil;
+import org.simantics.scenegraph.utils.NodeUtil.NodeProcedure;
+
+/**
+ * @author Tuukka Lehtonen
+ */
+public class ConnectionNode extends SingleElementNode implements InitValueSupport {
+
+    private static final long serialVersionUID = 8657421058765967521L;
+
+    private static class SetAlphaCompositeProcedure implements NodeProcedure<Object> {
+        public Composite composite;
+        @Override
+        public Object execute(INode node, String id) {
+            if (node instanceof EdgeNode)
+                ((EdgeNode) node).setAlphaComposite(composite);
+//          NodeUtil.setPropertyIfSupported("alpha", alphaComposite, edge);
+            return null;
+        }
+    };
+
+    private static class SetDynamicStrokeProcedure implements NodeProcedure<Object> {
+        public Stroke stroke;
+        @Override
+        public Object execute(INode node, String id) {
+            if (node instanceof EdgeNode)
+                ((EdgeNode) node).setDynamicStroke(stroke);
+            else if (node instanceof RouteGraphNode)
+                ((RouteGraphNode) node).setDynamicStroke(stroke);
+//          NodeUtil.setPropertyIfSupported("width", stroke, edge);
+            return null;
+        }
+    };
+
+    private static class SetDynamicColorProcedure implements NodeProcedure<Object> {
+        public Color color;
+        @Override
+        public Object execute(INode node, String id) {
+            if (node instanceof EdgeNode)
+                ((EdgeNode) node).setDynamicColor(color);
+            else if (node instanceof RouteGraphNode)
+                ((RouteGraphNode) node).setDynamicColor(color);
+//          NodeUtil.setPropertyIfSupported("color", color, edge);
+            return null;
+        }
+    };
+
+//    private static class SetValueProcedure implements NodeProcedure<Object> {
+//        public String key;
+//        public Object value;
+//        @Override
+//        public Object execute(INode node, String id) {
+//            NodeUtil.setPropertyIfSupported(key, value, node);
+////            edge.setValue(key, value);
+//            return null;
+//        }
+//    };
+
+    private static final SetAlphaCompositeProcedure setAlphaCompositeProcedure = new SetAlphaCompositeProcedure();
+    private static final SetDynamicStrokeProcedure  setDynamicStrokeProcedure  = new SetDynamicStrokeProcedure();
+    private static final SetDynamicColorProcedure   setDynamicColorProcedure   = new SetDynamicColorProcedure();
+
+    private static NodeProcedure<?> initValuesProcedure = new NodeProcedure<Object>() {
+        @Override
+        public Object execute(INode node, String id) {
+            if (node instanceof InitValueSupport)
+                ((InitValueSupport) node).initValues();
+            return null;
+        }
+    };
+
+    @PropertySetter("alpha")
+    public void setAlphaComposite(Composite alphaComposite) {
+        setAlphaCompositeProcedure.composite = alphaComposite;
+        forEdges(this, setAlphaCompositeProcedure);
+    }
+    
+    @PropertySetter("width")
+    public void setDynamicStroke(Stroke stroke) {
+        setDynamicStrokeProcedure.stroke = stroke;
+        forEdges(this, setDynamicStrokeProcedure);
+    }
+
+    @PropertySetter("color")
+    public void setDynamicColor(Color color) {
+        setDynamicColorProcedure.color = color;
+        forEdges(this, setDynamicColorProcedure);
+    }
+
+    @Override
+    public void initValues() {
+        NodeUtil.forChildren(this, initValuesProcedure, null);
+    }
+
+    private void forEdges(SingleElementNode node, NodeProcedure<?> procedure) {
+        for (String childId : node.getSortedNodesById()) {
+            IG2DNode child = node.getNode(childId);
+            if (child instanceof SingleElementNode) {
+                forEdges((SingleElementNode) child, procedure);
+            } else if (child instanceof EdgeNode) {
+                procedure.execute(child, childId);
+            } else if (child instanceof RouteGraphNode) {
+                procedure.execute(child, childId);
+            }
+        }
+    }
+
+    @Override
+    protected boolean hitTest(MouseEvent event) {
+        if (!super.hitTest(event))
+            return false;
+
+        Point2D pos = NodeUtil.worldToLocal(this, event.controlPosition, new Point2D.Double());
+        double tolerance = 0.5;
+        // TODO: change tolerance based on zoom level
+
+        for (IG2DNode child : getSortedNodes()) {
+            if (child instanceof RouteGraphNode) {
+                RouteGraphNode rgn = (RouteGraphNode) child;
+                RouteGraph rg = rgn.getRouteGraph();
+                Object pick = rg.pickLine(pos.getX(), pos.getY(), tolerance);
+                if (pick != null)
+                    return true;
+            }
+        }
+        return false;
+    }
+
+    @Override
+    public void beforeRender(Graphics2D g) {
+        g.setRenderingHint(G2DRenderingHints.KEY_BEGIN_ELEMENT, "connection");
+    }
+    
+    @Override
+    public void afterRender(Graphics2D g) {
+        g.setRenderingHint(G2DRenderingHints.KEY_END_ELEMENT, "connection");
+    }
+    
+}