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.Graphics2D;
  17 import java.awt.Stroke;
  18 import java.awt.geom.Point2D;
  19
  20 import org.simantics.diagram.connection.RouteGraph;
  21 import org.simantics.scenegraph.INode;
  22 import org.simantics.scenegraph.g2d.G2DRenderingHints;
  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;
  29
  30 /**
  31  * @author Tuukka Lehtonen
  32  */
  33 public class ConnectionNode extends SingleElementNode implements InitValueSupport {
  34
  35     private static final long serialVersionUID = 8657421058765967521L;
  36
  37     private static class SetAlphaCompositeProcedure implements NodeProcedure<Object> {
  38         public Composite composite;
  39         @Override
  40         public Object execute(INode node, String id) {
  41             if (node instanceof EdgeNode)
  42                 ((EdgeNode) node).setAlphaComposite(composite);
  43 //          NodeUtil.setPropertyIfSupported("alpha", alphaComposite, edge);
  44             return null;
  45         }
  46     };
  47
  48     private static class SetDynamicStrokeProcedure implements NodeProcedure<Object> {
  49         public Stroke stroke;
  50         @Override
  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);
  57             return null;
  58         }
  59     };
  60
  61     private static class SetDynamicColorProcedure implements NodeProcedure<Object> {
  62         public Color color;
  63         @Override
  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);
  70             return null;
  71         }
  72     };
  73
  74 //    private static class SetValueProcedure implements NodeProcedure<Object> {
  75 //        public String key;
  76 //        public Object value;
  77 //        @Override
  78 //        public Object execute(INode node, String id) {
  79 //            NodeUtil.setPropertyIfSupported(key, value, node);
  80 ////            edge.setValue(key, value);
  81 //            return null;
  82 //        }
  83 //    };
  84
  85     private static final SetAlphaCompositeProcedure setAlphaCompositeProcedure = new SetAlphaCompositeProcedure();
  86     private static final SetDynamicStrokeProcedure  setDynamicStrokeProcedure  = new SetDynamicStrokeProcedure();
  87     private static final SetDynamicColorProcedure   setDynamicColorProcedure   = new SetDynamicColorProcedure();
  88
  89     private static NodeProcedure<?> initValuesProcedure = new NodeProcedure<Object>() {
  90         @Override
  91         public Object execute(INode node, String id) {
  92             if (node instanceof InitValueSupport)
  93                 ((InitValueSupport) node).initValues();
  94             return null;
  95         }
  96     };
  97
  98     @PropertySetter("alpha")
  99     public void setAlphaComposite(Composite alphaComposite) {
 100         setAlphaCompositeProcedure.composite = alphaComposite;
 101         forEdges(this, setAlphaCompositeProcedure);
 102     }
 103
 104     @PropertySetter("width")
 105     public void setDynamicStroke(Stroke stroke) {
 106         setDynamicStrokeProcedure.stroke = stroke;
 107         forEdges(this, setDynamicStrokeProcedure);
 108     }
 109
 110     @PropertySetter("color")
 111     public void setDynamicColor(Color color) {
 112         setDynamicColorProcedure.color = color;
 113         forEdges(this, setDynamicColorProcedure);
 114     }
 115
 116     @Override
 117     public void initValues() {
 118         NodeUtil.forChildren(this, initValuesProcedure, null);
 119     }
 120
 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);
 130             }
 131         }
 132     }
 133
 134     @Override
 135     protected boolean hitTest(MouseEvent event) {
 136         if (!super.hitTest(event))
 137             return false;
 138
 139         Point2D pos = NodeUtil.worldToLocal(this, event.controlPosition, new Point2D.Double());
 140         double tolerance = 0.5;
 141         // TODO: change tolerance based on zoom level
 142
 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);
 148                 if (pick != null)
 149                     return true;
 150             }
 151         }
 152         return false;
 153     }
 154
 155     @Override
 156     public void beforeRender(Graphics2D g) {
 157         g.setRenderingHint(G2DRenderingHints.KEY_BEGIN_ELEMENT, "connection");
 158     }
 159
 160     @Override
 161     public void afterRender(Graphics2D g) {
 162         g.setRenderingHint(G2DRenderingHints.KEY_END_ELEMENT, "connection");
 163     }
 164
 165 }