bundles/org.simantics.scenegraph/src/org/simantics/scenegraph/g2d/nodes/ConnectionNode.java

   1 /*******************************************************************************
   2  * Copyright (c) 2007, 2010 Association for Decentralized Information Management
   3  * in Industry THTH ry.
   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
   8  *
   9  * Contributors:
  10  *     VTT Technical Research Centre of Finland - initial API and implementation
  11  *******************************************************************************/
  12 package org.simantics.scenegraph.g2d.nodes;
  13
  14 import java.awt.Color;
  15 import java.awt.Composite;
  16 import java.awt.Stroke;
  17 import java.awt.geom.Point2D;
  18
  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;
  27
  28 /**
  29  * @author Tuukka Lehtonen
  30  */
  31 public class ConnectionNode extends SingleElementNode implements InitValueSupport {
  32
  33     private static final long serialVersionUID = 8657421058765967521L;
  34
  35     private static class SetAlphaCompositeProcedure implements NodeProcedure<Object> {
  36         public Composite composite;
  37         @Override
  38         public Object execute(INode node, String id) {
  39             if (node instanceof EdgeNode)
  40                 ((EdgeNode) node).setAlphaComposite(composite);
  41 //          NodeUtil.setPropertyIfSupported("alpha", alphaComposite, edge);
  42             return null;
  43         }
  44     };
  45
  46     private static class SetDynamicStrokeProcedure implements NodeProcedure<Object> {
  47         public Stroke stroke;
  48         @Override
  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);
  55             return null;
  56         }
  57     };
  58
  59     private static class SetDynamicColorProcedure implements NodeProcedure<Object> {
  60         public Color color;
  61         @Override
  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);
  68             return null;
  69         }
  70     };
  71
  72 //    private static class SetValueProcedure implements NodeProcedure<Object> {
  73 //        public String key;
  74 //        public Object value;
  75 //        @Override
  76 //        public Object execute(INode node, String id) {
  77 //            NodeUtil.setPropertyIfSupported(key, value, node);
  78 ////            edge.setValue(key, value);
  79 //            return null;
  80 //        }
  81 //    };
  82
  83     private static final SetAlphaCompositeProcedure setAlphaCompositeProcedure = new SetAlphaCompositeProcedure();
  84     private static final SetDynamicStrokeProcedure  setDynamicStrokeProcedure  = new SetDynamicStrokeProcedure();
  85     private static final SetDynamicColorProcedure   setDynamicColorProcedure   = new SetDynamicColorProcedure();
  86
  87     private static NodeProcedure<?> initValuesProcedure = new NodeProcedure<Object>() {
  88         @Override
  89         public Object execute(INode node, String id) {
  90             if (node instanceof InitValueSupport)
  91                 ((InitValueSupport) node).initValues();
  92             return null;
  93         }
  94     };
  95
  96     @PropertySetter("alpha")
  97     public void setAlphaComposite(Composite alphaComposite) {
  98         setAlphaCompositeProcedure.composite = alphaComposite;
  99         forEdges(this, setAlphaCompositeProcedure);
 100     }
 101
 102     @PropertySetter("width")
 103     public void setDynamicStroke(Stroke stroke) {
 104         setDynamicStrokeProcedure.stroke = stroke;
 105         forEdges(this, setDynamicStrokeProcedure);
 106     }
 107
 108     @PropertySetter("color")
 109     public void setDynamicColor(Color color) {
 110         setDynamicColorProcedure.color = color;
 111         forEdges(this, setDynamicColorProcedure);
 112     }
 113
 114     @Override
 115     public void initValues() {
 116         NodeUtil.forChildren(this, initValuesProcedure, null);
 117     }
 118
 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);
 128             }
 129         }
 130     }
 131
 132     @Override
 133     protected boolean hitTest(MouseEvent event) {
 134         if (!super.hitTest(event))
 135             return false;
 136
 137         Point2D pos = NodeUtil.worldToLocal(this, event.controlPosition, new Point2D.Double());
 138         double tolerance = 0.5;
 139         // TODO: change tolerance based on zoom level
 140
 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);
 146                 if (pick != null)
 147                     return true;
 148             }
 149         }
 150         return false;
 151     }
 152
 153 }