bundles/org.simantics.g2d/src/org/simantics/g2d/element/handler/impl/Terminals.java

   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
   8  *\r
   9  * Contributors:\r
  10  *     VTT Technical Research Centre of Finland - initial API and implementation\r
  11  *******************************************************************************/\r
  12 package org.simantics.g2d.element.handler.impl;\r
  13 \r
  14 import gnu.trove.map.hash.THashMap;\r
  15 \r
  16 import java.awt.Shape;\r
  17 import java.awt.geom.AffineTransform;\r
  18 import java.util.ArrayList;\r
  19 import java.util.Collection;\r
  20 import java.util.Map;\r
  21 \r
  22 import org.simantics.g2d.diagram.handler.Topology.Terminal;\r
  23 import org.simantics.g2d.element.IElement;\r
  24 import org.simantics.g2d.element.handler.TerminalLayout;\r
  25 import org.simantics.g2d.element.handler.TerminalTopology;\r
  26 import org.simantics.g2d.utils.geom.DirectionSet;\r
  27 \r
  28 /**\r
  29  * @author Tuukka Lehtonen\r
  30  */\r
  31 public class Terminals implements TerminalLayout, TerminalTopology {\r
  32 \r
  33     private static final long             serialVersionUID = -6532093690907028016L;\r
  34 \r
  35     private final Map<Terminal, ObjectTerminal> terminalMap      = new THashMap<Terminal, ObjectTerminal>();\r
  36     private final ArrayList<Terminal>           terminals        = new ArrayList<Terminal>();\r
  37 \r
  38     public Terminals(Collection<ObjectTerminal> ts) {\r
  39         for (ObjectTerminal ti : ts) {\r
  40             terminals.add(ti);\r
  41             terminalMap.put(ti, ti);\r
  42         }\r
  43     }\r
  44 \r
  45     @Override\r
  46     public AffineTransform getTerminalPosition(IElement node, Terminal t) {\r
  47         ObjectTerminal ti = terminalMap.get(t);\r
  48         if (ti == null)\r
  49             return null;\r
  50         return new AffineTransform(ti.getTransform());\r
  51     }\r
  52 \r
  53     @Override\r
  54     public void getTerminals(IElement e, Collection<Terminal> result) {\r
  55         result.addAll(terminals);\r
  56     }\r
  57 \r
  58     @Override\r
  59     public boolean getTerminalDirection(IElement node, Terminal t, DirectionSet directions) {\r
  60         ObjectTerminal ti = terminalMap.get(t);\r
  61         if (ti == null)\r
  62             return false;\r
  63         directions.addAll(ti.getDirections());\r
  64         return true;\r
  65     }\r
  66 \r
  67     @Override\r
  68     public Shape getTerminalShape(IElement node, Terminal t) {\r
  69         ObjectTerminal ti = terminalMap.get(t);\r
  70         if (ti == null)\r
  71             return null;\r
  72         return ti.getShape();\r
  73     }\r
  74 \r
  75     @Override\r
  76     public int hashCode() {\r
  77         return terminalMap.hashCode();\r
  78     }\r
  79 \r
  80     @Override\r
  81     public boolean equals(Object obj) {\r
  82         if (this == obj)\r
  83             return true;\r
  84         if (obj == null)\r
  85             return false;\r
  86         if (getClass() != obj.getClass())\r
  87             return false;\r
  88         Terminals other = (Terminals) obj;\r
  89         return terminalMap.equals(other.terminalMap);\r
  90     }\r
  91 \r
  92 }