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.Stroke;
17 import java.awt.geom.Point2D;
19 import org.simantics.diagram.connection.RouteGraph;
20 import org.simantics.scenegraph.INode;
21 import org.simantics.scenegraph.g2d.IG2DNode;
22 import org.simantics.scenegraph.g2d.events.MouseEvent;
23 import org.simantics.scenegraph.g2d.nodes.connection.RouteGraphNode;
24 import org.simantics.scenegraph.utils.InitValueSupport;
25 import org.simantics.scenegraph.utils.NodeUtil;
26 import org.simantics.scenegraph.utils.NodeUtil.NodeProcedure;
29 * @author Tuukka Lehtonen
31 public class ConnectionNode extends SingleElementNode implements InitValueSupport {
33 private static final long serialVersionUID = 8657421058765967521L;
35 private static class SetAlphaCompositeProcedure implements NodeProcedure<Object> {
36 public Composite composite;
38 public Object execute(INode node, String id) {
39 if (node instanceof EdgeNode)
40 ((EdgeNode) node).setAlphaComposite(composite);
41 // NodeUtil.setPropertyIfSupported("alpha", alphaComposite, edge);
46 private static class SetDynamicStrokeProcedure implements NodeProcedure<Object> {
49 public Object execute(INode node, String id) {
50 if (node instanceof EdgeNode)
51 ((EdgeNode) node).setDynamicStroke(stroke);
52 else if (node instanceof RouteGraphNode)
53 ((RouteGraphNode) node).setDynamicStroke(stroke);
54 // NodeUtil.setPropertyIfSupported("width", stroke, edge);
59 private static class SetDynamicColorProcedure implements NodeProcedure<Object> {
62 public Object execute(INode node, String id) {
63 if (node instanceof EdgeNode)
64 ((EdgeNode) node).setDynamicColor(color);
65 else if (node instanceof RouteGraphNode)
66 ((RouteGraphNode) node).setDynamicColor(color);
67 // NodeUtil.setPropertyIfSupported("color", color, edge);
72 // private static class SetValueProcedure implements NodeProcedure<Object> {
74 // public Object value;
76 // public Object execute(INode node, String id) {
77 // NodeUtil.setPropertyIfSupported(key, value, node);
78 //// edge.setValue(key, value);
83 private static final SetAlphaCompositeProcedure setAlphaCompositeProcedure = new SetAlphaCompositeProcedure();
84 private static final SetDynamicStrokeProcedure setDynamicStrokeProcedure = new SetDynamicStrokeProcedure();
85 private static final SetDynamicColorProcedure setDynamicColorProcedure = new SetDynamicColorProcedure();
87 private static NodeProcedure<?> initValuesProcedure = new NodeProcedure<Object>() {
89 public Object execute(INode node, String id) {
90 if (node instanceof InitValueSupport)
91 ((InitValueSupport) node).initValues();
96 @PropertySetter("alpha")
97 public void setAlphaComposite(Composite alphaComposite) {
98 setAlphaCompositeProcedure.composite = alphaComposite;
99 forEdges(this, setAlphaCompositeProcedure);
102 @PropertySetter("width")
103 public void setDynamicStroke(Stroke stroke) {
104 setDynamicStrokeProcedure.stroke = stroke;
105 forEdges(this, setDynamicStrokeProcedure);
108 @PropertySetter("color")
109 public void setDynamicColor(Color color) {
110 setDynamicColorProcedure.color = color;
111 forEdges(this, setDynamicColorProcedure);
115 public void initValues() {
116 NodeUtil.forChildren(this, initValuesProcedure, null);
119 private void forEdges(SingleElementNode node, NodeProcedure<?> procedure) {
120 for (String childId : node.getSortedNodesById()) {
121 IG2DNode child = node.getNode(childId);
122 if (child instanceof SingleElementNode) {
123 forEdges((SingleElementNode) child, procedure);
124 } else if (child instanceof EdgeNode) {
125 procedure.execute(child, childId);
126 } else if (child instanceof RouteGraphNode) {
127 procedure.execute(child, childId);
133 protected boolean hitTest(MouseEvent event) {
134 if (!super.hitTest(event))
137 Point2D pos = NodeUtil.worldToLocal(this, event.controlPosition, new Point2D.Double());
138 double tolerance = 0.5;
139 // TODO: change tolerance based on zoom level
141 for (IG2DNode child : getSortedNodes()) {
142 if (child instanceof RouteGraphNode) {
143 RouteGraphNode rgn = (RouteGraphNode) child;
144 RouteGraph rg = rgn.getRouteGraph();
145 Object pick = rg.pickLine(pos.getX(), pos.getY(), tolerance);