1 /*******************************************************************************
2 * Copyright (c) 2007, 2010 Association for Decentralized Information Management
4 * All rights reserved. This program and the accompanying materials
5 * are made available under the terms of the Eclipse Public License v1.0
6 * which accompanies this distribution, and is available at
7 * http://www.eclipse.org/legal/epl-v10.html
10 * VTT Technical Research Centre of Finland - initial API and implementation
11 *******************************************************************************/
12 package org.simantics.scenegraph.g2d.nodes;
14 import java.awt.Color;
15 import java.awt.Composite;
16 import java.awt.Graphics2D;
17 import java.awt.Stroke;
18 import java.awt.geom.Point2D;
20 import org.simantics.diagram.connection.RouteGraph;
21 import org.simantics.scenegraph.INode;
22 import org.simantics.scenegraph.g2d.G2DHints;
23 import org.simantics.scenegraph.g2d.IG2DNode;
24 import org.simantics.scenegraph.g2d.events.MouseEvent;
25 import org.simantics.scenegraph.g2d.nodes.connection.RouteGraphNode;
26 import org.simantics.scenegraph.utils.InitValueSupport;
27 import org.simantics.scenegraph.utils.NodeUtil;
28 import org.simantics.scenegraph.utils.NodeUtil.NodeProcedure;
31 * @author Tuukka Lehtonen
33 public class ConnectionNode extends SingleElementNode implements InitValueSupport {
35 private static final long serialVersionUID = 8657421058765967521L;
37 private static class SetAlphaCompositeProcedure implements NodeProcedure<Object> {
38 public Composite composite;
40 public Object execute(INode node, String id) {
41 if (node instanceof EdgeNode)
42 ((EdgeNode) node).setAlphaComposite(composite);
43 // NodeUtil.setPropertyIfSupported("alpha", alphaComposite, edge);
48 private static class SetDynamicStrokeProcedure implements NodeProcedure<Object> {
51 public Object execute(INode node, String id) {
52 if (node instanceof EdgeNode)
53 ((EdgeNode) node).setDynamicStroke(stroke);
54 else if (node instanceof RouteGraphNode)
55 ((RouteGraphNode) node).setDynamicStroke(stroke);
56 // NodeUtil.setPropertyIfSupported("width", stroke, edge);
61 private static class SetDynamicColorProcedure implements NodeProcedure<Object> {
64 public Object execute(INode node, String id) {
65 if (node instanceof EdgeNode)
66 ((EdgeNode) node).setDynamicColor(color);
67 else if (node instanceof RouteGraphNode)
68 ((RouteGraphNode) node).setDynamicColor(color);
69 // NodeUtil.setPropertyIfSupported("color", color, edge);
74 // private static class SetValueProcedure implements NodeProcedure<Object> {
76 // public Object value;
78 // public Object execute(INode node, String id) {
79 // NodeUtil.setPropertyIfSupported(key, value, node);
80 //// edge.setValue(key, value);
85 private static final SetAlphaCompositeProcedure setAlphaCompositeProcedure = new SetAlphaCompositeProcedure();
86 private static final SetDynamicStrokeProcedure setDynamicStrokeProcedure = new SetDynamicStrokeProcedure();
87 private static final SetDynamicColorProcedure setDynamicColorProcedure = new SetDynamicColorProcedure();
89 private static NodeProcedure<?> initValuesProcedure = new NodeProcedure<Object>() {
91 public Object execute(INode node, String id) {
92 if (node instanceof InitValueSupport)
93 ((InitValueSupport) node).initValues();
98 @PropertySetter("alpha")
99 public void setAlphaComposite(Composite alphaComposite) {
100 setAlphaCompositeProcedure.composite = alphaComposite;
101 forEdges(this, setAlphaCompositeProcedure);
104 @PropertySetter("width")
105 public void setDynamicStroke(Stroke stroke) {
106 setDynamicStrokeProcedure.stroke = stroke;
107 forEdges(this, setDynamicStrokeProcedure);
110 @PropertySetter("color")
111 public void setDynamicColor(Color color) {
112 setDynamicColorProcedure.color = color;
113 forEdges(this, setDynamicColorProcedure);
117 public void initValues() {
118 NodeUtil.forChildren(this, initValuesProcedure, null);
121 private void forEdges(SingleElementNode node, NodeProcedure<?> procedure) {
122 for (String childId : node.getSortedNodesById()) {
123 IG2DNode child = node.getNode(childId);
124 if (child instanceof SingleElementNode) {
125 forEdges((SingleElementNode) child, procedure);
126 } else if (child instanceof EdgeNode) {
127 procedure.execute(child, childId);
128 } else if (child instanceof RouteGraphNode) {
129 procedure.execute(child, childId);
135 protected boolean hitTest(MouseEvent event) {
136 if (!super.hitTest(event))
139 Point2D pos = NodeUtil.worldToLocal(this, event.controlPosition, new Point2D.Double());
140 double tolerance = 0.5;
141 // TODO: change tolerance based on zoom level
143 for (IG2DNode child : getSortedNodes()) {
144 if (child instanceof RouteGraphNode) {
145 RouteGraphNode rgn = (RouteGraphNode) child;
146 RouteGraph rg = rgn.getRouteGraph();
147 Object pick = rg.pickLine(pos.getX(), pos.getY(), tolerance);
155 public void beforeRender(Graphics2D g) {
156 g.setRenderingHint(G2DHints.KEY_BEGIN_ELEMENT, "connection");
159 public void afterRender(Graphics2D g) {
160 g.setRenderingHint(G2DHints.KEY_END_ELEMENT, "connection");