1 /*******************************************************************************
\r
2 * Copyright (c) 2007, 2010 Association for Decentralized Information Management
\r
3 * in Industry THTH ry.
\r
4 * All rights reserved. This program and the accompanying materials
\r
5 * are made available under the terms of the Eclipse Public License v1.0
\r
6 * which accompanies this distribution, and is available at
\r
7 * http://www.eclipse.org/legal/epl-v10.html
\r
10 * VTT Technical Research Centre of Finland - initial API and implementation
\r
11 *******************************************************************************/
\r
12 package org.simantics.scenegraph.g2d.nodes;
\r
14 import java.awt.Color;
\r
15 import java.awt.Composite;
\r
16 import java.awt.Stroke;
\r
17 import java.awt.geom.Point2D;
\r
19 import org.simantics.diagram.connection.RouteGraph;
\r
20 import org.simantics.scenegraph.INode;
\r
21 import org.simantics.scenegraph.g2d.IG2DNode;
\r
22 import org.simantics.scenegraph.g2d.events.MouseEvent;
\r
23 import org.simantics.scenegraph.g2d.nodes.connection.RouteGraphNode;
\r
24 import org.simantics.scenegraph.utils.InitValueSupport;
\r
25 import org.simantics.scenegraph.utils.NodeUtil;
\r
26 import org.simantics.scenegraph.utils.NodeUtil.NodeProcedure;
\r
29 * @author Tuukka Lehtonen
\r
31 public class ConnectionNode extends SingleElementNode implements InitValueSupport {
\r
33 private static final long serialVersionUID = 8657421058765967521L;
\r
35 private static class SetAlphaCompositeProcedure implements NodeProcedure<Object> {
\r
36 public Composite composite;
\r
38 public Object execute(INode node, String id) {
\r
39 if (node instanceof EdgeNode)
\r
40 ((EdgeNode) node).setAlphaComposite(composite);
\r
41 // NodeUtil.setPropertyIfSupported("alpha", alphaComposite, edge);
\r
46 private static class SetDynamicStrokeProcedure implements NodeProcedure<Object> {
\r
47 public Stroke stroke;
\r
49 public Object execute(INode node, String id) {
\r
50 if (node instanceof EdgeNode)
\r
51 ((EdgeNode) node).setDynamicStroke(stroke);
\r
52 else if (node instanceof RouteGraphNode)
\r
53 ((RouteGraphNode) node).setDynamicStroke(stroke);
\r
54 // NodeUtil.setPropertyIfSupported("width", stroke, edge);
\r
59 private static class SetDynamicColorProcedure implements NodeProcedure<Object> {
\r
62 public Object execute(INode node, String id) {
\r
63 if (node instanceof EdgeNode)
\r
64 ((EdgeNode) node).setDynamicColor(color);
\r
65 else if (node instanceof RouteGraphNode)
\r
66 ((RouteGraphNode) node).setDynamicColor(color);
\r
67 // NodeUtil.setPropertyIfSupported("color", color, edge);
\r
72 // private static class SetValueProcedure implements NodeProcedure<Object> {
\r
73 // public String key;
\r
74 // public Object value;
\r
76 // public Object execute(INode node, String id) {
\r
77 // NodeUtil.setPropertyIfSupported(key, value, node);
\r
78 //// edge.setValue(key, value);
\r
83 private static final SetAlphaCompositeProcedure setAlphaCompositeProcedure = new SetAlphaCompositeProcedure();
\r
84 private static final SetDynamicStrokeProcedure setDynamicStrokeProcedure = new SetDynamicStrokeProcedure();
\r
85 private static final SetDynamicColorProcedure setDynamicColorProcedure = new SetDynamicColorProcedure();
\r
87 private static NodeProcedure<?> initValuesProcedure = new NodeProcedure<Object>() {
\r
89 public Object execute(INode node, String id) {
\r
90 if (node instanceof InitValueSupport)
\r
91 ((InitValueSupport) node).initValues();
\r
96 @PropertySetter("alpha")
\r
97 public void setAlphaComposite(Composite alphaComposite) {
\r
98 setAlphaCompositeProcedure.composite = alphaComposite;
\r
99 forEdges(this, setAlphaCompositeProcedure);
\r
102 @PropertySetter("width")
\r
103 public void setDynamicStroke(Stroke stroke) {
\r
104 setDynamicStrokeProcedure.stroke = stroke;
\r
105 forEdges(this, setDynamicStrokeProcedure);
\r
108 @PropertySetter("color")
\r
109 public void setDynamicColor(Color color) {
\r
110 setDynamicColorProcedure.color = color;
\r
111 forEdges(this, setDynamicColorProcedure);
\r
115 public void initValues() {
\r
116 NodeUtil.forChildren(this, initValuesProcedure, null);
\r
119 private void forEdges(SingleElementNode node, NodeProcedure<?> procedure) {
\r
120 for (String childId : node.getSortedNodesById()) {
\r
121 IG2DNode child = node.getNode(childId);
\r
122 if (child instanceof SingleElementNode) {
\r
123 forEdges((SingleElementNode) child, procedure);
\r
124 } else if (child instanceof EdgeNode) {
\r
125 procedure.execute(child, childId);
\r
126 } else if (child instanceof RouteGraphNode) {
\r
127 procedure.execute(child, childId);
\r
133 protected boolean hitTest(MouseEvent event) {
\r
134 if (!super.hitTest(event))
\r
137 Point2D pos = NodeUtil.worldToLocal(this, event.controlPosition, new Point2D.Double());
\r
138 double tolerance = 0.5;
\r
139 // TODO: change tolerance based on zoom level
\r
141 for (IG2DNode child : getSortedNodes()) {
\r
142 if (child instanceof RouteGraphNode) {
\r
143 RouteGraphNode rgn = (RouteGraphNode) child;
\r
144 RouteGraph rg = rgn.getRouteGraph();
\r
145 Object pick = rg.pickLine(pos.getX(), pos.getY(), tolerance);
\r